| 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 |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 81 // that no FrameWorker owns, or is decoding, this buffer. | 81 // that no FrameWorker owns, or is decoding, this buffer. |
| 82 VP9Worker *frame_worker_owner; | 82 VP9Worker *frame_worker_owner; |
| 83 | 83 |
| 84 // row and col indicate which position frame has been decoded to in real | 84 // row and col indicate which position frame has been decoded to in real |
| 85 // pixel unit. They are reset to -1 when decoding begins and set to INT_MAX | 85 // pixel unit. They are reset to -1 when decoding begins and set to INT_MAX |
| 86 // when the frame is fully decoded. | 86 // when the frame is fully decoded. |
| 87 int row; | 87 int row; |
| 88 int col; | 88 int col; |
| 89 } RefCntBuffer; | 89 } RefCntBuffer; |
| 90 | 90 |
| 91 typedef struct { | 91 typedef struct BufferPool { |
| 92 // Protect BufferPool from being accessed by several FrameWorkers at | 92 // Protect BufferPool from being accessed by several FrameWorkers at |
| 93 // the same time during frame parallel decode. | 93 // the same time during frame parallel decode. |
| 94 // TODO(hkuang): Try to use atomic variable instead of locking the whole pool. | 94 // TODO(hkuang): Try to use atomic variable instead of locking the whole pool. |
| 95 #if CONFIG_MULTITHREAD | 95 #if CONFIG_MULTITHREAD |
| 96 pthread_mutex_t pool_mutex; | 96 pthread_mutex_t pool_mutex; |
| 97 #endif | 97 #endif |
| 98 | 98 |
| 99 // Private data associated with the frame buffer callbacks. | 99 // Private data associated with the frame buffer callbacks. |
| 100 void *cb_priv; | 100 void *cb_priv; |
| 101 | 101 |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 177 int mb_cols, mi_cols; | 177 int mb_cols, mi_cols; |
| 178 int mi_stride; | 178 int mi_stride; |
| 179 | 179 |
| 180 /* profile settings */ | 180 /* profile settings */ |
| 181 TX_MODE tx_mode; | 181 TX_MODE tx_mode; |
| 182 | 182 |
| 183 int base_qindex; | 183 int base_qindex; |
| 184 int y_dc_delta_q; | 184 int y_dc_delta_q; |
| 185 int uv_dc_delta_q; | 185 int uv_dc_delta_q; |
| 186 int uv_ac_delta_q; | 186 int uv_ac_delta_q; |
| 187 int16_t y_dequant[MAX_SEGMENTS][2]; |
| 188 int16_t uv_dequant[MAX_SEGMENTS][2]; |
| 187 | 189 |
| 188 /* We allocate a MODE_INFO struct for each macroblock, together with | 190 /* We allocate a MODE_INFO struct for each macroblock, together with |
| 189 an extra row on top and column on the left to simplify prediction. */ | 191 an extra row on top and column on the left to simplify prediction. */ |
| 190 int mi_alloc_size; | 192 int mi_alloc_size; |
| 191 MODE_INFO *mip; /* Base of allocated array */ | 193 MODE_INFO *mip; /* Base of allocated array */ |
| 192 MODE_INFO *mi; /* Corresponds to upper left visible macroblock */ | 194 MODE_INFO *mi; /* Corresponds to upper left visible macroblock */ |
| 193 | 195 |
| 194 // TODO(agrange): Move prev_mi into encoder structure. | 196 // TODO(agrange): Move prev_mi into encoder structure. |
| 195 // prev_mip and prev_mi will only be allocated in VP9 encoder. | 197 // prev_mip and prev_mi will only be allocated in VP9 encoder. |
| 196 MODE_INFO *prev_mip; /* MODE_INFO array 'mip' from last decoded frame */ | 198 MODE_INFO *prev_mip; /* MODE_INFO array 'mip' from last decoded frame */ |
| 197 MODE_INFO *prev_mi; /* 'mi' from last frame (points into prev_mip) */ | 199 MODE_INFO *prev_mi; /* 'mi' from last frame (points into prev_mip) */ |
| 198 | 200 |
| 199 // Separate mi functions between encoder and decoder. | 201 // Separate mi functions between encoder and decoder. |
| 200 int (*alloc_mi)(struct VP9Common *cm, int mi_size); | 202 int (*alloc_mi)(struct VP9Common *cm, int mi_size); |
| 201 void (*free_mi)(struct VP9Common *cm); | 203 void (*free_mi)(struct VP9Common *cm); |
| 202 void (*setup_mi)(struct VP9Common *cm); | 204 void (*setup_mi)(struct VP9Common *cm); |
| 203 | 205 |
| 206 // Grid of pointers to 8x8 MODE_INFO structs. Any 8x8 not in the visible |
| 207 // area will be NULL. |
| 208 MODE_INFO **mi_grid_base; |
| 209 MODE_INFO **mi_grid_visible; |
| 210 MODE_INFO **prev_mi_grid_base; |
| 211 MODE_INFO **prev_mi_grid_visible; |
| 204 | 212 |
| 205 // Whether to use previous frame's motion vectors for prediction. | 213 // Whether to use previous frame's motion vectors for prediction. |
| 206 int use_prev_frame_mvs; | 214 int use_prev_frame_mvs; |
| 207 | 215 |
| 208 // Persistent mb segment id map used in prediction. | 216 // Persistent mb segment id map used in prediction. |
| 209 int seg_map_idx; | 217 int seg_map_idx; |
| 210 int prev_seg_map_idx; | 218 int prev_seg_map_idx; |
| 211 | 219 |
| 212 uint8_t *seg_map_array[NUM_PING_PONG_BUFFERS]; | 220 uint8_t *seg_map_array[NUM_PING_PONG_BUFFERS]; |
| 213 uint8_t *last_frame_seg_map; | 221 uint8_t *last_frame_seg_map; |
| (...skipping 150 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 364 int mi_rows, int mi_cols) { | 372 int mi_rows, int mi_cols) { |
| 365 xd->mb_to_top_edge = -((mi_row * MI_SIZE) * 8); | 373 xd->mb_to_top_edge = -((mi_row * MI_SIZE) * 8); |
| 366 xd->mb_to_bottom_edge = ((mi_rows - bh - mi_row) * MI_SIZE) * 8; | 374 xd->mb_to_bottom_edge = ((mi_rows - bh - mi_row) * MI_SIZE) * 8; |
| 367 xd->mb_to_left_edge = -((mi_col * MI_SIZE) * 8); | 375 xd->mb_to_left_edge = -((mi_col * MI_SIZE) * 8); |
| 368 xd->mb_to_right_edge = ((mi_cols - bw - mi_col) * MI_SIZE) * 8; | 376 xd->mb_to_right_edge = ((mi_cols - bw - mi_col) * MI_SIZE) * 8; |
| 369 | 377 |
| 370 // Are edges available for intra prediction? | 378 // Are edges available for intra prediction? |
| 371 xd->up_available = (mi_row != 0); | 379 xd->up_available = (mi_row != 0); |
| 372 xd->left_available = (mi_col > tile->mi_col_start); | 380 xd->left_available = (mi_col > tile->mi_col_start); |
| 373 if (xd->up_available) { | 381 if (xd->up_available) { |
| 374 xd->above_mi = xd->mi[-xd->mi_stride].src_mi; | 382 xd->above_mi = xd->mi[-xd->mi_stride]; |
| 383 // above_mi may be NULL in VP9 encoder's first pass. |
| 375 xd->above_mbmi = xd->above_mi ? &xd->above_mi->mbmi : NULL; | 384 xd->above_mbmi = xd->above_mi ? &xd->above_mi->mbmi : NULL; |
| 376 } else { | 385 } else { |
| 377 xd->above_mi = NULL; | 386 xd->above_mi = NULL; |
| 378 xd->above_mbmi = NULL; | 387 xd->above_mbmi = NULL; |
| 379 } | 388 } |
| 380 | 389 |
| 381 if (xd->left_available) { | 390 if (xd->left_available) { |
| 382 xd->left_mi = xd->mi[-1].src_mi; | 391 xd->left_mi = xd->mi[-1]; |
| 392 // left_mi may be NULL in VP9 encoder's first pass. |
| 383 xd->left_mbmi = xd->left_mi ? &xd->left_mi->mbmi : NULL; | 393 xd->left_mbmi = xd->left_mi ? &xd->left_mi->mbmi : NULL; |
| 384 } else { | 394 } else { |
| 385 xd->left_mi = NULL; | 395 xd->left_mi = NULL; |
| 386 xd->left_mbmi = NULL; | 396 xd->left_mbmi = NULL; |
| 387 } | 397 } |
| 388 } | 398 } |
| 389 | 399 |
| 390 static INLINE void update_partition_context(MACROBLOCKD *xd, | 400 static INLINE void update_partition_context(MACROBLOCKD *xd, |
| 391 int mi_row, int mi_col, | 401 int mi_row, int mi_col, |
| 392 BLOCK_SIZE subsize, | 402 BLOCK_SIZE subsize, |
| 393 BLOCK_SIZE bsize) { | 403 BLOCK_SIZE bsize) { |
| 394 PARTITION_CONTEXT *const above_ctx = xd->above_seg_context + mi_col; | 404 PARTITION_CONTEXT *const above_ctx = xd->above_seg_context + mi_col; |
| 395 PARTITION_CONTEXT *const left_ctx = xd->left_seg_context + (mi_row & MI_MASK); | 405 PARTITION_CONTEXT *const left_ctx = xd->left_seg_context + (mi_row & MI_MASK); |
| 396 | 406 |
| 397 // num_4x4_blocks_wide_lookup[bsize] / 2 | 407 // num_4x4_blocks_wide_lookup[bsize] / 2 |
| 398 const int bs = num_8x8_blocks_wide_lookup[bsize]; | 408 const int bs = num_8x8_blocks_wide_lookup[bsize]; |
| 399 | 409 |
| 400 // update the partition context at the end notes. set partition bits | 410 // update the partition context at the end notes. set partition bits |
| 401 // of block sizes larger than the current one to be one, and partition | 411 // of block sizes larger than the current one to be one, and partition |
| 402 // bits of smaller block sizes to be zero. | 412 // bits of smaller block sizes to be zero. |
| 403 vpx_memset(above_ctx, partition_context_lookup[subsize].above, bs); | 413 memset(above_ctx, partition_context_lookup[subsize].above, bs); |
| 404 vpx_memset(left_ctx, partition_context_lookup[subsize].left, bs); | 414 memset(left_ctx, partition_context_lookup[subsize].left, bs); |
| 405 } | 415 } |
| 406 | 416 |
| 407 static INLINE int partition_plane_context(const MACROBLOCKD *xd, | 417 static INLINE int partition_plane_context(const MACROBLOCKD *xd, |
| 408 int mi_row, int mi_col, | 418 int mi_row, int mi_col, |
| 409 BLOCK_SIZE bsize) { | 419 BLOCK_SIZE bsize) { |
| 410 const PARTITION_CONTEXT *above_ctx = xd->above_seg_context + mi_col; | 420 const PARTITION_CONTEXT *above_ctx = xd->above_seg_context + mi_col; |
| 411 const PARTITION_CONTEXT *left_ctx = xd->left_seg_context + (mi_row & MI_MASK); | 421 const PARTITION_CONTEXT *left_ctx = xd->left_seg_context + (mi_row & MI_MASK); |
| 412 | |
| 413 const int bsl = mi_width_log2_lookup[bsize]; | 422 const int bsl = mi_width_log2_lookup[bsize]; |
| 414 const int bs = 1 << bsl; | 423 int above = (*above_ctx >> bsl) & 1 , left = (*left_ctx >> bsl) & 1; |
| 415 int above = 0, left = 0, i; | |
| 416 | 424 |
| 417 assert(b_width_log2_lookup[bsize] == b_height_log2_lookup[bsize]); | 425 assert(b_width_log2_lookup[bsize] == b_height_log2_lookup[bsize]); |
| 418 assert(bsl >= 0); | 426 assert(bsl >= 0); |
| 419 | 427 |
| 420 for (i = 0; i < bs; i++) { | |
| 421 above |= above_ctx[i]; | |
| 422 left |= left_ctx[i]; | |
| 423 } | |
| 424 above = (above & bs) > 0; | |
| 425 left = (left & bs) > 0; | |
| 426 | |
| 427 return (left * 2 + above) + bsl * PARTITION_PLOFFSET; | 428 return (left * 2 + above) + bsl * PARTITION_PLOFFSET; |
| 428 } | 429 } |
| 429 | 430 |
| 430 #ifdef __cplusplus | 431 #ifdef __cplusplus |
| 431 } // extern "C" | 432 } // extern "C" |
| 432 #endif | 433 #endif |
| 433 | 434 |
| 434 #endif // VP9_COMMON_VP9_ONYXC_INT_H_ | 435 #endif // VP9_COMMON_VP9_ONYXC_INT_H_ |
| OLD | NEW |