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