7050074756
When no more data is available, vpx_codec_decode should be called with NULL as data and 0 as data_sz. vpx_codec_get_frame iterates over a list of the frames available for display. The iterator storage should be initialized to NULL to start the iteration. Iteration is complete when this function returns NULL. Also change the unit test to conform to the api spec. Change-Id: I4b258b309f5df3d37d10c82f01492c0394181c2a
114 lines
4.2 KiB
C++
114 lines
4.2 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();
|
|
API_REGISTER_STATE_CHECK(
|
|
res_dec = vpx_codec_decode(&decoder_,
|
|
cxdata, static_cast<unsigned int>(size),
|
|
user_priv, 0));
|
|
return res_dec;
|
|
}
|
|
|
|
bool Decoder::IsVP8() const {
|
|
const char *codec_name = GetDecoderName();
|
|
return strncmp(kVP8Name, codec_name, sizeof(kVP8Name) - 1) == 0;
|
|
}
|
|
|
|
void DecoderTest::HandlePeekResult(Decoder *const decoder,
|
|
CompressedVideoSource *video,
|
|
const vpx_codec_err_t res_peek) {
|
|
const bool is_vp8 = decoder->IsVP8();
|
|
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);
|
|
}
|
|
}
|
|
|
|
void DecoderTest::RunLoop(CompressedVideoSource *video,
|
|
const vpx_codec_dec_cfg_t &dec_cfg) {
|
|
Decoder* const decoder = codec_->CreateDecoder(dec_cfg, 0);
|
|
ASSERT_TRUE(decoder != NULL);
|
|
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);
|
|
HandlePeekResult(decoder, video, res_peek);
|
|
ASSERT_FALSE(::testing::Test::HasFailure());
|
|
|
|
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::RunLoop(CompressedVideoSource *video) {
|
|
vpx_codec_dec_cfg_t dec_cfg = {0};
|
|
RunLoop(video, dec_cfg);
|
|
}
|
|
|
|
} // namespace libvpx_test
|