diff options
Diffstat (limited to 'third_party/libwebrtc/test/configurable_frame_size_encoder.h')
-rw-r--r-- | third_party/libwebrtc/test/configurable_frame_size_encoder.h | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/configurable_frame_size_encoder.h b/third_party/libwebrtc/test/configurable_frame_size_encoder.h new file mode 100644 index 0000000000..747fc09d4b --- /dev/null +++ b/third_party/libwebrtc/test/configurable_frame_size_encoder.h @@ -0,0 +1,72 @@ +/* + * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. + * + * Use of this source code is governed by a BSD-style license + * that can be found in the LICENSE file in the root of the source + * tree. An additional intellectual property rights grant can be found + * in the file PATENTS. All contributing project authors may + * be found in the AUTHORS file in the root of the source tree. + */ + +#ifndef TEST_CONFIGURABLE_FRAME_SIZE_ENCODER_H_ +#define TEST_CONFIGURABLE_FRAME_SIZE_ENCODER_H_ + +#include <stddef.h> +#include <stdint.h> + +#include <functional> +#include <memory> +#include <vector> + +#include "absl/types/optional.h" +#include "api/video/video_bitrate_allocation.h" +#include "api/video/video_frame.h" +#include "api/video_codecs/video_codec.h" +#include "api/video_codecs/video_encoder.h" +#include "modules/video_coding/include/video_codec_interface.h" + +namespace webrtc { +namespace test { + +class ConfigurableFrameSizeEncoder : public VideoEncoder { + public: + explicit ConfigurableFrameSizeEncoder(size_t max_frame_size); + ~ConfigurableFrameSizeEncoder() override; + + void SetFecControllerOverride( + FecControllerOverride* fec_controller_override) override; + + int32_t InitEncode(const VideoCodec* codec_settings, + const Settings& settings) override; + + int32_t Encode(const VideoFrame& input_image, + const std::vector<VideoFrameType>* frame_types) override; + + int32_t RegisterEncodeCompleteCallback( + EncodedImageCallback* callback) override; + + int32_t Release() override; + + void SetRates(const RateControlParameters& parameters) override; + + EncoderInfo GetEncoderInfo() const override; + + int32_t SetFrameSize(size_t size); + + void SetCodecType(VideoCodecType codec_type_); + + void RegisterPostEncodeCallback( + std::function<void(void)> post_encode_callback); + + private: + EncodedImageCallback* callback_; + absl::optional<std::function<void(void)>> post_encode_callback_; + + size_t current_frame_size_; + VideoCodecType codec_type_; +}; + +} // namespace test +} // namespace webrtc + +#endif // TEST_CONFIGURABLE_FRAME_SIZE_ENCODER_H_ |