1923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@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. 9923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org */ 10923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 11923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include <cstdio> 12923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include <cstdlib> 13923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include <string> 14923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "third_party/googletest/src/include/gtest/gtest.h" 15923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/codec_factory.h" 16923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/encode_test_driver.h" 17923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/i420_video_source.h" 18923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/util.h" 19923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "test/md5_helper.h" 20923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org#include "vpx_mem/vpx_mem.h" 21923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 22923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.orgnamespace { 23923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.orgclass TileIndependenceTest : public ::libvpx_test::EncoderTest, 2447265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org public ::libvpx_test::CodecTestWithParam<int> { 25923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org protected: 2647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org TileIndependenceTest() 2747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org : EncoderTest(GET_PARAM(0)), 2847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org md5_fw_order_(), 2947265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org md5_inv_order_(), 3047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org n_tiles_(GET_PARAM(1)) { 31923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org init_flags_ = VPX_CODEC_USE_PSNR; 32d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org vpx_codec_dec_cfg_t cfg = vpx_codec_dec_cfg_t(); 33923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg.w = 704; 34923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg.h = 144; 35923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg.threads = 1; 36923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org fw_dec_ = codec_->CreateDecoder(cfg, 0); 37923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org inv_dec_ = codec_->CreateDecoder(cfg, 0); 38923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org inv_dec_->Control(VP9_INVERT_TILE_DECODE_ORDER, 1); 39923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 40923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 41923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org virtual ~TileIndependenceTest() { 42923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org delete fw_dec_; 43923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org delete inv_dec_; 44923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 45923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 46923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org virtual void SetUp() { 47923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org InitializeConfig(); 48923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org SetMode(libvpx_test::kTwoPassGood); 49923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 50923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 51923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org virtual void PreEncodeFrameHook(libvpx_test::VideoSource *video, 52923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org libvpx_test::Encoder *encoder) { 53923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org if (video->frame() == 1) { 54923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org encoder->Control(VP9E_SET_TILE_COLUMNS, n_tiles_); 55923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 56923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 57923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 58923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org void UpdateMD5(::libvpx_test::Decoder *dec, const vpx_codec_cx_pkt_t *pkt, 59923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org ::libvpx_test::MD5 *md5) { 6047265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org const vpx_codec_err_t res = dec->DecodeFrame( 6147265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org reinterpret_cast<uint8_t*>(pkt->data.frame.buf), pkt->data.frame.sz); 6210a9a0d835561a7f2300c561c514efcf374554d6fgalligan@chromium.org if (res != VPX_CODEC_OK) { 6310a9a0d835561a7f2300c561c514efcf374554d6fgalligan@chromium.org abort_ = true; 6410a9a0d835561a7f2300c561c514efcf374554d6fgalligan@chromium.org ASSERT_EQ(VPX_CODEC_OK, res); 6510a9a0d835561a7f2300c561c514efcf374554d6fgalligan@chromium.org } 66923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org const vpx_image_t *img = dec->GetDxData().Next(); 67923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org md5->Add(img); 68923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 69923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 70923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org virtual void FramePktHook(const vpx_codec_cx_pkt_t *pkt) { 71923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org UpdateMD5(fw_dec_, pkt, &md5_fw_order_); 72923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org UpdateMD5(inv_dec_, pkt, &md5_inv_order_); 73923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org } 74923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 75923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org ::libvpx_test::MD5 md5_fw_order_, md5_inv_order_; 76923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org ::libvpx_test::Decoder *fw_dec_, *inv_dec_; 7747265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org 7847265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org private: 7947265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org int n_tiles_; 80923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org}; 81923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 82923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org// run an encode with 2 or 4 tiles, and do the decode both in normal and 83923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org// inverted tile ordering. Ensure that the MD5 of the output in both cases 84923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org// is identical. If so, tiles are considered independent and the test passes. 85923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.orgTEST_P(TileIndependenceTest, MD5Match) { 86923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org const vpx_rational timebase = { 33333333, 1000000000 }; 87923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg_.g_timebase = timebase; 88923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg_.rc_target_bitrate = 500; 89923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg_.g_lag_in_frames = 25; 90923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org cfg_.rc_end_usage = VPX_VBR; 91923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 92923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 704, 144, 93923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org timebase.den, timebase.num, 0, 30); 94923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); 95923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 9647265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.org const char *md5_fw_str = md5_fw_order_.Get(); 97923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org const char *md5_inv_str = md5_inv_order_.Get(); 98923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 99923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org // could use ASSERT_EQ(!memcmp(.., .., 16) here, but this gives nicer 100923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org // output if it fails. Not sure if it's helpful since it's really just 101923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org // a MD5... 102923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org ASSERT_STREQ(md5_fw_str, md5_inv_str); 103923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org} 104923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 10547265f8fe3a36a426773454ad90d20c9aa616c24johannkoenig@chromium.orgVP9_INSTANTIATE_TEST_CASE(TileIndependenceTest, ::testing::Range(0, 2, 1)); 106923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org 107923acb6a717b7f2719ed21184c1bc533b0a14888johannkoenig@chromium.org} // namespace 108