xref: /aosp_15_r20/external/libvpx/vp9/encoder/vp9_noise_estimate.h (revision fb1b10ab9aebc7c7068eedab379b749d7e3900be)
1*fb1b10abSAndroid Build Coastguard Worker /*
2*fb1b10abSAndroid Build Coastguard Worker  *  Copyright (c) 2012 The WebM project authors. All Rights Reserved.
3*fb1b10abSAndroid Build Coastguard Worker  *
4*fb1b10abSAndroid Build Coastguard Worker  *  Use of this source code is governed by a BSD-style license
5*fb1b10abSAndroid Build Coastguard Worker  *  that can be found in the LICENSE file in the root of the source
6*fb1b10abSAndroid Build Coastguard Worker  *  tree. An additional intellectual property rights grant can be found
7*fb1b10abSAndroid Build Coastguard Worker  *  in the file PATENTS.  All contributing project authors may
8*fb1b10abSAndroid Build Coastguard Worker  *  be found in the AUTHORS file in the root of the source tree.
9*fb1b10abSAndroid Build Coastguard Worker  */
10*fb1b10abSAndroid Build Coastguard Worker 
11*fb1b10abSAndroid Build Coastguard Worker #ifndef VPX_VP9_ENCODER_VP9_NOISE_ESTIMATE_H_
12*fb1b10abSAndroid Build Coastguard Worker #define VPX_VP9_ENCODER_VP9_NOISE_ESTIMATE_H_
13*fb1b10abSAndroid Build Coastguard Worker 
14*fb1b10abSAndroid Build Coastguard Worker #include "vp9/encoder/vp9_block.h"
15*fb1b10abSAndroid Build Coastguard Worker #include "vp9/encoder/vp9_skin_detection.h"
16*fb1b10abSAndroid Build Coastguard Worker #include "vpx_scale/yv12config.h"
17*fb1b10abSAndroid Build Coastguard Worker 
18*fb1b10abSAndroid Build Coastguard Worker #if CONFIG_VP9_TEMPORAL_DENOISING
19*fb1b10abSAndroid Build Coastguard Worker #include "vp9/encoder/vp9_denoiser.h"
20*fb1b10abSAndroid Build Coastguard Worker #endif
21*fb1b10abSAndroid Build Coastguard Worker 
22*fb1b10abSAndroid Build Coastguard Worker #ifdef __cplusplus
23*fb1b10abSAndroid Build Coastguard Worker extern "C" {
24*fb1b10abSAndroid Build Coastguard Worker #endif
25*fb1b10abSAndroid Build Coastguard Worker 
26*fb1b10abSAndroid Build Coastguard Worker #define MAX_VAR_HIST_BINS 20
27*fb1b10abSAndroid Build Coastguard Worker 
28*fb1b10abSAndroid Build Coastguard Worker typedef enum noise_level { kLowLow, kLow, kMedium, kHigh } NOISE_LEVEL;
29*fb1b10abSAndroid Build Coastguard Worker 
30*fb1b10abSAndroid Build Coastguard Worker typedef struct noise_estimate {
31*fb1b10abSAndroid Build Coastguard Worker   int enabled;
32*fb1b10abSAndroid Build Coastguard Worker   NOISE_LEVEL level;
33*fb1b10abSAndroid Build Coastguard Worker   int value;
34*fb1b10abSAndroid Build Coastguard Worker   int thresh;
35*fb1b10abSAndroid Build Coastguard Worker   int adapt_thresh;
36*fb1b10abSAndroid Build Coastguard Worker   int count;
37*fb1b10abSAndroid Build Coastguard Worker   int last_w;
38*fb1b10abSAndroid Build Coastguard Worker   int last_h;
39*fb1b10abSAndroid Build Coastguard Worker   int num_frames_estimate;
40*fb1b10abSAndroid Build Coastguard Worker } NOISE_ESTIMATE;
41*fb1b10abSAndroid Build Coastguard Worker 
42*fb1b10abSAndroid Build Coastguard Worker struct VP9_COMP;
43*fb1b10abSAndroid Build Coastguard Worker 
44*fb1b10abSAndroid Build Coastguard Worker void vp9_noise_estimate_init(NOISE_ESTIMATE *const ne, int width, int height);
45*fb1b10abSAndroid Build Coastguard Worker 
46*fb1b10abSAndroid Build Coastguard Worker NOISE_LEVEL vp9_noise_estimate_extract_level(NOISE_ESTIMATE *const ne);
47*fb1b10abSAndroid Build Coastguard Worker 
48*fb1b10abSAndroid Build Coastguard Worker void vp9_update_noise_estimate(struct VP9_COMP *const cpi);
49*fb1b10abSAndroid Build Coastguard Worker 
50*fb1b10abSAndroid Build Coastguard Worker #ifdef __cplusplus
51*fb1b10abSAndroid Build Coastguard Worker }  // extern "C"
52*fb1b10abSAndroid Build Coastguard Worker #endif
53*fb1b10abSAndroid Build Coastguard Worker 
54*fb1b10abSAndroid Build Coastguard Worker #endif  // VPX_VP9_ENCODER_VP9_NOISE_ESTIMATE_H_
55