d970807e0c
The type rtc::scoped_refptr<T> is now part of api/. Please include it from api/scoped_refptr.h. More info: See: https://groups.google.com/forum/#!topic/discuss-webrtc/Mme2MSz4z4o. Bug: webrtc:9887, webrtc:8205 No-Try: True Change-Id: Ic6c7c81e226e59f12f7933e472f573ae097b55bf Reviewed-on: https://webrtc-review.googlesource.com/c/119041 Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#26414}
131 lines
4.3 KiB
C++
131 lines
4.3 KiB
C++
/*
|
|
* Copyright (c) 2016 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 "media/base/adapted_video_track_source.h"
|
|
|
|
#include "api/scoped_refptr.h"
|
|
#include "api/video/i420_buffer.h"
|
|
#include "api/video/video_frame_buffer.h"
|
|
#include "api/video/video_rotation.h"
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/time_utils.h"
|
|
|
|
namespace rtc {
|
|
|
|
AdaptedVideoTrackSource::AdaptedVideoTrackSource() {
|
|
thread_checker_.DetachFromThread();
|
|
}
|
|
|
|
AdaptedVideoTrackSource::AdaptedVideoTrackSource(int required_alignment)
|
|
: video_adapter_(required_alignment) {
|
|
thread_checker_.DetachFromThread();
|
|
}
|
|
AdaptedVideoTrackSource::~AdaptedVideoTrackSource() = default;
|
|
|
|
bool AdaptedVideoTrackSource::GetStats(Stats* stats) {
|
|
rtc::CritScope lock(&stats_crit_);
|
|
|
|
if (!stats_) {
|
|
return false;
|
|
}
|
|
|
|
*stats = *stats_;
|
|
return true;
|
|
}
|
|
|
|
void AdaptedVideoTrackSource::OnFrame(const webrtc::VideoFrame& frame) {
|
|
rtc::scoped_refptr<webrtc::VideoFrameBuffer> buffer(
|
|
frame.video_frame_buffer());
|
|
/* Note that this is a "best effort" approach to
|
|
wants.rotation_applied; apply_rotation_ can change from false to
|
|
true between the check of apply_rotation() and the call to
|
|
broadcaster_.OnFrame(), in which case we generate a frame with
|
|
pending rotation despite some sink with wants.rotation_applied ==
|
|
true was just added. The VideoBroadcaster enforces
|
|
synchronization for us in this case, by not passing the frame on
|
|
to sinks which don't want it. */
|
|
if (apply_rotation() && frame.rotation() != webrtc::kVideoRotation_0 &&
|
|
buffer->type() == webrtc::VideoFrameBuffer::Type::kI420) {
|
|
/* Apply pending rotation. */
|
|
webrtc::VideoFrame rotated_frame =
|
|
webrtc::VideoFrame::Builder()
|
|
.set_video_frame_buffer(webrtc::I420Buffer::Rotate(
|
|
*buffer->GetI420(), frame.rotation()))
|
|
.set_rotation(webrtc::kVideoRotation_0)
|
|
.set_timestamp_us(frame.timestamp_us())
|
|
.set_id(frame.id())
|
|
.build();
|
|
broadcaster_.OnFrame(rotated_frame);
|
|
} else {
|
|
broadcaster_.OnFrame(frame);
|
|
}
|
|
}
|
|
|
|
void AdaptedVideoTrackSource::AddOrUpdateSink(
|
|
rtc::VideoSinkInterface<webrtc::VideoFrame>* sink,
|
|
const rtc::VideoSinkWants& wants) {
|
|
RTC_DCHECK(thread_checker_.CalledOnValidThread());
|
|
|
|
broadcaster_.AddOrUpdateSink(sink, wants);
|
|
OnSinkWantsChanged(broadcaster_.wants());
|
|
}
|
|
|
|
void AdaptedVideoTrackSource::RemoveSink(
|
|
rtc::VideoSinkInterface<webrtc::VideoFrame>* sink) {
|
|
RTC_DCHECK(thread_checker_.CalledOnValidThread());
|
|
|
|
broadcaster_.RemoveSink(sink);
|
|
OnSinkWantsChanged(broadcaster_.wants());
|
|
}
|
|
|
|
bool AdaptedVideoTrackSource::apply_rotation() {
|
|
return broadcaster_.wants().rotation_applied;
|
|
}
|
|
|
|
void AdaptedVideoTrackSource::OnSinkWantsChanged(
|
|
const rtc::VideoSinkWants& wants) {
|
|
RTC_DCHECK(thread_checker_.CalledOnValidThread());
|
|
video_adapter_.OnResolutionFramerateRequest(
|
|
wants.target_pixel_count, wants.max_pixel_count, wants.max_framerate_fps);
|
|
}
|
|
|
|
bool AdaptedVideoTrackSource::AdaptFrame(int width,
|
|
int height,
|
|
int64_t time_us,
|
|
int* out_width,
|
|
int* out_height,
|
|
int* crop_width,
|
|
int* crop_height,
|
|
int* crop_x,
|
|
int* crop_y) {
|
|
{
|
|
rtc::CritScope lock(&stats_crit_);
|
|
stats_ = Stats{width, height};
|
|
}
|
|
|
|
if (!broadcaster_.frame_wanted()) {
|
|
return false;
|
|
}
|
|
|
|
if (!video_adapter_.AdaptFrameResolution(
|
|
width, height, time_us * rtc::kNumNanosecsPerMicrosec, crop_width,
|
|
crop_height, out_width, out_height)) {
|
|
broadcaster_.OnDiscardedFrame();
|
|
// VideoAdapter dropped the frame.
|
|
return false;
|
|
}
|
|
|
|
*crop_x = (width - *crop_width) / 2;
|
|
*crop_y = (height - *crop_height) / 2;
|
|
return true;
|
|
}
|
|
|
|
} // namespace rtc
|