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