test/level_test.cc

   1 /*
   2  *  Copyright (c) 2016 The WebM project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10 #include "third_party/googletest/src/include/gtest/gtest.h"
  11 #include "test/codec_factory.h"
  12 #include "test/encode_test_driver.h"
  13 #include "test/i420_video_source.h"
  14 #include "test/util.h"
  15
  16 namespace {
  17 class LevelTest
  18     : public ::libvpx_test::EncoderTest,
  19       public ::libvpx_test::CodecTestWith2Params<libvpx_test::TestMode, int> {
  20  protected:
  21   LevelTest()
  22       : EncoderTest(GET_PARAM(0)), encoding_mode_(GET_PARAM(1)),
  23         cpu_used_(GET_PARAM(2)), min_gf_internal_(24), target_level_(0),
  24         level_(0) {}
  25   virtual ~LevelTest() {}
  26
  27   virtual void SetUp() {
  28     InitializeConfig();
  29     SetMode(encoding_mode_);
  30     if (encoding_mode_ != ::libvpx_test::kRealTime) {
  31       cfg_.g_lag_in_frames = 25;
  32       cfg_.rc_end_usage = VPX_VBR;
  33     } else {
  34       cfg_.g_lag_in_frames = 0;
  35       cfg_.rc_end_usage = VPX_CBR;
  36     }
  37     cfg_.rc_2pass_vbr_minsection_pct = 5;
  38     cfg_.rc_2pass_vbr_maxsection_pct = 2000;
  39     cfg_.rc_target_bitrate = 400;
  40     cfg_.rc_max_quantizer = 63;
  41     cfg_.rc_min_quantizer = 0;
  42   }
  43
  44   virtual void PreEncodeFrameHook(::libvpx_test::VideoSource *video,
  45                                   ::libvpx_test::Encoder *encoder) {
  46     if (video->frame() == 0) {
  47       encoder->Control(VP8E_SET_CPUUSED, cpu_used_);
  48       encoder->Control(VP9E_SET_TARGET_LEVEL, target_level_);
  49       encoder->Control(VP9E_SET_MIN_GF_INTERVAL, min_gf_internal_);
  50       if (encoding_mode_ != ::libvpx_test::kRealTime) {
  51         encoder->Control(VP8E_SET_ENABLEAUTOALTREF, 1);
  52         encoder->Control(VP8E_SET_ARNR_MAXFRAMES, 7);
  53         encoder->Control(VP8E_SET_ARNR_STRENGTH, 5);
  54         encoder->Control(VP8E_SET_ARNR_TYPE, 3);
  55       }
  56     }
  57     encoder->Control(VP9E_GET_LEVEL, &level_);
  58     ASSERT_LE(level_, 51);
  59     ASSERT_GE(level_, 0);
  60   }
  61
  62   ::libvpx_test::TestMode encoding_mode_;
  63   int cpu_used_;
  64   int min_gf_internal_;
  65   int target_level_;
  66   int level_;
  67 };
  68
  69 TEST_P(LevelTest, TestTargetLevel11Large) {
  70   ASSERT_NE(encoding_mode_, ::libvpx_test::kRealTime);
  71   ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0,
  72                                        60);
  73   target_level_ = 11;
  74   cfg_.rc_target_bitrate = 150;
  75   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
  76   ASSERT_GE(target_level_, level_);
  77 }
  78
  79 TEST_P(LevelTest, TestTargetLevel20Large) {
  80   ASSERT_NE(encoding_mode_, ::libvpx_test::kRealTime);
  81   ::libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288,
  82                                        30, 1, 0, 60);
  83   target_level_ = 20;
  84   cfg_.rc_target_bitrate = 1200;
  85   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
  86   ASSERT_GE(target_level_, level_);
  87 }
  88
  89 TEST_P(LevelTest, TestTargetLevel31Large) {
  90   ASSERT_NE(encoding_mode_, ::libvpx_test::kRealTime);
  91   ::libvpx_test::I420VideoSource video("niklas_1280_720_30.y4m", 1280, 720, 30,
  92                                        1, 0, 60);
  93   target_level_ = 31;
  94   cfg_.rc_target_bitrate = 8000;
  95   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
  96   ASSERT_GE(target_level_, level_);
  97 }
  98
  99 // Test for keeping level stats only
 100 TEST_P(LevelTest, TestTargetLevel0) {
 101   ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0,
 102                                        40);
 103   target_level_ = 0;
 104   min_gf_internal_ = 4;
 105   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
 106   ASSERT_GE(11, level_);
 107
 108   cfg_.rc_target_bitrate = 1600;
 109   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
 110   ASSERT_GE(20, level_);
 111 }
 112
 113 // Test for level control being turned off
 114 TEST_P(LevelTest, TestTargetLevel255) {
 115   ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0,
 116                                        30);
 117   target_level_ = 255;
 118   ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
 119 }
 120
 121 TEST_P(LevelTest, TestTargetLevelApi) {
 122   ::libvpx_test::I420VideoSource video("hantro_odd.yuv", 208, 144, 30, 1, 0, 1);
 123   static const vpx_codec_iface_t *codec = &vpx_codec_vp9_cx_algo;
 124   vpx_codec_ctx_t enc;
 125   vpx_codec_enc_cfg_t cfg;
 126   EXPECT_EQ(VPX_CODEC_OK, vpx_codec_enc_config_default(codec, &cfg, 0));
 127   cfg.rc_target_bitrate = 100;
 128   EXPECT_EQ(VPX_CODEC_OK, vpx_codec_enc_init(&enc, codec, &cfg, 0));
 129   for (int level = 0; level <= 256; ++level) {
 130     if (level == 10 || level == 11 || level == 20 || level == 21 ||
 131         level == 30 || level == 31 || level == 40 || level == 41 ||
 132         level == 50 || level == 51 || level == 52 || level == 60 ||
 133         level == 61 || level == 62 || level == 0 || level == 1 || level == 255)
 134       EXPECT_EQ(VPX_CODEC_OK,
 135                 vpx_codec_control(&enc, VP9E_SET_TARGET_LEVEL, level));
 136     else
 137       EXPECT_EQ(VPX_CODEC_INVALID_PARAM,
 138                 vpx_codec_control(&enc, VP9E_SET_TARGET_LEVEL, level));
 139   }
 140   EXPECT_EQ(VPX_CODEC_OK, vpx_codec_destroy(&enc));
 141 }
 142
 143 VP9_INSTANTIATE_TEST_SUITE(LevelTest,
 144                            ::testing::Values(::libvpx_test::kTwoPassGood,
 145                                              ::libvpx_test::kOnePassGood),
 146                            ::testing::Range(0, 9));
 147 }  // namespace