| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the |
| * Alliance for Open Media Patent License 1.0 was not distributed with this |
| * source code in the PATENTS file, you can obtain it at |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #include <cstdio> |
| #include <cstdlib> |
| #include <string> |
| #include "third_party/googletest/src/googletest/include/gtest/gtest.h" |
| #include "test/codec_factory.h" |
| #include "test/encode_test_driver.h" |
| #include "test/i420_video_source.h" |
| #include "test/util.h" |
| #include "test/md5_helper.h" |
| #include "aom_mem/aom_mem.h" |
| |
| namespace { |
| class TileIndependenceTest |
| : public ::libaom_test::CodecTestWith3Params<int, int, int>, |
| public ::libaom_test::EncoderTest { |
| protected: |
| TileIndependenceTest() |
| : EncoderTest(GET_PARAM(0)), md5_fw_order_(), md5_inv_order_(), |
| n_tile_cols_(GET_PARAM(1)), n_tile_rows_(GET_PARAM(2)), |
| n_tile_groups_(GET_PARAM(3)) { |
| init_flags_ = AOM_CODEC_USE_PSNR; |
| aom_codec_dec_cfg_t cfg = aom_codec_dec_cfg_t(); |
| cfg.w = 704; |
| cfg.h = 576; |
| cfg.threads = 1; |
| fw_dec_ = codec_->CreateDecoder(cfg, 0); |
| inv_dec_ = codec_->CreateDecoder(cfg, 0); |
| inv_dec_->Control(AV1_INVERT_TILE_DECODE_ORDER, 1); |
| |
| if (fw_dec_->IsAV1() && inv_dec_->IsAV1()) { |
| fw_dec_->Control(AV1_SET_DECODE_TILE_ROW, -1); |
| fw_dec_->Control(AV1_SET_DECODE_TILE_COL, -1); |
| inv_dec_->Control(AV1_SET_DECODE_TILE_ROW, -1); |
| inv_dec_->Control(AV1_SET_DECODE_TILE_COL, -1); |
| } |
| } |
| |
| virtual ~TileIndependenceTest() { |
| delete fw_dec_; |
| delete inv_dec_; |
| } |
| |
| virtual void SetUp() { |
| InitializeConfig(); |
| SetMode(libaom_test::kOnePassGood); |
| } |
| |
| virtual void PreEncodeFrameHook(libaom_test::VideoSource *video, |
| libaom_test::Encoder *encoder) { |
| if (video->frame() == 0) { |
| encoder->Control(AV1E_SET_TILE_COLUMNS, n_tile_cols_); |
| encoder->Control(AV1E_SET_TILE_ROWS, n_tile_rows_); |
| SetCpuUsed(encoder); |
| } else if (video->frame() == 3) { |
| encoder->Control(AV1E_SET_NUM_TG, n_tile_groups_); |
| } |
| } |
| |
| virtual void SetCpuUsed(libaom_test::Encoder *encoder) { |
| static const int kCpuUsed = 3; |
| encoder->Control(AOME_SET_CPUUSED, kCpuUsed); |
| } |
| |
| void UpdateMD5(::libaom_test::Decoder *dec, const aom_codec_cx_pkt_t *pkt, |
| ::libaom_test::MD5 *md5) { |
| const aom_codec_err_t res = dec->DecodeFrame( |
| reinterpret_cast<uint8_t *>(pkt->data.frame.buf), pkt->data.frame.sz); |
| if (res != AOM_CODEC_OK) { |
| abort_ = true; |
| ASSERT_EQ(AOM_CODEC_OK, res); |
| } |
| const aom_image_t *img = dec->GetDxData().Next(); |
| md5->Add(img); |
| } |
| |
| virtual void FramePktHook(const aom_codec_cx_pkt_t *pkt) { |
| UpdateMD5(fw_dec_, pkt, &md5_fw_order_); |
| UpdateMD5(inv_dec_, pkt, &md5_inv_order_); |
| } |
| |
| void DoTest() { |
| const aom_rational timebase = { 33333333, 1000000000 }; |
| cfg_.g_timebase = timebase; |
| cfg_.rc_target_bitrate = 500; |
| cfg_.g_lag_in_frames = 12; |
| cfg_.rc_end_usage = AOM_VBR; |
| |
| libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 704, 576, |
| timebase.den, timebase.num, 0, 5); |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| |
| const char *md5_fw_str = md5_fw_order_.Get(); |
| const char *md5_inv_str = md5_inv_order_.Get(); |
| ASSERT_STREQ(md5_fw_str, md5_inv_str); |
| } |
| |
| ::libaom_test::MD5 md5_fw_order_, md5_inv_order_; |
| ::libaom_test::Decoder *fw_dec_, *inv_dec_; |
| |
| private: |
| int n_tile_cols_; |
| int n_tile_rows_; |
| int n_tile_groups_; |
| }; |
| |
| // run an encode with 2 or 4 tiles, and do the decode both in normal and |
| // inverted tile ordering. Ensure that the MD5 of the output in both cases |
| // is identical. If so, tiles are considered independent and the test passes. |
| TEST_P(TileIndependenceTest, MD5Match) { |
| cfg_.large_scale_tile = 0; |
| fw_dec_->Control(AV1_SET_TILE_MODE, 0); |
| inv_dec_->Control(AV1_SET_TILE_MODE, 0); |
| DoTest(); |
| } |
| |
| class TileIndependenceTestLarge : public TileIndependenceTest { |
| virtual void SetCpuUsed(libaom_test::Encoder *encoder) { |
| static const int kCpuUsed = 0; |
| encoder->Control(AOME_SET_CPUUSED, kCpuUsed); |
| } |
| }; |
| |
| TEST_P(TileIndependenceTestLarge, MD5Match) { |
| cfg_.large_scale_tile = 0; |
| fw_dec_->Control(AV1_SET_TILE_MODE, 0); |
| inv_dec_->Control(AV1_SET_TILE_MODE, 0); |
| DoTest(); |
| } |
| |
| AV1_INSTANTIATE_TEST_SUITE(TileIndependenceTest, ::testing::Values(0, 1), |
| ::testing::Values(0, 1), ::testing::Values(1, 2, 4)); |
| AV1_INSTANTIATE_TEST_SUITE(TileIndependenceTestLarge, ::testing::Values(0, 1), |
| ::testing::Values(0, 1), ::testing::Values(1, 2, 4)); |
| |
| class TileIndependenceLSTest : public TileIndependenceTest {}; |
| |
| TEST_P(TileIndependenceLSTest, MD5Match) { |
| cfg_.large_scale_tile = 1; |
| fw_dec_->Control(AV1_SET_TILE_MODE, 1); |
| fw_dec_->Control(AV1D_EXT_TILE_DEBUG, 1); |
| inv_dec_->Control(AV1_SET_TILE_MODE, 1); |
| inv_dec_->Control(AV1D_EXT_TILE_DEBUG, 1); |
| DoTest(); |
| } |
| |
| class TileIndependenceLSTestLarge : public TileIndependenceTestLarge {}; |
| |
| TEST_P(TileIndependenceLSTestLarge, MD5Match) { |
| cfg_.large_scale_tile = 1; |
| fw_dec_->Control(AV1_SET_TILE_MODE, 1); |
| fw_dec_->Control(AV1D_EXT_TILE_DEBUG, 1); |
| inv_dec_->Control(AV1_SET_TILE_MODE, 1); |
| inv_dec_->Control(AV1D_EXT_TILE_DEBUG, 1); |
| DoTest(); |
| } |
| |
| AV1_INSTANTIATE_TEST_SUITE(TileIndependenceLSTest, ::testing::Values(6), |
| ::testing::Values(6), ::testing::Values(1)); |
| AV1_INSTANTIATE_TEST_SUITE(TileIndependenceLSTestLarge, ::testing::Values(6), |
| ::testing::Values(6), ::testing::Values(1)); |
| } // namespace |