4765070b8d
This is a mechanical change since it affects so many files. I420VideoFrame -> VideoFrame and reformatted. Rationale: in the next CL I420VideoFrame will get an indication of Pixel Format (I420 for starters) and of storage type: usually UNOWNED, could be SHMEM, and in the near future will be possibly TEXTURE. See https://codereview.chromium.org/1154153003 for the change that happened in Cr. BUG=4730, chromium:440843 R=jiayl@webrtc.org, niklas.enbom@webrtc.org, pthatcher@webrtc.org Review URL: https://webrtc-codereview.appspot.com/52629004 Cr-Commit-Position: refs/heads/master@{#9339}
121 lines
3.2 KiB
C++
121 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) 2014 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 "webrtc/modules/video_coding/utility/include/quality_scaler.h"
|
|
|
|
namespace webrtc {
|
|
|
|
static const int kMinFps = 10;
|
|
static const int kMeasureSeconds = 5;
|
|
static const int kFramedropPercentThreshold = 60;
|
|
|
|
QualityScaler::QualityScaler()
|
|
: num_samples_(0), low_qp_threshold_(-1), downscale_shift_(0),
|
|
min_width_(0), min_height_(0) {
|
|
}
|
|
|
|
void QualityScaler::Init(int low_qp_threshold) {
|
|
ClearSamples();
|
|
low_qp_threshold_ = low_qp_threshold;
|
|
}
|
|
|
|
void QualityScaler::SetMinResolution(int min_width, int min_height) {
|
|
min_width_ = min_width;
|
|
min_height_ = min_height;
|
|
}
|
|
|
|
// Report framerate(fps) to estimate # of samples.
|
|
void QualityScaler::ReportFramerate(int framerate) {
|
|
num_samples_ = static_cast<size_t>(
|
|
kMeasureSeconds * (framerate < kMinFps ? kMinFps : framerate));
|
|
}
|
|
|
|
void QualityScaler::ReportQP(int qp) {
|
|
framedrop_percent_.AddSample(0);
|
|
average_qp_.AddSample(qp);
|
|
}
|
|
|
|
void QualityScaler::ReportDroppedFrame() {
|
|
framedrop_percent_.AddSample(100);
|
|
}
|
|
|
|
QualityScaler::Resolution QualityScaler::GetScaledResolution(
|
|
const VideoFrame& frame) {
|
|
// Should be set through InitEncode -> Should be set by now.
|
|
assert(low_qp_threshold_ >= 0);
|
|
assert(num_samples_ > 0);
|
|
|
|
Resolution res;
|
|
res.width = frame.width();
|
|
res.height = frame.height();
|
|
|
|
// Update scale factor.
|
|
int avg_drop = 0;
|
|
int avg_qp = 0;
|
|
if (framedrop_percent_.GetAverage(num_samples_, &avg_drop) &&
|
|
avg_drop >= kFramedropPercentThreshold) {
|
|
AdjustScale(false);
|
|
} else if (average_qp_.GetAverage(num_samples_, &avg_qp) &&
|
|
avg_qp <= low_qp_threshold_) {
|
|
AdjustScale(true);
|
|
}
|
|
|
|
assert(downscale_shift_ >= 0);
|
|
for (int shift = downscale_shift_;
|
|
shift > 0 && res.width > 1 && res.height > 1;
|
|
--shift) {
|
|
res.width >>= 1;
|
|
res.height >>= 1;
|
|
}
|
|
|
|
// Set this limitation for VP8 HW encoder to avoid crash.
|
|
if (min_width_ > 0 && res.width * res.height < min_width_ * min_height_) {
|
|
res.width = min_width_;
|
|
res.height = min_height_;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
const VideoFrame& QualityScaler::GetScaledFrame(const VideoFrame& frame) {
|
|
Resolution res = GetScaledResolution(frame);
|
|
if (res.width == frame.width())
|
|
return frame;
|
|
|
|
scaler_.Set(frame.width(),
|
|
frame.height(),
|
|
res.width,
|
|
res.height,
|
|
kI420,
|
|
kI420,
|
|
kScaleBox);
|
|
if (scaler_.Scale(frame, &scaled_frame_) != 0)
|
|
return frame;
|
|
|
|
scaled_frame_.set_ntp_time_ms(frame.ntp_time_ms());
|
|
scaled_frame_.set_timestamp(frame.timestamp());
|
|
scaled_frame_.set_render_time_ms(frame.render_time_ms());
|
|
|
|
return scaled_frame_;
|
|
}
|
|
|
|
void QualityScaler::ClearSamples() {
|
|
framedrop_percent_.Reset();
|
|
average_qp_.Reset();
|
|
}
|
|
|
|
void QualityScaler::AdjustScale(bool up) {
|
|
downscale_shift_ += up ? -1 : 1;
|
|
if (downscale_shift_ < 0)
|
|
downscale_shift_ = 0;
|
|
ClearSamples();
|
|
}
|
|
|
|
} // namespace webrtc
|