1 /*
2 * Copyright (c) 2019 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/testsupport/video_frame_writer.h"
12
13 #include <stdint.h>
14 #include <stdio.h>
15 #include <string.h>
16
17 #include <memory>
18 #include <string>
19
20 #include "absl/strings/string_view.h"
21 #include "api/test/video/video_frame_writer.h"
22 #include "api/video/i420_buffer.h"
23 #include "test/gtest.h"
24 #include "test/testsupport/file_utils.h"
25 #include "test/testsupport/frame_reader.h"
26
27 namespace webrtc {
28 namespace test {
29 namespace {
30
31 const size_t kFrameWidth = 50;
32 const size_t kFrameHeight = 20;
33 const size_t kFrameLength = 3 * kFrameWidth * kFrameHeight / 2; // I420.
34 const size_t kFrameRate = 30;
35
36 // Size of header: "YUV4MPEG2 W50 H20 F30:1 C420\n"
37 const size_t kFileHeaderSize = 29;
38 // Size of header: "FRAME\n"
39 const size_t kFrameHeaderSize = 6;
40
CreateI420Buffer(int width,int height)41 rtc::scoped_refptr<I420Buffer> CreateI420Buffer(int width, int height) {
42 rtc::scoped_refptr<I420Buffer> buffer(I420Buffer::Create(width, height));
43 for (int x = 0; x < width; x++) {
44 for (int y = 0; y < height; y++) {
45 buffer->MutableDataY()[x + y * width] = 128;
46 }
47 }
48 int chroma_width = buffer->ChromaWidth();
49 int chroma_height = buffer->ChromaHeight();
50 for (int x = 0; x < chroma_width; x++) {
51 for (int y = 0; y < chroma_height; y++) {
52 buffer->MutableDataU()[x + y * chroma_width] = 1;
53 buffer->MutableDataV()[x + y * chroma_width] = 255;
54 }
55 }
56 return buffer;
57 }
58
AssertI420BuffersEq(rtc::scoped_refptr<webrtc::I420BufferInterface> actual,rtc::scoped_refptr<webrtc::I420BufferInterface> expected)59 void AssertI420BuffersEq(
60 rtc::scoped_refptr<webrtc::I420BufferInterface> actual,
61 rtc::scoped_refptr<webrtc::I420BufferInterface> expected) {
62 ASSERT_TRUE(actual);
63
64 ASSERT_EQ(actual->width(), expected->width());
65 ASSERT_EQ(actual->height(), expected->height());
66 const int width = expected->width();
67 const int height = expected->height();
68 for (int x = 0; x < width; x++) {
69 for (int y = 0; y < height; y++) {
70 ASSERT_EQ(actual->DataY()[x + y * width],
71 expected->DataY()[x + y * width]);
72 }
73 }
74
75 ASSERT_EQ(actual->ChromaWidth(), expected->ChromaWidth());
76 ASSERT_EQ(actual->ChromaHeight(), expected->ChromaHeight());
77 int chroma_width = expected->ChromaWidth();
78 int chroma_height = expected->ChromaHeight();
79 for (int x = 0; x < chroma_width; x++) {
80 for (int y = 0; y < chroma_height; y++) {
81 ASSERT_EQ(actual->DataU()[x + y * chroma_width],
82 expected->DataU()[x + y * chroma_width]);
83 ASSERT_EQ(actual->DataV()[x + y * chroma_width],
84 expected->DataV()[x + y * chroma_width]);
85 }
86 }
87 }
88
89 } // namespace
90
91 class VideoFrameWriterTest : public ::testing::Test {
92 protected:
93 VideoFrameWriterTest() = default;
94 ~VideoFrameWriterTest() override = default;
95
SetUp()96 void SetUp() override {
97 temp_filename_ = webrtc::test::TempFilename(webrtc::test::OutputPath(),
98 "video_frame_writer_unittest");
99 frame_writer_ = CreateFrameWriter();
100 }
101
102 virtual std::unique_ptr<VideoFrameWriter> CreateFrameWriter() = 0;
103
TearDown()104 void TearDown() override { remove(temp_filename_.c_str()); }
105
106 std::unique_ptr<VideoFrameWriter> frame_writer_;
107 std::string temp_filename_;
108 };
109
110 class Y4mVideoFrameWriterTest : public VideoFrameWriterTest {
111 protected:
CreateFrameWriter()112 std::unique_ptr<VideoFrameWriter> CreateFrameWriter() override {
113 return std::make_unique<Y4mVideoFrameWriterImpl>(
114 temp_filename_, kFrameWidth, kFrameHeight, kFrameRate);
115 }
116 };
117
118 class YuvVideoFrameWriterTest : public VideoFrameWriterTest {
119 protected:
CreateFrameWriter()120 std::unique_ptr<VideoFrameWriter> CreateFrameWriter() override {
121 return std::make_unique<YuvVideoFrameWriterImpl>(temp_filename_,
122 kFrameWidth, kFrameHeight);
123 }
124 };
125
TEST_F(Y4mVideoFrameWriterTest,InitSuccess)126 TEST_F(Y4mVideoFrameWriterTest, InitSuccess) {}
127
TEST_F(Y4mVideoFrameWriterTest,WriteFrame)128 TEST_F(Y4mVideoFrameWriterTest, WriteFrame) {
129 rtc::scoped_refptr<I420Buffer> expected_buffer =
130 CreateI420Buffer(kFrameWidth, kFrameHeight);
131
132 VideoFrame frame =
133 VideoFrame::Builder().set_video_frame_buffer(expected_buffer).build();
134
135 ASSERT_TRUE(frame_writer_->WriteFrame(frame));
136 ASSERT_TRUE(frame_writer_->WriteFrame(frame));
137
138 frame_writer_->Close();
139 EXPECT_EQ(kFileHeaderSize + 2 * kFrameHeaderSize + 2 * kFrameLength,
140 GetFileSize(temp_filename_));
141
142 std::unique_ptr<FrameReader> frame_reader =
143 std::make_unique<Y4mFrameReaderImpl>(temp_filename_, kFrameWidth,
144 kFrameHeight);
145 ASSERT_TRUE(frame_reader->Init());
146 AssertI420BuffersEq(frame_reader->ReadFrame(), expected_buffer);
147 AssertI420BuffersEq(frame_reader->ReadFrame(), expected_buffer);
148 EXPECT_FALSE(frame_reader->ReadFrame()); // End of file.
149 frame_reader->Close();
150 }
151
TEST_F(YuvVideoFrameWriterTest,InitSuccess)152 TEST_F(YuvVideoFrameWriterTest, InitSuccess) {}
153
TEST_F(YuvVideoFrameWriterTest,WriteFrame)154 TEST_F(YuvVideoFrameWriterTest, WriteFrame) {
155 rtc::scoped_refptr<I420Buffer> expected_buffer =
156 CreateI420Buffer(kFrameWidth, kFrameHeight);
157
158 VideoFrame frame =
159 VideoFrame::Builder().set_video_frame_buffer(expected_buffer).build();
160
161 ASSERT_TRUE(frame_writer_->WriteFrame(frame));
162 ASSERT_TRUE(frame_writer_->WriteFrame(frame));
163
164 frame_writer_->Close();
165 EXPECT_EQ(2 * kFrameLength, GetFileSize(temp_filename_));
166
167 std::unique_ptr<FrameReader> frame_reader =
168 std::make_unique<YuvFrameReaderImpl>(temp_filename_, kFrameWidth,
169 kFrameHeight);
170 ASSERT_TRUE(frame_reader->Init());
171 AssertI420BuffersEq(frame_reader->ReadFrame(), expected_buffer);
172 AssertI420BuffersEq(frame_reader->ReadFrame(), expected_buffer);
173 EXPECT_FALSE(frame_reader->ReadFrame()); // End of file.
174 frame_reader->Close();
175 }
176
177 } // namespace test
178 } // namespace webrtc
179