1/*
2 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  tree. An additional intellectual property rights grant can be found
7 *  in the file PATENTS.  All contributing project authors may
8 *  be found in the AUTHORS file in the root of the source tree.
9 */
10
11#ifndef VP9_ENCODER_VP9_VARIANCE_H_
12#define VP9_ENCODER_VP9_VARIANCE_H_
13
14#include "vpx/vpx_integer.h"
15
16#ifdef __cplusplus
17extern "C" {
18#endif
19
20void variance(const uint8_t *src_ptr,
21              int  source_stride,
22              const uint8_t *ref_ptr,
23              int  recon_stride,
24              int  w,
25              int  h,
26              unsigned int *sse,
27              int *sum);
28
29typedef unsigned int(*vp9_sad_fn_t)(const uint8_t *src_ptr,
30                                    int source_stride,
31                                    const uint8_t *ref_ptr,
32                                    int ref_stride,
33                                    unsigned int max_sad);
34
35typedef unsigned int(*vp9_sad_avg_fn_t)(const uint8_t *src_ptr,
36                                        int source_stride,
37                                        const uint8_t *ref_ptr,
38                                        int ref_stride,
39                                        const uint8_t *second_pred,
40                                        unsigned int max_sad);
41
42typedef void (*vp9_sad_multi_fn_t)(const uint8_t *src_ptr,
43                                   int source_stride,
44                                   const uint8_t *ref_ptr,
45                                   int  ref_stride,
46                                   unsigned int *sad_array);
47
48typedef void (*vp9_sad_multi1_fn_t)(const uint8_t *src_ptr,
49                                    int source_stride,
50                                    const uint8_t *ref_ptr,
51                                    int  ref_stride,
52                                    unsigned int *sad_array);
53
54typedef void (*vp9_sad_multi_d_fn_t)(const uint8_t *src_ptr,
55                                     int source_stride,
56                                     const uint8_t* const ref_ptr[],
57                                     int  ref_stride, unsigned int *sad_array);
58
59typedef unsigned int (*vp9_variance_fn_t)(const uint8_t *src_ptr,
60                                          int source_stride,
61                                          const uint8_t *ref_ptr,
62                                          int ref_stride,
63                                          unsigned int *sse);
64
65typedef unsigned int (*vp9_subpixvariance_fn_t)(const uint8_t *src_ptr,
66                                                int source_stride,
67                                                int xoffset,
68                                                int yoffset,
69                                                const uint8_t *ref_ptr,
70                                                int Refstride,
71                                                unsigned int *sse);
72
73typedef unsigned int (*vp9_subp_avg_variance_fn_t)(const uint8_t *src_ptr,
74                                                   int source_stride,
75                                                   int xoffset,
76                                                   int yoffset,
77                                                   const uint8_t *ref_ptr,
78                                                   int Refstride,
79                                                   unsigned int *sse,
80                                                   const uint8_t *second_pred);
81
82typedef unsigned int (*vp9_getmbss_fn_t)(const short *);
83
84typedef unsigned int (*vp9_get16x16prederror_fn_t)(const uint8_t *src_ptr,
85                                                   int source_stride,
86                                                   const uint8_t *ref_ptr,
87                                                   int  ref_stride);
88
89typedef struct vp9_variance_vtable {
90  vp9_sad_fn_t               sdf;
91  vp9_sad_avg_fn_t           sdaf;
92  vp9_variance_fn_t          vf;
93  vp9_subpixvariance_fn_t    svf;
94  vp9_subp_avg_variance_fn_t svaf;
95  vp9_variance_fn_t          svf_halfpix_h;
96  vp9_variance_fn_t          svf_halfpix_v;
97  vp9_variance_fn_t          svf_halfpix_hv;
98  vp9_sad_multi_fn_t         sdx3f;
99  vp9_sad_multi1_fn_t        sdx8f;
100  vp9_sad_multi_d_fn_t       sdx4df;
101} vp9_variance_fn_ptr_t;
102
103void vp9_comp_avg_pred(uint8_t *comp_pred, const uint8_t *pred, int width,
104                       int height, const uint8_t *ref, int ref_stride);
105
106#ifdef __cplusplus
107}  // extern "C"
108#endif
109
110#endif  // VP9_ENCODER_VP9_VARIANCE_H_
111