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

Side by Side Diff: webrtc/test/fake_encoder.h

Issue 2686103002: Do not encode frames in MultithreadedFakeH264Encoder after Release(). (Closed)
Patch Set: sprang comments 1. 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 unified diff | Download patch
« no previous file with comments | « no previous file | webrtc/test/fake_encoder.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source 5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #ifndef WEBRTC_TEST_FAKE_ENCODER_H_ 11 #ifndef WEBRTC_TEST_FAKE_ENCODER_H_
12 #define WEBRTC_TEST_FAKE_ENCODER_H_ 12 #define WEBRTC_TEST_FAKE_ENCODER_H_
13 13
14 #include <vector> 14 #include <vector>
15 #include <memory>
15 16
16 #include "webrtc/base/criticalsection.h" 17 #include "webrtc/base/criticalsection.h"
18 #include "webrtc/base/sequenced_task_checker.h"
17 #include "webrtc/base/task_queue.h" 19 #include "webrtc/base/task_queue.h"
18 #include "webrtc/common_types.h" 20 #include "webrtc/common_types.h"
19 #include "webrtc/system_wrappers/include/clock.h" 21 #include "webrtc/system_wrappers/include/clock.h"
20 #include "webrtc/video_encoder.h" 22 #include "webrtc/video_encoder.h"
21 23
22 namespace webrtc { 24 namespace webrtc {
23 namespace test { 25 namespace test {
24 26
25 class FakeEncoder : public VideoEncoder { 27 class FakeEncoder : public VideoEncoder {
26 public: 28 public:
27 explicit FakeEncoder(Clock* clock); 29 explicit FakeEncoder(Clock* clock);
28 virtual ~FakeEncoder(); 30 virtual ~FakeEncoder() = default;
29 31
30 // Sets max bitrate. Not thread-safe, call before registering the encoder. 32 // Sets max bitrate. Not thread-safe, call before registering the encoder.
31 void SetMaxBitrate(int max_kbps); 33 void SetMaxBitrate(int max_kbps);
32 34
33 int32_t InitEncode(const VideoCodec* config, 35 int32_t InitEncode(const VideoCodec* config,
34 int32_t number_of_cores, 36 int32_t number_of_cores,
35 size_t max_payload_size) override; 37 size_t max_payload_size) override;
36 int32_t Encode(const VideoFrame& input_image, 38 int32_t Encode(const VideoFrame& input_image,
37 const CodecSpecificInfo* codec_specific_info, 39 const CodecSpecificInfo* codec_specific_info,
38 const std::vector<FrameType>* frame_types) override; 40 const std::vector<FrameType>* frame_types) override;
39 int32_t RegisterEncodeCompleteCallback( 41 int32_t RegisterEncodeCompleteCallback(
40 EncodedImageCallback* callback) override; 42 EncodedImageCallback* callback) override;
41 int32_t Release() override; 43 int32_t Release() override;
42 int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override; 44 int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override;
43 int32_t SetRateAllocation(const BitrateAllocation& rate_allocation, 45 int32_t SetRateAllocation(const BitrateAllocation& rate_allocation,
44 uint32_t framerate) override; 46 uint32_t framerate) override;
45 const char* ImplementationName() const override; 47 const char* ImplementationName() const override;
46 48
47 static const char* kImplementationName; 49 static const char* kImplementationName;
48 50
49 protected: 51 protected:
50 rtc::CriticalSection crit_sect_;
51 Clock* const clock_; 52 Clock* const clock_;
52 VideoCodec config_ GUARDED_BY(crit_sect_); 53 VideoCodec config_ GUARDED_BY(crit_sect_);
53 EncodedImageCallback* callback_ GUARDED_BY(crit_sect_); 54 EncodedImageCallback* callback_ GUARDED_BY(crit_sect_);
54 BitrateAllocation target_bitrate_ GUARDED_BY(crit_sect_); 55 BitrateAllocation target_bitrate_ GUARDED_BY(crit_sect_);
55 int max_target_bitrate_kbps_ GUARDED_BY(crit_sect_); 56 int max_target_bitrate_kbps_ GUARDED_BY(crit_sect_);
56 int64_t last_encode_time_ms_ GUARDED_BY(crit_sect_); 57 int64_t last_encode_time_ms_ GUARDED_BY(crit_sect_);
58 rtc::CriticalSection crit_sect_;
59
57 uint8_t encoded_buffer_[100000]; 60 uint8_t encoded_buffer_[100000];
58 }; 61 };
59 62
60 class FakeH264Encoder : public FakeEncoder, public EncodedImageCallback { 63 class FakeH264Encoder : public FakeEncoder, public EncodedImageCallback {
61 public: 64 public:
62 explicit FakeH264Encoder(Clock* clock); 65 explicit FakeH264Encoder(Clock* clock);
63 virtual ~FakeH264Encoder() {} 66 virtual ~FakeH264Encoder() = default;
64 67
65 int32_t RegisterEncodeCompleteCallback( 68 int32_t RegisterEncodeCompleteCallback(
66 EncodedImageCallback* callback) override; 69 EncodedImageCallback* callback) override;
67 70
68 Result OnEncodedImage(const EncodedImage& encodedImage, 71 Result OnEncodedImage(const EncodedImage& encodedImage,
69 const CodecSpecificInfo* codecSpecificInfo, 72 const CodecSpecificInfo* codecSpecificInfo,
70 const RTPFragmentationHeader* fragments) override; 73 const RTPFragmentationHeader* fragments) override;
71 74
72 private: 75 private:
73 rtc::CriticalSection local_crit_sect_;
74 EncodedImageCallback* callback_ GUARDED_BY(local_crit_sect_); 76 EncodedImageCallback* callback_ GUARDED_BY(local_crit_sect_);
75 int idr_counter_ GUARDED_BY(local_crit_sect_); 77 int idr_counter_ GUARDED_BY(local_crit_sect_);
78 rtc::CriticalSection local_crit_sect_;
76 }; 79 };
77 80
78 class DelayedEncoder : public test::FakeEncoder { 81 class DelayedEncoder : public test::FakeEncoder {
79 public: 82 public:
80 DelayedEncoder(Clock* clock, int delay_ms); 83 DelayedEncoder(Clock* clock, int delay_ms);
81 virtual ~DelayedEncoder() {} 84 virtual ~DelayedEncoder() = default;
82 85
83 void SetDelay(int delay_ms); 86 void SetDelay(int delay_ms);
84 int32_t Encode(const VideoFrame& input_image, 87 int32_t Encode(const VideoFrame& input_image,
85 const CodecSpecificInfo* codec_specific_info, 88 const CodecSpecificInfo* codec_specific_info,
86 const std::vector<FrameType>* frame_types) override; 89 const std::vector<FrameType>* frame_types) override;
87 90
88 private: 91 private:
89 rtc::CriticalSection local_crit_sect_; 92 int delay_ms_ ACCESS_ON(sequence_checker_);
90 int delay_ms_ GUARDED_BY(&local_crit_sect_); 93 rtc::SequencedTaskChecker sequence_checker_;
91 }; 94 };
92 95
93 // This class implements a multi-threaded fake encoder by posting 96 // This class implements a multi-threaded fake encoder by posting
94 // FakeH264Encoder::Encode(.) tasks to |queue1_| and |queue2_|, in an 97 // FakeH264Encoder::Encode(.) tasks to |queue1_| and |queue2_|, in an
95 // alternating fashion. 98 // alternating fashion. The class itself does not need to be thread safe,
99 // as it is called from the task queue in ViEEncoder.
96 class MultithreadedFakeH264Encoder : public test::FakeH264Encoder { 100 class MultithreadedFakeH264Encoder : public test::FakeH264Encoder {
97 public: 101 public:
98 explicit MultithreadedFakeH264Encoder(Clock* clock); 102 explicit MultithreadedFakeH264Encoder(Clock* clock);
99 virtual ~MultithreadedFakeH264Encoder() override; 103 virtual ~MultithreadedFakeH264Encoder() = default;
104
105 int32_t InitEncode(const VideoCodec* config,
106 int32_t number_of_cores,
107 size_t max_payload_size) override;
100 108
101 int32_t Encode(const VideoFrame& input_image, 109 int32_t Encode(const VideoFrame& input_image,
102 const CodecSpecificInfo* codec_specific_info, 110 const CodecSpecificInfo* codec_specific_info,
103 const std::vector<FrameType>* frame_types) override; 111 const std::vector<FrameType>* frame_types) override;
104 112
105 int32_t EncodeCallback(const VideoFrame& input_image, 113 int32_t EncodeCallback(const VideoFrame& input_image,
106 const CodecSpecificInfo* codec_specific_info, 114 const CodecSpecificInfo* codec_specific_info,
107 const std::vector<FrameType>* frame_types); 115 const std::vector<FrameType>* frame_types);
108 116
117 int32_t Release() override;
118
109 protected: 119 protected:
110 class EncodeTask; 120 class EncodeTask;
111 121
112 int current_queue_; 122 int current_queue_ ACCESS_ON(sequence_checker_);
113 rtc::TaskQueue queue1_; 123 std::unique_ptr<rtc::TaskQueue> queue1_ ACCESS_ON(sequence_checker_);
114 rtc::TaskQueue queue2_; 124 std::unique_ptr<rtc::TaskQueue> queue2_ ACCESS_ON(sequence_checker_);
125 rtc::SequencedTaskChecker sequence_checker_;
115 }; 126 };
116 127
117 } // namespace test 128 } // namespace test
118 } // namespace webrtc 129 } // namespace webrtc
119 130
120 #endif // WEBRTC_TEST_FAKE_ENCODER_H_ 131 #endif // WEBRTC_TEST_FAKE_ENCODER_H_
OLDNEW
« no previous file with comments | « no previous file | webrtc/test/fake_encoder.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698