00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00026 #ifndef AVCODEC_VP56_H
00027 #define AVCODEC_VP56_H
00028
00029 #include "vp56data.h"
00030 #include "dsputil.h"
00031 #include "get_bits.h"
00032 #include "bytestream.h"
00033 #include "vp3dsp.h"
00034 #include "vp56dsp.h"
00035
00036 typedef struct vp56_context VP56Context;
00037
00038 typedef struct VP56mv {
00039 DECLARE_ALIGNED(4, int16_t, x);
00040 int16_t y;
00041 } VP56mv;
00042
00043 #define VP56_SIZE_CHANGE 1
00044
00045 typedef void (*VP56ParseVectorAdjustment)(VP56Context *s,
00046 VP56mv *vect);
00047 typedef void (*VP56Filter)(VP56Context *s, uint8_t *dst, uint8_t *src,
00048 int offset1, int offset2, int stride,
00049 VP56mv mv, int mask, int select, int luma);
00050 typedef void (*VP56ParseCoeff)(VP56Context *s);
00051 typedef void (*VP56DefaultModelsInit)(VP56Context *s);
00052 typedef void (*VP56ParseVectorModels)(VP56Context *s);
00053 typedef int (*VP56ParseCoeffModels)(VP56Context *s);
00054 typedef int (*VP56ParseHeader)(VP56Context *s, const uint8_t *buf,
00055 int buf_size);
00056
00057 typedef struct VP56RangeCoder {
00058 int high;
00059 int bits;
00060
00061 const uint8_t *buffer;
00062 const uint8_t *end;
00063 unsigned int code_word;
00064 } VP56RangeCoder;
00065
00066 typedef struct VP56RefDc {
00067 uint8_t not_null_dc;
00068 VP56Frame ref_frame;
00069 DCTELEM dc_coeff;
00070 } VP56RefDc;
00071
00072 typedef struct VP56Macroblock {
00073 uint8_t type;
00074 VP56mv mv;
00075 } VP56Macroblock;
00076
00077 typedef struct VP56Model {
00078 uint8_t coeff_reorder[64];
00079 uint8_t coeff_index_to_pos[64];
00080 uint8_t vector_sig[2];
00081 uint8_t vector_dct[2];
00082 uint8_t vector_pdi[2][2];
00083 uint8_t vector_pdv[2][7];
00084 uint8_t vector_fdv[2][8];
00085 uint8_t coeff_dccv[2][11];
00086 uint8_t coeff_ract[2][3][6][11];
00087 uint8_t coeff_acct[2][3][3][6][5];
00088 uint8_t coeff_dcct[2][36][5];
00089 uint8_t coeff_runv[2][14];
00090 uint8_t mb_type[3][10][10];
00091 uint8_t mb_types_stats[3][10][2];
00092 } VP56Model;
00093
00094 struct vp56_context {
00095 AVCodecContext *avctx;
00096 DSPContext dsp;
00097 VP3DSPContext vp3dsp;
00098 VP56DSPContext vp56dsp;
00099 ScanTable scantable;
00100 AVFrame frames[4];
00101 AVFrame *framep[6];
00102 uint8_t *edge_emu_buffer_alloc;
00103 uint8_t *edge_emu_buffer;
00104 VP56RangeCoder c;
00105 VP56RangeCoder cc;
00106 VP56RangeCoder *ccp;
00107 int sub_version;
00108
00109
00110 int golden_frame;
00111 int plane_width[4];
00112 int plane_height[4];
00113 int mb_width;
00114 int mb_height;
00115 int block_offset[6];
00116
00117 int quantizer;
00118 uint16_t dequant_dc;
00119 uint16_t dequant_ac;
00120 int8_t *qscale_table;
00121
00122
00123 VP56RefDc *above_blocks;
00124 VP56RefDc left_block[4];
00125 int above_block_idx[6];
00126 DCTELEM prev_dc[3][3];
00127
00128
00129 VP56mb mb_type;
00130 VP56Macroblock *macroblocks;
00131 DECLARE_ALIGNED(16, DCTELEM, block_coeff)[6][64];
00132
00133
00134 VP56mv mv[6];
00135 VP56mv vector_candidate[2];
00136 int vector_candidate_pos;
00137
00138
00139 int filter_header;
00140 int deblock_filtering;
00141 int filter_selection;
00142 int filter_mode;
00143 int max_vector_length;
00144 int sample_variance_threshold;
00145
00146 uint8_t coeff_ctx[4][64];
00147 uint8_t coeff_ctx_last[4];
00148
00149 int has_alpha;
00150
00151
00152 int flip;
00153 int frbi;
00154 int srbi;
00155 int stride[4];
00156
00157 const uint8_t *vp56_coord_div;
00158 VP56ParseVectorAdjustment parse_vector_adjustment;
00159 VP56Filter filter;
00160 VP56ParseCoeff parse_coeff;
00161 VP56DefaultModelsInit default_models_init;
00162 VP56ParseVectorModels parse_vector_models;
00163 VP56ParseCoeffModels parse_coeff_models;
00164 VP56ParseHeader parse_header;
00165
00166
00167 VP56Context *alpha_context;
00168
00169 VP56Model *modelp;
00170 VP56Model model;
00171
00172
00173 int use_huffman;
00174 GetBitContext gb;
00175 VLC dccv_vlc[2];
00176 VLC runv_vlc[2];
00177 VLC ract_vlc[2][3][6];
00178 unsigned int nb_null[2][2];
00179 };
00180
00181
00182 void ff_vp56_init(AVCodecContext *avctx, int flip, int has_alpha);
00183 void ff_vp56_init_context(AVCodecContext *avctx, VP56Context *s,
00184 int flip, int has_alpha);
00185 int ff_vp56_free(AVCodecContext *avctx);
00186 int ff_vp56_free_context(VP56Context *s);
00187 void ff_vp56_init_dequant(VP56Context *s, int quantizer);
00188 int ff_vp56_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
00189 AVPacket *avpkt);
00190
00191
00196 extern const uint8_t ff_vp56_norm_shift[256];
00197 void ff_vp56_init_range_decoder(VP56RangeCoder *c, const uint8_t *buf, int buf_size);
00198
00199 static av_always_inline unsigned int vp56_rac_renorm(VP56RangeCoder *c)
00200 {
00201 int shift = ff_vp56_norm_shift[c->high];
00202 int bits = c->bits;
00203 unsigned int code_word = c->code_word;
00204
00205 c->high <<= shift;
00206 code_word <<= shift;
00207 bits += shift;
00208 if(bits >= 0 && c->buffer < c->end) {
00209 code_word |= bytestream_get_be16(&c->buffer) << bits;
00210 bits -= 16;
00211 }
00212 c->bits = bits;
00213 return code_word;
00214 }
00215
00216 #if ARCH_ARM
00217 #include "arm/vp56_arith.h"
00218 #elif ARCH_X86
00219 #include "x86/vp56_arith.h"
00220 #endif
00221
00222 #ifndef vp56_rac_get_prob
00223 #define vp56_rac_get_prob vp56_rac_get_prob
00224 static av_always_inline int vp56_rac_get_prob(VP56RangeCoder *c, uint8_t prob)
00225 {
00226 unsigned int code_word = vp56_rac_renorm(c);
00227 unsigned int low = 1 + (((c->high - 1) * prob) >> 8);
00228 unsigned int low_shift = low << 16;
00229 int bit = code_word >= low_shift;
00230
00231 c->high = bit ? c->high - low : low;
00232 c->code_word = bit ? code_word - low_shift : code_word;
00233
00234 return bit;
00235 }
00236 #endif
00237
00238 #ifndef vp56_rac_get_prob_branchy
00239
00240 static av_always_inline int vp56_rac_get_prob_branchy(VP56RangeCoder *c, int prob)
00241 {
00242 unsigned long code_word = vp56_rac_renorm(c);
00243 unsigned low = 1 + (((c->high - 1) * prob) >> 8);
00244 unsigned low_shift = low << 16;
00245
00246 if (code_word >= low_shift) {
00247 c->high -= low;
00248 c->code_word = code_word - low_shift;
00249 return 1;
00250 }
00251
00252 c->high = low;
00253 c->code_word = code_word;
00254 return 0;
00255 }
00256 #endif
00257
00258 static av_always_inline int vp56_rac_get(VP56RangeCoder *c)
00259 {
00260 unsigned int code_word = vp56_rac_renorm(c);
00261
00262 int low = (c->high + 1) >> 1;
00263 unsigned int low_shift = low << 16;
00264 int bit = code_word >= low_shift;
00265 if (bit) {
00266 c->high -= low;
00267 code_word -= low_shift;
00268 } else {
00269 c->high = low;
00270 }
00271
00272 c->code_word = code_word;
00273 return bit;
00274 }
00275
00276
00277 static av_always_inline int vp8_rac_get(VP56RangeCoder *c)
00278 {
00279 return vp56_rac_get_prob(c, 128);
00280 }
00281
00282 static av_unused int vp56_rac_gets(VP56RangeCoder *c, int bits)
00283 {
00284 int value = 0;
00285
00286 while (bits--) {
00287 value = (value << 1) | vp56_rac_get(c);
00288 }
00289
00290 return value;
00291 }
00292
00293 static av_unused int vp8_rac_get_uint(VP56RangeCoder *c, int bits)
00294 {
00295 int value = 0;
00296
00297 while (bits--) {
00298 value = (value << 1) | vp8_rac_get(c);
00299 }
00300
00301 return value;
00302 }
00303
00304
00305 static av_unused int vp8_rac_get_sint(VP56RangeCoder *c, int bits)
00306 {
00307 int v;
00308
00309 if (!vp8_rac_get(c))
00310 return 0;
00311
00312 v = vp8_rac_get_uint(c, bits);
00313
00314 if (vp8_rac_get(c))
00315 v = -v;
00316
00317 return v;
00318 }
00319
00320
00321 static av_unused int vp56_rac_gets_nn(VP56RangeCoder *c, int bits)
00322 {
00323 int v = vp56_rac_gets(c, 7) << 1;
00324 return v + !v;
00325 }
00326
00327 static av_unused int vp8_rac_get_nn(VP56RangeCoder *c)
00328 {
00329 int v = vp8_rac_get_uint(c, 7) << 1;
00330 return v + !v;
00331 }
00332
00333 static av_always_inline
00334 int vp56_rac_get_tree(VP56RangeCoder *c,
00335 const VP56Tree *tree,
00336 const uint8_t *probs)
00337 {
00338 while (tree->val > 0) {
00339 if (vp56_rac_get_prob(c, probs[tree->prob_idx]))
00340 tree += tree->val;
00341 else
00342 tree++;
00343 }
00344 return -tree->val;
00345 }
00346
00347
00348
00349 static av_always_inline int vp8_rac_get_tree(VP56RangeCoder *c, const int8_t (*tree)[2],
00350 const uint8_t *probs)
00351 {
00352 int i = 0;
00353
00354 do {
00355 i = tree[i][vp56_rac_get_prob(c, probs[i])];
00356 } while (i > 0);
00357
00358 return -i;
00359 }
00360
00361
00362 static av_always_inline int vp8_rac_get_coeff(VP56RangeCoder *c, const uint8_t *prob)
00363 {
00364 int v = 0;
00365
00366 do {
00367 v = (v<<1) + vp56_rac_get_prob(c, *prob++);
00368 } while (*prob);
00369
00370 return v;
00371 }
00372
00373 #endif