OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC 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 | 10 |
11 #include <stdio.h> | 11 #include <stdio.h> |
12 | 12 |
13 #include <memory> | 13 #include <memory> |
14 | 14 |
15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
| 16 #include "webrtc/base/optional.h" |
16 #include "webrtc/base/timeutils.h" | 17 #include "webrtc/base/timeutils.h" |
17 #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h" | 18 #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h" |
18 #include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" | 19 #include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" |
19 #include "webrtc/modules/video_coding/codecs/vp8/temporal_layers.h" | 20 #include "webrtc/modules/video_coding/codecs/vp8/temporal_layers.h" |
20 #include "webrtc/test/frame_utils.h" | 21 #include "webrtc/test/frame_utils.h" |
21 #include "webrtc/test/gtest.h" | 22 #include "webrtc/test/gtest.h" |
22 #include "webrtc/test/testsupport/fileutils.h" | 23 #include "webrtc/test/testsupport/fileutils.h" |
23 | 24 |
24 namespace webrtc { | 25 namespace webrtc { |
25 | 26 |
(...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
81 bool Vp8UnitTestEncodeCompleteCallback::EncodeComplete() { | 82 bool Vp8UnitTestEncodeCompleteCallback::EncodeComplete() { |
82 if (encode_complete_) { | 83 if (encode_complete_) { |
83 encode_complete_ = false; | 84 encode_complete_ = false; |
84 return true; | 85 return true; |
85 } | 86 } |
86 return false; | 87 return false; |
87 } | 88 } |
88 | 89 |
89 class Vp8UnitTestDecodeCompleteCallback : public webrtc::DecodedImageCallback { | 90 class Vp8UnitTestDecodeCompleteCallback : public webrtc::DecodedImageCallback { |
90 public: | 91 public: |
91 explicit Vp8UnitTestDecodeCompleteCallback(VideoFrame* frame) | 92 explicit Vp8UnitTestDecodeCompleteCallback(rtc::Optional<VideoFrame>* frame) |
92 : decoded_frame_(frame), decode_complete(false) {} | 93 : decoded_frame_(frame), decode_complete(false) {} |
93 int32_t Decoded(VideoFrame& frame) override; | 94 int32_t Decoded(VideoFrame& frame) override; |
94 int32_t Decoded(VideoFrame& frame, int64_t decode_time_ms) override { | 95 int32_t Decoded(VideoFrame& frame, int64_t decode_time_ms) override { |
95 RTC_NOTREACHED(); | 96 RTC_NOTREACHED(); |
96 return -1; | 97 return -1; |
97 } | 98 } |
98 bool DecodeComplete(); | 99 bool DecodeComplete(); |
99 | 100 |
100 private: | 101 private: |
101 VideoFrame* decoded_frame_; | 102 rtc::Optional<VideoFrame>* decoded_frame_; |
102 bool decode_complete; | 103 bool decode_complete; |
103 }; | 104 }; |
104 | 105 |
105 bool Vp8UnitTestDecodeCompleteCallback::DecodeComplete() { | 106 bool Vp8UnitTestDecodeCompleteCallback::DecodeComplete() { |
106 if (decode_complete) { | 107 if (decode_complete) { |
107 decode_complete = false; | 108 decode_complete = false; |
108 return true; | 109 return true; |
109 } | 110 } |
110 return false; | 111 return false; |
111 } | 112 } |
112 | 113 |
113 int Vp8UnitTestDecodeCompleteCallback::Decoded(VideoFrame& image) { | 114 int Vp8UnitTestDecodeCompleteCallback::Decoded(VideoFrame& image) { |
114 *decoded_frame_ = image; | 115 *decoded_frame_ = rtc::Optional<VideoFrame>(image); |
115 decode_complete = true; | 116 decode_complete = true; |
116 return 0; | 117 return 0; |
117 } | 118 } |
118 | 119 |
119 class TestVp8Impl : public ::testing::Test { | 120 class TestVp8Impl : public ::testing::Test { |
120 protected: | 121 protected: |
121 virtual void SetUp() { | 122 virtual void SetUp() { |
122 encoder_.reset(VP8Encoder::Create()); | 123 encoder_.reset(VP8Encoder::Create()); |
123 decoder_.reset(VP8Decoder::Create()); | 124 decoder_.reset(VP8Decoder::Create()); |
124 memset(&codec_inst_, 0, sizeof(codec_inst_)); | 125 memset(&codec_inst_, 0, sizeof(codec_inst_)); |
(...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
177 return encoded_frame_._length; | 178 return encoded_frame_._length; |
178 } | 179 } |
179 } | 180 } |
180 return 0; | 181 return 0; |
181 } | 182 } |
182 | 183 |
183 size_t WaitForDecodedFrame() const { | 184 size_t WaitForDecodedFrame() const { |
184 int64_t startTime = rtc::TimeMillis(); | 185 int64_t startTime = rtc::TimeMillis(); |
185 while (rtc::TimeMillis() - startTime < kMaxWaitDecTimeMs) { | 186 while (rtc::TimeMillis() - startTime < kMaxWaitDecTimeMs) { |
186 if (decode_complete_callback_->DecodeComplete()) { | 187 if (decode_complete_callback_->DecodeComplete()) { |
187 return CalcBufferSize(kI420, decoded_frame_.width(), | 188 return CalcBufferSize(kI420, decoded_frame_->width(), |
188 decoded_frame_.height()); | 189 decoded_frame_->height()); |
189 } | 190 } |
190 } | 191 } |
191 return 0; | 192 return 0; |
192 } | 193 } |
193 | 194 |
194 const int kWidth = 172; | 195 const int kWidth = 172; |
195 const int kHeight = 144; | 196 const int kHeight = 144; |
196 | 197 |
197 std::unique_ptr<Vp8UnitTestEncodeCompleteCallback> encode_complete_callback_; | 198 std::unique_ptr<Vp8UnitTestEncodeCompleteCallback> encode_complete_callback_; |
198 std::unique_ptr<Vp8UnitTestDecodeCompleteCallback> decode_complete_callback_; | 199 std::unique_ptr<Vp8UnitTestDecodeCompleteCallback> decode_complete_callback_; |
199 std::unique_ptr<uint8_t[]> source_buffer_; | 200 std::unique_ptr<uint8_t[]> source_buffer_; |
200 FILE* source_file_; | 201 FILE* source_file_; |
201 std::unique_ptr<VideoFrame> input_frame_; | 202 std::unique_ptr<VideoFrame> input_frame_; |
202 std::unique_ptr<VideoEncoder> encoder_; | 203 std::unique_ptr<VideoEncoder> encoder_; |
203 std::unique_ptr<VideoDecoder> decoder_; | 204 std::unique_ptr<VideoDecoder> decoder_; |
204 EncodedImage encoded_frame_; | 205 EncodedImage encoded_frame_; |
205 VideoFrame decoded_frame_; | 206 rtc::Optional<VideoFrame> decoded_frame_; |
206 VideoCodec codec_inst_; | 207 VideoCodec codec_inst_; |
207 TemporalLayersFactory tl_factory_; | 208 TemporalLayersFactory tl_factory_; |
208 }; | 209 }; |
209 | 210 |
210 TEST_F(TestVp8Impl, EncoderParameterTest) { | 211 TEST_F(TestVp8Impl, EncoderParameterTest) { |
211 strncpy(codec_inst_.plName, "VP8", 31); | 212 strncpy(codec_inst_.plName, "VP8", 31); |
212 codec_inst_.plType = 126; | 213 codec_inst_.plType = 126; |
213 codec_inst_.maxBitrate = 0; | 214 codec_inst_.maxBitrate = 0; |
214 codec_inst_.minBitrate = 0; | 215 codec_inst_.minBitrate = 0; |
215 codec_inst_.width = 1440; | 216 codec_inst_.width = 1440; |
(...skipping 29 matching lines...) Expand all Loading... |
245 TEST_F(TestVp8Impl, MAYBE_AlignedStrideEncodeDecode) { | 246 TEST_F(TestVp8Impl, MAYBE_AlignedStrideEncodeDecode) { |
246 SetUpEncodeDecode(); | 247 SetUpEncodeDecode(); |
247 encoder_->Encode(*input_frame_, NULL, NULL); | 248 encoder_->Encode(*input_frame_, NULL, NULL); |
248 EXPECT_GT(WaitForEncodedFrame(), 0u); | 249 EXPECT_GT(WaitForEncodedFrame(), 0u); |
249 // First frame should be a key frame. | 250 // First frame should be a key frame. |
250 encoded_frame_._frameType = kVideoFrameKey; | 251 encoded_frame_._frameType = kVideoFrameKey; |
251 encoded_frame_.ntp_time_ms_ = kTestNtpTimeMs; | 252 encoded_frame_.ntp_time_ms_ = kTestNtpTimeMs; |
252 EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, | 253 EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, |
253 decoder_->Decode(encoded_frame_, false, NULL)); | 254 decoder_->Decode(encoded_frame_, false, NULL)); |
254 EXPECT_GT(WaitForDecodedFrame(), 0u); | 255 EXPECT_GT(WaitForDecodedFrame(), 0u); |
| 256 ASSERT_TRUE(decoded_frame_); |
255 // Compute PSNR on all planes (faster than SSIM). | 257 // Compute PSNR on all planes (faster than SSIM). |
256 EXPECT_GT(I420PSNR(input_frame_.get(), &decoded_frame_), 36); | 258 EXPECT_GT(I420PSNR(input_frame_.get(), &*decoded_frame_), 36); |
257 EXPECT_EQ(kTestTimestamp, decoded_frame_.timestamp()); | 259 EXPECT_EQ(kTestTimestamp, decoded_frame_->timestamp()); |
258 EXPECT_EQ(kTestNtpTimeMs, decoded_frame_.ntp_time_ms()); | 260 EXPECT_EQ(kTestNtpTimeMs, decoded_frame_->ntp_time_ms()); |
259 } | 261 } |
260 | 262 |
261 #if defined(WEBRTC_ANDROID) | 263 #if defined(WEBRTC_ANDROID) |
262 #define MAYBE_DecodeWithACompleteKeyFrame DISABLED_DecodeWithACompleteKeyFrame | 264 #define MAYBE_DecodeWithACompleteKeyFrame DISABLED_DecodeWithACompleteKeyFrame |
263 #else | 265 #else |
264 #define MAYBE_DecodeWithACompleteKeyFrame DecodeWithACompleteKeyFrame | 266 #define MAYBE_DecodeWithACompleteKeyFrame DecodeWithACompleteKeyFrame |
265 #endif | 267 #endif |
266 TEST_F(TestVp8Impl, MAYBE_DecodeWithACompleteKeyFrame) { | 268 TEST_F(TestVp8Impl, MAYBE_DecodeWithACompleteKeyFrame) { |
267 SetUpEncodeDecode(); | 269 SetUpEncodeDecode(); |
268 encoder_->Encode(*input_frame_, NULL, NULL); | 270 encoder_->Encode(*input_frame_, NULL, NULL); |
269 EXPECT_GT(WaitForEncodedFrame(), 0u); | 271 EXPECT_GT(WaitForEncodedFrame(), 0u); |
270 // Setting complete to false -> should return an error. | 272 // Setting complete to false -> should return an error. |
271 encoded_frame_._completeFrame = false; | 273 encoded_frame_._completeFrame = false; |
272 EXPECT_EQ(WEBRTC_VIDEO_CODEC_ERROR, | 274 EXPECT_EQ(WEBRTC_VIDEO_CODEC_ERROR, |
273 decoder_->Decode(encoded_frame_, false, NULL)); | 275 decoder_->Decode(encoded_frame_, false, NULL)); |
274 // Setting complete back to true. Forcing a delta frame. | 276 // Setting complete back to true. Forcing a delta frame. |
275 encoded_frame_._frameType = kVideoFrameDelta; | 277 encoded_frame_._frameType = kVideoFrameDelta; |
276 encoded_frame_._completeFrame = true; | 278 encoded_frame_._completeFrame = true; |
277 EXPECT_EQ(WEBRTC_VIDEO_CODEC_ERROR, | 279 EXPECT_EQ(WEBRTC_VIDEO_CODEC_ERROR, |
278 decoder_->Decode(encoded_frame_, false, NULL)); | 280 decoder_->Decode(encoded_frame_, false, NULL)); |
279 // Now setting a key frame. | 281 // Now setting a key frame. |
280 encoded_frame_._frameType = kVideoFrameKey; | 282 encoded_frame_._frameType = kVideoFrameKey; |
281 EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, | 283 EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, |
282 decoder_->Decode(encoded_frame_, false, NULL)); | 284 decoder_->Decode(encoded_frame_, false, NULL)); |
283 EXPECT_GT(I420PSNR(input_frame_.get(), &decoded_frame_), 36); | 285 ASSERT_TRUE(decoded_frame_); |
| 286 EXPECT_GT(I420PSNR(input_frame_.get(), &*decoded_frame_), 36); |
284 } | 287 } |
285 | 288 |
286 } // namespace webrtc | 289 } // namespace webrtc |
OLD | NEW |