Moving src/webrtc into src/.
In order to eliminate the WebRTC Subtree mirror in Chromium, WebRTC is moving the content of the src/webrtc directory up to the src/ directory. NOPRESUBMIT=true NOTREECHECKS=true NOTRY=true TBR=tommi@webrtc.org Bug: chromium:611808 Change-Id: Iac59c5b51b950f174119565bac87955a7994bc38 Reviewed-on: https://webrtc-review.googlesource.com/1560 Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Henrik Kjellander <kjellander@webrtc.org> Cr-Commit-Position: refs/heads/master@{#19845}
This commit is contained in:
committed by
Commit Bot
parent
6674846b4a
commit
bb547203bf
113
modules/video_processing/util/noise_estimation.cc
Normal file
113
modules/video_processing/util/noise_estimation.cc
Normal file
@ -0,0 +1,113 @@
|
||||
/*
|
||||
* Copyright (c) 2016 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/util/noise_estimation.h"
|
||||
#if DISPLAYNEON
|
||||
#include <android/log.h>
|
||||
#endif
|
||||
|
||||
namespace webrtc {
|
||||
|
||||
void NoiseEstimation::Init(int width, int height, CpuType cpu_type) {
|
||||
int mb_cols = width >> 4;
|
||||
int mb_rows = height >> 4;
|
||||
consec_low_var_.reset(new uint32_t[mb_cols * mb_rows]());
|
||||
width_ = width;
|
||||
height_ = height;
|
||||
mb_cols_ = width_ >> 4;
|
||||
mb_rows_ = height_ >> 4;
|
||||
cpu_type_ = cpu_type;
|
||||
}
|
||||
|
||||
void NoiseEstimation::GetNoise(int mb_index, uint32_t var, uint32_t luma) {
|
||||
consec_low_var_[mb_index]++;
|
||||
num_static_block_++;
|
||||
if (consec_low_var_[mb_index] >= kConsecLowVarFrame &&
|
||||
(luma >> 6) < kAverageLumaMax && (luma >> 6) > kAverageLumaMin) {
|
||||
// Normalized var by the average luma value, this gives more weight to
|
||||
// darker blocks.
|
||||
int nor_var = var / (luma >> 10);
|
||||
noise_var_ +=
|
||||
nor_var > kBlockSelectionVarMax ? kBlockSelectionVarMax : nor_var;
|
||||
num_noisy_block_++;
|
||||
}
|
||||
}
|
||||
|
||||
void NoiseEstimation::ResetConsecLowVar(int mb_index) {
|
||||
consec_low_var_[mb_index] = 0;
|
||||
}
|
||||
|
||||
void NoiseEstimation::UpdateNoiseLevel() {
|
||||
// TODO(jackychen): Tune a threshold for numb_noisy_block > T to make the
|
||||
// condition more reasonable.
|
||||
// No enough samples implies the motion of the camera or too many moving
|
||||
// objects in the frame.
|
||||
if (num_static_block_ <
|
||||
(0.65 * mb_cols_ * mb_rows_ / NOISE_SUBSAMPLE_INTERVAL) ||
|
||||
!num_noisy_block_) {
|
||||
#if DISPLAY
|
||||
printf("Not enough samples. %d \n", num_static_block_);
|
||||
#elif DISPLAYNEON
|
||||
__android_log_print(ANDROID_LOG_DEBUG, "DISPLAY",
|
||||
"Not enough samples. %d \n", num_static_block_);
|
||||
#endif
|
||||
noise_var_ = 0;
|
||||
noise_var_accum_ = 0;
|
||||
num_noisy_block_ = 0;
|
||||
num_static_block_ = 0;
|
||||
return;
|
||||
} else {
|
||||
#if DISPLAY
|
||||
printf("%d %d fraction = %.3f\n", num_static_block_,
|
||||
mb_cols_ * mb_rows_ / NOISE_SUBSAMPLE_INTERVAL,
|
||||
percent_static_block_);
|
||||
#elif DISPLAYNEON
|
||||
__android_log_print(ANDROID_LOG_DEBUG, "DISPLAY", "%d %d fraction = %.3f\n",
|
||||
num_static_block_,
|
||||
mb_cols_ * mb_rows_ / NOISE_SUBSAMPLE_INTERVAL,
|
||||
percent_static_block_);
|
||||
#endif
|
||||
// Normalized by the number of noisy blocks.
|
||||
noise_var_ /= num_noisy_block_;
|
||||
// Get the percentage of static blocks.
|
||||
percent_static_block_ = static_cast<double>(num_static_block_) /
|
||||
(mb_cols_ * mb_rows_ / NOISE_SUBSAMPLE_INTERVAL);
|
||||
num_noisy_block_ = 0;
|
||||
num_static_block_ = 0;
|
||||
}
|
||||
// For the first frame just update the value with current noise_var_,
|
||||
// otherwise, use the averaging window.
|
||||
if (noise_var_accum_ == 0) {
|
||||
noise_var_accum_ = noise_var_;
|
||||
} else {
|
||||
noise_var_accum_ = (noise_var_accum_ * 15 + noise_var_) / 16;
|
||||
}
|
||||
#if DISPLAY
|
||||
printf("noise_var_accum_ = %.1f, noise_var_ = %d.\n", noise_var_accum_,
|
||||
noise_var_);
|
||||
#elif DISPLAYNEON
|
||||
__android_log_print(ANDROID_LOG_DEBUG, "DISPLAY",
|
||||
"noise_var_accum_ = %.1f, noise_var_ = %d.\n",
|
||||
noise_var_accum_, noise_var_);
|
||||
#endif
|
||||
// Reset noise_var_ for the next frame.
|
||||
noise_var_ = 0;
|
||||
}
|
||||
|
||||
uint8_t NoiseEstimation::GetNoiseLevel() {
|
||||
int noise_thr = cpu_type_ ? kNoiseThreshold : kNoiseThresholdNeon;
|
||||
UpdateNoiseLevel();
|
||||
if (noise_var_accum_ > noise_thr) {
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
} // namespace webrtc
|
||||
Reference in New Issue
Block a user