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

15
16
17
#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
18
#include "vpx_ports/mem.h"
19
#include "vp9/common/vp9_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;

Jim Bankoski's avatar
Jim Bankoski committed
27
typedef struct block {
John Koleszar's avatar
John Koleszar committed
28
29
30
31
32
33
34
35
36
37
  // 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;
Daniel Kang's avatar
Daniel Kang committed
38
  short *zbin_16x16;
John Koleszar's avatar
John Koleszar committed
39
40
  short *zrun_zbin_boost;
  short *zrun_zbin_boost_8x8;
Daniel Kang's avatar
Daniel Kang committed
41
  short *zrun_zbin_boost_16x16;
John Koleszar's avatar
John Koleszar committed
42
43
44
45
46
47
48
49
50
51
52
53
  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;
Daniel Kang's avatar
Daniel Kang committed
54
  int eob_max_offset_16x16;
John Koleszar's avatar
John Koleszar committed
55
56
} BLOCK;

John Koleszar's avatar
John Koleszar committed
57
58
59
60
61
62
63
typedef struct {
  int count;
  struct {
    B_PREDICTION_MODE mode;
    int_mv mv;
    int_mv second_mv;
  } bmi[16];
64
65
} PARTITION_INFO;

Adrian Grange's avatar
Adrian Grange committed
66
67
// 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
68
69
70
71
72
typedef struct {
  MODE_INFO mic;
  PARTITION_INFO partition_info;
  int_mv best_ref_mv;
  int_mv second_best_ref_mv;
73
  int_mv ref_mvs[MAX_REF_FRAMES][MAX_MV_REFS];
John Koleszar's avatar
John Koleszar committed
74
75
  int rate;
  int distortion;
76
  int64_t intra_error;
John Koleszar's avatar
John Koleszar committed
77
78
79
  int best_mode_index;
  int rddiv;
  int rdmult;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
80
81
82
  int hybrid_pred_diff;
  int comp_pred_diff;
  int single_pred_diff;
83
  int64_t txfm_rd_diff[NB_TXFM_MODES];
Adrian Grange's avatar
Adrian Grange committed
84
85
} PICK_MODE_CONTEXT;

Jim Bankoski's avatar
Jim Bankoski committed
86
typedef struct macroblock {
John Koleszar's avatar
John Koleszar committed
87
88
  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
89
#if !CONFIG_SUPERBLOCKS
John Koleszar's avatar
John Koleszar committed
90
91
92
  DECLARE_ALIGNED(16, unsigned char, thismb[256]);    // 16x16 Y

  unsigned char *thismb_ptr;
93
#endif
John Koleszar's avatar
John Koleszar committed
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
  // 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;

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

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

John Koleszar's avatar
John Koleszar committed
132
133
  int mbmode_cost[2][MB_MODE_COUNT];
  int intra_uv_mode_cost[2][MB_MODE_COUNT];
134
  int bmode_costs[VP9_KF_BINTRAMODES][VP9_KF_BINTRAMODES][VP9_KF_BINTRAMODES];
John Koleszar's avatar
John Koleszar committed
135
136
  int i8x8_mode_costs[MB_MODE_COUNT];
  int inter_bmode_costs[B_MODE_COUNT];
137
138
  int switchable_interp_costs[VP9_SWITCHABLE_FILTERS + 1]
                             [VP9_SWITCHABLE_FILTERS];
John Koleszar's avatar
John Koleszar committed
139

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

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

John Koleszar's avatar
John Koleszar committed
149
  int encode_breakout;
John Koleszar's avatar
John Koleszar committed
150

John Koleszar's avatar
John Koleszar committed
151
152
  // char * gf_active_ptr;
  signed char *gf_active_ptr;
153

John Koleszar's avatar
John Koleszar committed
154
  unsigned char *active_ptr;
John Koleszar's avatar
John Koleszar committed
155

156
  unsigned int token_costs[TX_SIZE_MAX][BLOCK_TYPES][COEF_BANDS]
Daniel Kang's avatar
Daniel Kang committed
157
    [PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];
158
159
  unsigned int hybrid_token_costs[TX_SIZE_MAX][BLOCK_TYPES][COEF_BANDS]
    [PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];
160

John Koleszar's avatar
John Koleszar committed
161
  int optimize;
John Koleszar's avatar
John Koleszar committed
162

John Koleszar's avatar
John Koleszar committed
163
164
165
  // 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
166
#if CONFIG_SUPERBLOCKS
John Koleszar's avatar
John Koleszar committed
167
168
  // when 4 MBs share coding parameters:
  PICK_MODE_CONTEXT sb_context[4];
Adrian Grange's avatar
Adrian Grange committed
169
170
#endif

171
172
  void (*vp9_short_fdct4x4)(short *input, short *output, int pitch);
  void (*vp9_short_fdct8x4)(short *input, short *output, int pitch);
John Koleszar's avatar
John Koleszar committed
173
  void (*short_walsh4x4)(short *input, short *output, int pitch);
174
175
  void (*quantize_b_4x4)(BLOCK *b, BLOCKD *d);
  void (*quantize_b_4x4_pair)(BLOCK *b1, BLOCK *b2, BLOCKD *d0, BLOCKD *d1);
176
177
  void (*vp9_short_fdct8x8)(short *input, short *output, int pitch);
  void (*vp9_short_fdct16x16)(short *input, short *output, int pitch);
John Koleszar's avatar
John Koleszar committed
178
  void (*short_fhaar2x2)(short *input, short *output, int pitch);
Daniel Kang's avatar
Daniel Kang committed
179
  void (*quantize_b_16x16)(BLOCK *b, BLOCKD *d);
John Koleszar's avatar
John Koleszar committed
180
181
  void (*quantize_b_8x8)(BLOCK *b, BLOCKD *d);
  void (*quantize_b_2x2)(BLOCK *b, BLOCKD *d);
John Koleszar's avatar
John Koleszar committed
182
183
184
185
186

} MACROBLOCK;


#endif