2 * Copyright (c) 2013 The WebM project authors. All Rights Reserved.
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.
12 #include "third_party/googletest/src/include/gtest/gtest.h"
13 #include "test/codec_factory.h"
14 #include "test/decode_test_driver.h"
15 #include "test/i420_video_source.h"
16 #include "vpx/svc_context.h"
17 #include "vpx/vp8cx.h"
18 #include "vpx/vpx_encoder.h"
22 using libvpx_test::CodecFactory;
23 using libvpx_test::Decoder;
24 using libvpx_test::VP9CodecFactory;
26 class SvcTest : public ::testing::Test {
28 static const uint32_t kWidth = 352;
29 static const uint32_t kHeight = 288;
33 test_file_name_("hantro_collage_w352h288.yuv"),
34 codec_initialized_(false),
36 memset(&svc_, 0, sizeof(svc_));
37 memset(&codec_, 0, sizeof(codec_));
38 memset(&codec_enc_, 0, sizeof(codec_enc_));
43 virtual void SetUp() {
44 svc_.encoding_mode = INTER_LAYER_PREDICTION_IP;
45 svc_.log_level = SVC_LOG_DEBUG;
48 codec_iface_ = vpx_codec_vp9_cx();
49 const vpx_codec_err_t res =
50 vpx_codec_enc_config_default(codec_iface_, &codec_enc_, 0);
51 EXPECT_EQ(VPX_CODEC_OK, res);
53 codec_enc_.g_w = kWidth;
54 codec_enc_.g_h = kHeight;
55 codec_enc_.g_timebase.num = 1;
56 codec_enc_.g_timebase.den = 60;
57 codec_enc_.kf_min_dist = 100;
58 codec_enc_.kf_max_dist = 100;
60 vpx_codec_dec_cfg_t dec_cfg = {0};
61 VP9CodecFactory codec_factory;
62 decoder_ = codec_factory.CreateDecoder(dec_cfg, 0);
65 virtual void TearDown() {
66 vpx_svc_release(&svc_);
68 if (codec_initialized_) vpx_codec_destroy(&codec_);
72 vpx_codec_ctx_t codec_;
73 struct vpx_codec_enc_cfg codec_enc_;
74 vpx_codec_iface_t *codec_iface_;
75 std::string test_file_name_;
76 bool codec_initialized_;
80 TEST_F(SvcTest, SvcInit) {
81 // test missing parameters
82 vpx_codec_err_t res = vpx_svc_init(NULL, &codec_, codec_iface_, &codec_enc_);
83 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
84 res = vpx_svc_init(&svc_, NULL, codec_iface_, &codec_enc_);
85 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
86 res = vpx_svc_init(&svc_, &codec_, NULL, &codec_enc_);
87 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
89 res = vpx_svc_init(&svc_, &codec_, codec_iface_, NULL);
90 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
92 svc_.spatial_layers = 6; // too many layers
93 res = vpx_svc_init(&svc_, &codec_, codec_iface_, &codec_enc_);
94 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
96 svc_.spatial_layers = 0; // use default layers
97 res = vpx_svc_init(&svc_, &codec_, codec_iface_, &codec_enc_);
98 EXPECT_EQ(VPX_CODEC_OK, res);
99 codec_initialized_ = true;
100 EXPECT_EQ(VPX_SS_DEFAULT_LAYERS, svc_.spatial_layers);
103 TEST_F(SvcTest, InitTwoLayers) {
104 svc_.spatial_layers = 2;
105 vpx_svc_set_scale_factors(&svc_, "4/16,16*16"); // invalid scale values
106 vpx_codec_err_t res = vpx_svc_init(&svc_, &codec_, codec_iface_, &codec_enc_);
107 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
109 vpx_svc_set_scale_factors(&svc_, "4/16,16/16"); // valid scale values
110 res = vpx_svc_init(&svc_, &codec_, codec_iface_, &codec_enc_);
111 EXPECT_EQ(VPX_CODEC_OK, res);
112 codec_initialized_ = true;
115 TEST_F(SvcTest, InvalidOptions) {
116 vpx_codec_err_t res = vpx_svc_set_options(&svc_, NULL);
117 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
119 res = vpx_svc_set_options(&svc_, "not-an-option=1");
120 EXPECT_EQ(VPX_CODEC_OK, res);
121 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
122 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
125 TEST_F(SvcTest, SetLayersOption) {
126 vpx_codec_err_t res = vpx_svc_set_options(&svc_, "layers=3");
127 EXPECT_EQ(VPX_CODEC_OK, res);
128 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
129 EXPECT_EQ(VPX_CODEC_OK, res);
130 codec_initialized_ = true;
131 EXPECT_EQ(3, svc_.spatial_layers);
134 TEST_F(SvcTest, SetEncodingMode) {
135 vpx_codec_err_t res = vpx_svc_set_options(&svc_, "encoding-mode=alt-ip");
136 EXPECT_EQ(VPX_CODEC_OK, res);
137 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
138 EXPECT_EQ(VPX_CODEC_OK, res);
139 codec_initialized_ = true;
140 EXPECT_EQ(ALT_INTER_LAYER_PREDICTION_IP, svc_.encoding_mode);
143 TEST_F(SvcTest, SetMultipleOptions) {
144 vpx_codec_err_t res = vpx_svc_set_options(&svc_, "layers=2 encoding-mode=ip");
145 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
146 EXPECT_EQ(VPX_CODEC_OK, res);
147 codec_initialized_ = true;
148 EXPECT_EQ(2, svc_.spatial_layers);
149 EXPECT_EQ(INTER_LAYER_PREDICTION_IP, svc_.encoding_mode);
152 TEST_F(SvcTest, SetScaleFactorsOption) {
153 svc_.spatial_layers = 2;
154 vpx_codec_err_t res =
155 vpx_svc_set_options(&svc_, "scale-factors=not-scale-factors");
156 EXPECT_EQ(VPX_CODEC_OK, res);
157 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
158 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
160 res = vpx_svc_set_options(&svc_, "scale-factors=1/3,2/3");
161 EXPECT_EQ(VPX_CODEC_OK, res);
162 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
163 EXPECT_EQ(VPX_CODEC_OK, res);
164 codec_initialized_ = true;
167 TEST_F(SvcTest, SetQuantizersOption) {
168 svc_.spatial_layers = 2;
169 vpx_codec_err_t res = vpx_svc_set_options(&svc_, "quantizers=not-quantizers");
170 EXPECT_EQ(VPX_CODEC_OK, res);
171 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
172 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
174 vpx_svc_set_options(&svc_, "quantizers=40,45");
175 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
176 EXPECT_EQ(VPX_CODEC_OK, res);
177 codec_initialized_ = true;
180 TEST_F(SvcTest, SetQuantizers) {
181 vpx_codec_err_t res = vpx_svc_set_quantizers(NULL, "40,30");
182 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
184 res = vpx_svc_set_quantizers(&svc_, NULL);
185 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
187 svc_.spatial_layers = 2;
188 res = vpx_svc_set_quantizers(&svc_, "40");
189 EXPECT_EQ(VPX_CODEC_OK, res);
190 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
191 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
193 res = vpx_svc_set_quantizers(&svc_, "40,30");
194 EXPECT_EQ(VPX_CODEC_OK, res);
195 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
196 EXPECT_EQ(VPX_CODEC_OK, res);
197 codec_initialized_ = true;
200 TEST_F(SvcTest, SetScaleFactors) {
201 vpx_codec_err_t res = vpx_svc_set_scale_factors(NULL, "4/16,16/16");
202 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
204 res = vpx_svc_set_scale_factors(&svc_, NULL);
205 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
207 svc_.spatial_layers = 2;
208 res = vpx_svc_set_scale_factors(&svc_, "4/16");
209 EXPECT_EQ(VPX_CODEC_OK, res);
210 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
211 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
213 res = vpx_svc_set_scale_factors(&svc_, "4/16,16/16");
214 EXPECT_EQ(VPX_CODEC_OK, res);
215 res = vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
216 EXPECT_EQ(VPX_CODEC_OK, res);
217 codec_initialized_ = true;
220 // Test that decoder can handle an SVC frame as the first frame in a sequence.
221 TEST_F(SvcTest, FirstFrameHasLayers) {
222 svc_.spatial_layers = 2;
223 vpx_svc_set_scale_factors(&svc_, "4/16,16/16");
224 vpx_svc_set_quantizers(&svc_, "40,30");
226 vpx_codec_err_t res =
227 vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
228 EXPECT_EQ(VPX_CODEC_OK, res);
229 codec_initialized_ = true;
231 libvpx_test::I420VideoSource video(test_file_name_, kWidth, kHeight,
232 codec_enc_.g_timebase.den,
233 codec_enc_.g_timebase.num, 0, 30);
236 res = vpx_svc_encode(&svc_, &codec_, video.img(), video.pts(),
237 video.duration(), VPX_DL_REALTIME);
238 EXPECT_EQ(VPX_CODEC_OK, res);
240 const vpx_codec_err_t res_dec = decoder_->DecodeFrame(
241 static_cast<const uint8_t *>(vpx_svc_get_buffer(&svc_)),
242 vpx_svc_get_frame_size(&svc_));
244 // this test fails with a decoder error
245 ASSERT_EQ(VPX_CODEC_OK, res_dec) << decoder_->DecodeError();
248 TEST_F(SvcTest, EncodeThreeFrames) {
249 svc_.spatial_layers = 2;
250 vpx_svc_set_scale_factors(&svc_, "4/16,16/16");
251 vpx_svc_set_quantizers(&svc_, "40,30");
253 vpx_codec_err_t res =
254 vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
255 ASSERT_EQ(VPX_CODEC_OK, res);
256 codec_initialized_ = true;
258 libvpx_test::I420VideoSource video(test_file_name_, kWidth, kHeight,
259 codec_enc_.g_timebase.den,
260 codec_enc_.g_timebase.num, 0, 30);
263 // This frame is a keyframe.
264 res = vpx_svc_encode(&svc_, &codec_, video.img(), video.pts(),
265 video.duration(), VPX_DL_REALTIME);
266 ASSERT_EQ(VPX_CODEC_OK, res);
267 EXPECT_EQ(1, vpx_svc_is_keyframe(&svc_));
269 vpx_codec_err_t res_dec = decoder_->DecodeFrame(
270 static_cast<const uint8_t *>(vpx_svc_get_buffer(&svc_)),
271 vpx_svc_get_frame_size(&svc_));
272 ASSERT_EQ(VPX_CODEC_OK, res_dec) << decoder_->DecodeError();
276 // This is a P-frame.
277 res = vpx_svc_encode(&svc_, &codec_, video.img(), video.pts(),
278 video.duration(), VPX_DL_REALTIME);
279 ASSERT_EQ(VPX_CODEC_OK, res);
280 EXPECT_EQ(0, vpx_svc_is_keyframe(&svc_));
282 res_dec = decoder_->DecodeFrame(
283 static_cast<const uint8_t *>(vpx_svc_get_buffer(&svc_)),
284 vpx_svc_get_frame_size(&svc_));
285 ASSERT_EQ(VPX_CODEC_OK, res_dec) << decoder_->DecodeError();
289 // This is a P-frame.
290 res = vpx_svc_encode(&svc_, &codec_, video.img(), video.pts(),
291 video.duration(), VPX_DL_REALTIME);
292 ASSERT_EQ(VPX_CODEC_OK, res);
293 EXPECT_EQ(0, vpx_svc_is_keyframe(&svc_));
295 res_dec = decoder_->DecodeFrame(
296 static_cast<const uint8_t *>(vpx_svc_get_buffer(&svc_)),
297 vpx_svc_get_frame_size(&svc_));
298 ASSERT_EQ(VPX_CODEC_OK, res_dec) << decoder_->DecodeError();
301 TEST_F(SvcTest, GetLayerResolution) {
302 svc_.spatial_layers = 2;
303 vpx_svc_set_scale_factors(&svc_, "4/16,8/16");
304 vpx_svc_set_quantizers(&svc_, "40,30");
306 vpx_codec_err_t res =
307 vpx_svc_init(&svc_, &codec_, vpx_codec_vp9_cx(), &codec_enc_);
308 EXPECT_EQ(VPX_CODEC_OK, res);
309 codec_initialized_ = true;
311 // ensure that requested layer is a valid layer
312 uint32_t layer_width, layer_height;
313 res = vpx_svc_get_layer_resolution(&svc_, svc_.spatial_layers,
314 &layer_width, &layer_height);
315 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
317 res = vpx_svc_get_layer_resolution(NULL, 0, &layer_width, &layer_height);
318 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
320 res = vpx_svc_get_layer_resolution(&svc_, 0, NULL, &layer_height);
321 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
323 res = vpx_svc_get_layer_resolution(&svc_, 0, &layer_width, NULL);
324 EXPECT_EQ(VPX_CODEC_INVALID_PARAM, res);
326 res = vpx_svc_get_layer_resolution(&svc_, 0, &layer_width, &layer_height);
327 EXPECT_EQ(VPX_CODEC_OK, res);
328 EXPECT_EQ(kWidth * 4 / 16, layer_width);
329 EXPECT_EQ(kHeight * 4 / 16, layer_height);
331 res = vpx_svc_get_layer_resolution(&svc_, 1, &layer_width, &layer_height);
332 EXPECT_EQ(VPX_CODEC_OK, res);
333 EXPECT_EQ(kWidth * 8 / 16, layer_width);
334 EXPECT_EQ(kHeight * 8 / 16, layer_height);