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 VP9_ENCODER_DENOISER_H_
12#define VP9_ENCODER_DENOISER_H_
13
14#include "vp9/encoder/vp9_block.h"
15#include "vpx_scale/yv12config.h"
16
17#ifdef __cplusplus
18extern "C" {
19#endif
20
21#define MOTION_MAGNITUDE_THRESHOLD (8 * 3)
22
23typedef enum vp9_denoiser_decision {
24  COPY_BLOCK,
25  FILTER_BLOCK
26} VP9_DENOISER_DECISION;
27
28typedef struct vp9_denoiser {
29  YV12_BUFFER_CONFIG running_avg_y[MAX_REF_FRAMES];
30  YV12_BUFFER_CONFIG mc_running_avg_y;
31  int increase_denoising;
32} VP9_DENOISER;
33
34void vp9_denoiser_update_frame_info(VP9_DENOISER *denoiser,
35                                    YV12_BUFFER_CONFIG src,
36                                    FRAME_TYPE frame_type,
37                                    int refresh_alt_ref_frame,
38                                    int refresh_golden_frame,
39                                    int refresh_last_frame);
40
41void vp9_denoiser_denoise(VP9_DENOISER *denoiser, MACROBLOCK *mb,
42                          int mi_row, int mi_col, BLOCK_SIZE bs,
43                          PICK_MODE_CONTEXT *ctx);
44
45void vp9_denoiser_reset_frame_stats(PICK_MODE_CONTEXT *ctx);
46
47void vp9_denoiser_update_frame_stats(MB_MODE_INFO *mbmi,
48                                     unsigned int sse, PREDICTION_MODE mode,
49                                     PICK_MODE_CONTEXT *ctx);
50
51int vp9_denoiser_alloc(VP9_DENOISER *denoiser, int width, int height,
52                       int ssx, int ssy,
53#if CONFIG_VP9_HIGHBITDEPTH
54                       int use_highbitdepth,
55#endif
56                       int border);
57
58void vp9_denoiser_free(VP9_DENOISER *denoiser);
59
60#ifdef __cplusplus
61}  // extern "C"
62#endif
63
64#endif  // VP9_ENCODER_DENOISER_H_
65