1ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org/* 2ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 3ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * 4ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Use of this source code is governed by a BSD-style license 5ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * that can be found in the LICENSE file in the root of the source 6ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * tree. An additional intellectual property rights grant can be found 7ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * in the file PATENTS. All contributing project authors may 8ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * be found in the AUTHORS file in the root of the source tree. 9ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org */ 10ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "third_party/googletest/src/include/gtest/gtest.h" 11923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/codec_factory.h" 12ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "test/encode_test_driver.h" 13ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "test/i420_video_source.h" 14923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/util.h" 15ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgnamespace { 16ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 17ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org// lookahead range: [kLookAheadMin, kLookAheadMax). 18ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgconst int kLookAheadMin = 5; 19ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgconst int kLookAheadMax = 26; 20ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 21923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.orgclass AltRefTest : public ::libvpx_test::EncoderTest, 22923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org public ::libvpx_test::CodecTestWithParam<int> { 23ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org protected: 24923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org AltRefTest() : EncoderTest(GET_PARAM(0)), altref_count_(0) {} 25ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org virtual ~AltRefTest() {} 26ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 27ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org virtual void SetUp() { 28ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org InitializeConfig(); 29ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org SetMode(libvpx_test::kTwoPassGood); 30ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 31ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 32ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org virtual void BeginPassHook(unsigned int pass) { 33ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org altref_count_ = 0; 34ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 35ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 36ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org virtual void PreEncodeFrameHook(libvpx_test::VideoSource *video, 37ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org libvpx_test::Encoder *encoder) { 38ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org if (video->frame() == 1) { 39ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org encoder->Control(VP8E_SET_ENABLEAUTOALTREF, 1); 40ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org encoder->Control(VP8E_SET_CPUUSED, 3); 41ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 42ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 43ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 44ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org virtual void FramePktHook(const vpx_codec_cx_pkt_t *pkt) { 45ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org if (pkt->data.frame.flags & VPX_FRAME_IS_INVISIBLE) ++altref_count_; 46ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 47ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 48ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int altref_count() const { return altref_count_; } 49ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 50ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org private: 51ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int altref_count_; 52ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org}; 53ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 54ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgTEST_P(AltRefTest, MonotonicTimestamps) { 55ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org const vpx_rational timebase = { 33333333, 1000000000 }; 56ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org cfg_.g_timebase = timebase; 57ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org cfg_.rc_target_bitrate = 1000; 58923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg_.g_lag_in_frames = GET_PARAM(1); 59ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 60ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 61ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org timebase.den, timebase.num, 0, 30); 62ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 63ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org EXPECT_GE(altref_count(), 1); 64ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org} 65ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 66923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 67923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.orgVP8_INSTANTIATE_TEST_CASE(AltRefTest, 68923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org ::testing::Range(kLookAheadMin, kLookAheadMax)); 69ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org} // namespace 70