annotate ffmpeg/libavcodec/intrax8.c @ 13:844d341cf643 tip

Back up before ISMIR
author Yading Song <yading.song@eecs.qmul.ac.uk>
date Thu, 31 Oct 2013 13:17:06 +0000
parents 6840f77b83aa
children
rev   line source
yading@10 1 /*
yading@10 2 * This file is part of FFmpeg.
yading@10 3 *
yading@10 4 * FFmpeg is free software; you can redistribute it and/or
yading@10 5 * modify it under the terms of the GNU Lesser General Public
yading@10 6 * License as published by the Free Software Foundation; either
yading@10 7 * version 2.1 of the License, or (at your option) any later version.
yading@10 8 *
yading@10 9 * FFmpeg is distributed in the hope that it will be useful,
yading@10 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
yading@10 12 * Lesser General Public License for more details.
yading@10 13 *
yading@10 14 * You should have received a copy of the GNU Lesser General Public
yading@10 15 * License along with FFmpeg; if not, write to the Free Software
yading@10 16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10 17 */
yading@10 18
yading@10 19 /**
yading@10 20 * @file
yading@10 21 * @brief IntraX8 (J-Frame) subdecoder, used by WMV2 and VC-1
yading@10 22 */
yading@10 23
yading@10 24 #include "libavutil/avassert.h"
yading@10 25 #include "avcodec.h"
yading@10 26 #include "error_resilience.h"
yading@10 27 #include "get_bits.h"
yading@10 28 #include "mpegvideo.h"
yading@10 29 #include "msmpeg4data.h"
yading@10 30 #include "intrax8huf.h"
yading@10 31 #include "intrax8.h"
yading@10 32 #include "intrax8dsp.h"
yading@10 33
yading@10 34 #define MAX_TABLE_DEPTH(table_bits, max_bits) ((max_bits+table_bits-1)/table_bits)
yading@10 35
yading@10 36 #define DC_VLC_BITS 9
yading@10 37 #define AC_VLC_BITS 9
yading@10 38 #define OR_VLC_BITS 7
yading@10 39
yading@10 40 #define DC_VLC_MTD MAX_TABLE_DEPTH(DC_VLC_BITS, MAX_DC_VLC_BITS)
yading@10 41 #define AC_VLC_MTD MAX_TABLE_DEPTH(AC_VLC_BITS, MAX_AC_VLC_BITS)
yading@10 42 #define OR_VLC_MTD MAX_TABLE_DEPTH(OR_VLC_BITS, MAX_OR_VLC_BITS)
yading@10 43
yading@10 44 static VLC j_ac_vlc[2][2][8]; //[quant<13],[intra/inter],[select]
yading@10 45 static VLC j_dc_vlc[2][8]; //[quant], [select]
yading@10 46 static VLC j_orient_vlc[2][4]; //[quant], [select]
yading@10 47
yading@10 48 static av_cold void x8_vlc_init(void){
yading@10 49 int i;
yading@10 50 int offset = 0;
yading@10 51 int sizeidx = 0;
yading@10 52 static const uint16_t sizes[8*4 + 8*2 + 2 + 4] = {
yading@10 53 576, 548, 582, 618, 546, 616, 560, 642,
yading@10 54 584, 582, 704, 664, 512, 544, 656, 640,
yading@10 55 512, 648, 582, 566, 532, 614, 596, 648,
yading@10 56 586, 552, 584, 590, 544, 578, 584, 624,
yading@10 57
yading@10 58 528, 528, 526, 528, 536, 528, 526, 544,
yading@10 59 544, 512, 512, 528, 528, 544, 512, 544,
yading@10 60
yading@10 61 128, 128, 128, 128, 128, 128};
yading@10 62
yading@10 63 static VLC_TYPE table[28150][2];
yading@10 64
yading@10 65 #define init_ac_vlc(dst,src) \
yading@10 66 dst.table = &table[offset]; \
yading@10 67 dst.table_allocated = sizes[sizeidx]; \
yading@10 68 offset += sizes[sizeidx++]; \
yading@10 69 init_vlc(&dst, \
yading@10 70 AC_VLC_BITS,77, \
yading@10 71 &src[1],4,2, \
yading@10 72 &src[0],4,2, \
yading@10 73 INIT_VLC_USE_NEW_STATIC)
yading@10 74 //set ac tables
yading@10 75 for(i=0;i<8;i++){
yading@10 76 init_ac_vlc( j_ac_vlc[0][0][i], x8_ac0_highquant_table[i][0] );
yading@10 77 init_ac_vlc( j_ac_vlc[0][1][i], x8_ac1_highquant_table[i][0] );
yading@10 78 init_ac_vlc( j_ac_vlc[1][0][i], x8_ac0_lowquant_table [i][0] );
yading@10 79 init_ac_vlc( j_ac_vlc[1][1][i], x8_ac1_lowquant_table [i][0] );
yading@10 80 }
yading@10 81 #undef init_ac_vlc
yading@10 82
yading@10 83 //set dc tables
yading@10 84 #define init_dc_vlc(dst,src) \
yading@10 85 dst.table = &table[offset]; \
yading@10 86 dst.table_allocated = sizes[sizeidx]; \
yading@10 87 offset += sizes[sizeidx++]; \
yading@10 88 init_vlc(&dst, \
yading@10 89 DC_VLC_BITS,34, \
yading@10 90 &src[1],4,2, \
yading@10 91 &src[0],4,2, \
yading@10 92 INIT_VLC_USE_NEW_STATIC);
yading@10 93 for(i=0;i<8;i++){
yading@10 94 init_dc_vlc( j_dc_vlc[0][i], x8_dc_highquant_table[i][0]);
yading@10 95 init_dc_vlc( j_dc_vlc[1][i], x8_dc_lowquant_table [i][0]);
yading@10 96 }
yading@10 97 #undef init_dc_vlc
yading@10 98
yading@10 99 //set orient tables
yading@10 100 #define init_or_vlc(dst,src) \
yading@10 101 dst.table = &table[offset]; \
yading@10 102 dst.table_allocated = sizes[sizeidx]; \
yading@10 103 offset += sizes[sizeidx++]; \
yading@10 104 init_vlc(&dst, \
yading@10 105 OR_VLC_BITS,12, \
yading@10 106 &src[1],4,2, \
yading@10 107 &src[0],4,2, \
yading@10 108 INIT_VLC_USE_NEW_STATIC);
yading@10 109 for(i=0;i<2;i++){
yading@10 110 init_or_vlc( j_orient_vlc[0][i], x8_orient_highquant_table[i][0]);
yading@10 111 }
yading@10 112 for(i=0;i<4;i++){
yading@10 113 init_or_vlc( j_orient_vlc[1][i], x8_orient_lowquant_table [i][0])
yading@10 114 }
yading@10 115 if (offset != sizeof(table)/sizeof(VLC_TYPE)/2)
yading@10 116 av_log(NULL, AV_LOG_ERROR, "table size %i does not match needed %i\n", (int)(sizeof(table)/sizeof(VLC_TYPE)/2), offset);
yading@10 117 }
yading@10 118 #undef init_or_vlc
yading@10 119
yading@10 120 static void x8_reset_vlc_tables(IntraX8Context * w){
yading@10 121 memset(w->j_dc_vlc,0,sizeof(w->j_dc_vlc));
yading@10 122 memset(w->j_ac_vlc,0,sizeof(w->j_ac_vlc));
yading@10 123 w->j_orient_vlc=NULL;
yading@10 124 }
yading@10 125
yading@10 126 static inline void x8_select_ac_table(IntraX8Context * const w , int mode){
yading@10 127 MpegEncContext * const s= w->s;
yading@10 128 int table_index;
yading@10 129
yading@10 130 av_assert2(mode<4);
yading@10 131
yading@10 132 if( w->j_ac_vlc[mode] ) return;
yading@10 133
yading@10 134 table_index = get_bits(&s->gb, 3);
yading@10 135 w->j_ac_vlc[mode] = &j_ac_vlc[w->quant<13][mode>>1][table_index];//2 modes use same tables
yading@10 136 av_assert2(w->j_ac_vlc[mode]);
yading@10 137 }
yading@10 138
yading@10 139 static inline int x8_get_orient_vlc(IntraX8Context * w){
yading@10 140 MpegEncContext * const s= w->s;
yading@10 141 int table_index;
yading@10 142
yading@10 143 if(!w->j_orient_vlc ){
yading@10 144 table_index = get_bits(&s->gb, 1+(w->quant<13) );
yading@10 145 w->j_orient_vlc = &j_orient_vlc[w->quant<13][table_index];
yading@10 146 }
yading@10 147
yading@10 148 return get_vlc2(&s->gb, w->j_orient_vlc->table, OR_VLC_BITS, OR_VLC_MTD);
yading@10 149 }
yading@10 150
yading@10 151 #define extra_bits(eb) (eb)
yading@10 152 #define extra_run (0xFF<<8)
yading@10 153 #define extra_level (0x00<<8)
yading@10 154 #define run_offset(r) ((r)<<16)
yading@10 155 #define level_offset(l) ((l)<<24)
yading@10 156 static const uint32_t ac_decode_table[]={
yading@10 157 /*46*/ extra_bits(3) | extra_run | run_offset(16) | level_offset( 0),
yading@10 158 /*47*/ extra_bits(3) | extra_run | run_offset(24) | level_offset( 0),
yading@10 159 /*48*/ extra_bits(2) | extra_run | run_offset( 4) | level_offset( 1),
yading@10 160 /*49*/ extra_bits(3) | extra_run | run_offset( 8) | level_offset( 1),
yading@10 161
yading@10 162 /*50*/ extra_bits(5) | extra_run | run_offset(32) | level_offset( 0),
yading@10 163 /*51*/ extra_bits(4) | extra_run | run_offset(16) | level_offset( 1),
yading@10 164
yading@10 165 /*52*/ extra_bits(2) | extra_level | run_offset( 0) | level_offset( 4),
yading@10 166 /*53*/ extra_bits(2) | extra_level | run_offset( 0) | level_offset( 8),
yading@10 167 /*54*/ extra_bits(2) | extra_level | run_offset( 0) | level_offset(12),
yading@10 168 /*55*/ extra_bits(3) | extra_level | run_offset( 0) | level_offset(16),
yading@10 169 /*56*/ extra_bits(3) | extra_level | run_offset( 0) | level_offset(24),
yading@10 170
yading@10 171 /*57*/ extra_bits(2) | extra_level | run_offset( 1) | level_offset( 3),
yading@10 172 /*58*/ extra_bits(3) | extra_level | run_offset( 1) | level_offset( 7),
yading@10 173
yading@10 174 /*59*/ extra_bits(2) | extra_run | run_offset(16) | level_offset( 0),
yading@10 175 /*60*/ extra_bits(2) | extra_run | run_offset(20) | level_offset( 0),
yading@10 176 /*61*/ extra_bits(2) | extra_run | run_offset(24) | level_offset( 0),
yading@10 177 /*62*/ extra_bits(2) | extra_run | run_offset(28) | level_offset( 0),
yading@10 178 /*63*/ extra_bits(4) | extra_run | run_offset(32) | level_offset( 0),
yading@10 179 /*64*/ extra_bits(4) | extra_run | run_offset(48) | level_offset( 0),
yading@10 180
yading@10 181 /*65*/ extra_bits(2) | extra_run | run_offset( 4) | level_offset( 1),
yading@10 182 /*66*/ extra_bits(3) | extra_run | run_offset( 8) | level_offset( 1),
yading@10 183 /*67*/ extra_bits(4) | extra_run | run_offset(16) | level_offset( 1),
yading@10 184
yading@10 185 /*68*/ extra_bits(2) | extra_level | run_offset( 0) | level_offset( 4),
yading@10 186 /*69*/ extra_bits(3) | extra_level | run_offset( 0) | level_offset( 8),
yading@10 187 /*70*/ extra_bits(4) | extra_level | run_offset( 0) | level_offset(16),
yading@10 188
yading@10 189 /*71*/ extra_bits(2) | extra_level | run_offset( 1) | level_offset( 3),
yading@10 190 /*72*/ extra_bits(3) | extra_level | run_offset( 1) | level_offset( 7),
yading@10 191 };
yading@10 192 //extra_bits = 3bits; extra_run/level = 1 bit; run_offset = 6bits; level_offset = 5 bits;
yading@10 193 #undef extra_bits
yading@10 194 #undef extra_run
yading@10 195 #undef extra_level
yading@10 196 #undef run_offset
yading@10 197 #undef level_offset
yading@10 198
yading@10 199 static void x8_get_ac_rlf(IntraX8Context * const w, const int mode,
yading@10 200 int * const run, int * const level, int * const final){
yading@10 201 MpegEncContext * const s= w->s;
yading@10 202 int i,e;
yading@10 203
yading@10 204 // x8_select_ac_table(w,mode);
yading@10 205 i = get_vlc2(&s->gb, w->j_ac_vlc[mode]->table, AC_VLC_BITS, AC_VLC_MTD);
yading@10 206
yading@10 207 if(i<46){ //[0-45]
yading@10 208 int t,l;
yading@10 209 if(i<0){
yading@10 210 (*level)=(*final)=//prevent 'may be used unilitialized'
yading@10 211 (*run)=64;//this would cause error exit in the ac loop
yading@10 212 return;
yading@10 213 }
yading@10 214
yading@10 215 (*final) = t = (i>22);
yading@10 216 i-=23*t;
yading@10 217 /*
yading@10 218 i== 0-15 r=0-15 l=0 ;r=i& %01111
yading@10 219 i==16-19 r=0-3 l=1 ;r=i& %00011
yading@10 220 i==20-21 r=0-1 l=2 ;r=i& %00001
yading@10 221 i==22 r=0 l=3 ;r=i& %00000
yading@10 222 l=lut_l[i/2]={0,0,0,0,0,0,0,0,1,1,2,3}[i>>1];// 11 10'01 01'00 00'00 00'00 00'00 00 => 0xE50000
yading@10 223 t=lut_mask[l]={0x0f,0x03,0x01,0x00}[l]; as i<256 the higher bits do not matter */
yading@10 224 l=(0xE50000>>(i&(0x1E)))&3;/*0x1E or (~1) or ((i>>1)<<1)*/
yading@10 225 t=(0x01030F>>(l<<3));
yading@10 226
yading@10 227 (*run) = i&t;
yading@10 228 (*level) = l;
yading@10 229 }else if(i<73){//[46-72]
yading@10 230 uint32_t sm;
yading@10 231 uint32_t mask;
yading@10 232
yading@10 233 i-=46;
yading@10 234 sm=ac_decode_table[i];
yading@10 235
yading@10 236 e=get_bits(&s->gb,sm&0xF);sm>>=8;//3bits
yading@10 237 mask=sm&0xff;sm>>=8; //1bit
yading@10 238
yading@10 239 (*run) =(sm&0xff) + (e&( mask));//6bits
yading@10 240 (*level)=(sm>>8) + (e&(~mask));//5bits
yading@10 241 (*final)=i>(58-46);
yading@10 242 }else if(i<75){//[73-74]
yading@10 243 static const uint8_t crazy_mix_runlevel[32]={
yading@10 244 0x22,0x32,0x33,0x53,0x23,0x42,0x43,0x63,
yading@10 245 0x24,0x52,0x34,0x73,0x25,0x62,0x44,0x83,
yading@10 246 0x26,0x72,0x35,0x54,0x27,0x82,0x45,0x64,
yading@10 247 0x28,0x92,0x36,0x74,0x29,0xa2,0x46,0x84};
yading@10 248
yading@10 249 (*final)=!(i&1);
yading@10 250 e=get_bits(&s->gb,5);//get the extra bits
yading@10 251 (*run) =crazy_mix_runlevel[e]>>4;
yading@10 252 (*level)=crazy_mix_runlevel[e]&0x0F;
yading@10 253 }else{
yading@10 254 (*level)=get_bits( &s->gb, 7-3*(i&1));
yading@10 255 (*run) =get_bits( &s->gb, 6);
yading@10 256 (*final)=get_bits1(&s->gb);
yading@10 257 }
yading@10 258 return;
yading@10 259 }
yading@10 260
yading@10 261 //static const uint8_t dc_extra_sbits[] ={0, 1,1, 1,1, 2,2, 3,3, 4,4, 5,5, 6,6, 7,7 };
yading@10 262 static const uint8_t dc_index_offset[] ={ 0, 1,2, 3,4, 5,7, 9,13, 17,25, 33,49, 65,97, 129,193};
yading@10 263
yading@10 264 static int x8_get_dc_rlf(IntraX8Context * const w,int const mode, int * const level, int * const final){
yading@10 265 MpegEncContext * const s= w->s;
yading@10 266 int i,e,c;
yading@10 267
yading@10 268 av_assert2(mode<3);
yading@10 269 if( !w->j_dc_vlc[mode] ) {
yading@10 270 int table_index;
yading@10 271 table_index = get_bits(&s->gb, 3);
yading@10 272 //4 modes, same table
yading@10 273 w->j_dc_vlc[mode]= &j_dc_vlc[w->quant<13][table_index];
yading@10 274 }
yading@10 275
yading@10 276 i=get_vlc2(&s->gb, w->j_dc_vlc[mode]->table, DC_VLC_BITS, DC_VLC_MTD);
yading@10 277
yading@10 278 /*(i>=17) {i-=17;final=1;}*/
yading@10 279 c= i>16;
yading@10 280 (*final)=c;
yading@10 281 i-=17*c;
yading@10 282
yading@10 283 if(i<=0){
yading@10 284 (*level)=0;
yading@10 285 return -i;
yading@10 286 }
yading@10 287 c=(i+1)>>1;//hackish way to calculate dc_extra_sbits[]
yading@10 288 c-=c>1;
yading@10 289
yading@10 290 e=get_bits(&s->gb,c);//get the extra bits
yading@10 291 i=dc_index_offset[i]+(e>>1);
yading@10 292
yading@10 293 e= -(e & 1);//0,0xffffff
yading@10 294 (*level)= (i ^ e) - e;// (i^0)-0 , (i^0xff)-(-1)
yading@10 295 return 0;
yading@10 296 }
yading@10 297 //end of huffman
yading@10 298
yading@10 299 static int x8_setup_spatial_predictor(IntraX8Context * const w, const int chroma){
yading@10 300 MpegEncContext * const s= w->s;
yading@10 301 int range;
yading@10 302 int sum;
yading@10 303 int quant;
yading@10 304
yading@10 305 w->dsp.setup_spatial_compensation(s->dest[chroma], s->edge_emu_buffer,
yading@10 306 s->current_picture.f.linesize[chroma>0],
yading@10 307 &range, &sum, w->edges);
yading@10 308 if(chroma){
yading@10 309 w->orient=w->chroma_orient;
yading@10 310 quant=w->quant_dc_chroma;
yading@10 311 }else{
yading@10 312 quant=w->quant;
yading@10 313 }
yading@10 314
yading@10 315 w->flat_dc=0;
yading@10 316 if(range < quant || range < 3){
yading@10 317 w->orient=0;
yading@10 318 if(range < 3){//yep you read right, a +-1 idct error may break decoding!
yading@10 319 w->flat_dc=1;
yading@10 320 sum+=9;
yading@10 321 w->predicted_dc = (sum*6899)>>17;//((1<<17)+9)/(8+8+1+2)=6899
yading@10 322 }
yading@10 323 }
yading@10 324 if(chroma)
yading@10 325 return 0;
yading@10 326
yading@10 327 av_assert2(w->orient < 3);
yading@10 328 if(range < 2*w->quant){
yading@10 329 if( (w->edges&3) == 0){
yading@10 330 if(w->orient==1) w->orient=11;
yading@10 331 if(w->orient==2) w->orient=10;
yading@10 332 }else{
yading@10 333 w->orient=0;
yading@10 334 }
yading@10 335 w->raw_orient=0;
yading@10 336 }else{
yading@10 337 static const uint8_t prediction_table[3][12]={
yading@10 338 {0,8,4, 10,11, 2,6,9,1,3,5,7},
yading@10 339 {4,0,8, 11,10, 3,5,2,6,9,1,7},
yading@10 340 {8,0,4, 10,11, 1,7,2,6,9,3,5}
yading@10 341 };
yading@10 342 w->raw_orient=x8_get_orient_vlc(w);
yading@10 343 if(w->raw_orient<0) return -1;
yading@10 344 av_assert2(w->raw_orient < 12 );
yading@10 345 av_assert2(w->orient<3);
yading@10 346 w->orient=prediction_table[w->orient][w->raw_orient];
yading@10 347 }
yading@10 348 return 0;
yading@10 349 }
yading@10 350
yading@10 351 static void x8_update_predictions(IntraX8Context * const w, const int orient, const int est_run ){
yading@10 352 MpegEncContext * const s= w->s;
yading@10 353
yading@10 354 w->prediction_table[s->mb_x*2+(s->mb_y&1)] = (est_run<<2) + 1*(orient==4) + 2*(orient==8);
yading@10 355 /*
yading@10 356 y=2n+0 ->//0 2 4
yading@10 357 y=2n+1 ->//1 3 5
yading@10 358 */
yading@10 359 }
yading@10 360 static void x8_get_prediction_chroma(IntraX8Context * const w){
yading@10 361 MpegEncContext * const s= w->s;
yading@10 362
yading@10 363 w->edges = 1*( !(s->mb_x>>1) );
yading@10 364 w->edges|= 2*( !(s->mb_y>>1) );
yading@10 365 w->edges|= 4*( s->mb_x >= (2*s->mb_width-1) );//mb_x for chroma would always be odd
yading@10 366
yading@10 367 w->raw_orient=0;
yading@10 368 if(w->edges&3){//lut_co[8]={inv,4,8,8, inv,4,8,8}<- =>{1,1,0,0;1,1,0,0} => 0xCC
yading@10 369 w->chroma_orient=4<<((0xCC>>w->edges)&1);
yading@10 370 return;
yading@10 371 }
yading@10 372 w->chroma_orient = (w->prediction_table[2*s->mb_x-2] & 0x03)<<2;//block[x-1][y|1-1)]
yading@10 373 }
yading@10 374
yading@10 375 static void x8_get_prediction(IntraX8Context * const w){
yading@10 376 MpegEncContext * const s= w->s;
yading@10 377 int a,b,c,i;
yading@10 378
yading@10 379 w->edges = 1*( !s->mb_x );
yading@10 380 w->edges|= 2*( !s->mb_y );
yading@10 381 w->edges|= 4*( s->mb_x >= (2*s->mb_width-1) );
yading@10 382
yading@10 383 switch(w->edges&3){
yading@10 384 case 0:
yading@10 385 break;
yading@10 386 case 1:
yading@10 387 //take the one from the above block[0][y-1]
yading@10 388 w->est_run = w->prediction_table[!(s->mb_y&1)]>>2;
yading@10 389 w->orient = 1;
yading@10 390 return;
yading@10 391 case 2:
yading@10 392 //take the one from the previous block[x-1][0]
yading@10 393 w->est_run = w->prediction_table[2*s->mb_x-2]>>2;
yading@10 394 w->orient = 2;
yading@10 395 return;
yading@10 396 case 3:
yading@10 397 w->est_run = 16;
yading@10 398 w->orient = 0;
yading@10 399 return;
yading@10 400 }
yading@10 401 //no edge cases
yading@10 402 b= w->prediction_table[2*s->mb_x + !(s->mb_y&1) ];//block[x ][y-1]
yading@10 403 a= w->prediction_table[2*s->mb_x-2 + (s->mb_y&1) ];//block[x-1][y ]
yading@10 404 c= w->prediction_table[2*s->mb_x-2 + !(s->mb_y&1) ];//block[x-1][y-1]
yading@10 405
yading@10 406 w->est_run = FFMIN(b,a);
yading@10 407 /* This condition has nothing to do with w->edges, even if it looks
yading@10 408 similar it would trigger if e.g. x=3;y=2;
yading@10 409 I guess somebody wrote something wrong and it became standard. */
yading@10 410 if( (s->mb_x & s->mb_y) != 0 ) w->est_run=FFMIN(c,w->est_run);
yading@10 411 w->est_run>>=2;
yading@10 412
yading@10 413 a&=3;
yading@10 414 b&=3;
yading@10 415 c&=3;
yading@10 416
yading@10 417 i=( 0xFFEAF4C4>>(2*b+8*a) )&3;
yading@10 418 if(i!=3) w->orient=i;
yading@10 419 else w->orient=( 0xFFEAD8>>(2*c+8*(w->quant>12)) )&3;
yading@10 420 /*
yading@10 421 lut1[b][a]={
yading@10 422 ->{0, 1, 0, pad},
yading@10 423 {0, 1, X, pad},
yading@10 424 {2, 2, 2, pad}}
yading@10 425 pad 2 2 2; pad X 1 0; pad 0 1 0 <-
yading@10 426 -> 11 10 '10 10 '11 11'01 00 '11 00'01 00=>0xEAF4C4
yading@10 427
yading@10 428 lut2[q>12][c]={
yading@10 429 ->{0,2,1,pad},
yading@10 430 {2,2,2,pad}}
yading@10 431 pad 2 2 2; pad 1 2 0 <-
yading@10 432 -> 11 10'10 10 '11 01'10 00=>0xEAD8
yading@10 433 */
yading@10 434 }
yading@10 435
yading@10 436
yading@10 437 static void x8_ac_compensation(IntraX8Context * const w, int const direction, int const dc_level){
yading@10 438 MpegEncContext * const s= w->s;
yading@10 439 int t;
yading@10 440 #define B(x,y) s->block[0][s->dsp.idct_permutation[(x)+(y)*8]]
yading@10 441 #define T(x) ((x) * dc_level + 0x8000) >> 16;
yading@10 442 switch(direction){
yading@10 443 case 0:
yading@10 444 t = T(3811);//h
yading@10 445 B(1,0) -= t;
yading@10 446 B(0,1) -= t;
yading@10 447
yading@10 448 t = T(487);//e
yading@10 449 B(2,0) -= t;
yading@10 450 B(0,2) -= t;
yading@10 451
yading@10 452 t = T(506);//f
yading@10 453 B(3,0) -= t;
yading@10 454 B(0,3) -= t;
yading@10 455
yading@10 456 t = T(135);//c
yading@10 457 B(4,0) -= t;
yading@10 458 B(0,4) -= t;
yading@10 459 B(2,1) += t;
yading@10 460 B(1,2) += t;
yading@10 461 B(3,1) += t;
yading@10 462 B(1,3) += t;
yading@10 463
yading@10 464 t = T(173);//d
yading@10 465 B(5,0) -= t;
yading@10 466 B(0,5) -= t;
yading@10 467
yading@10 468 t = T(61);//b
yading@10 469 B(6,0) -= t;
yading@10 470 B(0,6) -= t;
yading@10 471 B(5,1) += t;
yading@10 472 B(1,5) += t;
yading@10 473
yading@10 474 t = T(42); //a
yading@10 475 B(7,0) -= t;
yading@10 476 B(0,7) -= t;
yading@10 477 B(4,1) += t;
yading@10 478 B(1,4) += t;
yading@10 479 B(4,4) += t;
yading@10 480
yading@10 481 t = T(1084);//g
yading@10 482 B(1,1) += t;
yading@10 483
yading@10 484 s->block_last_index[0] = FFMAX(s->block_last_index[0], 7*8);
yading@10 485 break;
yading@10 486 case 1:
yading@10 487 B(0,1) -= T(6269);
yading@10 488 B(0,3) -= T( 708);
yading@10 489 B(0,5) -= T( 172);
yading@10 490 B(0,7) -= T( 73);
yading@10 491
yading@10 492 s->block_last_index[0] = FFMAX(s->block_last_index[0], 7*8);
yading@10 493 break;
yading@10 494 case 2:
yading@10 495 B(1,0) -= T(6269);
yading@10 496 B(3,0) -= T( 708);
yading@10 497 B(5,0) -= T( 172);
yading@10 498 B(7,0) -= T( 73);
yading@10 499
yading@10 500 s->block_last_index[0] = FFMAX(s->block_last_index[0], 7);
yading@10 501 break;
yading@10 502 }
yading@10 503 #undef B
yading@10 504 #undef T
yading@10 505 }
yading@10 506
yading@10 507 static void dsp_x8_put_solidcolor(uint8_t const pix, uint8_t * dst, int const linesize){
yading@10 508 int k;
yading@10 509 for(k=0;k<8;k++){
yading@10 510 memset(dst,pix,8);
yading@10 511 dst+=linesize;
yading@10 512 }
yading@10 513 }
yading@10 514
yading@10 515 static const int16_t quant_table[64] = {
yading@10 516 256, 256, 256, 256, 256, 256, 259, 262,
yading@10 517 265, 269, 272, 275, 278, 282, 285, 288,
yading@10 518 292, 295, 299, 303, 306, 310, 314, 317,
yading@10 519 321, 325, 329, 333, 337, 341, 345, 349,
yading@10 520 353, 358, 362, 366, 371, 375, 379, 384,
yading@10 521 389, 393, 398, 403, 408, 413, 417, 422,
yading@10 522 428, 433, 438, 443, 448, 454, 459, 465,
yading@10 523 470, 476, 482, 488, 493, 499, 505, 511
yading@10 524 };
yading@10 525
yading@10 526 static int x8_decode_intra_mb(IntraX8Context* const w, const int chroma){
yading@10 527 MpegEncContext * const s= w->s;
yading@10 528
yading@10 529 uint8_t * scantable;
yading@10 530 int final,run,level;
yading@10 531 int ac_mode,dc_mode,est_run,dc_level;
yading@10 532 int pos,n;
yading@10 533 int zeros_only;
yading@10 534 int use_quant_matrix;
yading@10 535 int sign;
yading@10 536
yading@10 537 av_assert2(w->orient<12);
yading@10 538 s->dsp.clear_block(s->block[0]);
yading@10 539
yading@10 540 if(chroma){
yading@10 541 dc_mode=2;
yading@10 542 }else{
yading@10 543 dc_mode=!!w->est_run;//0,1
yading@10 544 }
yading@10 545
yading@10 546 if(x8_get_dc_rlf(w, dc_mode, &dc_level, &final)) return -1;
yading@10 547 n=0;
yading@10 548 zeros_only=0;
yading@10 549 if(!final){//decode ac
yading@10 550 use_quant_matrix=w->use_quant_matrix;
yading@10 551 if(chroma){
yading@10 552 ac_mode = 1;
yading@10 553 est_run = 64;//not used
yading@10 554 }else{
yading@10 555 if (w->raw_orient < 3){
yading@10 556 use_quant_matrix = 0;
yading@10 557 }
yading@10 558 if(w->raw_orient > 4){
yading@10 559 ac_mode = 0;
yading@10 560 est_run = 64;
yading@10 561 }else{
yading@10 562 if(w->est_run > 1){
yading@10 563 ac_mode = 2;
yading@10 564 est_run=w->est_run;
yading@10 565 }else{
yading@10 566 ac_mode = 3;
yading@10 567 est_run = 64;
yading@10 568 }
yading@10 569 }
yading@10 570 }
yading@10 571 x8_select_ac_table(w,ac_mode);
yading@10 572 /*scantable_selector[12]={0,2,0,1,1,1,0,2,2,0,1,2};<-
yading@10 573 -> 10'01' 00'10' 10'00' 01'01' 01'00' 10'00 =>0x928548 */
yading@10 574 scantable = w->scantable[ (0x928548>>(2*w->orient))&3 ].permutated;
yading@10 575 pos=0;
yading@10 576 do {
yading@10 577 n++;
yading@10 578 if( n >= est_run ){
yading@10 579 ac_mode=3;
yading@10 580 x8_select_ac_table(w,3);
yading@10 581 }
yading@10 582
yading@10 583 x8_get_ac_rlf(w,ac_mode,&run,&level,&final);
yading@10 584
yading@10 585 pos+=run+1;
yading@10 586 if(pos>63){
yading@10 587 //this also handles vlc error in x8_get_ac_rlf
yading@10 588 return -1;
yading@10 589 }
yading@10 590 level= (level+1) * w->dquant;
yading@10 591 level+= w->qsum;
yading@10 592
yading@10 593 sign = - get_bits1(&s->gb);
yading@10 594 level = (level ^ sign) - sign;
yading@10 595
yading@10 596 if(use_quant_matrix){
yading@10 597 level = (level*quant_table[pos])>>8;
yading@10 598 }
yading@10 599 s->block[0][ scantable[pos] ]=level;
yading@10 600 }while(!final);
yading@10 601
yading@10 602 s->block_last_index[0]=pos;
yading@10 603 }else{//DC only
yading@10 604 s->block_last_index[0]=0;
yading@10 605 if(w->flat_dc && ((unsigned)(dc_level+1)) < 3){//[-1;1]
yading@10 606 int32_t divide_quant= !chroma ? w->divide_quant_dc_luma:
yading@10 607 w->divide_quant_dc_chroma;
yading@10 608 int32_t dc_quant = !chroma ? w->quant:
yading@10 609 w->quant_dc_chroma;
yading@10 610
yading@10 611 //original intent dc_level+=predicted_dc/quant; but it got lost somewhere in the rounding
yading@10 612 dc_level+= (w->predicted_dc*divide_quant + (1<<12) )>>13;
yading@10 613
yading@10 614 dsp_x8_put_solidcolor( av_clip_uint8((dc_level*dc_quant+4)>>3),
yading@10 615 s->dest[chroma], s->current_picture.f.linesize[!!chroma]);
yading@10 616
yading@10 617 goto block_placed;
yading@10 618 }
yading@10 619 zeros_only = (dc_level == 0);
yading@10 620 }
yading@10 621 if(!chroma){
yading@10 622 s->block[0][0] = dc_level*w->quant;
yading@10 623 }else{
yading@10 624 s->block[0][0] = dc_level*w->quant_dc_chroma;
yading@10 625 }
yading@10 626
yading@10 627 //there is !zero_only check in the original, but dc_level check is enough
yading@10 628 if( (unsigned int)(dc_level+1) >= 3 && (w->edges&3) != 3 ){
yading@10 629 int direction;
yading@10 630 /*ac_comp_direction[orient] = { 0, 3, 3, 1, 1, 0, 0, 0, 2, 2, 2, 1 };<-
yading@10 631 -> 01'10' 10'10' 00'00' 00'01' 01'11' 11'00 =>0x6A017C */
yading@10 632 direction= (0x6A017C>>(w->orient*2))&3;
yading@10 633 if (direction != 3){
yading@10 634 x8_ac_compensation(w, direction, s->block[0][0]);//modify block_last[]
yading@10 635 }
yading@10 636 }
yading@10 637
yading@10 638 if(w->flat_dc){
yading@10 639 dsp_x8_put_solidcolor(w->predicted_dc, s->dest[chroma], s->current_picture.f.linesize[!!chroma]);
yading@10 640 }else{
yading@10 641 w->dsp.spatial_compensation[w->orient]( s->edge_emu_buffer,
yading@10 642 s->dest[chroma],
yading@10 643 s->current_picture.f.linesize[!!chroma] );
yading@10 644 }
yading@10 645 if(!zeros_only)
yading@10 646 s->dsp.idct_add ( s->dest[chroma],
yading@10 647 s->current_picture.f.linesize[!!chroma],
yading@10 648 s->block[0] );
yading@10 649
yading@10 650 block_placed:
yading@10 651
yading@10 652 if(!chroma){
yading@10 653 x8_update_predictions(w,w->orient,n);
yading@10 654 }
yading@10 655
yading@10 656 if(s->loop_filter){
yading@10 657 uint8_t* ptr = s->dest[chroma];
yading@10 658 int linesize = s->current_picture.f.linesize[!!chroma];
yading@10 659
yading@10 660 if(!( (w->edges&2) || ( zeros_only && (w->orient|4)==4 ) )){
yading@10 661 w->dsp.h_loop_filter(ptr, linesize, w->quant);
yading@10 662 }
yading@10 663 if(!( (w->edges&1) || ( zeros_only && (w->orient|8)==8 ) )){
yading@10 664 w->dsp.v_loop_filter(ptr, linesize, w->quant);
yading@10 665 }
yading@10 666 }
yading@10 667 return 0;
yading@10 668 }
yading@10 669
yading@10 670 static void x8_init_block_index(MpegEncContext *s){ //FIXME maybe merge with ff_*
yading@10 671 //not s->linesize as this would be wrong for field pics
yading@10 672 //not that IntraX8 has interlacing support ;)
yading@10 673 const int linesize = s->current_picture.f.linesize[0];
yading@10 674 const int uvlinesize = s->current_picture.f.linesize[1];
yading@10 675
yading@10 676 s->dest[0] = s->current_picture.f.data[0];
yading@10 677 s->dest[1] = s->current_picture.f.data[1];
yading@10 678 s->dest[2] = s->current_picture.f.data[2];
yading@10 679
yading@10 680 s->dest[0] += s->mb_y * linesize << 3;
yading@10 681 s->dest[1] += ( s->mb_y&(~1) ) * uvlinesize << 2;//chroma blocks are on add rows
yading@10 682 s->dest[2] += ( s->mb_y&(~1) ) * uvlinesize << 2;
yading@10 683 }
yading@10 684
yading@10 685 /**
yading@10 686 * Initialize IntraX8 frame decoder.
yading@10 687 * Requires valid MpegEncContext with valid s->mb_width before calling.
yading@10 688 * @param w pointer to IntraX8Context
yading@10 689 * @param s pointer to MpegEncContext of the parent codec
yading@10 690 */
yading@10 691 av_cold void ff_intrax8_common_init(IntraX8Context * w, MpegEncContext * const s){
yading@10 692
yading@10 693 w->s=s;
yading@10 694 x8_vlc_init();
yading@10 695 av_assert0(s->mb_width>0);
yading@10 696 w->prediction_table=av_mallocz(s->mb_width*2*2);//two rows, 2 blocks per cannon mb
yading@10 697
yading@10 698 ff_init_scantable(s->dsp.idct_permutation, &w->scantable[0], ff_wmv1_scantable[0]);
yading@10 699 ff_init_scantable(s->dsp.idct_permutation, &w->scantable[1], ff_wmv1_scantable[2]);
yading@10 700 ff_init_scantable(s->dsp.idct_permutation, &w->scantable[2], ff_wmv1_scantable[3]);
yading@10 701
yading@10 702 ff_intrax8dsp_init(&w->dsp);
yading@10 703 }
yading@10 704
yading@10 705 /**
yading@10 706 * Destroy IntraX8 frame structure.
yading@10 707 * @param w pointer to IntraX8Context
yading@10 708 */
yading@10 709 av_cold void ff_intrax8_common_end(IntraX8Context * w)
yading@10 710 {
yading@10 711 av_freep(&w->prediction_table);
yading@10 712 }
yading@10 713
yading@10 714 /**
yading@10 715 * Decode single IntraX8 frame.
yading@10 716 * The parent codec must fill s->loopfilter and s->gb (bitstream).
yading@10 717 * The parent codec must call MPV_frame_start(), ff_er_frame_start() before calling this function.
yading@10 718 * The parent codec must call ff_er_frame_end(), MPV_frame_end() after calling this function.
yading@10 719 * This function does not use MPV_decode_mb().
yading@10 720 * lowres decoding is theoretically impossible.
yading@10 721 * @param w pointer to IntraX8Context
yading@10 722 * @param dquant doubled quantizer, it would be odd in case of VC-1 halfpq==1.
yading@10 723 * @param quant_offset offset away from zero
yading@10 724 */
yading@10 725 //FIXME extern uint8_t ff_wmv3_dc_scale_table[32];
yading@10 726 int ff_intrax8_decode_picture(IntraX8Context * const w, int dquant, int quant_offset){
yading@10 727 MpegEncContext * const s= w->s;
yading@10 728 int mb_xy;
yading@10 729 w->use_quant_matrix = get_bits1(&s->gb);
yading@10 730
yading@10 731 w->dquant = dquant;
yading@10 732 w->quant = dquant >> 1;
yading@10 733 w->qsum = quant_offset;
yading@10 734
yading@10 735 w->divide_quant_dc_luma = ((1<<16) + (w->quant>>1)) / w->quant;
yading@10 736 if(w->quant < 5){
yading@10 737 w->quant_dc_chroma = w->quant;
yading@10 738 w->divide_quant_dc_chroma = w->divide_quant_dc_luma;
yading@10 739 }else{
yading@10 740 w->quant_dc_chroma = w->quant+((w->quant+3)>>3);
yading@10 741 w->divide_quant_dc_chroma = ((1<<16) + (w->quant_dc_chroma>>1)) / w->quant_dc_chroma;
yading@10 742 }
yading@10 743 x8_reset_vlc_tables(w);
yading@10 744
yading@10 745 s->resync_mb_x=0;
yading@10 746 s->resync_mb_y=0;
yading@10 747
yading@10 748 for(s->mb_y=0; s->mb_y < s->mb_height*2; s->mb_y++){
yading@10 749 x8_init_block_index(s);
yading@10 750 mb_xy=(s->mb_y>>1)*s->mb_stride;
yading@10 751
yading@10 752 for(s->mb_x=0; s->mb_x < s->mb_width*2; s->mb_x++){
yading@10 753 x8_get_prediction(w);
yading@10 754 if(x8_setup_spatial_predictor(w,0)) goto error;
yading@10 755 if(x8_decode_intra_mb(w,0)) goto error;
yading@10 756
yading@10 757 if( s->mb_x & s->mb_y & 1 ){
yading@10 758 x8_get_prediction_chroma(w);
yading@10 759
yading@10 760 /*when setting up chroma, no vlc is read,
yading@10 761 so no error condition can be reached*/
yading@10 762 x8_setup_spatial_predictor(w,1);
yading@10 763 if(x8_decode_intra_mb(w,1)) goto error;
yading@10 764
yading@10 765 x8_setup_spatial_predictor(w,2);
yading@10 766 if(x8_decode_intra_mb(w,2)) goto error;
yading@10 767
yading@10 768 s->dest[1]+= 8;
yading@10 769 s->dest[2]+= 8;
yading@10 770
yading@10 771 /*emulate MB info in the relevant tables*/
yading@10 772 s->mbskip_table [mb_xy]=0;
yading@10 773 s->mbintra_table[mb_xy]=1;
yading@10 774 s->current_picture.qscale_table[mb_xy] = w->quant;
yading@10 775 mb_xy++;
yading@10 776 }
yading@10 777 s->dest[0]+= 8;
yading@10 778 }
yading@10 779 if(s->mb_y&1){
yading@10 780 ff_mpeg_draw_horiz_band(s, (s->mb_y-1)*8, 16);
yading@10 781 }
yading@10 782 }
yading@10 783
yading@10 784 error:
yading@10 785 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
yading@10 786 (s->mb_x>>1)-1, (s->mb_y>>1)-1,
yading@10 787 ER_MB_END );
yading@10 788 return 0;
yading@10 789 }