sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2015 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 | |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 11 | #include "test/frame_generator.h" |
| 12 | |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 13 | #include <assert.h> |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 14 | #include <stdio.h> |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 15 | #include <string.h> |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 16 | |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 17 | #include <cstdint> |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 18 | #include <memory> |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 19 | #include <string> |
| 20 | |
Mirko Bonadei | d970807 | 2019-01-25 20:26:48 +0100 | [diff] [blame] | 21 | #include "api/scoped_refptr.h" |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 22 | #include "api/video/video_frame_buffer.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 23 | #include "test/gtest.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 24 | #include "test/testsupport/file_utils.h" |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 25 | |
| 26 | namespace webrtc { |
| 27 | namespace test { |
| 28 | |
| 29 | static const int kFrameWidth = 4; |
| 30 | static const int kFrameHeight = 4; |
| 31 | |
| 32 | class FrameGeneratorTest : public ::testing::Test { |
| 33 | public: |
| 34 | void SetUp() override { |
| 35 | two_frame_filename_ = |
| 36 | test::TempFilename(test::OutputPath(), "2_frame_yuv_file"); |
| 37 | one_frame_filename_ = |
| 38 | test::TempFilename(test::OutputPath(), "1_frame_yuv_file"); |
| 39 | |
| 40 | FILE* file = fopen(two_frame_filename_.c_str(), "wb"); |
| 41 | WriteYuvFile(file, 0, 0, 0); |
| 42 | WriteYuvFile(file, 127, 127, 127); |
| 43 | fclose(file); |
| 44 | file = fopen(one_frame_filename_.c_str(), "wb"); |
| 45 | WriteYuvFile(file, 255, 255, 255); |
| 46 | fclose(file); |
| 47 | } |
| 48 | void TearDown() override { |
| 49 | remove(one_frame_filename_.c_str()); |
| 50 | remove(two_frame_filename_.c_str()); |
| 51 | } |
| 52 | |
| 53 | protected: |
| 54 | void WriteYuvFile(FILE* file, uint8_t y, uint8_t u, uint8_t v) { |
| 55 | assert(file); |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 56 | std::unique_ptr<uint8_t[]> plane_buffer(new uint8_t[y_size]); |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 57 | memset(plane_buffer.get(), y, y_size); |
| 58 | fwrite(plane_buffer.get(), 1, y_size, file); |
| 59 | memset(plane_buffer.get(), u, uv_size); |
| 60 | fwrite(plane_buffer.get(), 1, uv_size, file); |
| 61 | memset(plane_buffer.get(), v, uv_size); |
| 62 | fwrite(plane_buffer.get(), 1, uv_size, file); |
| 63 | } |
| 64 | |
Miguel Casas-Sanchez | 4765070 | 2015-05-29 17:21:40 -0700 | [diff] [blame] | 65 | void CheckFrameAndMutate(VideoFrame* frame, uint8_t y, uint8_t u, uint8_t v) { |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 66 | // Check that frame is valid, has the correct color and timestamp are clean. |
| 67 | ASSERT_NE(nullptr, frame); |
Magnus Jedvert | 90e3190 | 2017-06-07 11:32:50 +0200 | [diff] [blame] | 68 | rtc::scoped_refptr<I420BufferInterface> i420_buffer = |
| 69 | frame->video_frame_buffer()->ToI420(); |
nisse | c9c142f | 2016-05-17 04:05:47 -0700 | [diff] [blame] | 70 | const uint8_t* buffer; |
Magnus Jedvert | 90e3190 | 2017-06-07 11:32:50 +0200 | [diff] [blame] | 71 | buffer = i420_buffer->DataY(); |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 72 | for (int i = 0; i < y_size; ++i) |
| 73 | ASSERT_EQ(y, buffer[i]); |
Magnus Jedvert | 90e3190 | 2017-06-07 11:32:50 +0200 | [diff] [blame] | 74 | buffer = i420_buffer->DataU(); |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 75 | for (int i = 0; i < uv_size; ++i) |
| 76 | ASSERT_EQ(u, buffer[i]); |
Magnus Jedvert | 90e3190 | 2017-06-07 11:32:50 +0200 | [diff] [blame] | 77 | buffer = i420_buffer->DataV(); |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 78 | for (int i = 0; i < uv_size; ++i) |
| 79 | ASSERT_EQ(v, buffer[i]); |
| 80 | EXPECT_EQ(0, frame->ntp_time_ms()); |
| 81 | EXPECT_EQ(0, frame->render_time_ms()); |
| 82 | EXPECT_EQ(0u, frame->timestamp()); |
| 83 | |
| 84 | // Mutate to something arbitrary non-zero. |
| 85 | frame->set_ntp_time_ms(11); |
nisse | 1c0dea8 | 2017-01-30 02:43:18 -0800 | [diff] [blame] | 86 | frame->set_timestamp_us(12); |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 87 | frame->set_timestamp(13); |
| 88 | } |
| 89 | |
erikvarga | 579de6f | 2017-08-29 09:12:57 -0700 | [diff] [blame] | 90 | uint64_t Hash(VideoFrame* frame) { |
| 91 | // Generate a 64-bit hash from the frame's buffer. |
| 92 | uint64_t hash = 19; |
| 93 | rtc::scoped_refptr<I420BufferInterface> i420_buffer = |
| 94 | frame->video_frame_buffer()->ToI420(); |
| 95 | const uint8_t* buffer = i420_buffer->DataY(); |
| 96 | for (int i = 0; i < y_size; ++i) { |
| 97 | hash = (37 * hash) + buffer[i]; |
| 98 | } |
| 99 | buffer = i420_buffer->DataU(); |
| 100 | for (int i = 0; i < uv_size; ++i) { |
| 101 | hash = (37 * hash) + buffer[i]; |
| 102 | } |
| 103 | buffer = i420_buffer->DataV(); |
| 104 | for (int i = 0; i < uv_size; ++i) { |
| 105 | hash = (37 * hash) + buffer[i]; |
| 106 | } |
| 107 | return hash; |
| 108 | } |
| 109 | |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 110 | std::string two_frame_filename_; |
| 111 | std::string one_frame_filename_; |
| 112 | const int y_size = kFrameWidth * kFrameHeight; |
| 113 | const int uv_size = ((kFrameHeight + 1) / 2) * ((kFrameWidth + 1) / 2); |
| 114 | }; |
| 115 | |
| 116 | TEST_F(FrameGeneratorTest, SingleFrameFile) { |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 117 | std::unique_ptr<FrameGenerator> generator(FrameGenerator::CreateFromYuvFile( |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 118 | std::vector<std::string>(1, one_frame_filename_), kFrameWidth, |
| 119 | kFrameHeight, 1)); |
| 120 | CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255); |
| 121 | CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255); |
| 122 | } |
| 123 | |
| 124 | TEST_F(FrameGeneratorTest, TwoFrameFile) { |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 125 | std::unique_ptr<FrameGenerator> generator(FrameGenerator::CreateFromYuvFile( |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 126 | std::vector<std::string>(1, two_frame_filename_), kFrameWidth, |
| 127 | kFrameHeight, 1)); |
| 128 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 129 | CheckFrameAndMutate(generator->NextFrame(), 127, 127, 127); |
| 130 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 131 | } |
| 132 | |
| 133 | TEST_F(FrameGeneratorTest, MultipleFrameFiles) { |
| 134 | std::vector<std::string> files; |
| 135 | files.push_back(two_frame_filename_); |
| 136 | files.push_back(one_frame_filename_); |
| 137 | |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 138 | std::unique_ptr<FrameGenerator> generator( |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 139 | FrameGenerator::CreateFromYuvFile(files, kFrameWidth, kFrameHeight, 1)); |
| 140 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 141 | CheckFrameAndMutate(generator->NextFrame(), 127, 127, 127); |
| 142 | CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255); |
| 143 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 144 | } |
| 145 | |
| 146 | TEST_F(FrameGeneratorTest, TwoFrameFileWithRepeat) { |
| 147 | const int kRepeatCount = 3; |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 148 | std::unique_ptr<FrameGenerator> generator(FrameGenerator::CreateFromYuvFile( |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 149 | std::vector<std::string>(1, two_frame_filename_), kFrameWidth, |
| 150 | kFrameHeight, kRepeatCount)); |
| 151 | for (int i = 0; i < kRepeatCount; ++i) |
| 152 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 153 | for (int i = 0; i < kRepeatCount; ++i) |
| 154 | CheckFrameAndMutate(generator->NextFrame(), 127, 127, 127); |
| 155 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 156 | } |
| 157 | |
| 158 | TEST_F(FrameGeneratorTest, MultipleFrameFilesWithRepeat) { |
| 159 | const int kRepeatCount = 3; |
| 160 | std::vector<std::string> files; |
| 161 | files.push_back(two_frame_filename_); |
| 162 | files.push_back(one_frame_filename_); |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 163 | std::unique_ptr<FrameGenerator> generator(FrameGenerator::CreateFromYuvFile( |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 164 | files, kFrameWidth, kFrameHeight, kRepeatCount)); |
| 165 | for (int i = 0; i < kRepeatCount; ++i) |
| 166 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 167 | for (int i = 0; i < kRepeatCount; ++i) |
| 168 | CheckFrameAndMutate(generator->NextFrame(), 127, 127, 127); |
| 169 | for (int i = 0; i < kRepeatCount; ++i) |
| 170 | CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255); |
| 171 | CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0); |
| 172 | } |
| 173 | |
erikvarga | 579de6f | 2017-08-29 09:12:57 -0700 | [diff] [blame] | 174 | TEST_F(FrameGeneratorTest, SlideGenerator) { |
| 175 | const int kGenCount = 9; |
| 176 | const int kRepeatCount = 3; |
| 177 | std::unique_ptr<FrameGenerator> generator( |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 178 | FrameGenerator::CreateSlideGenerator(kFrameWidth, kFrameHeight, |
| 179 | kRepeatCount)); |
erikvarga | 579de6f | 2017-08-29 09:12:57 -0700 | [diff] [blame] | 180 | uint64_t hashes[kGenCount]; |
| 181 | for (int i = 0; i < kGenCount; ++i) { |
| 182 | hashes[i] = Hash(generator->NextFrame()); |
| 183 | } |
| 184 | // Check that the buffer changes only every |kRepeatCount| frames. |
| 185 | for (int i = 1; i < kGenCount; ++i) { |
| 186 | if (i % kRepeatCount == 0) { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 187 | EXPECT_NE(hashes[i - 1], hashes[i]); |
erikvarga | 579de6f | 2017-08-29 09:12:57 -0700 | [diff] [blame] | 188 | } else { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 189 | EXPECT_EQ(hashes[i - 1], hashes[i]); |
erikvarga | 579de6f | 2017-08-29 09:12:57 -0700 | [diff] [blame] | 190 | } |
| 191 | } |
| 192 | } |
| 193 | |
sprang@webrtc.org | 25dd1db | 2015-03-02 11:55:45 +0000 | [diff] [blame] | 194 | } // namespace test |
| 195 | } // namespace webrtc |