block.h 5.54 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
 */


#ifndef __INC_BLOCK_H
#define __INC_BLOCK_H

John Koleszar's avatar
John Koleszar committed
15
16
17
#include "vp8/common/onyx.h"
#include "vp8/common/entropymv.h"
#include "vp8/common/entropy.h"
John Koleszar's avatar
John Koleszar committed
18
#include "vpx_ports/mem.h"
19
#include "vp8/common/onyxc_int.h"
John Koleszar's avatar
John Koleszar committed
20
21

// motion search site
John Koleszar's avatar
John Koleszar committed
22
23
24
typedef struct {
  MV mv;
  int offset;
John Koleszar's avatar
John Koleszar committed
25
26
} search_site;

John Koleszar's avatar
John Koleszar committed
27
28
29
30
31
32
33
34
35
36
37
typedef struct {
  // 16 Y blocks, 4 U blocks, 4 V blocks each with 16 entries
  short *src_diff;
  short *coeff;

  // 16 Y blocks, 4 U blocks, 4 V blocks each with 16 entries
  short *quant;
  short *quant_fast;      // fast quant deprecated for now
  unsigned char *quant_shift;
  short *zbin;
  short *zbin_8x8;
38
#if CONFIG_TX16X16 || CONFIG_HYBRIDTRANSFORM16X16
Daniel Kang's avatar
Daniel Kang committed
39
40
  short *zbin_16x16;
#endif
John Koleszar's avatar
John Koleszar committed
41
42
  short *zrun_zbin_boost;
  short *zrun_zbin_boost_8x8;
43
#if CONFIG_TX16X16 || CONFIG_HYBRIDTRANSFORM16X16
Daniel Kang's avatar
Daniel Kang committed
44
45
  short *zrun_zbin_boost_16x16;
#endif
John Koleszar's avatar
John Koleszar committed
46
47
48
49
50
51
52
53
54
55
56
57
  short *round;

  // Zbin Over Quant value
  short zbin_extra;

  unsigned char **base_src;
  unsigned char **base_second_src;
  int src;
  int src_stride;

  int eob_max_offset;
  int eob_max_offset_8x8;
58
#if CONFIG_TX16X16 || CONFIG_HYBRIDTRANSFORM16X16
Daniel Kang's avatar
Daniel Kang committed
59
60
  int eob_max_offset_16x16;
#endif
John Koleszar's avatar
John Koleszar committed
61
62
} BLOCK;

John Koleszar's avatar
John Koleszar committed
63
64
65
66
67
68
69
typedef struct {
  int count;
  struct {
    B_PREDICTION_MODE mode;
    int_mv mv;
    int_mv second_mv;
  } bmi[16];
70
71
} PARTITION_INFO;

Adrian Grange's avatar
Adrian Grange committed
72
73
// Structure to hold snapshot of coding context during the mode picking process
// TODO Do we need all of these?
John Koleszar's avatar
John Koleszar committed
74
75
76
77
78
79
80
typedef struct {
  MODE_INFO mic;
  PARTITION_INFO partition_info;
  int_mv best_ref_mv;
  int_mv second_best_ref_mv;
  int rate;
  int distortion;
81
  int64_t intra_error;
John Koleszar's avatar
John Koleszar committed
82
83
84
  int best_mode_index;
  int rddiv;
  int rdmult;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
85
86
87
  int hybrid_pred_diff;
  int comp_pred_diff;
  int single_pred_diff;
88
89
90
#if CONFIG_TX_SELECT
  int64_t txfm_rd_diff[NB_TXFM_MODES];
#endif
Adrian Grange's avatar
Adrian Grange committed
91
92
} PICK_MODE_CONTEXT;

