vp9_lossless_test.cc revision b08e2e23eec181e9951df33cd704ac294c5407b6
191037db265ecdd914a26e056cf69207b4f50924ehkuang/* 2b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * Copyright (c) 2013 The WebM project authors. All Rights Reserved. 3b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * 4b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * Use of this source code is governed by a BSD-style license 5b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * that can be found in the LICENSE file in the root of the source 6b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * tree. An additional intellectual property rights grant can be found 7b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * in the file PATENTS. All contributing project authors may 8b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian * be found in the AUTHORS file in the root of the source tree. 9b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian */ 10b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian#include "./vpx_config.h" 1191037db265ecdd914a26e056cf69207b4f50924ehkuang#include "third_party/googletest/src/include/gtest/gtest.h" 1291037db265ecdd914a26e056cf69207b4f50924ehkuang#include "test/codec_factory.h" 1391037db265ecdd914a26e056cf69207b4f50924ehkuang#include "test/encode_test_driver.h" 1491037db265ecdd914a26e056cf69207b4f50924ehkuang#include "test/i420_video_source.h" 1591037db265ecdd914a26e056cf69207b4f50924ehkuang#include "test/util.h" 16b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian#include "test/y4m_video_source.h" 1791037db265ecdd914a26e056cf69207b4f50924ehkuang 1891037db265ecdd914a26e056cf69207b4f50924ehkuangnamespace { 1991037db265ecdd914a26e056cf69207b4f50924ehkuang 2091037db265ecdd914a26e056cf69207b4f50924ehkuangconst int kMaxPsnr = 100; 2191037db265ecdd914a26e056cf69207b4f50924ehkuang 2291037db265ecdd914a26e056cf69207b4f50924ehkuangclass LossLessTest : public ::libvpx_test::EncoderTest, 2391037db265ecdd914a26e056cf69207b4f50924ehkuang public ::libvpx_test::CodecTestWithParam<libvpx_test::TestMode> { 2491037db265ecdd914a26e056cf69207b4f50924ehkuang protected: 2591037db265ecdd914a26e056cf69207b4f50924ehkuang LossLessTest() : EncoderTest(GET_PARAM(0)), 2691037db265ecdd914a26e056cf69207b4f50924ehkuang psnr_(kMaxPsnr), 2791037db265ecdd914a26e056cf69207b4f50924ehkuang nframes_(0), 2891037db265ecdd914a26e056cf69207b4f50924ehkuang encoding_mode_(GET_PARAM(1)) { 2991037db265ecdd914a26e056cf69207b4f50924ehkuang } 3091037db265ecdd914a26e056cf69207b4f50924ehkuang 3191037db265ecdd914a26e056cf69207b4f50924ehkuang virtual ~LossLessTest() {} 3291037db265ecdd914a26e056cf69207b4f50924ehkuang 3391037db265ecdd914a26e056cf69207b4f50924ehkuang virtual void SetUp() { 3491037db265ecdd914a26e056cf69207b4f50924ehkuang InitializeConfig(); 3591037db265ecdd914a26e056cf69207b4f50924ehkuang SetMode(encoding_mode_); 3691037db265ecdd914a26e056cf69207b4f50924ehkuang } 3791037db265ecdd914a26e056cf69207b4f50924ehkuang 3891037db265ecdd914a26e056cf69207b4f50924ehkuang virtual void BeginPassHook(unsigned int /*pass*/) { 399b35249446b07f40ac5fcc3205f2c048616efacchkuang psnr_ = kMaxPsnr; 4091037db265ecdd914a26e056cf69207b4f50924ehkuang nframes_ = 0; 4191037db265ecdd914a26e056cf69207b4f50924ehkuang } 4291037db265ecdd914a26e056cf69207b4f50924ehkuang 4391037db265ecdd914a26e056cf69207b4f50924ehkuang virtual void PSNRPktHook(const vpx_codec_cx_pkt_t *pkt) { 4491037db265ecdd914a26e056cf69207b4f50924ehkuang if (pkt->data.psnr.psnr[0] < psnr_) 4591037db265ecdd914a26e056cf69207b4f50924ehkuang psnr_= pkt->data.psnr.psnr[0]; 4691037db265ecdd914a26e056cf69207b4f50924ehkuang } 4791037db265ecdd914a26e056cf69207b4f50924ehkuang 4891037db265ecdd914a26e056cf69207b4f50924ehkuang double GetMinPsnr() const { 4991037db265ecdd914a26e056cf69207b4f50924ehkuang return psnr_; 5091037db265ecdd914a26e056cf69207b4f50924ehkuang } 5191037db265ecdd914a26e056cf69207b4f50924ehkuang 5291037db265ecdd914a26e056cf69207b4f50924ehkuang private: 5391037db265ecdd914a26e056cf69207b4f50924ehkuang double psnr_; 5491037db265ecdd914a26e056cf69207b4f50924ehkuang unsigned int nframes_; 5591037db265ecdd914a26e056cf69207b4f50924ehkuang libvpx_test::TestMode encoding_mode_; 5691037db265ecdd914a26e056cf69207b4f50924ehkuang}; 5791037db265ecdd914a26e056cf69207b4f50924ehkuang 5891037db265ecdd914a26e056cf69207b4f50924ehkuangTEST_P(LossLessTest, TestLossLessEncoding) { 5991037db265ecdd914a26e056cf69207b4f50924ehkuang const vpx_rational timebase = { 33333333, 1000000000 }; 6091037db265ecdd914a26e056cf69207b4f50924ehkuang cfg_.g_timebase = timebase; 6191037db265ecdd914a26e056cf69207b4f50924ehkuang cfg_.rc_target_bitrate = 2000; 6291037db265ecdd914a26e056cf69207b4f50924ehkuang cfg_.g_lag_in_frames = 25; 6391037db265ecdd914a26e056cf69207b4f50924ehkuang cfg_.rc_min_quantizer = 0; 6491037db265ecdd914a26e056cf69207b4f50924ehkuang cfg_.rc_max_quantizer = 0; 6591037db265ecdd914a26e056cf69207b4f50924ehkuang 6691037db265ecdd914a26e056cf69207b4f50924ehkuang init_flags_ = VPX_CODEC_USE_PSNR; 6791037db265ecdd914a26e056cf69207b4f50924ehkuang 6891037db265ecdd914a26e056cf69207b4f50924ehkuang // intentionally changed the dimension for better testing coverage 699b35249446b07f40ac5fcc3205f2c048616efacchkuang libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 70b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian timebase.den, timebase.num, 0, 10); 71b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 72b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian const double psnr_lossless = GetMinPsnr(); 73b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian EXPECT_GE(psnr_lossless, kMaxPsnr); 74b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian} 75b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 76b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh VenkatasubramanianTEST_P(LossLessTest, TestLossLessEncoding444) { 77b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian libvpx_test::Y4mVideoSource video("rush_hour_444.y4m", 0, 10); 78b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 79b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian cfg_.g_profile = 1; 80b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian cfg_.g_timebase = video.timebase(); 81b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian cfg_.rc_target_bitrate = 2000; 82b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian cfg_.g_lag_in_frames = 25; 83b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian cfg_.rc_min_quantizer = 0; 84b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian cfg_.rc_max_quantizer = 0; 85b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 86b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian init_flags_ = VPX_CODEC_USE_PSNR; 87b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 889b35249446b07f40ac5fcc3205f2c048616efacchkuang ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 8991037db265ecdd914a26e056cf69207b4f50924ehkuang const double psnr_lossless = GetMinPsnr(); 9091037db265ecdd914a26e056cf69207b4f50924ehkuang EXPECT_GE(psnr_lossless, kMaxPsnr); 9191037db265ecdd914a26e056cf69207b4f50924ehkuang} 92b08e2e23eec181e9951df33cd704ac294c5407b6Vignesh Venkatasubramanian 9391037db265ecdd914a26e056cf69207b4f50924ehkuangVP9_INSTANTIATE_TEST_CASE(LossLessTest, ALL_TEST_MODES); 9491037db265ecdd914a26e056cf69207b4f50924ehkuang} // namespace 95