Adding upstream version 115.7.0esr.upstream/115.7.0esr

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

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