2 * Copyright (c) 2012 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.
10 #ifndef TEST_ENCODE_TEST_DRIVER_H_
11 #define TEST_ENCODE_TEST_DRIVER_H_
16 #include "third_party/googletest/src/include/gtest/gtest.h"
18 #include "./vpx_config.h"
19 #if CONFIG_VP8_ENCODER || CONFIG_VP9_ENCODER
20 #include "vpx/vp8cx.h"
22 #include "vpx/vpx_encoder.h"
24 namespace libvpx_test {
36 #define ALL_TEST_MODES ::testing::Values(::libvpx_test::kRealTime, \
37 ::libvpx_test::kOnePassGood, \
38 ::libvpx_test::kOnePassBest, \
39 ::libvpx_test::kTwoPassGood, \
40 ::libvpx_test::kTwoPassBest)
42 #define ONE_PASS_TEST_MODES ::testing::Values(::libvpx_test::kRealTime, \
43 ::libvpx_test::kOnePassGood, \
44 ::libvpx_test::kOnePassBest)
46 #define TWO_PASS_TEST_MODES ::testing::Values(::libvpx_test::kTwoPassGood, \
47 ::libvpx_test::kTwoPassBest)
50 // Provides an object to handle the libvpx get_cx_data() iteration pattern
51 class CxDataIterator {
53 explicit CxDataIterator(vpx_codec_ctx_t *encoder)
54 : encoder_(encoder), iter_(NULL) {}
56 const vpx_codec_cx_pkt_t *Next() {
57 return vpx_codec_get_cx_data(encoder_, &iter_);
61 vpx_codec_ctx_t *encoder_;
62 vpx_codec_iter_t iter_;
65 // Implements an in-memory store for libvpx twopass statistics
66 class TwopassStatsStore {
68 void Append(const vpx_codec_cx_pkt_t &pkt) {
69 buffer_.append(reinterpret_cast<char *>(pkt.data.twopass_stats.buf),
70 pkt.data.twopass_stats.sz);
73 vpx_fixed_buf_t buf() {
74 const vpx_fixed_buf_t buf = { &buffer_[0], buffer_.size() };
87 // Provides a simplified interface to manage one video encoding pass, given
88 // a configuration and video source.
90 // TODO(jkoleszar): The exact services it provides and the appropriate
91 // level of abstraction will be fleshed out as more tests are written.
94 Encoder(vpx_codec_enc_cfg_t cfg, unsigned long deadline,
95 const unsigned long init_flags, TwopassStatsStore *stats)
96 : cfg_(cfg), deadline_(deadline), init_flags_(init_flags), stats_(stats) {
97 memset(&encoder_, 0, sizeof(encoder_));
101 vpx_codec_destroy(&encoder_);
104 CxDataIterator GetCxData() {
105 return CxDataIterator(&encoder_);
108 void InitEncoder(VideoSource *video);
110 const vpx_image_t *GetPreviewFrame() {
111 return vpx_codec_get_preview_frame(&encoder_);
113 // This is a thin wrapper around vpx_codec_encode(), so refer to
114 // vpx_encoder.h for its semantics.
115 void EncodeFrame(VideoSource *video, const unsigned long frame_flags);
117 // Convenience wrapper for EncodeFrame()
118 void EncodeFrame(VideoSource *video) {
119 EncodeFrame(video, 0);
122 void Control(int ctrl_id, int arg) {
123 const vpx_codec_err_t res = vpx_codec_control_(&encoder_, ctrl_id, arg);
124 ASSERT_EQ(VPX_CODEC_OK, res) << EncoderError();
127 void Control(int ctrl_id, struct vpx_scaling_mode *arg) {
128 const vpx_codec_err_t res = vpx_codec_control_(&encoder_, ctrl_id, arg);
129 ASSERT_EQ(VPX_CODEC_OK, res) << EncoderError();
132 void Control(int ctrl_id, struct vpx_svc_layer_id *arg) {
133 const vpx_codec_err_t res = vpx_codec_control_(&encoder_, ctrl_id, arg);
134 ASSERT_EQ(VPX_CODEC_OK, res) << EncoderError();
137 void Control(int ctrl_id, struct vpx_svc_parameters *arg) {
138 const vpx_codec_err_t res = vpx_codec_control_(&encoder_, ctrl_id, arg);
139 ASSERT_EQ(VPX_CODEC_OK, res) << EncoderError();
141 #if CONFIG_VP8_ENCODER || CONFIG_VP9_ENCODER
142 void Control(int ctrl_id, vpx_active_map_t *arg) {
143 const vpx_codec_err_t res = vpx_codec_control_(&encoder_, ctrl_id, arg);
144 ASSERT_EQ(VPX_CODEC_OK, res) << EncoderError();
148 void Config(const vpx_codec_enc_cfg_t *cfg) {
149 const vpx_codec_err_t res = vpx_codec_enc_config_set(&encoder_, cfg);
150 ASSERT_EQ(VPX_CODEC_OK, res) << EncoderError();
154 void set_deadline(unsigned long deadline) {
155 deadline_ = deadline;
159 virtual vpx_codec_iface_t* CodecInterface() const = 0;
161 const char *EncoderError() {
162 const char *detail = vpx_codec_error_detail(&encoder_);
163 return detail ? detail : vpx_codec_error(&encoder_);
167 void EncodeFrameInternal(const VideoSource &video,
168 const unsigned long frame_flags);
170 // Flush the encoder on EOS
173 vpx_codec_ctx_t encoder_;
174 vpx_codec_enc_cfg_t cfg_;
175 unsigned long deadline_;
176 unsigned long init_flags_;
177 TwopassStatsStore *stats_;
180 // Common test functionality for all Encoder tests.
182 // This class is a mixin which provides the main loop common to all
183 // encoder tests. It provides hooks which can be overridden by subclasses
184 // to implement each test's specific behavior, while centralizing the bulk
185 // of the boilerplate. Note that it doesn't inherit the gtest testing
186 // classes directly, so that tests can be parameterized differently.
189 explicit EncoderTest(const CodecFactory *codec)
190 : codec_(codec), abort_(false), init_flags_(0), frame_flags_(0),
192 // Default to 1 thread.
196 virtual ~EncoderTest() {}
198 // Initialize the cfg_ member with the default configuration.
199 void InitializeConfig();
201 // Map the TestMode enum to the deadline_ and passes_ variables.
202 void SetMode(TestMode mode);
205 void set_init_flags(unsigned long flag) { // NOLINT(runtime/int)
210 virtual void RunLoop(VideoSource *video);
212 // Hook to be called at the beginning of a pass.
213 virtual void BeginPassHook(unsigned int /*pass*/) {}
215 // Hook to be called at the end of a pass.
216 virtual void EndPassHook() {}
218 // Hook to be called before encoding a frame.
219 virtual void PreEncodeFrameHook(VideoSource* /*video*/) {}
220 virtual void PreEncodeFrameHook(VideoSource* /*video*/,
221 Encoder* /*encoder*/) {}
223 // Hook to be called on every compressed data packet.
224 virtual void FramePktHook(const vpx_codec_cx_pkt_t* /*pkt*/) {}
226 // Hook to be called on every PSNR packet.
227 virtual void PSNRPktHook(const vpx_codec_cx_pkt_t* /*pkt*/) {}
229 // Hook to determine whether the encode loop should continue.
230 virtual bool Continue() const {
231 return !(::testing::Test::HasFatalFailure() || abort_);
234 const CodecFactory *codec_;
235 // Hook to determine whether to decode frame after encoding
236 virtual bool DoDecode() const { return 1; }
238 // Hook to handle encode/decode mismatch
239 virtual void MismatchHook(const vpx_image_t *img1,
240 const vpx_image_t *img2);
242 // Hook to be called on every decompressed frame.
243 virtual void DecompressedFrameHook(const vpx_image_t& /*img*/,
244 vpx_codec_pts_t /*pts*/) {}
246 // Hook to be called to handle decode result. Return true to continue.
247 virtual bool HandleDecodeResult(const vpx_codec_err_t res_dec,
248 const VideoSource& /*video*/,
250 EXPECT_EQ(VPX_CODEC_OK, res_dec) << decoder->DecodeError();
251 return VPX_CODEC_OK == res_dec;
254 // Hook that can modify the encoder's output data
255 virtual const vpx_codec_cx_pkt_t *MutateEncoderOutputHook(
256 const vpx_codec_cx_pkt_t *pkt) {
261 vpx_codec_enc_cfg_t cfg_;
262 vpx_codec_dec_cfg_t dec_cfg_;
263 unsigned int passes_;
264 unsigned long deadline_;
265 TwopassStatsStore stats_;
266 unsigned long init_flags_;
267 unsigned long frame_flags_;
268 vpx_codec_pts_t last_pts_;
271 } // namespace libvpx_test
273 #endif // TEST_ENCODE_TEST_DRIVER_H_