diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
commit | 26a029d407be480d791972afb5975cf62c9360a6 (patch) | |
tree | f435a8308119effd964b339f76abb83a57c29483 /third_party/libwebrtc/test/frame_generator_capturer.cc | |
parent | Initial commit. (diff) | |
download | firefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz firefox-26a029d407be480d791972afb5975cf62c9360a6.zip |
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/test/frame_generator_capturer.cc')
-rw-r--r-- | third_party/libwebrtc/test/frame_generator_capturer.cc | 233 |
1 files changed, 233 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/frame_generator_capturer.cc b/third_party/libwebrtc/test/frame_generator_capturer.cc new file mode 100644 index 0000000000..6ba0807a74 --- /dev/null +++ b/third_party/libwebrtc/test/frame_generator_capturer.cc @@ -0,0 +1,233 @@ +/* + * 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. + */ + +#include "test/frame_generator_capturer.h" + +#include <algorithm> +#include <cmath> +#include <cstddef> +#include <memory> +#include <utility> + +#include "absl/types/optional.h" +#include "api/task_queue/task_queue_base.h" +#include "api/task_queue/task_queue_factory.h" +#include "api/test/frame_generator_interface.h" +#include "api/units/time_delta.h" +#include "api/video/color_space.h" +#include "api/video/video_frame.h" +#include "api/video/video_rotation.h" +#include "api/video/video_sink_interface.h" +#include "api/video/video_source_interface.h" +#include "rtc_base/checks.h" +#include "rtc_base/logging.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 { + +FrameGeneratorCapturer::FrameGeneratorCapturer( + Clock* clock, + std::unique_ptr<FrameGeneratorInterface> frame_generator, + int target_fps, + TaskQueueFactory& task_queue_factory) + : clock_(clock), + sending_(true), + sink_wants_observer_(nullptr), + frame_generator_(std::move(frame_generator)), + source_fps_(target_fps), + target_capture_fps_(target_fps), + first_frame_capture_time_(-1), + task_queue_(task_queue_factory.CreateTaskQueue( + "FrameGenCapQ", + TaskQueueFactory::Priority::HIGH)) { + RTC_DCHECK(frame_generator_); + RTC_DCHECK_GT(target_fps, 0); +} + +FrameGeneratorCapturer::~FrameGeneratorCapturer() { + Stop(); +} + +void FrameGeneratorCapturer::SetFakeRotation(VideoRotation rotation) { + MutexLock lock(&lock_); + fake_rotation_ = rotation; +} + +void FrameGeneratorCapturer::SetFakeColorSpace( + absl::optional<ColorSpace> color_space) { + MutexLock lock(&lock_); + fake_color_space_ = color_space; +} + +bool FrameGeneratorCapturer::Init() { + // This check is added because frame_generator_ might be file based and should + // not crash because a file moved. + if (frame_generator_.get() == nullptr) + return false; + + frame_task_ = RepeatingTaskHandle::DelayedStart( + task_queue_.Get(), + TimeDelta::Seconds(1) / GetCurrentConfiguredFramerate(), + [this] { + InsertFrame(); + return TimeDelta::Seconds(1) / GetCurrentConfiguredFramerate(); + }, + TaskQueueBase::DelayPrecision::kHigh); + return true; +} + +void FrameGeneratorCapturer::InsertFrame() { + MutexLock lock(&lock_); + if (sending_) { + FrameGeneratorInterface::VideoFrameData frame_data = + frame_generator_->NextFrame(); + // TODO(srte): Use more advanced frame rate control to allow arbritrary + // fractions. + int decimation = + std::round(static_cast<double>(source_fps_) / target_capture_fps_); + for (int i = 1; i < decimation; ++i) + frame_data = frame_generator_->NextFrame(); + + VideoFrame frame = VideoFrame::Builder() + .set_video_frame_buffer(frame_data.buffer) + .set_rotation(fake_rotation_) + .set_timestamp_us(clock_->TimeInMicroseconds()) + .set_ntp_time_ms(clock_->CurrentNtpInMilliseconds()) + .set_update_rect(frame_data.update_rect) + .set_color_space(fake_color_space_) + .build(); + if (first_frame_capture_time_ == -1) { + first_frame_capture_time_ = frame.ntp_time_ms(); + } + + TestVideoCapturer::OnFrame(frame); + } +} + +absl::optional<FrameGeneratorCapturer::Resolution> +FrameGeneratorCapturer::GetResolution() const { + FrameGeneratorInterface::Resolution resolution = + frame_generator_->GetResolution(); + return Resolution{.width = static_cast<int>(resolution.width), + .height = static_cast<int>(resolution.height)}; +} + +void FrameGeneratorCapturer::Start() { + { + MutexLock lock(&lock_); + sending_ = true; + } + if (!frame_task_.Running()) { + frame_task_ = RepeatingTaskHandle::Start( + task_queue_.Get(), + [this] { + InsertFrame(); + return TimeDelta::Seconds(1) / GetCurrentConfiguredFramerate(); + }, + TaskQueueBase::DelayPrecision::kHigh); + } +} + +void FrameGeneratorCapturer::Stop() { + MutexLock lock(&lock_); + sending_ = false; +} + +void FrameGeneratorCapturer::ChangeResolution(size_t width, size_t height) { + MutexLock lock(&lock_); + frame_generator_->ChangeResolution(width, height); +} + +void FrameGeneratorCapturer::ChangeFramerate(int target_framerate) { + MutexLock lock(&lock_); + RTC_CHECK(target_capture_fps_ > 0); + if (target_framerate > source_fps_) + RTC_LOG(LS_WARNING) << "Target framerate clamped from " << target_framerate + << " to " << source_fps_; + if (source_fps_ % target_capture_fps_ != 0) { + int decimation = + std::round(static_cast<double>(source_fps_) / target_capture_fps_); + int effective_rate = target_capture_fps_ / decimation; + RTC_LOG(LS_WARNING) << "Target framerate, " << target_framerate + << ", is an uneven fraction of the source rate, " + << source_fps_ + << ". The framerate will be :" << effective_rate; + } + target_capture_fps_ = std::min(source_fps_, target_framerate); +} + +int FrameGeneratorCapturer::GetFrameWidth() const { + return static_cast<int>(frame_generator_->GetResolution().width); +} + +int FrameGeneratorCapturer::GetFrameHeight() const { + return static_cast<int>(frame_generator_->GetResolution().height); +} + +void FrameGeneratorCapturer::OnOutputFormatRequest( + int width, + int height, + const absl::optional<int>& max_fps) { + TestVideoCapturer::OnOutputFormatRequest(width, height, max_fps); +} + +void FrameGeneratorCapturer::SetSinkWantsObserver(SinkWantsObserver* observer) { + MutexLock lock(&lock_); + RTC_DCHECK(!sink_wants_observer_); + sink_wants_observer_ = observer; +} + +void FrameGeneratorCapturer::AddOrUpdateSink( + rtc::VideoSinkInterface<VideoFrame>* sink, + const rtc::VideoSinkWants& wants) { + TestVideoCapturer::AddOrUpdateSink(sink, wants); + MutexLock lock(&lock_); + if (sink_wants_observer_) { + // Tests need to observe unmodified sink wants. + sink_wants_observer_->OnSinkWantsChanged(sink, wants); + } + UpdateFps(GetSinkWants().max_framerate_fps); +} + +void FrameGeneratorCapturer::RemoveSink( + rtc::VideoSinkInterface<VideoFrame>* sink) { + TestVideoCapturer::RemoveSink(sink); + + MutexLock lock(&lock_); + UpdateFps(GetSinkWants().max_framerate_fps); +} + +void FrameGeneratorCapturer::UpdateFps(int max_fps) { + if (max_fps < target_capture_fps_) { + wanted_fps_.emplace(max_fps); + } else { + wanted_fps_.reset(); + } +} + +void FrameGeneratorCapturer::ForceFrame() { + // One-time non-repeating task, + task_queue_.PostTask([this] { InsertFrame(); }); +} + +int FrameGeneratorCapturer::GetCurrentConfiguredFramerate() { + MutexLock lock(&lock_); + if (wanted_fps_ && *wanted_fps_ < target_capture_fps_) + return *wanted_fps_; + return target_capture_fps_; +} + +} // namespace test +} // namespace webrtc |