James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 1 | /* |
Yaowu Xu | bde4ac8 | 2016-11-28 15:26:06 -0800 | [diff] [blame^] | 2 | * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 3 | * |
Yaowu Xu | bde4ac8 | 2016-11-28 15:26:06 -0800 | [diff] [blame^] | 4 | * This source code is subject to the terms of the BSD 2 Clause License and |
| 5 | * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| 6 | * was not distributed with this source code in the LICENSE file, you can |
| 7 | * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
| 8 | * Media Patent License 1.0 was not distributed with this source code in the |
| 9 | * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 10 | */ |
Yaowu Xu | bde4ac8 | 2016-11-28 15:26:06 -0800 | [diff] [blame^] | 11 | |
James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 12 | #include "third_party/googletest/src/include/gtest/gtest.h" |
John Koleszar | 706cafe | 2013-01-18 11:51:12 -0800 | [diff] [blame] | 13 | #include "test/codec_factory.h" |
James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 14 | #include "test/encode_test_driver.h" |
| 15 | #include "test/i420_video_source.h" |
John Koleszar | 706cafe | 2013-01-18 11:51:12 -0800 | [diff] [blame] | 16 | #include "test/util.h" |
James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 17 | namespace { |
| 18 | |
James Zern | 0dcd21f | 2016-03-17 18:46:10 -0700 | [diff] [blame] | 19 | class AltRefForcedKeyTestLarge |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 20 | : public ::libaom_test::EncoderTest, |
| 21 | public ::libaom_test::CodecTestWith2Params<libaom_test::TestMode, int> { |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 22 | protected: |
James Zern | 0dcd21f | 2016-03-17 18:46:10 -0700 | [diff] [blame] | 23 | AltRefForcedKeyTestLarge() |
clang-format | 3a826f1 | 2016-08-11 17:46:05 -0700 | [diff] [blame] | 24 | : EncoderTest(GET_PARAM(0)), encoding_mode_(GET_PARAM(1)), |
| 25 | cpu_used_(GET_PARAM(2)), forced_kf_frame_num_(1), frame_num_(0) {} |
James Zern | 0dcd21f | 2016-03-17 18:46:10 -0700 | [diff] [blame] | 26 | virtual ~AltRefForcedKeyTestLarge() {} |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 27 | |
| 28 | virtual void SetUp() { |
| 29 | InitializeConfig(); |
| 30 | SetMode(encoding_mode_); |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 31 | cfg_.rc_end_usage = AOM_VBR; |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 32 | cfg_.g_threads = 0; |
| 33 | } |
| 34 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 35 | virtual void PreEncodeFrameHook(::libaom_test::VideoSource *video, |
| 36 | ::libaom_test::Encoder *encoder) { |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 37 | if (video->frame() == 0) { |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 38 | encoder->Control(AOME_SET_CPUUSED, cpu_used_); |
| 39 | encoder->Control(AOME_SET_ENABLEAUTOALTREF, 1); |
| 40 | #if CONFIG_AV1_ENCODER |
clang-format | 3a826f1 | 2016-08-11 17:46:05 -0700 | [diff] [blame] | 41 | // override test default for tile columns if necessary. |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 42 | if (GET_PARAM(0) == &libaom_test::kAV1) { |
| 43 | encoder->Control(AV1E_SET_TILE_COLUMNS, 6); |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 44 | } |
| 45 | #endif |
| 46 | } |
| 47 | frame_flags_ = |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 48 | (video->frame() == forced_kf_frame_num_) ? AOM_EFLAG_FORCE_KF : 0; |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 49 | } |
| 50 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 51 | virtual void FramePktHook(const aom_codec_cx_pkt_t *pkt) { |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 52 | if (frame_num_ == forced_kf_frame_num_) { |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 53 | ASSERT_TRUE(!!(pkt->data.frame.flags & AOM_FRAME_IS_KEY)) |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 54 | << "Frame #" << frame_num_ << " isn't a keyframe!"; |
| 55 | } |
| 56 | ++frame_num_; |
| 57 | } |
| 58 | |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 59 | ::libaom_test::TestMode encoding_mode_; |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 60 | int cpu_used_; |
| 61 | unsigned int forced_kf_frame_num_; |
| 62 | unsigned int frame_num_; |
| 63 | }; |
| 64 | |
James Zern | 0dcd21f | 2016-03-17 18:46:10 -0700 | [diff] [blame] | 65 | TEST_P(AltRefForcedKeyTestLarge, Frame1IsKey) { |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 66 | const aom_rational timebase = { 1, 30 }; |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 67 | const int lag_values[] = { 3, 15, 25, -1 }; |
| 68 | |
| 69 | forced_kf_frame_num_ = 1; |
| 70 | for (int i = 0; lag_values[i] != -1; ++i) { |
| 71 | frame_num_ = 0; |
| 72 | cfg_.g_lag_in_frames = lag_values[i]; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 73 | libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 74 | timebase.den, timebase.num, 0, 30); |
| 75 | ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| 76 | } |
| 77 | } |
| 78 | |
James Zern | 0dcd21f | 2016-03-17 18:46:10 -0700 | [diff] [blame] | 79 | TEST_P(AltRefForcedKeyTestLarge, ForcedFrameIsKey) { |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 80 | const aom_rational timebase = { 1, 30 }; |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 81 | const int lag_values[] = { 3, 15, 25, -1 }; |
| 82 | |
| 83 | for (int i = 0; lag_values[i] != -1; ++i) { |
| 84 | frame_num_ = 0; |
| 85 | forced_kf_frame_num_ = lag_values[i] - 1; |
| 86 | cfg_.g_lag_in_frames = lag_values[i]; |
Yaowu Xu | c27fc14 | 2016-08-22 16:08:15 -0700 | [diff] [blame] | 87 | libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 88 | timebase.den, timebase.num, 0, 30); |
| 89 | ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| 90 | } |
| 91 | } |
John Koleszar | 706cafe | 2013-01-18 11:51:12 -0800 | [diff] [blame] | 92 | |
Yaowu Xu | f883b42 | 2016-08-30 14:01:10 -0700 | [diff] [blame] | 93 | AV1_INSTANTIATE_TEST_CASE(AltRefForcedKeyTestLarge, |
| 94 | ::testing::Values(::libaom_test::kOnePassGood), |
| 95 | ::testing::Range(0, 9)); |
James Zern | ac4c37c | 2016-02-19 11:05:33 -0800 | [diff] [blame] | 96 | |
James Zern | 429743c | 2012-08-07 17:12:10 -0700 | [diff] [blame] | 97 | } // namespace |