| Index: source/libvpx/vp8/encoder/onyx_if.c
|
| ===================================================================
|
| --- source/libvpx/vp8/encoder/onyx_if.c (revision 278778)
|
| +++ source/libvpx/vp8/encoder/onyx_if.c (working copy)
|
| @@ -98,6 +98,9 @@
|
| #ifdef OUTPUT_YUV_SRC
|
| FILE *yuv_file;
|
| #endif
|
| +#ifdef OUTPUT_YUV_DENOISED
|
| +FILE *yuv_denoised_file;
|
| +#endif
|
|
|
| #if 0
|
| FILE *framepsnr;
|
| @@ -1748,7 +1751,8 @@
|
| {
|
| int width = (cpi->oxcf.Width + 15) & ~15;
|
| int height = (cpi->oxcf.Height + 15) & ~15;
|
| - vp8_denoiser_allocate(&cpi->denoiser, width, height);
|
| + vp8_denoiser_allocate(&cpi->denoiser, width, height,
|
| + cpi->common.mb_rows, cpi->common.mb_cols);
|
| }
|
| }
|
| #endif
|
| @@ -1961,6 +1965,9 @@
|
| #ifdef OUTPUT_YUV_SRC
|
| yuv_file = fopen("bd.yuv", "ab");
|
| #endif
|
| +#ifdef OUTPUT_YUV_DENOISED
|
| + yuv_denoised_file = fopen("denoised.yuv", "ab");
|
| +#endif
|
|
|
| #if 0
|
| framepsnr = fopen("framepsnr.stt", "a");
|
| @@ -2410,6 +2417,9 @@
|
| #ifdef OUTPUT_YUV_SRC
|
| fclose(yuv_file);
|
| #endif
|
| +#ifdef OUTPUT_YUV_DENOISED
|
| + fclose(yuv_denoised_file);
|
| +#endif
|
|
|
| #if 0
|
|
|
| @@ -2610,10 +2620,9 @@
|
| }
|
|
|
|
|
| -#if OUTPUT_YUV_SRC
|
| -void vp8_write_yuv_frame(const char *name, YV12_BUFFER_CONFIG *s)
|
| +#if defined(OUTPUT_YUV_SRC) || defined(OUTPUT_YUV_DENOISED)
|
| +void vp8_write_yuv_frame(FILE *yuv_file, YV12_BUFFER_CONFIG *s)
|
| {
|
| - FILE *yuv_file = fopen(name, "ab");
|
| unsigned char *src = s->y_buffer;
|
| int h = s->y_height;
|
|
|
| @@ -2643,12 +2652,9 @@
|
| src += s->uv_stride;
|
| }
|
| while (--h);
|
| -
|
| - fclose(yuv_file);
|
| }
|
| #endif
|
|
|
| -
|
| static void scale_and_extend_source(YV12_BUFFER_CONFIG *sd, VP8_COMP *cpi)
|
| {
|
| VP8_COMMON *cm = &cpi->common;
|
| @@ -3895,7 +3901,7 @@
|
| #endif
|
|
|
| #ifdef OUTPUT_YUV_SRC
|
| - vp8_write_yuv_frame(cpi->Source);
|
| + vp8_write_yuv_frame(yuv_file, cpi->Source);
|
| #endif
|
|
|
| do
|
| @@ -4434,6 +4440,11 @@
|
|
|
| update_reference_frames(cpi);
|
|
|
| +#ifdef OUTPUT_YUV_DENOISED
|
| + vp8_write_yuv_frame(yuv_denoised_file,
|
| + &cpi->denoiser.yv12_running_avg[INTRA_FRAME]);
|
| +#endif
|
| +
|
| #if !(CONFIG_REALTIME_ONLY & CONFIG_ONTHEFLY_BITPACKING)
|
| if (cpi->oxcf.error_resilient_mode)
|
| {
|
|
|