/* * Copyright (c) 2019, 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/codec_factory.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 { class DatarateTest : public ::libaom_test::EncoderTest { public: explicit DatarateTest(const ::libaom_test::CodecFactory *codec) : EncoderTest(codec), set_cpu_used_(0), aq_mode_(0), speed_change_test_(false) {} protected: ~DatarateTest() override = default; virtual void ResetModel() { last_pts_ = 0; bits_in_buffer_model_ = cfg_.rc_target_bitrate * cfg_.rc_buf_initial_sz; frame_number_ = 0; tot_frame_number_ = 0; first_drop_ = 0; num_drops_ = 0; // Denoiser is off by default. denoiser_on_ = 0; bits_total_ = 0; denoiser_offon_test_ = 0; denoiser_offon_period_ = -1; tile_column_ = 0; screen_mode_ = false; max_perc_spike_ = 1.0; max_perc_spike_high_ = 1.0; num_spikes_ = 0; num_spikes_high_ = 0; frame_update_bitrate_ = 0; for (int i = 0; i < 3; i++) { target_bitrate_update_[i] = 0; frame_number_dynamic_[i] = 0; bits_total_dynamic_[i] = 0; effective_datarate_dynamic_[i] = 0.0; } } void PreEncodeFrameHook(::libaom_test::VideoSource *video, ::libaom_test::Encoder *encoder) override { if (video->frame() == 0) { encoder->Control(AOME_SET_CPUUSED, set_cpu_used_); encoder->Control(AV1E_SET_AQ_MODE, aq_mode_); encoder->Control(AV1E_SET_TILE_COLUMNS, tile_column_); encoder->Control(AV1E_SET_ROW_MT, 1); if (cfg_.g_usage == AOM_USAGE_REALTIME) { encoder->Control(AV1E_SET_ENABLE_GLOBAL_MOTION, 0); encoder->Control(AV1E_SET_ENABLE_WARPED_MOTION, 0); encoder->Control(AV1E_SET_ENABLE_RESTORATION, 0); encoder->Control(AV1E_SET_ENABLE_OBMC, 0); encoder->Control(AV1E_SET_DELTAQ_MODE, 0); encoder->Control(AV1E_SET_ENABLE_TPL_MODEL, 0); encoder->Control(AV1E_SET_ENABLE_CDEF, 1); encoder->Control(AV1E_SET_COEFF_COST_UPD_FREQ, 2); encoder->Control(AV1E_SET_MODE_COST_UPD_FREQ, 2); encoder->Control(AV1E_SET_MV_COST_UPD_FREQ, 2); encoder->Control(AV1E_SET_DV_COST_UPD_FREQ, 2); } if (screen_mode_) { encoder->Control(AV1E_SET_TUNE_CONTENT, AOM_CONTENT_SCREEN); encoder->Control(AV1E_SET_ENABLE_PALETTE, 1); encoder->Control(AV1E_SET_ENABLE_INTRABC, 0); } } if (speed_change_test_) { if (video->frame() == 0) { encoder->Control(AOME_SET_CPUUSED, 8); } else if (video->frame() == 30) { encoder->Control(AOME_SET_CPUUSED, 7); } else if (video->frame() == 60) { encoder->Control(AOME_SET_CPUUSED, 6); } else if (video->frame() == 90) { encoder->Control(AOME_SET_CPUUSED, 7); } } if (frame_update_bitrate_ > 0) { if (frame_number_ == frame_update_bitrate_) { cfg_.rc_target_bitrate = target_bitrate_update_[1]; encoder->Config(&cfg_); } else if (frame_number_ == 2 * frame_update_bitrate_) { cfg_.rc_target_bitrate = target_bitrate_update_[2]; encoder->Config(&cfg_); } } if (denoiser_offon_test_) { ASSERT_GT(denoiser_offon_period_, 0) << "denoiser_offon_period_ is not positive."; if ((video->frame() + 1) % denoiser_offon_period_ == 0) { // Flip denoiser_on_ periodically denoiser_on_ ^= 1; } } encoder->Control(AV1E_SET_NOISE_SENSITIVITY, denoiser_on_); const aom_rational_t tb = video->timebase(); timebase_ = static_cast(tb.num) / tb.den; duration_ = 0; } void FramePktHook(const aom_codec_cx_pkt_t *pkt) override { // Time since last timestamp = duration. aom_codec_pts_t duration = pkt->data.frame.pts - last_pts_; if (duration > 1) { // If first drop not set and we have a drop set it to this time. if (!first_drop_) first_drop_ = last_pts_ + 1; // Update the number of frame drops. num_drops_ += static_cast(duration - 1); // Update counter for total number of frames (#frames input to encoder). // Needed for setting the proper layer_id below. tot_frame_number_ += static_cast(duration - 1); } // Add to the buffer the bits we'd expect from a constant bitrate server. bits_in_buffer_model_ += static_cast( duration * timebase_ * cfg_.rc_target_bitrate * 1000); // Buffer should not go negative. ASSERT_GE(bits_in_buffer_model_, 0) << "Buffer Underrun at frame " << pkt->data.frame.pts; const size_t frame_size_in_bits = pkt->data.frame.sz * 8; // Update the total encoded bits. bits_total_ += frame_size_in_bits; // Update the most recent pts. last_pts_ = pkt->data.frame.pts; ++frame_number_; ++tot_frame_number_; const int per_frame_bandwidth = (cfg_.rc_target_bitrate * 1000) / 30; if (frame_size_in_bits > max_perc_spike_ * per_frame_bandwidth && frame_number_ > 1) num_spikes_++; if (frame_size_in_bits > max_perc_spike_high_ * per_frame_bandwidth && frame_number_ > 1) num_spikes_high_++; if (frame_update_bitrate_ > 0) { if (frame_number_ < frame_update_bitrate_) { bits_total_dynamic_[0] += frame_size_in_bits; frame_number_dynamic_[0]++; } else if (frame_number_ >= frame_update_bitrate_ && frame_number_ < 2 * frame_update_bitrate_) { bits_total_dynamic_[1] += frame_size_in_bits; frame_number_dynamic_[1]++; } else { bits_total_dynamic_[2] += frame_size_in_bits; frame_number_dynamic_[2]++; } } } void EndPassHook() override { duration_ = (last_pts_ + 1) * timebase_; // Effective file datarate: effective_datarate_ = (bits_total_ / 1000.0) / duration_; if (frame_update_bitrate_ > 0) { for (int i = 0; i < 3; i++) effective_datarate_dynamic_[i] = 30 * (bits_total_dynamic_[i] / 1000.0) / frame_number_dynamic_[i]; } } aom_codec_pts_t last_pts_; double timebase_; int frame_number_; // Counter for number of non-dropped/encoded frames. int tot_frame_number_; // Counter for total number of input frames. int64_t bits_total_; double duration_; double effective_datarate_; int set_cpu_used_; int64_t bits_in_buffer_model_; aom_codec_pts_t first_drop_; int num_drops_; int denoiser_on_; int denoiser_offon_test_; int denoiser_offon_period_; unsigned int aq_mode_; bool speed_change_test_; int tile_column_; bool screen_mode_; double max_perc_spike_; double max_perc_spike_high_; int num_spikes_; int num_spikes_high_; // These are use for test with dynamic bitrate change. // Used to verify that the encoder can respond and hit bitrate that is updated // during the sequence. int frame_update_bitrate_; int target_bitrate_update_[3]; double effective_datarate_dynamic_[3]; int64_t bits_total_dynamic_[3]; int frame_number_dynamic_[3]; }; } // namespace } // namespace datarate_test