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