test/av1_convolve_2d_test_util.h

   1 /*
   2  * Copyright (c) 2016, 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 #ifndef TEST_HIPREC_CONVOLVE_TEST_UTIL_H_
  13 #define TEST_HIPREC_CONVOLVE_TEST_UTIL_H_
  14
  15 #include "third_party/googletest/src/googletest/include/gtest/gtest.h"
  16 #include "test/acm_random.h"
  17 #include "test/util.h"
  18 #include "./av1_rtcd.h"
  19 #include "./aom_dsp_rtcd.h"
  20 #include "test/clear_system_state.h"
  21 #include "test/register_state_check.h"
  22
  23 namespace libaom_test {
  24
  25 namespace AV1Convolve2D {
  26
  27 typedef void (*convolve_2d_func)(const uint8_t *src, int src_stride,
  28                                  uint8_t *dst, int dst_stride, int w, int h,
  29                                  InterpFilterParams *filter_params_x,
  30                                  InterpFilterParams *filter_params_y,
  31                                  const int subpel_x_q4, const int subpel_y_q4,
  32                                  ConvolveParams *conv_params);
  33
  34 typedef std::tr1::tuple<convolve_2d_func, int, int, BLOCK_SIZE> Convolve2DParam;
  35
  36 ::testing::internal::ParamGenerator<Convolve2DParam> BuildParams(
  37     convolve_2d_func filter, int subx_exist, int suby_exist);
  38
  39 class AV1Convolve2DTest : public ::testing::TestWithParam<Convolve2DParam> {
  40  public:
  41   virtual ~AV1Convolve2DTest();
  42   virtual void SetUp();
  43
  44   virtual void TearDown();
  45
  46  protected:
  47   void RunCheckOutput(convolve_2d_func test_impl);
  48   void RunSpeedTest(convolve_2d_func test_impl);
  49
  50   libaom_test::ACMRandom rnd_;
  51 };
  52
  53 class AV1Convolve2DSrTest : public ::testing::TestWithParam<Convolve2DParam> {
  54  public:
  55   virtual ~AV1Convolve2DSrTest();
  56   virtual void SetUp();
  57
  58   virtual void TearDown();
  59
  60  protected:
  61   void RunCheckOutput(convolve_2d_func test_impl);
  62   void RunSpeedTest(convolve_2d_func test_impl);
  63
  64   libaom_test::ACMRandom rnd_;
  65 };
  66
  67 class AV1JntConvolve2DTest : public ::testing::TestWithParam<Convolve2DParam> {
  68  public:
  69   virtual ~AV1JntConvolve2DTest();
  70   virtual void SetUp();
  71
  72   virtual void TearDown();
  73
  74  protected:
  75   void RunCheckOutput(convolve_2d_func test_impl);
  76
  77   libaom_test::ACMRandom rnd_;
  78 };
  79
  80 }  // namespace AV1Convolve2D
  81
  82 namespace AV1HighbdConvolve2D {
  83 typedef void (*highbd_convolve_2d_func)(
  84     const uint16_t *src, int src_stride, uint16_t *dst, int dst_stride, int w,
  85     int h, InterpFilterParams *filter_params_x,
  86     InterpFilterParams *filter_params_y, const int subpel_x_q4,
  87     const int subpel_y_q4, ConvolveParams *conv_params, int bd);
  88
  89 typedef std::tr1::tuple<int, highbd_convolve_2d_func, int, int, BLOCK_SIZE>
  90     HighbdConvolve2DParam;
  91
  92 ::testing::internal::ParamGenerator<HighbdConvolve2DParam> BuildParams(
  93     highbd_convolve_2d_func filter, int subx_exist, int suby_exist);
  94
  95 class AV1HighbdConvolve2DTest
  96     : public ::testing::TestWithParam<HighbdConvolve2DParam> {
  97  public:
  98   virtual ~AV1HighbdConvolve2DTest();
  99   virtual void SetUp();
 100
 101   virtual void TearDown();
 102
 103  protected:
 104   void RunCheckOutput(highbd_convolve_2d_func test_impl);
 105
 106   libaom_test::ACMRandom rnd_;
 107 };
 108
 109 class AV1HighbdConvolve2DSrTest
 110     : public ::testing::TestWithParam<HighbdConvolve2DParam> {
 111  public:
 112   virtual ~AV1HighbdConvolve2DSrTest();
 113   virtual void SetUp();
 114
 115   virtual void TearDown();
 116
 117  protected:
 118   void RunCheckOutput(highbd_convolve_2d_func test_impl);
 119   void RunSpeedTest(highbd_convolve_2d_func test_impl);
 120
 121   libaom_test::ACMRandom rnd_;
 122 };
 123
 124 class AV1HighbdJntConvolve2DTest
 125     : public ::testing::TestWithParam<HighbdConvolve2DParam> {
 126  public:
 127   virtual ~AV1HighbdJntConvolve2DTest();
 128   virtual void SetUp();
 129
 130   virtual void TearDown();
 131
 132  protected:
 133   void RunCheckOutput(highbd_convolve_2d_func test_impl);
 134   void RunSpeedTest(highbd_convolve_2d_func test_impl);
 135
 136   libaom_test::ACMRandom rnd_;
 137 };
 138
 139 }  // namespace AV1HighbdConvolve2D
 140
 141 }  // namespace libaom_test
 142
 143 #endif  // TEST_HIPREC_CONVOLVE_TEST_UTIL_H_