|
1 /* |
|
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
|
3 * |
|
4 * Use of this source code is governed by a BSD-style license |
|
5 * that can be found in the LICENSE file in the root of the source |
|
6 * 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. |
|
9 */ |
|
10 |
|
11 |
|
12 #ifndef __INC_BLOCKD_H |
|
13 #define __INC_BLOCKD_H |
|
14 |
|
15 void vpx_log(const char *format, ...); |
|
16 |
|
17 #include "vpx_config.h" |
|
18 #include "vpx_scale/yv12config.h" |
|
19 #include "mv.h" |
|
20 #include "treecoder.h" |
|
21 #include "vpx_ports/mem.h" |
|
22 |
|
23 /*#define DCPRED 1*/ |
|
24 #define DCPREDSIMTHRESH 0 |
|
25 #define DCPREDCNTTHRESH 3 |
|
26 |
|
27 #define MB_FEATURE_TREE_PROBS 3 |
|
28 #define MAX_MB_SEGMENTS 4 |
|
29 |
|
30 #define MAX_REF_LF_DELTAS 4 |
|
31 #define MAX_MODE_LF_DELTAS 4 |
|
32 |
|
33 /* Segment Feature Masks */ |
|
34 #define SEGMENT_DELTADATA 0 |
|
35 #define SEGMENT_ABSDATA 1 |
|
36 |
|
37 typedef struct |
|
38 { |
|
39 int r, c; |
|
40 } POS; |
|
41 |
|
42 #define PLANE_TYPE_Y_NO_DC 0 |
|
43 #define PLANE_TYPE_Y2 1 |
|
44 #define PLANE_TYPE_UV 2 |
|
45 #define PLANE_TYPE_Y_WITH_DC 3 |
|
46 |
|
47 |
|
48 typedef char ENTROPY_CONTEXT; |
|
49 typedef struct |
|
50 { |
|
51 ENTROPY_CONTEXT y1[4]; |
|
52 ENTROPY_CONTEXT u[2]; |
|
53 ENTROPY_CONTEXT v[2]; |
|
54 ENTROPY_CONTEXT y2; |
|
55 } ENTROPY_CONTEXT_PLANES; |
|
56 |
|
57 extern const unsigned char vp8_block2left[25]; |
|
58 extern const unsigned char vp8_block2above[25]; |
|
59 |
|
60 #define VP8_COMBINEENTROPYCONTEXTS( Dest, A, B) \ |
|
61 Dest = (A)+(B); |
|
62 |
|
63 |
|
64 typedef enum |
|
65 { |
|
66 KEY_FRAME = 0, |
|
67 INTER_FRAME = 1 |
|
68 } FRAME_TYPE; |
|
69 |
|
70 typedef enum |
|
71 { |
|
72 DC_PRED, /* average of above and left pixels */ |
|
73 V_PRED, /* vertical prediction */ |
|
74 H_PRED, /* horizontal prediction */ |
|
75 TM_PRED, /* Truemotion prediction */ |
|
76 B_PRED, /* block based prediction, each block has its own prediction mode */ |
|
77 |
|
78 NEARESTMV, |
|
79 NEARMV, |
|
80 ZEROMV, |
|
81 NEWMV, |
|
82 SPLITMV, |
|
83 |
|
84 MB_MODE_COUNT |
|
85 } MB_PREDICTION_MODE; |
|
86 |
|
87 /* Macroblock level features */ |
|
88 typedef enum |
|
89 { |
|
90 MB_LVL_ALT_Q = 0, /* Use alternate Quantizer .... */ |
|
91 MB_LVL_ALT_LF = 1, /* Use alternate loop filter value... */ |
|
92 MB_LVL_MAX = 2 /* Number of MB level features supported */ |
|
93 |
|
94 } MB_LVL_FEATURES; |
|
95 |
|
96 /* Segment Feature Masks */ |
|
97 #define SEGMENT_ALTQ 0x01 |
|
98 #define SEGMENT_ALT_LF 0x02 |
|
99 |
|
100 #define VP8_YMODES (B_PRED + 1) |
|
101 #define VP8_UV_MODES (TM_PRED + 1) |
|
102 |
|
103 #define VP8_MVREFS (1 + SPLITMV - NEARESTMV) |
|
104 |
|
105 typedef enum |
|
106 { |
|
107 B_DC_PRED, /* average of above and left pixels */ |
|
108 B_TM_PRED, |
|
109 |
|
110 B_VE_PRED, /* vertical prediction */ |
|
111 B_HE_PRED, /* horizontal prediction */ |
|
112 |
|
113 B_LD_PRED, |
|
114 B_RD_PRED, |
|
115 |
|
116 B_VR_PRED, |
|
117 B_VL_PRED, |
|
118 B_HD_PRED, |
|
119 B_HU_PRED, |
|
120 |
|
121 LEFT4X4, |
|
122 ABOVE4X4, |
|
123 ZERO4X4, |
|
124 NEW4X4, |
|
125 |
|
126 B_MODE_COUNT |
|
127 } B_PREDICTION_MODE; |
|
128 |
|
129 #define VP8_BINTRAMODES (B_HU_PRED + 1) /* 10 */ |
|
130 #define VP8_SUBMVREFS (1 + NEW4X4 - LEFT4X4) |
|
131 |
|
132 /* For keyframes, intra block modes are predicted by the (already decoded) |
|
133 modes for the Y blocks to the left and above us; for interframes, there |
|
134 is a single probability table. */ |
|
135 |
|
136 union b_mode_info |
|
137 { |
|
138 B_PREDICTION_MODE as_mode; |
|
139 int_mv mv; |
|
140 }; |
|
141 |
|
142 typedef enum |
|
143 { |
|
144 INTRA_FRAME = 0, |
|
145 LAST_FRAME = 1, |
|
146 GOLDEN_FRAME = 2, |
|
147 ALTREF_FRAME = 3, |
|
148 MAX_REF_FRAMES = 4 |
|
149 } MV_REFERENCE_FRAME; |
|
150 |
|
151 typedef struct |
|
152 { |
|
153 uint8_t mode, uv_mode; |
|
154 uint8_t ref_frame; |
|
155 uint8_t is_4x4; |
|
156 int_mv mv; |
|
157 |
|
158 uint8_t partitioning; |
|
159 uint8_t mb_skip_coeff; /* does this mb has coefficients at all, 1=no coefficients, 0=need decode tokens */ |
|
160 uint8_t need_to_clamp_mvs; |
|
161 uint8_t segment_id; /* Which set of segmentation parameters should be used for this MB */ |
|
162 } MB_MODE_INFO; |
|
163 |
|
164 typedef struct modeinfo |
|
165 { |
|
166 MB_MODE_INFO mbmi; |
|
167 union b_mode_info bmi[16]; |
|
168 } MODE_INFO; |
|
169 |
|
170 #if CONFIG_MULTI_RES_ENCODING |
|
171 /* The mb-level information needed to be stored for higher-resolution encoder */ |
|
172 typedef struct |
|
173 { |
|
174 MB_PREDICTION_MODE mode; |
|
175 MV_REFERENCE_FRAME ref_frame; |
|
176 int_mv mv; |
|
177 int dissim; /* dissimilarity level of the macroblock */ |
|
178 } LOWER_RES_MB_INFO; |
|
179 |
|
180 /* The frame-level information needed to be stored for higher-resolution |
|
181 * encoder */ |
|
182 typedef struct |
|
183 { |
|
184 FRAME_TYPE frame_type; |
|
185 int is_frame_dropped; |
|
186 /* The frame number of each reference frames */ |
|
187 unsigned int low_res_ref_frames[MAX_REF_FRAMES]; |
|
188 LOWER_RES_MB_INFO *mb_info; |
|
189 } LOWER_RES_FRAME_INFO; |
|
190 #endif |
|
191 |
|
192 typedef struct blockd |
|
193 { |
|
194 short *qcoeff; |
|
195 short *dqcoeff; |
|
196 unsigned char *predictor; |
|
197 short *dequant; |
|
198 |
|
199 int offset; |
|
200 char *eob; |
|
201 |
|
202 union b_mode_info bmi; |
|
203 } BLOCKD; |
|
204 |
|
205 typedef void (*vp8_subpix_fn_t)(unsigned char *src, int src_pitch, int xofst, int yofst, unsigned char *dst, int dst_pitch); |
|
206 |
|
207 typedef struct macroblockd |
|
208 { |
|
209 DECLARE_ALIGNED(16, unsigned char, predictor[384]); |
|
210 DECLARE_ALIGNED(16, short, qcoeff[400]); |
|
211 DECLARE_ALIGNED(16, short, dqcoeff[400]); |
|
212 DECLARE_ALIGNED(16, char, eobs[25]); |
|
213 |
|
214 DECLARE_ALIGNED(16, short, dequant_y1[16]); |
|
215 DECLARE_ALIGNED(16, short, dequant_y1_dc[16]); |
|
216 DECLARE_ALIGNED(16, short, dequant_y2[16]); |
|
217 DECLARE_ALIGNED(16, short, dequant_uv[16]); |
|
218 |
|
219 /* 16 Y blocks, 4 U, 4 V, 1 DC 2nd order block, each with 16 entries. */ |
|
220 BLOCKD block[25]; |
|
221 int fullpixel_mask; |
|
222 |
|
223 YV12_BUFFER_CONFIG pre; /* Filtered copy of previous frame reconstruction */ |
|
224 YV12_BUFFER_CONFIG dst; |
|
225 |
|
226 MODE_INFO *mode_info_context; |
|
227 int mode_info_stride; |
|
228 |
|
229 FRAME_TYPE frame_type; |
|
230 |
|
231 int up_available; |
|
232 int left_available; |
|
233 |
|
234 unsigned char *recon_above[3]; |
|
235 unsigned char *recon_left[3]; |
|
236 int recon_left_stride[2]; |
|
237 |
|
238 /* Y,U,V,Y2 */ |
|
239 ENTROPY_CONTEXT_PLANES *above_context; |
|
240 ENTROPY_CONTEXT_PLANES *left_context; |
|
241 |
|
242 /* 0 indicates segmentation at MB level is not enabled. Otherwise the individual bits indicate which features are active. */ |
|
243 unsigned char segmentation_enabled; |
|
244 |
|
245 /* 0 (do not update) 1 (update) the macroblock segmentation map. */ |
|
246 unsigned char update_mb_segmentation_map; |
|
247 |
|
248 /* 0 (do not update) 1 (update) the macroblock segmentation feature data. */ |
|
249 unsigned char update_mb_segmentation_data; |
|
250 |
|
251 /* 0 (do not update) 1 (update) the macroblock segmentation feature data. */ |
|
252 unsigned char mb_segement_abs_delta; |
|
253 |
|
254 /* Per frame flags that define which MB level features (such as quantizer or loop filter level) */ |
|
255 /* are enabled and when enabled the proabilities used to decode the per MB flags in MB_MODE_INFO */ |
|
256 vp8_prob mb_segment_tree_probs[MB_FEATURE_TREE_PROBS]; /* Probability Tree used to code Segment number */ |
|
257 |
|
258 signed char segment_feature_data[MB_LVL_MAX][MAX_MB_SEGMENTS]; /* Segment parameters */ |
|
259 |
|
260 /* mode_based Loop filter adjustment */ |
|
261 unsigned char mode_ref_lf_delta_enabled; |
|
262 unsigned char mode_ref_lf_delta_update; |
|
263 |
|
264 /* Delta values have the range +/- MAX_LOOP_FILTER */ |
|
265 signed char last_ref_lf_deltas[MAX_REF_LF_DELTAS]; /* 0 = Intra, Last, GF, ARF */ |
|
266 signed char ref_lf_deltas[MAX_REF_LF_DELTAS]; /* 0 = Intra, Last, GF, ARF */ |
|
267 signed char last_mode_lf_deltas[MAX_MODE_LF_DELTAS]; /* 0 = BPRED, ZERO_MV, MV, SPLIT */ |
|
268 signed char mode_lf_deltas[MAX_MODE_LF_DELTAS]; /* 0 = BPRED, ZERO_MV, MV, SPLIT */ |
|
269 |
|
270 /* Distance of MB away from frame edges */ |
|
271 int mb_to_left_edge; |
|
272 int mb_to_right_edge; |
|
273 int mb_to_top_edge; |
|
274 int mb_to_bottom_edge; |
|
275 |
|
276 |
|
277 |
|
278 vp8_subpix_fn_t subpixel_predict; |
|
279 vp8_subpix_fn_t subpixel_predict8x4; |
|
280 vp8_subpix_fn_t subpixel_predict8x8; |
|
281 vp8_subpix_fn_t subpixel_predict16x16; |
|
282 |
|
283 void *current_bc; |
|
284 |
|
285 int corrupted; |
|
286 |
|
287 #if ARCH_X86 || ARCH_X86_64 |
|
288 /* This is an intermediate buffer currently used in sub-pixel motion search |
|
289 * to keep a copy of the reference area. This buffer can be used for other |
|
290 * purpose. |
|
291 */ |
|
292 DECLARE_ALIGNED(32, unsigned char, y_buf[22*32]); |
|
293 #endif |
|
294 } MACROBLOCKD; |
|
295 |
|
296 |
|
297 extern void vp8_build_block_doffsets(MACROBLOCKD *x); |
|
298 extern void vp8_setup_block_dptrs(MACROBLOCKD *x); |
|
299 |
|
300 #endif /* __INC_BLOCKD_H */ |