5106e4dfa8
There are two CreateDecoder functions and decode_test_driver is not calling the right function now. This bug is discovered during really enable the frame parallel flag inside libvpx. This bug does not affect any existing unit test though. Change-Id: Icd9633c4b66d50e422a09c4310ff791082878936
105 lines
3.8 KiB
C++
105 lines
3.8 KiB
C++
/*
|
|
* Copyright (c) 2012 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.
|
|
*/
|
|
#include "test/codec_factory.h"
|
|
#include "test/decode_test_driver.h"
|
|
#include "third_party/googletest/src/include/gtest/gtest.h"
|
|
#include "test/register_state_check.h"
|
|
#include "test/video_source.h"
|
|
|
|
namespace libvpx_test {
|
|
|
|
const char kVP8Name[] = "WebM Project VP8";
|
|
|
|
vpx_codec_err_t Decoder::PeekStream(const uint8_t *cxdata, size_t size,
|
|
vpx_codec_stream_info_t *stream_info) {
|
|
return vpx_codec_peek_stream_info(CodecInterface(),
|
|
cxdata, static_cast<unsigned int>(size),
|
|
stream_info);
|
|
}
|
|
|
|
vpx_codec_err_t Decoder::DecodeFrame(const uint8_t *cxdata, size_t size) {
|
|
return DecodeFrame(cxdata, size, NULL);
|
|
}
|
|
|
|
vpx_codec_err_t Decoder::DecodeFrame(const uint8_t *cxdata, size_t size,
|
|
void *user_priv) {
|
|
vpx_codec_err_t res_dec;
|
|
InitOnce();
|
|
REGISTER_STATE_CHECK(
|
|
res_dec = vpx_codec_decode(&decoder_,
|
|
cxdata, static_cast<unsigned int>(size),
|
|
user_priv, 0));
|
|
return res_dec;
|
|
}
|
|
|
|
void DecoderTest::RunLoop(CompressedVideoSource *video) {
|
|
Decoder* const decoder = codec_->CreateDecoder(cfg_, flags_, 0);
|
|
ASSERT_TRUE(decoder != NULL);
|
|
const char *codec_name = decoder->GetDecoderName();
|
|
const bool is_vp8 = strncmp(kVP8Name, codec_name, sizeof(kVP8Name) - 1) == 0;
|
|
bool end_of_file = false;
|
|
|
|
// Decode frames.
|
|
for (video->Begin(); !::testing::Test::HasFailure() && !end_of_file;
|
|
video->Next()) {
|
|
PreDecodeFrameHook(*video, decoder);
|
|
|
|
vpx_codec_stream_info_t stream_info;
|
|
stream_info.sz = sizeof(stream_info);
|
|
|
|
if (video->cxdata() != NULL) {
|
|
const vpx_codec_err_t res_peek = decoder->PeekStream(video->cxdata(),
|
|
video->frame_size(),
|
|
&stream_info);
|
|
if (is_vp8) {
|
|
/* Vp8's implementation of PeekStream returns an error if the frame you
|
|
* pass it is not a keyframe, so we only expect VPX_CODEC_OK on the
|
|
* first frame, which must be a keyframe. */
|
|
if (video->frame_number() == 0)
|
|
ASSERT_EQ(VPX_CODEC_OK, res_peek) << "Peek return failed: "
|
|
<< vpx_codec_err_to_string(res_peek);
|
|
} else {
|
|
/* The Vp9 implementation of PeekStream returns an error only if the
|
|
* data passed to it isn't a valid Vp9 chunk. */
|
|
ASSERT_EQ(VPX_CODEC_OK, res_peek) << "Peek return failed: "
|
|
<< vpx_codec_err_to_string(res_peek);
|
|
}
|
|
|
|
vpx_codec_err_t res_dec = decoder->DecodeFrame(video->cxdata(),
|
|
video->frame_size());
|
|
if (!HandleDecodeResult(res_dec, *video, decoder))
|
|
break;
|
|
} else {
|
|
// Signal end of the file to the decoder.
|
|
const vpx_codec_err_t res_dec = decoder->DecodeFrame(NULL, 0);
|
|
ASSERT_EQ(VPX_CODEC_OK, res_dec) << decoder->DecodeError();
|
|
end_of_file = true;
|
|
}
|
|
|
|
DxDataIterator dec_iter = decoder->GetDxData();
|
|
const vpx_image_t *img = NULL;
|
|
|
|
// Get decompressed data
|
|
while ((img = dec_iter.Next()))
|
|
DecompressedFrameHook(*img, video->frame_number());
|
|
}
|
|
|
|
delete decoder;
|
|
}
|
|
|
|
void DecoderTest::set_cfg(const vpx_codec_dec_cfg_t &dec_cfg) {
|
|
memcpy(&cfg_, &dec_cfg, sizeof(cfg_));
|
|
}
|
|
|
|
void DecoderTest::set_flags(const vpx_codec_flags_t flags) {
|
|
flags_ = flags;
|
|
}
|
|
} // namespace libvpx_test
|