vp9_lossless_test.cc 2.65 KB
Newer Older
Yaowu Xu's avatar
Yaowu Xu committed
1
/*
Frank Galligan's avatar
Frank Galligan committed
2 3 4 5 6 7 8 9
 *  Copyright (c) 2013 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */
10
#include "./vpx_config.h"
Yaowu Xu's avatar
Yaowu Xu committed
11 12 13 14 15
#include "third_party/googletest/src/include/gtest/gtest.h"
#include "test/codec_factory.h"
#include "test/encode_test_driver.h"
#include "test/i420_video_source.h"
#include "test/util.h"
16
#include "test/y4m_video_source.h"
Yaowu Xu's avatar
Yaowu Xu committed
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38

namespace {

const int kMaxPsnr = 100;

class LossLessTest : public ::libvpx_test::EncoderTest,
    public ::libvpx_test::CodecTestWithParam<libvpx_test::TestMode> {
 protected:
  LossLessTest() : EncoderTest(GET_PARAM(0)),
                   psnr_(kMaxPsnr),
                   nframes_(0),
                   encoding_mode_(GET_PARAM(1)) {
  }

  virtual ~LossLessTest() {}

  virtual void SetUp() {
    InitializeConfig();
    SetMode(encoding_mode_);
  }

  virtual void BeginPassHook(unsigned int /*pass*/) {
39
    psnr_ = kMaxPsnr;
Yaowu Xu's avatar
Yaowu Xu committed
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
    nframes_ = 0;
  }

  virtual void PSNRPktHook(const vpx_codec_cx_pkt_t *pkt) {
    if (pkt->data.psnr.psnr[0] < psnr_)
      psnr_= pkt->data.psnr.psnr[0];
  }

  double GetMinPsnr() const {
      return psnr_;
  }

 private:
  double psnr_;
  unsigned int nframes_;
  libvpx_test::TestMode encoding_mode_;
};

TEST_P(LossLessTest, TestLossLessEncoding) {
  const vpx_rational timebase = { 33333333, 1000000000 };
  cfg_.g_timebase = timebase;
  cfg_.rc_target_bitrate = 2000;
  cfg_.g_lag_in_frames = 25;
  cfg_.rc_min_quantizer = 0;
  cfg_.rc_max_quantizer = 0;

  init_flags_ = VPX_CODEC_USE_PSNR;

  // intentionally changed the dimension for better testing coverage
69
  libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288,
70
                                     timebase.den, timebase.num, 0, 10);
71
  ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
Yaowu Xu's avatar
Yaowu Xu committed
72 73 74
  const double psnr_lossless = GetMinPsnr();
  EXPECT_GE(psnr_lossless, kMaxPsnr);
}
75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94

#if CONFIG_NON420
TEST_P(LossLessTest, TestLossLessEncoding444) {
  libvpx_test::Y4mVideoSource video("rush_hour_444.y4m", 0, 10);

  cfg_.g_profile = 1;
  cfg_.g_timebase = video.timebase();
  cfg_.rc_target_bitrate = 2000;
  cfg_.g_lag_in_frames = 25;
  cfg_.rc_min_quantizer = 0;
  cfg_.rc_max_quantizer = 0;

  init_flags_ = VPX_CODEC_USE_PSNR;

  ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
  const double psnr_lossless = GetMinPsnr();
  EXPECT_GE(psnr_lossless, kMaxPsnr);
}
#endif

Yaowu Xu's avatar
Yaowu Xu committed
95 96
VP9_INSTANTIATE_TEST_CASE(LossLessTest, ALL_TEST_MODES);
}  // namespace