ad6fc5a05c
This was never turned on, contains a lot of complexity and somehow manages triggering a bug in a downstream project. BUG=webrtc:5066 R=marpan@webrtc.org TBR=mflodman@webrtc.org Review URL: https://codereview.webrtc.org/1917323002 . Cr-Commit-Position: refs/heads/master@{#12692}
127 lines
3.4 KiB
C++
127 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2012 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 "webrtc/modules/video_processing/frame_preprocessor.h"
|
|
|
|
#include "webrtc/modules/video_processing/video_denoiser.h"
|
|
|
|
namespace webrtc {
|
|
|
|
VPMFramePreprocessor::VPMFramePreprocessor()
|
|
: resampled_frame_(), frame_cnt_(0) {
|
|
spatial_resampler_ = new VPMSimpleSpatialResampler();
|
|
vd_ = new VPMVideoDecimator();
|
|
EnableDenoising(false);
|
|
denoised_frame_toggle_ = 0;
|
|
}
|
|
|
|
VPMFramePreprocessor::~VPMFramePreprocessor() {
|
|
Reset();
|
|
delete vd_;
|
|
delete spatial_resampler_;
|
|
}
|
|
|
|
void VPMFramePreprocessor::Reset() {
|
|
vd_->Reset();
|
|
spatial_resampler_->Reset();
|
|
frame_cnt_ = 0;
|
|
}
|
|
|
|
void VPMFramePreprocessor::EnableTemporalDecimation(bool enable) {
|
|
vd_->EnableTemporalDecimation(enable);
|
|
}
|
|
|
|
void VPMFramePreprocessor::SetInputFrameResampleMode(
|
|
VideoFrameResampling resampling_mode) {
|
|
spatial_resampler_->SetInputFrameResampleMode(resampling_mode);
|
|
}
|
|
|
|
int32_t VPMFramePreprocessor::SetTargetResolution(uint32_t width,
|
|
uint32_t height,
|
|
uint32_t frame_rate) {
|
|
if ((width == 0) || (height == 0) || (frame_rate == 0)) {
|
|
return VPM_PARAMETER_ERROR;
|
|
}
|
|
int32_t ret_val = 0;
|
|
ret_val = spatial_resampler_->SetTargetFrameSize(width, height);
|
|
|
|
if (ret_val < 0)
|
|
return ret_val;
|
|
|
|
vd_->SetTargetFramerate(frame_rate);
|
|
return VPM_OK;
|
|
}
|
|
|
|
void VPMFramePreprocessor::UpdateIncomingframe_rate() {
|
|
vd_->UpdateIncomingframe_rate();
|
|
}
|
|
|
|
uint32_t VPMFramePreprocessor::GetDecimatedFrameRate() {
|
|
return vd_->GetDecimatedFrameRate();
|
|
}
|
|
|
|
uint32_t VPMFramePreprocessor::GetDecimatedWidth() const {
|
|
return spatial_resampler_->TargetWidth();
|
|
}
|
|
|
|
uint32_t VPMFramePreprocessor::GetDecimatedHeight() const {
|
|
return spatial_resampler_->TargetHeight();
|
|
}
|
|
|
|
void VPMFramePreprocessor::EnableDenoising(bool enable) {
|
|
if (enable) {
|
|
denoiser_.reset(new VideoDenoiser(true));
|
|
} else {
|
|
denoiser_.reset();
|
|
}
|
|
}
|
|
|
|
const VideoFrame* VPMFramePreprocessor::PreprocessFrame(
|
|
const VideoFrame& frame) {
|
|
if (frame.IsZeroSize()) {
|
|
return nullptr;
|
|
}
|
|
|
|
vd_->UpdateIncomingframe_rate();
|
|
if (vd_->DropFrame()) {
|
|
return nullptr;
|
|
}
|
|
|
|
const VideoFrame* current_frame = &frame;
|
|
if (denoiser_) {
|
|
VideoFrame* denoised_frame = &denoised_frame_[0];
|
|
VideoFrame* denoised_frame_prev = &denoised_frame_[1];
|
|
// Swap the buffer to save one memcpy in DenoiseFrame.
|
|
if (denoised_frame_toggle_) {
|
|
denoised_frame = &denoised_frame_[1];
|
|
denoised_frame_prev = &denoised_frame_[0];
|
|
}
|
|
// Invert the flag.
|
|
denoised_frame_toggle_ ^= 1;
|
|
denoiser_->DenoiseFrame(*current_frame, denoised_frame, denoised_frame_prev,
|
|
true);
|
|
current_frame = denoised_frame;
|
|
}
|
|
|
|
if (spatial_resampler_->ApplyResample(current_frame->width(),
|
|
current_frame->height())) {
|
|
if (spatial_resampler_->ResampleFrame(*current_frame, &resampled_frame_) !=
|
|
VPM_OK) {
|
|
return nullptr;
|
|
}
|
|
current_frame = &resampled_frame_;
|
|
}
|
|
|
|
++frame_cnt_;
|
|
return current_frame;
|
|
}
|
|
|
|
} // namespace webrtc
|