Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2018 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 | |
Niels Möller | cbcbc22 | 2018-09-28 07:07:24 | [diff] [blame] | 11 | #ifndef TEST_VIDEO_ENCODER_PROXY_FACTORY_H_ |
| 12 | #define TEST_VIDEO_ENCODER_PROXY_FACTORY_H_ |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 13 | |
| 14 | #include <memory> |
| 15 | #include <vector> |
| 16 | |
Karl Wiberg | 918f50c | 2018-07-05 09:40:33 | [diff] [blame] | 17 | #include "absl/memory/memory.h" |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 18 | #include "api/video_codecs/video_encoder.h" |
| 19 | #include "api/video_codecs/video_encoder_factory.h" |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 20 | |
| 21 | namespace webrtc { |
| 22 | namespace test { |
| 23 | |
| 24 | // An encoder factory with a single underlying VideoEncoder object, |
| 25 | // intended for test purposes. Each call to CreateVideoEncoder returns |
| 26 | // a proxy for the same encoder, typically an instance of FakeEncoder. |
Niels Möller | cbcbc22 | 2018-09-28 07:07:24 | [diff] [blame] | 27 | class VideoEncoderProxyFactory final : public VideoEncoderFactory { |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 28 | public: |
Mirta Dvornicic | 1ec2a16 | 2018-12-10 09:47:34 | [diff] [blame] | 29 | explicit VideoEncoderProxyFactory(VideoEncoder* encoder) : encoder_(encoder) { |
| 30 | codec_info_.is_hardware_accelerated = false; |
| 31 | codec_info_.has_internal_source = false; |
| 32 | } |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 33 | |
| 34 | // Unused by tests. |
| 35 | std::vector<SdpVideoFormat> GetSupportedFormats() const override { |
| 36 | RTC_NOTREACHED(); |
| 37 | return {}; |
| 38 | } |
| 39 | |
Mirta Dvornicic | 1ec2a16 | 2018-12-10 09:47:34 | [diff] [blame] | 40 | CodecInfo QueryVideoEncoder(const SdpVideoFormat& format) const override { |
| 41 | return codec_info_; |
| 42 | } |
| 43 | |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 44 | std::unique_ptr<VideoEncoder> CreateVideoEncoder( |
| 45 | const SdpVideoFormat& format) override { |
Karl Wiberg | 918f50c | 2018-07-05 09:40:33 | [diff] [blame] | 46 | return absl::make_unique<EncoderProxy>(encoder_); |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 47 | } |
| 48 | |
Mirta Dvornicic | 1ec2a16 | 2018-12-10 09:47:34 | [diff] [blame] | 49 | void SetIsHardwareAccelerated(bool is_hardware_accelerated) { |
| 50 | codec_info_.is_hardware_accelerated = is_hardware_accelerated; |
| 51 | } |
| 52 | void SetHasInternalSource(bool has_internal_source) { |
| 53 | codec_info_.has_internal_source = has_internal_source; |
| 54 | } |
| 55 | |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 56 | private: |
| 57 | // Wrapper class, since CreateVideoEncoder needs to surrender |
| 58 | // ownership to the object it returns. |
| 59 | class EncoderProxy final : public VideoEncoder { |
| 60 | public: |
| 61 | explicit EncoderProxy(VideoEncoder* encoder) : encoder_(encoder) {} |
| 62 | |
| 63 | private: |
| 64 | int32_t Encode(const VideoFrame& input_image, |
| 65 | const CodecSpecificInfo* codec_specific_info, |
| 66 | const std::vector<FrameType>* frame_types) override { |
| 67 | return encoder_->Encode(input_image, codec_specific_info, frame_types); |
| 68 | } |
| 69 | int32_t InitEncode(const VideoCodec* config, |
| 70 | int32_t number_of_cores, |
| 71 | size_t max_payload_size) override { |
| 72 | return encoder_->InitEncode(config, number_of_cores, max_payload_size); |
| 73 | } |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 74 | int32_t RegisterEncodeCompleteCallback( |
| 75 | EncodedImageCallback* callback) override { |
| 76 | return encoder_->RegisterEncodeCompleteCallback(callback); |
| 77 | } |
| 78 | int32_t Release() override { return encoder_->Release(); } |
Erik Språng | 566124a | 2018-04-23 10:32:22 | [diff] [blame] | 79 | int32_t SetRateAllocation(const VideoBitrateAllocation& rate_allocation, |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 80 | uint32_t framerate) override { |
| 81 | return encoder_->SetRateAllocation(rate_allocation, framerate); |
| 82 | } |
Erik Språng | a1ead6f2 | 2018-11-01 16:45:06 | [diff] [blame] | 83 | VideoEncoder::EncoderInfo GetEncoderInfo() const override { |
| 84 | return encoder_->GetEncoderInfo(); |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 85 | } |
| 86 | |
| 87 | VideoEncoder* const encoder_; |
| 88 | }; |
| 89 | |
| 90 | VideoEncoder* const encoder_; |
Mirta Dvornicic | 1ec2a16 | 2018-12-10 09:47:34 | [diff] [blame] | 91 | CodecInfo codec_info_; |
Niels Möller | 4db138e | 2018-04-19 07:04:13 | [diff] [blame] | 92 | }; |
| 93 | |
| 94 | } // namespace test |
| 95 | } // namespace webrtc |
| 96 | |
Niels Möller | cbcbc22 | 2018-09-28 07:07:24 | [diff] [blame] | 97 | #endif // TEST_VIDEO_ENCODER_PROXY_FACTORY_H_ |