1/* 2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11#ifndef VP8_ENCODER_DENOISING_H_ 12#define VP8_ENCODER_DENOISING_H_ 13 14#include "block.h" 15#include "vp8/common/loopfilter.h" 16 17#ifdef __cplusplus 18extern "C" { 19#endif 20 21#define SUM_DIFF_THRESHOLD (16 * 16 * 2) 22#define SUM_DIFF_THRESHOLD_HIGH (600) 23#define MOTION_MAGNITUDE_THRESHOLD (8*3) 24 25#define SUM_DIFF_THRESHOLD_UV (96) // (8 * 8 * 1.5) 26#define SUM_DIFF_THRESHOLD_HIGH_UV (8 * 8 * 2) 27#define SUM_DIFF_FROM_AVG_THRESH_UV (8 * 8 * 8) 28#define MOTION_MAGNITUDE_THRESHOLD_UV (8*3) 29 30enum vp8_denoiser_decision 31{ 32 COPY_BLOCK, 33 FILTER_BLOCK 34}; 35 36enum vp8_denoiser_filter_state { 37 kNoFilter, 38 kFilterZeroMV, 39 kFilterNonZeroMV 40}; 41 42enum vp8_denoiser_mode { 43 kDenoiserOff, 44 kDenoiserOnYOnly, 45 kDenoiserOnYUV, 46 kDenoiserOnYUVAggressive, 47 kDenoiserOnAdaptive 48}; 49 50typedef struct { 51 // Scale factor on sse threshold above which no denoising is done. 52 unsigned int scale_sse_thresh; 53 // Scale factor on motion magnitude threshold above which no 54 // denoising is done. 55 unsigned int scale_motion_thresh; 56 // Scale factor on motion magnitude below which we increase the strength of 57 // the temporal filter (in function vp8_denoiser_filter). 58 unsigned int scale_increase_filter; 59 // Scale factor to bias to ZEROMV for denoising. 60 unsigned int denoise_mv_bias; 61 // Scale factor to bias to ZEROMV for coding mode selection. 62 unsigned int pickmode_mv_bias; 63 // Quantizer threshold below which we use the segmentation map to switch off 64 // loop filter for blocks that have been coded as ZEROMV-LAST a certain number 65 // (consec_zerolast) of consecutive frames. Note that the delta-QP is set to 66 // 0 when segmentation map is used for shutting off loop filter. 67 unsigned int qp_thresh; 68 // Threshold for number of consecutive frames for blocks coded as ZEROMV-LAST. 69 unsigned int consec_zerolast; 70 // Threshold for amount of spatial blur on Y channel. 0 means no spatial blur. 71 unsigned int spatial_blur; 72} denoise_params; 73 74typedef struct vp8_denoiser 75{ 76 YV12_BUFFER_CONFIG yv12_running_avg[MAX_REF_FRAMES]; 77 YV12_BUFFER_CONFIG yv12_mc_running_avg; 78 // TODO(marpan): Should remove yv12_last_source and use vp8_lookahead_peak. 79 YV12_BUFFER_CONFIG yv12_last_source; 80 unsigned char* denoise_state; 81 int num_mb_cols; 82 int denoiser_mode; 83 int threshold_aggressive_mode; 84 int nmse_source_diff; 85 int nmse_source_diff_count; 86 int qp_avg; 87 int qp_threshold_up; 88 int qp_threshold_down; 89 int bitrate_threshold; 90 denoise_params denoise_pars; 91} VP8_DENOISER; 92 93int vp8_denoiser_allocate(VP8_DENOISER *denoiser, int width, int height, 94 int num_mb_rows, int num_mb_cols, int mode); 95 96void vp8_denoiser_free(VP8_DENOISER *denoiser); 97 98void vp8_denoiser_set_parameters(VP8_DENOISER *denoiser, int mode); 99 100void vp8_denoiser_denoise_mb(VP8_DENOISER *denoiser, 101 MACROBLOCK *x, 102 unsigned int best_sse, 103 unsigned int zero_mv_sse, 104 int recon_yoffset, 105 int recon_uvoffset, 106 loop_filter_info_n *lfi_n, 107 int mb_row, 108 int mb_col, 109 int block_index); 110 111#ifdef __cplusplus 112} // extern "C" 113#endif 114 115#endif // VP8_ENCODER_DENOISING_H_ 116