vp9_block.h 7.06 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
#ifndef VP9_ENCODER_VP9_BLOCK_H_
#define VP9_ENCODER_VP9_BLOCK_H_
John Koleszar's avatar
John Koleszar committed
13

14
15
16
#include "vp9/common/vp9_onyx.h"
#include "vp9/common/vp9_entropymv.h"
#include "vp9/common/vp9_entropy.h"
John Koleszar's avatar
John Koleszar committed
17
#include "vpx_ports/mem.h"
18
#include "vp9/common/vp9_onyxc_int.h"
John Koleszar's avatar
John Koleszar committed
19
20

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

Adrian Grange's avatar
Adrian Grange committed
26
// Structure to hold snapshot of coding context during the mode picking process
John Koleszar's avatar
John Koleszar committed
27
28
typedef struct {
  MODE_INFO mic;
29
  uint8_t *zcoeff_blk;
30
31
32
33
34
  int16_t *coeff[MAX_MB_PLANE][2];
  int16_t *qcoeff[MAX_MB_PLANE][2];
  int16_t *dqcoeff[MAX_MB_PLANE][2];
  uint16_t *eobs[MAX_MB_PLANE][2];

35
  // dual buffer pointers, 0: in use, 1: best in store
36
37
38
39
40
  int16_t *coeff_pbuf[MAX_MB_PLANE][2];
  int16_t *qcoeff_pbuf[MAX_MB_PLANE][2];
  int16_t *dqcoeff_pbuf[MAX_MB_PLANE][2];
  uint16_t *eobs_pbuf[MAX_MB_PLANE][2];

41
  int is_coded;
42
  int num_4x4_blk;
43
  int skip;
John Koleszar's avatar
John Koleszar committed
44
45
  int_mv best_ref_mv;
  int_mv second_best_ref_mv;
Paul Wilkins's avatar
Paul Wilkins committed
46
  int_mv ref_mvs[MAX_REF_FRAMES][MAX_MV_REF_CANDIDATES];
John Koleszar's avatar
John Koleszar committed
47
48
  int rate;
  int distortion;
49
  int64_t intra_error;
John Koleszar's avatar
John Koleszar committed
50
51
52
  int best_mode_index;
  int rddiv;
  int rdmult;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
53
54
55
  int hybrid_pred_diff;
  int comp_pred_diff;
  int single_pred_diff;
56
  int64_t tx_rd_diff[TX_MODES];
57
  int64_t best_filter_diff[SWITCHABLE_FILTER_CONTEXTS];
Jim Bankoski's avatar
Jim Bankoski committed
58

Jingning Han's avatar
Jingning Han committed
59
60
61
62
  // motion vector cache for adaptive motion search control in partition
  // search loop
  int_mv pred_mv[MAX_REF_FRAMES];

Jim Bankoski's avatar
Jim Bankoski committed
63
64
65
66
67
  // Bit flag for each mode whether it has high error in comparison to others.
  unsigned int modes_with_high_error;

  // Bit flag for each ref frame whether it has high error compared to others.
  unsigned int frames_with_high_error;
Adrian Grange's avatar
Adrian Grange committed
68
69
} PICK_MODE_CONTEXT;

70
struct macroblock_plane {
71
  DECLARE_ALIGNED(16, int16_t, src_diff[64 * 64]);
72
  int16_t *coeff;
John Koleszar's avatar
John Koleszar committed
73
  struct buf_2d src;
74
75
76

  // Quantizer setings
  int16_t *quant;
77
  int16_t *quant_shift;
78
79
80
81
82
  int16_t *zbin;
  int16_t *round;

  // Zbin Over Quant value
  int16_t zbin_extra;
83
84
};

85
86
87
88
89
/* The [2] dimension is for whether we skip the EOB node (i.e. if previous
 * coefficient in this block was zero) or not. */
typedef unsigned int vp9_coeff_cost[BLOCK_TYPES][REF_TYPES][COEF_BANDS][2]
                                   [PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];

