OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef VP9_COMMON_VP9_ONYXC_INT_H_ | 11 #ifndef VP9_COMMON_VP9_ONYXC_INT_H_ |
12 #define VP9_COMMON_VP9_ONYXC_INT_H_ | 12 #define VP9_COMMON_VP9_ONYXC_INT_H_ |
13 | 13 |
14 #include "./vpx_config.h" | 14 #include "./vpx_config.h" |
15 #include "vpx/internal/vpx_codec_internal.h" | 15 #include "vpx/internal/vpx_codec_internal.h" |
16 #include "./vp9_rtcd.h" | 16 #include "./vp9_rtcd.h" |
17 #include "vp9/common/vp9_loopfilter.h" | 17 #include "vp9/common/vp9_loopfilter.h" |
18 #include "vp9/common/vp9_entropymv.h" | 18 #include "vp9/common/vp9_entropymv.h" |
19 #include "vp9/common/vp9_entropy.h" | 19 #include "vp9/common/vp9_entropy.h" |
20 #include "vp9/common/vp9_entropymode.h" | 20 #include "vp9/common/vp9_entropymode.h" |
| 21 #include "vp9/common/vp9_frame_buffers.h" |
21 #include "vp9/common/vp9_quant_common.h" | 22 #include "vp9/common/vp9_quant_common.h" |
22 #include "vp9/common/vp9_tile_common.h" | 23 #include "vp9/common/vp9_tile_common.h" |
23 | 24 |
24 #if CONFIG_VP9_POSTPROC | 25 #if CONFIG_VP9_POSTPROC |
25 #include "vp9/common/vp9_postproc.h" | 26 #include "vp9/common/vp9_postproc.h" |
26 #endif | 27 #endif |
27 | 28 |
28 #ifdef __cplusplus | 29 #ifdef __cplusplus |
29 extern "C" { | 30 extern "C" { |
30 #endif | 31 #endif |
(...skipping 22 matching lines...) Expand all Loading... |
53 vp9_prob partition_prob[PARTITION_CONTEXTS][PARTITION_TYPES - 1]; | 54 vp9_prob partition_prob[PARTITION_CONTEXTS][PARTITION_TYPES - 1]; |
54 vp9_coeff_probs_model coef_probs[TX_SIZES][PLANE_TYPES]; | 55 vp9_coeff_probs_model coef_probs[TX_SIZES][PLANE_TYPES]; |
55 vp9_prob switchable_interp_prob[SWITCHABLE_FILTER_CONTEXTS] | 56 vp9_prob switchable_interp_prob[SWITCHABLE_FILTER_CONTEXTS] |
56 [SWITCHABLE_FILTERS - 1]; | 57 [SWITCHABLE_FILTERS - 1]; |
57 vp9_prob inter_mode_probs[INTER_MODE_CONTEXTS][INTER_MODES - 1]; | 58 vp9_prob inter_mode_probs[INTER_MODE_CONTEXTS][INTER_MODES - 1]; |
58 vp9_prob intra_inter_prob[INTRA_INTER_CONTEXTS]; | 59 vp9_prob intra_inter_prob[INTRA_INTER_CONTEXTS]; |
59 vp9_prob comp_inter_prob[COMP_INTER_CONTEXTS]; | 60 vp9_prob comp_inter_prob[COMP_INTER_CONTEXTS]; |
60 vp9_prob single_ref_prob[REF_CONTEXTS][2]; | 61 vp9_prob single_ref_prob[REF_CONTEXTS][2]; |
61 vp9_prob comp_ref_prob[REF_CONTEXTS]; | 62 vp9_prob comp_ref_prob[REF_CONTEXTS]; |
62 struct tx_probs tx_probs; | 63 struct tx_probs tx_probs; |
63 vp9_prob mbskip_probs[MBSKIP_CONTEXTS]; | 64 vp9_prob skip_probs[SKIP_CONTEXTS]; |
64 nmv_context nmvc; | 65 nmv_context nmvc; |
65 } FRAME_CONTEXT; | 66 } FRAME_CONTEXT; |
66 | 67 |
67 typedef struct { | 68 typedef struct { |
68 unsigned int y_mode[BLOCK_SIZE_GROUPS][INTRA_MODES]; | 69 unsigned int y_mode[BLOCK_SIZE_GROUPS][INTRA_MODES]; |
69 unsigned int uv_mode[INTRA_MODES][INTRA_MODES]; | 70 unsigned int uv_mode[INTRA_MODES][INTRA_MODES]; |
70 unsigned int partition[PARTITION_CONTEXTS][PARTITION_TYPES]; | 71 unsigned int partition[PARTITION_CONTEXTS][PARTITION_TYPES]; |
71 vp9_coeff_count_model coef[TX_SIZES][PLANE_TYPES]; | 72 vp9_coeff_count_model coef[TX_SIZES][PLANE_TYPES]; |
72 unsigned int eob_branch[TX_SIZES][PLANE_TYPES][REF_TYPES] | 73 unsigned int eob_branch[TX_SIZES][PLANE_TYPES][REF_TYPES] |
73 [COEF_BANDS][COEFF_CONTEXTS]; | 74 [COEF_BANDS][COEFF_CONTEXTS]; |
74 unsigned int switchable_interp[SWITCHABLE_FILTER_CONTEXTS] | 75 unsigned int switchable_interp[SWITCHABLE_FILTER_CONTEXTS] |
75 [SWITCHABLE_FILTERS]; | 76 [SWITCHABLE_FILTERS]; |
76 unsigned int inter_mode[INTER_MODE_CONTEXTS][INTER_MODES]; | 77 unsigned int inter_mode[INTER_MODE_CONTEXTS][INTER_MODES]; |
77 unsigned int intra_inter[INTRA_INTER_CONTEXTS][2]; | 78 unsigned int intra_inter[INTRA_INTER_CONTEXTS][2]; |
78 unsigned int comp_inter[COMP_INTER_CONTEXTS][2]; | 79 unsigned int comp_inter[COMP_INTER_CONTEXTS][2]; |
79 unsigned int single_ref[REF_CONTEXTS][2][2]; | 80 unsigned int single_ref[REF_CONTEXTS][2][2]; |
80 unsigned int comp_ref[REF_CONTEXTS][2]; | 81 unsigned int comp_ref[REF_CONTEXTS][2]; |
81 struct tx_counts tx; | 82 struct tx_counts tx; |
82 unsigned int mbskip[MBSKIP_CONTEXTS][2]; | 83 unsigned int skip[SKIP_CONTEXTS][2]; |
83 nmv_context_counts mv; | 84 nmv_context_counts mv; |
84 } FRAME_COUNTS; | 85 } FRAME_COUNTS; |
85 | 86 |
86 | 87 |
87 typedef enum { | 88 typedef enum { |
88 SINGLE_REFERENCE = 0, | 89 SINGLE_REFERENCE = 0, |
89 COMPOUND_REFERENCE = 1, | 90 COMPOUND_REFERENCE = 1, |
90 REFERENCE_MODE_SELECT = 2, | 91 REFERENCE_MODE_SELECT = 2, |
91 REFERENCE_MODES = 3, | 92 REFERENCE_MODES = 3, |
92 } REFERENCE_MODE; | 93 } REFERENCE_MODE; |
93 | 94 |
| 95 |
| 96 typedef struct { |
| 97 int ref_count; |
| 98 vpx_codec_frame_buffer_t raw_frame_buffer; |
| 99 YV12_BUFFER_CONFIG buf; |
| 100 } RefCntBuffer; |
| 101 |
94 typedef struct VP9Common { | 102 typedef struct VP9Common { |
95 struct vpx_internal_error_info error; | 103 struct vpx_internal_error_info error; |
96 | 104 |
97 DECLARE_ALIGNED(16, int16_t, y_dequant[QINDEX_RANGE][8]); | 105 DECLARE_ALIGNED(16, int16_t, y_dequant[QINDEX_RANGE][8]); |
98 DECLARE_ALIGNED(16, int16_t, uv_dequant[QINDEX_RANGE][8]); | 106 DECLARE_ALIGNED(16, int16_t, uv_dequant[QINDEX_RANGE][8]); |
99 #if CONFIG_ALPHA | 107 #if CONFIG_ALPHA |
100 DECLARE_ALIGNED(16, int16_t, a_dequant[QINDEX_RANGE][8]); | 108 DECLARE_ALIGNED(16, int16_t, a_dequant[QINDEX_RANGE][8]); |
101 #endif | 109 #endif |
102 | 110 |
103 COLOR_SPACE color_space; | 111 COLOR_SPACE color_space; |
104 | 112 |
105 int width; | 113 int width; |
106 int height; | 114 int height; |
107 int display_width; | 115 int display_width; |
108 int display_height; | 116 int display_height; |
109 int last_width; | 117 int last_width; |
110 int last_height; | 118 int last_height; |
111 | 119 |
112 // TODO(jkoleszar): this implies chroma ss right now, but could vary per | 120 // TODO(jkoleszar): this implies chroma ss right now, but could vary per |
113 // plane. Revisit as part of the future change to YV12_BUFFER_CONFIG to | 121 // plane. Revisit as part of the future change to YV12_BUFFER_CONFIG to |
114 // support additional planes. | 122 // support additional planes. |
115 int subsampling_x; | 123 int subsampling_x; |
116 int subsampling_y; | 124 int subsampling_y; |
117 | 125 |
118 YV12_BUFFER_CONFIG *frame_to_show; | 126 YV12_BUFFER_CONFIG *frame_to_show; |
119 | 127 |
120 YV12_BUFFER_CONFIG yv12_fb[FRAME_BUFFERS]; | 128 RefCntBuffer frame_bufs[FRAME_BUFFERS]; |
121 int fb_idx_ref_cnt[FRAME_BUFFERS]; /* reference counts */ | 129 |
122 int ref_frame_map[REF_FRAMES]; /* maps fb_idx to reference slot */ | 130 int ref_frame_map[REF_FRAMES]; /* maps fb_idx to reference slot */ |
123 | 131 |
124 // TODO(jkoleszar): could expand active_ref_idx to 4, with 0 as intra, and | 132 // TODO(jkoleszar): could expand active_ref_idx to 4, with 0 as intra, and |
125 // roll new_fb_idx into it. | 133 // roll new_fb_idx into it. |
126 | 134 |
127 // Each frame can reference REFS_PER_FRAME buffers | 135 // Each frame can reference REFS_PER_FRAME buffers |
128 RefBuffer frame_refs[REFS_PER_FRAME]; | 136 RefBuffer frame_refs[REFS_PER_FRAME]; |
129 | 137 |
130 int new_fb_idx; | 138 int new_fb_idx; |
131 | 139 |
(...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
210 int version; | 218 int version; |
211 | 219 |
212 #if CONFIG_VP9_POSTPROC | 220 #if CONFIG_VP9_POSTPROC |
213 struct postproc_state postproc_state; | 221 struct postproc_state postproc_state; |
214 #endif | 222 #endif |
215 | 223 |
216 int error_resilient_mode; | 224 int error_resilient_mode; |
217 int frame_parallel_decoding_mode; | 225 int frame_parallel_decoding_mode; |
218 | 226 |
219 int log2_tile_cols, log2_tile_rows; | 227 int log2_tile_cols, log2_tile_rows; |
| 228 |
| 229 // Private data associated with the frame buffer callbacks. |
| 230 void *cb_priv; |
| 231 vpx_get_frame_buffer_cb_fn_t get_fb_cb; |
| 232 vpx_release_frame_buffer_cb_fn_t release_fb_cb; |
| 233 |
| 234 // Handles memory for the codec. |
| 235 InternalFrameBufferList int_frame_buffers; |
220 } VP9_COMMON; | 236 } VP9_COMMON; |
221 | 237 |
222 static YV12_BUFFER_CONFIG *get_frame_new_buffer(VP9_COMMON *cm) { | 238 static INLINE YV12_BUFFER_CONFIG *get_frame_new_buffer(VP9_COMMON *cm) { |
223 return &cm->yv12_fb[cm->new_fb_idx]; | 239 return &cm->frame_bufs[cm->new_fb_idx].buf; |
224 } | 240 } |
225 | 241 |
226 static int get_free_fb(VP9_COMMON *cm) { | 242 static INLINE int get_free_fb(VP9_COMMON *cm) { |
227 int i; | 243 int i; |
228 for (i = 0; i < FRAME_BUFFERS; i++) | 244 for (i = 0; i < FRAME_BUFFERS; i++) |
229 if (cm->fb_idx_ref_cnt[i] == 0) | 245 if (cm->frame_bufs[i].ref_count == 0) |
230 break; | 246 break; |
231 | 247 |
232 assert(i < FRAME_BUFFERS); | 248 assert(i < FRAME_BUFFERS); |
233 cm->fb_idx_ref_cnt[i] = 1; | 249 cm->frame_bufs[i].ref_count = 1; |
234 return i; | 250 return i; |
235 } | 251 } |
236 | 252 |
237 static void ref_cnt_fb(int *buf, int *idx, int new_idx) { | 253 static INLINE void ref_cnt_fb(RefCntBuffer *bufs, int *idx, int new_idx) { |
238 const int ref_index = *idx; | 254 const int ref_index = *idx; |
239 | 255 |
240 if (ref_index >= 0 && buf[ref_index] > 0) | 256 if (ref_index >= 0 && bufs[ref_index].ref_count > 0) |
241 buf[ref_index]--; | 257 bufs[ref_index].ref_count--; |
242 | 258 |
243 *idx = new_idx; | 259 *idx = new_idx; |
244 | 260 |
245 buf[new_idx]++; | 261 bufs[new_idx].ref_count++; |
246 } | 262 } |
247 | 263 |
248 static int mi_cols_aligned_to_sb(int n_mis) { | 264 static INLINE int mi_cols_aligned_to_sb(int n_mis) { |
249 return ALIGN_POWER_OF_TWO(n_mis, MI_BLOCK_SIZE_LOG2); | 265 return ALIGN_POWER_OF_TWO(n_mis, MI_BLOCK_SIZE_LOG2); |
250 } | 266 } |
251 | 267 |
252 static INLINE const vp9_prob* get_partition_probs(VP9_COMMON *cm, int ctx) { | 268 static INLINE const vp9_prob* get_partition_probs(VP9_COMMON *cm, int ctx) { |
253 return cm->frame_type == KEY_FRAME ? vp9_kf_partition_probs[ctx] | 269 return cm->frame_type == KEY_FRAME ? vp9_kf_partition_probs[ctx] |
254 : cm->fc.partition_prob[ctx]; | 270 : cm->fc.partition_prob[ctx]; |
255 } | 271 } |
256 | 272 |
257 static INLINE void set_skip_context( | 273 static INLINE void set_skip_context( |
258 MACROBLOCKD *xd, | 274 MACROBLOCKD *xd, |
259 ENTROPY_CONTEXT *above_context[MAX_MB_PLANE], | 275 ENTROPY_CONTEXT *above_context[MAX_MB_PLANE], |
260 ENTROPY_CONTEXT left_context[MAX_MB_PLANE][16], | 276 ENTROPY_CONTEXT left_context[MAX_MB_PLANE][16], |
261 int mi_row, int mi_col) { | 277 int mi_row, int mi_col) { |
262 const int above_idx = mi_col * 2; | 278 const int above_idx = mi_col * 2; |
263 const int left_idx = (mi_row * 2) & 15; | 279 const int left_idx = (mi_row * 2) & 15; |
264 int i; | 280 int i; |
265 for (i = 0; i < MAX_MB_PLANE; i++) { | 281 for (i = 0; i < MAX_MB_PLANE; i++) { |
266 struct macroblockd_plane *const pd = &xd->plane[i]; | 282 struct macroblockd_plane *const pd = &xd->plane[i]; |
267 pd->above_context = above_context[i] + (above_idx >> pd->subsampling_x); | 283 pd->above_context = above_context[i] + (above_idx >> pd->subsampling_x); |
268 pd->left_context = left_context[i] + (left_idx >> pd->subsampling_y); | 284 pd->left_context = left_context[i] + (left_idx >> pd->subsampling_y); |
269 } | 285 } |
270 } | 286 } |
271 | 287 |
272 static void set_mi_row_col(MACROBLOCKD *xd, const TileInfo *const tile, | 288 static INLINE void set_mi_row_col(MACROBLOCKD *xd, const TileInfo *const tile, |
273 int mi_row, int bh, | 289 int mi_row, int bh, |
274 int mi_col, int bw, | 290 int mi_col, int bw, |
275 int mi_rows, int mi_cols) { | 291 int mi_rows, int mi_cols) { |
276 xd->mb_to_top_edge = -((mi_row * MI_SIZE) * 8); | 292 xd->mb_to_top_edge = -((mi_row * MI_SIZE) * 8); |
277 xd->mb_to_bottom_edge = ((mi_rows - bh - mi_row) * MI_SIZE) * 8; | 293 xd->mb_to_bottom_edge = ((mi_rows - bh - mi_row) * MI_SIZE) * 8; |
278 xd->mb_to_left_edge = -((mi_col * MI_SIZE) * 8); | 294 xd->mb_to_left_edge = -((mi_col * MI_SIZE) * 8); |
279 xd->mb_to_right_edge = ((mi_cols - bw - mi_col) * MI_SIZE) * 8; | 295 xd->mb_to_right_edge = ((mi_cols - bw - mi_col) * MI_SIZE) * 8; |
280 | 296 |
281 // Are edges available for intra prediction? | 297 // Are edges available for intra prediction? |
282 xd->up_available = (mi_row != 0); | 298 xd->up_available = (mi_row != 0); |
283 xd->left_available = (mi_col > tile->mi_col_start); | 299 xd->left_available = (mi_col > tile->mi_col_start); |
284 } | 300 } |
285 | 301 |
(...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
338 left = (left & bs) > 0; | 354 left = (left & bs) > 0; |
339 | 355 |
340 return (left * 2 + above) + bsl * PARTITION_PLOFFSET; | 356 return (left * 2 + above) + bsl * PARTITION_PLOFFSET; |
341 } | 357 } |
342 | 358 |
343 #ifdef __cplusplus | 359 #ifdef __cplusplus |
344 } // extern "C" | 360 } // extern "C" |
345 #endif | 361 #endif |
346 | 362 |
347 #endif // VP9_COMMON_VP9_ONYXC_INT_H_ | 363 #endif // VP9_COMMON_VP9_ONYXC_INT_H_ |
OLD | NEW |