mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
9e5defcf74
This CL implements `ChangeResolution()` to let `Y4mFrameGenerator` generate I420 frame with resolution other than y4m input by scaling. The code is mostly copied from `IvfVideoFrameGenerator`. The test case is also added for this change. Bug: webrtc:15210 Change-Id: I690e427a545a72d93ed39b77fd0f602054a30508 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/311521 Commit-Queue: Jianhui J Dai <jianhui.j.dai@intel.com> Reviewed-by: Erik Språng <sprang@webrtc.org> Cr-Commit-Position: refs/heads/main@{#40426}
101 lines
3.4 KiB
C++
101 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2023 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/testsupport/y4m_frame_generator.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include <string>
|
|
|
|
#include "absl/strings/string_view.h"
|
|
#include "api/scoped_refptr.h"
|
|
#include "api/video/i420_buffer.h"
|
|
#include "rtc_base/checks.h"
|
|
#include "test/testsupport/frame_reader.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
namespace {
|
|
// Reading 30 bytes from the Y4M header should be enough to get width
|
|
// and heigth.
|
|
// The header starts with: `YUV4MPEG2 W<WIDTH> H<HEIGTH>`.
|
|
constexpr int kHeaderBytesToRead = 30;
|
|
} // namespace
|
|
|
|
Y4mFrameGenerator::Y4mFrameGenerator(absl::string_view filename,
|
|
RepeatMode repeat_mode)
|
|
: filename_(filename), repeat_mode_(repeat_mode) {
|
|
// Read resolution from the Y4M header.
|
|
FILE* file = fopen(filename_.c_str(), "r");
|
|
RTC_CHECK(file != NULL) << "Cannot open " << filename_;
|
|
char header[kHeaderBytesToRead];
|
|
RTC_CHECK(fgets(header, sizeof(header), file) != nullptr)
|
|
<< "File " << filename_ << " is too small";
|
|
fclose(file);
|
|
int fps_denominator;
|
|
RTC_CHECK_EQ(sscanf(header, "YUV4MPEG2 W%zu H%zu F%i:%i", &width_, &height_,
|
|
&fps_, &fps_denominator),
|
|
4);
|
|
fps_ /= fps_denominator;
|
|
RTC_CHECK_GT(width_, 0);
|
|
RTC_CHECK_GT(height_, 0);
|
|
|
|
// Delegate the actual reads (from NextFrame) to a Y4mReader.
|
|
frame_reader_ = webrtc::test::CreateY4mFrameReader(
|
|
filename_, ToYuvFrameReaderRepeatMode(repeat_mode_));
|
|
}
|
|
|
|
Y4mFrameGenerator::VideoFrameData Y4mFrameGenerator::NextFrame() {
|
|
webrtc::VideoFrame::UpdateRect update_rect{0, 0, static_cast<int>(width_),
|
|
static_cast<int>(height_)};
|
|
rtc::scoped_refptr<webrtc::I420Buffer> next_frame_buffer =
|
|
frame_reader_->PullFrame();
|
|
|
|
if (!next_frame_buffer ||
|
|
(static_cast<size_t>(next_frame_buffer->width()) == width_ &&
|
|
static_cast<size_t>(next_frame_buffer->height()) == height_)) {
|
|
return VideoFrameData(next_frame_buffer, update_rect);
|
|
}
|
|
|
|
// Allocate a new buffer and return scaled version.
|
|
rtc::scoped_refptr<webrtc::I420Buffer> scaled_buffer(
|
|
I420Buffer::Create(width_, height_));
|
|
webrtc::I420Buffer::SetBlack(scaled_buffer.get());
|
|
scaled_buffer->ScaleFrom(*next_frame_buffer->ToI420());
|
|
return VideoFrameData(scaled_buffer, update_rect);
|
|
}
|
|
|
|
void Y4mFrameGenerator::ChangeResolution(size_t width, size_t height) {
|
|
width_ = width;
|
|
height_ = height;
|
|
RTC_CHECK_GT(width_, 0);
|
|
RTC_CHECK_GT(height_, 0);
|
|
}
|
|
|
|
FrameGeneratorInterface::Resolution Y4mFrameGenerator::GetResolution() const {
|
|
return {.width = width_, .height = height_};
|
|
}
|
|
|
|
YuvFrameReaderImpl::RepeatMode Y4mFrameGenerator::ToYuvFrameReaderRepeatMode(
|
|
RepeatMode repeat_mode) const {
|
|
switch (repeat_mode) {
|
|
case RepeatMode::kSingle:
|
|
return YuvFrameReaderImpl::RepeatMode::kSingle;
|
|
case RepeatMode::kLoop:
|
|
return YuvFrameReaderImpl::RepeatMode::kRepeat;
|
|
case RepeatMode::kPingPong:
|
|
return YuvFrameReaderImpl::RepeatMode::kPingPong;
|
|
}
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|