1*77c1e3ccSAndroid Build Coastguard Worker /* 2*77c1e3ccSAndroid Build Coastguard Worker * Copyright (c) 2020, Alliance for Open Media. All rights reserved. 3*77c1e3ccSAndroid Build Coastguard Worker * 4*77c1e3ccSAndroid Build Coastguard Worker * This source code is subject to the terms of the BSD 2 Clause License and 5*77c1e3ccSAndroid Build Coastguard Worker * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License 6*77c1e3ccSAndroid Build Coastguard Worker * was not distributed with this source code in the LICENSE file, you can 7*77c1e3ccSAndroid Build Coastguard Worker * obtain it at www.aomedia.org/license/software. If the Alliance for Open 8*77c1e3ccSAndroid Build Coastguard Worker * Media Patent License 1.0 was not distributed with this source code in the 9*77c1e3ccSAndroid Build Coastguard Worker * PATENTS file, you can obtain it at www.aomedia.org/license/patent. 10*77c1e3ccSAndroid Build Coastguard Worker */ 11*77c1e3ccSAndroid Build Coastguard Worker 12*77c1e3ccSAndroid Build Coastguard Worker #ifndef AOM_AV1_ENCODER_AV1_NOISE_ESTIMATE_H_ 13*77c1e3ccSAndroid Build Coastguard Worker #define AOM_AV1_ENCODER_AV1_NOISE_ESTIMATE_H_ 14*77c1e3ccSAndroid Build Coastguard Worker 15*77c1e3ccSAndroid Build Coastguard Worker #include "av1/encoder/block.h" 16*77c1e3ccSAndroid Build Coastguard Worker #include "aom_scale/yv12config.h" 17*77c1e3ccSAndroid Build Coastguard Worker 18*77c1e3ccSAndroid Build Coastguard Worker #ifdef __cplusplus 19*77c1e3ccSAndroid Build Coastguard Worker extern "C" { 20*77c1e3ccSAndroid Build Coastguard Worker #endif 21*77c1e3ccSAndroid Build Coastguard Worker 22*77c1e3ccSAndroid Build Coastguard Worker #define MAX_VAR_HIST_BINS 20 23*77c1e3ccSAndroid Build Coastguard Worker 24*77c1e3ccSAndroid Build Coastguard Worker typedef enum noise_level { kLowLow, kLow, kMedium, kHigh } NOISE_LEVEL; 25*77c1e3ccSAndroid Build Coastguard Worker 26*77c1e3ccSAndroid Build Coastguard Worker typedef struct noise_estimate { 27*77c1e3ccSAndroid Build Coastguard Worker int enabled; 28*77c1e3ccSAndroid Build Coastguard Worker NOISE_LEVEL level; 29*77c1e3ccSAndroid Build Coastguard Worker int value; 30*77c1e3ccSAndroid Build Coastguard Worker int thresh; 31*77c1e3ccSAndroid Build Coastguard Worker int adapt_thresh; 32*77c1e3ccSAndroid Build Coastguard Worker int count; 33*77c1e3ccSAndroid Build Coastguard Worker int last_w; 34*77c1e3ccSAndroid Build Coastguard Worker int last_h; 35*77c1e3ccSAndroid Build Coastguard Worker int num_frames_estimate; 36*77c1e3ccSAndroid Build Coastguard Worker } NOISE_ESTIMATE; 37*77c1e3ccSAndroid Build Coastguard Worker 38*77c1e3ccSAndroid Build Coastguard Worker struct AV1_COMP; 39*77c1e3ccSAndroid Build Coastguard Worker 40*77c1e3ccSAndroid Build Coastguard Worker void av1_noise_estimate_init(NOISE_ESTIMATE *const ne, int width, int height); 41*77c1e3ccSAndroid Build Coastguard Worker 42*77c1e3ccSAndroid Build Coastguard Worker NOISE_LEVEL av1_noise_estimate_extract_level(NOISE_ESTIMATE *const ne); 43*77c1e3ccSAndroid Build Coastguard Worker 44*77c1e3ccSAndroid Build Coastguard Worker void av1_update_noise_estimate(struct AV1_COMP *const cpi); 45*77c1e3ccSAndroid Build Coastguard Worker 46*77c1e3ccSAndroid Build Coastguard Worker #ifdef __cplusplus 47*77c1e3ccSAndroid Build Coastguard Worker } // extern "C" 48*77c1e3ccSAndroid Build Coastguard Worker #endif 49*77c1e3ccSAndroid Build Coastguard Worker 50*77c1e3ccSAndroid Build Coastguard Worker #endif // AOM_AV1_ENCODER_AV1_NOISE_ESTIMATE_H_ 51