summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/test/frame_generator_capturer.h
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/libwebrtc/test/frame_generator_capturer.h
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/test/frame_generator_capturer.h')
-rw-r--r--third_party/libwebrtc/test/frame_generator_capturer.h191
1 files changed, 191 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/frame_generator_capturer.h b/third_party/libwebrtc/test/frame_generator_capturer.h
new file mode 100644
index 0000000000..e310e40129
--- /dev/null
+++ b/third_party/libwebrtc/test/frame_generator_capturer.h
@@ -0,0 +1,191 @@
+/*
+ * 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_FRAME_GENERATOR_CAPTURER_H_
+#define TEST_FRAME_GENERATOR_CAPTURER_H_
+
+#include <memory>
+#include <string>
+
+#include "api/task_queue/task_queue_factory.h"
+#include "api/test/frame_generator_interface.h"
+#include "api/video/video_frame.h"
+#include "rtc_base/synchronization/mutex.h"
+#include "rtc_base/task_queue.h"
+#include "rtc_base/task_utils/repeating_task.h"
+#include "system_wrappers/include/clock.h"
+#include "test/test_video_capturer.h"
+
+namespace webrtc {
+
+namespace test {
+namespace frame_gen_cap_impl {
+template <typename T>
+class AutoOpt : public absl::optional<T> {
+ public:
+ using absl::optional<T>::optional;
+ T* operator->() {
+ if (!absl::optional<T>::has_value())
+ this->emplace(T());
+ return absl::optional<T>::operator->();
+ }
+};
+} // namespace frame_gen_cap_impl
+struct FrameGeneratorCapturerConfig {
+ struct SquaresVideo {
+ int framerate = 30;
+ FrameGeneratorInterface::OutputType pixel_format =
+ FrameGeneratorInterface::OutputType::kI420;
+ int width = 320;
+ int height = 180;
+ int num_squares = 10;
+ };
+
+ struct SquareSlides {
+ int framerate = 30;
+ TimeDelta change_interval = TimeDelta::Seconds(10);
+ int width = 1600;
+ int height = 1200;
+ };
+
+ struct VideoFile {
+ int framerate = 30;
+ std::string name;
+ // Must be set to width and height of the source video file.
+ int width = 0;
+ int height = 0;
+ };
+
+ struct ImageSlides {
+ int framerate = 30;
+ TimeDelta change_interval = TimeDelta::Seconds(10);
+ struct Crop {
+ TimeDelta scroll_duration = TimeDelta::Seconds(0);
+ absl::optional<int> width;
+ absl::optional<int> height;
+ } crop;
+ int width = 1850;
+ int height = 1110;
+ std::vector<std::string> paths = {
+ "web_screenshot_1850_1110",
+ "presentation_1850_1110",
+ "photo_1850_1110",
+ "difficult_photo_1850_1110",
+ };
+ };
+
+ frame_gen_cap_impl::AutoOpt<SquaresVideo> squares_video;
+ frame_gen_cap_impl::AutoOpt<SquareSlides> squares_slides;
+ frame_gen_cap_impl::AutoOpt<VideoFile> video_file;
+ frame_gen_cap_impl::AutoOpt<ImageSlides> image_slides;
+};
+
+class FrameGeneratorCapturer : public TestVideoCapturer {
+ public:
+ class SinkWantsObserver {
+ public:
+ // OnSinkWantsChanged is called when FrameGeneratorCapturer::AddOrUpdateSink
+ // is called.
+ virtual void OnSinkWantsChanged(rtc::VideoSinkInterface<VideoFrame>* sink,
+ const rtc::VideoSinkWants& wants) = 0;
+
+ protected:
+ virtual ~SinkWantsObserver() {}
+ };
+
+ FrameGeneratorCapturer(
+ Clock* clock,
+ std::unique_ptr<FrameGeneratorInterface> frame_generator,
+ int target_fps,
+ TaskQueueFactory& task_queue_factory);
+ virtual ~FrameGeneratorCapturer();
+
+ static std::unique_ptr<FrameGeneratorCapturer> Create(
+ Clock* clock,
+ TaskQueueFactory& task_queue_factory,
+ FrameGeneratorCapturerConfig::SquaresVideo config);
+ static std::unique_ptr<FrameGeneratorCapturer> Create(
+ Clock* clock,
+ TaskQueueFactory& task_queue_factory,
+ FrameGeneratorCapturerConfig::SquareSlides config);
+ static std::unique_ptr<FrameGeneratorCapturer> Create(
+ Clock* clock,
+ TaskQueueFactory& task_queue_factory,
+ FrameGeneratorCapturerConfig::VideoFile config);
+ static std::unique_ptr<FrameGeneratorCapturer> Create(
+ Clock* clock,
+ TaskQueueFactory& task_queue_factory,
+ FrameGeneratorCapturerConfig::ImageSlides config);
+ static std::unique_ptr<FrameGeneratorCapturer> Create(
+ Clock* clock,
+ TaskQueueFactory& task_queue_factory,
+ const FrameGeneratorCapturerConfig& config);
+
+ void Start();
+ void Stop();
+ void ChangeResolution(size_t width, size_t height);
+ void ChangeFramerate(int target_framerate);
+
+ struct Resolution {
+ int width;
+ int height;
+ };
+ absl::optional<Resolution> GetResolution();
+
+ void OnOutputFormatRequest(int width,
+ int height,
+ const absl::optional<int>& max_fps);
+
+ void SetSinkWantsObserver(SinkWantsObserver* observer);
+
+ void AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink,
+ const rtc::VideoSinkWants& wants) override;
+ void RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) override;
+
+ void ForceFrame();
+ void SetFakeRotation(VideoRotation rotation);
+ void SetFakeColorSpace(absl::optional<ColorSpace> color_space);
+
+ int64_t first_frame_capture_time() const { return first_frame_capture_time_; }
+
+ bool Init();
+
+ private:
+ void InsertFrame();
+ static bool Run(void* obj);
+ int GetCurrentConfiguredFramerate();
+ void UpdateFps(int max_fps) RTC_EXCLUSIVE_LOCKS_REQUIRED(&lock_);
+
+ Clock* const clock_;
+ RepeatingTaskHandle frame_task_;
+ bool sending_;
+ SinkWantsObserver* sink_wants_observer_ RTC_GUARDED_BY(&lock_);
+
+ Mutex lock_;
+ std::unique_ptr<FrameGeneratorInterface> frame_generator_;
+
+ int source_fps_ RTC_GUARDED_BY(&lock_);
+ int target_capture_fps_ RTC_GUARDED_BY(&lock_);
+ absl::optional<int> wanted_fps_ RTC_GUARDED_BY(&lock_);
+ VideoRotation fake_rotation_ = kVideoRotation_0;
+ absl::optional<ColorSpace> fake_color_space_ RTC_GUARDED_BY(&lock_);
+
+ int64_t first_frame_capture_time_;
+
+ Mutex stats_lock_;
+ absl::optional<Resolution> source_resolution_ RTC_GUARDED_BY(&stats_lock_);
+
+ // Must be the last field, so it will be deconstructed first as tasks
+ // in the TaskQueue access other fields of the instance of this class.
+ rtc::TaskQueue task_queue_;
+};
+} // namespace test
+} // namespace webrtc
+
+#endif // TEST_FRAME_GENERATOR_CAPTURER_H_