1 /*
2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3  *
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
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #ifndef WEBRTC_TEST_FAKE_ENCODER_H_
12 #define WEBRTC_TEST_FAKE_ENCODER_H_
13 
14 #include <vector>
15 
16 #include "webrtc/common_types.h"
17 #include "webrtc/system_wrappers/include/clock.h"
18 #include "webrtc/video_encoder.h"
19 
20 namespace webrtc {
21 namespace test {
22 
23 class FakeEncoder : public VideoEncoder {
24  public:
25   explicit FakeEncoder(Clock* clock);
26   virtual ~FakeEncoder();
27 
28   // Sets max bitrate. Not thread-safe, call before registering the encoder.
29   void SetMaxBitrate(int max_kbps);
30 
31   int32_t InitEncode(const VideoCodec* config,
32                      int32_t number_of_cores,
33                      size_t max_payload_size) override;
34   int32_t Encode(const VideoFrame& input_image,
35                  const CodecSpecificInfo* codec_specific_info,
36                  const std::vector<FrameType>* frame_types) override;
37   int32_t RegisterEncodeCompleteCallback(
38       EncodedImageCallback* callback) override;
39   int32_t Release() override;
40   int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override;
41   int32_t SetRates(uint32_t new_target_bitrate, uint32_t framerate) override;
42   const char* ImplementationName() const override;
43 
44   static const char* kImplementationName;
45 
46  protected:
47   Clock* const clock_;
48   VideoCodec config_;
49   EncodedImageCallback* callback_;
50   int target_bitrate_kbps_;
51   int max_target_bitrate_kbps_;
52   int64_t last_encode_time_ms_;
53   uint8_t encoded_buffer_[100000];
54 };
55 
56 class FakeH264Encoder : public FakeEncoder, public EncodedImageCallback {
57  public:
58   explicit FakeH264Encoder(Clock* clock);
~FakeH264Encoder()59   virtual ~FakeH264Encoder() {}
60 
61   int32_t RegisterEncodeCompleteCallback(
62       EncodedImageCallback* callback) override;
63 
64   int32_t Encoded(const EncodedImage& encodedImage,
65                   const CodecSpecificInfo* codecSpecificInfo,
66                   const RTPFragmentationHeader* fragments) override;
67 
68  private:
69   EncodedImageCallback* callback_;
70   int idr_counter_;
71 };
72 
73 class DelayedEncoder : public test::FakeEncoder {
74  public:
75   DelayedEncoder(Clock* clock, int delay_ms);
~DelayedEncoder()76   virtual ~DelayedEncoder() {}
77 
78   int32_t Encode(const VideoFrame& input_image,
79                  const CodecSpecificInfo* codec_specific_info,
80                  const std::vector<FrameType>* frame_types) override;
81 
82  private:
83   const int delay_ms_;
84 };
85 }  // namespace test
86 }  // namespace webrtc
87 
88 #endif  // WEBRTC_TEST_FAKE_ENCODER_H_
89