Gitiles
Code Review Sign In
gerrit.openfyde.cn / webrtc.googlesource.com / src / 9c55f0f957534144d2b8a64154f0a479249b34be / . / talk / media / webrtc / webrtcvideoframe_unittest.cc
blob: e63c5d5e5611fb1576dce56e370eba86bc8fdca7 [file] [log] [blame]
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000[diff] [blame]1/*
2 * libjingle
3 * Copyright 2011 Google Inc.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright notice,
9 * this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright notice,
11 * this list of conditions and the following disclaimer in the documentation
12 * and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28#include "talk/base/flags.h"
29#include "talk/media/base/videoframe_unittest.h"
30#include "talk/media/webrtc/webrtcvideoframe.h"
31
32extern int FLAG_yuvconverter_repeat; // From lmivideoframe_unittest.cc.
33
34class WebRtcVideoFrameTest : public VideoFrameTest<cricket::WebRtcVideoFrame> {
35 public:
36 WebRtcVideoFrameTest() {
37 repeat_ = FLAG_yuvconverter_repeat;
38 }
39
40 void TestInit(int cropped_width, int cropped_height) {
41 const int frame_width = 1920;
42 const int frame_height = 1080;
43
44 // Build the CapturedFrame.
45 cricket::CapturedFrame captured_frame;
46 captured_frame.fourcc = cricket::FOURCC_I420;
47 captured_frame.pixel_width = 1;
48 captured_frame.pixel_height = 1;
49 captured_frame.elapsed_time = 1234;
50 captured_frame.time_stamp = 5678;
51 captured_frame.rotation = 0;
52 captured_frame.width = frame_width;
53 captured_frame.height = frame_height;
54 captured_frame.data_size = (frame_width * frame_height) +
55 ((frame_width + 1) / 2) * ((frame_height + 1) / 2) * 2;
wu@webrtc.org97077a32013-10-25 21:18:33 +0000[diff] [blame]56 talk_base::scoped_ptr<uint8[]> captured_frame_buffer(
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000[diff] [blame]57 new uint8[captured_frame.data_size]);
58 captured_frame.data = captured_frame_buffer.get();
59
60 // Create the new frame from the CapturedFrame.
61 cricket::WebRtcVideoFrame frame;
62 EXPECT_TRUE(frame.Init(&captured_frame, cropped_width, cropped_height));
63
64 // Verify the new frame.
65 EXPECT_EQ(1u, frame.GetPixelWidth());
66 EXPECT_EQ(1u, frame.GetPixelHeight());
67 EXPECT_EQ(1234, frame.GetElapsedTime());
68 EXPECT_EQ(5678, frame.GetTimeStamp());
69 EXPECT_EQ(0, frame.GetRotation());
70 // The size of the new frame should have been cropped to multiple of 4.
71 EXPECT_EQ(static_cast<size_t>(cropped_width & ~3), frame.GetWidth());
72 EXPECT_EQ(static_cast<size_t>(cropped_height & ~3), frame.GetHeight());
73 }
74};
75
76#define TEST_WEBRTCVIDEOFRAME(X) TEST_F(WebRtcVideoFrameTest, X) { \
77 VideoFrameTest<cricket::WebRtcVideoFrame>::X(); \
78}
79
80TEST_WEBRTCVIDEOFRAME(ConstructI420)
81TEST_WEBRTCVIDEOFRAME(ConstructI422)
82TEST_WEBRTCVIDEOFRAME(ConstructYuy2)
83TEST_WEBRTCVIDEOFRAME(ConstructYuy2Unaligned)
84TEST_WEBRTCVIDEOFRAME(ConstructYuy2Wide)
85TEST_WEBRTCVIDEOFRAME(ConstructYV12)
86TEST_WEBRTCVIDEOFRAME(ConstructUyvy)
87TEST_WEBRTCVIDEOFRAME(ConstructM420)
88TEST_WEBRTCVIDEOFRAME(ConstructQ420)
89TEST_WEBRTCVIDEOFRAME(ConstructNV21)
90TEST_WEBRTCVIDEOFRAME(ConstructNV12)
91TEST_WEBRTCVIDEOFRAME(ConstructABGR)
92TEST_WEBRTCVIDEOFRAME(ConstructARGB)
93TEST_WEBRTCVIDEOFRAME(ConstructARGBWide)
94TEST_WEBRTCVIDEOFRAME(ConstructBGRA)
95TEST_WEBRTCVIDEOFRAME(Construct24BG)
96TEST_WEBRTCVIDEOFRAME(ConstructRaw)
97TEST_WEBRTCVIDEOFRAME(ConstructRGB565)
98TEST_WEBRTCVIDEOFRAME(ConstructARGB1555)
99TEST_WEBRTCVIDEOFRAME(ConstructARGB4444)
100
101TEST_WEBRTCVIDEOFRAME(ConstructI420Mirror)
102TEST_WEBRTCVIDEOFRAME(ConstructI420Rotate0)
103TEST_WEBRTCVIDEOFRAME(ConstructI420Rotate90)
104TEST_WEBRTCVIDEOFRAME(ConstructI420Rotate180)
105TEST_WEBRTCVIDEOFRAME(ConstructI420Rotate270)
106TEST_WEBRTCVIDEOFRAME(ConstructYV12Rotate0)
107TEST_WEBRTCVIDEOFRAME(ConstructYV12Rotate90)
108TEST_WEBRTCVIDEOFRAME(ConstructYV12Rotate180)
109TEST_WEBRTCVIDEOFRAME(ConstructYV12Rotate270)
110TEST_WEBRTCVIDEOFRAME(ConstructNV12Rotate0)
111TEST_WEBRTCVIDEOFRAME(ConstructNV12Rotate90)
112TEST_WEBRTCVIDEOFRAME(ConstructNV12Rotate180)
113TEST_WEBRTCVIDEOFRAME(ConstructNV12Rotate270)
114TEST_WEBRTCVIDEOFRAME(ConstructNV21Rotate0)
115TEST_WEBRTCVIDEOFRAME(ConstructNV21Rotate90)
116TEST_WEBRTCVIDEOFRAME(ConstructNV21Rotate180)
117TEST_WEBRTCVIDEOFRAME(ConstructNV21Rotate270)
118TEST_WEBRTCVIDEOFRAME(ConstructUYVYRotate0)
119TEST_WEBRTCVIDEOFRAME(ConstructUYVYRotate90)
120TEST_WEBRTCVIDEOFRAME(ConstructUYVYRotate180)
121TEST_WEBRTCVIDEOFRAME(ConstructUYVYRotate270)
122TEST_WEBRTCVIDEOFRAME(ConstructYUY2Rotate0)
123TEST_WEBRTCVIDEOFRAME(ConstructYUY2Rotate90)
124TEST_WEBRTCVIDEOFRAME(ConstructYUY2Rotate180)
125TEST_WEBRTCVIDEOFRAME(ConstructYUY2Rotate270)
126TEST_WEBRTCVIDEOFRAME(ConstructI4201Pixel)
127TEST_WEBRTCVIDEOFRAME(ConstructI4205Pixel)
128// TODO(juberti): WebRtcVideoFrame does not support horizontal crop.
129// Re-evaluate once it supports 3 independent planes, since we might want to
130// just Init normally and then crop by adjusting pointers.
131// TEST_WEBRTCVIDEOFRAME(ConstructI420CropHorizontal)
132TEST_WEBRTCVIDEOFRAME(ConstructI420CropVertical)
133// TODO(juberti): WebRtcVideoFrame is not currently refcounted.
134// TEST_WEBRTCVIDEOFRAME(ConstructCopy)
135// TEST_WEBRTCVIDEOFRAME(ConstructCopyIsRef)
136TEST_WEBRTCVIDEOFRAME(ConstructBlack)
137// TODO(fbarchard): Implement Jpeg
138// TEST_WEBRTCVIDEOFRAME(ConstructMjpgI420)
139// TEST_WEBRTCVIDEOFRAME(ConstructMjpgI422)
140// TEST_WEBRTCVIDEOFRAME(ConstructMjpgI444)
141// TEST_WEBRTCVIDEOFRAME(ConstructMjpgI411)
142// TEST_WEBRTCVIDEOFRAME(ConstructMjpgI400)
143// TEST_WEBRTCVIDEOFRAME(ValidateMjpgI420)
144// TEST_WEBRTCVIDEOFRAME(ValidateMjpgI422)
145// TEST_WEBRTCVIDEOFRAME(ValidateMjpgI444)
146// TEST_WEBRTCVIDEOFRAME(ValidateMjpgI411)
147// TEST_WEBRTCVIDEOFRAME(ValidateMjpgI400)
148TEST_WEBRTCVIDEOFRAME(ValidateI420)
149TEST_WEBRTCVIDEOFRAME(ValidateI420SmallSize)
150TEST_WEBRTCVIDEOFRAME(ValidateI420LargeSize)
151TEST_WEBRTCVIDEOFRAME(ValidateI420HugeSize)
152// TEST_WEBRTCVIDEOFRAME(ValidateMjpgI420InvalidSize)
153// TEST_WEBRTCVIDEOFRAME(ValidateI420InvalidSize)
154
155// TODO(fbarchard): WebRtcVideoFrame does not support odd sizes.
156// Re-evaluate once WebRTC switches to libyuv
157// TEST_WEBRTCVIDEOFRAME(ConstructYuy2AllSizes)
158// TEST_WEBRTCVIDEOFRAME(ConstructARGBAllSizes)
159TEST_WEBRTCVIDEOFRAME(Reset)
160TEST_WEBRTCVIDEOFRAME(ConvertToABGRBuffer)
161TEST_WEBRTCVIDEOFRAME(ConvertToABGRBufferStride)
162TEST_WEBRTCVIDEOFRAME(ConvertToABGRBufferInverted)
163TEST_WEBRTCVIDEOFRAME(ConvertToARGB1555Buffer)
164TEST_WEBRTCVIDEOFRAME(ConvertToARGB1555BufferStride)
165TEST_WEBRTCVIDEOFRAME(ConvertToARGB1555BufferInverted)
166TEST_WEBRTCVIDEOFRAME(ConvertToARGB4444Buffer)
167TEST_WEBRTCVIDEOFRAME(ConvertToARGB4444BufferStride)
168TEST_WEBRTCVIDEOFRAME(ConvertToARGB4444BufferInverted)
169TEST_WEBRTCVIDEOFRAME(ConvertToARGBBuffer)
170TEST_WEBRTCVIDEOFRAME(ConvertToARGBBufferStride)
171TEST_WEBRTCVIDEOFRAME(ConvertToARGBBufferInverted)
172TEST_WEBRTCVIDEOFRAME(ConvertToBGRABuffer)
173TEST_WEBRTCVIDEOFRAME(ConvertToBGRABufferStride)
174TEST_WEBRTCVIDEOFRAME(ConvertToBGRABufferInverted)
175TEST_WEBRTCVIDEOFRAME(ConvertToRAWBuffer)
176TEST_WEBRTCVIDEOFRAME(ConvertToRAWBufferStride)
177TEST_WEBRTCVIDEOFRAME(ConvertToRAWBufferInverted)
178TEST_WEBRTCVIDEOFRAME(ConvertToRGB24Buffer)
179TEST_WEBRTCVIDEOFRAME(ConvertToRGB24BufferStride)
180TEST_WEBRTCVIDEOFRAME(ConvertToRGB24BufferInverted)
181TEST_WEBRTCVIDEOFRAME(ConvertToRGB565Buffer)
182TEST_WEBRTCVIDEOFRAME(ConvertToRGB565BufferStride)
183TEST_WEBRTCVIDEOFRAME(ConvertToRGB565BufferInverted)
184TEST_WEBRTCVIDEOFRAME(ConvertToBayerBGGRBuffer)
185TEST_WEBRTCVIDEOFRAME(ConvertToBayerBGGRBufferStride)
186TEST_WEBRTCVIDEOFRAME(ConvertToBayerBGGRBufferInverted)
187TEST_WEBRTCVIDEOFRAME(ConvertToBayerGRBGBuffer)
188TEST_WEBRTCVIDEOFRAME(ConvertToBayerGRBGBufferStride)
189TEST_WEBRTCVIDEOFRAME(ConvertToBayerGRBGBufferInverted)
190TEST_WEBRTCVIDEOFRAME(ConvertToBayerGBRGBuffer)
191TEST_WEBRTCVIDEOFRAME(ConvertToBayerGBRGBufferStride)
192TEST_WEBRTCVIDEOFRAME(ConvertToBayerGBRGBufferInverted)
193TEST_WEBRTCVIDEOFRAME(ConvertToBayerRGGBBuffer)
194TEST_WEBRTCVIDEOFRAME(ConvertToBayerRGGBBufferStride)
195TEST_WEBRTCVIDEOFRAME(ConvertToBayerRGGBBufferInverted)
196TEST_WEBRTCVIDEOFRAME(ConvertToI400Buffer)
197TEST_WEBRTCVIDEOFRAME(ConvertToI400BufferStride)
198TEST_WEBRTCVIDEOFRAME(ConvertToI400BufferInverted)
199TEST_WEBRTCVIDEOFRAME(ConvertToYUY2Buffer)
200TEST_WEBRTCVIDEOFRAME(ConvertToYUY2BufferStride)
201TEST_WEBRTCVIDEOFRAME(ConvertToYUY2BufferInverted)
202TEST_WEBRTCVIDEOFRAME(ConvertToUYVYBuffer)
203TEST_WEBRTCVIDEOFRAME(ConvertToUYVYBufferStride)
204TEST_WEBRTCVIDEOFRAME(ConvertToUYVYBufferInverted)
205TEST_WEBRTCVIDEOFRAME(ConvertFromABGRBuffer)
206TEST_WEBRTCVIDEOFRAME(ConvertFromABGRBufferStride)
207TEST_WEBRTCVIDEOFRAME(ConvertFromABGRBufferInverted)
208TEST_WEBRTCVIDEOFRAME(ConvertFromARGB1555Buffer)
209TEST_WEBRTCVIDEOFRAME(ConvertFromARGB1555BufferStride)
210TEST_WEBRTCVIDEOFRAME(ConvertFromARGB1555BufferInverted)
211TEST_WEBRTCVIDEOFRAME(ConvertFromARGB4444Buffer)
212TEST_WEBRTCVIDEOFRAME(ConvertFromARGB4444BufferStride)
213TEST_WEBRTCVIDEOFRAME(ConvertFromARGB4444BufferInverted)
214TEST_WEBRTCVIDEOFRAME(ConvertFromARGBBuffer)
215TEST_WEBRTCVIDEOFRAME(ConvertFromARGBBufferStride)
216TEST_WEBRTCVIDEOFRAME(ConvertFromARGBBufferInverted)
217TEST_WEBRTCVIDEOFRAME(ConvertFromBGRABuffer)
218TEST_WEBRTCVIDEOFRAME(ConvertFromBGRABufferStride)
219TEST_WEBRTCVIDEOFRAME(ConvertFromBGRABufferInverted)
220TEST_WEBRTCVIDEOFRAME(ConvertFromRAWBuffer)
221TEST_WEBRTCVIDEOFRAME(ConvertFromRAWBufferStride)
222TEST_WEBRTCVIDEOFRAME(ConvertFromRAWBufferInverted)
223TEST_WEBRTCVIDEOFRAME(ConvertFromRGB24Buffer)
224TEST_WEBRTCVIDEOFRAME(ConvertFromRGB24BufferStride)
225TEST_WEBRTCVIDEOFRAME(ConvertFromRGB24BufferInverted)
226TEST_WEBRTCVIDEOFRAME(ConvertFromRGB565Buffer)
227TEST_WEBRTCVIDEOFRAME(ConvertFromRGB565BufferStride)
228TEST_WEBRTCVIDEOFRAME(ConvertFromRGB565BufferInverted)
229TEST_WEBRTCVIDEOFRAME(ConvertFromBayerBGGRBuffer)
230TEST_WEBRTCVIDEOFRAME(ConvertFromBayerBGGRBufferStride)
231TEST_WEBRTCVIDEOFRAME(ConvertFromBayerBGGRBufferInverted)
232TEST_WEBRTCVIDEOFRAME(ConvertFromBayerGRBGBuffer)
233TEST_WEBRTCVIDEOFRAME(ConvertFromBayerGRBGBufferStride)
234TEST_WEBRTCVIDEOFRAME(ConvertFromBayerGRBGBufferInverted)
235TEST_WEBRTCVIDEOFRAME(ConvertFromBayerGBRGBuffer)
236TEST_WEBRTCVIDEOFRAME(ConvertFromBayerGBRGBufferStride)
237TEST_WEBRTCVIDEOFRAME(ConvertFromBayerGBRGBufferInverted)
238TEST_WEBRTCVIDEOFRAME(ConvertFromBayerRGGBBuffer)
239TEST_WEBRTCVIDEOFRAME(ConvertFromBayerRGGBBufferStride)
240TEST_WEBRTCVIDEOFRAME(ConvertFromBayerRGGBBufferInverted)
241TEST_WEBRTCVIDEOFRAME(ConvertFromI400Buffer)
242TEST_WEBRTCVIDEOFRAME(ConvertFromI400BufferStride)
243TEST_WEBRTCVIDEOFRAME(ConvertFromI400BufferInverted)
244TEST_WEBRTCVIDEOFRAME(ConvertFromYUY2Buffer)
245TEST_WEBRTCVIDEOFRAME(ConvertFromYUY2BufferStride)
246TEST_WEBRTCVIDEOFRAME(ConvertFromYUY2BufferInverted)
247TEST_WEBRTCVIDEOFRAME(ConvertFromUYVYBuffer)
248TEST_WEBRTCVIDEOFRAME(ConvertFromUYVYBufferStride)
249TEST_WEBRTCVIDEOFRAME(ConvertFromUYVYBufferInverted)
250// TEST_WEBRTCVIDEOFRAME(ConvertToI422Buffer)
251TEST_WEBRTCVIDEOFRAME(ConvertARGBToBayerGRBG)
252TEST_WEBRTCVIDEOFRAME(ConvertARGBToBayerGBRG)
253TEST_WEBRTCVIDEOFRAME(ConvertARGBToBayerBGGR)
254TEST_WEBRTCVIDEOFRAME(ConvertARGBToBayerRGGB)
255TEST_WEBRTCVIDEOFRAME(CopyToBuffer)
256TEST_WEBRTCVIDEOFRAME(CopyToFrame)
257TEST_WEBRTCVIDEOFRAME(Write)
258TEST_WEBRTCVIDEOFRAME(CopyToBuffer1Pixel)
259// TEST_WEBRTCVIDEOFRAME(ConstructARGBBlackWhitePixel)
260
261TEST_WEBRTCVIDEOFRAME(StretchToFrame)
262TEST_WEBRTCVIDEOFRAME(Copy)
263TEST_WEBRTCVIDEOFRAME(CopyIsRef)
264TEST_WEBRTCVIDEOFRAME(MakeExclusive)
265
266// These functions test implementation-specific details.
wu@webrtc.org16d62542013-11-05 23:45:14 +0000[diff] [blame]267TEST_F(WebRtcVideoFrameTest, Alias) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000[diff] [blame]268 cricket::WebRtcVideoFrame frame1, frame2;
269 ASSERT_TRUE(LoadFrameNoRepeat(&frame1));
wu@webrtc.org16d62542013-11-05 23:45:14 +0000[diff] [blame]270 const int64 time_stamp = INT64_C(0x7FFFFFFFFFFFFFF0);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000[diff] [blame]271 frame1.SetTimeStamp(time_stamp);
272 EXPECT_EQ(time_stamp, frame1.GetTimeStamp());
wu@webrtc.org16d62542013-11-05 23:45:14 +0000[diff] [blame]273 frame2.Alias(frame1.frame()->Buffer(), frame1.frame()->Size(),
274 kWidth, kHeight, 1, 1,
275 frame1.GetElapsedTime(), frame1.GetTimeStamp(), 0);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000[diff] [blame]276 EXPECT_TRUE(IsEqual(frame1, frame2, 0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000[diff] [blame]277}
278
279// Tests the Init function with different cropped size.
280TEST_F(WebRtcVideoFrameTest, InitEvenSize) {
281 TestInit(640, 360);
282}
283
284TEST_F(WebRtcVideoFrameTest, InitOddWidth) {
285 TestInit(601, 480);
286}
287
288TEST_F(WebRtcVideoFrameTest, InitOddHeight) {
289 TestInit(360, 765);
290}
291
292TEST_F(WebRtcVideoFrameTest, InitOddWidthHeight) {
293 TestInit(355, 1021);
294}