summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/test/frame_generator_unittest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/libwebrtc/test/frame_generator_unittest.cc')
-rw-r--r--third_party/libwebrtc/test/frame_generator_unittest.cc284
1 files changed, 284 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/frame_generator_unittest.cc b/third_party/libwebrtc/test/frame_generator_unittest.cc
new file mode 100644
index 0000000000..ece37a547f
--- /dev/null
+++ b/third_party/libwebrtc/test/frame_generator_unittest.cc
@@ -0,0 +1,284 @@
+/*
+ * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
+ *
+ * Use of this source code is governed by a BSD-style license
+ * that can be found in the LICENSE file in the root of the source
+ * tree. An additional intellectual property rights grant can be found
+ * in the file PATENTS. All contributing project authors may
+ * be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "test/frame_generator.h"
+
+#include <stdio.h>
+#include <string.h>
+
+#include <cstdint>
+#include <memory>
+#include <string>
+
+#include "api/scoped_refptr.h"
+#include "api/test/create_frame_generator.h"
+#include "api/test/frame_generator_interface.h"
+#include "api/video/video_frame_buffer.h"
+#include "test/gtest.h"
+#include "test/testsupport/file_utils.h"
+
+namespace webrtc {
+namespace test {
+
+constexpr int kFrameWidth = 4;
+constexpr int kFrameHeight = 4;
+constexpr int y_size = kFrameWidth * kFrameHeight;
+constexpr int uv_size = ((kFrameHeight + 1) / 2) * ((kFrameWidth + 1) / 2);
+
+class FrameGeneratorTest : public ::testing::Test {
+ public:
+ void SetUp() override {
+ two_frame_yuv_filename_ =
+ test::TempFilename(test::OutputPath(), "2_frame_yuv_file");
+ one_frame_yuv_filename_ =
+ test::TempFilename(test::OutputPath(), "1_frame_yuv_file");
+ two_frame_nv12_filename_ =
+ test::TempFilename(test::OutputPath(), "2_frame_nv12_file");
+ one_frame_nv12_filename_ =
+ test::TempFilename(test::OutputPath(), "1_frame_nv12_file");
+
+ FILE* file = fopen(two_frame_yuv_filename_.c_str(), "wb");
+ WriteYuvFile(file, 0, 0, 0);
+ WriteYuvFile(file, 127, 128, 129);
+ fclose(file);
+ file = fopen(one_frame_yuv_filename_.c_str(), "wb");
+ WriteYuvFile(file, 255, 255, 255);
+ fclose(file);
+ file = fopen(two_frame_nv12_filename_.c_str(), "wb");
+ WriteNV12File(file, 0, 0, 0);
+ WriteNV12File(file, 127, 128, 129);
+ fclose(file);
+ file = fopen(one_frame_nv12_filename_.c_str(), "wb");
+ WriteNV12File(file, 255, 255, 255);
+ fclose(file);
+ }
+
+ void TearDown() override {
+ remove(one_frame_yuv_filename_.c_str());
+ remove(two_frame_yuv_filename_.c_str());
+ remove(one_frame_nv12_filename_.c_str());
+ remove(two_frame_nv12_filename_.c_str());
+ }
+
+ protected:
+ void WriteYuvFile(FILE* file, uint8_t y, uint8_t u, uint8_t v) {
+ RTC_DCHECK(file);
+ std::unique_ptr<uint8_t[]> plane_buffer(new uint8_t[y_size]);
+ memset(plane_buffer.get(), y, y_size);
+ fwrite(plane_buffer.get(), 1, y_size, file);
+ memset(plane_buffer.get(), u, uv_size);
+ fwrite(plane_buffer.get(), 1, uv_size, file);
+ memset(plane_buffer.get(), v, uv_size);
+ fwrite(plane_buffer.get(), 1, uv_size, file);
+ }
+
+ void WriteNV12File(FILE* file, uint8_t y, uint8_t u, uint8_t v) {
+ RTC_DCHECK(file);
+ uint8_t plane_buffer[y_size];
+
+ memset(&plane_buffer, y, y_size);
+ fwrite(&plane_buffer, 1, y_size, file);
+ for (size_t i = 0; i < uv_size; ++i) {
+ plane_buffer[2 * i] = u;
+ plane_buffer[2 * i + 1] = v;
+ }
+ fwrite(&plane_buffer, 1, 2 * uv_size, file);
+ }
+
+ void CheckFrameAndMutate(const FrameGeneratorInterface::VideoFrameData& frame,
+ uint8_t y,
+ uint8_t u,
+ uint8_t v) {
+ // Check that frame is valid, has the correct color and timestamp are clean.
+ rtc::scoped_refptr<I420BufferInterface> i420_buffer =
+ frame.buffer->ToI420();
+ const uint8_t* buffer;
+ buffer = i420_buffer->DataY();
+ for (int i = 0; i < y_size; ++i)
+ ASSERT_EQ(y, buffer[i]);
+ buffer = i420_buffer->DataU();
+ for (int i = 0; i < uv_size; ++i)
+ ASSERT_EQ(u, buffer[i]);
+ buffer = i420_buffer->DataV();
+ for (int i = 0; i < uv_size; ++i)
+ ASSERT_EQ(v, buffer[i]);
+ }
+
+ uint64_t Hash(const FrameGeneratorInterface::VideoFrameData& frame) {
+ // Generate a 64-bit hash from the frame's buffer.
+ uint64_t hash = 19;
+ rtc::scoped_refptr<I420BufferInterface> i420_buffer =
+ frame.buffer->ToI420();
+ const uint8_t* buffer = i420_buffer->DataY();
+ for (int i = 0; i < y_size; ++i) {
+ hash = (37 * hash) + buffer[i];
+ }
+ buffer = i420_buffer->DataU();
+ for (int i = 0; i < uv_size; ++i) {
+ hash = (37 * hash) + buffer[i];
+ }
+ buffer = i420_buffer->DataV();
+ for (int i = 0; i < uv_size; ++i) {
+ hash = (37 * hash) + buffer[i];
+ }
+ return hash;
+ }
+
+ std::string two_frame_yuv_filename_;
+ std::string one_frame_yuv_filename_;
+ std::string two_frame_nv12_filename_;
+ std::string one_frame_nv12_filename_;
+};
+
+TEST_F(FrameGeneratorTest, SingleFrameYuvFile) {
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromYuvFileFrameGenerator(
+ std::vector<std::string>(1, one_frame_yuv_filename_), kFrameWidth,
+ kFrameHeight, 1));
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+}
+
+TEST_F(FrameGeneratorTest, TwoFrameYuvFile) {
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromYuvFileFrameGenerator(
+ std::vector<std::string>(1, two_frame_yuv_filename_), kFrameWidth,
+ kFrameHeight, 1));
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, MultipleFrameYuvFiles) {
+ std::vector<std::string> files;
+ files.push_back(two_frame_yuv_filename_);
+ files.push_back(one_frame_yuv_filename_);
+
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromYuvFileFrameGenerator(files, kFrameWidth, kFrameHeight, 1));
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, TwoFrameYuvFileWithRepeat) {
+ const int kRepeatCount = 3;
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromYuvFileFrameGenerator(
+ std::vector<std::string>(1, two_frame_yuv_filename_), kFrameWidth,
+ kFrameHeight, kRepeatCount));
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, MultipleFrameYuvFilesWithRepeat) {
+ const int kRepeatCount = 3;
+ std::vector<std::string> files;
+ files.push_back(two_frame_yuv_filename_);
+ files.push_back(one_frame_yuv_filename_);
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromYuvFileFrameGenerator(files, kFrameWidth, kFrameHeight,
+ kRepeatCount));
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, SingleFrameNV12File) {
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromNV12FileFrameGenerator(
+ std::vector<std::string>(1, one_frame_nv12_filename_), kFrameWidth,
+ kFrameHeight, 1));
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+}
+
+TEST_F(FrameGeneratorTest, TwoFrameNV12File) {
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromNV12FileFrameGenerator(
+ std::vector<std::string>(1, two_frame_nv12_filename_), kFrameWidth,
+ kFrameHeight, 1));
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, MultipleFrameNV12Files) {
+ std::vector<std::string> files;
+ files.push_back(two_frame_nv12_filename_);
+ files.push_back(one_frame_nv12_filename_);
+
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromNV12FileFrameGenerator(files, kFrameWidth, kFrameHeight, 1));
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, TwoFrameNV12FileWithRepeat) {
+ const int kRepeatCount = 3;
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromNV12FileFrameGenerator(
+ std::vector<std::string>(1, two_frame_nv12_filename_), kFrameWidth,
+ kFrameHeight, kRepeatCount));
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, MultipleFrameNV12FilesWithRepeat) {
+ const int kRepeatCount = 3;
+ std::vector<std::string> files;
+ files.push_back(two_frame_nv12_filename_);
+ files.push_back(one_frame_nv12_filename_);
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateFromNV12FileFrameGenerator(files, kFrameWidth, kFrameHeight,
+ kRepeatCount));
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 127, 128, 129);
+ for (int i = 0; i < kRepeatCount; ++i)
+ CheckFrameAndMutate(generator->NextFrame(), 255, 255, 255);
+ CheckFrameAndMutate(generator->NextFrame(), 0, 0, 0);
+}
+
+TEST_F(FrameGeneratorTest, SlideGenerator) {
+ const int kGenCount = 9;
+ const int kRepeatCount = 3;
+ std::unique_ptr<FrameGeneratorInterface> generator(
+ CreateSlideFrameGenerator(kFrameWidth, kFrameHeight, kRepeatCount));
+ uint64_t hashes[kGenCount];
+ for (int i = 0; i < kGenCount; ++i) {
+ hashes[i] = Hash(generator->NextFrame());
+ }
+ // Check that the buffer changes only every `kRepeatCount` frames.
+ for (int i = 1; i < kGenCount; ++i) {
+ if (i % kRepeatCount == 0) {
+ EXPECT_NE(hashes[i - 1], hashes[i]);
+ } else {
+ EXPECT_EQ(hashes[i - 1], hashes[i]);
+ }
+ }
+}
+
+} // namespace test
+} // namespace webrtc
generated by cgit v1.2.3 (git 2.39.1) at 2024-08-28 16:24:08 +0000