diff options
Diffstat (limited to 'third_party/libwebrtc/test/test_video_capturer.cc')
-rw-r--r-- | third_party/libwebrtc/test/test_video_capturer.cc | 107 |
1 files changed, 107 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/test_video_capturer.cc b/third_party/libwebrtc/test/test_video_capturer.cc new file mode 100644 index 0000000000..4a4adc61d7 --- /dev/null +++ b/third_party/libwebrtc/test/test_video_capturer.cc @@ -0,0 +1,107 @@ +/* + * Copyright (c) 2017 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/test_video_capturer.h" + +#include <algorithm> + +#include "api/scoped_refptr.h" +#include "api/video/i420_buffer.h" +#include "api/video/video_frame_buffer.h" +#include "api/video/video_rotation.h" + +namespace webrtc { +namespace test { +TestVideoCapturer::~TestVideoCapturer() = default; + +void TestVideoCapturer::OnOutputFormatRequest( + int width, + int height, + const absl::optional<int>& max_fps) { + absl::optional<std::pair<int, int>> target_aspect_ratio = + std::make_pair(width, height); + absl::optional<int> max_pixel_count = width * height; + video_adapter_.OnOutputFormatRequest(target_aspect_ratio, max_pixel_count, + max_fps); +} + +void TestVideoCapturer::OnFrame(const VideoFrame& original_frame) { + int cropped_width = 0; + int cropped_height = 0; + int out_width = 0; + int out_height = 0; + + VideoFrame frame = MaybePreprocess(original_frame); + + if (!video_adapter_.AdaptFrameResolution( + frame.width(), frame.height(), frame.timestamp_us() * 1000, + &cropped_width, &cropped_height, &out_width, &out_height)) { + // Drop frame in order to respect frame rate constraint. + return; + } + + if (out_height != frame.height() || out_width != frame.width()) { + // Video adapter has requested a down-scale. Allocate a new buffer and + // return scaled version. + // For simplicity, only scale here without cropping. + rtc::scoped_refptr<I420Buffer> scaled_buffer = + I420Buffer::Create(out_width, out_height); + scaled_buffer->ScaleFrom(*frame.video_frame_buffer()->ToI420()); + VideoFrame::Builder new_frame_builder = + VideoFrame::Builder() + .set_video_frame_buffer(scaled_buffer) + .set_rotation(kVideoRotation_0) + .set_timestamp_us(frame.timestamp_us()) + .set_id(frame.id()); + if (frame.has_update_rect()) { + VideoFrame::UpdateRect new_rect = frame.update_rect().ScaleWithFrame( + frame.width(), frame.height(), 0, 0, frame.width(), frame.height(), + out_width, out_height); + new_frame_builder.set_update_rect(new_rect); + } + broadcaster_.OnFrame(new_frame_builder.build()); + + } else { + // No adaptations needed, just return the frame as is. + broadcaster_.OnFrame(frame); + } +} + +rtc::VideoSinkWants TestVideoCapturer::GetSinkWants() { + return broadcaster_.wants(); +} + +void TestVideoCapturer::AddOrUpdateSink( + rtc::VideoSinkInterface<VideoFrame>* sink, + const rtc::VideoSinkWants& wants) { + broadcaster_.AddOrUpdateSink(sink, wants); + UpdateVideoAdapter(); +} + +void TestVideoCapturer::RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) { + broadcaster_.RemoveSink(sink); + UpdateVideoAdapter(); +} + +void TestVideoCapturer::UpdateVideoAdapter() { + video_adapter_.OnSinkWants(broadcaster_.wants()); +} + +VideoFrame TestVideoCapturer::MaybePreprocess(const VideoFrame& frame) { + MutexLock lock(&lock_); + if (preprocessor_ != nullptr) { + return preprocessor_->Preprocess(frame); + } else { + return frame; + } +} + +} // namespace test +} // namespace webrtc |