c3bdffb0a5
subpel functions will be moved in another patch. Change-Id: Idb2e049bad0b9b32ac42cc7731cd6903de2826ce
136 lines
4.8 KiB
C
136 lines
4.8 KiB
C
/*
|
|
* Copyright (c) 2010 The WebM 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.
|
|
*/
|
|
|
|
#ifndef VP9_ENCODER_VP9_VARIANCE_H_
|
|
#define VP9_ENCODER_VP9_VARIANCE_H_
|
|
|
|
#include "vpx/vpx_integer.h"
|
|
#include "vpx_ports/mem.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
// TODO(johannkoenig): All functions which depend on
|
|
// [highbd_][8|10|12_]variance should be refactored or moved to vpx_dsp.
|
|
static void variance(const uint8_t *a, int a_stride,
|
|
const uint8_t *b, int b_stride,
|
|
int w, int h, unsigned int *sse, int *sum) {
|
|
int i, j;
|
|
|
|
*sum = 0;
|
|
*sse = 0;
|
|
|
|
for (i = 0; i < h; i++) {
|
|
for (j = 0; j < w; j++) {
|
|
const int diff = a[j] - b[j];
|
|
*sum += diff;
|
|
*sse += diff * diff;
|
|
}
|
|
|
|
a += a_stride;
|
|
b += b_stride;
|
|
}
|
|
}
|
|
|
|
#if CONFIG_VP9_HIGHBITDEPTH
|
|
static void highbd_variance64(const uint8_t *a8, int a_stride,
|
|
const uint8_t *b8, int b_stride,
|
|
int w, int h, uint64_t *sse, uint64_t *sum) {
|
|
int i, j;
|
|
|
|
uint16_t *a = CONVERT_TO_SHORTPTR(a8);
|
|
uint16_t *b = CONVERT_TO_SHORTPTR(b8);
|
|
*sum = 0;
|
|
*sse = 0;
|
|
|
|
for (i = 0; i < h; i++) {
|
|
for (j = 0; j < w; j++) {
|
|
const int diff = a[j] - b[j];
|
|
*sum += diff;
|
|
*sse += diff * diff;
|
|
}
|
|
a += a_stride;
|
|
b += b_stride;
|
|
}
|
|
}
|
|
static void highbd_8_variance(const uint8_t *a8, int a_stride,
|
|
const uint8_t *b8, int b_stride,
|
|
int w, int h, unsigned int *sse, int *sum) {
|
|
uint64_t sse_long = 0;
|
|
uint64_t sum_long = 0;
|
|
highbd_variance64(a8, a_stride, b8, b_stride, w, h, &sse_long, &sum_long);
|
|
*sse = (unsigned int)sse_long;
|
|
*sum = (int)sum_long;
|
|
}
|
|
#endif
|
|
|
|
typedef unsigned int(*vp9_sad_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
const uint8_t *ref_ptr,
|
|
int ref_stride);
|
|
|
|
typedef unsigned int(*vp9_sad_avg_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
const uint8_t *ref_ptr,
|
|
int ref_stride,
|
|
const uint8_t *second_pred);
|
|
|
|
typedef void (*vp9_sad_multi_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
const uint8_t *ref_ptr,
|
|
int ref_stride,
|
|
unsigned int *sad_array);
|
|
|
|
typedef void (*vp9_sad_multi_d_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
const uint8_t* const ref_ptr[],
|
|
int ref_stride, unsigned int *sad_array);
|
|
|
|
typedef unsigned int (*vp9_variance_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
const uint8_t *ref_ptr,
|
|
int ref_stride,
|
|
unsigned int *sse);
|
|
|
|
typedef unsigned int (*vp9_subpixvariance_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
int xoffset,
|
|
int yoffset,
|
|
const uint8_t *ref_ptr,
|
|
int Refstride,
|
|
unsigned int *sse);
|
|
|
|
typedef unsigned int (*vp9_subp_avg_variance_fn_t)(const uint8_t *src_ptr,
|
|
int source_stride,
|
|
int xoffset,
|
|
int yoffset,
|
|
const uint8_t *ref_ptr,
|
|
int Refstride,
|
|
unsigned int *sse,
|
|
const uint8_t *second_pred);
|
|
|
|
typedef struct vp9_variance_vtable {
|
|
vp9_sad_fn_t sdf;
|
|
vp9_sad_avg_fn_t sdaf;
|
|
vp9_variance_fn_t vf;
|
|
vp9_subpixvariance_fn_t svf;
|
|
vp9_subp_avg_variance_fn_t svaf;
|
|
vp9_sad_multi_fn_t sdx3f;
|
|
vp9_sad_multi_fn_t sdx8f;
|
|
vp9_sad_multi_d_fn_t sdx4df;
|
|
} vp9_variance_fn_ptr_t;
|
|
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|
|
|
|
#endif // VP9_ENCODER_VP9_VARIANCE_H_
|