pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 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 | */ |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 10 | #ifndef TEST_FRAME_GENERATOR_CAPTURER_H_ |
| 11 | #define TEST_FRAME_GENERATOR_CAPTURER_H_ |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 12 | |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 13 | #include <memory> |
sprang@webrtc.org | 131bea8 | 2015-02-18 12:46:06 +0000 | [diff] [blame] | 14 | #include <string> |
| 15 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 16 | #include "api/video/video_frame.h" |
| 17 | #include "rtc_base/criticalsection.h" |
| 18 | #include "rtc_base/task_queue.h" |
Emircan Uysaler | 03e6ec9 | 2018-03-09 15:03:26 -0800 | [diff] [blame] | 19 | #include "test/frame_generator.h" |
Sebastian Jansson | f1f363f | 2018-08-13 14:24:58 +0200 | [diff] [blame^] | 20 | #include "test/test_video_capturer.h" |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 21 | |
| 22 | namespace webrtc { |
| 23 | |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 24 | namespace test { |
| 25 | |
| 26 | class FrameGenerator; |
| 27 | |
Sebastian Jansson | f1f363f | 2018-08-13 14:24:58 +0200 | [diff] [blame^] | 28 | class FrameGeneratorCapturer : public TestVideoCapturer { |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 29 | public: |
perkj | 803d97f | 2016-11-01 11:45:46 -0700 | [diff] [blame] | 30 | class SinkWantsObserver { |
| 31 | public: |
| 32 | // OnSinkWantsChanged is called when FrameGeneratorCapturer::AddOrUpdateSink |
| 33 | // is called. |
| 34 | virtual void OnSinkWantsChanged(rtc::VideoSinkInterface<VideoFrame>* sink, |
| 35 | const rtc::VideoSinkWants& wants) = 0; |
| 36 | |
| 37 | protected: |
| 38 | virtual ~SinkWantsObserver() {} |
| 39 | }; |
| 40 | |
Emircan Uysaler | 207a75d | 2018-03-12 14:12:14 -0700 | [diff] [blame] | 41 | // |type| has the default value OutputType::I420. |num_squares| has the |
| 42 | // default value 10. |
Emircan Uysaler | 03e6ec9 | 2018-03-09 15:03:26 -0800 | [diff] [blame] | 43 | static FrameGeneratorCapturer* Create( |
| 44 | int width, |
| 45 | int height, |
Danil Chapovalov | 431abd9 | 2018-06-18 12:54:17 +0200 | [diff] [blame] | 46 | absl::optional<FrameGenerator::OutputType> type, |
| 47 | absl::optional<int> num_squares, |
Emircan Uysaler | 03e6ec9 | 2018-03-09 15:03:26 -0800 | [diff] [blame] | 48 | int target_fps, |
| 49 | Clock* clock); |
Taylor Brandstetter | 081136f | 2018-03-08 01:54:10 +0000 | [diff] [blame] | 50 | |
perkj | a49cbd3 | 2016-09-16 07:53:41 -0700 | [diff] [blame] | 51 | static FrameGeneratorCapturer* CreateFromYuvFile(const std::string& file_name, |
andresp@webrtc.org | ab65495 | 2013-09-19 12:14:03 +0000 | [diff] [blame] | 52 | size_t width, |
| 53 | size_t height, |
| 54 | int target_fps, |
| 55 | Clock* clock); |
erikvarga | 579de6f | 2017-08-29 09:12:57 -0700 | [diff] [blame] | 56 | |
| 57 | static FrameGeneratorCapturer* CreateSlideGenerator(int width, |
| 58 | int height, |
| 59 | int frame_repeat_count, |
| 60 | int target_fps, |
| 61 | Clock* clock); |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 62 | virtual ~FrameGeneratorCapturer(); |
| 63 | |
kjellander@webrtc.org | 14665ff | 2015-03-04 12:58:35 +0000 | [diff] [blame] | 64 | void Start() override; |
| 65 | void Stop() override; |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 66 | void ChangeResolution(size_t width, size_t height); |
perkj | a49cbd3 | 2016-09-16 07:53:41 -0700 | [diff] [blame] | 67 | |
perkj | 803d97f | 2016-11-01 11:45:46 -0700 | [diff] [blame] | 68 | void SetSinkWantsObserver(SinkWantsObserver* observer); |
| 69 | |
perkj | a49cbd3 | 2016-09-16 07:53:41 -0700 | [diff] [blame] | 70 | void AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink, |
| 71 | const rtc::VideoSinkWants& wants) override; |
| 72 | void RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) override; |
| 73 | |
sprang | 867fb52 | 2015-08-03 04:38:41 -0700 | [diff] [blame] | 74 | void ForceFrame(); |
Per | ba7dc72 | 2016-04-19 15:01:23 +0200 | [diff] [blame] | 75 | void SetFakeRotation(VideoRotation rotation); |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 76 | |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 77 | int64_t first_frame_capture_time() const { return first_frame_capture_time_; } |
| 78 | |
andresp@webrtc.org | ab65495 | 2013-09-19 12:14:03 +0000 | [diff] [blame] | 79 | FrameGeneratorCapturer(Clock* clock, |
perkj | a8ba195 | 2017-02-27 06:52:10 -0800 | [diff] [blame] | 80 | std::unique_ptr<FrameGenerator> frame_generator, |
pbos@webrtc.org | af8d5af | 2013-07-09 08:02:33 +0000 | [diff] [blame] | 81 | int target_fps); |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 82 | bool Init(); |
sprang@webrtc.org | 131bea8 | 2015-02-18 12:46:06 +0000 | [diff] [blame] | 83 | |
| 84 | private: |
ilnik | baded15 | 2017-03-17 05:55:25 -0700 | [diff] [blame] | 85 | class InsertFrameTask; |
| 86 | |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 87 | void InsertFrame(); |
| 88 | static bool Run(void* obj); |
sprang | c5d62e2 | 2017-04-02 23:53:04 -0700 | [diff] [blame] | 89 | int GetCurrentConfiguredFramerate(); |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 90 | |
pbos@webrtc.org | de1429e | 2014-04-28 13:00:21 +0000 | [diff] [blame] | 91 | Clock* const clock_; |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 92 | bool sending_; |
danilchap | a37de39 | 2017-09-09 04:17:22 -0700 | [diff] [blame] | 93 | rtc::VideoSinkInterface<VideoFrame>* sink_ RTC_GUARDED_BY(&lock_); |
| 94 | SinkWantsObserver* sink_wants_observer_ RTC_GUARDED_BY(&lock_); |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 95 | |
Peter Boström | f2f8283 | 2015-05-01 13:00:41 +0200 | [diff] [blame] | 96 | rtc::CriticalSection lock_; |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 97 | std::unique_ptr<FrameGenerator> frame_generator_; |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 98 | |
danilchap | a37de39 | 2017-09-09 04:17:22 -0700 | [diff] [blame] | 99 | int target_fps_ RTC_GUARDED_BY(&lock_); |
Danil Chapovalov | 431abd9 | 2018-06-18 12:54:17 +0200 | [diff] [blame] | 100 | absl::optional<int> wanted_fps_ RTC_GUARDED_BY(&lock_); |
Per | ba7dc72 | 2016-04-19 15:01:23 +0200 | [diff] [blame] | 101 | VideoRotation fake_rotation_ = kVideoRotation_0; |
wu@webrtc.org | cd70119 | 2014-04-24 22:10:24 +0000 | [diff] [blame] | 102 | |
| 103 | int64_t first_frame_capture_time_; |
ilnik | baded15 | 2017-03-17 05:55:25 -0700 | [diff] [blame] | 104 | // Must be the last field, so it will be deconstructed first as tasks |
| 105 | // in the TaskQueue access other fields of the instance of this class. |
| 106 | rtc::TaskQueue task_queue_; |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 107 | }; |
ilnik | baded15 | 2017-03-17 05:55:25 -0700 | [diff] [blame] | 108 | } // namespace test |
| 109 | } // namespace webrtc |
pbos@webrtc.org | 26d1210 | 2013-05-29 13:41:03 +0000 | [diff] [blame] | 110 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 111 | #endif // TEST_FRAME_GENERATOR_CAPTURER_H_ |