diff options
author | Matt A. Tobin <email@mattatobin.com> | 2020-04-07 23:30:51 -0400 |
---|---|---|
committer | Matt A. Tobin <email@mattatobin.com> | 2020-04-07 23:30:51 -0400 |
commit | 5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb (patch) | |
tree | 45d55e3e5e73c4255c4d71258d9be5b2d004d28f /third_party/aom/test/decode_multithreaded_test.cc | |
parent | 50f1986697a7412e4160976fa5e11217b4ef1f44 (diff) | |
download | UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar.gz UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar.lz UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.tar.xz UXP-5545a8983ff0ef1fb52e64aef8e66fa9b13c1cbb.zip |
Move aom source to a sub-directory under media/libaom
There is no damned reason to treat this differently than any other media lib given its license and there never was.
Diffstat (limited to 'third_party/aom/test/decode_multithreaded_test.cc')
-rw-r--r-- | third_party/aom/test/decode_multithreaded_test.cc | 185 |
1 files changed, 0 insertions, 185 deletions
diff --git a/third_party/aom/test/decode_multithreaded_test.cc b/third_party/aom/test/decode_multithreaded_test.cc deleted file mode 100644 index cea1d144f..000000000 --- a/third_party/aom/test/decode_multithreaded_test.cc +++ /dev/null @@ -1,185 +0,0 @@ -/* - * Copyright (c) 2018, Alliance for Open Media. All rights reserved - * - * This source code is subject to the terms of the BSD 2 Clause License and - * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License - * was not distributed with this source code in the LICENSE file, you can - * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent. - */ - -#include <cstdio> -#include <cstdlib> -#include <string> - -#include "aom_mem/aom_mem.h" -#include "test/codec_factory.h" -#include "test/encode_test_driver.h" -#include "test/i420_video_source.h" -#include "test/md5_helper.h" -#include "test/util.h" -#include "third_party/googletest/src/googletest/include/gtest/gtest.h" - -namespace { - -static const int kNumMultiThreadDecoders = 3; - -class AV1DecodeMultiThreadedTest - : public ::libaom_test::CodecTestWith5Params<int, int, int, int, int>, - public ::libaom_test::EncoderTest { - protected: - AV1DecodeMultiThreadedTest() - : EncoderTest(GET_PARAM(0)), md5_single_thread_(), md5_multi_thread_(), - n_tile_cols_(GET_PARAM(1)), n_tile_rows_(GET_PARAM(2)), - n_tile_groups_(GET_PARAM(3)), set_cpu_used_(GET_PARAM(4)), - row_mt_(GET_PARAM(5)) { - 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; - cfg.allow_lowbitdepth = 1; - single_thread_dec_ = codec_->CreateDecoder(cfg, 0); - - // Test cfg.threads == powers of 2. - for (int i = 0; i < kNumMultiThreadDecoders; ++i) { - cfg.threads <<= 1; - multi_thread_dec_[i] = codec_->CreateDecoder(cfg, 0); - multi_thread_dec_[i]->Control(AV1D_SET_ROW_MT, row_mt_); - } - - if (single_thread_dec_->IsAV1()) { - single_thread_dec_->Control(AV1D_EXT_TILE_DEBUG, 1); - single_thread_dec_->Control(AV1_SET_DECODE_TILE_ROW, -1); - single_thread_dec_->Control(AV1_SET_DECODE_TILE_COL, -1); - } - for (int i = 0; i < kNumMultiThreadDecoders; ++i) { - if (multi_thread_dec_[i]->IsAV1()) { - multi_thread_dec_[i]->Control(AV1D_EXT_TILE_DEBUG, 1); - multi_thread_dec_[i]->Control(AV1_SET_DECODE_TILE_ROW, -1); - multi_thread_dec_[i]->Control(AV1_SET_DECODE_TILE_COL, -1); - } - } - } - - virtual ~AV1DecodeMultiThreadedTest() { - delete single_thread_dec_; - for (int i = 0; i < kNumMultiThreadDecoders; ++i) - delete multi_thread_dec_[i]; - } - - virtual void SetUp() { - InitializeConfig(); - SetMode(libaom_test::kTwoPassGood); - } - - virtual void PreEncodeFrameHook(libaom_test::VideoSource *video, - libaom_test::Encoder *encoder) { - if (video->frame() == 1) { - encoder->Control(AV1E_SET_TILE_COLUMNS, n_tile_cols_); - encoder->Control(AV1E_SET_TILE_ROWS, n_tile_rows_); - encoder->Control(AV1E_SET_NUM_TG, n_tile_groups_); - encoder->Control(AOME_SET_CPUUSED, set_cpu_used_); - } - } - - 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(single_thread_dec_, pkt, &md5_single_thread_); - - for (int i = 0; i < kNumMultiThreadDecoders; ++i) - UpdateMD5(multi_thread_dec_[i], pkt, &md5_multi_thread_[i]); - } - - 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_single_thread_str = md5_single_thread_.Get(); - - for (int i = 0; i < kNumMultiThreadDecoders; ++i) { - const char *md5_multi_thread_str = md5_multi_thread_[i].Get(); - ASSERT_STREQ(md5_single_thread_str, md5_multi_thread_str); - } - } - - ::libaom_test::MD5 md5_single_thread_; - ::libaom_test::MD5 md5_multi_thread_[kNumMultiThreadDecoders]; - ::libaom_test::Decoder *single_thread_dec_; - ::libaom_test::Decoder *multi_thread_dec_[kNumMultiThreadDecoders]; - - private: - int n_tile_cols_; - int n_tile_rows_; - int n_tile_groups_; - int set_cpu_used_; - int row_mt_; -}; - -// run an encode and do the decode both in single thread -// and multi thread. Ensure that the MD5 of the output in both cases -// is identical. If so, the test passes. -TEST_P(AV1DecodeMultiThreadedTest, MD5Match) { - cfg_.large_scale_tile = 0; - single_thread_dec_->Control(AV1_SET_TILE_MODE, 0); - for (int i = 0; i < kNumMultiThreadDecoders; ++i) - multi_thread_dec_[i]->Control(AV1_SET_TILE_MODE, 0); - DoTest(); -} - -class AV1DecodeMultiThreadedTestLarge : public AV1DecodeMultiThreadedTest {}; - -TEST_P(AV1DecodeMultiThreadedTestLarge, MD5Match) { - cfg_.large_scale_tile = 0; - single_thread_dec_->Control(AV1_SET_TILE_MODE, 0); - for (int i = 0; i < kNumMultiThreadDecoders; ++i) - multi_thread_dec_[i]->Control(AV1_SET_TILE_MODE, 0); - DoTest(); -} - -// TODO(ranjit): More tests have to be added using pre-generated MD5. -AV1_INSTANTIATE_TEST_CASE(AV1DecodeMultiThreadedTest, ::testing::Values(1, 2), - ::testing::Values(1, 2), ::testing::Values(1), - ::testing::Values(3), ::testing::Values(0, 1)); -AV1_INSTANTIATE_TEST_CASE(AV1DecodeMultiThreadedTestLarge, - ::testing::Values(0, 1, 2, 6), - ::testing::Values(0, 1, 2, 6), - ::testing::Values(1, 4), ::testing::Values(0), - ::testing::Values(0, 1)); - -class AV1DecodeMultiThreadedLSTestLarge - : public AV1DecodeMultiThreadedTestLarge {}; - -TEST_P(AV1DecodeMultiThreadedLSTestLarge, MD5Match) { - cfg_.large_scale_tile = 1; - single_thread_dec_->Control(AV1_SET_TILE_MODE, 1); - for (int i = 0; i < kNumMultiThreadDecoders; ++i) - multi_thread_dec_[i]->Control(AV1_SET_TILE_MODE, 1); - DoTest(); -} - -AV1_INSTANTIATE_TEST_CASE(AV1DecodeMultiThreadedLSTestLarge, - ::testing::Values(6), ::testing::Values(6), - ::testing::Values(1), ::testing::Values(0, 3), - ::testing::Values(0, 1)); - -} // namespace |