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     DECLARE_ALIGNED(16, unsigned char, thismb[256]);
  69
  70     unsigned char *thismb_ptr;
  71     // 16 Y blocks, 4 U blocks, 4 V blocks, 1 DC 2nd order block each with 16 entries
  72     BLOCK block[25];
  73
  74     YV12_BUFFER_CONFIG src;
  75
  76     MACROBLOCKD e_mbd;
  77     PARTITION_INFO *partition_info; /* work pointer */
  78     PARTITION_INFO *pi;   /* Corresponds to upper left visible macroblock */
  79     PARTITION_INFO *pip;  /* Base of allocated array */
  80
  81     search_site *ss;
  82     int ss_count;
  83     int searches_per_step;
  84
  85     int errorperbit;
  86     int sadperbit16;
  87     int sadperbit4;
  88     int rddiv;
  89     int rdmult;
  90     unsigned int * mb_activity_ptr;
  91     int * mb_norm_activity_ptr;
  92     signed int act_zbin_adj;
  93
  94     int mvcosts[2][MVvals+1];
  95     int *mvcost[2];
  96     int mvsadcosts[2][MVfpvals+1];
  97     int *mvsadcost[2];
  98     int mbmode_cost[2][MB_MODE_COUNT];
  99     int intra_uv_mode_cost[2][MB_MODE_COUNT];
 100     unsigned int bmode_costs[10][10][10];
 101     unsigned int inter_bmode_costs[B_MODE_COUNT];
 102
 103     // These define limits to motion vector components to prevent them from extending outside the UMV borders
 104     int mv_col_min;
 105     int mv_col_max;
 106     int mv_row_min;
 107     int mv_row_max;
 108
 109     int vector_range;    // Used to monitor limiting range of recent vectors to guide search.
 110     int skip;
 111
 112     int encode_breakout;
 113
 114     //char * gf_active_ptr;
 115     signed char *gf_active_ptr;
 116
 117     unsigned char *active_ptr;
 118     MV_CONTEXT *mvc;
 119
 120     unsigned int token_costs[BLOCK_TYPES] [COEF_BANDS] [PREV_COEF_CONTEXTS] [MAX_ENTROPY_TOKENS];
 121     int optimize;
 122     int q_index;
 123
 124     void (*vp8_short_fdct4x4)(short *input, short *output, int pitch);
 125     void (*vp8_short_fdct8x4)(short *input, short *output, int pitch);
 126     void (*short_walsh4x4)(short *input, short *output, int pitch);
 127     void (*quantize_b)(BLOCK *b, BLOCKD *d);
 128     void (*quantize_b_pair)(BLOCK *b1, BLOCK *b2, BLOCKD *d0, BLOCKD *d1);
 129
 130 } MACROBLOCK;
 131
 132
 133 #endif