09202d8071
Change-Id: Ieebea089095d9073b3a94932791099f614ce120c
529 lines
14 KiB
C
529 lines
14 KiB
C
/*
|
|
* Copyright (c) 2010 The VP8 project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
|
|
#include "variance.h"
|
|
|
|
const int vp8_six_tap[8][6] =
|
|
{
|
|
{ 0, 0, 128, 0, 0, 0 }, // note that 1/8 pel positions are just as per alpha -0.5 bicubic
|
|
{ 0, -6, 123, 12, -1, 0 },
|
|
{ 2, -11, 108, 36, -8, 1 }, // New 1/4 pel 6 tap filter
|
|
{ 0, -9, 93, 50, -6, 0 },
|
|
{ 3, -16, 77, 77, -16, 3 }, // New 1/2 pel 6 tap filter
|
|
{ 0, -6, 50, 93, -9, 0 },
|
|
{ 1, -8, 36, 108, -11, 2 }, // New 1/4 pel 6 tap filter
|
|
{ 0, -1, 12, 123, -6, 0 }
|
|
};
|
|
|
|
|
|
#ifdef USEBILINEAR
|
|
const int VP8_FILTER_WEIGHT = 128;
|
|
const int VP8_FILTER_SHIFT = 7;
|
|
const int vp8_bilinear_taps[8][2] =
|
|
{
|
|
{ 128, 0 },
|
|
{ 112, 16 },
|
|
{ 96, 32 },
|
|
{ 80, 48 },
|
|
{ 64, 64 },
|
|
{ 48, 80 },
|
|
{ 32, 96 },
|
|
{ 16, 112 }
|
|
};
|
|
|
|
unsigned int vp8_get_mb_ss_c
|
|
(
|
|
short *src_ptr
|
|
)
|
|
{
|
|
unsigned int i = 0, sum = 0;
|
|
|
|
do
|
|
{
|
|
sum += (src_ptr[i] * src_ptr[i]);
|
|
i++;
|
|
}
|
|
while (i < 256);
|
|
|
|
return sum;
|
|
}
|
|
|
|
|
|
void vp8_variance(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
int w,
|
|
int h,
|
|
unsigned int *sse,
|
|
int *sum)
|
|
{
|
|
int i, j;
|
|
int diff;
|
|
|
|
*sum = 0;
|
|
*sse = 0;
|
|
|
|
for (i = 0; i < h; i++)
|
|
{
|
|
for (j = 0; j < w; j++)
|
|
{
|
|
diff = src_ptr[j] - ref_ptr[j];
|
|
*sum += diff;
|
|
*sse += diff * diff;
|
|
}
|
|
|
|
src_ptr += source_stride;
|
|
ref_ptr += recon_stride;
|
|
}
|
|
}
|
|
|
|
unsigned int
|
|
vp8_get8x8var_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *SSE,
|
|
int *Sum
|
|
)
|
|
{
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 8, 8, SSE, Sum);
|
|
return (*SSE - (((*Sum) * (*Sum)) >> 6));
|
|
}
|
|
|
|
unsigned int
|
|
vp8_get16x16var_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *SSE,
|
|
int *Sum
|
|
)
|
|
{
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 16, 16, SSE, Sum);
|
|
return (*SSE - (((*Sum) * (*Sum)) >> 8));
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned int vp8_variance16x16_c(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *sse)
|
|
{
|
|
unsigned int var;
|
|
int avg;
|
|
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 16, 16, &var, &avg);
|
|
*sse = var;
|
|
return (var - ((avg * avg) >> 8));
|
|
}
|
|
|
|
unsigned int vp8_variance8x16_c(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *sse)
|
|
{
|
|
unsigned int var;
|
|
int avg;
|
|
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 8, 16, &var, &avg);
|
|
*sse = var;
|
|
return (var - ((avg * avg) >> 7));
|
|
}
|
|
|
|
unsigned int vp8_variance16x8_c(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *sse)
|
|
{
|
|
unsigned int var;
|
|
int avg;
|
|
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 16, 8, &var, &avg);
|
|
*sse = var;
|
|
return (var - ((avg * avg) >> 7));
|
|
}
|
|
|
|
|
|
unsigned int vp8_variance8x8_c(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *sse)
|
|
{
|
|
unsigned int var;
|
|
int avg;
|
|
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 8, 8, &var, &avg);
|
|
*sse = var;
|
|
return (var - ((avg * avg) >> 6));
|
|
}
|
|
|
|
unsigned int vp8_variance4x4_c(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *sse)
|
|
{
|
|
unsigned int var;
|
|
int avg;
|
|
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 4, 4, &var, &avg);
|
|
*sse = var;
|
|
return (var - ((avg * avg) >> 4));
|
|
}
|
|
|
|
|
|
unsigned int vp8_mse16x16_c(
|
|
unsigned char *src_ptr,
|
|
int source_stride,
|
|
unsigned char *ref_ptr,
|
|
int recon_stride,
|
|
unsigned int *sse)
|
|
{
|
|
unsigned int var;
|
|
int avg;
|
|
|
|
vp8_variance(src_ptr, source_stride, ref_ptr, recon_stride, 16, 16, &var, &avg);
|
|
*sse = var;
|
|
return var;
|
|
}
|
|
|
|
|
|
/****************************************************************************
|
|
*
|
|
* ROUTINE : filter_block2d_bil_first_pass
|
|
*
|
|
* INPUTS : UINT8 *src_ptr : Pointer to source block.
|
|
* UINT32 src_pixels_per_line : Stride of input block.
|
|
* UINT32 pixel_step : Offset between filter input samples (see notes).
|
|
* UINT32 output_height : Input block height.
|
|
* UINT32 output_width : Input block width.
|
|
* INT32 *vp8_filter : Array of 2 bi-linear filter taps.
|
|
*
|
|
* OUTPUTS : INT32 *output_ptr : Pointer to filtered block.
|
|
*
|
|
* RETURNS : void
|
|
*
|
|
* FUNCTION : Applies a 1-D 2-tap bi-linear filter to the source block in
|
|
* either horizontal or vertical direction to produce the
|
|
* filtered output block. Used to implement first-pass
|
|
* of 2-D separable filter.
|
|
*
|
|
* SPECIAL NOTES : Produces INT32 output to retain precision for next pass.
|
|
* Two filter taps should sum to VP8_FILTER_WEIGHT.
|
|
* pixel_step defines whether the filter is applied
|
|
* horizontally (pixel_step=1) or vertically (pixel_step=stride).
|
|
* It defines the offset required to move from one input
|
|
* to the next.
|
|
*
|
|
****************************************************************************/
|
|
void vp8e_filter_block2d_bil_first_pass
|
|
(
|
|
unsigned char *src_ptr,
|
|
unsigned short *output_ptr,
|
|
unsigned int src_pixels_per_line,
|
|
int pixel_step,
|
|
unsigned int output_height,
|
|
unsigned int output_width,
|
|
const int *vp8_filter
|
|
)
|
|
{
|
|
unsigned int i, j;
|
|
|
|
for (i = 0; i < output_height; i++)
|
|
{
|
|
for (j = 0; j < output_width; j++)
|
|
{
|
|
// Apply bilinear filter
|
|
output_ptr[j] = (((int)src_ptr[0] * vp8_filter[0]) +
|
|
((int)src_ptr[pixel_step] * vp8_filter[1]) +
|
|
(VP8_FILTER_WEIGHT / 2)) >> VP8_FILTER_SHIFT;
|
|
src_ptr++;
|
|
}
|
|
|
|
// Next row...
|
|
src_ptr += src_pixels_per_line - output_width;
|
|
output_ptr += output_width;
|
|
}
|
|
}
|
|
|
|
/****************************************************************************
|
|
*
|
|
* ROUTINE : filter_block2d_bil_second_pass
|
|
*
|
|
* INPUTS : INT32 *src_ptr : Pointer to source block.
|
|
* UINT32 src_pixels_per_line : Stride of input block.
|
|
* UINT32 pixel_step : Offset between filter input samples (see notes).
|
|
* UINT32 output_height : Input block height.
|
|
* UINT32 output_width : Input block width.
|
|
* INT32 *vp8_filter : Array of 2 bi-linear filter taps.
|
|
*
|
|
* OUTPUTS : UINT16 *output_ptr : Pointer to filtered block.
|
|
*
|
|
* RETURNS : void
|
|
*
|
|
* FUNCTION : Applies a 1-D 2-tap bi-linear filter to the source block in
|
|
* either horizontal or vertical direction to produce the
|
|
* filtered output block. Used to implement second-pass
|
|
* of 2-D separable filter.
|
|
*
|
|
* SPECIAL NOTES : Requires 32-bit input as produced by filter_block2d_bil_first_pass.
|
|
* Two filter taps should sum to VP8_FILTER_WEIGHT.
|
|
* pixel_step defines whether the filter is applied
|
|
* horizontally (pixel_step=1) or vertically (pixel_step=stride).
|
|
* It defines the offset required to move from one input
|
|
* to the next.
|
|
*
|
|
****************************************************************************/
|
|
void vp8e_filter_block2d_bil_second_pass
|
|
(
|
|
unsigned short *src_ptr,
|
|
unsigned char *output_ptr,
|
|
unsigned int src_pixels_per_line,
|
|
unsigned int pixel_step,
|
|
unsigned int output_height,
|
|
unsigned int output_width,
|
|
const int *vp8_filter
|
|
)
|
|
{
|
|
unsigned int i, j;
|
|
int Temp;
|
|
|
|
for (i = 0; i < output_height; i++)
|
|
{
|
|
for (j = 0; j < output_width; j++)
|
|
{
|
|
// Apply filter
|
|
Temp = ((int)src_ptr[0] * vp8_filter[0]) +
|
|
((int)src_ptr[pixel_step] * vp8_filter[1]) +
|
|
(VP8_FILTER_WEIGHT / 2);
|
|
output_ptr[j] = (unsigned int)(Temp >> VP8_FILTER_SHIFT);
|
|
src_ptr++;
|
|
}
|
|
|
|
// Next row...
|
|
src_ptr += src_pixels_per_line - output_width;
|
|
output_ptr += output_width;
|
|
}
|
|
}
|
|
|
|
|
|
/****************************************************************************
|
|
*
|
|
* ROUTINE : filter_block2d_bil
|
|
*
|
|
* INPUTS : UINT8 *src_ptr : Pointer to source block.
|
|
* UINT32 src_pixels_per_line : Stride of input block.
|
|
* INT32 *HFilter : Array of 2 horizontal filter taps.
|
|
* INT32 *VFilter : Array of 2 vertical filter taps.
|
|
*
|
|
* OUTPUTS : UINT16 *output_ptr : Pointer to filtered block.
|
|
*
|
|
* RETURNS : void
|
|
*
|
|
* FUNCTION : 2-D filters an 8x8 input block by applying a 2-tap
|
|
* bi-linear filter horizontally followed by a 2-tap
|
|
* bi-linear filter vertically on the result.
|
|
*
|
|
* SPECIAL NOTES : The intermediate horizontally filtered block must produce
|
|
* 1 more point than the input block in each column. This
|
|
* is to ensure that the 2-tap filter has one extra data-point
|
|
* at the top of each column so filter taps do not extend
|
|
* beyond data. Thus the output of the first stage filter
|
|
* is an 8x9 (hx_v) block.
|
|
*
|
|
****************************************************************************/
|
|
void vp8e_filter_block2d_bil
|
|
(
|
|
unsigned char *src_ptr,
|
|
unsigned char *output_ptr,
|
|
unsigned int src_pixels_per_line,
|
|
int *HFilter,
|
|
int *VFilter
|
|
)
|
|
{
|
|
|
|
unsigned short FData[20*16]; // Temp data bufffer used in filtering
|
|
|
|
// First filter 1-D horizontally...
|
|
vp8e_filter_block2d_bil_first_pass(src_ptr, FData, src_pixels_per_line, 1, 9, 8, HFilter);
|
|
|
|
// then 1-D vertically...
|
|
vp8e_filter_block2d_bil_second_pass(FData, output_ptr, 8, 8, 8, 8, VFilter);
|
|
}
|
|
|
|
|
|
|
|
unsigned int vp8_sub_pixel_variance4x4_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int src_pixels_per_line,
|
|
int xoffset,
|
|
int yoffset,
|
|
unsigned char *dst_ptr,
|
|
int dst_pixels_per_line,
|
|
unsigned int *sse
|
|
)
|
|
{
|
|
unsigned char temp2[20*16];
|
|
const int *HFilter, *VFilter;
|
|
unsigned short FData3[5*4]; // Temp data bufffer used in filtering
|
|
|
|
HFilter = vp8_bilinear_taps[xoffset];
|
|
VFilter = vp8_bilinear_taps[yoffset];
|
|
|
|
// First filter 1d Horizontal
|
|
vp8e_filter_block2d_bil_first_pass(src_ptr, FData3, src_pixels_per_line, 1, 5, 4, HFilter);
|
|
|
|
// Now filter Verticaly
|
|
vp8e_filter_block2d_bil_second_pass(FData3, temp2, 4, 4, 4, 4, VFilter);
|
|
|
|
return vp8_variance4x4_c(temp2, 4, dst_ptr, dst_pixels_per_line, sse);
|
|
}
|
|
|
|
|
|
unsigned int vp8_sub_pixel_variance8x8_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int src_pixels_per_line,
|
|
int xoffset,
|
|
int yoffset,
|
|
unsigned char *dst_ptr,
|
|
int dst_pixels_per_line,
|
|
unsigned int *sse
|
|
)
|
|
{
|
|
unsigned short FData3[9*8]; // Temp data bufffer used in filtering
|
|
unsigned char temp2[20*16];
|
|
const int *HFilter, *VFilter;
|
|
|
|
HFilter = vp8_bilinear_taps[xoffset];
|
|
VFilter = vp8_bilinear_taps[yoffset];
|
|
|
|
vp8e_filter_block2d_bil_first_pass(src_ptr, FData3, src_pixels_per_line, 1, 9, 8, HFilter);
|
|
vp8e_filter_block2d_bil_second_pass(FData3, temp2, 8, 8, 8, 8, VFilter);
|
|
|
|
return vp8_variance8x8_c(temp2, 8, dst_ptr, dst_pixels_per_line, sse);
|
|
}
|
|
|
|
unsigned int vp8_sub_pixel_variance16x16_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int src_pixels_per_line,
|
|
int xoffset,
|
|
int yoffset,
|
|
unsigned char *dst_ptr,
|
|
int dst_pixels_per_line,
|
|
unsigned int *sse
|
|
)
|
|
{
|
|
unsigned short FData3[17*16]; // Temp data bufffer used in filtering
|
|
unsigned char temp2[20*16];
|
|
const int *HFilter, *VFilter;
|
|
|
|
HFilter = vp8_bilinear_taps[xoffset];
|
|
VFilter = vp8_bilinear_taps[yoffset];
|
|
|
|
vp8e_filter_block2d_bil_first_pass(src_ptr, FData3, src_pixels_per_line, 1, 17, 16, HFilter);
|
|
vp8e_filter_block2d_bil_second_pass(FData3, temp2, 16, 16, 16, 16, VFilter);
|
|
|
|
return vp8_variance16x16_c(temp2, 16, dst_ptr, dst_pixels_per_line, sse);
|
|
}
|
|
|
|
unsigned int vp8_sub_pixel_mse16x16_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int src_pixels_per_line,
|
|
int xoffset,
|
|
int yoffset,
|
|
unsigned char *dst_ptr,
|
|
int dst_pixels_per_line,
|
|
unsigned int *sse
|
|
)
|
|
{
|
|
vp8_sub_pixel_variance16x16_c(src_ptr, src_pixels_per_line, xoffset, yoffset, dst_ptr, dst_pixels_per_line, sse);
|
|
return *sse;
|
|
}
|
|
|
|
unsigned int vp8_sub_pixel_variance16x8_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int src_pixels_per_line,
|
|
int xoffset,
|
|
int yoffset,
|
|
unsigned char *dst_ptr,
|
|
int dst_pixels_per_line,
|
|
unsigned int *sse
|
|
)
|
|
{
|
|
unsigned short FData3[16*9]; // Temp data bufffer used in filtering
|
|
unsigned char temp2[20*16];
|
|
const int *HFilter, *VFilter;
|
|
|
|
HFilter = vp8_bilinear_taps[xoffset];
|
|
VFilter = vp8_bilinear_taps[yoffset];
|
|
|
|
vp8e_filter_block2d_bil_first_pass(src_ptr, FData3, src_pixels_per_line, 1, 9, 16, HFilter);
|
|
vp8e_filter_block2d_bil_second_pass(FData3, temp2, 16, 16, 8, 16, VFilter);
|
|
|
|
return vp8_variance16x8_c(temp2, 16, dst_ptr, dst_pixels_per_line, sse);
|
|
}
|
|
|
|
unsigned int vp8_sub_pixel_variance8x16_c
|
|
(
|
|
unsigned char *src_ptr,
|
|
int src_pixels_per_line,
|
|
int xoffset,
|
|
int yoffset,
|
|
unsigned char *dst_ptr,
|
|
int dst_pixels_per_line,
|
|
unsigned int *sse
|
|
)
|
|
{
|
|
unsigned short FData3[9*16]; // Temp data bufffer used in filtering
|
|
unsigned char temp2[20*16];
|
|
const int *HFilter, *VFilter;
|
|
|
|
|
|
HFilter = vp8_bilinear_taps[xoffset];
|
|
VFilter = vp8_bilinear_taps[yoffset];
|
|
|
|
|
|
vp8e_filter_block2d_bil_first_pass(src_ptr, FData3, src_pixels_per_line, 1, 17, 8, HFilter);
|
|
vp8e_filter_block2d_bil_second_pass(FData3, temp2, 8, 8, 16, 8, VFilter);
|
|
|
|
return vp8_variance8x16_c(temp2, 8, dst_ptr, dst_pixels_per_line, sse);
|
|
}
|
|
#endif
|