147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org/* 2d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * Copyright (c) 2013 The WebM project authors. All Rights Reserved. 3d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * 4d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * Use of this source code is governed by a BSD-style license 5d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * that can be found in the LICENSE file in the root of the source 6d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * tree. An additional intellectual property rights grant can be found 7d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * in the file PATENTS. All contributing project authors may 8d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org * be found in the AUTHORS file in the root of the source tree. 9d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org */ 10dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org#include "./vpx_config.h" 1147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org#include "third_party/googletest/src/include/gtest/gtest.h" 1247265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org#include "test/codec_factory.h" 1347265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org#include "test/encode_test_driver.h" 1447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org#include "test/i420_video_source.h" 1547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org#include "test/util.h" 16dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org#include "test/y4m_video_source.h" 1747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 1847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.orgnamespace { 1947265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 2047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.orgconst int kMaxPsnr = 100; 2147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 2293a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.orgclass LosslessTestLarge : public ::libvpx_test::EncoderTest, 2347265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org public ::libvpx_test::CodecTestWithParam<libvpx_test::TestMode> { 2447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org protected: 2593a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.org LosslessTestLarge() 2693a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.org : EncoderTest(GET_PARAM(0)), 2793a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.org psnr_(kMaxPsnr), 2893a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.org nframes_(0), 2993a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.org encoding_mode_(GET_PARAM(1)) { 3047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org } 3147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 3293a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.org virtual ~LosslessTestLarge() {} 3347265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 3447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org virtual void SetUp() { 3547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org InitializeConfig(); 3647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org SetMode(encoding_mode_); 3747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org } 3847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 39e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org virtual void PreEncodeFrameHook(::libvpx_test::VideoSource *video, 40e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ::libvpx_test::Encoder *encoder) { 41e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (video->frame() == 1) { 42e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Only call Control if quantizer > 0 to verify that using quantizer 43e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // alone will activate lossless 44e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (cfg_.rc_max_quantizer > 0 || cfg_.rc_min_quantizer > 0) { 45e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org encoder->Control(VP9E_SET_LOSSLESS, 1); 46e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 47e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 48e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 49e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 5047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org virtual void BeginPassHook(unsigned int /*pass*/) { 51d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org psnr_ = kMaxPsnr; 5247265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org nframes_ = 0; 5347265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org } 5447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 5547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org virtual void PSNRPktHook(const vpx_codec_cx_pkt_t *pkt) { 5647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org if (pkt->data.psnr.psnr[0] < psnr_) 5747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org psnr_= pkt->data.psnr.psnr[0]; 5847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org } 5947265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 6047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org double GetMinPsnr() const { 6147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org return psnr_; 6247265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org } 6347265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 6447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org private: 6547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org double psnr_; 6647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org unsigned int nframes_; 6747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org libvpx_test::TestMode encoding_mode_; 6847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org}; 6947265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 7093a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.orgTEST_P(LosslessTestLarge, TestLossLessEncoding) { 7147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org const vpx_rational timebase = { 33333333, 1000000000 }; 7247265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org cfg_.g_timebase = timebase; 7347265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org cfg_.rc_target_bitrate = 2000; 7447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org cfg_.g_lag_in_frames = 25; 7547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org cfg_.rc_min_quantizer = 0; 7647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org cfg_.rc_max_quantizer = 0; 7747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 7847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org init_flags_ = VPX_CODEC_USE_PSNR; 7947265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 8047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org // intentionally changed the dimension for better testing coverage 81d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 82d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org timebase.den, timebase.num, 0, 10); 83d851b91d14ef0bd71acdce7b90c9a8f1af1181adjohannkoenig@chromium.org ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 8447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org const double psnr_lossless = GetMinPsnr(); 8547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org EXPECT_GE(psnr_lossless, kMaxPsnr); 8647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org} 87dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org 8893a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.orgTEST_P(LosslessTestLarge, TestLossLessEncoding444) { 89dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org libvpx_test::Y4mVideoSource video("rush_hour_444.y4m", 0, 10); 90dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org 91dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org cfg_.g_profile = 1; 92dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org cfg_.g_timebase = video.timebase(); 93dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org cfg_.rc_target_bitrate = 2000; 94dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org cfg_.g_lag_in_frames = 25; 95dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org cfg_.rc_min_quantizer = 0; 96dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org cfg_.rc_max_quantizer = 0; 97dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org 98dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org init_flags_ = VPX_CODEC_USE_PSNR; 99dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org 100dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 101dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org const double psnr_lossless = GetMinPsnr(); 102dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org EXPECT_GE(psnr_lossless, kMaxPsnr); 103dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org} 104dddee1ec7cedf276305b107429f684539b105276johannkoenig@chromium.org 105e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgTEST_P(LosslessTestLarge, TestLossLessEncodingCtrl) { 106e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const vpx_rational timebase = { 33333333, 1000000000 }; 107e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cfg_.g_timebase = timebase; 108e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cfg_.rc_target_bitrate = 2000; 109e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cfg_.g_lag_in_frames = 25; 110e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Intentionally set Q > 0, to make sure control can be used to activate 111e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // lossless 112e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cfg_.rc_min_quantizer = 10; 113e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org cfg_.rc_max_quantizer = 20; 114e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 115e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org init_flags_ = VPX_CODEC_USE_PSNR; 116e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 117e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, 118e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org timebase.den, timebase.num, 0, 10); 119e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 120e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org const double psnr_lossless = GetMinPsnr(); 121e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org EXPECT_GE(psnr_lossless, kMaxPsnr); 122e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 123e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 12493a74791c8e808ea76001ee07693aa2a5fdd3500johannkoenig@chromium.orgVP9_INSTANTIATE_TEST_CASE(LosslessTestLarge, ALL_TEST_MODES); 12547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org} // namespace 126