comparison intrax8.c @ 5887:83ac4620c6ed libavcodec

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