90
91
typedef struct macroblock MACROBLOCK;
struct macroblock {
92
  struct macroblock_plane plane[MAX_MB_PLANE];
John Koleszar's avatar
John Koleszar committed
93
94

  MACROBLOCKD e_mbd;
John Koleszar's avatar
John Koleszar committed
95
  int skip_block;
96
97
  int select_txfm_size;
  int skip_optimize;
John Koleszar's avatar
John Koleszar committed
98
99
100
101
102
103
104
105
106
107

  search_site *ss;
  int ss_count;
  int searches_per_step;

  int errorperbit;
  int sadperbit16;
  int sadperbit4;
  int rddiv;
  int rdmult;
108
  unsigned int mb_energy;
John Koleszar's avatar
John Koleszar committed
109
110
111
112
  unsigned int *mb_activity_ptr;
  int *mb_norm_activity_ptr;
  signed int act_zbin_adj;

113
  int mv_best_ref_index[MAX_REF_FRAMES];
114
  unsigned int max_mv_context[MAX_REF_FRAMES];
115
  unsigned int source_variance;
116

117
118
119
120
121
  int nmvjointcost[MV_JOINTS];
  int nmvcosts[2][MV_VALS];
  int *nmvcost[2];
  int nmvcosts_hp[2][MV_VALS];
  int *nmvcost_hp[2];
122
  int **mvcost;
123
124
125
126
127
128

  int nmvjointsadcost[MV_JOINTS];
  int nmvsadcosts[2][MV_VALS];
  int *nmvsadcost[2];
  int nmvsadcosts_hp[2][MV_VALS];
  int *nmvsadcost_hp[2];
129
  int **mvsadcost;
130

131
  int mbmode_cost[MB_MODE_COUNT];
132
  unsigned inter_mode_cost[INTER_MODE_CONTEXTS][INTER_MODES];
John Koleszar's avatar
John Koleszar committed
133
  int intra_uv_mode_cost[2][MB_MODE_COUNT];
134
  int y_mode_costs[INTRA_MODES][INTRA_MODES][INTRA_MODES];
135
  int switchable_interp_costs[SWITCHABLE_FILTER_CONTEXTS][SWITCHABLE_FILTERS];
John Koleszar's avatar
John Koleszar committed
136

137
138
  // These define limits to motion vector components to prevent them
  // from extending outside the UMV borders
John Koleszar's avatar
John Koleszar committed
139
140
141
142
  int mv_col_min;
  int mv_col_max;
  int mv_row_min;
  int mv_row_max;
John Koleszar's avatar
John Koleszar committed
143

144
  uint8_t zcoeff_blk[TX_SIZES][256];
John Koleszar's avatar
John Koleszar committed
145
  int skip;
John Koleszar's avatar
John Koleszar committed
146

John Koleszar's avatar
John Koleszar committed
147
  int encode_breakout;
John Koleszar's avatar
John Koleszar committed
148

John Koleszar's avatar
John Koleszar committed
149
  unsigned char *active_ptr;
John Koleszar's avatar
John Koleszar committed
150

151
  // note that token_costs is the cost when eob node is skipped
152
  vp9_coeff_cost token_costs[TX_SIZES];
153
  DECLARE_ALIGNED(16, uint8_t, token_cache[1024]);
154

John Koleszar's avatar
John Koleszar committed
155
  int optimize;
John Koleszar's avatar
John Koleszar committed
156

157
  // indicate if it is in the rd search loop or encoding process
158
  int use_lp32x32fdct;
159
  int skip_encode;
160

161
162
  // Used to store sub partition's choices.
  int fast_ms;
Jingning Han's avatar
Jingning Han committed
163
  int_mv pred_mv[MAX_REF_FRAMES];
164
165
  int subblock_ref;

166
167
  // TODO(jingning): Need to refactor the structure arrays that buffers the
  // coding mode decisions of each partition type.
168
169
170
171
  PICK_MODE_CONTEXT ab4x4_context[4][4][4];
  PICK_MODE_CONTEXT sb8x4_context[4][4][4];
  PICK_MODE_CONTEXT sb4x8_context[4][4][4];
  PICK_MODE_CONTEXT sb8x8_context[4][4][4];
172
173
  PICK_MODE_CONTEXT sb8x16_context[4][4][2];
  PICK_MODE_CONTEXT sb16x8_context[4][4][2];
Ronald S. Bultje's avatar
Ronald S. Bultje committed
174
  PICK_MODE_CONTEXT mb_context[4][4];
175
176
  PICK_MODE_CONTEXT sb32x16_context[4][2];
  PICK_MODE_CONTEXT sb16x32_context[4][2];
John Koleszar's avatar
John Koleszar committed
177
  // when 4 MBs share coding parameters:
Ronald S. Bultje's avatar
Ronald S. Bultje committed
178
  PICK_MODE_CONTEXT sb32_context[4];
179
180
  PICK_MODE_CONTEXT sb32x64_context[2];
  PICK_MODE_CONTEXT sb64x32_context[2];
Ronald S. Bultje's avatar
Ronald S. Bultje committed
181
  PICK_MODE_CONTEXT sb64_context;
182
  int partition_cost[PARTITION_CONTEXTS][PARTITION_TYPES];
Adrian Grange's avatar
Adrian Grange committed
183

184
185
186
187
  BLOCK_SIZE b_partitioning[4][4][4];
  BLOCK_SIZE mb_partitioning[4][4];
  BLOCK_SIZE sb_partitioning[4];
  BLOCK_SIZE sb64_partitioning;
188

189
  void (*fwd_txm4x4)(const int16_t *input, int16_t *output, int stride);
190
};
John Koleszar's avatar
John Koleszar committed
191

