1 /* 2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 #include <climits> 11 #include <vector> 12 #include "third_party/googletest/src/include/gtest/gtest.h" 13 #include "test/codec_factory.h" 14 #include "test/encode_test_driver.h" 15 #include "test/i420_video_source.h" 16 #include "test/util.h" 17 18 namespace { 19 20 class KeyframeTest 21 : public ::libvpx_test::EncoderTest, 22 public ::libvpx_test::CodecTestWithParam<libvpx_test::TestMode> { 23 protected: 24 KeyframeTest() : EncoderTest(GET_PARAM(0)) {} 25 virtual ~KeyframeTest() {} 26 27 virtual void SetUp() { 28 InitializeConfig(); 29 SetMode(GET_PARAM(1)); 30 kf_count_ = 0; 31 kf_count_max_ = INT_MAX; 32 kf_do_force_kf_ = false; 33 set_cpu_used_ = 0; 34 } 35 36 virtual void PreEncodeFrameHook(::libvpx_test::VideoSource *video, 37 ::libvpx_test::Encoder *encoder) { 38 if (kf_do_force_kf_) { 39 frame_flags_ = (video->frame() % 3) ? 0 : VPX_EFLAG_FORCE_KF; 40 } 41 if (set_cpu_used_ && video->frame() == 1) { 42 encoder->Control(VP8E_SET_CPUUSED, set_cpu_used_); 43 } 44 } 45 46 virtual void FramePktHook(const vpx_codec_cx_pkt_t *pkt) { 47 if (pkt->data.frame.flags & VPX_FRAME_IS_KEY) { 48 kf_pts_list_.push_back(pkt->data.frame.pts); 49 kf_count_++; 50 abort_ |= kf_count_ > kf_count_max_; 51 } 52 } 53 54 bool kf_do_force_kf_; 55 int kf_count_; 56 int kf_count_max_; 57 std::vector<vpx_codec_pts_t> kf_pts_list_; 58 int set_cpu_used_; 59 }; 60 61 TEST_P(KeyframeTest, TestRandomVideoSource) { 62 // Validate that encoding the RandomVideoSource produces multiple keyframes. 63 // This validates the results of the TestDisableKeyframes test. 64 kf_count_max_ = 2; // early exit successful tests. 65 66 ::libvpx_test::RandomVideoSource video; 67 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 68 69 // In realtime mode - auto placed keyframes are exceedingly rare, don't 70 // bother with this check if(GetParam() > 0) 71 if (GET_PARAM(1) > 0) EXPECT_GT(kf_count_, 1); 72 } 73 74 TEST_P(KeyframeTest, TestDisableKeyframes) { 75 cfg_.kf_mode = VPX_KF_DISABLED; 76 kf_count_max_ = 1; // early exit failed tests. 77 78 ::libvpx_test::RandomVideoSource video; 79 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 80 81 EXPECT_EQ(1, kf_count_); 82 } 83 84 TEST_P(KeyframeTest, TestForceKeyframe) { 85 cfg_.kf_mode = VPX_KF_DISABLED; 86 kf_do_force_kf_ = true; 87 88 ::libvpx_test::DummyVideoSource video; 89 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 90 91 // verify that every third frame is a keyframe. 92 for (std::vector<vpx_codec_pts_t>::const_iterator iter = kf_pts_list_.begin(); 93 iter != kf_pts_list_.end(); ++iter) { 94 ASSERT_EQ(0, *iter % 3) << "Unexpected keyframe at frame " << *iter; 95 } 96 } 97 98 TEST_P(KeyframeTest, TestKeyframeMaxDistance) { 99 cfg_.kf_max_dist = 25; 100 101 ::libvpx_test::DummyVideoSource video; 102 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 103 104 // verify that keyframe interval matches kf_max_dist 105 for (std::vector<vpx_codec_pts_t>::const_iterator iter = kf_pts_list_.begin(); 106 iter != kf_pts_list_.end(); ++iter) { 107 ASSERT_EQ(0, *iter % 25) << "Unexpected keyframe at frame " << *iter; 108 } 109 } 110 111 TEST_P(KeyframeTest, TestAutoKeyframe) { 112 cfg_.kf_mode = VPX_KF_AUTO; 113 kf_do_force_kf_ = false; 114 115 // Force a deterministic speed step in Real Time mode, as the faster modes 116 // may not produce a keyframe like we expect. This is necessary when running 117 // on very slow environments (like Valgrind). The step -11 was determined 118 // experimentally as the fastest mode that still throws the keyframe. 119 if (deadline_ == VPX_DL_REALTIME) set_cpu_used_ = -11; 120 121 // This clip has a cut scene every 30 frames -> Frame 0, 30, 60, 90, 120. 122 // I check only the first 40 frames to make sure there's a keyframe at frame 123 // 0 and 30. 124 ::libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 125 30, 1, 0, 40); 126 127 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 128 129 // In realtime mode - auto placed keyframes are exceedingly rare, don't 130 // bother with this check 131 if (GET_PARAM(1) > 0) 132 EXPECT_EQ(2u, kf_pts_list_.size()) << " Not the right number of keyframes "; 133 134 // Verify that keyframes match the file keyframes in the file. 135 for (std::vector<vpx_codec_pts_t>::const_iterator iter = kf_pts_list_.begin(); 136 iter != kf_pts_list_.end(); ++iter) { 137 if (deadline_ == VPX_DL_REALTIME && *iter > 0) 138 EXPECT_EQ(0, (*iter - 1) % 30) << "Unexpected keyframe at frame " 139 << *iter; 140 else 141 EXPECT_EQ(0, *iter % 30) << "Unexpected keyframe at frame " << *iter; 142 } 143 } 144 145 VP8_INSTANTIATE_TEST_CASE(KeyframeTest, ALL_TEST_MODES); 146 } // namespace 147
