| /* |
| * 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() |
| : content_metrics_(nullptr), |
| resampled_frame_(), |
| enable_ca_(false), |
| frame_cnt_(0) { |
| spatial_resampler_ = new VPMSimpleSpatialResampler(); |
| ca_ = new VPMContentAnalysis(true); |
| vd_ = new VPMVideoDecimator(); |
| } |
| |
| VPMFramePreprocessor::~VPMFramePreprocessor() { |
| Reset(); |
| delete ca_; |
| delete vd_; |
| delete spatial_resampler_; |
| } |
| |
| void VPMFramePreprocessor::Reset() { |
| ca_->Release(); |
| vd_->Reset(); |
| content_metrics_ = nullptr; |
| spatial_resampler_->Reset(); |
| enable_ca_ = false; |
| frame_cnt_ = 0; |
| } |
| |
| void VPMFramePreprocessor::EnableTemporalDecimation(bool enable) { |
| vd_->EnableTemporalDecimation(enable); |
| } |
| |
| void VPMFramePreprocessor::EnableContentAnalysis(bool enable) { |
| enable_ca_ = 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::SetTargetFramerate(int frame_rate) { |
| if (frame_rate == -1) { |
| vd_->EnableTemporalDecimation(false); |
| } else { |
| vd_->EnableTemporalDecimation(true); |
| vd_->SetTargetFramerate(frame_rate); |
| } |
| } |
| |
| 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::EnableDenosing(bool enable) { |
| denoiser_.reset(new VideoDenoiser()); |
| } |
| |
| 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_) { |
| denoiser_->DenoiseFrame(*current_frame, &denoised_frame_); |
| 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_; |
| } |
| |
| // Perform content analysis on the frame to be encoded. |
| if (enable_ca_ && frame_cnt_ % kSkipFrameCA == 0) { |
| // Compute new metrics every |kSkipFramesCA| frames, starting with |
| // the first frame. |
| content_metrics_ = ca_->ComputeContentMetrics(*current_frame); |
| } |
| ++frame_cnt_; |
| return current_frame; |
| } |
| |
| VideoContentMetrics* VPMFramePreprocessor::GetContentMetrics() const { |
| return content_metrics_; |
| } |
| |
| } // namespace webrtc |