192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
// TODO(jingning): the variables used here are little complicated. need further
// refactoring on organizing the temporary buffers, when recursive
// partition down to 4x4 block size is enabled.
static PICK_MODE_CONTEXT *get_block_context(MACROBLOCK *x, BLOCK_SIZE bsize) {
  MACROBLOCKD *const xd = &x->e_mbd;

  switch (bsize) {
    case BLOCK_64X64:
      return &x->sb64_context;
    case BLOCK_64X32:
      return &x->sb64x32_context[xd->sb_index];
    case BLOCK_32X64:
      return &x->sb32x64_context[xd->sb_index];
    case BLOCK_32X32:
      return &x->sb32_context[xd->sb_index];
    case BLOCK_32X16:
      return &x->sb32x16_context[xd->sb_index][xd->mb_index];
    case BLOCK_16X32:
      return &x->sb16x32_context[xd->sb_index][xd->mb_index];
    case BLOCK_16X16:
      return &x->mb_context[xd->sb_index][xd->mb_index];
    case BLOCK_16X8:
      return &x->sb16x8_context[xd->sb_index][xd->mb_index][xd->b_index];
    case BLOCK_8X16:
      return &x->sb8x16_context[xd->sb_index][xd->mb_index][xd->b_index];
    case BLOCK_8X8:
      return &x->sb8x8_context[xd->sb_index][xd->mb_index][xd->b_index];
    case BLOCK_8X4:
      return &x->sb8x4_context[xd->sb_index][xd->mb_index][xd->b_index];
    case BLOCK_4X8:
      return &x->sb4x8_context[xd->sb_index][xd->mb_index][xd->b_index];
    case BLOCK_4X4:
      return &x->ab4x4_context[xd->sb_index][xd->mb_index][xd->b_index];
    default:
      assert(0);
      return NULL;
  }
}

231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
struct rdcost_block_args {
  MACROBLOCK *x;
  ENTROPY_CONTEXT t_above[16];
  ENTROPY_CONTEXT t_left[16];
  TX_SIZE tx_size;
  int bw;
  int bh;
  int rate;
  int64_t dist;
  int64_t sse;
  int this_rate;
  int64_t this_dist;
  int64_t this_sse;
  int64_t this_rd;
  int64_t best_rd;
  int skip;
  const int16_t *scan, *nb;
};

250
#endif  // VP9_ENCODER_VP9_BLOCK_H_