| /* | |
| * Copyright (c) 2020, 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 <ostream> | |
| #include "aom/aom_codec.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" | |
| namespace { | |
| typedef struct { | |
| const unsigned int min_kf_dist; | |
| const unsigned int max_kf_dist; | |
| } kfIntervalParam; | |
| const kfIntervalParam kfTestParams[] = { | |
| { 1, 1 }, { 0, 10 }, { 10, 10 }, { 0, 30 }, { 30, 30 } | |
| }; | |
| std::ostream &operator<<(std::ostream &os, const kfIntervalParam &test_arg) { | |
| return os << "kfIntervalParam { min_kf_dist:" << test_arg.min_kf_dist | |
| << " max_kf_dist:" << test_arg.max_kf_dist << " }"; | |
| } | |
| // This class is used to test the presence of forward key frame. | |
| class KeyFrameIntervalTestLarge | |
| : public ::libaom_test::CodecTestWith3Params<libaom_test::TestMode, | |
| kfIntervalParam, aom_rc_mode>, | |
| public ::libaom_test::EncoderTest { | |
| protected: | |
| KeyFrameIntervalTestLarge() | |
| : EncoderTest(GET_PARAM(0)), encoding_mode_(GET_PARAM(1)), | |
| kf_dist_param_(GET_PARAM(2)), end_usage_check_(GET_PARAM(3)) { | |
| kf_dist_ = -1; | |
| is_kf_interval_violated_ = false; | |
| } | |
| virtual ~KeyFrameIntervalTestLarge() {} | |
| virtual void SetUp() { | |
| InitializeConfig(); | |
| SetMode(encoding_mode_); | |
| const aom_rational timebase = { 1, 30 }; | |
| cfg_.g_timebase = timebase; | |
| cfg_.rc_end_usage = end_usage_check_; | |
| cfg_.g_threads = 1; | |
| cfg_.kf_min_dist = kf_dist_param_.min_kf_dist; | |
| cfg_.kf_max_dist = kf_dist_param_.max_kf_dist; | |
| cfg_.g_lag_in_frames = 19; | |
| } | |
| virtual bool DoDecode() const { return 1; } | |
| virtual void PreEncodeFrameHook(::libaom_test::VideoSource *video, | |
| ::libaom_test::Encoder *encoder) { | |
| if (video->frame() == 0) { | |
| encoder->Control(AOME_SET_CPUUSED, 5); | |
| encoder->Control(AOME_SET_ENABLEAUTOALTREF, 1); | |
| } | |
| } | |
| virtual bool HandleDecodeResult(const aom_codec_err_t res_dec, | |
| libaom_test::Decoder *decoder) { | |
| EXPECT_EQ(AOM_CODEC_OK, res_dec) << decoder->DecodeError(); | |
| if (AOM_CODEC_OK == res_dec) { | |
| aom_codec_ctx_t *ctx_dec = decoder->GetDecoder(); | |
| int frame_flags = 0; | |
| AOM_CODEC_CONTROL_TYPECHECKED(ctx_dec, AOMD_GET_FRAME_FLAGS, | |
| &frame_flags); | |
| if (kf_dist_ != -1) { | |
| kf_dist_++; | |
| if (kf_dist_ > (int)kf_dist_param_.max_kf_dist) { | |
| is_kf_interval_violated_ = true; | |
| } | |
| } | |
| if ((frame_flags & AOM_FRAME_IS_KEY) == | |
| static_cast<aom_codec_frame_flags_t>(AOM_FRAME_IS_KEY)) { | |
| if (kf_dist_ != -1 && kf_dist_ < (int)kf_dist_param_.min_kf_dist) { | |
| is_kf_interval_violated_ = true; | |
| } | |
| kf_dist_ = 0; | |
| } | |
| } | |
| return AOM_CODEC_OK == res_dec; | |
| } | |
| ::libaom_test::TestMode encoding_mode_; | |
| const kfIntervalParam kf_dist_param_; | |
| int kf_dist_; | |
| bool is_kf_interval_violated_; | |
| aom_rc_mode end_usage_check_; | |
| }; | |
| TEST_P(KeyFrameIntervalTestLarge, KeyFrameIntervalTest) { | |
| libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, | |
| cfg_.g_timebase.den, cfg_.g_timebase.num, | |
| 0, 75); | |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); | |
| ASSERT_EQ(is_kf_interval_violated_, false) << kf_dist_param_; | |
| } | |
| // This class tests for presence and placement of application forced key frames. | |
| class ForcedKeyTestLarge | |
| : public ::libaom_test::CodecTestWith5Params<libaom_test::TestMode, int, | |
| int, int, aom_rc_mode>, | |
| public ::libaom_test::EncoderTest { | |
| protected: | |
| ForcedKeyTestLarge() | |
| : EncoderTest(GET_PARAM(0)), encoding_mode_(GET_PARAM(1)), | |
| auto_alt_ref_(GET_PARAM(2)), fwd_kf_enabled_(GET_PARAM(3)), | |
| cpu_used_(GET_PARAM(4)), rc_end_usage_(GET_PARAM(5)) { | |
| forced_kf_frame_num_ = 1; | |
| frame_num_ = 0; | |
| is_kf_placement_violated_ = false; | |
| } | |
| virtual ~ForcedKeyTestLarge() {} | |
| virtual void SetUp() { | |
| InitializeConfig(); | |
| SetMode(encoding_mode_); | |
| cfg_.rc_end_usage = rc_end_usage_; | |
| cfg_.g_threads = 0; | |
| cfg_.kf_max_dist = 30; | |
| cfg_.kf_min_dist = 0; | |
| cfg_.fwd_kf_enabled = fwd_kf_enabled_; | |
| } | |
| virtual void PreEncodeFrameHook(::libaom_test::VideoSource *video, | |
| ::libaom_test::Encoder *encoder) { | |
| if (video->frame() == 0) { | |
| encoder->Control(AOME_SET_CPUUSED, cpu_used_); | |
| encoder->Control(AOME_SET_ENABLEAUTOALTREF, auto_alt_ref_); | |
| #if CONFIG_AV1_ENCODER | |
| // override test default for tile columns if necessary. | |
| if (GET_PARAM(0) == &libaom_test::kAV1) { | |
| encoder->Control(AV1E_SET_TILE_COLUMNS, 6); | |
| } | |
| #endif | |
| } | |
| frame_flags_ = | |
| ((int)video->frame() == forced_kf_frame_num_) ? AOM_EFLAG_FORCE_KF : 0; | |
| } | |
| virtual bool HandleDecodeResult(const aom_codec_err_t res_dec, | |
| libaom_test::Decoder *decoder) { | |
| EXPECT_EQ(AOM_CODEC_OK, res_dec) << decoder->DecodeError(); | |
| if (AOM_CODEC_OK == res_dec) { | |
| if ((int)frame_num_ == forced_kf_frame_num_) { | |
| aom_codec_ctx_t *ctx_dec = decoder->GetDecoder(); | |
| int frame_flags = 0; | |
| AOM_CODEC_CONTROL_TYPECHECKED(ctx_dec, AOMD_GET_FRAME_FLAGS, | |
| &frame_flags); | |
| if ((frame_flags & AOM_FRAME_IS_KEY) != | |
| static_cast<aom_codec_frame_flags_t>(AOM_FRAME_IS_KEY)) { | |
| is_kf_placement_violated_ = true; | |
| } | |
| } | |
| ++frame_num_; | |
| } | |
| return AOM_CODEC_OK == res_dec; | |
| } | |
| ::libaom_test::TestMode encoding_mode_; | |
| int auto_alt_ref_; | |
| int fwd_kf_enabled_; | |
| int cpu_used_; | |
| aom_rc_mode rc_end_usage_; | |
| int forced_kf_frame_num_; | |
| unsigned int frame_num_; | |
| bool is_kf_placement_violated_; | |
| }; | |
| TEST_P(ForcedKeyTestLarge, Frame1IsKey) { | |
| const aom_rational timebase = { 1, 30 }; | |
| const int lag_values[] = { 3, 15, 25, -1 }; | |
| forced_kf_frame_num_ = 1; | |
| for (int i = 0; lag_values[i] != -1; ++i) { | |
| frame_num_ = 0; | |
| cfg_.g_lag_in_frames = lag_values[i]; | |
| is_kf_placement_violated_ = false; | |
| libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, | |
| timebase.den, timebase.num, 0, | |
| fwd_kf_enabled_ ? 60 : 30); | |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); | |
| ASSERT_EQ(is_kf_placement_violated_, false) | |
| << "Frame #" << frame_num_ << " isn't a keyframe!"; | |
| } | |
| } | |
| // This class checks the presence and placement of application | |
| // forced key frames. | |
| TEST_P(ForcedKeyTestLarge, ForcedFrameIsKey) { | |
| const aom_rational timebase = { 1, 30 }; | |
| const int lag_values[] = { 3, 15, 25, -1 }; | |
| for (int i = 0; lag_values[i] != -1; ++i) { | |
| frame_num_ = 0; | |
| forced_kf_frame_num_ = lag_values[i] - 1; | |
| cfg_.g_lag_in_frames = lag_values[i]; | |
| is_kf_placement_violated_ = false; | |
| libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, | |
| timebase.den, timebase.num, 0, | |
| fwd_kf_enabled_ ? 60 : 30); | |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); | |
| ASSERT_EQ(is_kf_placement_violated_, false) | |
| << "Frame #" << frame_num_ << " isn't a keyframe!"; | |
| // Two pass and single pass CBR are currently segfaulting for the case when | |
| // forced kf is placed after lag in frames. | |
| // TODO(anyone): Enable(uncomment) below test once above bug is fixed. | |
| // frame_num_ = 0; | |
| // forced_kf_frame_num_ = lag_values[i] + 1; | |
| // cfg_.g_lag_in_frames = lag_values[i]; | |
| // is_kf_placement_violated_ = false; | |
| // ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); | |
| // ASSERT_EQ(is_kf_placement_violated_, false) | |
| // << "Frame #" << frame_num_ << " isn't a keyframe!"; | |
| } | |
| } | |
| TEST_P(ForcedKeyTestLarge, ForcedFrameIsKeyCornerCases) { | |
| // TODO(aomedia:2915): Remove this if statement. | |
| if (encoding_mode_ == ::libaom_test::kTwoPassGood && auto_alt_ref_ == 1 && | |
| fwd_kf_enabled_ == 1) { | |
| std::cerr << "The test will divide by zero (crbug.com/aomedia/2915). Skip " | |
| "the test." | |
| << std::endl; | |
| return; | |
| } | |
| const aom_rational timebase = { 1, 30 }; | |
| const int kf_offsets[] = { -2, -1, 1, 2, 0 }; | |
| cfg_.g_lag_in_frames = 35; | |
| for (int i = 0; kf_offsets[i] != 0; ++i) { | |
| frame_num_ = 0; | |
| forced_kf_frame_num_ = (int)cfg_.kf_max_dist + kf_offsets[i]; | |
| forced_kf_frame_num_ = forced_kf_frame_num_ > 0 ? forced_kf_frame_num_ : 1; | |
| is_kf_placement_violated_ = false; | |
| libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, | |
| timebase.den, timebase.num, 0, | |
| fwd_kf_enabled_ ? 60 : 30); | |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); | |
| ASSERT_EQ(is_kf_placement_violated_, false) | |
| << "Frame #" << frame_num_ << " isn't a keyframe!"; | |
| } | |
| } | |
| AV1_INSTANTIATE_TEST_SUITE(KeyFrameIntervalTestLarge, | |
| testing::Values(::libaom_test::kOnePassGood, | |
| ::libaom_test::kTwoPassGood), | |
| ::testing::ValuesIn(kfTestParams), | |
| ::testing::Values(AOM_Q, AOM_VBR, AOM_CBR, AOM_CQ)); | |
| // TODO(anyone): Add CBR to list of rc_modes once forced kf placement after | |
| // lag in frames bug is fixed. | |
| AV1_INSTANTIATE_TEST_SUITE(ForcedKeyTestLarge, | |
| ::testing::Values(::libaom_test::kOnePassGood, | |
| ::libaom_test::kTwoPassGood), | |
| ::testing::Values(0, 1), ::testing::Values(0, 1), | |
| ::testing::Values(2, 5), | |
| ::testing::Values(AOM_Q, AOM_VBR, AOM_CQ)); | |
| } // namespace |