mcomp.h 4.3 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5
6
 *  that can be found in the LICENSE file in the root of the source
 *  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.
John Koleszar's avatar
John Koleszar committed
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 */


#ifndef __INC_MCOMP_H
#define __INC_MCOMP_H

#include "block.h"
#include "variance.h"

#ifdef ENTROPY_STATS
extern void init_mv_ref_counts();
extern void accum_mv_refs(MB_PREDICTION_MODE, const int near_mv_ref_cts[4]);
#endif


#define MAX_MVSEARCH_STEPS 8                                    // The maximum number of steps in a step search given the largest allowed initial step
#define MAX_FULL_PEL_VAL ((1 << (MAX_MVSEARCH_STEPS+3)) - 8)    // Max full pel mv specified in 1/8 pel units
#define MAX_FIRST_STEP (1 << (MAX_MVSEARCH_STEPS-1))            // Maximum size of the first step in full pel units

extern void print_mode_context(void);
Scott LaVarnway's avatar
Scott LaVarnway committed
29
extern int vp8_mv_bit_cost(int_mv *mv, int_mv *ref, int *mvcost[2], int Weight);
John Koleszar's avatar
John Koleszar committed
30
31
32
33
34
35
36
37
38
extern void vp8_init_dsmotion_compensation(MACROBLOCK *x, int stride);
extern void vp8_init3smotion_compensation(MACROBLOCK *x,  int stride);


extern int vp8_hex_search
(
    MACROBLOCK *x,
    BLOCK *b,
    BLOCKD *d,
Scott LaVarnway's avatar
Scott LaVarnway committed
39
40
    int_mv *ref_mv,
    int_mv *best_mv,
John Koleszar's avatar
John Koleszar committed
41
42
    int search_param,
    int error_per_bit,
43
    const vp8_variance_fn_ptr_t *vf,
John Koleszar's avatar
John Koleszar committed
44
    int *mvsadcost[2],
45
    int *mvcost[2],
Scott LaVarnway's avatar
Scott LaVarnway committed
46
    int_mv *center_mv
John Koleszar's avatar
John Koleszar committed
47
48
);

49
typedef int (fractional_mv_step_fp)
Scott LaVarnway's avatar
Scott LaVarnway committed
50
51
52
    (MACROBLOCK *x, BLOCK *b, BLOCKD *d, int_mv *bestmv, int_mv *ref_mv,
     int error_per_bit, const vp8_variance_fn_ptr_t *vfp, int *mvcost[2],
     int *distortion, unsigned int *sse);
John Koleszar's avatar
John Koleszar committed
53
54
55
56
57
58
59
60
61
62
63
extern fractional_mv_step_fp vp8_find_best_sub_pixel_step_iteratively;
extern fractional_mv_step_fp vp8_find_best_sub_pixel_step;
extern fractional_mv_step_fp vp8_find_best_half_pixel_step;
extern fractional_mv_step_fp vp8_skip_fractional_mv_step;

#define prototype_full_search_sad(sym)\
    int (sym)\
    (\
     MACROBLOCK *x, \
     BLOCK *b, \
     BLOCKD *d, \
Scott LaVarnway's avatar
Scott LaVarnway committed
64
     int_mv *ref_mv, \
65
     int sad_per_bit, \
John Koleszar's avatar
John Koleszar committed
66
67
68
     int distance, \
     vp8_variance_fn_ptr_t *fn_ptr, \
     int *mvcost[2], \
Scott LaVarnway's avatar
Scott LaVarnway committed
69
     int_mv *center_mv \
John Koleszar's avatar
John Koleszar committed
70
71
    )

72
73
74
75
76
77
#define prototype_refining_search_sad(sym)\
    int (sym)\
    (\
     MACROBLOCK *x, \
     BLOCK *b, \
     BLOCKD *d, \
Scott LaVarnway's avatar
Scott LaVarnway committed
78
     int_mv *ref_mv, \
79
     int sad_per_bit, \
80
81
82
     int distance, \
     vp8_variance_fn_ptr_t *fn_ptr, \
     int *mvcost[2], \
Scott LaVarnway's avatar
Scott LaVarnway committed
83
     int_mv *center_mv \
84
85
    )

John Koleszar's avatar
John Koleszar committed
86
87
88
89
90
91
#define prototype_diamond_search_sad(sym)\
    int (sym)\
    (\
     MACROBLOCK *x, \
     BLOCK *b, \
     BLOCKD *d, \
Scott LaVarnway's avatar
Scott LaVarnway committed
92
93
     int_mv *ref_mv, \
     int_mv *best_mv, \
John Koleszar's avatar
John Koleszar committed
94
     int search_param, \
95
     int sad_per_bit, \
John Koleszar's avatar
John Koleszar committed
96
97
     int *num00, \
     vp8_variance_fn_ptr_t *fn_ptr, \
98
     int *mvcost[2], \
Scott LaVarnway's avatar
Scott LaVarnway committed
99
     int_mv *center_mv \
John Koleszar's avatar
John Koleszar committed
100
101
102
103
104
105
106
107
108
    )

#if ARCH_X86 || ARCH_X86_64
#include "x86/mcomp_x86.h"
#endif

typedef prototype_full_search_sad(*vp8_full_search_fn_t);
extern prototype_full_search_sad(vp8_full_search_sad);
extern prototype_full_search_sad(vp8_full_search_sadx3);
109
extern prototype_full_search_sad(vp8_full_search_sadx8);
John Koleszar's avatar
John Koleszar committed
110

111
112
113
114
typedef prototype_refining_search_sad(*vp8_refining_search_fn_t);
extern prototype_refining_search_sad(vp8_refining_search_sad);
extern prototype_refining_search_sad(vp8_refining_search_sadx4);

John Koleszar's avatar
John Koleszar committed
115
116
117
118
119
120
121
122
123
typedef prototype_diamond_search_sad(*vp8_diamond_search_fn_t);
extern prototype_diamond_search_sad(vp8_diamond_search_sad);
extern prototype_diamond_search_sad(vp8_diamond_search_sadx4);

#ifndef vp8_search_full_search
#define vp8_search_full_search vp8_full_search_sad
#endif
extern prototype_full_search_sad(vp8_search_full_search);

124
125
126
127
128
#ifndef vp8_search_refining_search
#define vp8_search_refining_search vp8_refining_search_sad
#endif
extern prototype_refining_search_sad(vp8_search_refining_search);

John Koleszar's avatar
John Koleszar committed
129
130
131
132
133
134
135
136
#ifndef vp8_search_diamond_search
#define vp8_search_diamond_search vp8_diamond_search_sad
#endif
extern prototype_diamond_search_sad(vp8_search_diamond_search);

typedef struct
{
    prototype_full_search_sad(*full_search);
137
    prototype_refining_search_sad(*refining_search);
John Koleszar's avatar
John Koleszar committed
138
139
140
141
142
143
144
145
146
147
    prototype_diamond_search_sad(*diamond_search);
} vp8_search_rtcd_vtable_t;

#if CONFIG_RUNTIME_CPU_DETECT
#define SEARCH_INVOKE(ctx,fn) (ctx)->fn
#else
#define SEARCH_INVOKE(ctx,fn) vp8_search_##fn
#endif

#endif