test/decode_scalability_test.cc

   1 /*
   2  * Copyright (c) 2021, Alliance for Open Media. All rights reserved
   3  *
   4  * This source code is subject to the terms of the BSD 2 Clause License and
   5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
   6  * was not distributed with this source code in the LICENSE file, you can
   7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
   8  * Media Patent License 1.0 was not distributed with this source code in the
   9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
  10  */
  11
  12 #include <ostream>
  13
  14 #include "test/codec_factory.h"
  15 #include "test/decode_test_driver.h"
  16 #include "test/ivf_video_source.h"
  17 #include "test/util.h"
  18 #include "test/video_source.h"
  19 #include "third_party/googletest/src/googletest/include/gtest/gtest.h"
  20
  21 namespace {
  22
  23 struct ObuExtensionHeader {
  24   int temporal_id;
  25   int spatial_id;
  26 };
  27
  28 struct DecodeParam {
  29   const char *filename;
  30   const ObuExtensionHeader *headers;
  31   size_t num_headers;
  32 };
  33
  34 std::ostream &operator<<(std::ostream &os, const DecodeParam &dp) {
  35   return os << "file: " << dp.filename;
  36 }
  37
  38 class DecodeScalabilityTest
  39     : public ::libaom_test::DecoderTest,
  40       public ::libaom_test::CodecTestWithParam<DecodeParam> {
  41  protected:
  42   DecodeScalabilityTest()
  43       : DecoderTest(GET_PARAM(0)), headers_(GET_PARAM(1).headers),
  44         num_headers_(GET_PARAM(1).num_headers) {}
  45
  46   ~DecodeScalabilityTest() override {}
  47
  48   void PreDecodeFrameHook(const libaom_test::CompressedVideoSource &video,
  49                           libaom_test::Decoder *decoder) override {
  50     if (video.frame_number() == 0)
  51       decoder->Control(AV1D_SET_OUTPUT_ALL_LAYERS, 1);
  52   }
  53
  54   void DecompressedFrameHook(const aom_image_t &img,
  55                              const unsigned int /*frame_number*/) override {
  56     const ObuExtensionHeader &header = headers_[header_index_];
  57     EXPECT_EQ(img.temporal_id, header.temporal_id);
  58     EXPECT_EQ(img.spatial_id, header.spatial_id);
  59     header_index_ = (header_index_ + 1) % num_headers_;
  60   }
  61
  62   void RunTest() {
  63     const DecodeParam input = GET_PARAM(1);
  64     aom_codec_dec_cfg_t cfg = { 1, 0, 0, !FORCE_HIGHBITDEPTH_DECODING };
  65     libaom_test::IVFVideoSource decode_video(input.filename);
  66     decode_video.Init();
  67
  68     ASSERT_NO_FATAL_FAILURE(RunLoop(&decode_video, cfg));
  69   }
  70
  71  private:
  72   const ObuExtensionHeader *const headers_;
  73   const size_t num_headers_;
  74   size_t header_index_ = 0;
  75 };
  76
  77 TEST_P(DecodeScalabilityTest, ObuExtensionHeader) { RunTest(); }
  78
  79 // For all test files, we have:
  80 //   operatingPoint = 0
  81 //   OperatingPointIdc = operating_point_idc[ 0 ]
  82
  83 // av1-1-b8-01-size-16x16.ivf:
  84 //   operating_points_cnt_minus_1 = 0
  85 //   operating_point_idc[ 0 ] = 0x0
  86 const ObuExtensionHeader kSize16x16Headers[1] = { { 0, 0 } };
  87
  88 #if !CONFIG_REALTIME_ONLY
  89 // av1-1-b8-22-svc-L1T2.ivf:
  90 //   operating_points_cnt_minus_1 = 1
  91 //   operating_point_idc[ 0 ] = 0x103
  92 //   operating_point_idc[ 1 ] = 0x101
  93 const ObuExtensionHeader kL1T2Headers[2] = { { 0, 0 }, { 1, 0 } };
  94
  95 // av1-1-b8-22-svc-L2T1.ivf:
  96 //   operating_points_cnt_minus_1 = 1
  97 //   operating_point_idc[ 0 ] = 0x301
  98 //   operating_point_idc[ 1 ] = 0x101
  99 const ObuExtensionHeader kL2T1Headers[2] = { { 0, 0 }, { 0, 1 } };
 100
 101 // av1-1-b8-22-svc-L2T2.ivf:
 102 //   operating_points_cnt_minus_1 = 3
 103 //   operating_point_idc[ 0 ] = 0x303
 104 //   operating_point_idc[ 1 ] = 0x301
 105 //   operating_point_idc[ 2 ] = 0x103
 106 //   operating_point_idc[ 3 ] = 0x101
 107 const ObuExtensionHeader kL2T2Headers[4] = {
 108   { 0, 0 }, { 0, 1 }, { 1, 0 }, { 1, 1 }
 109 };
 110 #endif  // !CONFIG_REALTIME_ONLY
 111
 112 const DecodeParam kAV1DecodeScalabilityTests[] = {
 113   // { filename, headers, num_headers }
 114   { "av1-1-b8-01-size-16x16.ivf", kSize16x16Headers, 1 },
 115 #if !CONFIG_REALTIME_ONLY
 116   // These test vectors use loop restoration, which is not supported in
 117   // the CONFIG_REALTIME_ONLY builds.
 118   { "av1-1-b8-22-svc-L1T2.ivf", kL1T2Headers, 2 },
 119   { "av1-1-b8-22-svc-L2T1.ivf", kL2T1Headers, 2 },
 120   { "av1-1-b8-22-svc-L2T2.ivf", kL2T2Headers, 4 },
 121 #endif  // !CONFIG_REALTIME_ONLY
 122 };
 123
 124 AV1_INSTANTIATE_TEST_SUITE(DecodeScalabilityTest,
 125                            ::testing::ValuesIn(kAV1DecodeScalabilityTests));
 126
 127 }  // namespace