vp9_block.h 6.68 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
30
  uint8_t *zcoeff_blk;
  int num_4x4_blk;
31
  int skip;
John Koleszar's avatar
John Koleszar committed
32
33
  int_mv best_ref_mv;
  int_mv second_best_ref_mv;
Paul Wilkins's avatar
Paul Wilkins committed
34
  int_mv ref_mvs[MAX_REF_FRAMES][MAX_MV_REF_CANDIDATES];
John Koleszar's avatar
John Koleszar committed
35
36
  int rate;
  int distortion;
37
  int64_t intra_error;
John Koleszar's avatar
John Koleszar committed
38
39
40
  int best_mode_index;
  int rddiv;
  int rdmult;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
41
42
43
  int hybrid_pred_diff;
  int comp_pred_diff;
  int single_pred_diff;
44
  int64_t tx_rd_diff[TX_MODES];
45
  int64_t best_filter_diff[SWITCHABLE_FILTER_CONTEXTS];
Jim Bankoski's avatar
Jim Bankoski committed
46

Jingning Han's avatar
Jingning Han committed
47
48
49
50
  // 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
51
52
53
54
55
  // 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
56
57
} PICK_MODE_CONTEXT;

58
struct macroblock_plane {
59
60
  DECLARE_ALIGNED(16, int16_t, src_diff[64 * 64]);
  DECLARE_ALIGNED(16, int16_t, coeff[64 * 64]);
John Koleszar's avatar
John Koleszar committed
61
  struct buf_2d src;
62
63
64

  // Quantizer setings
  int16_t *quant;
65
  int16_t *quant_shift;
66
67
68
69
70
  int16_t *zbin;
  int16_t *round;

  // Zbin Over Quant value
  int16_t zbin_extra;
71
72
};

73
74
75
76
77
/* 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];

78
79
typedef struct macroblock MACROBLOCK;
struct macroblock {
80
  struct macroblock_plane plane[MAX_MB_PLANE];
John Koleszar's avatar
John Koleszar committed
81
82

  MACROBLOCKD e_mbd;
John Koleszar's avatar
John Koleszar committed
83
  int skip_block;
John Koleszar's avatar
John Koleszar committed
84
85
86
87
88
89
90
91
92
93

  search_site *ss;
  int ss_count;
  int searches_per_step;

  int errorperbit;
  int sadperbit16;
  int sadperbit4;
  int rddiv;
  int rdmult;
94
  unsigned int mb_energy;
John Koleszar's avatar
John Koleszar committed
95
96
97
98
  unsigned int *mb_activity_ptr;
  int *mb_norm_activity_ptr;
  signed int act_zbin_adj;

99
  int mv_best_ref_index[MAX_REF_FRAMES];
100
  unsigned int max_mv_context[MAX_REF_FRAMES];
101
  unsigned int source_variance;
102

103
104
105
106
107
  int nmvjointcost[MV_JOINTS];
  int nmvcosts[2][MV_VALS];
  int *nmvcost[2];
  int nmvcosts_hp[2][MV_VALS];
  int *nmvcost_hp[2];
108
  int **mvcost;
109
110
111
112
113
114

  int nmvjointsadcost[MV_JOINTS];
  int nmvsadcosts[2][MV_VALS];
  int *nmvsadcost[2];
  int nmvsadcosts_hp[2][MV_VALS];
  int *nmvsadcost_hp[2];
115
  int **mvsadcost;
116

117
  int mbmode_cost[MB_MODE_COUNT];
118
  unsigned inter_mode_cost[INTER_MODE_CONTEXTS][INTER_MODES];
John Koleszar's avatar
John Koleszar committed
119
  int intra_uv_mode_cost[2][MB_MODE_COUNT];
120
  int y_mode_costs[INTRA_MODES][INTRA_MODES][INTRA_MODES];
121
  int switchable_interp_costs[SWITCHABLE_FILTER_CONTEXTS][SWITCHABLE_FILTERS];
John Koleszar's avatar
John Koleszar committed
122

123
124
  // These define limits to motion vector components to prevent them
  // from extending outside the UMV borders
John Koleszar's avatar
John Koleszar committed
125
126
127
128
  int mv_col_min;
  int mv_col_max;
  int mv_row_min;
  int mv_row_max;
John Koleszar's avatar
John Koleszar committed
129

130
  uint8_t zcoeff_blk[TX_SIZES][256];
John Koleszar's avatar
John Koleszar committed
131
  int skip;
John Koleszar's avatar
John Koleszar committed
132

John Koleszar's avatar
John Koleszar committed
133
  int encode_breakout;
John Koleszar's avatar
John Koleszar committed
134

John Koleszar's avatar
John Koleszar committed
135
  unsigned char *active_ptr;
John Koleszar's avatar
John Koleszar committed
136

137
  // note that token_costs is the cost when eob node is skipped
138
  vp9_coeff_cost token_costs[TX_SIZES];
139
  DECLARE_ALIGNED(16, uint8_t, token_cache[1024]);
140

John Koleszar's avatar
John Koleszar committed
141
  int optimize;
John Koleszar's avatar
John Koleszar committed
142

143
  // indicate if it is in the rd search loop or encoding process
144
  int use_lp32x32fdct;
145
  int skip_encode;
146

147
148
  // Used to store sub partition's choices.
  int fast_ms;
Jingning Han's avatar
Jingning Han committed
149
  int_mv pred_mv[MAX_REF_FRAMES];
150
151
  int subblock_ref;

152
153
  // TODO(jingning): Need to refactor the structure arrays that buffers the
  // coding mode decisions of each partition type.
154
155
156
157
  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];
158
159
  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
160
  PICK_MODE_CONTEXT mb_context[4][4];
161
162
  PICK_MODE_CONTEXT sb32x16_context[4][2];
  PICK_MODE_CONTEXT sb16x32_context[4][2];
John Koleszar's avatar
John Koleszar committed
163
  // when 4 MBs share coding parameters:
Ronald S. Bultje's avatar
Ronald S. Bultje committed
164
  PICK_MODE_CONTEXT sb32_context[4];
165
166
  PICK_MODE_CONTEXT sb32x64_context[2];
  PICK_MODE_CONTEXT sb64x32_context[2];
Ronald S. Bultje's avatar
Ronald S. Bultje committed
167
  PICK_MODE_CONTEXT sb64_context;
168
  int partition_cost[PARTITION_CONTEXTS][PARTITION_TYPES];
Adrian Grange's avatar
Adrian Grange committed
169

170
171
172
173
  BLOCK_SIZE b_partitioning[4][4][4];
  BLOCK_SIZE mb_partitioning[4][4];
  BLOCK_SIZE sb_partitioning[4];
  BLOCK_SIZE sb64_partitioning;
174

175
  void (*fwd_txm4x4)(const int16_t *input, int16_t *output, int stride);
176
};
John Koleszar's avatar
John Koleszar committed
177

178
179
180
181
182
183
184
185
186
187
188
189
190
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
// 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;
  }
}

217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
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;
};

236
#endif  // VP9_ENCODER_VP9_BLOCK_H_