vp8/encoder/block.h

   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_BLOCK_H
  13 #define __INC_BLOCK_H
  14
  15 #include "vp8/common/onyx.h"
  16 #include "vp8/common/blockd.h"
  17 #include "vp8/common/entropymv.h"
  18 #include "vp8/common/entropy.h"
  19 #include "vpx_ports/mem.h"
  20
  21 // motion search site
  22 typedef struct
  23 {
  24     MV mv;
  25     int offset;
  26 } search_site;
  27
  28 typedef struct
  29 {
  30     // 16 Y blocks, 4 U blocks, 4 V blocks each with 16 entries
  31     short *src_diff;
  32     short *coeff;
  33
  34     // 16 Y blocks, 4 U blocks, 4 V blocks each with 16 entries
  35     short *quant;
  36     short *quant_fast;
  37     unsigned char *quant_shift;
  38     short *zbin;
  39     short *zrun_zbin_boost;
  40     short *round;
  41
  42     // Zbin Over Quant value
  43     short zbin_extra;
  44
  45     unsigned char **base_src;
  46     int src;
  47     int src_stride;
  48
  49 //  MV  enc_mv;
  50     int force_empty;
  51
  52 } BLOCK;
  53
  54 typedef struct
  55 {
  56     int count;
  57     struct
  58     {
  59         B_PREDICTION_MODE mode;
  60         int_mv mv;
  61     } bmi[16];
  62 } PARTITION_INFO;
  63
  64 typedef struct
  65 {
  66     DECLARE_ALIGNED(16, short, src_diff[400]);       // 16x16 Y 8x8 U 8x8 V 4x4 2nd Y
  67     DECLARE_ALIGNED(16, short, coeff[400]);     // 16x16 Y 8x8 U 8x8 V 4x4 2nd Y
  68
  69     // 16 Y blocks, 4 U blocks, 4 V blocks, 1 DC 2nd order block each with 16 entries
  70     BLOCK block[25];
  71
  72     YV12_BUFFER_CONFIG src;
  73
  74     MACROBLOCKD e_mbd;
  75     PARTITION_INFO *partition_info; /* work pointer */
  76     PARTITION_INFO *pi;   /* Corresponds to upper left visible macroblock */
  77     PARTITION_INFO *pip;  /* Base of allocated array */
  78
  79     search_site *ss;
  80     int ss_count;
  81     int searches_per_step;
  82
  83     int errorperbit;
  84     int sadperbit16;
  85     int sadperbit4;
  86     int rddiv;
  87     int rdmult;
  88     unsigned int * mb_activity_ptr;
  89     int * mb_norm_activity_ptr;
  90     signed int act_zbin_adj;
  91
  92     int mvcosts[2][MVvals+1];
  93     int *mvcost[2];
  94     int mvsadcosts[2][MVfpvals+1];
  95     int *mvsadcost[2];
  96     int mbmode_cost[2][MB_MODE_COUNT];
  97     int intra_uv_mode_cost[2][MB_MODE_COUNT];
  98     unsigned int bmode_costs[10][10][10];
  99     unsigned int inter_bmode_costs[B_MODE_COUNT];
 100
 101     // These define limits to motion vector components to prevent them from extending outside the UMV borders
 102     int mv_col_min;
 103     int mv_col_max;
 104     int mv_row_min;
 105     int mv_row_max;
 106
 107     int vector_range;    // Used to monitor limiting range of recent vectors to guide search.
 108     int skip;
 109
 110     int encode_breakout;
 111
 112     //char * gf_active_ptr;
 113     signed char *gf_active_ptr;
 114
 115     unsigned char *active_ptr;
 116     MV_CONTEXT *mvc;
 117
 118     unsigned int token_costs[BLOCK_TYPES] [COEF_BANDS] [PREV_COEF_CONTEXTS] [vp8_coef_tokens];
 119     int optimize;
 120     int q_index;
 121
 122     void (*vp8_short_fdct4x4)(short *input, short *output, int pitch);
 123     void (*vp8_short_fdct8x4)(short *input, short *output, int pitch);
 124     void (*short_walsh4x4)(short *input, short *output, int pitch);
 125     void (*quantize_b)(BLOCK *b, BLOCKD *d);
 126     void (*quantize_b_pair)(BLOCK *b1, BLOCK *b2, BLOCKD *d0, BLOCKD *d1);
 127
 128 } MACROBLOCK;
 129
 130
 131 #endif