Index: source/libvpx/vp9/encoder/vp9_denoiser.h |
=================================================================== |
--- source/libvpx/vp9/encoder/vp9_denoiser.h (revision 281795) |
+++ source/libvpx/vp9/encoder/vp9_denoiser.h (working copy) |
@@ -18,14 +18,22 @@ |
extern "C" { |
#endif |
-enum vp9_denoiser_decision { |
+typedef enum vp9_denoiser_decision { |
COPY_BLOCK, |
FILTER_BLOCK |
-}; |
+} VP9_DENOISER_DECISION; |
typedef struct vp9_denoiser { |
YV12_BUFFER_CONFIG running_avg_y[MAX_REF_FRAMES]; |
YV12_BUFFER_CONFIG mc_running_avg_y; |
+ |
+ unsigned int zero_mv_sse; |
+ unsigned int best_sse; |
+ int increase_denoising; |
+ PREDICTION_MODE best_sse_inter_mode; |
+ int_mv best_sse_mv; |
+ MV_REFERENCE_FRAME best_reference_frame; |
+ MV_REFERENCE_FRAME best_zeromv_reference_frame; |
} VP9_DENOISER; |
void vp9_denoiser_update_frame_info(VP9_DENOISER *denoiser, |
@@ -38,8 +46,11 @@ |
void vp9_denoiser_denoise(VP9_DENOISER *denoiser, MACROBLOCK *mb, |
int mi_row, int mi_col, BLOCK_SIZE bs); |
-void vp9_denoiser_update_frame_stats(); |
+void vp9_denoiser_reset_frame_stats(VP9_DENOISER *denoiser); |
+void vp9_denoiser_update_frame_stats(VP9_DENOISER *denoiser, MB_MODE_INFO *mbmi, |
+ unsigned int sse, PREDICTION_MODE mode); |
+ |
int vp9_denoiser_alloc(VP9_DENOISER *denoiser, int width, int height, |
int ssx, int ssy, int border); |