From 21aa398f3d9d949b81fc0888efcae68f83986ce8 Mon Sep 17 00:00:00 2001 From: Alex Beregszaszi Date: Sat, 22 Jan 2005 17:52:01 +0000 Subject: [PATCH] preliminary vc9 bitstream decoder, committing to make syncing and team-work on it easier Originally committed as revision 3861 to svn://svn.ffmpeg.org/ffmpeg/trunk --- libavcodec/Makefile | 2 +- libavcodec/avcodec.h | 4 + libavcodec/vc9.c | 1723 ++++++++++++++++++++++++++++++++++++++++++ libavcodec/vc9data.h | 373 +++++++++ 4 files changed, 2101 insertions(+), 1 deletion(-) create mode 100644 libavcodec/vc9.c create mode 100644 libavcodec/vc9data.h diff --git a/libavcodec/Makefile b/libavcodec/Makefile index fe468ca5ad..e95b888a34 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -21,7 +21,7 @@ OBJS= bitstream.o utils.o mem.o allcodecs.o \ msvideo1.o vqavideo.o idcinvideo.o adx.o rational.o faandct.o 8bps.o \ smc.o parser.o flicvideo.o truemotion1.o vmdav.o lcl.o qtrle.o g726.o \ flac.o vp3dsp.o integer.o snow.o tscc.o sonic.o ulti.o h264idct.o \ - qdrw.o xl.o rangecoder.o png.o pnm.o qpeg.o + qdrw.o xl.o rangecoder.o png.o pnm.o qpeg.o vc9.o AMROBJS= ifeq ($(AMR_NB),yes) diff --git a/libavcodec/avcodec.h b/libavcodec/avcodec.h index 921ea68b29..8cf6500ca3 100644 --- a/libavcodec/avcodec.h +++ b/libavcodec/avcodec.h @@ -102,6 +102,8 @@ enum CodecID { CODEC_ID_FFVHUFF, CODEC_ID_RV30, CODEC_ID_RV40, + CODEC_ID_VC9, + CODEC_ID_WMV3, /* various pcm "codecs" */ @@ -1886,6 +1888,8 @@ extern AVCodec msmpeg4v2_decoder; extern AVCodec msmpeg4v3_decoder; extern AVCodec wmv1_decoder; extern AVCodec wmv2_decoder; +extern AVCodec vc9_decoder; +extern AVCodec wmv3_decoder; extern AVCodec mpeg1video_decoder; extern AVCodec mpeg2video_decoder; extern AVCodec mpegvideo_decoder; diff --git a/libavcodec/vc9.c b/libavcodec/vc9.c new file mode 100644 index 0000000000..8f469d074c --- /dev/null +++ b/libavcodec/vc9.c @@ -0,0 +1,1723 @@ +/* + * VC-9 and WMV3 decoder + * Copyright (c) 2005 Anonymous + * Copyright (c) 2005 Alex Beregszaszi + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + * + */ + +/** + * @file vc9.c + * VC-9 and WMV3 decoder + * + * TODO: Norm-6 bitplane imode, most AP stuff, optimize, all of MB layer :) + * TODO: use MPV_ !! + */ +#include "common.h" +#include "dsputil.h" +#include "avcodec.h" +#include "mpegvideo.h" +#include "vc9data.h" + +/* Some inhibiting stuff */ +#define HAS_ADVANCED_PROFILE 1 +#define TRACE 1 + +#if TRACE +# define INIT_VLC(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ + codes, codes_wrap, codes_size, use_static) \ + if (init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ + codes, codes_wrap, codes_size, use_static) < 0) \ + { \ + av_log(v->avctx, AV_LOG_ERROR, "Error for " # vlc " (%i)\n", i); \ + return -1; \ + } +#else +# define INIT_VLC(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ + codes, codes_wrap, codes_size, use_static) \ + init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ + codes, codes_wrap, codes_size, use_static) +#endif + +#define PROFILE_SIMPLE 0 +#define PROFILE_MAIN 1 +#define PROFILE_ADVANCED 3 + +#define QUANT_FRAME_IMPLICIT 0 +#define QUANT_FRAME_EXPLICIT 1 +#define QUANT_NON_UNIFORM 2 +#define QUANT_UNIFORM 3 + +/* Where quant can be changed */ +#define DQPROFILE_FOUR_EDGES 0 +#define DQPROFILE_DOUBLE_EDGES 1 +#define DQPROFILE_SINGLE_EDGE 2 +#define DQPROFILE_ALL_MBS 3 + +/* Which edge is quantized with ALTPQUANT */ +#define DQSINGLE_BEDGE_LEFT 0 +#define DQSINGLE_BEDGE_TOP 1 +#define DQSINGLE_BEDGE_RIGHT 2 +#define DQSINGLE_BEDGE_BOTTOM 3 + +/* Which pair of edges is quantized with ALTPQUANT */ +#define DQDOUBLE_BEDGE_TOPLEFT 0 +#define DQDOUBLE_BEDGE_TOPRIGHT 1 +#define DQDOUBLE_BEDGE_BOTTOMRIGHT 2 +#define DQDOUBLE_BEDGE_BOTTOMLEFT 3 + + +/* Start Codes */ +#define SEQ_SC 0x00000010F /* Sequence Start Code */ +#define SEQ_EC 0x00000000A /* Sequence End code */ +#define SEQ_HDR /* Sequence Header */ +#define ENTRY_SC 0x00000010E /* Entry Point Start Code */ +#define ENTRY_HDR /* Entry Point Header */ +#define FRM_SC 0x00000010D /* Frame Start Code */ +#define FRM_DAT /* Frame Data (includes a Frame Header) */ +#define FLD_SC 0x00000010C /* Field Start Code */ +#define FLD1_DAT /* Field 1 Data (includes a Frame Header) */ +#define FLD2_DAT /* Field 2 Data (includes a Field Header) */ +#define SLC_SC 0x00000010B /* Slice Start Code */ +#define SLC_HDR /* Slice Header */ +#define SLC_DAT /* Slice Data (FrH or FiH possible) */ + +/* MV P modes */ +#define MV_PMODE_1MV_HPEL_BILIN 0 +#define MV_PMODE_1MV 1 +#define MV_PMODE_1MV_HPEL 2 +#define MV_PMODE_MIXED_MV 3 +#define MV_PMODE_INTENSITY_COMP 4 + +/* MV P mode - the 5th element is only used for mode 1 */ +static const uint8_t mv_pmode_table[2][5] = { + { MV_PMODE_1MV_HPEL_BILIN, MV_PMODE_1MV, MV_PMODE_1MV_HPEL, MV_PMODE_MIXED_MV, MV_PMODE_INTENSITY_COMP }, + { MV_PMODE_1MV, MV_PMODE_MIXED_MV, MV_PMODE_1MV_HPEL, MV_PMODE_1MV_HPEL_BILIN, MV_PMODE_INTENSITY_COMP } +}; + +/* One more frame type */ +#define BI_TYPE 7 + +/* FIXME Worse than ugly */ +static const int fps_nr[5] = { 24, 25, 30, 50, 60 }, + fps_dr[2] = { 1000, 1001 }; +static const uint8_t pquant_table[3][32] = { + { /* Implicit quantizer */ + 0, 1, 2, 3, 4, 5, 6, 7, 8, 6, 7, 8, 9, 10, 11, 12, + 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 27, 29, 31 + }, + { /* Explicit quantizer, pquantizer uniform */ + 0, 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 + }, + { /* Explicit quantizer, pquantizer non-uniform */ + 0, 1, 1, 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, 29, 31 + } +}; + +// FIXME move this into the context +#define VC9_BFRACTION_VLC_BITS 7 +static VLC vc9_bfraction_vlc; +#define VC9_IMODE_VLC_BITS 4 +static VLC vc9_imode_vlc; +#define VC9_NORM2_VLC_BITS 3 +static VLC vc9_norm2_vlc; +/* Could be optimized, one table only needs 8 bits */ +#define VC9_TTMB_VLC_BITS 12 +static VLC vc9_ttmb_vlc[3]; +#define VC9_MV_DIFF_VLC_BITS 15 +static VLC vc9_mv_diff_vlc[4]; +#define VC9_CBPCY_I_VLC_BITS 13 +static VLC vc9_cbpcy_i_vlc; +#define VC9_CBPCY_P_VLC_BITS 14 +static VLC vc9_cbpcy_p_vlc[4]; +#define VC9_4MV_BLOCK_PATTERN_VLC_BITS 6 +static VLC vc9_4mv_block_pattern_vlc[4]; +#define VC9_LUMA_DC_VLC_BITS 26 +static VLC vc9_luma_dc_vlc[2]; + +typedef struct VC9Context{ + /* No MpegEnc context, might be good to use it */ + GetBitContext gb; + AVCodecContext *avctx; + + /***************************/ + /* Sequence Header */ + /***************************/ + /* Simple/Main Profile */ + int res_sm; //reserved, 2b + int res_x8; //reserved + int multires; //frame-level RESPIC syntax element present + int res_fasttx; //always 1 + int res_transtab; //always 0 + int syncmarker; //Sync markers presents + int rangered; //RANGEREDFRM (range reduction) syntax element present + int res_rtm_flag; //reserved, set to 1 + int reserved; //duh + +#if HAS_ADVANCED_PROFILE + /* Advanced Profile */ + int level; //3 + int chromaformat; //2 + int postprocflag; //frame-based processing use + int broadcast; //TFF/RFF present + int interlace; //Progressive/interlaced (RPTFTM syntax element) + int tfcntrflag; //TFCNTR present + int panscanflag; //NUMPANSCANWIN, TOPLEFT{X,Y}, BOTRIGHT{X,Y} presents + int extended_dmv; + int color_prim; //8 + int transfer_char; //8 + int matrix_coef; //8 + int hrd_param_flag; +#endif + + /* All Profiles */ + /* TODO: move all int to flags */ + int profile; //2 + int frmrtq_postproc; //3 + int bitrtq_postproc; //5 + int loopfilter; + int fastuvmc; //Rounding of qpel vector to hpel ? (not in Simple) + int extended_mv; //Ext MV in P/B (not in Simple) + int dquant; //Q varies with MBs, 2bits (not in Simple) + int vstransform; //variable-size transform46 + + int overlap; //overlapped transforms in use + int quantizer_mode; //2, quantizer mode used for sequence, see QUANT_* + int finterpflag; //INTERPFRM present + + + /*****************************/ + /* Frame decoding */ + /*****************************/ + /* All profiles */ + uint8_t mv_mode, mv_mode2; /* MV coding mode */ + uint8_t pict_type; /* Picture type, mapped on MPEG types */ + uint8_t pq, altpq; /* Quantizers */ + uint8_t dquantfrm, dqprofile, dqsbedge, dqbilevel; /* pquant parameters */ + int width_mb, height_mb; + int tile; /* 3x2 if (width_mb%3) else 2x3 */ + int transacfrm2, transacfrm, transacdctab; //1bit elements + uint8_t ttmbf, ttfrm; /* Transform type */ + uint8_t lumscale, lumshift; /* Luma compensation parameters */ + int16_t bfraction; /* Relative position % anchors=> how to scale MVs */ + uint8_t halfpq; /* Uniform quant over image and qp+.5 */ + uint8_t respic; + /* Ranges: + * 0 -> [-64n 63.f] x [-32, 31.f] + * 1 -> [-128, 127.f] x [-64, 63.f] + * 2 -> [-512, 511.f] x [-128, 127.f] + * 3 -> [-1024, 1023.f] x [-256, 255.f] + */ + uint8_t mvrange; + uint8_t pquantizer; + VLC *cbpcy_vlc /* Current CBPCY VLC table */, + *mv_diff_vlc /* Current MV Diff VLC table */, + *ttmb_vlc /* Current MB Transform Type VLC table */; + uint8_t *mv_type_mb_plane; /* bitplane for mv_type == "raw" */ + uint8_t *skip_mb_plane, /* bitplane for skipped MBs */ + *direct_mb_plane; /* bitplane for "direct" MBs */ + + /* S/M only ? */ + uint8_t rangeredfrm, interpfrm; + +#if HAS_ADVANCED_PROFILE + /* Advanced */ + uint8_t fcm; //0->Progressive, 2->Frame-Interlace, 3->Field-Interlace + uint8_t numpanscanwin; + uint8_t tfcntr; + uint8_t rptfrm, tff, rff; + uint8_t topleftx; + uint8_t toplefty; + uint8_t bottomrightx; + uint8_t bottomrighty; + uint8_t rndctrl; + uint8_t uvsamp; + uint8_t postproc; + int hrd_num_leaky_buckets; + uint8_t bit_rate_exponent; + uint8_t buffer_size_exponent; + uint8_t *ac_pred_plane; + uint8_t *over_flags_plane; + uint16_t *hrd_rate, *hrd_buffer; +#endif +} VC9Context; + +/* FIXME Slow and ugly */ +static int get_prefix(GetBitContext *gb, int stop, int len) +{ + int i = 0, tmp = !stop; + + while (i != len && tmp != stop) + { + tmp = get_bits(gb, 1); + i++; + } + return i; +} + +static int init_common(VC9Context *v) +{ + static int done = 0; + int i; + + v->mv_type_mb_plane = v->direct_mb_plane = v->skip_mb_plane = NULL; + v->pq = -1; +#if HAS_ADVANCED_PROFILE + v->ac_pred_plane = v->over_flags_plane = NULL; + v->hrd_rate = v->hrd_buffer = NULL; +#endif + if(!done) + { + done = 1; + INIT_VLC(&vc9_bfraction_vlc, VC9_BFRACTION_VLC_BITS, 23, + vc9_bfraction_bits, 1, 1, + vc9_bfraction_codes, 1, 1, 1); + INIT_VLC(&vc9_norm2_vlc, VC9_NORM2_VLC_BITS, 4, + vc9_norm2_bits, 1, 1, + vc9_norm2_codes, 1, 1, 1); + INIT_VLC(&vc9_cbpcy_i_vlc, VC9_CBPCY_I_VLC_BITS, 64, + vc9_cbpcy_i_bits, 1, 1, + vc9_cbpcy_i_codes, 2, 2, 1); + INIT_VLC(&vc9_imode_vlc, VC9_IMODE_VLC_BITS, 7, + vc9_imode_bits, 1, 1, + vc9_imode_codes, 1, 1, 1); + for(i=0; i<3; i++) + { + INIT_VLC(&vc9_4mv_block_pattern_vlc[i], VC9_4MV_BLOCK_PATTERN_VLC_BITS, 16, + vc9_4mv_block_pattern_bits[i], 1, 1, + vc9_4mv_block_pattern_codes[i], 1, 1, 1); + INIT_VLC(&vc9_cbpcy_p_vlc[i], VC9_CBPCY_P_VLC_BITS, 64, + vc9_cbpcy_p_bits[i], 1, 1, + vc9_cbpcy_p_codes[i], 2, 2, 1); + } + for (i=0; i<2; i++) + { + INIT_VLC(&vc9_mv_diff_vlc[i], VC9_MV_DIFF_VLC_BITS, 73, + vc9_mv_diff_bits[i], 1, 1, + vc9_mv_diff_codes[i], 2, 2, 1); + INIT_VLC(&vc9_luma_dc_vlc[i], VC9_LUMA_DC_VLC_BITS, 26, + vc9_luma_dc_bits[i], 1, 1, + vc9_luma_dc_codes[i], 4, 4, 1); + INIT_VLC(&vc9_ttmb_vlc[i], VC9_TTMB_VLC_BITS, 16, + vc9_ttmb_bits[i], 1, 1, + vc9_ttmb_codes[i], 2, 2, 1); + } + } + + return 0; +} + +#if HAS_ADVANCED_PROFILE +static int decode_hrd(VC9Context *v, GetBitContext *gb) +{ + int i, num; + + num = get_bits(gb, 5); + + if (v->hrd_rate || num != v->hrd_num_leaky_buckets) + { + av_free(v->hrd_rate); + v->hrd_rate = NULL; + } + if (!v->hrd_rate) v->hrd_rate = av_malloc(num); + if (!v->hrd_rate) return -1; + + if (v->hrd_buffer || num != v->hrd_num_leaky_buckets) + { + av_free(v->hrd_buffer); + v->hrd_buffer = NULL; + } + if (!v->hrd_buffer) v->hrd_buffer = av_malloc(num); + if (!v->hrd_buffer) return -1; + + v->hrd_num_leaky_buckets = num; + + //exponent in base-2 for rate + v->bit_rate_exponent = get_bits(gb, 4); + //exponent in base-2 for buffer_size + v->buffer_size_exponent = get_bits(gb, 4); + + for (i=0; ihrd_rate[i] = get_bits(gb, 16); + if (i && v->hrd_rate[i-1]>=v->hrd_rate[i]) + { + av_log(v, AV_LOG_ERROR, "HDR Rates aren't strictly increasing:" + "%i vs %i\n", v->hrd_rate[i-1], v->hrd_rate[i]); + return -1; + } + v->hrd_buffer[i] = get_bits(gb, 16); + if (i && v->hrd_buffer[i-1]hrd_buffer[i]) + { + av_log(v, AV_LOG_ERROR, "HDR Buffers aren't decreasing:" + "%i vs %i\n", v->hrd_buffer[i-1], v->hrd_buffer[i]); + return -1; + } + } + return 0; +} + +static int decode_advanced_sequence_header(AVCodecContext *avctx, GetBitContext *gb) +{ + VC9Context *v = avctx->priv_data; + int nr, dr, aspect_ratio; + + v->postprocflag = get_bits(gb, 1); + v->broadcast = get_bits(gb, 1); + v->interlace = get_bits(gb, 1); + + v->tfcntrflag = get_bits(gb, 1); + v->finterpflag = get_bits(gb, 1); //common + v->panscanflag = get_bits(gb, 1); + v->reserved = get_bits(gb, 1); + if (v->reserved) + { + av_log(avctx, AV_LOG_ERROR, "RESERVED should be 0 (is %i)\n", + v->reserved); + return -1; + } + if (v->extended_mv) + v->extended_dmv = get_bits(gb, 1); + + if (get_bits(gb, 1) /* pic_size_flag */) + { + avctx->coded_width = get_bits(gb, 12); + avctx->coded_height = get_bits(gb, 12); + if ( get_bits(gb, 1) /* disp_size_flag */) + { + avctx->width = get_bits(gb, 14); + avctx->height = get_bits(gb, 14); + } + + if ( get_bits(gb, 1) /* aspect_ratio_flag */) + { + aspect_ratio = get_bits(gb, 4); //SAR + if (aspect_ratio == 0x0F) //FF_ASPECT_EXTENDED + { + avctx->sample_aspect_ratio.num = get_bits(gb, 8); + avctx->sample_aspect_ratio.den = get_bits(gb, 8); + } + else if (aspect_ratio == 0x0E) + { + av_log(avctx, AV_LOG_DEBUG, "Reserved AR found\n"); + } + else + { + avctx->sample_aspect_ratio = vc9_pixel_aspect[aspect_ratio]; + } + } + } + else + { + avctx->coded_width = avctx->width; + avctx->coded_height = avctx->height; + } + + if ( get_bits(gb, 1) /* framerateflag */) + { + if ( get_bits(gb, 1) /* framerateind */) + { + nr = get_bits(gb, 8); + dr = get_bits(gb, 4); + if (nr<1) + { + av_log(avctx, AV_LOG_ERROR, "0 is forbidden for FRAMERATENR\n"); + return -1; + } + if (nr>5) + { + av_log(avctx, AV_LOG_ERROR, + "Reserved FRAMERATENR %i not handled\n", nr); + } + if (dr<1) + { + av_log(avctx, AV_LOG_ERROR, "0 is forbidden for FRAMERATEDR\n"); + } + if (dr>2) + { + av_log(avctx, AV_LOG_ERROR, + "Reserved FRAMERATEDR %i not handled\n", dr); + } + avctx->frame_rate_base = fps_nr[dr]; + avctx->frame_rate = fps_nr[nr]; + } + else + { + nr = get_bits(gb, 16); + // 0.03125->2048Hz / 0.03125Hz + avctx->frame_rate = 1000000; + avctx->frame_rate_base = 31250*(1+nr); + } + } + + if ( get_bits(gb, 1) /* color_format_flag */) + { + //Chromacity coordinates of color primaries + //like ITU-R BT.709-2, BT.470-2, ... + v->color_prim = get_bits(gb, 8); + if (v->color_prim<1) + { + av_log(avctx, AV_LOG_ERROR, "0 for COLOR_PRIM is reserved\n"); + return -1; + } + if (v->color_prim == 3 || v->color_prim>6) + { + av_log(avctx, AV_LOG_DEBUG, "Reserved COLOR_PRIM %i found\n", + v->color_prim); + return -1; + } + + //Opto-electronic transfer characteristics + v->transfer_char = get_bits(gb, 8); + if (v->transfer_char == 3 || v->transfer_char>8) + { + av_log(avctx, AV_LOG_DEBUG, "Reserved TRANSFERT_CHAR %i found\n", + v->color_prim); + return -1; + } + + //Matrix coefficient for primariev->YCbCr + v->matrix_coef = get_bits(gb, 8); + if (v->matrix_coef < 1) return -1; //forbidden + if ((v->matrix_coef>3 && v->matrix_coef<6) || v->matrix_coef>7) + { + av_log(avctx, AV_LOG_DEBUG, "Reserved MATRIX_COEF %i found\n", + v->color_prim); + return -1; + } + } + + //Hypothetical reference decoder indicator flag + v->hrd_param_flag = get_bits(gb, 1); + if (v->hrd_param_flag) + { + if (decode_hrd(v, gb) < 0) return -1; + } + + av_log(avctx, AV_LOG_DEBUG, "Advanced profile not supported yet\n"); + return -1; +} +#endif + +static int decode_sequence_header(AVCodecContext *avctx, GetBitContext *gb) +{ + VC9Context *v = avctx->priv_data; + + v->profile = get_bits(gb, 2); + av_log(avctx, AV_LOG_DEBUG, "Profile: %i\n", v->profile); + +#if HAS_ADVANCED_PROFILE + if (v->profile > PROFILE_MAIN) + { + v->level = get_bits(gb, 3); + v->chromaformat = get_bits(gb, 2); + if (v->chromaformat != 1) + { + av_log(avctx, AV_LOG_ERROR, + "Only 4:2:0 chroma format supported\n"); + return -1; + } + } + else +#endif + { + v->res_sm = get_bits(gb, 2); //reserved + if (v->res_sm) + { + av_log(avctx, AV_LOG_ERROR, + "Reserved RES_SM=%i is forbidden\n", v->res_sm); + //return -1; + } + } + + // (fps-2)/4 (->30) + v->frmrtq_postproc = get_bits(gb, 3); //common + // (bitrate-32kbps)/64kbps + v->bitrtq_postproc = get_bits(gb, 5); //common + v->loopfilter = get_bits(gb, 1); //common + +#if HAS_ADVANCED_PROFILE + if (v->profile <= PROFILE_MAIN) +#endif + { + v->res_x8 = get_bits(gb, 1); //reserved + if (v->res_x8) + { + av_log(avctx, AV_LOG_ERROR, + "1 for reserved RES_X8 is forbidden\n"); + return -1; + } + v->multires = get_bits(gb, 1); + v->res_fasttx = get_bits(gb, 1); + if (!v->res_fasttx) + { + av_log(avctx, AV_LOG_ERROR, + "0 for reserved RES_FASTTX is forbidden\n"); + //return -1; + } + } + + v->fastuvmc = get_bits(gb, 1); //common + if (!v->profile && !v->fastuvmc) + { + av_log(avctx, AV_LOG_ERROR, + "FASTUVMC unavailable in Simple Profile\n"); + return -1; + } + v->extended_mv = get_bits(gb, 1); //common + if (!v->profile && v->extended_mv) + { + av_log(avctx, AV_LOG_ERROR, + "Extended MVs unavailable in Simple Profile\n"); + return -1; + } + v->dquant = get_bits(gb, 2); //common + v->vstransform = get_bits(gb, 1); //common + +#if HAS_ADVANCED_PROFILE + if (v->profile <= PROFILE_MAIN) +#endif + { + v->res_transtab = get_bits(gb, 1); + if (v->res_transtab) + { + av_log(avctx, AV_LOG_ERROR, + "1 for reserved RES_TRANSTAB is forbidden\n"); + return -1; + } + } + + v->overlap = get_bits(gb, 1); //common + +#if HAS_ADVANCED_PROFILE + if (v->profile <= PROFILE_MAIN) +#endif + { + v->syncmarker = get_bits(gb, 1); + v->rangered = get_bits(gb, 1); + } + + avctx->max_b_frames = get_bits(gb, 3); //common + v->quantizer_mode = get_bits(gb, 2); //common + +#if HAS_ADVANCED_PROFILE + if (v->profile <= PROFILE_MAIN) +#endif + { + v->finterpflag = get_bits(gb, 1); //common + v->res_rtm_flag = get_bits(gb, 1); //reserved + if (!v->res_rtm_flag) + { + av_log(avctx, AV_LOG_ERROR, + "0 for reserved RES_RTM_FLAG is forbidden\n"); + //return -1; + } +#if TRACE + av_log(avctx, AV_LOG_INFO, + "Profile %i:\nfrmrtq_postproc=%i, bitrtq_postproc=%i\n" + "LoopFilter=%i, MultiRes=%i, FastUVMV=%i, Extended MV=%i\n" + "Rangered=%i, VSTransform=%i, Overlap=%i, SyncMarker=%i\n" + "DQuant=%i, Quantizer mode=%i, Max B frames=%i\n", + v->profile, v->frmrtq_postproc, v->bitrtq_postproc, + v->loopfilter, v->multires, v->fastuvmc, v->extended_mv, + v->rangered, v->vstransform, v->overlap, v->syncmarker, + v->dquant, v->quantizer_mode, avctx->max_b_frames + ); +#endif + } +#if HAS_ADVANCED_PROFILE + else decode_advanced_sequence_header(avctx, gb); +#endif +} + + +#if HAS_ADVANCED_PROFILE +/*****************************************************************************/ +/* Entry point decoding (Advanced Profile) */ +/*****************************************************************************/ +static int advanced_entry_point_process(AVCodecContext *avctx, GetBitContext *gb) +{ + VC9Context *v = avctx->priv_data; + int range_mapy_flag, range_mapuv_flag, i; + if (v->profile != PROFILE_ADVANCED) + { + av_log(avctx, AV_LOG_ERROR, + "Entry point are only defined in Advanced Profile!\n"); + return -1; //Only for advanced profile! + } + if (v->hrd_param_flag) + { + //Update buffer fullness + av_log(avctx, AV_LOG_DEBUG, "Buffer fullness update\n"); + for (i=0; ihrd_num_leaky_buckets; i++) + skip_bits(gb, 8); + } + if ((range_mapy_flag = get_bits(gb, 1))) + { + //RANGE_MAPY + av_log(avctx, AV_LOG_DEBUG, "RANGE_MAPY\n"); + skip_bits(gb, 3); + } + if ((range_mapuv_flag = get_bits(gb, 1))) + { + //RANGE_MAPUV + av_log(avctx, AV_LOG_DEBUG, "RANGE_MAPUV\n"); + skip_bits(gb, 3); + } + if (v->panscanflag) + { + //NUMPANSCANWIN + v->numpanscanwin = get_bits(gb, 3); + av_log(avctx, AV_LOG_DEBUG, "NUMPANSCANWIN: %u\n", v->numpanscanwin); + } + return 0; +} +#endif + +/******************************************************************************/ +/* Bitplane decoding */ +/******************************************************************************/ +#define IMODE_RAW 0 +#define IMODE_NORM2 1 +#define IMODE_DIFF2 2 +#define IMODE_NORM6 3 +#define IMODE_DIFF6 4 +#define IMODE_ROWSKIP 5 +#define IMODE_COLSKIP 6 + +//FIXME optimize +static int bitplane_decoding(uint8_t* plane, int width, int height, VC9Context *v) +{ + int imode, x, y, code; + uint8_t invert, *planep = plane; + + invert = get_bits(&v->gb, 1); + imode = get_vlc2(&v->gb, vc9_imode_vlc.table, VC9_IMODE_VLC_BITS, 2); + av_log(v->avctx, AV_LOG_DEBUG, "Bitplane: imode=%i, invert=%i\n", + imode, invert); + + switch (imode) + { + case IMODE_RAW: + for (y=0; ygb, 1)); //-1=0xFF + planep += width; + } + break; + case IMODE_DIFF2: + case IMODE_NORM2: + if ((height*width) & 1) *(++planep) = get_bits(&v->gb, 1); + code = get_vlc2(&v->gb, vc9_norm2_vlc.table, VC9_NORM2_VLC_BITS, 2); + *(++planep) = code&1; //lsb => left + *(++planep) = code&2; //msb => right - this is a bitplane, so only !0 matters + break; + case IMODE_DIFF6: + case IMODE_NORM6: + av_log(v->avctx, AV_LOG_ERROR, "Imode using Norm-6 isn't supported\n"); + return -1; + break; + case IMODE_ROWSKIP: + for (y=0; ygb, 1)) //rowskip + memset(planep, 0, width); + else for (x=0; xgb, 1); + planep += width; + } + break; + case IMODE_COLSKIP: //Teh ugly + for (x=0; xgb, 1)) //colskip + { + for (y=0; ygb, 1); + planep += width; + } + } + } + break; + default: break; + } + + /* Applying diff operator */ + if (imode == IMODE_DIFF2 || imode == IMODE_DIFF2) + { + /* And what about j=0 !? */ + planep = plane + width; + memset(plane, invert, width); + for (y=0; ydquant == 2) + { + pqdiff = get_bits(&v->gb, 3); + if (pqdiff == 7) v->altpq = get_bits(&v->gb, 5); + else v->altpq = v->pq + pqdiff + 1; + } + else + { + v->dquantfrm = get_bits(&v->gb, 1); + if ( v->dquantfrm ) + { + v->dqprofile = get_bits(&v->gb, 2); + switch (v->dqprofile) + { + case DQPROFILE_SINGLE_EDGE: + case DQPROFILE_DOUBLE_EDGES: + v->dqsbedge = get_bits(&v->gb, 2); + break; + case DQPROFILE_ALL_MBS: + v->dqbilevel = get_bits(&v->gb, 1); + default: break; //Forbidden ? + } + if (!v->dqbilevel || v->dqprofile != DQPROFILE_ALL_MBS) + { + pqdiff = get_bits(&v->gb, 3); + if (pqdiff == 7) v->altpq = get_bits(&v->gb, 5); + else v->altpq = v->pq + pqdiff + 1; + } + } + } + return 0; +} + +/*****************************************************************************/ +/* All Profiles picture header decoding specific functions */ +/* Only pro/epilog differs between Simple/Main and Advanced => check caller */ +/*****************************************************************************/ +static int decode_bi_picture_header(VC9Context *v) +{ + /* Very particular case: + - for S/M Profiles, decode_b_picture_header reads BF, + bfraction then determine if this is a BI frame, calling + this function afterwards + - for A Profile, PTYPE already tells so and we can go + directly there + */ + int pqindex; + + /* Read the quantization stuff */ + pqindex = get_bits(&v->gb, 5); + if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) + v->pq = pquant_table[0][pqindex]; + else + { + v->pq = pquant_table[v->quantizer_mode-1][pqindex]; + } + if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); + if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) + v->pquantizer = get_bits(&v->gb, 1); + + /* Read the MV type/mode */ + if (v->extended_mv == 1) + v->mvrange = get_prefix(&v->gb, 0, 3); + + /* FIXME: what table are used in that case ? */ + v->mv_diff_vlc = &vc9_mv_diff_vlc[0]; + v->cbpcy_vlc = &vc9_cbpcy_i_vlc; + + av_log(v->avctx, AV_LOG_DEBUG, "B frame, QP=%i\n", v->pq); + av_log(v->avctx, AV_LOG_ERROR, "BI_TYPE not supported yet\n"); + /* Epilog should be done in caller */ + return -1; +} + +static int decode_b_picture_header(VC9Context *v) +{ + int pqindex; + + /* Prolog common to all frametypes should be done in caller */ + if (v->profile == PROFILE_SIMPLE) + { + av_log(v, AV_LOG_ERROR, "Found a B frame while in Simple Profile!\n"); + return FRAME_SKIPED; + } + + v->bfraction = vc9_bfraction_lut[get_vlc2(&v->gb, vc9_bfraction_vlc.table, + VC9_BFRACTION_VLC_BITS, 2)]; + if (v->bfraction < -1) + { + av_log(v, AV_LOG_ERROR, "Invalid BFRaction\n"); + return FRAME_SKIPED; + } + else if (!v->bfraction) + { + /* We actually have a BI frame */ + return decode_bi_picture_header(v); + } + + /* Read the quantization stuff */ + pqindex = get_bits(&v->gb, 5); + if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) + v->pq = pquant_table[0][pqindex]; + else + { + v->pq = pquant_table[v->quantizer_mode-1][pqindex]; + } + if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); + if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) + v->pquantizer = get_bits(&v->gb, 1); + + /* Read the MV type/mode */ + if (v->extended_mv == 1) + v->mvrange = get_prefix(&v->gb, 0, 3); + v->mv_mode = get_bits(&v->gb, 1); + if (v->pq < 13) + { + if (!v->mv_mode) + { + v->mv_mode = get_bits(&v->gb, 2); + if (v->mv_mode) + av_log(v, AV_LOG_ERROR, + "mv_mode for lowquant B frame was %i\n", v->mv_mode); + } + } + else + { + if (!v->mv_mode) + { + if (get_bits(&v->gb, 1)) + av_log(v, AV_LOG_ERROR, + "mv_mode for highquant B frame was %i\n", v->mv_mode); + } + v->mv_mode = 1-v->mv_mode; //To match (pq < 13) mapping + } + + if (v->mv_mode == MV_PMODE_MIXED_MV) + { + if (bitplane_decoding( v->mv_type_mb_plane, v->width_mb, + v->height_mb, v)<0) + return -1; + } + + //bitplane + bitplane_decoding(v->direct_mb_plane, v->width_mb, v->height_mb, v); + bitplane_decoding(v->skip_mb_plane, v->width_mb, v->height_mb, v); + + /* FIXME: what is actually chosen for B frames ? */ + v->mv_diff_vlc = &vc9_mv_diff_vlc[get_bits(&v->gb, 2)]; + v->cbpcy_vlc = &vc9_cbpcy_p_vlc[get_bits(&v->gb, 2)]; + if (v->dquant) + { + vop_dquant_decoding(v); + } + + if (v->vstransform) + { + v->ttmbf = get_bits(&v->gb, 1); + if (v->ttmbf) + { + v->ttfrm = get_bits(&v->gb, 2); + av_log(v, AV_LOG_INFO, "Transform used: %ix%i\n", + (v->ttfrm & 2) ? 4 : 8, (v->ttfrm & 1) ? 4 : 8); + } + } + /* Epilog should be done in caller */ + return 0; +} + +static int decode_i_picture_header(VC9Context *v) +{ + int pqindex, status = 0, ac_pred, condover; + + /* Prolog common to all frametypes should be done in caller */ + //BF = Buffer Fullness + if (v->profile <= PROFILE_MAIN && get_bits(&v->gb, 7)) + { + av_log(v, AV_LOG_DEBUG, "I BufferFullness not 0\n"); + } + + /* Quantizer stuff */ + pqindex = get_bits(&v->gb, 5); + if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) + v->pq = pquant_table[0][pqindex]; + else + { + v->pq = pquant_table[v->quantizer_mode-1][pqindex]; + } + if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); + if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) + v->pquantizer = get_bits(&v->gb, 1); + av_log(v->avctx, AV_LOG_DEBUG, "I frame: QP=%i (+%i/2)\n", + v->pq, v->halfpq); +#if HAS_ADVANCED_PROFILE + if (v->profile <= PROFILE_MAIN) +#endif + { + if (v->extended_mv) v->mvrange = get_prefix(&v->gb, 0, 3); + if (v->multires) v->respic = get_bits(&v->gb, 2); + } +#if HAS_ADVANCED_PROFILE + else + { + ac_pred = get_bits(&v->gb, 1); + if (v->postprocflag) v->postproc = get_bits(&v->gb, 1); + /* 7.1.1.34 + 8.5.2 */ + if (v->overlap && v->pq<9) + { + condover = get_bits(&v->gb, 1); + if (condover) + { + condover = 2+get_bits(&v->gb, 1); + if (condover == 3) + status = bitplane_decoding(v->over_flags_plane, + v->width_mb, v->height_mb, v); + } + } + } +#endif + + /* Epilog should be done in caller */ + return status; +} + +static int decode_p_picture_header(VC9Context *v) +{ + /* INTERFRM, FRMCNT, RANGEREDFRM read in caller */ + int lowquant, pqindex; + + pqindex = get_bits(&v->gb, 5); + if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) + v->pq = pquant_table[0][pqindex]; + else + { + v->pq = pquant_table[v->quantizer_mode-1][pqindex]; + } + if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); + if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) + v->pquantizer = get_bits(&v->gb, 1); + av_log(v->avctx, AV_LOG_DEBUG, "P Frame: QP=%i (+%i/2)\n", + v->pq, v->halfpq); + if (v->extended_mv == 1) v->mvrange = get_prefix(&v->gb, 0, 3); +#if HAS_ADVANCED_PROFILE + if (v->profile > PROFILE_MAIN) + { + if (v->postprocflag) v->postproc = get_bits(&v->gb, 1); + } + else +#endif + if (v->multires) v->respic = get_bits(&v->gb, 2); + lowquant = (v->pquantizer>12) ? 0 : 1; + v->mv_mode = mv_pmode_table[lowquant][get_prefix(&v->gb, 1, 4)]; + if (v->mv_mode == MV_PMODE_INTENSITY_COMP) + { + v->mv_mode2 = mv_pmode_table[lowquant][get_prefix(&v->gb, 1, 3)]; + v->lumscale = get_bits(&v->gb, 6); + v->lumshift = get_bits(&v->gb, 6); + } + + if ((v->mv_mode == MV_PMODE_INTENSITY_COMP && + v->mv_mode2 == MV_PMODE_MIXED_MV) + || v->mv_mode == MV_PMODE_MIXED_MV) + { + if (bitplane_decoding(v->mv_type_mb_plane, v->width_mb, + v->height_mb, v) < 0) + return -1; + } + + if (bitplane_decoding(v->skip_mb_plane, v->width_mb, + v->height_mb, v) < 0) + return -1; + + /* Hopefully this is correct for P frames */ + v->mv_diff_vlc = &vc9_mv_diff_vlc[get_bits(&v->gb, 2)]; + v->cbpcy_vlc = &vc9_cbpcy_p_vlc[get_bits(&v->gb, 2)]; + + if (v->dquant) + { + av_log(v->avctx, AV_LOG_INFO, "VOP DQuant info\n"); + vop_dquant_decoding(v); + } + + if (v->vstransform) + { + v->ttmbf = get_bits(&v->gb, 1); + if (v->ttmbf) + { + v->ttfrm = get_bits(&v->gb, 2); + av_log(v->avctx, AV_LOG_INFO, "Transform used: %ix%i\n", + (v->ttfrm & 2) ? 4 : 8, (v->ttfrm & 1) ? 4 : 8); + } + } + /* Epilog should be done in caller */ + return 0; +} + + +static int standard_decode_picture_header(VC9Context *v) +{ + int status = 0; + + if (v->finterpflag) v->interpfrm = get_bits(&v->gb, 1); + skip_bits(&v->gb, 2); //framecnt unused + if (v->rangered) v->rangeredfrm = get_bits(&v->gb, 1); + v->pict_type = get_bits(&v->gb, 1); + if (v->avctx->max_b_frames && !v->pict_type) + { + if (get_bits(&v->gb, 1)) v->pict_type = I_TYPE; + else v->pict_type = P_TYPE; + } + else v->pict_type++; //P_TYPE + + switch (v->pict_type) + { + case I_TYPE: status = decode_i_picture_header(v); break; + case BI_TYPE: status = decode_b_picture_header(v); break; + case P_TYPE: status = decode_p_picture_header(v); break; + case B_TYPE: status = decode_b_picture_header(v); break; + } + + if (status == FRAME_SKIPED) + { + av_log(v, AV_LOG_INFO, "Skipping frame...\n"); + return status; + } + + /* AC/DC Syntax */ + v->transacfrm = get_bits(&v->gb, 1); + if (v->transacfrm) v->transacfrm += get_bits(&v->gb, 1); + if (v->pict_type == I_TYPE || v->pict_type == BI_TYPE) + { + v->transacfrm2 = get_bits(&v->gb, 1); + if (v->transacfrm2) v->transacfrm2 += get_bits(&v->gb, 1); + } + v->transacdctab = get_bits(&v->gb, 1); + + return 0; +} + + +#if HAS_ADVANCED_PROFILE +/******************************************************************************/ +/* Advanced Profile picture header decoding specific functions */ +/******************************************************************************/ +static int advanced_decode_picture_header(VC9Context *v) +{ + static const int type_table[4] = { P_TYPE, B_TYPE, I_TYPE, BI_TYPE }; + int type, i, ret; + + if (v->interlace) + { + v->fcm = get_bits(&v->gb, 1); + if (v->fcm) v->fcm = 2+get_bits(&v->gb, 1); + } + + type = get_prefix(&v->gb, 0, 4); + if (type > 4 || type < 0) return FRAME_SKIPED; + v->pict_type = type_table[type]; + av_log(v->avctx, AV_LOG_INFO, "AP Frame Type: %i\n", v->pict_type); + + if (v->tfcntrflag) v->tfcntr = get_bits(&v->gb, 8); + if (v->broadcast) + { + if (!v->interlace) v->rptfrm = get_bits(&v->gb, 2); + else + { + v->tff = get_bits(&v->gb, 1); + v->rff = get_bits(&v->gb, 1); + } + } + + if (v->panscanflag) + { +#if 0 + for (i=0; inumpanscanwin; i++) + { + v->topleftx[i] = get_bits(&v->gb, 16); + v->toplefty[i] = get_bits(&v->gb, 16); + v->bottomrightx[i] = get_bits(&v->gb, 16); + v->bottomrighty[i] = get_bits(&v->gb, 16); + } +#else + skip_bits(&v->gb, 16*4*v->numpanscanwin); +#endif + } + v->rndctrl = get_bits(&v->gb, 1); + v->uvsamp = get_bits(&v->gb, 1); + if (v->finterpflag == 1) v->interpfrm = get_bits(&v->gb, 1); + + switch(v->pict_type) + { + case I_TYPE: if (decode_i_picture_header(v) < 0) return -1; + case P_TYPE: if (decode_p_picture_header(v) < 0) return -1; + case BI_TYPE: + case B_TYPE: if (decode_b_picture_header(v) < 0) return FRAME_SKIPED; + default: break; + } + + /* AC/DC Syntax */ + v->transacfrm = get_bits(&v->gb, 1); + if (v->transacfrm) v->transacfrm += get_bits(&v->gb, 1); + if (v->pict_type == I_TYPE || v->pict_type == BI_TYPE) + { + v->transacfrm2 = get_bits(&v->gb, 1); + if (v->transacfrm2) v->transacfrm2 += get_bits(&v->gb, 1); + } + v->transacdctab = get_bits(&v->gb, 1); + if (v->pict_type == I_TYPE) vop_dquant_decoding(v); + + return 0; +} +#endif + +/******************************************************************************/ +/* MacroBlock decoding functions */ +/******************************************************************************/ +static int standard_decode_i_mbs(VC9Context *v) +{ + int x, y, ac_pred, cbpcy; + + /* Select ttmb table depending on pq */ + if (v->pq < 5) v->ttmb_vlc = &vc9_ttmb_vlc[0]; + else if (v->pq < 13) v->ttmb_vlc = &vc9_ttmb_vlc[1]; + else v->ttmb_vlc = &vc9_ttmb_vlc[2]; + + for (y=0; yheight_mb; y++) + { + for (x=0; xwidth_mb; x++) + { + cbpcy = get_vlc2(&v->gb, vc9_cbpcy_i_vlc.table, + VC9_CBPCY_I_VLC_BITS, 2); + ac_pred = get_bits(&v->gb, 1); + //Decode blocks from that mb wrt cbpcy + } + } + return 0; +} + +#define GET_MQUANT() \ + if (v->dquantfrm) \ + { \ + if (v->dqprofile == DQPROFILE_ALL_MBS) \ + { \ + if (v->dqbilevel) \ + { \ + mquant = (get_bits(&v->gb, 1)) ? v->pq : v->altpq; \ + } \ + else \ + { \ + mqdiff = get_bits(&v->gb, 3); \ + if (mqdiff != 7) mquant = v->pq + mqdiff; \ + else mquant = get_bits(&v->gb, 5); \ + } \ + } \ + } + +/* MVDATA decoding from 8.3.5.2 */ +#define GET_MVDATA() \ + index = 1 + get_vlc2(&v->gb, v->mv_diff_vlc->table, \ + VC9_MV_DIFF_VLC_BITS, 2); \ + if (index > 36) \ + { \ + mb_has_coeffs = 1; \ + index -= 37; \ + } \ + else mb_has_coeffs = 0; \ + mb_is_intra = 0; \ + if (!index) { dmv_x = dmv_y = 0; } \ + else if (index == 35) \ + { \ + dmv_x = get_bits(&v->gb, k_x); \ + dmv_y = get_bits(&v->gb, k_y); \ + mb_is_intra = 1; \ + } \ + else \ + { \ + index1 = index%6; \ + if (hpel_flag && index1 == 5) val = 1; \ + else val = 0; \ + val = get_bits(&v->gb, size_table[index1] - val); \ + sign = 0 - (val&1); \ + dmv_x = (sign ^ ((val>>1) + offset_table[index1])) - sign; \ + \ + index1 = index/6; \ + if (hpel_flag && index1 == 5) val = 1; \ + else val = 0; \ + val = get_bits(&v->gb, size_table[index1] - val); \ + sign = 0 - (val&1); \ + dmv_y = (sign ^ ((val>>1) + offset_table[index1])) - sign; \ + } + +static int decode_p_mbs(VC9Context *v) +{ + int x, y, current_mb = 0, i; /* MB/Block Position info */ + int skip_mb_bit = 0, cbpcy; /* MB/B skip */ + int hybrid_pred, ac_pred; /* Prediction types */ + int mb_has_coeffs = 1 /* last_flag */, mb_is_intra; + int dmv_x, dmv_y; /* Differential MV components */ + int mv_mode_bit = 0; /* mv_mode_bit: 1MV=0, 4MV=0 */ + int mqdiff, mquant; /* MB quantization */ + int tt_block; /* MB Transform type */ + static const int size_table[6] = { 0, 2, 3, 4, 5, 8 }, + offset_table[6] = { 0, 1, 3, 7, 15, 31 }; + int k_x, k_y; /* Long MV fixed bitlength */ + int hpel_flag, intra_flag; /* Some MB properties */ + int index, index1; /* LUT indices */ + int val, sign; + + /* Select ttmb table depending on pq */ + if (v->pq < 5) v->ttmb_vlc = &vc9_ttmb_vlc[0]; + else if (v->pq < 13) v->ttmb_vlc = &vc9_ttmb_vlc[1]; + else v->ttmb_vlc = &vc9_ttmb_vlc[2]; + + /* Select proper long MV range */ + switch (v->mvrange) + { + case 1: k_x = 10; k_y = 9; break; + case 2: k_x = 12; k_y = 10; break; + case 3: k_x = 13; k_y = 11; break; + default: /*case 0 too */ k_x = 9; k_y = 8; break; + } + + hpel_flag = v->mv_mode & 1; //MV_PMODE is HPEL + k_x -= hpel_flag; + k_y -= hpel_flag; + + for (y=0; yheight_mb; y++) + { + for (x=0; xwidth_mb; x++) + { + if (v->mv_type_mb_plane[current_mb]) + mv_mode_bit = get_bits(&v->gb, 1); + if (0) //skipmb is rawmode + skip_mb_bit = get_bits(&v->gb, 1); + if (!mv_mode_bit) /* 1MV mode */ + { + if (!v->skip_mb_plane[current_mb]) + { + GET_MVDATA(); + + /* hybrid mv pred, 8.3.5.3.4 */ + if (v->mv_mode == MV_PMODE_1MV || + v->mv_mode == MV_PMODE_MIXED_MV) + hybrid_pred = get_bits(&v->gb, 1); + if (mb_is_intra && !mb_has_coeffs) + { + GET_MQUANT(); + ac_pred = get_bits(&v->gb, 1); + } + else if (mb_has_coeffs) + { + if (mb_is_intra) ac_pred = get_bits(&v->gb, 1); + cbpcy = get_vlc2(&v->gb, v->cbpcy_vlc->table, + VC9_CBPCY_P_VLC_BITS, 2); + GET_MQUANT(); + } + if (!v->ttmbf) + v->ttfrm = get_vlc2(&v->gb, v->ttmb_vlc->table, + VC9_TTMB_VLC_BITS, 12); + //Decode blocks from that mb wrt cbpcy + } + else //Skipped + { + /* hybrid mv pred, 8.3.5.3.4 */ + if (v->mv_mode == MV_PMODE_1MV || + v->mv_mode == MV_PMODE_MIXED_MV) + hybrid_pred = get_bits(&v->gb, 1); + } + } //1MV mode + else //4MV mode + { + if (!v->skip_mb_plane[current_mb] /* unskipped MB */) + { + cbpcy = get_vlc2(&v->gb, v->cbpcy_vlc->table, + VC9_CBPCY_P_VLC_BITS, 2); + for (i=0; i<4; i++) //For all 4 Y blocks + { + if (cbpcy & (1<<6) /* cbpcy set for this block */) + { + GET_MVDATA(); + } + if (v->mv_mode == MV_PMODE_MIXED_MV /* Hybrid pred */) + hybrid_pred = get_bits(&v->gb, 1); + GET_MQUANT(); + if (mb_is_intra /* One of the 4 blocks is intra */ && + index /* non-zero pred for that block */) + ac_pred = get_bits(&v->gb, 1); + if (!v->ttmbf) + tt_block = get_vlc2(&v->gb, v->ttmb_vlc->table, + VC9_TTMB_VLC_BITS, 12); + + /* TODO: Process blocks wrt cbpcy */ + + /* Prepare cbpcy for next block */ + cbpcy <<= 1; + } + } + else //Skipped MB + { + for (i=0; i<4; i++) //All 4 Y blocks + { + if (v->mv_mode == MV_PMODE_MIXED_MV /* Hybrid pred */) + hybrid_pred = get_bits(&v->gb, 1); + + /* FIXME: do something */ + } + } + } + } + current_mb++; + } + return 0; +} + +static int decode_b_mbs(VC9Context *v) +{ + int x, y, current_mb = 0 , last_mb = v->height_mb*v->width_mb, + i /* MB / B postion information */; + int direct_b_bit = 0, skip_mb_bit = 0; + int ac_pred; + int b_mv1 = 0, b_mv2 = 0, b_mv_type = 0; + int mquant, mqdiff; /* MB quant stuff */ + int tt_block; /* Block transform type */ + + for (y=0; yheight_mb; y++) + { + for (x=0; xwidth_mb; x++) + { + if (v->direct_mb_plane[current_mb]) + direct_b_bit = get_bits(&v->gb, 1); + if (1 /* Skip mode is raw */) + { + /* FIXME getting tired commenting */ +#if 0 + skip_mb_bit = get_bits(&v->gb, n); //vlc +#endif + } + if (!direct_b_bit) + { + if (skip_mb_bit) + { + /* FIXME getting tired commenting */ +#if 0 + b_mv_type = get_bits(&v->gb, n); //vlc +#endif + } + else + { + /* FIXME getting tired commenting */ +#if 0 + b_mv1 = get_bits(&v->gb, n); //VLC +#endif + if (1 /* b_mv1 isn't intra */) + { + /* FIXME: actually read it */ + b_mv_type = 0; //vlc + } + } + } + if (!skip_mb_bit) + { + if (b_mv1 != last_mb) + { + GET_MQUANT(); + if (1 /* intra mb */) + ac_pred = get_bits(&v->gb, 1); + } + else + { + if (1 /* forward_mb is interpolate */) + { + /* FIXME: actually read it */ + b_mv2 = 0; //vlc + } + if (1 /* b_mv2 isn't the last */) + { + if (1 /* intra_mb */) + ac_pred = get_bits(&v->gb, 1); + GET_MQUANT(); + } + } + } + //End1 + /* FIXME getting tired, commenting */ +#if 0 + if (v->ttmbf) + v->ttmb = get_bits(&v->gb, n); //vlc +#endif + } + //End2 + for (i=0; i<6; i++) + { + /* FIXME: process the block */ + } + + current_mb++; + } + return 0; +} + +#if HAS_ADVANCED_PROFILE +static int advanced_decode_i_mbs(VC9Context *v) +{ + int i, x, y, cbpcy, mqdiff, absmq, mquant, ac_pred, condover, + current_mb = 0, over_flags_mb = 0; + + for (y=0; yheight_mb; y++) + { + for (x=0; xwidth_mb; x++) + { + if (v->ac_pred_plane[i]) + ac_pred = get_bits(&v->gb, 1); + if (condover == 3 && v->over_flags_plane) + over_flags_mb = get_bits(&v->gb, 1); + GET_MQUANT(); + } + current_mb++; + } + return 0; +} +#endif + +static int vc9_decode_init(AVCodecContext *avctx) +{ + VC9Context *v = avctx->priv_data; + GetBitContext gb; + + if (!avctx->extradata_size || !avctx->extradata) return -1; + avctx->pix_fmt = PIX_FMT_YUV420P; + v->avctx = avctx; + + if (init_common(v) < 0) return -1; + + if (avctx->codec_id == CODEC_ID_WMV3) + { + int count = 0; + + // looks like WMV3 has a sequence header stored in the extradata + // advanced sequence header may be before the first frame + // the last byte of the extradata is a version number, 1 for the + // samples we can decode + + init_get_bits(&gb, avctx->extradata, avctx->extradata_size); + + decode_sequence_header(avctx, &gb); + + count = avctx->extradata_size*8 - get_bits_count(&gb); + if (count>0) + { + av_log(avctx, AV_LOG_INFO, "Extra data: %i bits left, value: %X\n", + count, get_bits(&gb, count)); + } + else + { + av_log(avctx, AV_LOG_INFO, "Read %i bits in overflow\n", -count); + } + } + + /* Done with header parsing */ + //FIXME I feel like this is wrong + v->width_mb = (avctx->coded_width+15)>>4; + v->height_mb = (avctx->coded_height+15)>>4; + + /* Allocate mb bitplanes */ + v->mv_type_mb_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); + if (!v->mv_type_mb_plane) return -1; + v->skip_mb_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); + if (!v->skip_mb_plane) return -1; + v->direct_mb_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); + if (!v->direct_mb_plane) return -1; + +#if HAS_ADVANCED_PROFILE + if (v->profile > PROFILE_MAIN) + { + v->over_flags_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); + if (!v->over_flags_plane) return -1; + v->ac_pred_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); + if (!v->ac_pred_plane) return -1; + } +#endif + + return 0; +} + +static int vc9_decode_frame(AVCodecContext *avctx, + void *data, int *data_size, + uint8_t *buf, int buf_size) +{ + VC9Context *v = avctx->priv_data; + int ret = FRAME_SKIPED, len, start_code; + AVFrame *pict = data; + uint8_t *tmp_buf; + v->avctx = avctx; + + //buf_size = 0 -> last frame + if (!buf_size) return 0; + + len = avpicture_get_size(avctx->pix_fmt, avctx->width, + avctx->height); + tmp_buf = (uint8_t *)av_mallocz(len); + avpicture_fill((AVPicture *)pict, tmp_buf, avctx->pix_fmt, + avctx->width, avctx->height); + + if (avctx->codec_id == CODEC_ID_WMV3) + { + init_get_bits(&v->gb, buf, buf_size*8); + av_log(avctx, AV_LOG_INFO, "Frame: %i bits to decode\n", buf_size*8); + +#if HAS_ADVANCED_PROFILE + if (v->profile > PROFILE_MAIN) + { + if (advanced_decode_picture_header(v) == FRAME_SKIPED) return buf_size; + switch(v->pict_type) + { + case I_TYPE: ret = advanced_decode_i_mbs(v); break; + case P_TYPE: ret = decode_p_mbs(v); break; + case B_TYPE: + case BI_TYPE: ret = decode_b_mbs(v); break; + default: ret = FRAME_SKIPED; + } + if (ret == FRAME_SKIPED) return buf_size; //We ignore for now failures + } + else +#endif + { + if (standard_decode_picture_header(v) == FRAME_SKIPED) return buf_size; + switch(v->pict_type) + { + case I_TYPE: ret = standard_decode_i_mbs(v); break; + case P_TYPE: ret = decode_p_mbs(v); break; + case B_TYPE: + case BI_TYPE: ret = decode_b_mbs(v); break; + default: ret = FRAME_SKIPED; + } + if (ret == FRAME_SKIPED) return buf_size; + } + + /* Size of the output data = image */ + av_log(avctx, AV_LOG_DEBUG, "Consumed %i/%i bits\n", + get_bits_count(&v->gb), buf_size*8); + } + else + { +#if 0 + // search for IDU's + // FIXME + uint32_t scp = 0; + int scs = 0, i = 0; + + while (i < buf_size) + { + for (; i < buf_size && scp != 0x000001; i++) + scp = ((scp<<8)|buf[i])&0xffffff; + + if (scp != 0x000001) + break; // eof ? + + scs = buf[i++]; + + init_get_bits(&v->gb, buf+i, (buf_size-i)*8); + + switch(scs) + { + case 0xf: + decode_sequence_header(avctx, &v->gb); + break; + // to be finished + } + + i += get_bits_count(&v->gb)*8; + } +#else + av_abort(); +#endif + } + + *data_size = len; + + /* Fake consumption of all data */ + return buf_size; //Number of bytes consumed +} + +static int vc9_decode_end(AVCodecContext *avctx) +{ + VC9Context *v = avctx->priv_data; + +#if HAS_ADVANCED_PROFILE + if (v->hrd_rate) av_free(v->hrd_rate); + if (v->hrd_buffer) av_free(v->hrd_buffer); +#endif + if (v->mv_type_mb_plane) av_free(v->mv_type_mb_plane); + if (v->skip_mb_plane) av_free(v->skip_mb_plane); + if (v->direct_mb_plane) av_free(v->direct_mb_plane); + return 0; +} + +AVCodec vc9_decoder = { + "vc9", + CODEC_TYPE_VIDEO, + CODEC_ID_VC9, + sizeof(VC9Context), + vc9_decode_init, + NULL, + vc9_decode_end, + vc9_decode_frame, + 0, + NULL +}; + +AVCodec wmv3_decoder = { + "wmv3", + CODEC_TYPE_VIDEO, + CODEC_ID_WMV3, + sizeof(VC9Context), + vc9_decode_init, + NULL, + vc9_decode_end, + vc9_decode_frame, + 0, + NULL +}; diff --git a/libavcodec/vc9data.h b/libavcodec/vc9data.h new file mode 100644 index 0000000000..c3826f1a63 --- /dev/null +++ b/libavcodec/vc9data.h @@ -0,0 +1,373 @@ +/** + * @file vc9data.h + * VC9 tables. + */ + +#ifndef VC9DATA_H +#define VC9DATA_H + +/* bfraction is fractional, we scale to the GCD 3*5*7*8 = 840 */ +const int16_t vc9_bfraction_lut[23] = { + 420 /*1/2*/, 280 /*1/3*/, 560 /*2/3*/, 210 /*1/4*/, + 630 /*3/4*/, 168 /*1/5*/, 336 /*2/5*/, + 504 /*3/5*/, 672 /*4/5*/, 140 /*1/6*/, 700 /*5/6*/, + 120 /*1/7*/, 240 /*2/7*/, 360 /*3/7*/, 480 /*4/7*/, + 600 /*5/7*/, 720 /*6/7*/, 105 /*1/8*/, 315 /*3/8*/, + 525 /*5/8*/, 735 /*7/8*/, + -1 /*inv.*/, 0 /*BI fm*/ +}; +const uint8_t vc9_bfraction_bits[23] = { + 3, 3, 3, 3, + 3, 3, 3, + 7, 7, 7, 7, + 7, 7, 7, 7, + 7, 7, 7, 7, + 7, 7, + 7, 7 +}; +const uint8_t vc9_bfraction_codes[23] = { + 0, 1, 2, 3, + 4, 5, 6, + 112, 113, 114, 115, + 116, 117, 118, 119, + 120, 121, 122, 123, + 124, 125, + 126, 127 +}; + +//Same as H.264 +static const AVRational vc9_pixel_aspect[16]={ + {0, 1}, + {1, 1}, + {12, 11}, + {10, 11}, + {16, 11}, + {40, 33}, + {24, 11}, + {20, 11}, + {32, 11}, + {80, 33}, + {18, 11}, + {15, 11}, + {64, 33}, + {160, 99}, + {0, 1}, + {0, 1} +}; + +/* BitPlane IMODE - such a small table... */ +static const uint8_t vc9_imode_codes[7] = { + 0, 2, 1, 3, 1, 2, 3 +}; +static const uint8_t vc9_imode_bits[7] = { + 4, 2, 3, 2, 4, 3, 3 +}; + +/* Normal-2 imode */ +static const uint8_t vc9_norm2_codes[4] = { + 0, 4, 5, 3 +}; +static const uint8_t vc9_norm2_bits[4] = { + 1, 3, 3, 2 +}; + +/* 4MV Block pattern VLC tables */ +static const uint8_t vc9_4mv_block_pattern_codes[4][16] = { + { 14, 58, 59, 25, 12, 26, 15, 15, 13, 24, 27, 0, 28, 1, 2, 2}, + { 8, 18, 19, 4, 20, 5, 30, 11, 21, 31, 6, 12, 7, 13, 14, 0}, + { 15, 6, 7, 2, 8, 3, 28, 9, 10, 29, 4, 11, 5, 12, 13, 0}, + { 0, 11, 12, 4, 13, 5, 30, 16, 14, 31, 6, 17, 7, 18, 19, 19} +}; +static const uint8_t vc9_4mv_block_pattern_bits[4][16] = { + { 5, 6, 6, 5, 5, 5, 5, 4, 5, 5, 5, 3, 5, 3, 3, 2}, + { 4, 5, 5, 4, 5, 4, 5, 4, 5, 5, 4, 4, 4, 4, 4, 2}, + { 4, 4, 4, 4, 4, 4, 5, 4, 4, 5, 4, 4, 4, 4, 4, 3}, + { 2, 4, 4, 4, 4, 4, 5, 5, 4, 5, 4, 5, 4, 5, 5, 4} +}; + +/* I-Picture CBPCY VLC tables */ +static const uint16_t vc9_cbpcy_i_codes[64] = { + 1, 23, 9, 5, 6, 71, 32, 16, + 2, 124, 58, 29, 2, 236, 119, 0, + 3, 183, 44, 19, 1, 360, 70, 63, + 30, 1810, 181, 66, 34, 453, 286, 135, + 6, 3, 30, 28, 18, 904, 68, 112, + 31, 574, 57, 142, 1, 454, 182, 69, + 20, 575, 125, 24, 7, 455, 134, 25, + 21, 475, 2, 70, 13, 1811, 474, 361 +}; +static const uint8_t vc9_cbpcy_i_bits[64] = { + 1, 6, 5, 5, 5, 9, 7, 7, 5, 9, 7, 7, 6, 9, 8, 8, + 5, 9, 7, 7, 6, 10, 8, 8, 6, 13, 9, 8, 7, 11, 10, 9, + 4, 9, 7, 6, 7, 12, 9, 9, 6, 11, 8, 9, 7, 11, 9, 9, + 6, 11, 9, 9, 7, 11, 9, 9, 6, 10, 9, 9, 8, 13, 10, 10 +}; + +/* P-Picture CBPCY VLC tables */ +static const uint16_t vc9_cbpcy_p_codes[4][64] = { + { + 0, 1, 1, 4, 5, 1, 12, 4, 13, 14, 10, 11, 12, 7, 13, 2, + 15, 1, 96, 1, 49, 97, 2, 100, 3, 4, 5, 101, 102, 52, 53, 4, + 6, 7, 54, 103, 8, 9, 10, 110, 11, 12, 111, 56, 114, 58, 115, 5, + 13, 7, 8, 9, 10, 11, 12, 30, 13, 14, 15, 118, 119, 62, 63, 3 + }, + { + 0, 1, 2, 1, 3, 1, 16, 17, 5, 18, 12, 19, 13, 1, 28, 58, + 1, 1, 1, 2, 3, 2, 3, 236, 237, 4, 5, 238, 6, 7, 239, 8, + 9, 240, 10, 11, 121, 122, 12, 13, 14, 15, 241, 246, 16, 17, 124, 63, + 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 247, 125 + }, + { + 0, 1, 2, 3, 2, 3, 1, 4, 5, 24, 7, 13, 16, 17, 9, 5, + 25, 1, 1, 1, 2, 3, 96, 194, 1, 2, 98, 99, 195, 200, 101, 26, + 201, 102, 412, 413, 414, 54, 220, 111, 221, 3, 224, 113, 225, 114, 230, 29, + 231, 415, 240, 4, 241, 484, 5, 243, 3, 244, 245, 485, 492, 493, 247, 31 + }, + { + 0, 1, 1, 1, 2, 2, 3, 4, 3, 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, 28, 29, 30, 31 + } +}; +static const uint8_t vc9_cbpcy_p_bits[4][64] = { + { + 13, 6, 5, 6, 6, 7, 7, 5, 7, 7, 6, 6, 6, 5, 6, 3, + 7, 8, 8, 13, 7, 8, 13, 8, 13, 13, 13, 8, 8, 7, 7, 3, + 13, 13, 7, 8, 13, 13, 13, 8, 13, 13, 8, 7, 8, 7, 8, 3, + 13, 12, 12, 12, 12, 12, 12, 6, 12, 12, 12, 8, 8, 7, 7, 2 + }, + { + 14, 3, 3, 5, 3, 4, 5, 5, 3, 5, 4, 5, 4, 6, 5, 6, + 8, 14, 13, 8, 8, 13, 13, 8, 8, 13, 13, 8, 13, 13, 8, 13, + 13, 8, 13, 13, 7, 7, 13, 13, 13, 13, 8, 8, 13, 13, 7, 6, + 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 13, 8, 7 + }, + { + 13, 5, 5, 5, 4, 4, 6, 4, 4, 6, 4, 5, 5, 5, 4, 3, + 6, 8, 10, 9, 8, 8, 7, 8, 13, 13, 7, 7, 8, 8, 7, 5, + 8, 7, 9, 9, 9, 6, 8, 7, 8, 13, 8, 7, 8, 7, 8, 5, + 8, 9, 8, 13, 8, 9, 13, 8, 12, 8, 8, 9, 9, 9, 8, 5 + }, + { + 9, 2, 3, 9, 2, 9, 9, 9, 2, 9, 9, 9, 9, 9, 9, 9, + 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, + 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, + 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 8, 8, 8, 8 + } +}; + +/* MacroBlock Transform Type + * 8x8:B + * 8x4:B:btm 8x4:B:top 8x4:B:both, + * 4x8:B:right 4x8:B:left 4x8:B:both + * 4x4:B 8x8:MB + * 8x4:MB:btm 8x4:MB:top 8x4,MB,both + * 4x8,MB,right 4x8,MB,left + * 4x4,MB */ +static const uint16_t vc9_ttmb_codes[3][16] = { + { + 0x0003, + 0x002E, 0x005F, 0x0000, + 0x0016, 0x0015, 0x0001, + 0x0004, 0x0014, + 0x02F1, 0x0179, 0x017B, + 0x0BC0, 0x0BC1, 0x05E1, + 0x017A + }, + { + 0x0006, + 0x0006, 0x0003, 0x0007, + 0x000F, 0x000E, 0x0000, + 0x0002, 0x0002, + 0x0014, 0x0011, 0x000B, + 0x0009, 0x0021, 0x0015, + 0x0020 + }, + { + 0x0006, + 0x0000, 0x000E, 0x0005, + 0x0002, 0x0003, 0x0003, + 0x000F, 0x0002, + 0x0081, 0x0011, 0x0009, + 0x0101, 0x0041, 0x0011, + 0x0100 + } +}; + +static const uint8_t vc9_ttmb_bits[3][16] = { + { + 2, + 6, 7, 2, + 5, 5, 2, + 3, 5, + 10, 9, 9, + 12, 12, 11, + 9 + }, + { + 3, + 4, 4, 4, + 4, 4, 3, + 3, 2, + 7, 7, 6, + 6, 8, 7, + 8 + }, + { + 3, + 3, 4, 5, + 3, 3, 4, + 4, 2, + 9, 7, 6, + 10, 9, 7, + 11 + } +}; + + +/* MV differential tables, p265 */ +static const uint16_t vc9_mv_diff_codes[4][73] = { + { + 0, 2, 3, 8, 576, 3, 2, 6, + 5, 577, 578, 7, 8, 9, 40, 19, + 37, 82, 21, 22, 23, 579, 580, 166, + 96, 167, 49, 194, 195, 581, 582, 583, + 292, 293, 294, 13, 2, 7, 24, 50, + 102, 295, 13, 7, 8, 18, 50, 103, + 38, 20, 21, 22, 39, 204, 103, 23, + 24, 25, 104, 410, 105, 106, 107, 108, + 109, 220, 411, 442, 222, 443, 446, 447, + 7 /* 73 elements */ + }, + { + 0, 4, 5, 3, 4, 3, 4, 5, + 20, 6, 21, 44, 45, 46, 3008, 95, + 112, 113, 57, 3009, 3010, 116, 117, 3011, + 118, 3012, 3013, 3014, 3015, 3016, 3017, 3018, + 3019, 3020, 3021, 3022, 1, 4, 15, 160, + 161, 41, 6, 11, 42, 162, 43, 119, + 56, 57, 58, 163, 236, 237, 3023, 119, + 120, 242, 122, 486, 1512, 487, 246, 494, + 1513, 495, 1514, 1515, 1516, 1517, 1518, 1519, + 31 /* 73 elements */ + }, + { + 0, 512, 513, 514, 515, 2, 3, 258, + 259, 260, 261, 262, 263, 264, 265, 266, + 267, 268, 269, 270, 271, 272, 273, 274, + 275, 276, 277, 278, 279, 280, 281, 282, + 283, 284, 285, 286, 1, 5, 287, 288, + 289, 290, 6, 7, 291, 292, 293, 294, + 295, 296, 297, 298, 299, 300, 301, 302, + 303, 304, 305, 306, 307, 308, 309, 310, + 311, 312, 313, 314, 315, 316, 317, 318, + 319 /* 73 elements */ + }, + { + 0, 1, 1, 2, 3, 4, 1, 5, + 4, 3, 5, 8, 6, 9, 10, 11, + 12, 7, 104, 14, 105, 4, 10, 15, + 11, 6, 14, 8, 106, 107, 108, 15, + 109, 9, 55, 10, 1, 2, 1, 2, + 3, 12, 6, 2, 6, 7, 28, 7, + 15, 8, 5, 18, 29, 152, 77, 24, + 25, 26, 39, 108, 13, 109, 55, 56, + 57, 116, 11, 153, 234, 235, 118, 119, + 15 /* 73 elements */ + } +}; +static const uint8_t vc9_mv_diff_bits[4][73] = { + { + 6, 7, 7, 8, 14, 6, 5, 6, 7, 14, 14, 6, 6, 6, 8, 9, + 10, 9, 7, 7, 7, 14, 14, 10, 9, 10, 8, 10, 10, 14, 14, 14, + 13, 13, 13, 6, 3, 5, 6, 8, 9, 13, 5, 4, 4, 5, 7, 9, + 6, 5, 5, 5, 6, 9, 8, 5, 5, 5, 7, 10, 7, 7, 7, 7, + 7, 8, 10, 9, 8, 9, 9, 9, 3 /* 73 elements */ + }, + { + 5, 7, 7, 6, 6, 5, 5, 6, 7, 5, 7, 8, 8, 8, 14, 9, + 9, 9, 8, 14, 14, 9, 9, 14, 9, 14, 14, 14, 14, 14, 14, 14, + 14, 14, 14, 14, 2, 3, 6, 8, 8, 6, 3, 4, 6, 8, 6, 9, + 6, 6, 6, 8, 8, 8, 14, 7, 7, 8, 7, 9, 13, 9, 8, 9, + 13, 9, 13, 13, 13, 13, 13, 13, 5 /* 73 elements */ + + }, + { + 3, 12, 12, 12, 12, 3, 4, 11, 11, 11, 11, 11, 11, 11, 11, 11, + 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, + 11, 11, 11, 11, 1, 5, 11, 11, 11, 11, 4, 4, 11, 11, 11, 11, + 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, 11, + 11, 11, 11, 11, 11, 11, 11, 11, 11 /* 73 elements */ + }, + { + 15, 11, 15, 15, 15, 15, 12, 15, 12, 11, 12, 12, 15, 12, 12, 12, + 12, 15, 15, 12, 15, 10, 11, 12, 11, 10, 11, 10, 15, 15, 15, 11, + 15, 10, 14, 10, 4, 4, 5, 7, 8, 9, 5, 3, 4, 5, 6, 8, + 5, 4, 3, 5, 6, 8, 7, 5, 5, 5, 6, 7, 9, 7, 6, 6, + 6, 7, 10, 8, 8, 8, 7, 7, 4 /* 73 elements */ + } +}; + +/* DC luma differentials low+hi-mo, p217 */ +static const uint32_t vc9_luma_dc_codes[2][120] = { + { + 1, 1, 1, 1, 5, 7, 8, 12, + 0, 2, 18, 26, 3, 7, 39, 55, + 5, 76, 108, 109, 8, 25, 155, 27, + 154, 19, 52, 53, 97, 72, 196, 74, + 198, 199, 146, 395, 147, 387, 386, 150, + 151, 384, 788, 789, 1541, 1540, 1542, 3086, + 197581, 197577, 197576, 197578, 197579, 197580, 197582, 197583, + 197584, 197585, 197586, 197587, 197588, 197589, 197590, 197591, + 197592, 197593, 197594, 197595, 197596, 197597, 197598, 197599, + 197600, 197601, 197602, 197603, 197604, 197605, 197606, 197607, + 197608, 197609, 197610, 197611, 197612, 197613, 197614, 197615, + 197616, 197617, 197618, 197619, 197620, 197621, 197622, 197623, + 197624, 197625, 197626, 197627, 197628, 197629, 197630, 197631, + 395136, 395137, 395138, 395139, 395140, 395141, 395142, 395143, + 395144, 395145, 395146, 395147, 395148, 395149, 395150, 395151 + }, + { + 2, 3, 3, 2, 5, 1, 3, 8, + 0, 5, 13, 15, 19, 8, 24, 28, + 36, 4, 6, 18, 50, 59, 74, 75, + 11, 38, 39, 102, 116, 117, 20, 28, + 31, 29, 43, 61, 413, 415, 84, 825, + 824, 829, 171, 241, 1656, 242, 480, 481, + 340, 3314, 972, 683, 6631, 974, 6630, 1364, + 1951, 1365, 3901, 3895, 3900, 3893, 7789, 7784, + 15576, 15571, 15577, 31140, 996538, 996532, 996533, 996534, + 996535, 996536, 996537, 996539, 996540, 996541, 996542, 996543, + 1993024, 1993025, 1993026, 1993027, 1993029, 1993029, 1993030, 1993031, + 1993032, 1993033, 1993034, 1993035, 1993036, 1993037, 1993038, 1993039, + 1993040, 1993041, 1993042, 1993043, 1993044, 1993045, 1993046, 1993047, + 1993048, 1993049, 1993050, 1993051, 1993052, 1993053, 1993054, 1993055, + 1993056, 1993057, 1993058, 1993059, 1993060, 1993061, 1993062, 1993063 + } +}; +static const uint8_t vc9_luma_dc_bits[2][120] = { + { + 1, 2, 4, 5, 5, 5, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, + 9, 9, 9, 9, 10, 10, 10, 10, 10, 11, 11, 11, 12, 13, 13, 13, + 13, 13, 14, 14, 14, 14, 14, 14, 14, 14, 15, 15, 16, 16, 16, 17, + 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, + 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, + 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, 23, + 23, 23, 23, 23, 23, 23, 23, 23, 24, 24, 24, 24, 24, 24, 24, 24, + 24, 24, 24, 24, 24, 24, 24, 24 /* lowmo, 120 values */ + }, + { + 2, 2, 3, 4, 4, 5, 5, 5, 6, 6, 6, 6, 6, 7, 7, 7, + 7, 8, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 10, 10, + 10, 10, 11, 11, 11, 11, 12, 12, 12, 12, 13, 13, 13, 13, 14, 14, + 14, 14, 15, 15, 15, 15, 15, 16, 16, 16, 17, 17, 17, 17, 18, 18, + 19, 19, 19, 20, 25, 25, 25, 25, 25, 25, 25, 25, 25, 25, 25, 25, + 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, 26, + 26, 26, 26, 26, 26, 26, 26, 26 /* himo, 120 values */ + } +}; + +/* DC chroma differentials low+hi-mo */ + +#endif /* VC9DATA_H */