active_map_test.cc 3.39 KB
Newer Older
Alex Converse's avatar
Alex Converse committed
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
Alex Converse's avatar
Alex Converse committed
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
4
5
6
7
8
9
10
11
 * 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.
*/

Alex Converse's avatar
Alex Converse committed
12
13
#include <climits>
#include <vector>
14
#include "third_party/googletest/src/googletest/include/gtest/gtest.h"
Alex Converse's avatar
Alex Converse committed
15
16
17
18
19
20
21
22
#include "test/codec_factory.h"
#include "test/encode_test_driver.h"
#include "test/i420_video_source.h"
#include "test/util.h"

namespace {

class ActiveMapTest
23
24
    : public ::libaom_test::CodecTestWith2Params<libaom_test::TestMode, int>,
      public ::libaom_test::EncoderTest {
Alex Converse's avatar
Alex Converse committed
25
26
27
28
29
30
31
32
33
34
35
36
37
 protected:
  static const int kWidth = 208;
  static const int kHeight = 144;

  ActiveMapTest() : EncoderTest(GET_PARAM(0)) {}
  virtual ~ActiveMapTest() {}

  virtual void SetUp() {
    InitializeConfig();
    SetMode(GET_PARAM(1));
    cpu_used_ = GET_PARAM(2);
  }

38
39
  virtual void PreEncodeFrameHook(::libaom_test::VideoSource *video,
                                  ::libaom_test::Encoder *encoder) {
Alex Converse's avatar
Alex Converse committed
40
    if (video->frame() == 1) {
Yaowu Xu's avatar
Yaowu Xu committed
41
      encoder->Control(AOME_SET_CPUUSED, cpu_used_);
Alex Converse's avatar
Alex Converse committed
42
    } else if (video->frame() == 3) {
Yaowu Xu's avatar
Yaowu Xu committed
43
      aom_active_map_t map = aom_active_map_t();
clang-format's avatar
clang-format committed
44
      /* clang-format off */
Alex Converse's avatar
Alex Converse committed
45
46
47
48
49
50
51
52
53
54
55
      uint8_t active_map[9 * 13] = {
        1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0,
        1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0,
        1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0,
        1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0,
        0, 0, 0, 0, 1, 1, 0, 0, 0, 1, 0, 1, 1,
        0, 0, 0, 0, 1, 1, 0, 0, 1, 0, 1, 0, 1,
        0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 1, 0, 1,
        0, 0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 1, 1,
        1, 1, 1, 1, 0, 0, 0, 0, 1, 1, 1, 1, 0,
      };
clang-format's avatar
clang-format committed
56
      /* clang-format on */
Alex Converse's avatar
Alex Converse committed
57
58
59
60
61
      map.cols = (kWidth + 15) / 16;
      map.rows = (kHeight + 15) / 16;
      ASSERT_EQ(map.cols, 13u);
      ASSERT_EQ(map.rows, 9u);
      map.active_map = active_map;
Yaowu Xu's avatar
Yaowu Xu committed
62
      encoder->Control(AOME_SET_ACTIVEMAP, &map);
Alex Converse's avatar
Alex Converse committed
63
    } else if (video->frame() == 15) {
Yaowu Xu's avatar
Yaowu Xu committed
64
      aom_active_map_t map = aom_active_map_t();
Alex Converse's avatar
Alex Converse committed
65
66
67
      map.cols = (kWidth + 15) / 16;
      map.rows = (kHeight + 15) / 16;
      map.active_map = NULL;
Yaowu Xu's avatar
Yaowu Xu committed
68
      encoder->Control(AOME_SET_ACTIVEMAP, &map);
Alex Converse's avatar
Alex Converse committed
69
70
71
    }
  }

72
73
74
75
76
  void DoTest() {
    // Validate that this non multiple of 64 wide clip encodes
    cfg_.g_lag_in_frames = 0;
    cfg_.rc_target_bitrate = 400;
    cfg_.rc_resize_allowed = 0;
Yaowu Xu's avatar
Yaowu Xu committed
77
78
    cfg_.g_pass = AOM_RC_ONE_PASS;
    cfg_.rc_end_usage = AOM_CBR;
79
    cfg_.kf_max_dist = 90000;
80
    ::libaom_test::I420VideoSource video("hantro_odd.yuv", kWidth, kHeight, 30,
81
82
83
84
85
                                         1, 0, 20);

    ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
  }

Alex Converse's avatar
Alex Converse committed
86
87
88
  int cpu_used_;
};

clang-format's avatar
clang-format committed
89
TEST_P(ActiveMapTest, Test) { DoTest(); }
Alex Converse's avatar
Alex Converse committed
90

91
class ActiveMapTestLarge : public ActiveMapTest {};
Alex Converse's avatar
Alex Converse committed
92

clang-format's avatar
clang-format committed
93
TEST_P(ActiveMapTestLarge, Test) { DoTest(); }
Alex Converse's avatar
Alex Converse committed
94

Yaowu Xu's avatar
Yaowu Xu committed
95
96
97
AV1_INSTANTIATE_TEST_CASE(ActiveMapTestLarge,
                          ::testing::Values(::libaom_test::kRealTime),
                          ::testing::Range(0, 5));
Geza Lore's avatar
Geza Lore committed
98

Yaowu Xu's avatar
Yaowu Xu committed
99
100
101
AV1_INSTANTIATE_TEST_CASE(ActiveMapTest,
                          ::testing::Values(::libaom_test::kRealTime),
                          ::testing::Range(5, 9));
Geza Lore's avatar
Geza Lore committed
102

Alex Converse's avatar
Alex Converse committed
103
}  // namespace