mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 05:40:42 +01:00
c230da0f1b
To reuse same environment in video encoder and thus avoid creating duplicated environment. Bug: webrtc:15860, b/326933307 Change-Id: I1c56966301a9b453d615c45626407fede2a6d8b5 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/344143 Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Cr-Commit-Position: refs/heads/main@{#41956}
110 lines
3.6 KiB
C++
110 lines
3.6 KiB
C++
/*
|
|
* Copyright (c) 2019 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 "api/test/create_frame_generator.h"
|
|
|
|
#include <cstdio>
|
|
#include <utility>
|
|
|
|
#include "absl/base/nullability.h"
|
|
#include "absl/strings/string_view.h"
|
|
#include "api/environment/environment_factory.h"
|
|
#include "rtc_base/checks.h"
|
|
#include "test/frame_generator.h"
|
|
#include "test/testsupport/ivf_video_frame_generator.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
std::unique_ptr<FrameGeneratorInterface> CreateSquareFrameGenerator(
|
|
int width,
|
|
int height,
|
|
absl::optional<FrameGeneratorInterface::OutputType> type,
|
|
absl::optional<int> num_squares) {
|
|
return std::make_unique<SquareGenerator>(
|
|
width, height, type.value_or(FrameGeneratorInterface::OutputType::kI420),
|
|
num_squares.value_or(10));
|
|
}
|
|
|
|
std::unique_ptr<FrameGeneratorInterface> CreateFromYuvFileFrameGenerator(
|
|
std::vector<std::string> filenames,
|
|
size_t width,
|
|
size_t height,
|
|
int frame_repeat_count) {
|
|
RTC_DCHECK(!filenames.empty());
|
|
std::vector<FILE*> files;
|
|
for (const std::string& filename : filenames) {
|
|
FILE* file = fopen(filename.c_str(), "rb");
|
|
RTC_DCHECK(file != nullptr) << "Failed to open: '" << filename << "'\n";
|
|
files.push_back(file);
|
|
}
|
|
|
|
return std::make_unique<YuvFileGenerator>(files, width, height,
|
|
frame_repeat_count);
|
|
}
|
|
|
|
std::unique_ptr<FrameGeneratorInterface> CreateFromNV12FileFrameGenerator(
|
|
std::vector<std::string> filenames,
|
|
size_t width,
|
|
size_t height,
|
|
int frame_repeat_count) {
|
|
RTC_DCHECK(!filenames.empty());
|
|
std::vector<FILE*> files;
|
|
for (const std::string& filename : filenames) {
|
|
FILE* file = fopen(filename.c_str(), "rb");
|
|
RTC_DCHECK(file != nullptr) << "Failed to open: '" << filename << "'\n";
|
|
files.push_back(file);
|
|
}
|
|
|
|
return std::make_unique<NV12FileGenerator>(files, width, height,
|
|
frame_repeat_count);
|
|
}
|
|
|
|
std::unique_ptr<FrameGeneratorInterface> CreateFromIvfFileFrameGenerator(
|
|
std::string filename) {
|
|
return CreateFromIvfFileFrameGenerator(CreateEnvironment(), filename);
|
|
}
|
|
|
|
absl::Nonnull<std::unique_ptr<FrameGeneratorInterface>>
|
|
CreateFromIvfFileFrameGenerator(const Environment& env,
|
|
absl::string_view filename) {
|
|
return std::make_unique<IvfVideoFrameGenerator>(env, filename);
|
|
}
|
|
|
|
std::unique_ptr<FrameGeneratorInterface>
|
|
CreateScrollingInputFromYuvFilesFrameGenerator(
|
|
Clock* clock,
|
|
std::vector<std::string> filenames,
|
|
size_t source_width,
|
|
size_t source_height,
|
|
size_t target_width,
|
|
size_t target_height,
|
|
int64_t scroll_time_ms,
|
|
int64_t pause_time_ms) {
|
|
RTC_DCHECK(!filenames.empty());
|
|
std::vector<FILE*> files;
|
|
for (const std::string& filename : filenames) {
|
|
FILE* file = fopen(filename.c_str(), "rb");
|
|
RTC_DCHECK(file != nullptr);
|
|
files.push_back(file);
|
|
}
|
|
|
|
return std::make_unique<ScrollingImageFrameGenerator>(
|
|
clock, files, source_width, source_height, target_width, target_height,
|
|
scroll_time_ms, pause_time_ms);
|
|
}
|
|
|
|
std::unique_ptr<FrameGeneratorInterface>
|
|
CreateSlideFrameGenerator(int width, int height, int frame_repeat_count) {
|
|
return std::make_unique<SlideGenerator>(width, height, frame_repeat_count);
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|