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 BMVTypes {
00111     BMV_TYPE_BACKWARD,
00112     BMV_TYPE_FORWARD,
00113     BMV_TYPE_INTERPOLATED
00114 };
00116 
00119 enum TransformTypes {
00120     TT_8X8,
00121     TT_8X4_BOTTOM,
00122     TT_8X4_TOP,
00123     TT_8X4, //Both halves
00124     TT_4X8_RIGHT,
00125     TT_4X8_LEFT,
00126     TT_4X8, //Both halves
00127     TT_4X4
00128 };
00130 
00131 enum CodingSet {
00132     CS_HIGH_MOT_INTRA = 0,
00133     CS_HIGH_MOT_INTER,
00134     CS_LOW_MOT_INTRA,
00135     CS_LOW_MOT_INTER,
00136     CS_MID_RATE_INTRA,
00137     CS_MID_RATE_INTER,
00138     CS_HIGH_RATE_INTRA,
00139     CS_HIGH_RATE_INTER
00140 };
00141 
00144 enum COTypes {
00145     CONDOVER_NONE = 0,
00146     CONDOVER_ALL,
00147     CONDOVER_SELECT
00148 };
00150 
00151 
00156 typedef struct VC1Context{
00157     MpegEncContext s;
00158     IntraX8Context x8;
00159     VC1DSPContext vc1dsp;
00160 
00161     int bits;
00162 
00165     int res_sprite;       
00166     int res_y411;         
00167     int res_x8;           
00168     int multires;         
00169     int res_fasttx;       
00170     int res_transtab;     
00171     int rangered;         
00172 
00173     int res_rtm_flag;     
00174     int reserved;         
00175 
00176 
00179     int level;            
00180     int chromaformat;     
00181     int postprocflag;     
00182     int broadcast;        
00183     int interlace;        
00184     int tfcntrflag;       
00185     int panscanflag;      
00186     int refdist_flag;     
00187     int extended_dmv;     
00188     int color_prim;       
00189     int transfer_char;    
00190     int matrix_coef;      
00191     int hrd_param_flag;   
00192 
00193     int psf;              
00194 
00195 
00200     int profile;          
00201     int frmrtq_postproc;  
00202     int bitrtq_postproc;  
00203     int fastuvmc;         
00204     int extended_mv;      
00205     int dquant;           
00206     int vstransform;      
00207     int overlap;          
00208     int quantizer_mode;   
00209     int finterpflag;      
00210 
00211 
00214     uint8_t mv_mode;      
00215     uint8_t mv_mode2;     
00216     int k_x;              
00217     int k_y;              
00218     int range_x, range_y; 
00219     uint8_t pq, altpq;    
00220     uint8_t zz_8x8[4][64];
00221     int left_blk_sh, top_blk_sh; 
00222     const uint8_t* zz_8x4;
00223     const uint8_t* zz_4x8;
00224 
00226     uint8_t dquantfrm;
00227     uint8_t dqprofile;
00228     uint8_t dqsbedge;
00229     uint8_t dqbilevel;
00231 
00235     int c_ac_table_index; 
00236     int y_ac_table_index; 
00237 
00238     int ttfrm;            
00239     uint8_t ttmbf;        
00240     int *ttblk_base, *ttblk; 
00241     int codingset;        
00242     int codingset2;       
00243     int pqindex;          
00244     int a_avail, c_avail;
00245     uint8_t *mb_type_base, *mb_type[3];
00246 
00247 
00250     uint8_t lumscale;
00251     uint8_t lumshift;
00253     int16_t bfraction;    
00254     uint8_t halfpq;       
00255     uint8_t respic;       
00256     int buffer_fullness;  
00257 
00263     uint8_t mvrange;
00264     uint8_t pquantizer;           
00265     VLC *cbpcy_vlc;               
00266     int tt_index;                 
00267     uint8_t* mv_type_mb_plane;    
00268     uint8_t* direct_mb_plane;     
00269     int mv_type_is_raw;           
00270     int dmb_is_raw;               
00271     int skip_is_raw;              
00272     uint8_t luty[256], lutuv[256]; // lookup tables used for intensity compensation
00273     int use_ic;                   
00274     int rnd;                      
00275 
00278     uint8_t rangeredfrm; 
00279     uint8_t interpfrm;
00281 
00284     uint8_t fcm; 
00285     uint8_t numpanscanwin;
00286     uint8_t tfcntr;
00287     uint8_t rptfrm, tff, rff;
00288     uint16_t topleftx;
00289     uint16_t toplefty;
00290     uint16_t bottomrightx;
00291     uint16_t bottomrighty;
00292     uint8_t uvsamp;
00293     uint8_t postproc;
00294     int hrd_num_leaky_buckets;
00295     uint8_t bit_rate_exponent;
00296     uint8_t buffer_size_exponent;
00297     uint8_t* acpred_plane;       
00298     int acpred_is_raw;
00299     uint8_t* over_flags_plane;   
00300     int overflg_is_raw;
00301     uint8_t condover;
00302     uint16_t *hrd_rate, *hrd_buffer;
00303     uint8_t *hrd_fullness;
00304     uint8_t range_mapy_flag;
00305     uint8_t range_mapuv_flag;
00306     uint8_t range_mapy;
00307     uint8_t range_mapuv;
00309 
00312     int new_sprite;
00313     int two_sprites;
00315 
00316     int p_frame_skipped;
00317     int bi_type;
00318     int x8_type;
00319 
00320     DCTELEM (*block)[6][64];
00321     int n_allocated_blks, cur_blk_idx, left_blk_idx, topleft_blk_idx, top_blk_idx;
00322     uint32_t *cbp_base, *cbp;
00323     uint8_t *is_intra_base, *is_intra;
00324     int16_t (*luma_mv_base)[2], (*luma_mv)[2];
00325     uint8_t bfraction_lut_index;
00326     uint8_t broken_link;        
00327     uint8_t closed_entry;       
00328 
00329     int parse_only;             
00330 
00331     int warn_interlaced;
00332 } VC1Context;
00333 
00337 static av_always_inline const uint8_t* find_next_marker(const uint8_t *src, const uint8_t *end)
00338 {
00339     uint32_t mrk = 0xFFFFFFFF;
00340 
00341     if(end-src < 4) return end;
00342     while(src < end){
00343         mrk = (mrk << 8) | *src++;
00344         if(IS_MARKER(mrk))
00345             return src-4;
00346     }
00347     return end;
00348 }
00349 
00350 static av_always_inline int vc1_unescape_buffer(const uint8_t *src, int size, uint8_t *dst)
00351 {
00352     int dsize = 0, i;
00353 
00354     if(size < 4){
00355         for(dsize = 0; dsize < size; dsize++) *dst++ = *src++;
00356         return size;
00357     }
00358     for(i = 0; i < size; i++, src++) {
00359         if(src[0] == 3 && i >= 2 && !src[-1] && !src[-2] && i < size-1 && src[1] < 4) {
00360             dst[dsize++] = src[1];
00361             src++;
00362             i++;
00363         } else
00364             dst[dsize++] = *src;
00365     }
00366     return dsize;
00367 }
00368 
00376 int vc1_decode_sequence_header(AVCodecContext *avctx, VC1Context *v, GetBitContext *gb);
00377 
00378 int vc1_decode_entry_point(AVCodecContext *avctx, VC1Context *v, GetBitContext *gb);
00379 
00380 int vc1_parse_frame_header    (VC1Context *v, GetBitContext *gb);
00381 int vc1_parse_frame_header_adv(VC1Context *v, GetBitContext *gb);
00382 
00383 #endif /* AVCODEC_VC1_H */