1 /*
2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "test/frame_utils.h"
12
13 #include <stdio.h>
14 #include <string.h>
15
16 #include "api/video/i420_buffer.h"
17 #include "api/video/video_frame.h"
18
19 namespace webrtc {
20 namespace test {
21
EqualPlane(const uint8_t * data1,const uint8_t * data2,int stride1,int stride2,int width,int height)22 bool EqualPlane(const uint8_t* data1,
23 const uint8_t* data2,
24 int stride1,
25 int stride2,
26 int width,
27 int height) {
28 for (int y = 0; y < height; ++y) {
29 if (memcmp(data1, data2, width) != 0)
30 return false;
31 data1 += stride1;
32 data2 += stride2;
33 }
34 return true;
35 }
36
FramesEqual(const webrtc::VideoFrame & f1,const webrtc::VideoFrame & f2)37 bool FramesEqual(const webrtc::VideoFrame& f1, const webrtc::VideoFrame& f2) {
38 if (f1.timestamp() != f2.timestamp() ||
39 f1.ntp_time_ms() != f2.ntp_time_ms() ||
40 f1.render_time_ms() != f2.render_time_ms()) {
41 return false;
42 }
43 return FrameBufsEqual(f1.video_frame_buffer(), f2.video_frame_buffer());
44 }
45
FrameBufsEqual(const rtc::scoped_refptr<webrtc::VideoFrameBuffer> & f1,const rtc::scoped_refptr<webrtc::VideoFrameBuffer> & f2)46 bool FrameBufsEqual(const rtc::scoped_refptr<webrtc::VideoFrameBuffer>& f1,
47 const rtc::scoped_refptr<webrtc::VideoFrameBuffer>& f2) {
48 if (f1 == f2) {
49 return true;
50 }
51 // Exlude nullptr (except if both are nullptr, as above)
52 if (!f1 || !f2) {
53 return false;
54 }
55
56 if (f1->width() != f2->width() || f1->height() != f2->height() ||
57 f1->type() != f2->type()) {
58 return false;
59 }
60
61 rtc::scoped_refptr<webrtc::I420BufferInterface> f1_i420 = f1->ToI420();
62 rtc::scoped_refptr<webrtc::I420BufferInterface> f2_i420 = f2->ToI420();
63 return EqualPlane(f1_i420->DataY(), f2_i420->DataY(), f1_i420->StrideY(),
64 f2_i420->StrideY(), f1_i420->width(), f1_i420->height()) &&
65 EqualPlane(f1_i420->DataU(), f2_i420->DataU(), f1_i420->StrideU(),
66 f2_i420->StrideU(), f1_i420->ChromaWidth(),
67 f1_i420->ChromaHeight()) &&
68 EqualPlane(f1_i420->DataV(), f2_i420->DataV(), f1_i420->StrideV(),
69 f2_i420->StrideV(), f1_i420->ChromaWidth(),
70 f1_i420->ChromaHeight());
71 }
72
ReadI420Buffer(int width,int height,FILE * f)73 rtc::scoped_refptr<I420Buffer> ReadI420Buffer(int width, int height, FILE* f) {
74 int half_width = (width + 1) / 2;
75 rtc::scoped_refptr<I420Buffer> buffer(
76 // Explicit stride, no padding between rows.
77 I420Buffer::Create(width, height, width, half_width, half_width));
78 size_t size_y = static_cast<size_t>(width) * height;
79 size_t size_uv = static_cast<size_t>(half_width) * ((height + 1) / 2);
80
81 if (fread(buffer->MutableDataY(), 1, size_y, f) < size_y)
82 return nullptr;
83 if (fread(buffer->MutableDataU(), 1, size_uv, f) < size_uv)
84 return nullptr;
85 if (fread(buffer->MutableDataV(), 1, size_uv, f) < size_uv)
86 return nullptr;
87 return buffer;
88 }
89
90 } // namespace test
91 } // namespace webrtc
92