FFmpeg: libavcodec/vc1.h Source File

00001 /*
00002  * VC-1 and WMV3 decoder
00003  * Copyright (c) 2006-2007 Konstantin Shishkov
00004  * Partly based on vc9.c (c) 2005 Anonymous, Alex Beregszaszi, Michael Niedermayer
00005  *
00006  * This file is part of FFmpeg.
00007  *
00008  * FFmpeg is free software; you can redistribute it and/or
00009  * modify it under the terms of the GNU Lesser General Public
00010  * License as published by the Free Software Foundation; either
00011  * version 2.1 of the License, or (at your option) any later version.
00012  *
00013  * FFmpeg is distributed in the hope that it will be useful,
00014  * but WITHOUT ANY WARRANTY; without even the implied warranty of
00015  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00016  * Lesser General Public License for more details.
00017  *
00018  * You should have received a copy of the GNU Lesser General Public
00019  * License along with FFmpeg; if not, write to the Free Software
00020  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
00021  */
00022 
00023 #ifndef AVCODEC_VC1_H
00024 #define AVCODEC_VC1_H
00025 
00026 #include "avcodec.h"
00027 #include "mpegvideo.h"
00028 #include "intrax8.h"
00029 #include "vc1dsp.h"
00030 
00033 enum VC1Code {
00034     VC1_CODE_RES0       = 0x00000100,
00035     VC1_CODE_ENDOFSEQ   = 0x0000010A,
00036     VC1_CODE_SLICE,
00037     VC1_CODE_FIELD,
00038     VC1_CODE_FRAME,
00039     VC1_CODE_ENTRYPOINT,
00040     VC1_CODE_SEQHDR,
00041 };
00043 
00044 #define IS_MARKER(x) (((x) & ~0xFF) == VC1_CODE_RES0)
00045 
00048 enum Profile {
00049     PROFILE_SIMPLE,
00050     PROFILE_MAIN,
00051     PROFILE_COMPLEX, 
00052     PROFILE_ADVANCED
00053 };
00055 
00058 enum QuantMode {
00059     QUANT_FRAME_IMPLICIT,    
00060     QUANT_FRAME_EXPLICIT,    
00061     QUANT_NON_UNIFORM,       
00062     QUANT_UNIFORM            
00063 };
00065 
00068 enum DQProfile {
00069     DQPROFILE_FOUR_EDGES,
00070     DQPROFILE_DOUBLE_EDGES,
00071     DQPROFILE_SINGLE_EDGE,
00072     DQPROFILE_ALL_MBS
00073 };
00075 
00079 enum DQSingleEdge {
00080     DQSINGLE_BEDGE_LEFT,
00081     DQSINGLE_BEDGE_TOP,
00082     DQSINGLE_BEDGE_RIGHT,
00083     DQSINGLE_BEDGE_BOTTOM
00084 };
00086 
00089 enum DQDoubleEdge {
00090     DQDOUBLE_BEDGE_TOPLEFT,
00091     DQDOUBLE_BEDGE_TOPRIGHT,
00092     DQDOUBLE_BEDGE_BOTTOMRIGHT,
00093     DQDOUBLE_BEDGE_BOTTOMLEFT
00094 };
00096 
00099 enum MVModes {
00100     MV_PMODE_1MV_HPEL_BILIN,
00101     MV_PMODE_1MV,
00102     MV_PMODE_1MV_HPEL,
00103     MV_PMODE_MIXED_MV,
00104     MV_PMODE_INTENSITY_COMP
00105 };
00107 
00110 enum MBModesIntfr {
00111     MV_PMODE_INTFR_1MV,
00112     MV_PMODE_INTFR_2MV_FIELD,
00113     MV_PMODE_INTFR_2MV,
00114     MV_PMODE_INTFR_4MV_FIELD,
00115     MV_PMODE_INTFR_4MV,
00116     MV_PMODE_INTFR_INTRA,
00117 };
00119 
00122 enum BMVTypes {
00123     BMV_TYPE_BACKWARD,
00124     BMV_TYPE_FORWARD,
00125     BMV_TYPE_INTERPOLATED,
00126     BMV_TYPE_DIRECT
00127 };
00129 
00132 enum TransformTypes {
00133     TT_8X8,
00134     TT_8X4_BOTTOM,
00135     TT_8X4_TOP,
00136     TT_8X4,         // both halves
00137     TT_4X8_RIGHT,
00138     TT_4X8_LEFT,
00139     TT_4X8,         // both halves
00140     TT_4X4
00141 };
00143 
00144 enum CodingSet {
00145     CS_HIGH_MOT_INTRA = 0,
00146     CS_HIGH_MOT_INTER,
00147     CS_LOW_MOT_INTRA,
00148     CS_LOW_MOT_INTER,
00149     CS_MID_RATE_INTRA,
00150     CS_MID_RATE_INTER,
00151     CS_HIGH_RATE_INTRA,
00152     CS_HIGH_RATE_INTER
00153 };
00154 
00157 enum COTypes {
00158     CONDOVER_NONE = 0,
00159     CONDOVER_ALL,
00160     CONDOVER_SELECT
00161 };
00163 
00169 enum FrameCodingMode {
00170     PROGRESSIVE = 0,    
00171     ILACE_FRAME,        
00172     ILACE_FIELD         
00173 };
00174 
00179 typedef struct VC1Context{
00180     MpegEncContext s;
00181     IntraX8Context x8;
00182     VC1DSPContext vc1dsp;
00183 
00184     int bits;
00185 
00188     int res_sprite;       
00189     int res_y411;         
00190     int res_x8;           
00191     int multires;         
00192     int res_fasttx;       
00193     int res_transtab;     
00194     int rangered;         
00195 
00196     int res_rtm_flag;     
00197     int reserved;         
00198 
00199 
00202     int level;            
00203     int chromaformat;     
00204     int postprocflag;     
00205     int broadcast;        
00206     int interlace;        
00207     int tfcntrflag;       
00208     int panscanflag;      
00209     int refdist_flag;     
00210     int extended_dmv;     
00211     int color_prim;       
00212     int transfer_char;    
00213     int matrix_coef;      
00214     int hrd_param_flag;   
00215 
00216     int psf;              
00217 
00218 
00223     int profile;          
00224     int frmrtq_postproc;  
00225     int bitrtq_postproc;  
00226     int fastuvmc;         
00227     int extended_mv;      
00228     int dquant;           
00229     int vstransform;      
00230     int overlap;          
00231     int quantizer_mode;   
00232     int finterpflag;      
00233 
00234 
00237     uint8_t mv_mode;             
00238     uint8_t mv_mode2;            
00239     int k_x;                     
00240     int k_y;                     
00241     int range_x, range_y;        
00242     uint8_t pq, altpq;           
00243     uint8_t zz_8x8[4][64];       
00244     int left_blk_sh, top_blk_sh; 
00245     const uint8_t* zz_8x4;       
00246     const uint8_t* zz_4x8;       
00247 
00249     uint8_t dquantfrm;
00250     uint8_t dqprofile;
00251     uint8_t dqsbedge;
00252     uint8_t dqbilevel;
00254 
00258     int c_ac_table_index;    
00259     int y_ac_table_index;    
00260 
00261     int ttfrm;               
00262     uint8_t ttmbf;           
00263     int *ttblk_base, *ttblk; 
00264     int codingset;           
00265     int codingset2;          
00266     int pqindex;             
00267     int a_avail, c_avail;
00268     uint8_t *mb_type_base, *mb_type[3];
00269 
00270 
00273     uint8_t lumscale;
00274     uint8_t lumshift;
00276     int16_t bfraction;    
00277     uint8_t halfpq;       
00278     uint8_t respic;       
00279     int buffer_fullness;  
00280 
00286     uint8_t mvrange;                
00287     uint8_t pquantizer;             
00288     VLC *cbpcy_vlc;                 
00289     int tt_index;                   
00290     uint8_t* mv_type_mb_plane;      
00291     uint8_t* direct_mb_plane;       
00292     uint8_t* forward_mb_plane;      
00293     int mv_type_is_raw;             
00294     int dmb_is_raw;                 
00295     int fmb_is_raw;                 
00296     int skip_is_raw;                
00297     uint8_t luty[256], lutuv[256];  
00298     int use_ic;                     
00299     int rnd;                        
00300 
00303     uint8_t rangeredfrm;            
00304     uint8_t interpfrm;
00306 
00309     enum FrameCodingMode fcm;
00310     uint8_t numpanscanwin;
00311     uint8_t tfcntr;
00312     uint8_t rptfrm, tff, rff;
00313     uint16_t topleftx;
00314     uint16_t toplefty;
00315     uint16_t bottomrightx;
00316     uint16_t bottomrighty;
00317     uint8_t uvsamp;
00318     uint8_t postproc;
00319     int hrd_num_leaky_buckets;
00320     uint8_t bit_rate_exponent;
00321     uint8_t buffer_size_exponent;
00322     uint8_t* acpred_plane;       
00323     int acpred_is_raw;
00324     uint8_t* over_flags_plane;   
00325     int overflg_is_raw;
00326     uint8_t condover;
00327     uint16_t *hrd_rate, *hrd_buffer;
00328     uint8_t *hrd_fullness;
00329     uint8_t range_mapy_flag;
00330     uint8_t range_mapuv_flag;
00331     uint8_t range_mapy;
00332     uint8_t range_mapuv;
00334 
00336     uint8_t dmvrange;   
00337     int fourmvswitch;
00338     int intcomp;
00339     uint8_t lumscale2;  
00340     uint8_t lumshift2;
00341     uint8_t luty2[256], lutuv2[256]; // lookup tables used for intensity compensation
00342     VLC* mbmode_vlc;
00343     VLC* imv_vlc;
00344     VLC* twomvbp_vlc;
00345     VLC* fourmvbp_vlc;
00346     uint8_t twomvbp;
00347     uint8_t fourmvbp;
00348     uint8_t* fieldtx_plane;
00349     int fieldtx_is_raw;
00350     int8_t zzi_8x8[64];
00351     uint8_t *blk_mv_type_base, *blk_mv_type;    
00352     uint8_t *mv_f_base, *mv_f[2];               
00353     uint8_t *mv_f_last_base, *mv_f_last[2];
00354     uint8_t *mv_f_next_base, *mv_f_next[2];
00355     int field_mode;         
00356     int fptype;
00357     int second_field;
00358     int refdist;            
00359     int numref;             
00360                             // 0 corresponds to 1 and 1 corresponds to 2 references
00361     int reffield;           
00362                             // field to use among the two fields from previous frame
00363     int intcompfield;       
00364                             // 0: both fields, 1: bottom field, 2: top field
00365     int cur_field_type;     
00366     int ref_field_type[2];  
00367     int blocks_off, mb_off;
00368     int qs_last;            
00369     int bmvtype;
00370     int frfd, brfd;         
00371     int pic_header_flag;
00372 
00375     int new_sprite;
00376     int two_sprites;
00377     AVFrame sprite_output_frame;
00378     int output_width, output_height, sprite_width, sprite_height;
00379     uint8_t* sr_rows[2][2];      
00380 
00381 
00382     int p_frame_skipped;
00383     int bi_type;
00384     int x8_type;
00385 
00386     DCTELEM (*block)[6][64];
00387     int n_allocated_blks, cur_blk_idx, left_blk_idx, topleft_blk_idx, top_blk_idx;
00388     uint32_t *cbp_base, *cbp;
00389     uint8_t *is_intra_base, *is_intra;
00390     int16_t (*luma_mv_base)[2], (*luma_mv)[2];
00391     uint8_t bfraction_lut_index; 
00392     uint8_t broken_link;         
00393     uint8_t closed_entry;        
00394 
00395     int parse_only;              
00396 
00397     int warn_interlaced;
00398 } VC1Context;
00399 
00403 static av_always_inline const uint8_t* find_next_marker(const uint8_t *src, const uint8_t *end)
00404 {
00405     uint32_t mrk = 0xFFFFFFFF;
00406 
00407     if (end-src < 4)
00408         return end;
00409     while (src < end) {
00410         mrk = (mrk << 8) | *src++;
00411         if (IS_MARKER(mrk))
00412             return src - 4;
00413     }
00414     return end;
00415 }
00416 
00417 static av_always_inline int vc1_unescape_buffer(const uint8_t *src, int size, uint8_t *dst)
00418 {
00419     int dsize = 0, i;
00420 
00421     if (size < 4) {
00422         for (dsize = 0; dsize < size; dsize++)
00423             *dst++ = *src++;
00424         return size;
00425     }
00426     for (i = 0; i < size; i++, src++) {
00427         if (src[0] == 3 && i >= 2 && !src[-1] && !src[-2] && i < size-1 && src[1] < 4) {
00428             dst[dsize++] = src[1];
00429             src++;
00430             i++;
00431         } else
00432             dst[dsize++] = *src;
00433     }
00434     return dsize;
00435 }
00436 
00444 int vc1_decode_sequence_header(AVCodecContext *avctx, VC1Context *v, GetBitContext *gb);
00445 
00446 int vc1_decode_entry_point(AVCodecContext *avctx, VC1Context *v, GetBitContext *gb);
00447 
00448 int vc1_parse_frame_header    (VC1Context *v, GetBitContext *gb);
00449 int vc1_parse_frame_header_adv(VC1Context *v, GetBitContext *gb);
00450 
00451 #endif /* AVCODEC_VC1_H */