/* * Copyright (c) 2016, Alliance for Open Media. All rights reserved * * This source code is subject to the terms of the BSD 2 Clause License and * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License * was not distributed with this source code in the LICENSE file, you can * obtain it at www.aomedia.org/license/software. If the Alliance for Open * Media Patent License 1.0 was not distributed with this source code in the * PATENTS file, you can obtain it at www.aomedia.org/license/patent. */ #include "config/aom_config.h" #include "third_party/googletest/src/googletest/include/gtest/gtest.h" #include "test/acm_random.h" #include "test/codec_factory.h" #include "test/datarate_test.h" #include "test/encode_test_driver.h" #include "test/i420_video_source.h" #include "test/util.h" #include "test/y4m_video_source.h" #include "aom/aom_codec.h" namespace datarate_test { namespace { // Params: test mode, speed, aq mode and index for bitrate array. class DatarateTestLarge : public ::libaom_test::CodecTestWith4Params<libaom_test::TestMode, int, unsigned int, int>, public DatarateTest { public: DatarateTestLarge() : DatarateTest(GET_PARAM(0)) { set_cpu_used_ = GET_PARAM(2); aq_mode_ = GET_PARAM(3); } protected: ~DatarateTestLarge() override = default; void SetUp() override { InitializeConfig(GET_PARAM(1)); ResetModel(); } virtual void BasicRateTargetingVBRTest() { cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.g_error_resilient = 0; cfg_.rc_end_usage = AOM_VBR; cfg_.g_lag_in_frames = 0; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 140); const int bitrate_array[2] = { 400, 800 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.7) << " The datarate for the file is lower than target by too much!"; // FIXME(jingning): Lower this test threshold after vbr mode can render // sufficiently accurate bit rate. ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.45) << " The datarate for the file is greater than target by too much!"; } virtual void BasicRateTargetingCBRTest() { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 140); const int bitrate_array[2] = { 150, 550 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.19) << " The datarate for the file is greater than target by too much!"; } virtual void BasicRateTargetingCBRSpikeTest() { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 0; cfg_.rc_min_quantizer = 2; cfg_.rc_max_quantizer = 56; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; cfg_.kf_max_dist = 3000; cfg_.kf_min_dist = 3000; ::libaom_test::I420VideoSource video("desktopqvga2.320_240.yuv", 320, 240, 30, 1, 0, 800); const int bitrate_array[2] = { 100, 200 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); max_perc_spike_ = 3.0; max_perc_spike_high_ = 8.0; ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.19) << " The datarate for the file is greater than target by too much!"; ASSERT_LE(num_spikes_, 8); ASSERT_LT(num_spikes_high_, 1); } virtual void BasicRateTargetingCBRDynamicBitrateTest() { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 0; cfg_.rc_min_quantizer = 2; cfg_.rc_max_quantizer = 56; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; cfg_.kf_max_dist = 3000; cfg_.kf_min_dist = 3000; ::libaom_test::I420VideoSource video("desktop1.320_180.yuv", 320, 180, 30, 1, 0, 800); const int bitrate_array[2] = { 100, 200 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); target_bitrate_update_[0] = cfg_.rc_target_bitrate; target_bitrate_update_[1] = static_cast<int>(1.3 * cfg_.rc_target_bitrate); target_bitrate_update_[2] = static_cast<int>(0.7 * cfg_.rc_target_bitrate); frame_update_bitrate_ = 250; ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); for (int i = 0; i < 3; i++) { ASSERT_GE(effective_datarate_dynamic_[i], target_bitrate_update_[i] * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_dynamic_[i], target_bitrate_update_[i] * 1.20) << " The datarate for the file is greater than target by too much!"; } } virtual void BasicRateTargetingMultiThreadCBRTest() { ::libaom_test::I420VideoSource video("niklas_640_480_30.yuv", 640, 480, 30, 1, 0, 400); cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; cfg_.g_threads = 4; const int bitrate_array[2] = { 250, 650 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); tile_column_ = 2; ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 0.85) << " The datarate for the file exceeds the target by too much!"; ASSERT_LE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 1.15) << " The datarate for the file missed the target!" << cfg_.rc_target_bitrate << " " << effective_datarate_; } virtual void ErrorResilienceOnSceneCuts() { if (GET_PARAM(4) > 0) return; cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 0; cfg_.g_error_resilient = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 300); cfg_.rc_target_bitrate = 500; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.15) << " The datarate for the file is greater than target by too much!"; } virtual void BasicRateTargetingCBRPeriodicKeyFrameTest() { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; // Periodic keyframe cfg_.kf_max_dist = 50; ::libaom_test::I420VideoSource video("pixel_capture_w320h240.yuv", 320, 240, 30, 1, 0, 310); const int bitrate_array[2] = { 150, 550 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.15) << " The datarate for the file is greater than target by too much!"; } virtual void CBRPeriodicKeyFrameOnSceneCuts() { if (GET_PARAM(4) > 0) return; cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 0; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; // Periodic keyframe cfg_.kf_max_dist = 30; cfg_.kf_min_dist = 30; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 300); cfg_.rc_target_bitrate = 500; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.3) << " The datarate for the file is greater than target by too much!"; } virtual void BasicRateTargetingAQModeOnOffCBRTest() { if (GET_PARAM(4) > 0) return; cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 0; cfg_.rc_min_quantizer = 2; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_lag_in_frames = 0; cfg_.g_error_resilient = 0; cfg_.g_pass = AOM_RC_ONE_PASS; cfg_.g_usage = AOM_USAGE_REALTIME; cfg_.kf_mode = AOM_KF_DISABLED; ::libaom_test::I420VideoSource video("pixel_capture_w320h240.yuv", 320, 240, 30, 1, 0, 310); cfg_.rc_target_bitrate = 60; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.15) << " The datarate for the file is greater than target by too much!"; } virtual void BasicRateTargeting444CBRScreenTest() { ::libaom_test::Y4mVideoSource video("rush_hour_444.y4m", 0, 140); cfg_.g_profile = 1; cfg_.g_timebase = video.timebase(); cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; const int bitrate_array[2] = { 250, 650 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); screen_mode_ = true; ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 0.85) << " The datarate for the file exceeds the target by too much!"; ASSERT_LE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 1.15) << " The datarate for the file missed the target!" << cfg_.rc_target_bitrate << " " << effective_datarate_; } virtual void BasicRateTargetingSuperresCBR() { ::libaom_test::I420VideoSource video("desktopqvga2.320_240.yuv", 320, 240, 30, 1, 0, 800); cfg_.g_profile = 0; cfg_.g_timebase = video.timebase(); cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.rc_superres_mode = AOM_SUPERRES_FIXED; cfg_.rc_superres_denominator = 16; cfg_.rc_superres_kf_denominator = 16; const int bitrate_array[2] = { 250, 650 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 0.85) << " The datarate for the file exceeds the target by too much!"; ASSERT_LE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 1.15) << " The datarate for the file missed the target!" << cfg_.rc_target_bitrate << " " << effective_datarate_; } virtual void BasicRateTargetingSuperresCBRMultiThreads() { ::libaom_test::I420VideoSource video("niklas_640_480_30.yuv", 640, 480, 30, 1, 0, 400); cfg_.g_profile = 0; cfg_.g_timebase = video.timebase(); cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_dropframe_thresh = 1; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 63; cfg_.rc_end_usage = AOM_CBR; cfg_.g_threads = 2; cfg_.rc_superres_mode = AOM_SUPERRES_FIXED; cfg_.rc_superres_denominator = 16; cfg_.rc_superres_kf_denominator = 16; const int bitrate_array[2] = { 250, 650 }; cfg_.rc_target_bitrate = bitrate_array[GET_PARAM(4)]; ResetModel(); tile_column_ = 1; ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 0.85) << " The datarate for the file exceeds the target by too much!"; ASSERT_LE(static_cast<double>(cfg_.rc_target_bitrate), effective_datarate_ * 1.15) << " The datarate for the file missed the target!" << cfg_.rc_target_bitrate << " " << effective_datarate_; } }; // Params: test mode, speed, aq mode. class DatarateTestFrameDropLarge : public ::libaom_test::CodecTestWith3Params<libaom_test::TestMode, int, unsigned int>, public DatarateTest { public: DatarateTestFrameDropLarge() : DatarateTest(GET_PARAM(0)) { set_cpu_used_ = GET_PARAM(2); aq_mode_ = GET_PARAM(3); } protected: ~DatarateTestFrameDropLarge() override = default; void SetUp() override { InitializeConfig(GET_PARAM(1)); ResetModel(); } virtual void ChangingDropFrameThreshTest() { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_undershoot_pct = 20; cfg_.rc_undershoot_pct = 20; cfg_.rc_dropframe_thresh = 10; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 50; cfg_.rc_end_usage = AOM_CBR; cfg_.rc_target_bitrate = 200; cfg_.g_lag_in_frames = 0; cfg_.g_error_resilient = 1; // TODO(marpan): Investigate datarate target failures with a smaller // keyframe interval (128). cfg_.kf_max_dist = 9999; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 100); const int kDropFrameThreshTestStep = 30; aom_codec_pts_t last_drop = 140; int last_num_drops = 0; for (int i = 40; i < 100; i += kDropFrameThreshTestStep) { cfg_.rc_dropframe_thresh = i; ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.85) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.40) << " The datarate for the file is greater than target by too much!"; if (last_drop > 0) { ASSERT_LE(first_drop_, last_drop) << " The first dropped frame for drop_thresh " << i << " > first dropped frame for drop_thresh " << i - kDropFrameThreshTestStep; } ASSERT_GE(num_drops_, last_num_drops * 0.7) << " The number of dropped frames for drop_thresh " << i << " < number of dropped frames for drop_thresh " << i - kDropFrameThreshTestStep; last_drop = first_drop_; last_num_drops = num_drops_; } } }; // Check basic rate targeting for VBR mode. TEST_P(DatarateTestLarge, BasicRateTargetingVBR) { BasicRateTargetingVBRTest(); } // Check basic rate targeting for CBR. TEST_P(DatarateTestLarge, BasicRateTargetingCBR) { BasicRateTargetingCBRTest(); } // Check basic rate targeting for CBR, with 4 threads TEST_P(DatarateTestLarge, BasicRateTargetingMultiThreadCBR) { BasicRateTargetingMultiThreadCBRTest(); } // Check basic rate targeting for periodic key frame. TEST_P(DatarateTestLarge, PeriodicKeyFrameCBR) { BasicRateTargetingCBRPeriodicKeyFrameTest(); } // Check basic rate targeting for periodic key frame, aligned with scene change. TEST_P(DatarateTestLarge, PeriodicKeyFrameCBROnSceneCuts) { CBRPeriodicKeyFrameOnSceneCuts(); } // Check basic rate targeting with error resilience on for scene cuts. TEST_P(DatarateTestLarge, ErrorResilienceOnSceneCuts) { ErrorResilienceOnSceneCuts(); } // Check basic rate targeting for CBR, for 444 input screen mode. #if defined(CONFIG_MAX_DECODE_PROFILE) && CONFIG_MAX_DECODE_PROFILE < 1 TEST_P(DatarateTestLarge, DISABLED_BasicRateTargeting444CBRScreen) { #else TEST_P(DatarateTestLarge, BasicRateTargeting444CBRScreen) { #endif BasicRateTargeting444CBRScreenTest(); } // Check basic rate targeting for Superres mode with CBR. TEST_P(DatarateTestLarge, BasicRateTargetingSuperresCBR) { BasicRateTargetingSuperresCBR(); } // Check basic rate targeting for Superres mode with CBR and multi-threads. TEST_P(DatarateTestLarge, BasicRateTargetingSuperresCBRMultiThreads) { BasicRateTargetingSuperresCBRMultiThreads(); } // Check that (1) the first dropped frame gets earlier and earlier // as the drop frame threshold is increased, and (2) that the total number of // frame drops does not decrease as we increase frame drop threshold. // Use a lower qp-max to force some frame drops. TEST_P(DatarateTestFrameDropLarge, ChangingDropFrameThresh) { ChangingDropFrameThreshTest(); } TEST_P(DatarateTestLarge, BasicRateTargetingAQModeOnOffCBR) { BasicRateTargetingAQModeOnOffCBRTest(); } class DatarateTestRealtime : public DatarateTestLarge {}; class DatarateTestFrameDropRealtime : public DatarateTestFrameDropLarge {}; // Params: aq mode. class DatarateTestSpeedChangeRealtime : public ::libaom_test::CodecTestWith2Params<libaom_test::TestMode, unsigned int>, public DatarateTest { public: DatarateTestSpeedChangeRealtime() : DatarateTest(GET_PARAM(0)) { aq_mode_ = GET_PARAM(1); speed_change_test_ = true; } protected: ~DatarateTestSpeedChangeRealtime() override = default; void SetUp() override { InitializeConfig(GET_PARAM(1)); ResetModel(); } virtual void ChangingSpeedTest() { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_undershoot_pct = 20; cfg_.rc_undershoot_pct = 20; cfg_.rc_dropframe_thresh = 10; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 50; cfg_.rc_end_usage = AOM_CBR; cfg_.rc_target_bitrate = 200; cfg_.g_lag_in_frames = 0; cfg_.g_error_resilient = 1; // TODO(marpan): Investigate datarate target failures with a smaller // keyframe interval (128). cfg_.kf_max_dist = 9999; cfg_.rc_dropframe_thresh = 0; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 100); ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); ASSERT_GE(effective_datarate_, cfg_.rc_target_bitrate * 0.83) << " The datarate for the file is lower than target by too much!"; ASSERT_LE(effective_datarate_, cfg_.rc_target_bitrate * 1.35) << " The datarate for the file is greater than target by too much!"; } }; // Check basic rate targeting for VBR mode. TEST_P(DatarateTestRealtime, BasicRateTargetingVBR) { BasicRateTargetingVBRTest(); } // Check basic rate targeting for CBR. TEST_P(DatarateTestRealtime, BasicRateTargetingCBR) { BasicRateTargetingCBRTest(); } // Check basic rate targeting for CBR. Use a longer clip, // and verify #encode size spikes above threshold. TEST_P(DatarateTestRealtime, BasicRateTargetingCBRSpike) { BasicRateTargetingCBRSpikeTest(); } // Check basic rate targeting for CBR. Use a longer clip, // and verify encoder can respnd and hit new bitrates updated // within the stream. TEST_P(DatarateTestRealtime, BasicRateTargetingCBRDynamicBitrate) { BasicRateTargetingCBRDynamicBitrateTest(); } // Check basic rate targeting for CBR, with 4 threads TEST_P(DatarateTestRealtime, BasicRateTargetingMultiThreadCBR) { BasicRateTargetingMultiThreadCBRTest(); } // Check basic rate targeting for periodic key frame. TEST_P(DatarateTestRealtime, PeriodicKeyFrameCBR) { BasicRateTargetingCBRPeriodicKeyFrameTest(); } // Check basic rate targeting for periodic key frame, aligned with scene change. TEST_P(DatarateTestRealtime, PeriodicKeyFrameCBROnSceneCuts) { CBRPeriodicKeyFrameOnSceneCuts(); } // Check basic rate targeting with error resilience on for scene cuts. TEST_P(DatarateTestRealtime, ErrorResilienceOnSceneCuts) { ErrorResilienceOnSceneCuts(); } // Check basic rate targeting for CBR for 444 screen mode. #if defined(CONFIG_MAX_DECODE_PROFILE) && CONFIG_MAX_DECODE_PROFILE < 1 TEST_P(DatarateTestRealtime, DISABLED_BasicRateTargeting444CBRScreen) { #else TEST_P(DatarateTestRealtime, BasicRateTargeting444CBRScreen) { #endif BasicRateTargeting444CBRScreenTest(); } // Check basic rate targeting for Superres mode with CBR. TEST_P(DatarateTestRealtime, BasicRateTargetingSuperresCBR) { BasicRateTargetingSuperresCBR(); } // Check basic rate targeting for Superres mode with CBR and multi-threads. TEST_P(DatarateTestRealtime, BasicRateTargetingSuperresCBRMultiThreads) { BasicRateTargetingSuperresCBRMultiThreads(); } // Check that (1) the first dropped frame gets earlier and earlier // as the drop frame threshold is increased, and (2) that the total number of // frame drops does not decrease as we increase frame drop threshold. // Use a lower qp-max to force some frame drops. TEST_P(DatarateTestFrameDropRealtime, ChangingDropFrameThresh) { ChangingDropFrameThreshTest(); } TEST_P(DatarateTestSpeedChangeRealtime, ChangingSpeedTest) { ChangingSpeedTest(); } class DatarateTestSetFrameQpRealtime : public DatarateTest, public ::testing::TestWithParam<const libaom_test::AV1CodecFactory *> { public: DatarateTestSetFrameQpRealtime() : DatarateTest(GetParam()), frame_(0) {} protected: ~DatarateTestSetFrameQpRealtime() override = default; void SetUp() override { InitializeConfig(libaom_test::kRealTime); ResetModel(); } void PreEncodeFrameHook(::libaom_test::VideoSource *video, ::libaom_test::Encoder *encoder) override { set_cpu_used_ = 7; DatarateTest::PreEncodeFrameHook(video, encoder); frame_qp_ = rnd_.PseudoUniform(63); encoder->Control(AV1E_SET_QUANTIZER_ONE_PASS, frame_qp_); frame_++; } void PostEncodeFrameHook(::libaom_test::Encoder *encoder) override { if (frame_ >= total_frames_) return; int qp = 0; encoder->Control(AOME_GET_LAST_QUANTIZER_64, &qp); ASSERT_EQ(qp, frame_qp_); } protected: int total_frames_; private: int frame_qp_; int frame_; libaom_test::ACMRandom rnd_; }; TEST_P(DatarateTestSetFrameQpRealtime, SetFrameQpOnePass) { cfg_.rc_buf_initial_sz = 500; cfg_.rc_buf_optimal_sz = 500; cfg_.rc_buf_sz = 1000; cfg_.rc_undershoot_pct = 20; cfg_.rc_undershoot_pct = 20; cfg_.rc_min_quantizer = 0; cfg_.rc_max_quantizer = 50; cfg_.rc_end_usage = AOM_CBR; cfg_.rc_target_bitrate = 200; cfg_.g_lag_in_frames = 0; cfg_.g_error_resilient = 1; cfg_.kf_max_dist = 9999; cfg_.rc_dropframe_thresh = 0; total_frames_ = 100; ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 30, 1, 0, 100); ResetModel(); ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); } AV1_INSTANTIATE_TEST_SUITE(DatarateTestLarge, ::testing::Values(::libaom_test::kRealTime), ::testing::Range(5, 7), ::testing::Values(0, 3), ::testing::Values(0, 1)); AV1_INSTANTIATE_TEST_SUITE(DatarateTestFrameDropLarge, ::testing::Values(::libaom_test::kRealTime), ::testing::Range(5, 7), ::testing::Values(0, 3)); AV1_INSTANTIATE_TEST_SUITE(DatarateTestRealtime, ::testing::Values(::libaom_test::kRealTime), ::testing::Range(7, 12), ::testing::Values(0, 3), ::testing::Values(0, 1)); AV1_INSTANTIATE_TEST_SUITE(DatarateTestFrameDropRealtime, ::testing::Values(::libaom_test::kRealTime), ::testing::Range(7, 12), ::testing::Values(0, 3)); AV1_INSTANTIATE_TEST_SUITE(DatarateTestSpeedChangeRealtime, ::testing::Values(::libaom_test::kRealTime), ::testing::Values(0, 3)); INSTANTIATE_TEST_SUITE_P( AV1, DatarateTestSetFrameQpRealtime, ::testing::Values( static_cast<const libaom_test::CodecFactory *>(&libaom_test::kAV1))); } // namespace } // namespace datarate_test