summaryrefslogtreecommitdiff
path: root/vp8/encoder/block.h
blob: 98bc132777ba3b381cbe90fc88dd90260c467da2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */


#ifndef __INC_BLOCK_H
#define __INC_BLOCK_H

#include "vp8/common/onyx.h"
#include "vp8/common/blockd.h"
#include "vp8/common/entropymv.h"
#include "vp8/common/entropy.h"
#include "vpx_ports/mem.h"

// motion search site
typedef struct
{
    MV mv;
    int offset;
} search_site;

typedef struct
{
    // 16 Y blocks, 4 U blocks, 4 V blocks each with 16 entries
    short *src_diff;
    short *coeff;

    // 16 Y blocks, 4 U blocks, 4 V blocks each with 16 entries
    short *quant;
    short *quant_fast;      // fast quant deprecated for now
    unsigned char *quant_shift;
    short *zbin;
    short *zbin_8x8;
    short *zrun_zbin_boost;
    short *zrun_zbin_boost_8x8;
    short *round;

    // Zbin Over Quant value
    short zbin_extra;

    unsigned char **base_src;
    unsigned char **base_second_src;
    int src;
    int src_stride;

    int eob_max_offset;
    int eob_max_offset_8x8;

} BLOCK;

typedef struct
{
    int count;
    struct
    {
        B_PREDICTION_MODE mode;
        int_mv mv;
        int_mv second_mv;
    } bmi[16];
} PARTITION_INFO;

// Structure to hold snapshot of coding context during the mode picking process
// TODO Do we need all of these?
typedef struct
{
    MODE_INFO mic;
    PARTITION_INFO partition_info;
    int_mv best_ref_mv;
    int rate;
    int distortion;
    int intra_error;
    int best_mode_index;
    int rddiv;
    int rdmult;

} PICK_MODE_CONTEXT;

typedef struct
{
    DECLARE_ALIGNED(16, short, src_diff[400]);  // 16x16 Y 8x8 U 8x8 V 4x4 2nd Y
    DECLARE_ALIGNED(16, short, coeff[400]);     // 16x16 Y 8x8 U 8x8 V 4x4 2nd Y
    DECLARE_ALIGNED(16, unsigned char, thismb[256]);    // 16x16 Y

    unsigned char *thismb_ptr;
    // 16 Y blocks, 4 U blocks, 4 V blocks,
    // 1 DC 2nd order block each with 16 entries
    BLOCK block[25];

    YV12_BUFFER_CONFIG src;

    MACROBLOCKD e_mbd;
    PARTITION_INFO *partition_info; /* work pointer */
    PARTITION_INFO *pi;   /* Corresponds to upper left visible macroblock */
    PARTITION_INFO *pip;  /* Base of allocated array */

    search_site *ss;
    int ss_count;
    int searches_per_step;

    int errorperbit;
    int sadperbit16;
    int sadperbit4;
    int rddiv;
    int rdmult;
    unsigned int * mb_activity_ptr;
    int * mb_norm_activity_ptr;
    signed int act_zbin_adj;

    int mvcosts[2][MVvals+1];
    int *mvcost[2];
    int mvsadcosts[2][MVfpvals+1];
    int *mvsadcost[2];
#if CONFIG_HIGH_PRECISION_MV
    int mvcosts_hp[2][MVvals_hp+1];
    int *mvcost_hp[2];
    int mvsadcosts_hp[2][MVfpvals_hp+1];
    int *mvsadcost_hp[2];
#endif
    int mbmode_cost[2][MB_MODE_COUNT];
    int intra_uv_mode_cost[2][MB_MODE_COUNT];
    unsigned int bmode_costs[VP8_BINTRAMODES][VP8_BINTRAMODES][VP8_BINTRAMODES];
    unsigned int i8x8_mode_costs[MB_MODE_COUNT];
    unsigned int inter_bmode_costs[B_MODE_COUNT];

    // These define limits to motion vector components to prevent them from extending outside the UMV borders
    int mv_col_min;
    int mv_col_max;
    int mv_row_min;
    int mv_row_max;

    int skip;

    int encode_breakout;

    //char * gf_active_ptr;
    signed char *gf_active_ptr;

    unsigned char *active_ptr;
    MV_CONTEXT *mvc;
#if CONFIG_HIGH_PRECISION_MV
    MV_CONTEXT_HP *mvc_hp;
#endif

    unsigned int token_costs[BLOCK_TYPES] [COEF_BANDS]
                            [PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];
    unsigned int token_costs_8x8[BLOCK_TYPES_8X8] [COEF_BANDS]
                            [PREV_COEF_CONTEXTS] [MAX_ENTROPY_TOKENS];

    int optimize;
    int q_index;

    int encode_as_sb;

    // Structure to hold context for each of the 4 MBs within a SB:
    // when encoded as 4 independent MBs:
    PICK_MODE_CONTEXT mb_context[4];
#if CONFIG_SUPERBLOCKS
    // when 4 MBs share coding parameters:
    PICK_MODE_CONTEXT sb_context[4];
#endif

    void (*vp8_short_fdct4x4)(short *input, short *output, int pitch);
    void (*vp8_short_fdct8x4)(short *input, short *output, int pitch);
    void (*short_walsh4x4)(short *input, short *output, int pitch);
    void (*quantize_b)(BLOCK *b, BLOCKD *d);
    void (*quantize_b_pair)(BLOCK *b1, BLOCK *b2, BLOCKD *d0, BLOCKD *d1);
    void (*vp8_short_fdct8x8)(short *input, short *output, int pitch);
    void (*short_fhaar2x2)(short *input, short *output, int pitch);
    void (*quantize_b_8x8)(BLOCK *b, BLOCKD *d);
    void (*quantize_b_2x2)(BLOCK *b, BLOCKD *d);

} MACROBLOCK;


#endif