vp8/common/alloccommon.c

   1 /*
   2  *  Copyright (c) 2010 The VP8 project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license and patent
   5  *  grant that can be found in the LICENSE file in the root of the source
   6  *  tree. All contributing project authors may be found in the AUTHORS
   7  *  file in the root of the source tree.
   8  */
   9
  10
  11 #include "blockd.h"
  12 #include "vpx_mem/vpx_mem.h"
  13 #include "onyxc_int.h"
  14 #include "findnearmv.h"
  15 #include "entropymode.h"
  16 #include "systemdependent.h"
  17 #include "vpxerrors.h"
  18
  19 #ifdef HAVE_CONFIG_H
  20 #include "vpx_config.h"
  21 #endif
  22
  23 extern  void vp8_init_scan_order_mask();
  24
  25 void vp8_update_mode_info_border(MODE_INFO *mi, int rows, int cols)
  26 {
  27     int i;
  28     vpx_memset(mi - cols - 1, 0, sizeof(MODE_INFO) * cols + 1);
  29
  30     for (i = 0; i < rows; i++)
  31     {
  32         vpx_memset(&mi[i*cols-1], 0, sizeof(MODE_INFO));
  33     }
  34 }
  35 void vp8_de_alloc_frame_buffers(VP8_COMMON *oci)
  36 {
  37     vp8_yv12_de_alloc_frame_buffer(&oci->temp_scale_frame);
  38     vp8_yv12_de_alloc_frame_buffer(&oci->new_frame);
  39     vp8_yv12_de_alloc_frame_buffer(&oci->last_frame);
  40     vp8_yv12_de_alloc_frame_buffer(&oci->golden_frame);
  41     vp8_yv12_de_alloc_frame_buffer(&oci->alt_ref_frame);
  42     vp8_yv12_de_alloc_frame_buffer(&oci->post_proc_buffer);
  43
  44     vpx_free(oci->above_context[Y1CONTEXT]);
  45     vpx_free(oci->above_context[UCONTEXT]);
  46     vpx_free(oci->above_context[VCONTEXT]);
  47     vpx_free(oci->above_context[Y2CONTEXT]);
  48     vpx_free(oci->mip);
  49
  50     oci->above_context[Y1CONTEXT] = 0;
  51     oci->above_context[UCONTEXT]  = 0;
  52     oci->above_context[VCONTEXT]  = 0;
  53     oci->above_context[Y2CONTEXT] = 0;
  54     oci->mip = 0;
  55
  56     // Structure used to minitor GF useage
  57     if (oci->gf_active_flags != 0)
  58         vpx_free(oci->gf_active_flags);
  59
  60     oci->gf_active_flags = 0;
  61 }
  62
  63 int vp8_alloc_frame_buffers(VP8_COMMON *oci, int width, int height)
  64 {
  65     vp8_de_alloc_frame_buffers(oci);
  66
  67     // our internal buffers are always multiples of 16
  68     if ((width & 0xf) != 0)
  69         width += 16 - (width & 0xf);
  70
  71     if ((height & 0xf) != 0)
  72         height += 16 - (height & 0xf);
  73
  74
  75     if (vp8_yv12_alloc_frame_buffer(&oci->temp_scale_frame,   width, 16, VP8BORDERINPIXELS) < 0)
  76     {
  77         vp8_de_alloc_frame_buffers(oci);
  78         return ALLOC_FAILURE;
  79     }
  80
  81
  82     if (vp8_yv12_alloc_frame_buffer(&oci->new_frame,   width, height, VP8BORDERINPIXELS) < 0)
  83     {
  84         vp8_de_alloc_frame_buffers(oci);
  85         return ALLOC_FAILURE;
  86     }
  87
  88     if (vp8_yv12_alloc_frame_buffer(&oci->last_frame,  width, height, VP8BORDERINPIXELS) < 0)
  89     {
  90         vp8_de_alloc_frame_buffers(oci);
  91         return ALLOC_FAILURE;
  92     }
  93
  94     if (vp8_yv12_alloc_frame_buffer(&oci->golden_frame, width, height, VP8BORDERINPIXELS) < 0)
  95     {
  96         vp8_de_alloc_frame_buffers(oci);
  97         return ALLOC_FAILURE;
  98     }
  99
 100     if (vp8_yv12_alloc_frame_buffer(&oci->alt_ref_frame, width, height, VP8BORDERINPIXELS) < 0)
 101     {
 102         vp8_de_alloc_frame_buffers(oci);
 103         return ALLOC_FAILURE;
 104     }
 105
 106     if (vp8_yv12_alloc_frame_buffer(&oci->post_proc_buffer, width, height, VP8BORDERINPIXELS) < 0)
 107     {
 108         vp8_de_alloc_frame_buffers(oci);
 109         return ALLOC_FAILURE;
 110     }
 111
 112     oci->mb_rows = height >> 4;
 113     oci->mb_cols = width >> 4;
 114     oci->MBs = oci->mb_rows * oci->mb_cols;
 115     oci->mode_info_stride = oci->mb_cols + 1;
 116     oci->mip = vpx_calloc((oci->mb_cols + 1) * (oci->mb_rows + 1), sizeof(MODE_INFO));
 117
 118     if (!oci->mip)
 119     {
 120         vp8_de_alloc_frame_buffers(oci);
 121         return ALLOC_FAILURE;
 122     }
 123
 124     oci->mi = oci->mip + oci->mode_info_stride + 1;
 125
 126
 127     oci->above_context[Y1CONTEXT] = vpx_calloc(sizeof(ENTROPY_CONTEXT) * oci->mb_cols * 4 , 1);
 128
 129     if (!oci->above_context[Y1CONTEXT])
 130     {
 131         vp8_de_alloc_frame_buffers(oci);
 132         return ALLOC_FAILURE;
 133     }
 134
 135     oci->above_context[UCONTEXT]  = vpx_calloc(sizeof(ENTROPY_CONTEXT) * oci->mb_cols * 2 , 1);
 136
 137     if (!oci->above_context[UCONTEXT])
 138     {
 139         vp8_de_alloc_frame_buffers(oci);
 140         return ALLOC_FAILURE;
 141     }
 142
 143     oci->above_context[VCONTEXT]  = vpx_calloc(sizeof(ENTROPY_CONTEXT) * oci->mb_cols * 2 , 1);
 144
 145     if (!oci->above_context[VCONTEXT])
 146     {
 147         vp8_de_alloc_frame_buffers(oci);
 148         return ALLOC_FAILURE;
 149     }
 150
 151     oci->above_context[Y2CONTEXT] = vpx_calloc(sizeof(ENTROPY_CONTEXT) * oci->mb_cols     , 1);
 152
 153     if (!oci->above_context[Y2CONTEXT])
 154     {
 155         vp8_de_alloc_frame_buffers(oci);
 156         return ALLOC_FAILURE;
 157     }
 158
 159     vp8_update_mode_info_border(oci->mi, oci->mb_rows, oci->mb_cols);
 160
 161     // Structures used to minitor GF usage
 162     if (oci->gf_active_flags != 0)
 163         vpx_free(oci->gf_active_flags);
 164
 165     oci->gf_active_flags = (unsigned char *)vpx_calloc(oci->mb_rows * oci->mb_cols, 1);
 166
 167     if (!oci->gf_active_flags)
 168     {
 169         vp8_de_alloc_frame_buffers(oci);
 170         return ALLOC_FAILURE;
 171     }
 172
 173     oci->gf_active_count = oci->mb_rows * oci->mb_cols;
 174
 175     return 0;
 176 }
 177 void vp8_setup_version(VP8_COMMON *cm)
 178 {
 179     switch (cm->version)
 180     {
 181     case 0:
 182         cm->no_lpf = 0;
 183         cm->simpler_lpf = 0;
 184         cm->use_bilinear_mc_filter = 0;
 185         cm->full_pixel = 0;
 186         break;
 187     case 1:
 188         cm->no_lpf = 0;
 189         cm->simpler_lpf = 1;
 190         cm->use_bilinear_mc_filter = 1;
 191         cm->full_pixel = 0;
 192         break;
 193     case 2:
 194         cm->no_lpf = 1;
 195         cm->simpler_lpf = 0;
 196         cm->use_bilinear_mc_filter = 1;
 197         cm->full_pixel = 0;
 198         break;
 199     case 3:
 200         cm->no_lpf = 1;
 201         cm->simpler_lpf = 1;
 202         cm->use_bilinear_mc_filter = 1;
 203         cm->full_pixel = 1;
 204         break;
 205     default:
 206         //4,5,6,7 are reserved for future use
 207         cm->no_lpf = 0;
 208         cm->simpler_lpf = 0;
 209         cm->use_bilinear_mc_filter = 0;
 210         cm->full_pixel = 0;
 211         break;
 212     }
 213 }
 214 void vp8_create_common(VP8_COMMON *oci)
 215 {
 216     vp8_machine_specific_config(oci);
 217     vp8_default_coef_probs(oci);
 218     vp8_init_mbmode_probs(oci);
 219     vp8_default_bmode_probs(oci->fc.bmode_prob);
 220
 221     oci->mb_no_coeff_skip = 1;
 222     oci->no_lpf = 0;
 223     oci->simpler_lpf = 0;
 224     oci->use_bilinear_mc_filter = 0;
 225     oci->full_pixel = 0;
 226     oci->multi_token_partition = ONE_PARTITION;
 227     oci->clr_type = REG_YUV;
 228     oci->clamp_type = RECON_CLAMP_REQUIRED;
 229
 230     // Initialise reference frame sign bias structure to defaults
 231     vpx_memset(oci->ref_frame_sign_bias, 0, sizeof(oci->ref_frame_sign_bias));
 232
 233     // Default disable buffer to buffer copying
 234     oci->copy_buffer_to_gf = 0;
 235     oci->copy_buffer_to_arf = 0;
 236 }
 237
 238 void vp8_remove_common(VP8_COMMON *oci)
 239 {
 240     vp8_de_alloc_frame_buffers(oci);
 241 }
 242
 243 void vp8_initialize_common()
 244 {
 245     vp8_coef_tree_initialize();
 246
 247     vp8_entropy_mode_init();
 248
 249     vp8_init_scan_order_mask();
 250
 251 }