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 18 extern "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 30 enum vp8_denoiser_decision 31 { 32 COPY_BLOCK, 33 FILTER_BLOCK 34 }; 35 36 enum vp8_denoiser_filter_state { 37 kNoFilter, 38 kFilterZeroMV, 39 kFilterNonZeroMV 40 }; 41 42 enum vp8_denoiser_mode { 43 kDenoiserOff, 44 kDenoiserOnYOnly, 45 kDenoiserOnYUV, 46 kDenoiserOnYUVAggressive, 47 kDenoiserOnAdaptive 48 }; 49 50 typedef 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 74 typedef 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 93 int vp8_denoiser_allocate(VP8_DENOISER *denoiser, int width, int height, 94 int num_mb_rows, int num_mb_cols, int mode); 95 96 void vp8_denoiser_free(VP8_DENOISER *denoiser); 97 98 void vp8_denoiser_set_parameters(VP8_DENOISER *denoiser, int mode); 99 100 void 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
