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 |