15c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org/* 25c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 35c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * 45c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * Use of this source code is governed by a BSD-style license 55c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * that can be found in the LICENSE file in the root of the source 65c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * tree. An additional intellectual property rights grant can be found 75c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * in the file PATENTS. All contributing project authors may 85c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org * be found in the AUTHORS file in the root of the source tree. 95c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org */ 105c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 1141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org#include <limits.h> 1241294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 135c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org#include "denoising.h" 145c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 155c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org#include "vp8/common/reconinter.h" 165c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org#include "vpx/vpx_integer.h" 175c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org#include "vpx_mem/vpx_mem.h" 186fefe538d859300e7febe78271828198c10f1b52fgalligan@chromium.org#include "vp8_rtcd.h" 195c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 205c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.orgstatic const unsigned int NOISE_MOTION_THRESHOLD = 25 * 25; 21ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org/* SSE_DIFF_THRESHOLD is selected as ~95% confidence assuming 22ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * var(noise) ~= 100. 23ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org */ 245c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.orgstatic const unsigned int SSE_DIFF_THRESHOLD = 16 * 16 * 20; 255c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.orgstatic const unsigned int SSE_THRESHOLD = 16 * 16 * 40; 26118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.orgstatic const unsigned int SSE_THRESHOLD_HIGH = 16 * 16 * 60; 275c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 28ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org/* 29ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * The filter function was modified to reduce the computational complexity. 30ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Step 1: 31ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Instead of applying tap coefficients for each pixel, we calculated the 32ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * pixel adjustments vs. pixel diff value ahead of time. 33ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * adjustment = filtered_value - current_raw 34ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * = (filter_coefficient * diff + 128) >> 8 35ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * where 36ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * filter_coefficient = (255 << 8) / (256 + ((absdiff * 330) >> 3)); 37ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * filter_coefficient += filter_coefficient / 38ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * (3 + motion_magnitude_adjustment); 39ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * filter_coefficient is clamped to 0 ~ 255. 40ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * 41ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Step 2: 42ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * The adjustment vs. diff curve becomes flat very quick when diff increases. 43ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * This allowed us to use only several levels to approximate the curve without 44ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * changing the filtering algorithm too much. 45ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * The adjustments were further corrected by checking the motion magnitude. 46ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * The levels used are: 47ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * diff adjustment w/o motion correction adjustment w/ motion correction 48ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [-255, -16] -6 -7 49ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [-15, -8] -4 -5 50ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [-7, -4] -3 -4 51ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [-3, 3] diff diff 52ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [4, 7] 3 4 53ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [8, 15] 4 5 54ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * [16, 255] 6 7 55ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org */ 565c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 577765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.orgint vp8_denoiser_filter_c(unsigned char *mc_running_avg_y, int mc_avg_y_stride, 587765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org unsigned char *running_avg_y, int avg_y_stride, 597765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org unsigned char *sig, int sig_stride, 60118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org unsigned int motion_magnitude, 61118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org int increase_denoising) 625c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org{ 637765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org unsigned char *running_avg_y_start = running_avg_y; 647765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org unsigned char *sig_start = sig; 65118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org int sum_diff_thresh; 66118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org int r, c; 675c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org int sum_diff = 0; 68ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int adj_val[3] = {3, 4, 6}; 69118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org int shift_inc1 = 0; 70118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org int shift_inc2 = 1; 7187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org int col_sum[16] = {0, 0, 0, 0, 7287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 0, 0, 0, 0, 7387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 0, 0, 0, 0, 7487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 0, 0, 0, 0}; 75ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* If motion_magnitude is small, making the denoiser more aggressive by 76118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org * increasing the adjustment for each level. Add another increment for 77118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org * blocks that are labeled for increase denoising. */ 78ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org if (motion_magnitude <= MOTION_MAGNITUDE_THRESHOLD) 795c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 80118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org if (increase_denoising) { 81118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org shift_inc1 = 1; 82118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org shift_inc2 = 2; 83118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org } 84118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org adj_val[0] += shift_inc2; 85118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org adj_val[1] += shift_inc2; 86118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org adj_val[2] += shift_inc2; 87ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 885c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 89ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (r = 0; r < 16; ++r) 90ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org { 915c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org for (c = 0; c < 16; ++c) 925c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 93ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int diff = 0; 94ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int adjustment = 0; 95ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int absdiff = 0; 965c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 97ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org diff = mc_running_avg_y[c] - sig[c]; 98ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org absdiff = abs(diff); 995c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 100118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org // When |diff| <= |3 + shift_inc1|, use pixel value from 101118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org // last denoised raw. 102118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org if (absdiff <= 3 + shift_inc1) 1035c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 104ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org running_avg_y[c] = mc_running_avg_y[c]; 10587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] += diff; 1065c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 1075c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org else 1085c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 10987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org if (absdiff >= 4 + shift_inc1 && absdiff <= 7) 110ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org adjustment = adj_val[0]; 111ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org else if (absdiff >= 8 && absdiff <= 15) 112ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org adjustment = adj_val[1]; 113ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org else 114ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org adjustment = adj_val[2]; 115ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 116ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org if (diff > 0) 117ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org { 118ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org if ((sig[c] + adjustment) > 255) 119ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org running_avg_y[c] = 255; 120ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org else 121ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org running_avg_y[c] = sig[c] + adjustment; 122ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 12387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] += adjustment; 124ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 125ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org else 126ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org { 127ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org if ((sig[c] - adjustment) < 0) 128ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org running_avg_y[c] = 0; 129ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org else 130ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org running_avg_y[c] = sig[c] - adjustment; 131ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 13287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] -= adjustment; 133ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 1345c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 1355c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 1365c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 137ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* Update pointers for next iteration. */ 1385c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org sig += sig_stride; 1395c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mc_running_avg_y += mc_avg_y_stride; 1405c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org running_avg_y += avg_y_stride; 1415c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 142ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 14387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org for (c = 0; c < 16; ++c) { 14487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // Below we clip the value in the same way which SSE code use. 14587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // When adopting aggressive denoiser, the adj_val for each pixel 14687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // could be at most 8 (this is current max adjustment of the map). 14787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // In SSE code, we calculate the sum of adj_val for 14887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // the columns, so the sum could be upto 128(16 rows). However, 14987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // the range of the value is -128 ~ 127 in SSE code, that's why 15087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // we do this change in C code. 15187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // We don't do this for UV denoiser, since there are only 8 rows, 15287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // and max adjustments <= 8, so the sum of the columns will not 15387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // exceed 64. 15487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org if (col_sum[c] >= 128) { 15587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] = 127; 15687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 15787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org sum_diff += col_sum[c]; 15887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 15987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 160118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org sum_diff_thresh= SUM_DIFF_THRESHOLD; 161118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org if (increase_denoising) sum_diff_thresh = SUM_DIFF_THRESHOLD_HIGH; 16288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (abs(sum_diff) > sum_diff_thresh) { 16388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Before returning to copy the block (i.e., apply no denoising), check 16488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // if we can still apply some (weaker) temporal filtering to this block, 16588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // that would otherwise not be denoised at all. Simplest is to apply 16688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // an additional adjustment to running_avg_y to bring it closer to sig. 16788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // The adjustment is capped by a maximum delta, and chosen such that 16888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // in most cases the resulting sum_diff will be within the 16988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // accceptable range given by sum_diff_thresh. 17088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org 17188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // The delta is set by the excess of absolute pixel diff over threshold. 17288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int delta = ((abs(sum_diff) - sum_diff_thresh) >> 8) + 1; 17388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Only apply the adjustment for max delta up to 3. 17488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (delta < 4) { 17588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org sig -= sig_stride * 16; 17688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org mc_running_avg_y -= mc_avg_y_stride * 16; 17788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org running_avg_y -= avg_y_stride * 16; 17888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org for (r = 0; r < 16; ++r) { 17988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org for (c = 0; c < 16; ++c) { 18088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int diff = mc_running_avg_y[c] - sig[c]; 18188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int adjustment = abs(diff); 18288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (adjustment > delta) 18388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org adjustment = delta; 18488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (diff > 0) { 18588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Bring denoised signal down. 18688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (running_avg_y[c] - adjustment < 0) 18788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org running_avg_y[c] = 0; 18888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org else 18988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org running_avg_y[c] = running_avg_y[c] - adjustment; 19087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] -= adjustment; 19188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } else if (diff < 0) { 19288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Bring denoised signal up. 19388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (running_avg_y[c] + adjustment > 255) 19488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org running_avg_y[c] = 255; 19588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org else 19688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org running_avg_y[c] = running_avg_y[c] + adjustment; 19787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] += adjustment; 19888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 19988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 20088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // TODO(marpan): Check here if abs(sum_diff) has gone below the 20188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // threshold sum_diff_thresh, and if so, we can exit the row loop. 20288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org sig += sig_stride; 20388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org mc_running_avg_y += mc_avg_y_stride; 20488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org running_avg_y += avg_y_stride; 20588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 20687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 20787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org sum_diff = 0; 20887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org for (c = 0; c < 16; ++c) { 20987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org if (col_sum[c] >= 128) { 21087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org col_sum[c] = 127; 21187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 21287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org sum_diff += col_sum[c]; 21387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 21487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org 21588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (abs(sum_diff) > sum_diff_thresh) 21688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org return COPY_BLOCK; 21788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } else { 2185c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org return COPY_BLOCK; 21988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 22088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 221ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 2227765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org vp8_copy_mem16x16(running_avg_y_start, avg_y_stride, sig_start, sig_stride); 2235c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org return FILTER_BLOCK; 2245c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org} 2255c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 226e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.orgint vp8_denoiser_filter_uv_c(unsigned char *mc_running_avg_uv, 227e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int mc_avg_uv_stride, 228e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *running_avg_uv, 229e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int avg_uv_stride, 230e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *sig, 231e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int sig_stride, 232e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned int motion_magnitude, 233e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int increase_denoising) { 234e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *running_avg_uv_start = running_avg_uv; 235e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *sig_start = sig; 236e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int sum_diff_thresh; 237e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int r, c; 238e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int sum_diff = 0; 239e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int sum_block = 0; 240e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int adj_val[3] = {3, 4, 6}; 241e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int shift_inc1 = 0; 242e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int shift_inc2 = 1; 243e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org /* If motion_magnitude is small, making the denoiser more aggressive by 244e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * increasing the adjustment for each level. Add another increment for 245e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org * blocks that are labeled for increase denoising. */ 246e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (motion_magnitude <= MOTION_MAGNITUDE_THRESHOLD_UV) { 247e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (increase_denoising) { 248e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org shift_inc1 = 1; 249e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org shift_inc2 = 2; 250e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 251e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adj_val[0] += shift_inc2; 252e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adj_val[1] += shift_inc2; 253e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adj_val[2] += shift_inc2; 254e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 255e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 256e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Avoid denoising color signal if its close to average level. 257e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (r = 0; r < 8; ++r) { 258e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (c = 0; c < 8; ++c) { 259e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_block += sig[c]; 260e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 261e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sig += sig_stride; 262e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 263e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (abs(sum_block - (128 * 8 * 8)) < SUM_DIFF_FROM_AVG_THRESH_UV) { 264e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return COPY_BLOCK; 265e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 266e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 267e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sig -= sig_stride * 8; 268e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (r = 0; r < 8; ++r) { 269e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (c = 0; c < 8; ++c) { 270e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int diff = 0; 271e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int adjustment = 0; 272e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int absdiff = 0; 273e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 274e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org diff = mc_running_avg_uv[c] - sig[c]; 275e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org absdiff = abs(diff); 276e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 277e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // When |diff| <= |3 + shift_inc1|, use pixel value from 278e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // last denoised raw. 279e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (absdiff <= 3 + shift_inc1) { 280e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = mc_running_avg_uv[c]; 281e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_diff += diff; 282e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } else { 283e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (absdiff >= 4 && absdiff <= 7) 284e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adjustment = adj_val[0]; 285e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org else if (absdiff >= 8 && absdiff <= 15) 286e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adjustment = adj_val[1]; 287e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org else 288e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adjustment = adj_val[2]; 289e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (diff > 0) { 290e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if ((sig[c] + adjustment) > 255) 291e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = 255; 292e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org else 293e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = sig[c] + adjustment; 294e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_diff += adjustment; 295e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } else { 296e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if ((sig[c] - adjustment) < 0) 297e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = 0; 298e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org else 299e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = sig[c] - adjustment; 300e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_diff -= adjustment; 301e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 302e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 303e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 304e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org /* Update pointers for next iteration. */ 305e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sig += sig_stride; 306e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org mc_running_avg_uv += mc_avg_uv_stride; 307e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv += avg_uv_stride; 308e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 309e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 310e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_diff_thresh= SUM_DIFF_THRESHOLD_UV; 311e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (increase_denoising) sum_diff_thresh = SUM_DIFF_THRESHOLD_HIGH_UV; 312e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (abs(sum_diff) > sum_diff_thresh) { 313e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Before returning to copy the block (i.e., apply no denoising), check 314e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // if we can still apply some (weaker) temporal filtering to this block, 315e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // that would otherwise not be denoised at all. Simplest is to apply 316e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // an additional adjustment to running_avg_y to bring it closer to sig. 317e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // The adjustment is capped by a maximum delta, and chosen such that 318e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // in most cases the resulting sum_diff will be within the 319e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // accceptable range given by sum_diff_thresh. 320e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 321e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // The delta is set by the excess of absolute pixel diff over threshold. 322e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int delta = ((abs(sum_diff) - sum_diff_thresh) >> 8) + 1; 323e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Only apply the adjustment for max delta up to 3. 324e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (delta < 4) { 325e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sig -= sig_stride * 8; 326e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org mc_running_avg_uv -= mc_avg_uv_stride * 8; 327e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv -= avg_uv_stride * 8; 328e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (r = 0; r < 8; ++r) { 329e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org for (c = 0; c < 8; ++c) { 330e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int diff = mc_running_avg_uv[c] - sig[c]; 331e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int adjustment = abs(diff); 332e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (adjustment > delta) 333e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org adjustment = delta; 334e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (diff > 0) { 335e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Bring denoised signal down. 336e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (running_avg_uv[c] - adjustment < 0) 337e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = 0; 338e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org else 339e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = running_avg_uv[c] - adjustment; 340e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_diff -= adjustment; 341e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } else if (diff < 0) { 342e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Bring denoised signal up. 343e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (running_avg_uv[c] + adjustment > 255) 344e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = 255; 345e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org else 346e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv[c] = running_avg_uv[c] + adjustment; 347e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sum_diff += adjustment; 348e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 349e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 350e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // TODO(marpan): Check here if abs(sum_diff) has gone below the 351e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // threshold sum_diff_thresh, and if so, we can exit the row loop. 352e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sig += sig_stride; 353e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org mc_running_avg_uv += mc_avg_uv_stride; 354e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_uv += avg_uv_stride; 355e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 356e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (abs(sum_diff) > sum_diff_thresh) 357e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return COPY_BLOCK; 358e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } else { 359e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return COPY_BLOCK; 360e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 361e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 362e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 363e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp8_copy_mem8x8(running_avg_uv_start, avg_uv_stride, sig_start, 364e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org sig_stride); 365e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org return FILTER_BLOCK; 366e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org} 367e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org 36841294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.orgvoid vp8_denoiser_set_parameters(VP8_DENOISER *denoiser, int mode) { 36941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org assert(mode > 0); // Denoiser is allocated only if mode > 0. 37041294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (mode == 1) { 37141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoiser_mode = kDenoiserOnYOnly; 37241294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } else if (mode == 2) { 37341294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoiser_mode = kDenoiserOnYUV; 37441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } else if (mode == 3) { 37541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoiser_mode = kDenoiserOnYUVAggressive; 37641294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } else { 37741294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoiser_mode = kDenoiserOnAdaptive; 37841294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } 37941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (denoiser->denoiser_mode != kDenoiserOnYUVAggressive) { 38041294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_sse_thresh = 1; 38141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_motion_thresh = 8; 38241294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_increase_filter = 0; 38341294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.denoise_mv_bias = 95; 38441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.pickmode_mv_bias = 100; 38541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.qp_thresh = 0; 38641294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.consec_zerolast = UINT_MAX; 38787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->denoise_pars.spatial_blur = 0; 38841294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } else { 38941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_sse_thresh = 2; 39041294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_motion_thresh = 16; 39141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_increase_filter = 1; 39241294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.denoise_mv_bias = 60; 39341294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.pickmode_mv_bias = 60; 39441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.qp_thresh = 100; 39541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.consec_zerolast = 10; 39687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->denoise_pars.spatial_blur = 20; 39741294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } 39841294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org} 39941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 40088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.orgint vp8_denoiser_allocate(VP8_DENOISER *denoiser, int width, int height, 40141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org int num_mb_rows, int num_mb_cols, int mode) 4025c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org{ 4035c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org int i; 4045c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org assert(denoiser); 40588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->num_mb_cols = num_mb_cols; 4065c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 407d348b8d765c019ee7250075d663a83db00c65c08tomfinegan@chromium.org for (i = 0; i < MAX_REF_FRAMES; i++) 4085c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 4095c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org denoiser->yv12_running_avg[i].flags = 0; 4105c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 4115c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (vp8_yv12_alloc_frame_buffer(&(denoiser->yv12_running_avg[i]), width, 4125c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org height, VP8BORDERINPIXELS) 4135c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org < 0) 4145c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 4155c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_denoiser_free(denoiser); 4165c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org return 1; 4175c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 4185c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vpx_memset(denoiser->yv12_running_avg[i].buffer_alloc, 0, 4195c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org denoiser->yv12_running_avg[i].frame_size); 4205c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 4215c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 4225c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org denoiser->yv12_mc_running_avg.flags = 0; 4235c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 4245c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (vp8_yv12_alloc_frame_buffer(&(denoiser->yv12_mc_running_avg), width, 4255c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org height, VP8BORDERINPIXELS) < 0) 4265c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 4275c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_denoiser_free(denoiser); 4285c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org return 1; 4295c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 4305c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 4315c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vpx_memset(denoiser->yv12_mc_running_avg.buffer_alloc, 0, 4325c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org denoiser->yv12_mc_running_avg.frame_size); 43388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org 43441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (vp8_yv12_alloc_frame_buffer(&denoiser->yv12_last_source, width, 43541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org height, VP8BORDERINPIXELS) < 0) { 43641294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org vp8_denoiser_free(denoiser); 43741294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org return 1; 43841294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org } 43941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org vpx_memset(denoiser->yv12_last_source.buffer_alloc, 0, 44041294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->yv12_last_source.frame_size); 44141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 44288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state = vpx_calloc((num_mb_rows * num_mb_cols), 1); 44388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org vpx_memset(denoiser->denoise_state, 0, (num_mb_rows * num_mb_cols)); 44441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org vp8_denoiser_set_parameters(denoiser, mode); 44541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->nmse_source_diff = 0; 44641294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->nmse_source_diff_count = 0; 44787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->qp_avg = 0; 44887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // QP threshold below which we can go up to aggressive mode. 44987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->qp_threshold_up = 80; 45087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // QP threshold above which we can go back down to normal mode. 45187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // For now keep this second threshold high, so not used currently. 45287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->qp_threshold_down = 128; 45387997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // Bitrate thresholds and noise metric (nmse) thresholds for switching to 45487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org // aggressive mode. 45541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org // TODO(marpan): Adjust thresholds, including effect on resolution. 45687997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->bitrate_threshold = 200000; // (bits/sec). 457d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org denoiser->threshold_aggressive_mode = 35; 45887997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org if (width * height > 640 * 480) { 45987997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->bitrate_threshold = 500000; 46087997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->threshold_aggressive_mode = 100; 46187997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } else if (width * height > 960 * 540) { 46287997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->bitrate_threshold = 800000; 463d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org denoiser->threshold_aggressive_mode = 150; 46487997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } else if (width * height > 1280 * 720) { 46587997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org denoiser->bitrate_threshold = 2000000; 466d95585fb0ec024f6abd96f7b02e0df58019d46afjohannkoenig@chromium.org denoiser->threshold_aggressive_mode = 1400; 46787997d490ae52aa962a985c95b3cddf7f8832641johannkoenig@chromium.org } 4685c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org return 0; 4695c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org} 4705c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 47141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 4725c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.orgvoid vp8_denoiser_free(VP8_DENOISER *denoiser) 4735c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org{ 4745c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org int i; 4755c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org assert(denoiser); 4765c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 477d348b8d765c019ee7250075d663a83db00c65c08tomfinegan@chromium.org for (i = 0; i < MAX_REF_FRAMES ; i++) 4785c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 4795c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_yv12_de_alloc_frame_buffer(&denoiser->yv12_running_avg[i]); 4805c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 4815c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_yv12_de_alloc_frame_buffer(&denoiser->yv12_mc_running_avg); 4824e6bc04ba5f53c570e057fe1df5c470a67aaf144marpan@google.com vp8_yv12_de_alloc_frame_buffer(&denoiser->yv12_last_source); 483f36ea3335c459f693513c163c7516a99320ef717marpan@google.com vpx_free(denoiser->denoise_state); 4845c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org} 4855c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 4865c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 4875c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.orgvoid vp8_denoiser_denoise_mb(VP8_DENOISER *denoiser, 4885c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org MACROBLOCK *x, 4895c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org unsigned int best_sse, 4905c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org unsigned int zero_mv_sse, 4915c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org int recon_yoffset, 49288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int recon_uvoffset, 49388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org loop_filter_info_n *lfi_n, 49488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int mb_row, 49588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int mb_col, 49641294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org int block_index) 49741294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 4985c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org{ 4995c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org int mv_row; 5005c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org int mv_col; 50141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org unsigned int motion_threshold; 5025c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org unsigned int motion_magnitude2; 503118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org unsigned int sse_thresh; 50488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int sse_diff_thresh = 0; 50588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Spatial loop filter: only applied selectively based on 50688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // temporal filter state of block relative to top/left neighbors. 50788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int apply_spatial_loop_filter = 1; 5085c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org MV_REFERENCE_FRAME frame = x->best_reference_frame; 5095c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org MV_REFERENCE_FRAME zero_frame = x->best_zeromv_reference_frame; 5105c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5115c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org enum vp8_denoiser_decision decision = FILTER_BLOCK; 5123983288b51bb3c8f48a2d5f77523ef3ff1e94cd2johannkoenig@chromium.org enum vp8_denoiser_decision decision_u = COPY_BLOCK; 5133983288b51bb3c8f48a2d5f77523ef3ff1e94cd2johannkoenig@chromium.org enum vp8_denoiser_decision decision_v = COPY_BLOCK; 5145c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5155c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (zero_frame) 5165c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 5175c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org YV12_BUFFER_CONFIG *src = &denoiser->yv12_running_avg[frame]; 5185c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org YV12_BUFFER_CONFIG *dst = &denoiser->yv12_mc_running_avg; 5195c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org YV12_BUFFER_CONFIG saved_pre,saved_dst; 5205c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org MB_MODE_INFO saved_mbmi; 5215c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org MACROBLOCKD *filter_xd = &x->e_mbd; 5225c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org MB_MODE_INFO *mbmi = &filter_xd->mode_info_context->mbmi; 52388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int sse_diff = 0; 52488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Bias on zero motion vector sse. 52541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org const int zero_bias = denoiser->denoise_pars.denoise_mv_bias; 52688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org zero_mv_sse = (unsigned int)((int64_t)zero_mv_sse * zero_bias / 100); 52788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org sse_diff = zero_mv_sse - best_sse; 5285c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5295c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org saved_mbmi = *mbmi; 5305c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 531ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* Use the best MV for the compensation. */ 5325c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->ref_frame = x->best_reference_frame; 5335c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->mode = x->best_sse_inter_mode; 5345c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->mv = x->best_sse_mv; 5355c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->need_to_clamp_mvs = x->need_to_clamp_best_mvs; 5365c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mv_col = x->best_sse_mv.as_mv.col; 5375c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mv_row = x->best_sse_mv.as_mv.row; 53888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Bias to zero_mv if small amount of motion. 53988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Note sse_diff_thresh is intialized to zero, so this ensures 54088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // we will always choose zero_mv for denoising if 54188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // zero_mv_see <= best_sse (i.e., sse_diff <= 0). 54288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if ((unsigned int)(mv_row * mv_row + mv_col * mv_col) 54388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org <= NOISE_MOTION_THRESHOLD) 54488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org sse_diff_thresh = (int)SSE_DIFF_THRESHOLD; 5455c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5465c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (frame == INTRA_FRAME || 54788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org sse_diff <= sse_diff_thresh) 5485c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 549ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* 550ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Handle intra blocks as referring to last frame with zero motion 551ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * and let the absolute pixel difference affect the filter factor. 552ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Also consider small amount of motion as being random walk due 553ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * to noise, if it doesn't mean that we get a much bigger error. 554ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Note that any changes to the mode info only affects the 555ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * denoising. 556ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org */ 5575c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->ref_frame = 5585c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org x->best_zeromv_reference_frame; 5595c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5605c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org src = &denoiser->yv12_running_avg[zero_frame]; 5615c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5625c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->mode = ZEROMV; 5635c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mbmi->mv.as_int = 0; 5645c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org x->best_sse_inter_mode = ZEROMV; 5655c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org x->best_sse_mv.as_int = 0; 5665c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org best_sse = zero_mv_sse; 5675c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 5685c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5695c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org saved_pre = filter_xd->pre; 5705c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org saved_dst = filter_xd->dst; 5715c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 572ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* Compensate the running average. */ 5735c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->pre.y_buffer = src->y_buffer + recon_yoffset; 5745c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->pre.u_buffer = src->u_buffer + recon_uvoffset; 5755c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->pre.v_buffer = src->v_buffer + recon_uvoffset; 576ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* Write the compensated running average to the destination buffer. */ 5775c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.y_buffer = dst->y_buffer + recon_yoffset; 5785c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.u_buffer = dst->u_buffer + recon_uvoffset; 5795c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.v_buffer = dst->v_buffer + recon_uvoffset; 5805c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5815c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (!x->skip) 5825c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 5835c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_build_inter_predictors_mb(filter_xd); 5845c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 5855c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org else 5865c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 5875c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_build_inter16x16_predictors_mb(filter_xd, 5885c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.y_buffer, 5895c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.u_buffer, 5905c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.v_buffer, 5915c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.y_stride, 5925c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst.uv_stride); 5935c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 5945c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->pre = saved_pre; 5955c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org filter_xd->dst = saved_dst; 5965c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org *mbmi = saved_mbmi; 5975c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 5985c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 5995c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 6005c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mv_row = x->best_sse_mv.as_mv.row; 6015c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org mv_col = x->best_sse_mv.as_mv.col; 6025c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org motion_magnitude2 = mv_row * mv_row + mv_col * mv_col; 60341294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org motion_threshold = denoiser->denoise_pars.scale_motion_thresh * 60441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org NOISE_MOTION_THRESHOLD; 605118f379ec73bf762ee63784bc5f41ffd41107470johannkoenig@chromium.org 60641294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (motion_magnitude2 < 60741294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org denoiser->denoise_pars.scale_increase_filter * NOISE_MOTION_THRESHOLD) 60841294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org x->increase_denoising = 1; 60941294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 61041294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org sse_thresh = denoiser->denoise_pars.scale_sse_thresh * SSE_THRESHOLD; 61141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (x->increase_denoising) 61241294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org sse_thresh = denoiser->denoise_pars.scale_sse_thresh * SSE_THRESHOLD_HIGH; 61341294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org 61441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (best_sse > sse_thresh || motion_magnitude2 > motion_threshold) 61541294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org decision = COPY_BLOCK; 6165c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org 6175c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (decision == FILTER_BLOCK) 6185c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 6197765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org unsigned char *mc_running_avg_y = 6207765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org denoiser->yv12_mc_running_avg.y_buffer + recon_yoffset; 6217765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org int mc_avg_y_stride = denoiser->yv12_mc_running_avg.y_stride; 6227765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org unsigned char *running_avg_y = 6237765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].y_buffer + recon_yoffset; 6247765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org int avg_y_stride = denoiser->yv12_running_avg[INTRA_FRAME].y_stride; 6257765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org 626ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* Filter. */ 6277765c078fa920ba6c949c15f16b6cc979d8bb95bjohannkoenig@chromium.org decision = vp8_denoiser_filter(mc_running_avg_y, mc_avg_y_stride, 628e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_y, avg_y_stride, 629e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->thismb, 16, motion_magnitude2, 630e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->increase_denoising); 63188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state[block_index] = motion_magnitude2 > 0 ? 63288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org kFilterNonZeroMV : kFilterZeroMV; 633e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Only denoise UV for zero motion, and if y channel was denoised. 63441294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (denoiser->denoiser_mode != kDenoiserOnYOnly && 635e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org motion_magnitude2 == 0 && 636e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org decision == FILTER_BLOCK) { 637e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *mc_running_avg_u = 638e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_mc_running_avg.u_buffer + recon_uvoffset; 639e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *running_avg_u = 640e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].u_buffer + recon_uvoffset; 641e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *mc_running_avg_v = 642e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_mc_running_avg.v_buffer + recon_uvoffset; 643e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org unsigned char *running_avg_v = 644e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].v_buffer + recon_uvoffset; 645e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int mc_avg_uv_stride = denoiser->yv12_mc_running_avg.uv_stride; 646e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int avg_uv_stride = denoiser->yv12_running_avg[INTRA_FRAME].uv_stride; 647e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org int signal_stride = x->block[16].src_stride; 648e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org decision_u = 649e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp8_denoiser_filter_uv(mc_running_avg_u, mc_avg_uv_stride, 650e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_u, avg_uv_stride, 651e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->block[16].src + *x->block[16].base_src, 652e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org signal_stride, motion_magnitude2, 0); 653e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org decision_v = 654e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp8_denoiser_filter_uv(mc_running_avg_v, mc_avg_uv_stride, 655e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org running_avg_v, avg_uv_stride, 656e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->block[20].src + *x->block[20].base_src, 657e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org signal_stride, motion_magnitude2, 0); 658e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 6595c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 6605c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org if (decision == COPY_BLOCK) 6615c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org { 662ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org /* No filtering of this block; it differs too much from the predictor, 663ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * or the motion vector magnitude is considered too big. 664ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org */ 6655c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org vp8_copy_mem16x16( 6665c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org x->thismb, 16, 667d348b8d765c019ee7250075d663a83db00c65c08tomfinegan@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].y_buffer + recon_yoffset, 668d348b8d765c019ee7250075d663a83db00c65c08tomfinegan@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].y_stride); 66988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state[block_index] = kNoFilter; 67088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 67141294d96d7dbf9bc215b09832a8336c5fb158f0bjohannkoenig@chromium.org if (denoiser->denoiser_mode != kDenoiserOnYOnly) { 672e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (decision_u == COPY_BLOCK) { 673e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp8_copy_mem8x8( 674e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->block[16].src + *x->block[16].base_src, x->block[16].src_stride, 675e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].u_buffer + recon_uvoffset, 676e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].uv_stride); 677e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 678e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org if (decision_v == COPY_BLOCK) { 679e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org vp8_copy_mem8x8( 680e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org x->block[20].src + *x->block[20].base_src, x->block[16].src_stride, 681e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].v_buffer + recon_uvoffset, 682e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].uv_stride); 683e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 684e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org } 685e2064011d36b2008099446503f28e64d445060ecjohannkoenig@chromium.org // Option to selectively deblock the denoised signal, for y channel only. 68688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (apply_spatial_loop_filter) { 68788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org loop_filter_info lfi; 68888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int apply_filter_col = 0; 68988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int apply_filter_row = 0; 69088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int apply_filter = 0; 69188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int y_stride = denoiser->yv12_running_avg[INTRA_FRAME].y_stride; 69288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int uv_stride =denoiser->yv12_running_avg[INTRA_FRAME].uv_stride; 69388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org 69488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Fix filter level to some nominal value for now. 69588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int filter_level = 32; 69688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org 69788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org int hev_index = lfi_n->hev_thr_lut[INTER_FRAME][filter_level]; 69888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org lfi.mblim = lfi_n->mblim[filter_level]; 69988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org lfi.blim = lfi_n->blim[filter_level]; 70088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org lfi.lim = lfi_n->lim[filter_level]; 70188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org lfi.hev_thr = lfi_n->hev_thr[hev_index]; 70288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org 70388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Apply filter if there is a difference in the denoiser filter state 70488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // between the current and left/top block, or if non-zero motion vector 70588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // is used for the motion-compensated filtering. 70688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (mb_col > 0) { 70788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org apply_filter_col = !((denoiser->denoise_state[block_index] == 70888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state[block_index - 1]) && 70988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state[block_index] != kFilterNonZeroMV); 71088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (apply_filter_col) { 71188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Filter left vertical edge. 71288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org apply_filter = 1; 71388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org vp8_loop_filter_mbv( 71488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].y_buffer + recon_yoffset, 71588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org NULL, NULL, y_stride, uv_stride, &lfi); 71688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 71788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 71888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (mb_row > 0) { 71988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org apply_filter_row = !((denoiser->denoise_state[block_index] == 72088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state[block_index - denoiser->num_mb_cols]) && 72188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->denoise_state[block_index] != kFilterNonZeroMV); 72288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (apply_filter_row) { 72388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Filter top horizontal edge. 72488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org apply_filter = 1; 72588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org vp8_loop_filter_mbh( 72688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].y_buffer + recon_yoffset, 72788b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org NULL, NULL, y_stride, uv_stride, &lfi); 72888b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 72988b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 73088b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org if (apply_filter) { 73188b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // Update the signal block |x|. Pixel changes are only to top and/or 73288b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org // left boundary pixels: can we avoid full block copy here. 73388b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org vp8_copy_mem16x16( 73488b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org denoiser->yv12_running_avg[INTRA_FRAME].y_buffer + recon_yoffset, 73588b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org y_stride, x->thismb, 16); 73688b47b29cc274dd19cddc37c1ce1834d97df282efgalligan@chromium.org } 7375c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org } 7385c1d3b27608a3f3f6028c069b9bf066a4de474b6hclam@chromium.org} 739