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 <limits.h>
11
12 #include "test/codec_factory.h"
13 #include "test/encode_test_driver.h"
14 #include "test/util.h"
15 #include "test/video_source.h"
16 #include "third_party/googletest/src/include/gtest/gtest.h"
17
18 namespace {
19
20 const int kVideoSourceWidth = 320;
21 const int kVideoSourceHeight = 240;
22 const int kFramesToEncode = 2;
23
24 class RealtimeTest
25 : public ::libvpx_test::EncoderTest,
26 public ::libvpx_test::CodecTestWithParam<libvpx_test::TestMode> {
27 protected:
RealtimeTest()28 RealtimeTest() : EncoderTest(GET_PARAM(0)), frame_packets_(0) {}
~RealtimeTest()29 ~RealtimeTest() override {}
30
SetUp()31 void SetUp() override {
32 InitializeConfig();
33 cfg_.g_lag_in_frames = 0;
34 SetMode(::libvpx_test::kRealTime);
35 }
36
BeginPassHook(unsigned int)37 void BeginPassHook(unsigned int /*pass*/) override {
38 #if !CONFIG_REALTIME_ONLY
39 // TODO(tomfinegan): We're changing the pass value here to make sure
40 // we get frames when real time mode is combined with |g_pass| set to
41 // VPX_RC_FIRST_PASS. This is necessary because EncoderTest::RunLoop() sets
42 // the pass value based on the mode passed into EncoderTest::SetMode(),
43 // which overrides the one specified in SetUp() above.
44 cfg_.g_pass = VPX_RC_FIRST_PASS;
45 #endif
46 }
47
PreEncodeFrameHook(::libvpx_test::VideoSource * video,::libvpx_test::Encoder * encoder)48 void PreEncodeFrameHook(::libvpx_test::VideoSource *video,
49 ::libvpx_test::Encoder *encoder) override {
50 if (video->frame() == 0 && set_cpu_used_) {
51 encoder->Control(VP8E_SET_CPUUSED, 8);
52 }
53 }
54
FramePktHook(const vpx_codec_cx_pkt_t *)55 void FramePktHook(const vpx_codec_cx_pkt_t * /*pkt*/) override {
56 frame_packets_++;
57 }
58
IsVP9() const59 bool IsVP9() const {
60 #if CONFIG_VP9_ENCODER
61 return codec_ == &libvpx_test::kVP9;
62 #else
63 return false;
64 #endif
65 }
66
TestIntegerOverflow(unsigned int width,unsigned int height)67 void TestIntegerOverflow(unsigned int width, unsigned int height) {
68 ::libvpx_test::RandomVideoSource video;
69 video.SetSize(width, height);
70 video.set_limit(20);
71 cfg_.rc_target_bitrate = UINT_MAX;
72 ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
73 }
74
TestEncode()75 void TestEncode() {
76 ::libvpx_test::RandomVideoSource video;
77 video.SetSize(kVideoSourceWidth, kVideoSourceHeight);
78 video.set_limit(kFramesToEncode);
79 ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
80 EXPECT_EQ(kFramesToEncode, frame_packets_);
81 }
82
83 int frame_packets_;
84 bool set_cpu_used_ = true;
85 };
86
TEST_P(RealtimeTest,RealtimeFirstPassProducesFrames)87 TEST_P(RealtimeTest, RealtimeFirstPassProducesFrames) { TestEncode(); }
88
TEST_P(RealtimeTest,RealtimeDefaultCpuUsed)89 TEST_P(RealtimeTest, RealtimeDefaultCpuUsed) {
90 set_cpu_used_ = false;
91 TestEncode();
92 }
93
TEST_P(RealtimeTest,IntegerOverflow)94 TEST_P(RealtimeTest, IntegerOverflow) { TestIntegerOverflow(2048, 2048); }
95
TEST_P(RealtimeTest,IntegerOverflowLarge)96 TEST_P(RealtimeTest, IntegerOverflowLarge) {
97 if (IsVP9()) {
98 #if VPX_ARCH_X86_64
99 TestIntegerOverflow(16384, 16384);
100 #else
101 TestIntegerOverflow(4096, 4096);
102 #endif
103 } else {
104 GTEST_SKIP()
105 << "TODO(https://crbug.com/webm/1748,https://crbug.com/webm/1751):"
106 << " Enable this test after bitstream errors & undefined sanitizer "
107 "warnings are fixed.";
108 // TestIntegerOverflow(16383, 16383);
109 }
110 }
111
112 VP8_INSTANTIATE_TEST_SUITE(RealtimeTest,
113 ::testing::Values(::libvpx_test::kRealTime));
114 VP9_INSTANTIATE_TEST_SUITE(RealtimeTest,
115 ::testing::Values(::libvpx_test::kRealTime));
116
117 } // namespace
118