Make quality scaler downscale faster.
Include dropped frames by the encoder in the frame drop percentage.
To react faster at low framerates:
- Use ExpFilter instead of MovingAverage to filter QP values.
- Reduce sampling interval while waiting for minimum number of needed frames (when not in fast rampup mode).
A separate slower ExpFilter is used for upscaling.
Bug: webrtc:9169
Change-Id: If7ff6c3bd4201fda2da67125889838fe96ce7061
Reviewed-on: https://webrtc-review.googlesource.com/70761
Commit-Queue: Åsa Persson <asapersson@webrtc.org>
Reviewed-by: Karl Wiberg <kwiberg@webrtc.org>
Reviewed-by: Rasmus Brandt <brandtr@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#23014}
diff --git a/modules/video_coding/BUILD.gn b/modules/video_coding/BUILD.gn
index 619f15e..b2fd91d 100644
--- a/modules/video_coding/BUILD.gn
+++ b/modules/video_coding/BUILD.gn
@@ -243,6 +243,7 @@
"../../rtc_base:rtc_numerics",
"../../rtc_base:rtc_task_queue",
"../../rtc_base:sequenced_task_checker",
+ "../../rtc_base/experiments:quality_scaling_experiment",
"../../system_wrappers",
"../rtp_rtcp:rtp_rtcp_format",
]
diff --git a/modules/video_coding/utility/quality_scaler.cc b/modules/video_coding/utility/quality_scaler.cc
index e839f9e..273d7a6 100644
--- a/modules/video_coding/utility/quality_scaler.cc
+++ b/modules/video_coding/utility/quality_scaler.cc
@@ -17,7 +17,9 @@
#include "rtc_base/checks.h"
#include "rtc_base/logging.h"
+#include "rtc_base/numerics/exp_filter.h"
#include "rtc_base/task_queue.h"
+#include "rtc_base/timeutils.h"
// TODO(kthelgason): Some versions of Android have issues with log2.
// See https://code.google.com/p/android/issues/detail?id=212634 for details
@@ -37,6 +39,33 @@
} // namespace
+class QualityScaler::QpSmoother {
+ public:
+ explicit QpSmoother(float alpha)
+ : alpha_(alpha), last_sample_ms_(rtc::TimeMillis()), smoother_(alpha) {}
+
+ rtc::Optional<int> GetAvg() const {
+ float value = smoother_.filtered();
+ if (value == rtc::ExpFilter::kValueUndefined) {
+ return rtc::nullopt;
+ }
+ return static_cast<int>(value);
+ }
+
+ void Add(float sample) {
+ int64_t now_ms = rtc::TimeMillis();
+ smoother_.Apply(static_cast<float>(now_ms - last_sample_ms_), sample);
+ last_sample_ms_ = now_ms;
+ }
+
+ void Reset() { smoother_.Reset(alpha_); }
+
+ private:
+ const float alpha_;
+ int64_t last_sample_ms_;
+ rtc::ExpFilter smoother_;
+};
+
class QualityScaler::CheckQpTask : public rtc::QueuedTask {
public:
explicit CheckQpTask(QualityScaler* scaler) : scaler_(scaler) {
@@ -81,8 +110,16 @@
fast_rampup_(true),
// Arbitrarily choose size based on 30 fps for 5 seconds.
average_qp_(5 * 30),
- framedrop_percent_(5 * 30) {
+ framedrop_percent_media_opt_(5 * 30),
+ framedrop_percent_all_(5 * 30),
+ experiment_enabled_(QualityScalingExperiment::Enabled()),
+ observed_enough_frames_(false) {
RTC_DCHECK_CALLED_SEQUENTIALLY(&task_checker_);
+ if (experiment_enabled_) {
+ config_ = QualityScalingExperiment::GetConfig();
+ qp_smoother_high_.reset(new QpSmoother(config_.alpha_high));
+ qp_smoother_low_.reset(new QpSmoother(config_.alpha_low));
+ }
RTC_DCHECK(observer_ != nullptr);
check_qp_task_ = new CheckQpTask(this);
RTC_LOG(LS_INFO) << "QP thresholds: low: " << thresholds_.low
@@ -96,19 +133,36 @@
int64_t QualityScaler::GetSamplingPeriodMs() const {
RTC_DCHECK_CALLED_SEQUENTIALLY(&task_checker_);
- return fast_rampup_ ? sampling_period_ms_
- : (sampling_period_ms_ * kSamplePeriodScaleFactor);
+ if (fast_rampup_) {
+ return sampling_period_ms_;
+ }
+ if (experiment_enabled_ && !observed_enough_frames_) {
+ // Use half the interval while waiting for enough frames.
+ return sampling_period_ms_ / 2;
+ }
+ return sampling_period_ms_ * kSamplePeriodScaleFactor;
}
-void QualityScaler::ReportDroppedFrame() {
+void QualityScaler::ReportDroppedFrameByMediaOpt() {
RTC_DCHECK_CALLED_SEQUENTIALLY(&task_checker_);
- framedrop_percent_.AddSample(100);
+ framedrop_percent_media_opt_.AddSample(100);
+ framedrop_percent_all_.AddSample(100);
+}
+
+void QualityScaler::ReportDroppedFrameByEncoder() {
+ RTC_DCHECK_CALLED_SEQUENTIALLY(&task_checker_);
+ framedrop_percent_all_.AddSample(100);
}
void QualityScaler::ReportQp(int qp) {
RTC_DCHECK_CALLED_SEQUENTIALLY(&task_checker_);
- framedrop_percent_.AddSample(0);
+ framedrop_percent_media_opt_.AddSample(0);
+ framedrop_percent_all_.AddSample(0);
average_qp_.AddSample(qp);
+ if (qp_smoother_high_)
+ qp_smoother_high_->Add(qp);
+ if (qp_smoother_low_)
+ qp_smoother_low_->Add(qp);
}
void QualityScaler::CheckQp() {
@@ -118,11 +172,19 @@
// If we have not observed at least this many frames we can't make a good
// scaling decision.
- if (framedrop_percent_.size() < kMinFramesNeededToScale)
+ const size_t frames = config_.use_all_drop_reasons
+ ? framedrop_percent_all_.size()
+ : framedrop_percent_media_opt_.size();
+ if (frames < kMinFramesNeededToScale) {
+ observed_enough_frames_ = false;
return;
+ }
+ observed_enough_frames_ = true;
// Check if we should scale down due to high frame drop.
- const rtc::Optional<int> drop_rate = framedrop_percent_.GetAverage();
+ const rtc::Optional<int> drop_rate =
+ config_.use_all_drop_reasons ? framedrop_percent_all_.GetAverage()
+ : framedrop_percent_media_opt_.GetAverage();
if (drop_rate && *drop_rate >= kFramedropPercentThreshold) {
RTC_LOG(LS_INFO) << "Reporting high QP, framedrop percent " << *drop_rate;
ReportQpHigh();
@@ -130,14 +192,19 @@
}
// Check if we should scale up or down based on QP.
- const rtc::Optional<int> avg_qp = average_qp_.GetAverage();
- if (avg_qp) {
- RTC_LOG(LS_INFO) << "Checking average QP " << *avg_qp;
- if (*avg_qp > thresholds_.high) {
+ const rtc::Optional<int> avg_qp_high = qp_smoother_high_
+ ? qp_smoother_high_->GetAvg()
+ : average_qp_.GetAverage();
+ const rtc::Optional<int> avg_qp_low =
+ qp_smoother_low_ ? qp_smoother_low_->GetAvg() : average_qp_.GetAverage();
+ if (avg_qp_high && avg_qp_low) {
+ RTC_LOG(LS_INFO) << "Checking average QP " << *avg_qp_high << " ("
+ << *avg_qp_low << ").";
+ if (*avg_qp_high > thresholds_.high) {
ReportQpHigh();
return;
}
- if (*avg_qp <= thresholds_.low) {
+ if (*avg_qp_low <= thresholds_.low) {
// QP has been low. We want to try a higher resolution.
ReportQpLow();
return;
@@ -163,7 +230,12 @@
void QualityScaler::ClearSamples() {
RTC_DCHECK_CALLED_SEQUENTIALLY(&task_checker_);
- framedrop_percent_.Reset();
+ framedrop_percent_media_opt_.Reset();
+ framedrop_percent_all_.Reset();
average_qp_.Reset();
+ if (qp_smoother_high_)
+ qp_smoother_high_->Reset();
+ if (qp_smoother_low_)
+ qp_smoother_low_->Reset();
}
} // namespace webrtc
diff --git a/modules/video_coding/utility/quality_scaler.h b/modules/video_coding/utility/quality_scaler.h
index e823461..996b9f2 100644
--- a/modules/video_coding/utility/quality_scaler.h
+++ b/modules/video_coding/utility/quality_scaler.h
@@ -11,12 +11,14 @@
#ifndef MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_
#define MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_
+#include <memory>
#include <utility>
#include "api/optional.h"
#include "api/video_codecs/video_encoder.h"
#include "common_types.h" // NOLINT(build/include)
#include "modules/video_coding/utility/moving_average.h"
+#include "rtc_base/experiments/quality_scaling_experiment.h"
#include "rtc_base/sequenced_task_checker.h"
namespace webrtc {
@@ -49,8 +51,9 @@
QualityScaler(AdaptationObserverInterface* observer,
VideoEncoder::QpThresholds thresholds);
virtual ~QualityScaler();
- // Should be called each time the encoder drops a frame.
- void ReportDroppedFrame();
+ // Should be called each time a frame is dropped at encoding.
+ void ReportDroppedFrameByMediaOpt();
+ void ReportDroppedFrameByEncoder();
// Inform the QualityScaler of the last seen QP.
void ReportQp(int qp);
@@ -62,6 +65,7 @@
private:
class CheckQpTask;
+ class QpSmoother;
void CheckQp();
void ClearSamples();
void ReportQpLow();
@@ -76,7 +80,15 @@
const int64_t sampling_period_ms_;
bool fast_rampup_ RTC_GUARDED_BY(&task_checker_);
MovingAverage average_qp_ RTC_GUARDED_BY(&task_checker_);
- MovingAverage framedrop_percent_ RTC_GUARDED_BY(&task_checker_);
+ MovingAverage framedrop_percent_media_opt_ RTC_GUARDED_BY(&task_checker_);
+ MovingAverage framedrop_percent_all_ RTC_GUARDED_BY(&task_checker_);
+
+ // Used by QualityScalingExperiment.
+ const bool experiment_enabled_;
+ QualityScalingExperiment::Config config_ RTC_GUARDED_BY(&task_checker_);
+ std::unique_ptr<QpSmoother> qp_smoother_high_ RTC_GUARDED_BY(&task_checker_);
+ std::unique_ptr<QpSmoother> qp_smoother_low_ RTC_GUARDED_BY(&task_checker_);
+ bool observed_enough_frames_ RTC_GUARDED_BY(&task_checker_);
};
} // namespace webrtc
diff --git a/modules/video_coding/utility/quality_scaler_unittest.cc b/modules/video_coding/utility/quality_scaler_unittest.cc
index 824826d..58a381c 100644
--- a/modules/video_coding/utility/quality_scaler_unittest.cc
+++ b/modules/video_coding/utility/quality_scaler_unittest.cc
@@ -11,9 +11,11 @@
#include "modules/video_coding/utility/quality_scaler.h"
#include <memory>
+#include <string>
#include "rtc_base/event.h"
#include "rtc_base/task_queue.h"
+#include "test/field_trial.h"
#include "test/gmock.h"
#include "test/gtest.h"
@@ -63,7 +65,8 @@
: QualityScaler(observer, thresholds, 5) {}
};
-class QualityScalerTest : public ::testing::Test {
+class QualityScalerTest : public ::testing::Test,
+ public ::testing::WithParamInterface<std::string> {
protected:
enum ScaleDirection {
kKeepScaleAboveLowQp,
@@ -74,7 +77,8 @@
};
QualityScalerTest()
- : q_(new rtc::TaskQueue("QualityScalerTestQueue")),
+ : scoped_field_trial_(GetParam()),
+ q_(new rtc::TaskQueue("QualityScalerTestQueue")),
observer_(new MockAdaptationObserver()) {
DO_SYNC(q_, {
qs_ = std::unique_ptr<QualityScaler>(new QualityScalerUnderTest(
@@ -96,7 +100,7 @@
qs_->ReportQp(kLowQp);
break;
case kScaleDown:
- qs_->ReportDroppedFrame();
+ qs_->ReportDroppedFrameByMediaOpt();
break;
case kKeepScaleAtHighQp:
qs_->ReportQp(kHighQp);
@@ -108,37 +112,45 @@
}
}
+ test::ScopedFieldTrials scoped_field_trial_;
std::unique_ptr<rtc::TaskQueue> q_;
std::unique_ptr<QualityScaler> qs_;
std::unique_ptr<MockAdaptationObserver> observer_;
};
-TEST_F(QualityScalerTest, DownscalesAfterContinuousFramedrop) {
+INSTANTIATE_TEST_CASE_P(
+ FieldTrials,
+ QualityScalerTest,
+ ::testing::Values(
+ "WebRTC-Video-QualityScaling/Enabled-1,2,3,4,5,6,7,8,0.9,0.99,1/",
+ ""));
+
+TEST_P(QualityScalerTest, DownscalesAfterContinuousFramedrop) {
DO_SYNC(q_, { TriggerScale(kScaleDown); });
EXPECT_TRUE(observer_->event.Wait(kDefaultTimeoutMs));
EXPECT_EQ(1, observer_->adapt_down_events_);
EXPECT_EQ(0, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, KeepsScaleAtHighQp) {
+TEST_P(QualityScalerTest, KeepsScaleAtHighQp) {
DO_SYNC(q_, { TriggerScale(kKeepScaleAtHighQp); });
EXPECT_FALSE(observer_->event.Wait(kDefaultTimeoutMs));
EXPECT_EQ(0, observer_->adapt_down_events_);
EXPECT_EQ(0, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, DownscalesAboveHighQp) {
+TEST_P(QualityScalerTest, DownscalesAboveHighQp) {
DO_SYNC(q_, { TriggerScale(kScaleDownAboveHighQp); });
EXPECT_TRUE(observer_->event.Wait(kDefaultTimeoutMs));
EXPECT_EQ(1, observer_->adapt_down_events_);
EXPECT_EQ(0, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, DownscalesAfterTwoThirdsFramedrop) {
+TEST_P(QualityScalerTest, DownscalesAfterTwoThirdsFramedrop) {
DO_SYNC(q_, {
for (int i = 0; i < kFramerate * 5; ++i) {
- qs_->ReportDroppedFrame();
- qs_->ReportDroppedFrame();
+ qs_->ReportDroppedFrameByMediaOpt();
+ qs_->ReportDroppedFrameByMediaOpt();
qs_->ReportQp(kHighQp);
}
});
@@ -147,10 +159,10 @@
EXPECT_EQ(0, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, DoesNotDownscaleAfterHalfFramedrop) {
+TEST_P(QualityScalerTest, DoesNotDownscaleAfterHalfFramedrop) {
DO_SYNC(q_, {
for (int i = 0; i < kFramerate * 5; ++i) {
- qs_->ReportDroppedFrame();
+ qs_->ReportDroppedFrameByMediaOpt();
qs_->ReportQp(kHighQp);
}
});
@@ -159,21 +171,35 @@
EXPECT_EQ(0, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, KeepsScaleOnNormalQp) {
+TEST_P(QualityScalerTest, DownscalesAfterTwoThirdsIfFieldTrialEnabled) {
+ const bool kDownScaleExpected = !GetParam().empty();
+ DO_SYNC(q_, {
+ for (int i = 0; i < kFramerate * 5; ++i) {
+ qs_->ReportDroppedFrameByMediaOpt();
+ qs_->ReportDroppedFrameByEncoder();
+ qs_->ReportQp(kHighQp);
+ }
+ });
+ EXPECT_EQ(kDownScaleExpected, observer_->event.Wait(kDefaultTimeoutMs));
+ EXPECT_EQ(kDownScaleExpected ? 1 : 0, observer_->adapt_down_events_);
+ EXPECT_EQ(0, observer_->adapt_up_events_);
+}
+
+TEST_P(QualityScalerTest, KeepsScaleOnNormalQp) {
DO_SYNC(q_, { TriggerScale(kKeepScaleAboveLowQp); });
EXPECT_FALSE(observer_->event.Wait(kDefaultTimeoutMs));
EXPECT_EQ(0, observer_->adapt_down_events_);
EXPECT_EQ(0, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, UpscalesAfterLowQp) {
+TEST_P(QualityScalerTest, UpscalesAfterLowQp) {
DO_SYNC(q_, { TriggerScale(kScaleUp); });
EXPECT_TRUE(observer_->event.Wait(kDefaultTimeoutMs));
EXPECT_EQ(0, observer_->adapt_down_events_);
EXPECT_EQ(1, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, ScalesDownAndBackUp) {
+TEST_P(QualityScalerTest, ScalesDownAndBackUp) {
DO_SYNC(q_, { TriggerScale(kScaleDown); });
EXPECT_TRUE(observer_->event.Wait(kDefaultTimeoutMs));
EXPECT_EQ(1, observer_->adapt_down_events_);
@@ -184,7 +210,7 @@
EXPECT_EQ(1, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, DoesNotScaleUntilEnoughFramesObserved) {
+TEST_P(QualityScalerTest, DoesNotScaleUntilEnoughFramesObserved) {
DO_SYNC(q_, {
// Not enough frames to make a decision.
for (int i = 0; i < kMinFramesNeededToScale - 1; ++i) {
@@ -210,7 +236,7 @@
EXPECT_EQ(1, observer_->adapt_up_events_);
}
-TEST_F(QualityScalerTest, ScalesDownAndBackUpWithMinFramesNeeded) {
+TEST_P(QualityScalerTest, ScalesDownAndBackUpWithMinFramesNeeded) {
DO_SYNC(q_, {
for (int i = 0; i < kMinFramesNeededToScale; ++i) {
qs_->ReportQp(kHighQp + 1);