Index: source/libvpx/vp9/common/vp9_alloccommon.c |
=================================================================== |
--- source/libvpx/vp9/common/vp9_alloccommon.c (revision 240950) |
+++ source/libvpx/vp9/common/vp9_alloccommon.c (working copy) |
@@ -34,7 +34,7 @@ |
void vp9_free_frame_buffers(VP9_COMMON *cm) { |
int i; |
- for (i = 0; i < NUM_YV12_BUFFERS; i++) |
+ for (i = 0; i < cm->fb_count; i++) |
vp9_free_frame_buffer(&cm->yv12_fb[i]); |
vp9_free_frame_buffer(&cm->post_proc_buffer); |
@@ -75,10 +75,60 @@ |
cm->mode_info_stride * (cm->mi_rows + 1) * |
sizeof(*cm->mi_grid_base)); |
- vp9_update_mode_info_border(cm, cm->mip); |
vp9_update_mode_info_border(cm, cm->prev_mip); |
} |
+int vp9_resize_frame_buffers(VP9_COMMON *cm, int width, int height) { |
+ const int aligned_width = ALIGN_POWER_OF_TWO(width, MI_SIZE_LOG2); |
+ const int aligned_height = ALIGN_POWER_OF_TWO(height, MI_SIZE_LOG2); |
+ const int ss_x = cm->subsampling_x; |
+ const int ss_y = cm->subsampling_y; |
+ int mi_size; |
+ |
+ if (vp9_realloc_frame_buffer(&cm->post_proc_buffer, width, height, ss_x, ss_y, |
+ VP9BORDERINPIXELS, NULL, NULL, NULL) < 0) |
+ goto fail; |
+ |
+ set_mb_mi(cm, aligned_width, aligned_height); |
+ |
+ // Allocation |
+ mi_size = cm->mode_info_stride * (cm->mi_rows + MI_BLOCK_SIZE); |
+ |
+ vpx_free(cm->mip); |
+ cm->mip = vpx_calloc(mi_size, sizeof(MODE_INFO)); |
+ if (!cm->mip) |
+ goto fail; |
+ |
+ vpx_free(cm->prev_mip); |
+ cm->prev_mip = vpx_calloc(mi_size, sizeof(MODE_INFO)); |
+ if (!cm->prev_mip) |
+ goto fail; |
+ |
+ vpx_free(cm->mi_grid_base); |
+ cm->mi_grid_base = vpx_calloc(mi_size, sizeof(*cm->mi_grid_base)); |
+ if (!cm->mi_grid_base) |
+ goto fail; |
+ |
+ vpx_free(cm->prev_mi_grid_base); |
+ cm->prev_mi_grid_base = vpx_calloc(mi_size, sizeof(*cm->prev_mi_grid_base)); |
+ if (!cm->prev_mi_grid_base) |
+ goto fail; |
+ |
+ setup_mi(cm); |
+ |
+ // Create the segmentation map structure and set to 0. |
+ vpx_free(cm->last_frame_seg_map); |
+ cm->last_frame_seg_map = vpx_calloc(cm->mi_rows * cm->mi_cols, 1); |
+ if (!cm->last_frame_seg_map) |
+ goto fail; |
+ |
+ return 0; |
+ |
+ fail: |
+ vp9_free_frame_buffers(cm); |
+ return 1; |
+} |
+ |
int vp9_alloc_frame_buffers(VP9_COMMON *cm, int width, int height) { |
int i; |
@@ -88,22 +138,34 @@ |
const int ss_y = cm->subsampling_y; |
int mi_size; |
+ if (cm->fb_count == 0) { |
+ cm->fb_count = FRAME_BUFFERS; |
+ CHECK_MEM_ERROR(cm, cm->yv12_fb, |
+ vpx_calloc(cm->fb_count, sizeof(*cm->yv12_fb))); |
+ CHECK_MEM_ERROR(cm, cm->fb_idx_ref_cnt, |
+ vpx_calloc(cm->fb_count, sizeof(*cm->fb_idx_ref_cnt))); |
+ if (cm->fb_lru) { |
+ CHECK_MEM_ERROR(cm, cm->fb_idx_ref_lru, |
+ vpx_calloc(cm->fb_count, sizeof(*cm->fb_idx_ref_lru))); |
+ } |
+ } |
+ |
vp9_free_frame_buffers(cm); |
- for (i = 0; i < NUM_YV12_BUFFERS; i++) { |
+ for (i = 0; i < cm->fb_count; i++) { |
cm->fb_idx_ref_cnt[i] = 0; |
if (vp9_alloc_frame_buffer(&cm->yv12_fb[i], width, height, ss_x, ss_y, |
VP9BORDERINPIXELS) < 0) |
goto fail; |
} |
- cm->new_fb_idx = NUM_YV12_BUFFERS - 1; |
+ cm->new_fb_idx = cm->fb_count - 1; |
cm->fb_idx_ref_cnt[cm->new_fb_idx] = 1; |
- for (i = 0; i < ALLOWED_REFS_PER_FRAME; i++) |
+ for (i = 0; i < REFS_PER_FRAME; i++) |
cm->active_ref_idx[i] = i; |
- for (i = 0; i < NUM_REF_FRAMES; i++) { |
+ for (i = 0; i < REF_FRAMES; i++) { |
cm->ref_frame_map[i] = i; |
cm->fb_idx_ref_cnt[i] = 1; |
} |
@@ -149,20 +211,22 @@ |
void vp9_create_common(VP9_COMMON *cm) { |
vp9_machine_specific_config(cm); |
- |
- cm->tx_mode = ONLY_4X4; |
- cm->comp_pred_mode = HYBRID_PREDICTION; |
} |
void vp9_remove_common(VP9_COMMON *cm) { |
vp9_free_frame_buffers(cm); |
+ |
+ vpx_free(cm->yv12_fb); |
+ vpx_free(cm->fb_idx_ref_cnt); |
+ vpx_free(cm->fb_idx_ref_lru); |
+ |
+ cm->yv12_fb = NULL; |
+ cm->fb_idx_ref_cnt = NULL; |
+ cm->fb_idx_ref_lru = NULL; |
} |
void vp9_initialize_common() { |
vp9_init_neighbors(); |
- vp9_coef_tree_initialize(); |
- vp9_entropy_mode_init(); |
- vp9_entropy_mv_init(); |
} |
void vp9_update_frame_size(VP9_COMMON *cm) { |