OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 #include "test/codec_factory.h" | 10 #include "test/codec_factory.h" |
(...skipping 21 matching lines...) Expand all Loading... |
32 void *user_priv) { | 32 void *user_priv) { |
33 vpx_codec_err_t res_dec; | 33 vpx_codec_err_t res_dec; |
34 InitOnce(); | 34 InitOnce(); |
35 API_REGISTER_STATE_CHECK( | 35 API_REGISTER_STATE_CHECK( |
36 res_dec = vpx_codec_decode(&decoder_, | 36 res_dec = vpx_codec_decode(&decoder_, |
37 cxdata, static_cast<unsigned int>(size), | 37 cxdata, static_cast<unsigned int>(size), |
38 user_priv, 0)); | 38 user_priv, 0)); |
39 return res_dec; | 39 return res_dec; |
40 } | 40 } |
41 | 41 |
| 42 bool Decoder::IsVP8() const { |
| 43 const char *codec_name = GetDecoderName(); |
| 44 return strncmp(kVP8Name, codec_name, sizeof(kVP8Name) - 1) == 0; |
| 45 } |
| 46 |
| 47 void DecoderTest::HandlePeekResult(Decoder *const decoder, |
| 48 CompressedVideoSource *video, |
| 49 const vpx_codec_err_t res_peek) { |
| 50 const bool is_vp8 = decoder->IsVP8(); |
| 51 if (is_vp8) { |
| 52 /* Vp8's implementation of PeekStream returns an error if the frame you |
| 53 * pass it is not a keyframe, so we only expect VPX_CODEC_OK on the first |
| 54 * frame, which must be a keyframe. */ |
| 55 if (video->frame_number() == 0) |
| 56 ASSERT_EQ(VPX_CODEC_OK, res_peek) << "Peek return failed: " |
| 57 << vpx_codec_err_to_string(res_peek); |
| 58 } else { |
| 59 /* The Vp9 implementation of PeekStream returns an error only if the |
| 60 * data passed to it isn't a valid Vp9 chunk. */ |
| 61 ASSERT_EQ(VPX_CODEC_OK, res_peek) << "Peek return failed: " |
| 62 << vpx_codec_err_to_string(res_peek); |
| 63 } |
| 64 } |
| 65 |
42 void DecoderTest::RunLoop(CompressedVideoSource *video, | 66 void DecoderTest::RunLoop(CompressedVideoSource *video, |
43 const vpx_codec_dec_cfg_t &dec_cfg) { | 67 const vpx_codec_dec_cfg_t &dec_cfg) { |
44 Decoder* const decoder = codec_->CreateDecoder(dec_cfg, 0); | 68 Decoder* const decoder = codec_->CreateDecoder(dec_cfg, 0); |
45 ASSERT_TRUE(decoder != NULL); | 69 ASSERT_TRUE(decoder != NULL); |
46 const char *codec_name = decoder->GetDecoderName(); | |
47 const bool is_vp8 = strncmp(kVP8Name, codec_name, sizeof(kVP8Name) - 1) == 0; | |
48 | 70 |
49 // Decode frames. | 71 // Decode frames. |
50 for (video->Begin(); !::testing::Test::HasFailure() && video->cxdata(); | 72 for (video->Begin(); !::testing::Test::HasFailure() && video->cxdata(); |
51 video->Next()) { | 73 video->Next()) { |
52 PreDecodeFrameHook(*video, decoder); | 74 PreDecodeFrameHook(*video, decoder); |
53 | 75 |
54 vpx_codec_stream_info_t stream_info; | 76 vpx_codec_stream_info_t stream_info; |
55 stream_info.sz = sizeof(stream_info); | 77 stream_info.sz = sizeof(stream_info); |
56 const vpx_codec_err_t res_peek = decoder->PeekStream(video->cxdata(), | 78 const vpx_codec_err_t res_peek = decoder->PeekStream(video->cxdata(), |
57 video->frame_size(), | 79 video->frame_size(), |
58 &stream_info); | 80 &stream_info); |
59 if (is_vp8) { | 81 HandlePeekResult(decoder, video, res_peek); |
60 /* Vp8's implementation of PeekStream returns an error if the frame you | 82 ASSERT_FALSE(::testing::Test::HasFailure()); |
61 * pass it is not a keyframe, so we only expect VPX_CODEC_OK on the first | |
62 * frame, which must be a keyframe. */ | |
63 if (video->frame_number() == 0) | |
64 ASSERT_EQ(VPX_CODEC_OK, res_peek) << "Peek return failed: " | |
65 << vpx_codec_err_to_string(res_peek); | |
66 } else { | |
67 /* The Vp9 implementation of PeekStream returns an error only if the | |
68 * data passed to it isn't a valid Vp9 chunk. */ | |
69 ASSERT_EQ(VPX_CODEC_OK, res_peek) << "Peek return failed: " | |
70 << vpx_codec_err_to_string(res_peek); | |
71 } | |
72 | 83 |
73 vpx_codec_err_t res_dec = decoder->DecodeFrame(video->cxdata(), | 84 vpx_codec_err_t res_dec = decoder->DecodeFrame(video->cxdata(), |
74 video->frame_size()); | 85 video->frame_size()); |
75 if (!HandleDecodeResult(res_dec, *video, decoder)) | 86 if (!HandleDecodeResult(res_dec, *video, decoder)) |
76 break; | 87 break; |
77 | 88 |
78 DxDataIterator dec_iter = decoder->GetDxData(); | 89 DxDataIterator dec_iter = decoder->GetDxData(); |
79 const vpx_image_t *img = NULL; | 90 const vpx_image_t *img = NULL; |
80 | 91 |
81 // Get decompressed data | 92 // Get decompressed data |
82 while ((img = dec_iter.Next())) | 93 while ((img = dec_iter.Next())) |
83 DecompressedFrameHook(*img, video->frame_number()); | 94 DecompressedFrameHook(*img, video->frame_number()); |
84 } | 95 } |
85 delete decoder; | 96 delete decoder; |
86 } | 97 } |
87 | 98 |
88 void DecoderTest::RunLoop(CompressedVideoSource *video) { | 99 void DecoderTest::RunLoop(CompressedVideoSource *video) { |
89 vpx_codec_dec_cfg_t dec_cfg = {0}; | 100 vpx_codec_dec_cfg_t dec_cfg = {0}; |
90 RunLoop(video, dec_cfg); | 101 RunLoop(video, dec_cfg); |
91 } | 102 } |
92 | 103 |
93 } // namespace libvpx_test | 104 } // namespace libvpx_test |
OLD | NEW |