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 11 #ifndef API_TEST_VIDEO_FUNCTION_VIDEO_DECODER_FACTORY_H_ 12 #define API_TEST_VIDEO_FUNCTION_VIDEO_DECODER_FACTORY_H_ 13 14 #include <functional> 15 #include <memory> 16 #include <utility> 17 #include <vector> 18 19 #include "api/video_codecs/sdp_video_format.h" 20 #include "api/video_codecs/video_decoder.h" 21 #include "api/video_codecs/video_decoder_factory.h" 22 #include "rtc_base/checks.h" 23 24 namespace webrtc { 25 namespace test { 26 27 // A decoder factory producing decoders by calling a supplied create function. 28 class FunctionVideoDecoderFactory final : public VideoDecoderFactory { 29 public: FunctionVideoDecoderFactory(std::function<std::unique_ptr<VideoDecoder> ()> create)30 explicit FunctionVideoDecoderFactory( 31 std::function<std::unique_ptr<VideoDecoder>()> create) 32 : create_([create = std::move(create)](const SdpVideoFormat&) { 33 return create(); 34 }) {} FunctionVideoDecoderFactory(std::function<std::unique_ptr<VideoDecoder> (const SdpVideoFormat &)> create)35 explicit FunctionVideoDecoderFactory( 36 std::function<std::unique_ptr<VideoDecoder>(const SdpVideoFormat&)> 37 create) 38 : create_(std::move(create)) {} FunctionVideoDecoderFactory(std::function<std::unique_ptr<VideoDecoder> ()> create,std::vector<SdpVideoFormat> sdp_video_formats)39 FunctionVideoDecoderFactory( 40 std::function<std::unique_ptr<VideoDecoder>()> create, 41 std::vector<SdpVideoFormat> sdp_video_formats) 42 : create_([create = std::move(create)](const SdpVideoFormat&) { 43 return create(); 44 }), 45 sdp_video_formats_(std::move(sdp_video_formats)) {} 46 GetSupportedFormats()47 std::vector<SdpVideoFormat> GetSupportedFormats() const override { 48 return sdp_video_formats_; 49 } 50 CreateVideoDecoder(const SdpVideoFormat & format)51 std::unique_ptr<VideoDecoder> CreateVideoDecoder( 52 const SdpVideoFormat& format) override { 53 return create_(format); 54 } 55 56 private: 57 const std::function<std::unique_ptr<VideoDecoder>(const SdpVideoFormat&)> 58 create_; 59 const std::vector<SdpVideoFormat> sdp_video_formats_; 60 }; 61 62 } // namespace test 63 } // namespace webrtc 64 65 #endif // API_TEST_VIDEO_FUNCTION_VIDEO_DECODER_FACTORY_H_ 66