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 <math.h> | 11 #include <math.h> |
12 #include <string.h> | 12 #include <string.h> |
13 | 13 |
14 #include <memory> | 14 #include <memory> |
15 | 15 |
16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
17 #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h" | 17 #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h" |
18 #include "webrtc/test/testsupport/fileutils.h" | 18 #include "webrtc/test/testsupport/fileutils.h" |
19 #include "webrtc/video_frame.h" | 19 #include "webrtc/video_frame.h" |
20 | 20 |
21 namespace webrtc { | 21 namespace webrtc { |
22 | 22 |
| 23 namespace { |
| 24 void Calc16ByteAlignedStride(int width, int* stride_y, int* stride_uv) { |
| 25 *stride_y = 16 * ((width + 15) / 16); |
| 26 *stride_uv = 16 * ((width + 31) / 32); |
| 27 } |
| 28 |
| 29 } // Anonymous namespace |
| 30 |
23 class TestLibYuv : public ::testing::Test { | 31 class TestLibYuv : public ::testing::Test { |
24 protected: | 32 protected: |
25 TestLibYuv(); | 33 TestLibYuv(); |
26 virtual void SetUp(); | 34 virtual void SetUp(); |
27 virtual void TearDown(); | 35 virtual void TearDown(); |
28 | 36 |
29 FILE* source_file_; | 37 FILE* source_file_; |
30 VideoFrame orig_frame_; | 38 VideoFrame orig_frame_; |
31 std::unique_ptr<uint8_t[]> orig_buffer_; | 39 std::unique_ptr<uint8_t[]> orig_buffer_; |
32 const int width_; | 40 const int width_; |
(...skipping 99 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
132 EXPECT_EQ(0, ConvertFromI420(orig_frame_, kUYVY, 0, out_uyvy_buffer.get())); | 140 EXPECT_EQ(0, ConvertFromI420(orig_frame_, kUYVY, 0, out_uyvy_buffer.get())); |
133 EXPECT_EQ(0, ConvertToI420(kUYVY, out_uyvy_buffer.get(), 0, 0, width_, | 141 EXPECT_EQ(0, ConvertToI420(kUYVY, out_uyvy_buffer.get(), 0, 0, width_, |
134 height_, 0, kVideoRotation_0, &res_i420_frame)); | 142 height_, 0, kVideoRotation_0, &res_i420_frame)); |
135 psnr = I420PSNR(&orig_frame_, &res_i420_frame); | 143 psnr = I420PSNR(&orig_frame_, &res_i420_frame); |
136 EXPECT_EQ(48.0, psnr); | 144 EXPECT_EQ(48.0, psnr); |
137 if (PrintVideoFrame(res_i420_frame, output_file) < 0) { | 145 if (PrintVideoFrame(res_i420_frame, output_file) < 0) { |
138 return; | 146 return; |
139 } | 147 } |
140 j++; | 148 j++; |
141 | 149 |
142 printf("\nConvert #%d I420 <-> YV12\n", j); | |
143 std::unique_ptr<uint8_t[]> outYV120Buffer(new uint8_t[frame_length_]); | |
144 std::unique_ptr<uint8_t[]> res_i420_buffer(new uint8_t[frame_length_]); | |
145 VideoFrame yv12_frame; | |
146 EXPECT_EQ(0, ConvertFromI420(orig_frame_, kYV12, 0, outYV120Buffer.get())); | |
147 yv12_frame.CreateFrame(outYV120Buffer.get(), | |
148 outYV120Buffer.get() + size_y_, | |
149 outYV120Buffer.get() + size_y_ + size_uv_, | |
150 width_, height_, | |
151 width_, (width_ + 1) / 2, (width_ + 1) / 2, | |
152 kVideoRotation_0); | |
153 EXPECT_EQ(0, ConvertFromYV12(yv12_frame, kI420, 0, res_i420_buffer.get())); | |
154 if (fwrite(res_i420_buffer.get(), 1, frame_length_, output_file) != | |
155 frame_length_) { | |
156 return; | |
157 } | |
158 | |
159 ConvertToI420(kI420, res_i420_buffer.get(), 0, 0, width_, height_, 0, | |
160 kVideoRotation_0, &res_i420_frame); | |
161 psnr = I420PSNR(&orig_frame_, &res_i420_frame); | |
162 EXPECT_EQ(48.0, psnr); | |
163 j++; | |
164 | |
165 printf("\nConvert #%d I420 <-> YUY2\n", j); | 150 printf("\nConvert #%d I420 <-> YUY2\n", j); |
166 std::unique_ptr<uint8_t[]> out_yuy2_buffer(new uint8_t[width_ * height_ * 2]); | 151 std::unique_ptr<uint8_t[]> out_yuy2_buffer(new uint8_t[width_ * height_ * 2]); |
167 EXPECT_EQ(0, ConvertFromI420(orig_frame_, kYUY2, 0, out_yuy2_buffer.get())); | 152 EXPECT_EQ(0, ConvertFromI420(orig_frame_, kYUY2, 0, out_yuy2_buffer.get())); |
168 | 153 |
169 EXPECT_EQ(0, ConvertToI420(kYUY2, out_yuy2_buffer.get(), 0, 0, width_, | 154 EXPECT_EQ(0, ConvertToI420(kYUY2, out_yuy2_buffer.get(), 0, 0, width_, |
170 height_, 0, kVideoRotation_0, &res_i420_frame)); | 155 height_, 0, kVideoRotation_0, &res_i420_frame)); |
171 | 156 |
172 if (PrintVideoFrame(res_i420_frame, output_file) < 0) { | 157 if (PrintVideoFrame(res_i420_frame, output_file) < 0) { |
173 return; | 158 return; |
174 } | 159 } |
(...skipping 87 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
262 0, kVideoRotation_90, &rotated_res_i420_frame)); | 247 0, kVideoRotation_90, &rotated_res_i420_frame)); |
263 EXPECT_EQ(0, ConvertToI420(kI420, orig_buffer_.get(), 0, 0, width_, height_, | 248 EXPECT_EQ(0, ConvertToI420(kI420, orig_buffer_.get(), 0, 0, width_, height_, |
264 0, kVideoRotation_270, &rotated_res_i420_frame)); | 249 0, kVideoRotation_270, &rotated_res_i420_frame)); |
265 rotated_res_i420_frame.CreateEmptyFrame(width_, height_, | 250 rotated_res_i420_frame.CreateEmptyFrame(width_, height_, |
266 width_, (width_ + 1) / 2, | 251 width_, (width_ + 1) / 2, |
267 (width_ + 1) / 2); | 252 (width_ + 1) / 2); |
268 EXPECT_EQ(0, ConvertToI420(kI420, orig_buffer_.get(), 0, 0, width_, height_, | 253 EXPECT_EQ(0, ConvertToI420(kI420, orig_buffer_.get(), 0, 0, width_, height_, |
269 0, kVideoRotation_180, &rotated_res_i420_frame)); | 254 0, kVideoRotation_180, &rotated_res_i420_frame)); |
270 } | 255 } |
271 | 256 |
272 TEST_F(TestLibYuv, alignment) { | |
273 int value = 0x3FF; // 1023 | |
274 EXPECT_EQ(0x400, AlignInt(value, 128)); // Low 7 bits are zero. | |
275 EXPECT_EQ(0x400, AlignInt(value, 64)); // Low 6 bits are zero. | |
276 EXPECT_EQ(0x400, AlignInt(value, 32)); // Low 5 bits are zero. | |
277 } | |
278 | |
279 TEST_F(TestLibYuv, StrideAlignment) { | |
280 int stride_y = 0; | |
281 int stride_uv = 0; | |
282 int width = 52; | |
283 Calc16ByteAlignedStride(width, &stride_y, &stride_uv); | |
284 EXPECT_EQ(64, stride_y); | |
285 EXPECT_EQ(32, stride_uv); | |
286 width = 128; | |
287 Calc16ByteAlignedStride(width, &stride_y, &stride_uv); | |
288 EXPECT_EQ(128, stride_y); | |
289 EXPECT_EQ(64, stride_uv); | |
290 width = 127; | |
291 Calc16ByteAlignedStride(width, &stride_y, &stride_uv); | |
292 EXPECT_EQ(128, stride_y); | |
293 EXPECT_EQ(64, stride_uv); | |
294 } | |
295 | |
296 } // namespace webrtc | 257 } // namespace webrtc |
OLD | NEW |