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 18namespace { 19 20class CpuSpeedTest : public ::libvpx_test::EncoderTest, 21 public ::libvpx_test::CodecTestWith2Params< 22 libvpx_test::TestMode, int> { 23 protected: 24 CpuSpeedTest() : EncoderTest(GET_PARAM(0)) {} 25 26 virtual void SetUp() { 27 InitializeConfig(); 28 SetMode(GET_PARAM(1)); 29 set_cpu_used_ = GET_PARAM(2); 30 } 31 32 virtual void PreEncodeFrameHook(::libvpx_test::VideoSource *video, 33 ::libvpx_test::Encoder *encoder) { 34 if (video->frame() == 1) { 35 encoder->Control(VP8E_SET_CPUUSED, set_cpu_used_); 36 encoder->Control(VP8E_SET_ENABLEAUTOALTREF, 1); 37 encoder->Control(VP8E_SET_ARNR_MAXFRAMES, 7); 38 encoder->Control(VP8E_SET_ARNR_STRENGTH, 5); 39 encoder->Control(VP8E_SET_ARNR_TYPE, 3); 40 } 41 } 42 43 virtual void FramePktHook(const vpx_codec_cx_pkt_t *pkt) { 44 if (pkt->data.frame.flags & VPX_FRAME_IS_KEY) { 45 } 46 } 47 int set_cpu_used_; 48}; 49 50TEST_P(CpuSpeedTest, TestQ0) { 51 // Validate that this non multiple of 64 wide clip encodes and decodes 52 // without a mismatch when passing in a very low max q. This pushes 53 // the encoder to producing lots of big partitions which will likely 54 // extend into the border and test the border condition. 55 cfg_.g_lag_in_frames = 25; 56 cfg_.rc_2pass_vbr_minsection_pct = 5; 57 cfg_.rc_2pass_vbr_minsection_pct = 2000; 58 cfg_.rc_target_bitrate = 400; 59 cfg_.rc_max_quantizer = 0; 60 cfg_.rc_min_quantizer = 0; 61 62 ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0, 63 20); 64 65 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 66} 67 68 69TEST_P(CpuSpeedTest, TestEncodeHighBitrate) { 70 // Validate that this non multiple of 64 wide clip encodes and decodes 71 // without a mismatch when passing in a very low max q. This pushes 72 // the encoder to producing lots of big partitions which will likely 73 // extend into the border and test the border condition. 74 cfg_.g_lag_in_frames = 25; 75 cfg_.rc_2pass_vbr_minsection_pct = 5; 76 cfg_.rc_2pass_vbr_minsection_pct = 2000; 77 cfg_.rc_target_bitrate = 12000; 78 cfg_.rc_max_quantizer = 10; 79 cfg_.rc_min_quantizer = 0; 80 81 ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0, 82 40); 83 84 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 85} 86TEST_P(CpuSpeedTest, TestLowBitrate) { 87 // Validate that this clip encodes and decodes without a mismatch 88 // when passing in a very high min q. This pushes the encoder to producing 89 // lots of small partitions which might will test the other condition. 90 91 cfg_.g_lag_in_frames = 25; 92 cfg_.rc_2pass_vbr_minsection_pct = 5; 93 cfg_.rc_2pass_vbr_minsection_pct = 2000; 94 cfg_.rc_target_bitrate = 200; 95 cfg_.rc_min_quantizer = 40; 96 97 ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0, 98 40); 99 100 ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 101} 102 103using std::tr1::make_tuple; 104 105#define VP9_FACTORY \ 106 static_cast<const libvpx_test::CodecFactory*> (&libvpx_test::kVP9) 107 108VP9_INSTANTIATE_TEST_CASE( 109 CpuSpeedTest, 110 ::testing::Values(::libvpx_test::kTwoPassGood), 111 ::testing::Range(0, 5)); 112} // namespace 113