Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(201)

Unified Diff: webrtc/test/fake_encoder.h

Issue 2686103002: Do not encode frames in MultithreadedFakeH264Encoder after Release(). (Closed)
Patch Set: Rebase. Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « no previous file | webrtc/test/fake_encoder.cc » ('j') | webrtc/test/fake_encoder.cc » ('J')
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/test/fake_encoder.h
diff --git a/webrtc/test/fake_encoder.h b/webrtc/test/fake_encoder.h
index 39c3e3e67f599d50eb9ed404aa6634e3c75d480c..c07fbb36e1c49088e343d0f048b90359e0f651f1 100644
--- a/webrtc/test/fake_encoder.h
+++ b/webrtc/test/fake_encoder.h
@@ -12,8 +12,10 @@
#define WEBRTC_TEST_FAKE_ENCODER_H_
#include <vector>
+#include <memory>
#include "webrtc/base/criticalsection.h"
+#include "webrtc/base/sequenced_task_checker.h"
#include "webrtc/base/task_queue.h"
#include "webrtc/common_types.h"
#include "webrtc/system_wrappers/include/clock.h"
@@ -25,7 +27,7 @@ namespace test {
class FakeEncoder : public VideoEncoder {
public:
explicit FakeEncoder(Clock* clock);
- virtual ~FakeEncoder();
+ virtual ~FakeEncoder() = default;
// Sets max bitrate. Not thread-safe, call before registering the encoder.
void SetMaxBitrate(int max_kbps);
@@ -47,20 +49,21 @@ class FakeEncoder : public VideoEncoder {
static const char* kImplementationName;
protected:
- rtc::CriticalSection crit_sect_;
Clock* const clock_;
VideoCodec config_ GUARDED_BY(crit_sect_);
EncodedImageCallback* callback_ GUARDED_BY(crit_sect_);
BitrateAllocation target_bitrate_ GUARDED_BY(crit_sect_);
int max_target_bitrate_kbps_ GUARDED_BY(crit_sect_);
int64_t last_encode_time_ms_ GUARDED_BY(crit_sect_);
+ rtc::CriticalSection crit_sect_;
+
uint8_t encoded_buffer_[100000];
};
class FakeH264Encoder : public FakeEncoder, public EncodedImageCallback {
public:
explicit FakeH264Encoder(Clock* clock);
- virtual ~FakeH264Encoder() {}
+ virtual ~FakeH264Encoder() = default;
int32_t RegisterEncodeCompleteCallback(
EncodedImageCallback* callback) override;
@@ -70,15 +73,15 @@ class FakeH264Encoder : public FakeEncoder, public EncodedImageCallback {
const RTPFragmentationHeader* fragments) override;
private:
- rtc::CriticalSection local_crit_sect_;
EncodedImageCallback* callback_ GUARDED_BY(local_crit_sect_);
int idr_counter_ GUARDED_BY(local_crit_sect_);
+ rtc::CriticalSection local_crit_sect_;
};
class DelayedEncoder : public test::FakeEncoder {
public:
DelayedEncoder(Clock* clock, int delay_ms);
- virtual ~DelayedEncoder() {}
+ virtual ~DelayedEncoder() = default;
void SetDelay(int delay_ms);
int32_t Encode(const VideoFrame& input_image,
@@ -86,17 +89,22 @@ class DelayedEncoder : public test::FakeEncoder {
const std::vector<FrameType>* frame_types) override;
private:
- rtc::CriticalSection local_crit_sect_;
- int delay_ms_ GUARDED_BY(&local_crit_sect_);
+ int delay_ms_ ACCESS_ON(sequence_checker_);
+ rtc::SequencedTaskChecker sequence_checker_;
};
// This class implements a multi-threaded fake encoder by posting
// FakeH264Encoder::Encode(.) tasks to |queue1_| and |queue2_|, in an
-// alternating fashion.
+// alternating fashion. The class itself does not need to be thread safe,
+// as it is called from the task queue in ViEEncoder.
class MultithreadedFakeH264Encoder : public test::FakeH264Encoder {
public:
explicit MultithreadedFakeH264Encoder(Clock* clock);
- virtual ~MultithreadedFakeH264Encoder() override;
+ virtual ~MultithreadedFakeH264Encoder() = default;
+
+ int32_t InitEncode(const VideoCodec* config,
+ int32_t number_of_cores,
+ size_t max_payload_size) override;
int32_t Encode(const VideoFrame& input_image,
const CodecSpecificInfo* codec_specific_info,
@@ -106,12 +114,15 @@ class MultithreadedFakeH264Encoder : public test::FakeH264Encoder {
const CodecSpecificInfo* codec_specific_info,
const std::vector<FrameType>* frame_types);
+ int32_t Release() override;
+
protected:
class EncodeTask;
- int current_queue_;
- rtc::TaskQueue queue1_;
- rtc::TaskQueue queue2_;
+ int current_queue_ ACCESS_ON(sequence_checker_);
+ std::unique_ptr<rtc::TaskQueue> queue1_ ACCESS_ON(sequence_checker_);
+ std::unique_ptr<rtc::TaskQueue> queue2_ ACCESS_ON(sequence_checker_);
+ rtc::SequencedTaskChecker sequence_checker_;
};
} // namespace test
« no previous file with comments | « no previous file | webrtc/test/fake_encoder.cc » ('j') | webrtc/test/fake_encoder.cc » ('J')

Powered by Google App Engine
This is Rietveld 408576698