John Koleszar's avatar
John Koleszar committed
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
typedef struct {
  DECLARE_ALIGNED(16, short, src_diff[400]);  // 16x16 Y 8x8 U 8x8 V 4x4 2nd Y
  DECLARE_ALIGNED(16, short, coeff[400]);     // 16x16 Y 8x8 U 8x8 V 4x4 2nd Y
  DECLARE_ALIGNED(16, unsigned char, thismb[256]);    // 16x16 Y

  unsigned char *thismb_ptr;
  // 16 Y blocks, 4 U blocks, 4 V blocks,
  // 1 DC 2nd order block each with 16 entries
  BLOCK block[25];

  YV12_BUFFER_CONFIG src;

  MACROBLOCKD e_mbd;
  PARTITION_INFO *partition_info; /* work pointer */
  PARTITION_INFO *pi;   /* Corresponds to upper left visible macroblock */
  PARTITION_INFO *pip;  /* Base of allocated array */

  search_site *ss;
  int ss_count;
  int searches_per_step;

  int errorperbit;
  int sadperbit16;
  int sadperbit4;
  int rddiv;
  int rdmult;
  unsigned int *mb_activity_ptr;
  int *mb_norm_activity_ptr;
  signed int act_zbin_adj;

123
124
125
126
127
128
129
130
131
132
133
134
135
#if CONFIG_NEWMVENTROPY
  int nmvjointcost[MV_JOINTS];
  int nmvcosts[2][MV_VALS];
  int *nmvcost[2];
  int nmvcosts_hp[2][MV_VALS];
  int *nmvcost_hp[2];

  int nmvjointsadcost[MV_JOINTS];
  int nmvsadcosts[2][MV_VALS];
  int *nmvsadcost[2];
  int nmvsadcosts_hp[2][MV_VALS];
  int *nmvsadcost_hp[2];
#else
John Koleszar's avatar
John Koleszar committed
136
137
138
139
140
141
142
143
  int mvcosts[2][MVvals + 1];
  int *mvcost[2];
  int mvsadcosts[2][MVfpvals + 1];
  int *mvsadcost[2];
  int mvcosts_hp[2][MVvals_hp + 1];
  int *mvcost_hp[2];
  int mvsadcosts_hp[2][MVfpvals_hp + 1];
  int *mvsadcost_hp[2];
144
#endif  /* CONFIG_NEWMVENTROPY */
145

John Koleszar's avatar
John Koleszar committed
146
147
148
149
150
  int mbmode_cost[2][MB_MODE_COUNT];
  int intra_uv_mode_cost[2][MB_MODE_COUNT];
  int bmode_costs[VP8_BINTRAMODES][VP8_BINTRAMODES][VP8_BINTRAMODES];
  int i8x8_mode_costs[MB_MODE_COUNT];
  int inter_bmode_costs[B_MODE_COUNT];
151
152
153
154
#if CONFIG_SWITCHABLE_INTERP
  int switchable_interp_costs[VP8_SWITCHABLE_FILTERS+1]
                             [VP8_SWITCHABLE_FILTERS];
#endif
John Koleszar's avatar
John Koleszar committed
155
156
157
158
159
160

  // These define limits to motion vector components to prevent them from extending outside the UMV borders
  int mv_col_min;
  int mv_col_max;
  int mv_row_min;
  int mv_row_max;
John Koleszar's avatar
John Koleszar committed
161

John Koleszar's avatar
John Koleszar committed
162
  int skip;
John Koleszar's avatar
John Koleszar committed
163

John Koleszar's avatar
John Koleszar committed
164
  int encode_breakout;
John Koleszar's avatar
John Koleszar committed
165

John Koleszar's avatar
John Koleszar committed
166
167
  // char * gf_active_ptr;
  signed char *gf_active_ptr;
168

John Koleszar's avatar
John Koleszar committed
169
  unsigned char *active_ptr;
John Koleszar's avatar
John Koleszar committed
170

171
  unsigned int token_costs[TX_SIZE_MAX][BLOCK_TYPES][COEF_BANDS]
Daniel Kang's avatar
Daniel Kang committed
172
    [PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];
173
174
175
176
#if CONFIG_HYBRIDTRANSFORM || CONFIG_HYBRIDTRANSFORM8X8 || CONFIG_HYBRIDTRANSFORM16X16
  unsigned int hybrid_token_costs[TX_SIZE_MAX][BLOCK_TYPES][COEF_BANDS]
    [PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];
#endif
177

John Koleszar's avatar
John Koleszar committed
178
179
  int optimize;
  int q_index;
John Koleszar's avatar
John Koleszar committed
180

John Koleszar's avatar
John Koleszar committed
181
182
183
  // Structure to hold context for each of the 4 MBs within a SB:
  // when encoded as 4 independent MBs:
  PICK_MODE_CONTEXT mb_context[4];
Adrian Grange's avatar
Adrian Grange committed
184
#if CONFIG_SUPERBLOCKS
John Koleszar's avatar
John Koleszar committed
185
186
  // when 4 MBs share coding parameters:
  PICK_MODE_CONTEXT sb_context[4];
Adrian Grange's avatar
Adrian Grange committed
187
188
#endif

John Koleszar's avatar
John Koleszar committed
189
190
191
192
193
194
  void (*vp8_short_fdct4x4)(short *input, short *output, int pitch);
  void (*vp8_short_fdct8x4)(short *input, short *output, int pitch);
  void (*short_walsh4x4)(short *input, short *output, int pitch);
  void (*quantize_b)(BLOCK *b, BLOCKD *d);
  void (*quantize_b_pair)(BLOCK *b1, BLOCK *b2, BLOCKD *d0, BLOCKD *d1);
  void (*vp8_short_fdct8x8)(short *input, short *output, int pitch);
195
#if CONFIG_TX16X16 || CONFIG_HYBRIDTRANSFORM16X16
Daniel Kang's avatar
Daniel Kang committed
196
197
  void (*vp8_short_fdct16x16)(short *input, short *output, int pitch);
#endif
John Koleszar's avatar
John Koleszar committed
198
  void (*short_fhaar2x2)(short *input, short *output, int pitch);
199
#if CONFIG_TX16X16 || CONFIG_HYBRIDTRANSFORM16X16
Daniel Kang's avatar
Daniel Kang committed
200
201
  void (*quantize_b_16x16)(BLOCK *b, BLOCKD *d);
#endif
John Koleszar's avatar
John Koleszar committed
202
203
  void (*quantize_b_8x8)(BLOCK *b, BLOCKD *d);
  void (*quantize_b_2x2)(BLOCK *b, BLOCKD *d);
John Koleszar's avatar
John Koleszar committed
204
205
206
207
208

} MACROBLOCK;


#endif