2 * H263/MPEG4 backend for ffmpeg encoder and decoder
3 * Copyright (c) 2000,2001 Fabrice Bellard
5 * Copyright (c) 2001 Juan J. Sierralta P
6 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
8 * ac prediction encoding, B-frame support, error resilience, optimizations,
9 * qpel decoding, gmc decoding, interlaced decoding
10 * by Michael Niedermayer <michaelni@gmx.at>
12 * This file is part of FFmpeg.
14 * FFmpeg is free software; you can redistribute it and/or
15 * modify it under the terms of the GNU Lesser General Public
16 * License as published by the Free Software Foundation; either
17 * version 2.1 of the License, or (at your option) any later version.
19 * FFmpeg is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22 * Lesser General Public License for more details.
24 * You should have received a copy of the GNU Lesser General Public
25 * License along with FFmpeg; if not, write to the Free Software
26 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
30 * @file libavcodec/h263.c
39 #include "mpegvideo.h"
41 #include "mpeg4data.h"
47 #define INTRA_MCBPC_VLC_BITS 6
48 #define INTER_MCBPC_VLC_BITS 7
49 #define CBPY_VLC_BITS 6
52 #define SPRITE_TRAJ_VLC_BITS 6
53 #define MB_TYPE_B_VLC_BITS 4
54 #define TEX_VLC_BITS 9
55 #define H263_MBTYPE_B_VLC_BITS 6
56 #define CBPC_B_VLC_BITS 3
59 static void h263_encode_block(MpegEncContext * s, DCTELEM * block,
61 static void h263p_encode_umotion(MpegEncContext * s, int val);
62 static inline void mpeg4_encode_block(MpegEncContext * s, DCTELEM * block,
63 int n, int dc, uint8_t *scan_table,
64 PutBitContext *dc_pb, PutBitContext *ac_pb);
65 static int mpeg4_get_block_length(MpegEncContext * s, DCTELEM * block, int n, int intra_dc,
69 static int h263_decode_motion(MpegEncContext * s, int pred, int fcode);
70 static int h263p_decode_umotion(MpegEncContext * s, int pred);
71 static int h263_decode_block(MpegEncContext * s, DCTELEM * block,
73 static inline int mpeg4_decode_dc(MpegEncContext * s, int n, int *dir_ptr);
74 static inline int mpeg4_decode_block(MpegEncContext * s, DCTELEM * block,
75 int n, int coded, int intra, int rvlc);
77 static int h263_pred_dc(MpegEncContext * s, int n, int16_t **dc_val_ptr);
78 static void mpeg4_encode_visual_object_header(MpegEncContext * s);
79 static void mpeg4_encode_vol_header(MpegEncContext * s, int vo_number, int vol_number);
80 #endif //CONFIG_ENCODERS
81 static void mpeg4_decode_sprite_trajectory(MpegEncContext * s, GetBitContext *gb);
82 static inline int ff_mpeg4_pred_dc(MpegEncContext * s, int n, int level, int *dir_ptr, int encoding);
85 static uint8_t uni_DCtab_lum_len[512];
86 static uint8_t uni_DCtab_chrom_len[512];
87 static uint16_t uni_DCtab_lum_bits[512];
88 static uint16_t uni_DCtab_chrom_bits[512];
90 static uint8_t mv_penalty[MAX_FCODE+1][MAX_MV*2+1];
91 static uint8_t fcode_tab[MAX_MV*2+1];
92 static uint8_t umv_fcode_tab[MAX_MV*2+1];
94 static uint32_t uni_mpeg4_intra_rl_bits[64*64*2*2];
95 static uint8_t uni_mpeg4_intra_rl_len [64*64*2*2];
96 static uint32_t uni_mpeg4_inter_rl_bits[64*64*2*2];
97 static uint8_t uni_mpeg4_inter_rl_len [64*64*2*2];
98 static uint8_t uni_h263_intra_aic_rl_len [64*64*2*2];
99 static uint8_t uni_h263_inter_rl_len [64*64*2*2];
100 //#define UNI_MPEG4_ENC_INDEX(last,run,level) ((last)*128 + (run)*256 + (level))
101 //#define UNI_MPEG4_ENC_INDEX(last,run,level) ((last)*128*64 + (run) + (level)*64)
102 #define UNI_MPEG4_ENC_INDEX(last,run,level) ((last)*128*64 + (run)*128 + (level))
115 static uint8_t static_rl_table_store[5][2][2*MAX_RUN + MAX_LEVEL + 3];
117 #if 0 //3IV1 is quite rare and it slows things down a tiny bit
118 #define IS_3IV1 s->codec_tag == AV_RL32("3IV1")
123 int h263_get_picture_format(int width, int height)
127 if (width == 128 && height == 96)
129 else if (width == 176 && height == 144)
131 else if (width == 352 && height == 288)
133 else if (width == 704 && height == 576)
135 else if (width == 1408 && height == 1152)
142 static void show_pict_info(MpegEncContext *s){
143 av_log(s->avctx, AV_LOG_DEBUG, "qp:%d %c size:%d rnd:%d%s%s%s%s%s%s%s%s%s %d/%d\n",
144 s->qscale, av_get_pict_type_char(s->pict_type),
145 s->gb.size_in_bits, 1-s->no_rounding,
146 s->obmc ? " AP" : "",
147 s->umvplus ? " UMV" : "",
148 s->h263_long_vectors ? " LONG" : "",
149 s->h263_plus ? " +" : "",
150 s->h263_aic ? " AIC" : "",
151 s->alt_inter_vlc ? " AIV" : "",
152 s->modified_quant ? " MQ" : "",
153 s->loop_filter ? " LOOP" : "",
154 s->h263_slice_structured ? " SS" : "",
155 s->avctx->time_base.den, s->avctx->time_base.num
161 static void aspect_to_info(MpegEncContext * s, AVRational aspect){
164 if(aspect.num==0) aspect= (AVRational){1,1};
167 if(av_cmp_q(pixel_aspect[i], aspect) == 0){
168 s->aspect_ratio_info=i;
173 s->aspect_ratio_info= FF_ASPECT_EXTENDED;
176 void ff_flv_encode_picture_header(MpegEncContext * s, int picture_number)
180 align_put_bits(&s->pb);
182 put_bits(&s->pb, 17, 1);
183 put_bits(&s->pb, 5, (s->h263_flv-1)); /* 0: h263 escape codes 1: 11-bit escape codes */
184 put_bits(&s->pb, 8, (((int64_t)s->picture_number * 30 * s->avctx->time_base.num) / //FIXME use timestamp
185 s->avctx->time_base.den) & 0xff); /* TemporalReference */
186 if (s->width == 352 && s->height == 288)
188 else if (s->width == 176 && s->height == 144)
190 else if (s->width == 128 && s->height == 96)
192 else if (s->width == 320 && s->height == 240)
194 else if (s->width == 160 && s->height == 120)
196 else if (s->width <= 255 && s->height <= 255)
197 format = 0; /* use 1 byte width & height */
199 format = 1; /* use 2 bytes width & height */
200 put_bits(&s->pb, 3, format); /* PictureSize */
202 put_bits(&s->pb, 8, s->width);
203 put_bits(&s->pb, 8, s->height);
204 } else if (format == 1) {
205 put_bits(&s->pb, 16, s->width);
206 put_bits(&s->pb, 16, s->height);
208 put_bits(&s->pb, 2, s->pict_type == FF_P_TYPE); /* PictureType */
209 put_bits(&s->pb, 1, 1); /* DeblockingFlag: on */
210 put_bits(&s->pb, 5, s->qscale); /* Quantizer */
211 put_bits(&s->pb, 1, 0); /* ExtraInformation */
215 s->c_dc_scale_table= ff_aic_dc_scale_table;
218 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
222 void h263_encode_picture_header(MpegEncContext * s, int picture_number)
224 int format, coded_frame_rate, coded_frame_rate_base, i, temp_ref;
225 int best_clock_code=1;
227 int best_error= INT_MAX;
232 div= (s->avctx->time_base.num*1800000LL + 500LL*s->avctx->time_base.den) / ((1000LL+i)*s->avctx->time_base.den);
233 div= av_clip(div, 1, 127);
234 error= FFABS(s->avctx->time_base.num*1800000LL - (1000LL+i)*s->avctx->time_base.den*div);
235 if(error < best_error){
242 s->custom_pcf= best_clock_code!=1 || best_divisor!=60;
243 coded_frame_rate= 1800000;
244 coded_frame_rate_base= (1000+best_clock_code)*best_divisor;
246 align_put_bits(&s->pb);
248 /* Update the pointer to last GOB */
249 s->ptr_lastgob = pbBufPtr(&s->pb);
250 put_bits(&s->pb, 22, 0x20); /* PSC */
251 temp_ref= s->picture_number * (int64_t)coded_frame_rate * s->avctx->time_base.num / //FIXME use timestamp
252 (coded_frame_rate_base * (int64_t)s->avctx->time_base.den);
253 put_sbits(&s->pb, 8, temp_ref); /* TemporalReference */
255 put_bits(&s->pb, 1, 1); /* marker */
256 put_bits(&s->pb, 1, 0); /* h263 id */
257 put_bits(&s->pb, 1, 0); /* split screen off */
258 put_bits(&s->pb, 1, 0); /* camera off */
259 put_bits(&s->pb, 1, 0); /* freeze picture release off */
261 format = h263_get_picture_format(s->width, s->height);
264 put_bits(&s->pb, 3, format);
265 put_bits(&s->pb, 1, (s->pict_type == FF_P_TYPE));
266 /* By now UMV IS DISABLED ON H.263v1, since the restrictions
267 of H.263v1 UMV implies to check the predicted MV after
268 calculation of the current MB to see if we're on the limits */
269 put_bits(&s->pb, 1, 0); /* Unrestricted Motion Vector: off */
270 put_bits(&s->pb, 1, 0); /* SAC: off */
271 put_bits(&s->pb, 1, s->obmc); /* Advanced Prediction */
272 put_bits(&s->pb, 1, 0); /* only I/P frames, no PB frame */
273 put_bits(&s->pb, 5, s->qscale);
274 put_bits(&s->pb, 1, 0); /* Continuous Presence Multipoint mode: off */
278 /* H.263 Plus PTYPE */
280 put_bits(&s->pb, 3, 7);
281 put_bits(&s->pb,3,ufep); /* Update Full Extended PTYPE */
283 put_bits(&s->pb,3,6); /* Custom Source Format */
285 put_bits(&s->pb, 3, format);
287 put_bits(&s->pb,1, s->custom_pcf);
288 put_bits(&s->pb,1, s->umvplus); /* Unrestricted Motion Vector */
289 put_bits(&s->pb,1,0); /* SAC: off */
290 put_bits(&s->pb,1,s->obmc); /* Advanced Prediction Mode */
291 put_bits(&s->pb,1,s->h263_aic); /* Advanced Intra Coding */
292 put_bits(&s->pb,1,s->loop_filter); /* Deblocking Filter */
293 put_bits(&s->pb,1,s->h263_slice_structured); /* Slice Structured */
294 put_bits(&s->pb,1,0); /* Reference Picture Selection: off */
295 put_bits(&s->pb,1,0); /* Independent Segment Decoding: off */
296 put_bits(&s->pb,1,s->alt_inter_vlc); /* Alternative Inter VLC */
297 put_bits(&s->pb,1,s->modified_quant); /* Modified Quantization: */
298 put_bits(&s->pb,1,1); /* "1" to prevent start code emulation */
299 put_bits(&s->pb,3,0); /* Reserved */
301 put_bits(&s->pb, 3, s->pict_type == FF_P_TYPE);
303 put_bits(&s->pb,1,0); /* Reference Picture Resampling: off */
304 put_bits(&s->pb,1,0); /* Reduced-Resolution Update: off */
305 put_bits(&s->pb,1,s->no_rounding); /* Rounding Type */
306 put_bits(&s->pb,2,0); /* Reserved */
307 put_bits(&s->pb,1,1); /* "1" to prevent start code emulation */
309 /* This should be here if PLUSPTYPE */
310 put_bits(&s->pb, 1, 0); /* Continuous Presence Multipoint mode: off */
313 /* Custom Picture Format (CPFMT) */
314 aspect_to_info(s, s->avctx->sample_aspect_ratio);
316 put_bits(&s->pb,4,s->aspect_ratio_info);
317 put_bits(&s->pb,9,(s->width >> 2) - 1);
318 put_bits(&s->pb,1,1); /* "1" to prevent start code emulation */
319 put_bits(&s->pb,9,(s->height >> 2));
320 if (s->aspect_ratio_info == FF_ASPECT_EXTENDED){
321 put_bits(&s->pb, 8, s->avctx->sample_aspect_ratio.num);
322 put_bits(&s->pb, 8, s->avctx->sample_aspect_ratio.den);
327 put_bits(&s->pb, 1, best_clock_code);
328 put_bits(&s->pb, 7, best_divisor);
330 put_sbits(&s->pb, 2, temp_ref>>8);
333 /* Unlimited Unrestricted Motion Vectors Indicator (UUI) */
335 // put_bits(&s->pb,1,1); /* Limited according tables of Annex D */
336 //FIXME check actual requested range
337 put_bits(&s->pb,2,1); /* unlimited */
338 if(s->h263_slice_structured)
339 put_bits(&s->pb,2,0); /* no weird submodes */
341 put_bits(&s->pb, 5, s->qscale);
344 put_bits(&s->pb, 1, 0); /* no PEI */
346 if(s->h263_slice_structured){
347 put_bits(&s->pb, 1, 1);
349 assert(s->mb_x == 0 && s->mb_y == 0);
350 ff_h263_encode_mba(s);
352 put_bits(&s->pb, 1, 1);
357 s->c_dc_scale_table= ff_aic_dc_scale_table;
360 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
365 * Encodes a group of blocks header.
367 void h263_encode_gob_header(MpegEncContext * s, int mb_line)
369 put_bits(&s->pb, 17, 1); /* GBSC */
371 if(s->h263_slice_structured){
372 put_bits(&s->pb, 1, 1);
374 ff_h263_encode_mba(s);
377 put_bits(&s->pb, 1, 1);
378 put_bits(&s->pb, 5, s->qscale); /* GQUANT */
379 put_bits(&s->pb, 1, 1);
380 put_bits(&s->pb, 2, s->pict_type == FF_I_TYPE); /* GFID */
382 int gob_number= mb_line / s->gob_index;
384 put_bits(&s->pb, 5, gob_number); /* GN */
385 put_bits(&s->pb, 2, s->pict_type == FF_I_TYPE); /* GFID */
386 put_bits(&s->pb, 5, s->qscale); /* GQUANT */
390 static inline int get_block_rate(MpegEncContext * s, DCTELEM block[64], int block_last_index, uint8_t scantable[64]){
395 for(j=1; j<=block_last_index; j++){
396 const int index= scantable[j];
397 int level= block[index];
400 if((level&(~127)) == 0){
401 if(j<block_last_index) rate+= s->intra_ac_vlc_length [UNI_AC_ENC_INDEX(j-last-1, level)];
402 else rate+= s->intra_ac_vlc_last_length[UNI_AC_ENC_INDEX(j-last-1, level)];
404 rate += s->ac_esc_length;
414 static inline int decide_ac_pred(MpegEncContext * s, DCTELEM block[6][64], int dir[6], uint8_t *st[6], int zigzag_last_index[6])
418 int8_t * const qscale_table= s->current_picture.qscale_table;
420 memcpy(zigzag_last_index, s->block_last_index, sizeof(int)*6);
423 int16_t *ac_val, *ac_val1;
425 score -= get_block_rate(s, block[n], s->block_last_index[n], s->intra_scantable.permutated);
427 ac_val = s->ac_val[0][0] + s->block_index[n] * 16;
430 const int xy= s->mb_x + s->mb_y*s->mb_stride - s->mb_stride;
432 ac_val-= s->block_wrap[n]*16;
433 if(s->mb_y==0 || s->qscale == qscale_table[xy] || n==2 || n==3){
436 const int level= block[n][s->dsp.idct_permutation[i ]];
437 block[n][s->dsp.idct_permutation[i ]] = level - ac_val[i+8];
438 ac_val1[i ]= block[n][s->dsp.idct_permutation[i<<3]];
442 /* different qscale, we must rescale */
444 const int level= block[n][s->dsp.idct_permutation[i ]];
445 block[n][s->dsp.idct_permutation[i ]] = level - ROUNDED_DIV(ac_val[i + 8]*qscale_table[xy], s->qscale);
446 ac_val1[i ]= block[n][s->dsp.idct_permutation[i<<3]];
450 st[n]= s->intra_h_scantable.permutated;
452 const int xy= s->mb_x-1 + s->mb_y*s->mb_stride;
453 /* left prediction */
455 if(s->mb_x==0 || s->qscale == qscale_table[xy] || n==1 || n==3){
458 const int level= block[n][s->dsp.idct_permutation[i<<3]];
459 block[n][s->dsp.idct_permutation[i<<3]]= level - ac_val[i];
461 ac_val1[i+8]= block[n][s->dsp.idct_permutation[i ]];
464 /* different qscale, we must rescale */
466 const int level= block[n][s->dsp.idct_permutation[i<<3]];
467 block[n][s->dsp.idct_permutation[i<<3]]= level - ROUNDED_DIV(ac_val[i]*qscale_table[xy], s->qscale);
469 ac_val1[i+8]= block[n][s->dsp.idct_permutation[i ]];
472 st[n]= s->intra_v_scantable.permutated;
475 for(i=63; i>0; i--) //FIXME optimize
476 if(block[n][ st[n][i] ]) break;
477 s->block_last_index[n]= i;
479 score += get_block_rate(s, block[n], s->block_last_index[n], st[n]);
485 static inline void restore_ac_coeffs(MpegEncContext * s, DCTELEM block[6][64], int dir[6], uint8_t *st[6], int zigzag_last_index[6])
488 memcpy(s->block_last_index, zigzag_last_index, sizeof(int)*6);
491 int16_t *ac_val = s->ac_val[0][0] + s->block_index[n] * 16;
493 st[n]= s->intra_scantable.permutated;
497 block[n][s->dsp.idct_permutation[i ]] = ac_val[i+8];
500 /* left prediction */
502 block[n][s->dsp.idct_permutation[i<<3]]= ac_val[i ];
509 * init s->current_picture.qscale_table from s->lambda_table
511 static void ff_init_qscale_tab(MpegEncContext *s){
512 int8_t * const qscale_table= s->current_picture.qscale_table;
515 for(i=0; i<s->mb_num; i++){
516 unsigned int lam= s->lambda_table[ s->mb_index2xy[i] ];
517 int qp= (lam*139 + FF_LAMBDA_SCALE*64) >> (FF_LAMBDA_SHIFT + 7);
518 qscale_table[ s->mb_index2xy[i] ]= av_clip(qp, s->avctx->qmin, s->avctx->qmax);
523 * modify qscale so that encoding is acually possible in h263 (limit difference to -2..2)
525 void ff_clean_h263_qscales(MpegEncContext *s){
527 int8_t * const qscale_table= s->current_picture.qscale_table;
529 ff_init_qscale_tab(s);
531 for(i=1; i<s->mb_num; i++){
532 if(qscale_table[ s->mb_index2xy[i] ] - qscale_table[ s->mb_index2xy[i-1] ] >2)
533 qscale_table[ s->mb_index2xy[i] ]= qscale_table[ s->mb_index2xy[i-1] ]+2;
535 for(i=s->mb_num-2; i>=0; i--){
536 if(qscale_table[ s->mb_index2xy[i] ] - qscale_table[ s->mb_index2xy[i+1] ] >2)
537 qscale_table[ s->mb_index2xy[i] ]= qscale_table[ s->mb_index2xy[i+1] ]+2;
540 if(s->codec_id != CODEC_ID_H263P){
541 for(i=1; i<s->mb_num; i++){
542 int mb_xy= s->mb_index2xy[i];
544 if(qscale_table[mb_xy] != qscale_table[s->mb_index2xy[i-1]] && (s->mb_type[mb_xy]&CANDIDATE_MB_TYPE_INTER4V)){
545 s->mb_type[mb_xy]|= CANDIDATE_MB_TYPE_INTER;
552 * modify mb_type & qscale so that encoding is acually possible in mpeg4
554 void ff_clean_mpeg4_qscales(MpegEncContext *s){
556 int8_t * const qscale_table= s->current_picture.qscale_table;
558 ff_clean_h263_qscales(s);
560 if(s->pict_type== FF_B_TYPE){
562 /* ok, come on, this isn't funny anymore, there's more code for handling this mpeg4 mess than for the actual adaptive quantization */
564 for(i=0; i<s->mb_num; i++){
565 int mb_xy= s->mb_index2xy[i];
566 odd += qscale_table[mb_xy]&1;
569 if(2*odd > s->mb_num) odd=1;
572 for(i=0; i<s->mb_num; i++){
573 int mb_xy= s->mb_index2xy[i];
574 if((qscale_table[mb_xy]&1) != odd)
575 qscale_table[mb_xy]++;
576 if(qscale_table[mb_xy] > 31)
577 qscale_table[mb_xy]= 31;
580 for(i=1; i<s->mb_num; i++){
581 int mb_xy= s->mb_index2xy[i];
582 if(qscale_table[mb_xy] != qscale_table[s->mb_index2xy[i-1]] && (s->mb_type[mb_xy]&CANDIDATE_MB_TYPE_DIRECT)){
583 s->mb_type[mb_xy]|= CANDIDATE_MB_TYPE_BIDIR;
589 #endif //CONFIG_ENCODERS
591 #define tab_size ((signed)FF_ARRAY_ELEMS(s->direct_scale_mv[0]))
592 #define tab_bias (tab_size/2)
594 void ff_mpeg4_init_direct_mv(MpegEncContext *s){
596 for(i=0; i<tab_size; i++){
597 s->direct_scale_mv[0][i] = (i-tab_bias)*s->pb_time/s->pp_time;
598 s->direct_scale_mv[1][i] = (i-tab_bias)*(s->pb_time-s->pp_time)/s->pp_time;
602 static inline void ff_mpeg4_set_one_direct_mv(MpegEncContext *s, int mx, int my, int i){
603 int xy= s->block_index[i];
604 uint16_t time_pp= s->pp_time;
605 uint16_t time_pb= s->pb_time;
608 p_mx= s->next_picture.motion_val[0][xy][0];
609 if((unsigned)(p_mx + tab_bias) < tab_size){
610 s->mv[0][i][0] = s->direct_scale_mv[0][p_mx + tab_bias] + mx;
611 s->mv[1][i][0] = mx ? s->mv[0][i][0] - p_mx
612 : s->direct_scale_mv[1][p_mx + tab_bias];
614 s->mv[0][i][0] = p_mx*time_pb/time_pp + mx;
615 s->mv[1][i][0] = mx ? s->mv[0][i][0] - p_mx
616 : p_mx*(time_pb - time_pp)/time_pp;
618 p_my= s->next_picture.motion_val[0][xy][1];
619 if((unsigned)(p_my + tab_bias) < tab_size){
620 s->mv[0][i][1] = s->direct_scale_mv[0][p_my + tab_bias] + my;
621 s->mv[1][i][1] = my ? s->mv[0][i][1] - p_my
622 : s->direct_scale_mv[1][p_my + tab_bias];
624 s->mv[0][i][1] = p_my*time_pb/time_pp + my;
625 s->mv[1][i][1] = my ? s->mv[0][i][1] - p_my
626 : p_my*(time_pb - time_pp)/time_pp;
635 * @return the mb_type
637 int ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my){
638 const int mb_index= s->mb_x + s->mb_y*s->mb_stride;
639 const int colocated_mb_type= s->next_picture.mb_type[mb_index];
640 uint16_t time_pp= s->pp_time;
641 uint16_t time_pb= s->pb_time;
644 //FIXME avoid divides
645 // try special case with shifts for 1 and 3 B-frames?
647 if(IS_8X8(colocated_mb_type)){
648 s->mv_type = MV_TYPE_8X8;
650 ff_mpeg4_set_one_direct_mv(s, mx, my, i);
652 return MB_TYPE_DIRECT2 | MB_TYPE_8x8 | MB_TYPE_L0L1;
653 } else if(IS_INTERLACED(colocated_mb_type)){
654 s->mv_type = MV_TYPE_FIELD;
656 int field_select= s->next_picture.ref_index[0][s->block_index[2*i]];
657 s->field_select[0][i]= field_select;
658 s->field_select[1][i]= i;
659 if(s->top_field_first){
660 time_pp= s->pp_field_time - field_select + i;
661 time_pb= s->pb_field_time - field_select + i;
663 time_pp= s->pp_field_time + field_select - i;
664 time_pb= s->pb_field_time + field_select - i;
666 s->mv[0][i][0] = s->p_field_mv_table[i][0][mb_index][0]*time_pb/time_pp + mx;
667 s->mv[0][i][1] = s->p_field_mv_table[i][0][mb_index][1]*time_pb/time_pp + my;
668 s->mv[1][i][0] = mx ? s->mv[0][i][0] - s->p_field_mv_table[i][0][mb_index][0]
669 : s->p_field_mv_table[i][0][mb_index][0]*(time_pb - time_pp)/time_pp;
670 s->mv[1][i][1] = my ? s->mv[0][i][1] - s->p_field_mv_table[i][0][mb_index][1]
671 : s->p_field_mv_table[i][0][mb_index][1]*(time_pb - time_pp)/time_pp;
673 return MB_TYPE_DIRECT2 | MB_TYPE_16x8 | MB_TYPE_L0L1 | MB_TYPE_INTERLACED;
675 ff_mpeg4_set_one_direct_mv(s, mx, my, 0);
676 s->mv[0][1][0] = s->mv[0][2][0] = s->mv[0][3][0] = s->mv[0][0][0];
677 s->mv[0][1][1] = s->mv[0][2][1] = s->mv[0][3][1] = s->mv[0][0][1];
678 s->mv[1][1][0] = s->mv[1][2][0] = s->mv[1][3][0] = s->mv[1][0][0];
679 s->mv[1][1][1] = s->mv[1][2][1] = s->mv[1][3][1] = s->mv[1][0][1];
680 if((s->avctx->workaround_bugs & FF_BUG_DIRECT_BLOCKSIZE) || !s->quarter_sample)
681 s->mv_type= MV_TYPE_16X16;
683 s->mv_type= MV_TYPE_8X8;
684 return MB_TYPE_DIRECT2 | MB_TYPE_16x16 | MB_TYPE_L0L1; //Note see prev line
688 void ff_h263_update_motion_val(MpegEncContext * s){
689 const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
690 //FIXME a lot of that is only needed for !low_delay
691 const int wrap = s->b8_stride;
692 const int xy = s->block_index[0];
694 s->current_picture.mbskip_table[mb_xy]= s->mb_skipped;
696 if(s->mv_type != MV_TYPE_8X8){
697 int motion_x, motion_y;
701 } else if (s->mv_type == MV_TYPE_16X16) {
702 motion_x = s->mv[0][0][0];
703 motion_y = s->mv[0][0][1];
704 } else /*if (s->mv_type == MV_TYPE_FIELD)*/ {
706 motion_x = s->mv[0][0][0] + s->mv[0][1][0];
707 motion_y = s->mv[0][0][1] + s->mv[0][1][1];
708 motion_x = (motion_x>>1) | (motion_x&1);
710 s->p_field_mv_table[i][0][mb_xy][0]= s->mv[0][i][0];
711 s->p_field_mv_table[i][0][mb_xy][1]= s->mv[0][i][1];
713 s->current_picture.ref_index[0][xy ]=
714 s->current_picture.ref_index[0][xy + 1]= s->field_select[0][0];
715 s->current_picture.ref_index[0][xy + wrap ]=
716 s->current_picture.ref_index[0][xy + wrap + 1]= s->field_select[0][1];
719 /* no update if 8X8 because it has been done during parsing */
720 s->current_picture.motion_val[0][xy][0] = motion_x;
721 s->current_picture.motion_val[0][xy][1] = motion_y;
722 s->current_picture.motion_val[0][xy + 1][0] = motion_x;
723 s->current_picture.motion_val[0][xy + 1][1] = motion_y;
724 s->current_picture.motion_val[0][xy + wrap][0] = motion_x;
725 s->current_picture.motion_val[0][xy + wrap][1] = motion_y;
726 s->current_picture.motion_val[0][xy + 1 + wrap][0] = motion_x;
727 s->current_picture.motion_val[0][xy + 1 + wrap][1] = motion_y;
730 if(s->encoding){ //FIXME encoding MUST be cleaned up
731 if (s->mv_type == MV_TYPE_8X8)
732 s->current_picture.mb_type[mb_xy]= MB_TYPE_L0 | MB_TYPE_8x8;
734 s->current_picture.mb_type[mb_xy]= MB_TYPE_INTRA;
736 s->current_picture.mb_type[mb_xy]= MB_TYPE_L0 | MB_TYPE_16x16;
742 static inline int h263_get_motion_length(MpegEncContext * s, int val, int f_code){
743 int l, bit_size, code;
748 bit_size = f_code - 1;
749 /* modulo encoding */
750 l= INT_BIT - 6 - bit_size;
753 code = (val >> bit_size) + 1;
755 return mvtab[code][1] + 1 + bit_size;
759 static inline void ff_h263_encode_motion_vector(MpegEncContext * s, int x, int y, int f_code){
760 if(s->flags2 & CODEC_FLAG2_NO_OUTPUT){
761 skip_put_bits(&s->pb,
762 h263_get_motion_length(s, x, f_code)
763 +h263_get_motion_length(s, y, f_code));
765 ff_h263_encode_motion(s, x, f_code);
766 ff_h263_encode_motion(s, y, f_code);
770 static inline int get_p_cbp(MpegEncContext * s,
771 DCTELEM block[6][64],
772 int motion_x, int motion_y){
775 if(s->flags & CODEC_FLAG_CBP_RD){
776 int best_cbpy_score= INT_MAX;
777 int best_cbpc_score= INT_MAX;
778 int cbpc = (-1), cbpy= (-1);
779 const int offset= (s->mv_type==MV_TYPE_16X16 ? 0 : 16) + (s->dquant ? 8 : 0);
780 const int lambda= s->lambda2 >> (FF_LAMBDA_SHIFT - 6);
783 int score= inter_MCBPC_bits[i + offset] * lambda;
784 if(i&1) score += s->coded_score[5];
785 if(i&2) score += s->coded_score[4];
787 if(score < best_cbpc_score){
788 best_cbpc_score= score;
794 int score= cbpy_tab[i ^ 0xF][1] * lambda;
795 if(i&1) score += s->coded_score[3];
796 if(i&2) score += s->coded_score[2];
797 if(i&4) score += s->coded_score[1];
798 if(i&8) score += s->coded_score[0];
800 if(score < best_cbpy_score){
801 best_cbpy_score= score;
806 if ((motion_x | motion_y | s->dquant) == 0 && s->mv_type==MV_TYPE_16X16){
807 if(best_cbpy_score + best_cbpc_score + 2*lambda >= 0)
811 for (i = 0; i < 6; i++) {
812 if (s->block_last_index[i] >= 0 && ((cbp >> (5 - i))&1)==0 ){
813 s->block_last_index[i]= -1;
814 s->dsp.clear_block(s->block[i]);
819 for (i = 0; i < 6; i++) {
820 if (s->block_last_index[i] >= 0)
827 static inline int get_b_cbp(MpegEncContext * s, DCTELEM block[6][64],
828 int motion_x, int motion_y, int mb_type){
831 if(s->flags & CODEC_FLAG_CBP_RD){
833 const int lambda= s->lambda2 >> (FF_LAMBDA_SHIFT - 6);
836 if(s->coded_score[i] < 0){
837 score += s->coded_score[i];
844 if ((motion_x | motion_y | s->dquant | mb_type) == 0){
845 zero_score-= 4; //2*MV + mb_type + cbp bit
849 if(zero_score <= score){
854 for (i = 0; i < 6; i++) {
855 if (s->block_last_index[i] >= 0 && ((cbp >> (5 - i))&1)==0 ){
856 s->block_last_index[i]= -1;
857 s->dsp.clear_block(s->block[i]);
861 for (i = 0; i < 6; i++) {
862 if (s->block_last_index[i] >= 0)
869 static inline void mpeg4_encode_blocks(MpegEncContext * s, DCTELEM block[6][64], int intra_dc[6],
870 uint8_t **scan_table, PutBitContext *dc_pb, PutBitContext *ac_pb){
874 if(s->flags2 & CODEC_FLAG2_NO_OUTPUT){
875 for (i = 0; i < 6; i++) {
876 skip_put_bits(&s->pb, mpeg4_get_block_length(s, block[i], i, intra_dc[i], scan_table[i]));
879 /* encode each block */
880 for (i = 0; i < 6; i++) {
881 mpeg4_encode_block(s, block[i], i, intra_dc[i], scan_table[i], dc_pb, ac_pb);
885 if(s->flags2 & CODEC_FLAG2_NO_OUTPUT){
886 for (i = 0; i < 6; i++) {
887 skip_put_bits(&s->pb, mpeg4_get_block_length(s, block[i], i, 0, s->intra_scantable.permutated));
890 /* encode each block */
891 for (i = 0; i < 6; i++) {
892 mpeg4_encode_block(s, block[i], i, 0, s->intra_scantable.permutated, dc_pb, ac_pb);
898 static const int dquant_code[5]= {1,0,9,2,3};
900 void mpeg4_encode_mb(MpegEncContext * s,
901 DCTELEM block[6][64],
902 int motion_x, int motion_y)
904 int cbpc, cbpy, pred_x, pred_y;
905 PutBitContext * const pb2 = s->data_partitioning ? &s->pb2 : &s->pb;
906 PutBitContext * const tex_pb = s->data_partitioning && s->pict_type!=FF_B_TYPE ? &s->tex_pb : &s->pb;
907 PutBitContext * const dc_pb = s->data_partitioning && s->pict_type!=FF_I_TYPE ? &s->pb2 : &s->pb;
908 const int interleaved_stats= (s->flags&CODEC_FLAG_PASS1) && !s->data_partitioning ? 1 : 0;
910 // printf("**mb x=%d y=%d\n", s->mb_x, s->mb_y);
914 if(s->pict_type==FF_B_TYPE){
915 static const int mb_type_table[8]= {-1, 3, 2, 1,-1,-1,-1, 0}; /* convert from mv_dir to type */
916 int mb_type= mb_type_table[s->mv_dir];
923 s->last_mv[i][1][1]= 0;
927 assert(s->dquant>=-2 && s->dquant<=2);
928 assert((s->dquant&1)==0);
931 /* nothing to do if this MB was skipped in the next P Frame */
932 if(s->next_picture.mbskip_table[s->mb_y * s->mb_stride + s->mb_x]){ //FIXME avoid DCT & ...
938 s->mv_dir= MV_DIR_FORWARD; //doesn't matter
939 s->qscale -= s->dquant;
945 cbp= get_b_cbp(s, block, motion_x, motion_y, mb_type);
947 if ((cbp | motion_x | motion_y | mb_type) ==0) {
948 /* direct MB with MV={0,0} */
949 assert(s->dquant==0);
951 put_bits(&s->pb, 1, 1); /* mb not coded modb1=1 */
953 if(interleaved_stats){
961 put_bits(&s->pb, 1, 0); /* mb coded modb1=0 */
962 put_bits(&s->pb, 1, cbp ? 0 : 1); /* modb2 */ //FIXME merge
963 put_bits(&s->pb, mb_type+1, 1); // this table is so simple that we don't need it :)
964 if(cbp) put_bits(&s->pb, 6, cbp);
968 put_bits(&s->pb, 2, (s->dquant>>2)+3);
970 put_bits(&s->pb, 1, 0);
972 s->qscale -= s->dquant;
974 if(!s->progressive_sequence){
976 put_bits(&s->pb, 1, s->interlaced_dct);
977 if(mb_type) // not direct mode
978 put_bits(&s->pb, 1, s->mv_type == MV_TYPE_FIELD);
981 if(interleaved_stats){
982 s->misc_bits+= get_bits_diff(s);
986 assert(s->mv_dir & MV_DIRECT);
987 ff_h263_encode_motion_vector(s, motion_x, motion_y, 1);
991 assert(mb_type > 0 && mb_type < 4);
992 if(s->mv_type != MV_TYPE_FIELD){
993 if(s->mv_dir & MV_DIR_FORWARD){
994 ff_h263_encode_motion_vector(s, s->mv[0][0][0] - s->last_mv[0][0][0],
995 s->mv[0][0][1] - s->last_mv[0][0][1], s->f_code);
996 s->last_mv[0][0][0]= s->last_mv[0][1][0]= s->mv[0][0][0];
997 s->last_mv[0][0][1]= s->last_mv[0][1][1]= s->mv[0][0][1];
1000 if(s->mv_dir & MV_DIR_BACKWARD){
1001 ff_h263_encode_motion_vector(s, s->mv[1][0][0] - s->last_mv[1][0][0],
1002 s->mv[1][0][1] - s->last_mv[1][0][1], s->b_code);
1003 s->last_mv[1][0][0]= s->last_mv[1][1][0]= s->mv[1][0][0];
1004 s->last_mv[1][0][1]= s->last_mv[1][1][1]= s->mv[1][0][1];
1008 if(s->mv_dir & MV_DIR_FORWARD){
1009 put_bits(&s->pb, 1, s->field_select[0][0]);
1010 put_bits(&s->pb, 1, s->field_select[0][1]);
1012 if(s->mv_dir & MV_DIR_BACKWARD){
1013 put_bits(&s->pb, 1, s->field_select[1][0]);
1014 put_bits(&s->pb, 1, s->field_select[1][1]);
1016 if(s->mv_dir & MV_DIR_FORWARD){
1018 ff_h263_encode_motion_vector(s, s->mv[0][i][0] - s->last_mv[0][i][0] ,
1019 s->mv[0][i][1] - s->last_mv[0][i][1]/2, s->f_code);
1020 s->last_mv[0][i][0]= s->mv[0][i][0];
1021 s->last_mv[0][i][1]= s->mv[0][i][1]*2;
1025 if(s->mv_dir & MV_DIR_BACKWARD){
1027 ff_h263_encode_motion_vector(s, s->mv[1][i][0] - s->last_mv[1][i][0] ,
1028 s->mv[1][i][1] - s->last_mv[1][i][1]/2, s->b_code);
1029 s->last_mv[1][i][0]= s->mv[1][i][0];
1030 s->last_mv[1][i][1]= s->mv[1][i][1]*2;
1037 if(interleaved_stats){
1038 s->mv_bits+= get_bits_diff(s);
1041 mpeg4_encode_blocks(s, block, NULL, NULL, NULL, &s->pb);
1043 if(interleaved_stats){
1044 s->p_tex_bits+= get_bits_diff(s);
1047 }else{ /* s->pict_type==FF_B_TYPE */
1048 cbp= get_p_cbp(s, block, motion_x, motion_y);
1050 if ((cbp | motion_x | motion_y | s->dquant) == 0 && s->mv_type==MV_TYPE_16X16) {
1051 /* check if the B frames can skip it too, as we must skip it if we skip here
1052 why didn't they just compress the skip-mb bits instead of reusing them ?! */
1053 if(s->max_b_frames>0){
1060 if(x+16 > s->width) x= s->width-16;
1061 if(y+16 > s->height) y= s->height-16;
1063 offset= x + y*s->linesize;
1064 p_pic= s->new_picture.data[0] + offset;
1067 for(i=0; i<s->max_b_frames; i++){
1070 Picture *pic= s->reordered_input_picture[i+1];
1072 if(pic==NULL || pic->pict_type!=FF_B_TYPE) break;
1074 b_pic= pic->data[0] + offset;
1075 if(pic->type != FF_BUFFER_TYPE_SHARED)
1076 b_pic+= INPLACE_OFFSET;
1077 diff= s->dsp.sad[0](NULL, p_pic, b_pic, s->linesize, 16);
1078 if(diff>s->qscale*70){ //FIXME check that 70 is optimal
1086 if(s->mb_skipped==1){
1087 /* skip macroblock */
1088 put_bits(&s->pb, 1, 1);
1090 if(interleaved_stats){
1100 put_bits(&s->pb, 1, 0); /* mb coded */
1104 if(s->mv_type==MV_TYPE_16X16){
1105 if(s->dquant) cbpc+= 8;
1107 inter_MCBPC_bits[cbpc],
1108 inter_MCBPC_code[cbpc]);
1110 put_bits(pb2, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1112 put_bits(pb2, 2, dquant_code[s->dquant+2]);
1114 if(!s->progressive_sequence){
1116 put_bits(pb2, 1, s->interlaced_dct);
1117 put_bits(pb2, 1, 0);
1120 if(interleaved_stats){
1121 s->misc_bits+= get_bits_diff(s);
1124 /* motion vectors: 16x16 mode */
1125 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
1127 ff_h263_encode_motion_vector(s, motion_x - pred_x,
1128 motion_y - pred_y, s->f_code);
1129 }else if(s->mv_type==MV_TYPE_FIELD){
1130 if(s->dquant) cbpc+= 8;
1132 inter_MCBPC_bits[cbpc],
1133 inter_MCBPC_code[cbpc]);
1135 put_bits(pb2, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1137 put_bits(pb2, 2, dquant_code[s->dquant+2]);
1139 assert(!s->progressive_sequence);
1141 put_bits(pb2, 1, s->interlaced_dct);
1142 put_bits(pb2, 1, 1);
1144 if(interleaved_stats){
1145 s->misc_bits+= get_bits_diff(s);
1148 /* motion vectors: 16x8 interlaced mode */
1149 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
1152 put_bits(&s->pb, 1, s->field_select[0][0]);
1153 put_bits(&s->pb, 1, s->field_select[0][1]);
1155 ff_h263_encode_motion_vector(s, s->mv[0][0][0] - pred_x,
1156 s->mv[0][0][1] - pred_y, s->f_code);
1157 ff_h263_encode_motion_vector(s, s->mv[0][1][0] - pred_x,
1158 s->mv[0][1][1] - pred_y, s->f_code);
1160 assert(s->mv_type==MV_TYPE_8X8);
1162 inter_MCBPC_bits[cbpc+16],
1163 inter_MCBPC_code[cbpc+16]);
1164 put_bits(pb2, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1166 if(!s->progressive_sequence){
1168 put_bits(pb2, 1, s->interlaced_dct);
1171 if(interleaved_stats){
1172 s->misc_bits+= get_bits_diff(s);
1176 /* motion vectors: 8x8 mode*/
1177 h263_pred_motion(s, i, 0, &pred_x, &pred_y);
1179 ff_h263_encode_motion_vector(s, s->current_picture.motion_val[0][ s->block_index[i] ][0] - pred_x,
1180 s->current_picture.motion_val[0][ s->block_index[i] ][1] - pred_y, s->f_code);
1184 if(interleaved_stats){
1185 s->mv_bits+= get_bits_diff(s);
1188 mpeg4_encode_blocks(s, block, NULL, NULL, NULL, tex_pb);
1190 if(interleaved_stats){
1191 s->p_tex_bits+= get_bits_diff(s);
1197 int dc_diff[6]; //dc values with the dc prediction subtracted
1198 int dir[6]; //prediction direction
1199 int zigzag_last_index[6];
1200 uint8_t *scan_table[6];
1204 dc_diff[i]= ff_mpeg4_pred_dc(s, i, block[i][0], &dir[i], 1);
1207 if(s->flags & CODEC_FLAG_AC_PRED){
1208 s->ac_pred= decide_ac_pred(s, block, dir, scan_table, zigzag_last_index);
1210 restore_ac_coeffs(s, block, dir, scan_table, zigzag_last_index);
1213 scan_table[i]= s->intra_scantable.permutated;
1218 for (i = 0; i < 6; i++) {
1219 if (s->block_last_index[i] >= 1)
1220 cbp |= 1 << (5 - i);
1224 if (s->pict_type == FF_I_TYPE) {
1225 if(s->dquant) cbpc+=4;
1227 intra_MCBPC_bits[cbpc],
1228 intra_MCBPC_code[cbpc]);
1230 if(s->dquant) cbpc+=8;
1231 put_bits(&s->pb, 1, 0); /* mb coded */
1233 inter_MCBPC_bits[cbpc + 4],
1234 inter_MCBPC_code[cbpc + 4]);
1236 put_bits(pb2, 1, s->ac_pred);
1238 put_bits(pb2, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1240 put_bits(dc_pb, 2, dquant_code[s->dquant+2]);
1242 if(!s->progressive_sequence){
1243 put_bits(dc_pb, 1, s->interlaced_dct);
1246 if(interleaved_stats){
1247 s->misc_bits+= get_bits_diff(s);
1250 mpeg4_encode_blocks(s, block, dc_diff, scan_table, dc_pb, tex_pb);
1252 if(interleaved_stats){
1253 s->i_tex_bits+= get_bits_diff(s);
1257 /* restore ac coeffs & last_index stuff if we messed them up with the prediction */
1259 restore_ac_coeffs(s, block, dir, scan_table, zigzag_last_index);
1263 void h263_encode_mb(MpegEncContext * s,
1264 DCTELEM block[6][64],
1265 int motion_x, int motion_y)
1267 int cbpc, cbpy, i, cbp, pred_x, pred_y;
1269 int16_t rec_intradc[6];
1271 const int interleaved_stats= (s->flags&CODEC_FLAG_PASS1);
1273 //printf("**mb x=%d y=%d\n", s->mb_x, s->mb_y);
1276 cbp= get_p_cbp(s, block, motion_x, motion_y);
1278 if ((cbp | motion_x | motion_y | s->dquant | (s->mv_type - MV_TYPE_16X16)) == 0) {
1279 /* skip macroblock */
1280 put_bits(&s->pb, 1, 1);
1281 if(interleaved_stats){
1289 put_bits(&s->pb, 1, 0); /* mb coded */
1293 if(s->alt_inter_vlc==0 || cbpc!=3)
1295 if(s->dquant) cbpc+= 8;
1296 if(s->mv_type==MV_TYPE_16X16){
1298 inter_MCBPC_bits[cbpc],
1299 inter_MCBPC_code[cbpc]);
1301 put_bits(&s->pb, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1303 put_bits(&s->pb, 2, dquant_code[s->dquant+2]);
1305 if(interleaved_stats){
1306 s->misc_bits+= get_bits_diff(s);
1309 /* motion vectors: 16x16 mode */
1310 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
1313 ff_h263_encode_motion_vector(s, motion_x - pred_x,
1314 motion_y - pred_y, 1);
1317 h263p_encode_umotion(s, motion_x - pred_x);
1318 h263p_encode_umotion(s, motion_y - pred_y);
1319 if (((motion_x - pred_x) == 1) && ((motion_y - pred_y) == 1))
1320 /* To prevent Start Code emulation */
1321 put_bits(&s->pb,1,1);
1325 inter_MCBPC_bits[cbpc+16],
1326 inter_MCBPC_code[cbpc+16]);
1327 put_bits(&s->pb, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1329 put_bits(&s->pb, 2, dquant_code[s->dquant+2]);
1331 if(interleaved_stats){
1332 s->misc_bits+= get_bits_diff(s);
1336 /* motion vectors: 8x8 mode*/
1337 h263_pred_motion(s, i, 0, &pred_x, &pred_y);
1339 motion_x= s->current_picture.motion_val[0][ s->block_index[i] ][0];
1340 motion_y= s->current_picture.motion_val[0][ s->block_index[i] ][1];
1342 ff_h263_encode_motion_vector(s, motion_x - pred_x,
1343 motion_y - pred_y, 1);
1346 h263p_encode_umotion(s, motion_x - pred_x);
1347 h263p_encode_umotion(s, motion_y - pred_y);
1348 if (((motion_x - pred_x) == 1) && ((motion_y - pred_y) == 1))
1349 /* To prevent Start Code emulation */
1350 put_bits(&s->pb,1,1);
1355 if(interleaved_stats){
1356 s->mv_bits+= get_bits_diff(s);
1359 assert(s->mb_intra);
1364 for(i=0; i<6; i++) {
1365 int16_t level = block[i][0];
1368 if(i<4) scale= s->y_dc_scale;
1369 else scale= s->c_dc_scale;
1371 pred_dc = h263_pred_dc(s, i, &dc_ptr[i]);
1375 level = (level + (scale>>1))/scale;
1377 level = (level - (scale>>1))/scale;
1379 /* AIC can change CBP */
1380 if (level == 0 && s->block_last_index[i] == 0)
1381 s->block_last_index[i] = -1;
1383 if(!s->modified_quant){
1386 else if (level > 127)
1390 block[i][0] = level;
1391 /* Reconstruction */
1392 rec_intradc[i] = scale*level + pred_dc;
1394 rec_intradc[i] |= 1;
1395 //if ((rec_intradc[i] % 2) == 0)
1396 // rec_intradc[i]++;
1398 if (rec_intradc[i] < 0)
1400 else if (rec_intradc[i] > 2047)
1401 rec_intradc[i] = 2047;
1403 /* Update AC/DC tables */
1404 *dc_ptr[i] = rec_intradc[i];
1405 if (s->block_last_index[i] >= 0)
1406 cbp |= 1 << (5 - i);
1409 for(i=0; i<6; i++) {
1411 if (s->block_last_index[i] >= 1)
1412 cbp |= 1 << (5 - i);
1417 if (s->pict_type == FF_I_TYPE) {
1418 if(s->dquant) cbpc+=4;
1420 intra_MCBPC_bits[cbpc],
1421 intra_MCBPC_code[cbpc]);
1423 if(s->dquant) cbpc+=8;
1424 put_bits(&s->pb, 1, 0); /* mb coded */
1426 inter_MCBPC_bits[cbpc + 4],
1427 inter_MCBPC_code[cbpc + 4]);
1430 /* XXX: currently, we do not try to use ac prediction */
1431 put_bits(&s->pb, 1, 0); /* no AC prediction */
1434 put_bits(&s->pb, cbpy_tab[cbpy][1], cbpy_tab[cbpy][0]);
1436 put_bits(&s->pb, 2, dquant_code[s->dquant+2]);
1438 if(interleaved_stats){
1439 s->misc_bits+= get_bits_diff(s);
1443 for(i=0; i<6; i++) {
1444 /* encode each block */
1445 h263_encode_block(s, block[i], i);
1447 /* Update INTRADC for decoding */
1448 if (s->h263_aic && s->mb_intra) {
1449 block[i][0] = rec_intradc[i];
1454 if(interleaved_stats){
1456 s->p_tex_bits+= get_bits_diff(s);
1459 s->i_tex_bits+= get_bits_diff(s);
1466 void ff_h263_loop_filter(MpegEncContext * s){
1468 const int linesize = s->linesize;
1469 const int uvlinesize= s->uvlinesize;
1470 const int xy = s->mb_y * s->mb_stride + s->mb_x;
1471 uint8_t *dest_y = s->dest[0];
1472 uint8_t *dest_cb= s->dest[1];
1473 uint8_t *dest_cr= s->dest[2];
1475 // if(s->pict_type==FF_B_TYPE && !s->readable) return;
1481 if(!IS_SKIP(s->current_picture.mb_type[xy])){
1483 s->dsp.h263_v_loop_filter(dest_y+8*linesize , linesize, qp_c);
1484 s->dsp.h263_v_loop_filter(dest_y+8*linesize+8, linesize, qp_c);
1489 int qp_dt, qp_tt, qp_tc;
1491 if(IS_SKIP(s->current_picture.mb_type[xy-s->mb_stride]))
1494 qp_tt= s->current_picture.qscale_table[xy-s->mb_stride];
1502 const int chroma_qp= s->chroma_qscale_table[qp_tc];
1503 s->dsp.h263_v_loop_filter(dest_y , linesize, qp_tc);
1504 s->dsp.h263_v_loop_filter(dest_y+8, linesize, qp_tc);
1506 s->dsp.h263_v_loop_filter(dest_cb , uvlinesize, chroma_qp);
1507 s->dsp.h263_v_loop_filter(dest_cr , uvlinesize, chroma_qp);
1511 s->dsp.h263_h_loop_filter(dest_y-8*linesize+8 , linesize, qp_tt);
1514 if(qp_tt || IS_SKIP(s->current_picture.mb_type[xy-1-s->mb_stride]))
1517 qp_dt= s->current_picture.qscale_table[xy-1-s->mb_stride];
1520 const int chroma_qp= s->chroma_qscale_table[qp_dt];
1521 s->dsp.h263_h_loop_filter(dest_y -8*linesize , linesize, qp_dt);
1522 s->dsp.h263_h_loop_filter(dest_cb-8*uvlinesize, uvlinesize, chroma_qp);
1523 s->dsp.h263_h_loop_filter(dest_cr-8*uvlinesize, uvlinesize, chroma_qp);
1529 s->dsp.h263_h_loop_filter(dest_y +8, linesize, qp_c);
1530 if(s->mb_y + 1 == s->mb_height)
1531 s->dsp.h263_h_loop_filter(dest_y+8*linesize+8, linesize, qp_c);
1536 if(qp_c || IS_SKIP(s->current_picture.mb_type[xy-1]))
1539 qp_lc= s->current_picture.qscale_table[xy-1];
1542 s->dsp.h263_h_loop_filter(dest_y, linesize, qp_lc);
1543 if(s->mb_y + 1 == s->mb_height){
1544 const int chroma_qp= s->chroma_qscale_table[qp_lc];
1545 s->dsp.h263_h_loop_filter(dest_y +8* linesize, linesize, qp_lc);
1546 s->dsp.h263_h_loop_filter(dest_cb , uvlinesize, chroma_qp);
1547 s->dsp.h263_h_loop_filter(dest_cr , uvlinesize, chroma_qp);
1554 static int h263_pred_dc(MpegEncContext * s, int n, int16_t **dc_val_ptr)
1556 int x, y, wrap, a, c, pred_dc, scale;
1559 /* find prediction */
1561 x = 2 * s->mb_x + (n & 1);
1562 y = 2 * s->mb_y + ((n & 2) >> 1);
1563 wrap = s->b8_stride;
1564 dc_val = s->dc_val[0];
1565 scale = s->y_dc_scale;
1569 wrap = s->mb_stride;
1570 dc_val = s->dc_val[n - 4 + 1];
1571 scale = s->c_dc_scale;
1576 a = dc_val[(x - 1) + (y) * wrap];
1577 c = dc_val[(x) + (y - 1) * wrap];
1579 /* No prediction outside GOB boundary */
1580 if(s->first_slice_line && n!=3){
1582 if(n!=1 && s->mb_x == s->resync_mb_x) a= 1024;
1585 /* just DC prediction */
1586 if (a != 1024 && c != 1024)
1587 pred_dc = (a + c) >> 1;
1593 /* we assume pred is positive */
1594 //pred_dc = (pred_dc + (scale >> 1)) / scale;
1595 *dc_val_ptr = &dc_val[x + y * wrap];
1598 #endif /* CONFIG_ENCODERS */
1600 static void h263_pred_acdc(MpegEncContext * s, DCTELEM *block, int n)
1602 int x, y, wrap, a, c, pred_dc, scale, i;
1603 int16_t *dc_val, *ac_val, *ac_val1;
1605 /* find prediction */
1607 x = 2 * s->mb_x + (n & 1);
1608 y = 2 * s->mb_y + (n>> 1);
1609 wrap = s->b8_stride;
1610 dc_val = s->dc_val[0];
1611 ac_val = s->ac_val[0][0];
1612 scale = s->y_dc_scale;
1616 wrap = s->mb_stride;
1617 dc_val = s->dc_val[n - 4 + 1];
1618 ac_val = s->ac_val[n - 4 + 1][0];
1619 scale = s->c_dc_scale;
1622 ac_val += ((y) * wrap + (x)) * 16;
1628 a = dc_val[(x - 1) + (y) * wrap];
1629 c = dc_val[(x) + (y - 1) * wrap];
1631 /* No prediction outside GOB boundary */
1632 if(s->first_slice_line && n!=3){
1634 if(n!=1 && s->mb_x == s->resync_mb_x) a= 1024;
1639 if (s->h263_aic_dir) {
1640 /* left prediction */
1644 block[s->dsp.idct_permutation[i<<3]] += ac_val[i];
1649 /* top prediction */
1651 ac_val -= 16 * wrap;
1653 block[s->dsp.idct_permutation[i ]] += ac_val[i + 8];
1659 /* just DC prediction */
1660 if (a != 1024 && c != 1024)
1661 pred_dc = (a + c) >> 1;
1668 /* we assume pred is positive */
1669 block[0]=block[0]*scale + pred_dc;
1676 /* Update AC/DC tables */
1677 dc_val[(x) + (y) * wrap] = block[0];
1681 ac_val1[i ] = block[s->dsp.idct_permutation[i<<3]];
1684 ac_val1[8 + i] = block[s->dsp.idct_permutation[i ]];
1687 int16_t *h263_pred_motion(MpegEncContext * s, int block, int dir,
1691 int16_t *A, *B, *C, (*mot_val)[2];
1692 static const int off[4]= {2, 1, 1, -1};
1694 wrap = s->b8_stride;
1695 mot_val = s->current_picture.motion_val[dir] + s->block_index[block];
1698 /* special case for first (slice) line */
1699 if (s->first_slice_line && block<3) {
1700 // we can't just change some MVs to simulate that as we need them for the B frames (and ME)
1701 // and if we ever support non rectangular objects than we need to do a few ifs here anyway :(
1702 if(block==0){ //most common case
1703 if(s->mb_x == s->resync_mb_x){ //rare
1705 }else if(s->mb_x + 1 == s->resync_mb_x && s->h263_pred){ //rare
1706 C = mot_val[off[block] - wrap];
1711 *px = mid_pred(A[0], 0, C[0]);
1712 *py = mid_pred(A[1], 0, C[1]);
1719 if(s->mb_x + 1 == s->resync_mb_x && s->h263_pred){ //rare
1720 C = mot_val[off[block] - wrap];
1721 *px = mid_pred(A[0], 0, C[0]);
1722 *py = mid_pred(A[1], 0, C[1]);
1727 }else{ /* block==2*/
1728 B = mot_val[ - wrap];
1729 C = mot_val[off[block] - wrap];
1730 if(s->mb_x == s->resync_mb_x) //rare
1733 *px = mid_pred(A[0], B[0], C[0]);
1734 *py = mid_pred(A[1], B[1], C[1]);
1737 B = mot_val[ - wrap];
1738 C = mot_val[off[block] - wrap];
1739 *px = mid_pred(A[0], B[0], C[0]);
1740 *py = mid_pred(A[1], B[1], C[1]);
1746 void ff_h263_encode_motion(MpegEncContext * s, int val, int f_code)
1748 int range, l, bit_size, sign, code, bits;
1753 put_bits(&s->pb, mvtab[code][1], mvtab[code][0]);
1755 bit_size = f_code - 1;
1756 range = 1 << bit_size;
1757 /* modulo encoding */
1758 l= INT_BIT - 6 - bit_size;
1761 val= (val^sign)-sign;
1765 code = (val >> bit_size) + 1;
1766 bits = val & (range - 1);
1768 put_bits(&s->pb, mvtab[code][1] + 1, (mvtab[code][0] << 1) | sign);
1770 put_bits(&s->pb, bit_size, bits);
1775 /* Encode MV differences on H.263+ with Unrestricted MV mode */
1776 static void h263p_encode_umotion(MpegEncContext * s, int val)
1786 put_bits(&s->pb, 1, 1);
1788 put_bits(&s->pb, 3, 0);
1790 put_bits(&s->pb, 3, 2);
1793 sval = ((val < 0) ? (short)(-val):(short)val);
1796 while (temp_val != 0) {
1797 temp_val = temp_val >> 1;
1803 tcode = (sval & (1 << (i-1))) >> (i-1);
1804 tcode = (tcode << 1) | 1;
1805 code = (code << 2) | tcode;
1808 code = ((code << 1) | (val < 0)) << 1;
1809 put_bits(&s->pb, (2*n_bits)+1, code);
1810 //printf("\nVal = %d\tCode = %d", sval, code);
1814 static void init_mv_penalty_and_fcode(MpegEncContext *s)
1819 for(f_code=1; f_code<=MAX_FCODE; f_code++){
1820 for(mv=-MAX_MV; mv<=MAX_MV; mv++){
1823 if(mv==0) len= mvtab[0][1];
1825 int val, bit_size, range, code;
1827 bit_size = f_code - 1;
1828 range = 1 << bit_size;
1834 code = (val >> bit_size) + 1;
1836 len= mvtab[code][1] + 1 + bit_size;
1838 len= mvtab[32][1] + av_log2(code>>5) + 2 + bit_size;
1842 mv_penalty[f_code][mv+MAX_MV]= len;
1846 for(f_code=MAX_FCODE; f_code>0; f_code--){
1847 for(mv=-(16<<f_code); mv<(16<<f_code); mv++){
1848 fcode_tab[mv+MAX_MV]= f_code;
1852 for(mv=0; mv<MAX_MV*2+1; mv++){
1853 umv_fcode_tab[mv]= 1;
1857 static void init_uni_dc_tab(void)
1859 int level, uni_code, uni_len;
1861 for(level=-256; level<256; level++){
1863 /* find number of bits */
1872 l= (-level) ^ ((1 << size) - 1);
1877 uni_code= DCtab_lum[size][0];
1878 uni_len = DCtab_lum[size][1];
1881 uni_code<<=size; uni_code|=l;
1884 uni_code<<=1; uni_code|=1;
1888 uni_DCtab_lum_bits[level+256]= uni_code;
1889 uni_DCtab_lum_len [level+256]= uni_len;
1892 uni_code= DCtab_chrom[size][0];
1893 uni_len = DCtab_chrom[size][1];
1896 uni_code<<=size; uni_code|=l;
1899 uni_code<<=1; uni_code|=1;
1903 uni_DCtab_chrom_bits[level+256]= uni_code;
1904 uni_DCtab_chrom_len [level+256]= uni_len;
1909 static void init_uni_mpeg4_rl_tab(RLTable *rl, uint32_t *bits_tab, uint8_t *len_tab){
1910 int slevel, run, last;
1912 assert(MAX_LEVEL >= 64);
1913 assert(MAX_RUN >= 63);
1915 for(slevel=-64; slevel<64; slevel++){
1916 if(slevel==0) continue;
1917 for(run=0; run<64; run++){
1918 for(last=0; last<=1; last++){
1919 const int index= UNI_MPEG4_ENC_INDEX(last, run, slevel+64);
1920 int level= slevel < 0 ? -slevel : slevel;
1921 int sign= slevel < 0 ? 1 : 0;
1922 int bits, len, code;
1925 len_tab[index]= 100;
1928 code= get_rl_index(rl, last, run, level);
1929 bits= rl->table_vlc[code][0];
1930 len= rl->table_vlc[code][1];
1931 bits=bits*2+sign; len++;
1933 if(code!=rl->n && len < len_tab[index]){
1934 bits_tab[index]= bits;
1935 len_tab [index]= len;
1939 bits= rl->table_vlc[rl->n][0];
1940 len= rl->table_vlc[rl->n][1];
1941 bits=bits*2; len++; //esc1
1942 level1= level - rl->max_level[last][run];
1944 code= get_rl_index(rl, last, run, level1);
1945 bits<<= rl->table_vlc[code][1];
1946 len += rl->table_vlc[code][1];
1947 bits += rl->table_vlc[code][0];
1948 bits=bits*2+sign; len++;
1950 if(code!=rl->n && len < len_tab[index]){
1951 bits_tab[index]= bits;
1952 len_tab [index]= len;
1958 bits= rl->table_vlc[rl->n][0];
1959 len= rl->table_vlc[rl->n][1];
1960 bits=bits*4+2; len+=2; //esc2
1961 run1 = run - rl->max_run[last][level] - 1;
1963 code= get_rl_index(rl, last, run1, level);
1964 bits<<= rl->table_vlc[code][1];
1965 len += rl->table_vlc[code][1];
1966 bits += rl->table_vlc[code][0];
1967 bits=bits*2+sign; len++;
1969 if(code!=rl->n && len < len_tab[index]){
1970 bits_tab[index]= bits;
1971 len_tab [index]= len;
1976 bits= rl->table_vlc[rl->n][0];
1977 len = rl->table_vlc[rl->n][1];
1978 bits=bits*4+3; len+=2; //esc3
1979 bits=bits*2+last; len++;
1980 bits=bits*64+run; len+=6;
1981 bits=bits*2+1; len++; //marker
1982 bits=bits*4096+(slevel&0xfff); len+=12;
1983 bits=bits*2+1; len++; //marker
1985 if(len < len_tab[index]){
1986 bits_tab[index]= bits;
1987 len_tab [index]= len;
1994 static void init_uni_h263_rl_tab(RLTable *rl, uint32_t *bits_tab, uint8_t *len_tab){
1995 int slevel, run, last;
1997 assert(MAX_LEVEL >= 64);
1998 assert(MAX_RUN >= 63);
2000 for(slevel=-64; slevel<64; slevel++){
2001 if(slevel==0) continue;
2002 for(run=0; run<64; run++){
2003 for(last=0; last<=1; last++){
2004 const int index= UNI_MPEG4_ENC_INDEX(last, run, slevel+64);
2005 int level= slevel < 0 ? -slevel : slevel;
2006 int sign= slevel < 0 ? 1 : 0;
2007 int bits, len, code;
2009 len_tab[index]= 100;
2012 code= get_rl_index(rl, last, run, level);
2013 bits= rl->table_vlc[code][0];
2014 len= rl->table_vlc[code][1];
2015 bits=bits*2+sign; len++;
2017 if(code!=rl->n && len < len_tab[index]){
2018 if(bits_tab) bits_tab[index]= bits;
2019 len_tab [index]= len;
2022 bits= rl->table_vlc[rl->n][0];
2023 len = rl->table_vlc[rl->n][1];
2024 bits=bits*2+last; len++;
2025 bits=bits*64+run; len+=6;
2026 bits=bits*256+(level&0xff); len+=8;
2028 if(len < len_tab[index]){
2029 if(bits_tab) bits_tab[index]= bits;
2030 len_tab [index]= len;
2037 void h263_encode_init(MpegEncContext *s)
2039 static int done = 0;
2046 init_rl(&rl_inter, static_rl_table_store[0]);
2047 init_rl(&rl_intra, static_rl_table_store[1]);
2048 init_rl(&rl_intra_aic, static_rl_table_store[2]);
2050 init_uni_mpeg4_rl_tab(&rl_intra, uni_mpeg4_intra_rl_bits, uni_mpeg4_intra_rl_len);
2051 init_uni_mpeg4_rl_tab(&rl_inter, uni_mpeg4_inter_rl_bits, uni_mpeg4_inter_rl_len);
2053 init_uni_h263_rl_tab(&rl_intra_aic, NULL, uni_h263_intra_aic_rl_len);
2054 init_uni_h263_rl_tab(&rl_inter , NULL, uni_h263_inter_rl_len);
2056 init_mv_penalty_and_fcode(s);
2058 s->me.mv_penalty= mv_penalty; //FIXME exact table for msmpeg4 & h263p
2060 s->intra_ac_vlc_length =s->inter_ac_vlc_length = uni_h263_inter_rl_len;
2061 s->intra_ac_vlc_last_length=s->inter_ac_vlc_last_length= uni_h263_inter_rl_len + 128*64;
2063 s->intra_ac_vlc_length = uni_h263_intra_aic_rl_len;
2064 s->intra_ac_vlc_last_length= uni_h263_intra_aic_rl_len + 128*64;
2066 s->ac_esc_length= 7+1+6+8;
2068 // use fcodes >1 only for mpeg4 & h263 & h263p FIXME
2069 switch(s->codec_id){
2070 case CODEC_ID_MPEG4:
2071 s->fcode_tab= fcode_tab;
2072 s->min_qcoeff= -2048;
2073 s->max_qcoeff= 2047;
2074 s->intra_ac_vlc_length = uni_mpeg4_intra_rl_len;
2075 s->intra_ac_vlc_last_length= uni_mpeg4_intra_rl_len + 128*64;
2076 s->inter_ac_vlc_length = uni_mpeg4_inter_rl_len;
2077 s->inter_ac_vlc_last_length= uni_mpeg4_inter_rl_len + 128*64;
2078 s->luma_dc_vlc_length= uni_DCtab_lum_len;
2079 s->chroma_dc_vlc_length= uni_DCtab_chrom_len;
2080 s->ac_esc_length= 7+2+1+6+1+12+1;
2081 s->y_dc_scale_table= ff_mpeg4_y_dc_scale_table;
2082 s->c_dc_scale_table= ff_mpeg4_c_dc_scale_table;
2084 if(s->flags & CODEC_FLAG_GLOBAL_HEADER){
2086 s->avctx->extradata= av_malloc(1024);
2087 init_put_bits(&s->pb, s->avctx->extradata, 1024);
2089 if(!(s->workaround_bugs & FF_BUG_MS))
2090 mpeg4_encode_visual_object_header(s);
2091 mpeg4_encode_vol_header(s, 0, 0);
2093 // ff_mpeg4_stuffing(&s->pb); ?
2094 flush_put_bits(&s->pb);
2095 s->avctx->extradata_size= (put_bits_count(&s->pb)+7)>>3;
2099 case CODEC_ID_H263P:
2101 s->fcode_tab= umv_fcode_tab;
2102 if(s->modified_quant){
2103 s->min_qcoeff= -2047;
2104 s->max_qcoeff= 2047;
2106 s->min_qcoeff= -127;
2110 //Note for mpeg4 & h263 the dc-scale table will be set per frame as needed later
2112 if (s->h263_flv > 1) {
2113 s->min_qcoeff= -1023;
2114 s->max_qcoeff= 1023;
2116 s->min_qcoeff= -127;
2119 s->y_dc_scale_table=
2120 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
2122 default: //nothing needed - default table already set in mpegvideo.c
2123 s->min_qcoeff= -127;
2125 s->y_dc_scale_table=
2126 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
2131 * encodes a 8x8 block.
2132 * @param block the 8x8 block
2133 * @param n block index (0-3 are luma, 4-5 are chroma)
2135 static void h263_encode_block(MpegEncContext * s, DCTELEM * block, int n)
2137 int level, run, last, i, j, last_index, last_non_zero, sign, slevel, code;
2141 if (s->mb_intra && !s->h263_aic) {
2144 /* 255 cannot be represented, so we clamp */
2149 /* 0 cannot be represented also */
2150 else if (level < 1) {
2154 if (level == 128) //FIXME check rv10
2155 put_bits(&s->pb, 8, 0xff);
2157 put_bits(&s->pb, 8, level);
2161 if (s->h263_aic && s->mb_intra)
2164 if(s->alt_inter_vlc && !s->mb_intra){
2166 int inter_vlc_bits=0;
2170 last_index = s->block_last_index[n];
2171 last_non_zero = i - 1;
2172 for (; i <= last_index; i++) {
2173 j = s->intra_scantable.permutated[i];
2176 run = i - last_non_zero - 1;
2177 last = (i == last_index);
2179 if(level<0) level= -level;
2181 code = get_rl_index(rl, last, run, level);
2182 aic_code = get_rl_index(&rl_intra_aic, last, run, level);
2183 inter_vlc_bits += rl->table_vlc[code][1]+1;
2184 aic_vlc_bits += rl_intra_aic.table_vlc[aic_code][1]+1;
2186 if (code == rl->n) {
2187 inter_vlc_bits += 1+6+8-1;
2189 if (aic_code == rl_intra_aic.n) {
2190 aic_vlc_bits += 1+6+8-1;
2191 wrong_pos += run + 1;
2193 wrong_pos += wrong_run[aic_code];
2198 if(aic_vlc_bits < inter_vlc_bits && wrong_pos > 63)
2204 last_index = s->block_last_index[n];
2205 last_non_zero = i - 1;
2206 for (; i <= last_index; i++) {
2207 j = s->intra_scantable.permutated[i];
2210 run = i - last_non_zero - 1;
2211 last = (i == last_index);
2218 code = get_rl_index(rl, last, run, level);
2219 put_bits(&s->pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
2220 if (code == rl->n) {
2221 if(s->h263_flv <= 1){
2222 put_bits(&s->pb, 1, last);
2223 put_bits(&s->pb, 6, run);
2225 assert(slevel != 0);
2228 put_sbits(&s->pb, 8, slevel);
2230 put_bits(&s->pb, 8, 128);
2231 put_sbits(&s->pb, 5, slevel);
2232 put_sbits(&s->pb, 6, slevel>>5);
2235 if(level < 64) { // 7-bit level
2236 put_bits(&s->pb, 1, 0);
2237 put_bits(&s->pb, 1, last);
2238 put_bits(&s->pb, 6, run);
2240 put_sbits(&s->pb, 7, slevel);
2243 put_bits(&s->pb, 1, 1);
2244 put_bits(&s->pb, 1, last);
2245 put_bits(&s->pb, 6, run);
2247 put_sbits(&s->pb, 11, slevel);
2251 put_bits(&s->pb, 1, sign);
2258 /***************************************************/
2260 * add mpeg4 stuffing bits (01...1)
2262 void ff_mpeg4_stuffing(PutBitContext * pbc)
2265 put_bits(pbc, 1, 0);
2266 length= (-put_bits_count(pbc))&7;
2267 if(length) put_bits(pbc, length, (1<<length)-1);
2270 /* must be called before writing the header */
2271 void ff_set_mpeg4_time(MpegEncContext * s){
2272 if(s->pict_type==FF_B_TYPE){
2273 ff_mpeg4_init_direct_mv(s);
2275 s->last_time_base= s->time_base;
2276 s->time_base= s->time/s->avctx->time_base.den;
2280 static void mpeg4_encode_gop_header(MpegEncContext * s){
2281 int hours, minutes, seconds;
2284 put_bits(&s->pb, 16, 0);
2285 put_bits(&s->pb, 16, GOP_STARTCODE);
2287 time= s->current_picture_ptr->pts;
2288 if(s->reordered_input_picture[1])
2289 time= FFMIN(time, s->reordered_input_picture[1]->pts);
2290 time= time*s->avctx->time_base.num;
2292 seconds= time/s->avctx->time_base.den;
2293 minutes= seconds/60; seconds %= 60;
2294 hours= minutes/60; minutes %= 60;
2297 put_bits(&s->pb, 5, hours);
2298 put_bits(&s->pb, 6, minutes);
2299 put_bits(&s->pb, 1, 1);
2300 put_bits(&s->pb, 6, seconds);
2302 put_bits(&s->pb, 1, !!(s->flags&CODEC_FLAG_CLOSED_GOP));
2303 put_bits(&s->pb, 1, 0); //broken link == NO
2305 s->last_time_base= time / s->avctx->time_base.den;
2307 ff_mpeg4_stuffing(&s->pb);
2310 static void mpeg4_encode_visual_object_header(MpegEncContext * s){
2311 int profile_and_level_indication;
2314 if(s->avctx->profile != FF_PROFILE_UNKNOWN){
2315 profile_and_level_indication = s->avctx->profile << 4;
2316 }else if(s->max_b_frames || s->quarter_sample){
2317 profile_and_level_indication= 0xF0; // adv simple
2319 profile_and_level_indication= 0x00; // simple
2322 if(s->avctx->level != FF_LEVEL_UNKNOWN){
2323 profile_and_level_indication |= s->avctx->level;
2325 profile_and_level_indication |= 1; //level 1
2328 if(profile_and_level_indication>>4 == 0xF){
2336 put_bits(&s->pb, 16, 0);
2337 put_bits(&s->pb, 16, VOS_STARTCODE);
2339 put_bits(&s->pb, 8, profile_and_level_indication);
2341 put_bits(&s->pb, 16, 0);
2342 put_bits(&s->pb, 16, VISUAL_OBJ_STARTCODE);
2344 put_bits(&s->pb, 1, 1);
2345 put_bits(&s->pb, 4, vo_ver_id);
2346 put_bits(&s->pb, 3, 1); //priority
2348 put_bits(&s->pb, 4, 1); //visual obj type== video obj
2350 put_bits(&s->pb, 1, 0); //video signal type == no clue //FIXME
2352 ff_mpeg4_stuffing(&s->pb);
2355 static void mpeg4_encode_vol_header(MpegEncContext * s, int vo_number, int vol_number)
2359 if (!CONFIG_MPEG4_ENCODER) return;
2361 if(s->max_b_frames || s->quarter_sample){
2363 s->vo_type= ADV_SIMPLE_VO_TYPE;
2366 s->vo_type= SIMPLE_VO_TYPE;
2369 put_bits(&s->pb, 16, 0);
2370 put_bits(&s->pb, 16, 0x100 + vo_number); /* video obj */
2371 put_bits(&s->pb, 16, 0);
2372 put_bits(&s->pb, 16, 0x120 + vol_number); /* video obj layer */
2374 put_bits(&s->pb, 1, 0); /* random access vol */
2375 put_bits(&s->pb, 8, s->vo_type); /* video obj type indication */
2376 if(s->workaround_bugs & FF_BUG_MS) {
2377 put_bits(&s->pb, 1, 0); /* is obj layer id= no */
2379 put_bits(&s->pb, 1, 1); /* is obj layer id= yes */
2380 put_bits(&s->pb, 4, vo_ver_id); /* is obj layer ver id */
2381 put_bits(&s->pb, 3, 1); /* is obj layer priority */
2384 aspect_to_info(s, s->avctx->sample_aspect_ratio);
2386 put_bits(&s->pb, 4, s->aspect_ratio_info);/* aspect ratio info */
2387 if (s->aspect_ratio_info == FF_ASPECT_EXTENDED){
2388 put_bits(&s->pb, 8, s->avctx->sample_aspect_ratio.num);
2389 put_bits(&s->pb, 8, s->avctx->sample_aspect_ratio.den);
2392 if(s->workaround_bugs & FF_BUG_MS) { //
2393 put_bits(&s->pb, 1, 0); /* vol control parameters= no @@@ */
2395 put_bits(&s->pb, 1, 1); /* vol control parameters= yes */
2396 put_bits(&s->pb, 2, 1); /* chroma format YUV 420/YV12 */
2397 put_bits(&s->pb, 1, s->low_delay);
2398 put_bits(&s->pb, 1, 0); /* vbv parameters= no */
2401 put_bits(&s->pb, 2, RECT_SHAPE); /* vol shape= rectangle */
2402 put_bits(&s->pb, 1, 1); /* marker bit */
2404 put_bits(&s->pb, 16, s->avctx->time_base.den);
2405 if (s->time_increment_bits < 1)
2406 s->time_increment_bits = 1;
2407 put_bits(&s->pb, 1, 1); /* marker bit */
2408 put_bits(&s->pb, 1, 0); /* fixed vop rate=no */
2409 put_bits(&s->pb, 1, 1); /* marker bit */
2410 put_bits(&s->pb, 13, s->width); /* vol width */
2411 put_bits(&s->pb, 1, 1); /* marker bit */
2412 put_bits(&s->pb, 13, s->height); /* vol height */
2413 put_bits(&s->pb, 1, 1); /* marker bit */
2414 put_bits(&s->pb, 1, s->progressive_sequence ? 0 : 1);
2415 put_bits(&s->pb, 1, 1); /* obmc disable */
2416 if (vo_ver_id == 1) {
2417 put_bits(&s->pb, 1, s->vol_sprite_usage); /* sprite enable */
2419 put_bits(&s->pb, 2, s->vol_sprite_usage); /* sprite enable */
2422 put_bits(&s->pb, 1, 0); /* not 8 bit == false */
2423 put_bits(&s->pb, 1, s->mpeg_quant); /* quant type= (0=h263 style)*/
2426 ff_write_quant_matrix(&s->pb, s->avctx->intra_matrix);
2427 ff_write_quant_matrix(&s->pb, s->avctx->inter_matrix);
2431 put_bits(&s->pb, 1, s->quarter_sample);
2432 put_bits(&s->pb, 1, 1); /* complexity estimation disable */
2433 s->resync_marker= s->rtp_mode;
2434 put_bits(&s->pb, 1, s->resync_marker ? 0 : 1);/* resync marker disable */
2435 put_bits(&s->pb, 1, s->data_partitioning ? 1 : 0);
2436 if(s->data_partitioning){
2437 put_bits(&s->pb, 1, 0); /* no rvlc */
2440 if (vo_ver_id != 1){
2441 put_bits(&s->pb, 1, 0); /* newpred */
2442 put_bits(&s->pb, 1, 0); /* reduced res vop */
2444 put_bits(&s->pb, 1, 0); /* scalability */
2446 ff_mpeg4_stuffing(&s->pb);
2449 if(!(s->flags & CODEC_FLAG_BITEXACT)){
2450 put_bits(&s->pb, 16, 0);
2451 put_bits(&s->pb, 16, 0x1B2); /* user_data */
2452 ff_put_string(&s->pb, LIBAVCODEC_IDENT, 0);
2456 /* write mpeg4 VOP header */
2457 void mpeg4_encode_picture_header(MpegEncContext * s, int picture_number)
2460 int time_div, time_mod;
2462 if(s->pict_type==FF_I_TYPE){
2463 if(!(s->flags&CODEC_FLAG_GLOBAL_HEADER)){
2464 if(s->strict_std_compliance < FF_COMPLIANCE_VERY_STRICT) //HACK, the reference sw is buggy
2465 mpeg4_encode_visual_object_header(s);
2466 if(s->strict_std_compliance < FF_COMPLIANCE_VERY_STRICT || picture_number==0) //HACK, the reference sw is buggy
2467 mpeg4_encode_vol_header(s, 0, 0);
2469 if(!(s->workaround_bugs & FF_BUG_MS))
2470 mpeg4_encode_gop_header(s);
2473 s->partitioned_frame= s->data_partitioning && s->pict_type!=FF_B_TYPE;
2475 //printf("num:%d rate:%d base:%d\n", s->picture_number, s->time_base.den, FRAME_RATE_BASE);
2477 put_bits(&s->pb, 16, 0); /* vop header */
2478 put_bits(&s->pb, 16, VOP_STARTCODE); /* vop header */
2479 put_bits(&s->pb, 2, s->pict_type - 1); /* pict type: I = 0 , P = 1 */
2482 time_div= s->time/s->avctx->time_base.den;
2483 time_mod= s->time%s->avctx->time_base.den;
2484 time_incr= time_div - s->last_time_base;
2485 assert(time_incr >= 0);
2487 put_bits(&s->pb, 1, 1);
2489 put_bits(&s->pb, 1, 0);
2491 put_bits(&s->pb, 1, 1); /* marker */
2492 put_bits(&s->pb, s->time_increment_bits, time_mod); /* time increment */
2493 put_bits(&s->pb, 1, 1); /* marker */
2494 put_bits(&s->pb, 1, 1); /* vop coded */
2495 if ( s->pict_type == FF_P_TYPE
2496 || (s->pict_type == FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE)) {
2497 put_bits(&s->pb, 1, s->no_rounding); /* rounding type */
2499 put_bits(&s->pb, 3, 0); /* intra dc VLC threshold */
2500 if(!s->progressive_sequence){
2501 put_bits(&s->pb, 1, s->current_picture_ptr->top_field_first);
2502 put_bits(&s->pb, 1, s->alternate_scan);
2504 //FIXME sprite stuff
2506 put_bits(&s->pb, 5, s->qscale);
2508 if (s->pict_type != FF_I_TYPE)
2509 put_bits(&s->pb, 3, s->f_code); /* fcode_for */
2510 if (s->pict_type == FF_B_TYPE)
2511 put_bits(&s->pb, 3, s->b_code); /* fcode_back */
2512 // printf("****frame %d\n", picture_number);
2515 #endif //CONFIG_ENCODERS
2519 * encoding quantized level -> quantized diff
2520 * decoding quantized diff -> quantized level
2521 * @param n block index (0-3 are luma, 4-5 are chroma)
2522 * @param dir_ptr pointer to an integer where the prediction direction will be stored
2524 static inline int ff_mpeg4_pred_dc(MpegEncContext * s, int n, int level, int *dir_ptr, int encoding)
2526 int a, b, c, wrap, pred, scale, ret;
2529 /* find prediction */
2531 scale = s->y_dc_scale;
2533 scale = s->c_dc_scale;
2538 wrap= s->block_wrap[n];
2539 dc_val = s->dc_val[0] + s->block_index[n];
2545 b = dc_val[ - 1 - wrap];
2546 c = dc_val[ - wrap];
2548 /* outside slice handling (we can't do that by memset as we need the dc for error resilience) */
2549 if(s->first_slice_line && n!=3){
2551 if(n!=1 && s->mb_x == s->resync_mb_x) b=a= 1024;
2553 if(s->mb_x == s->resync_mb_x && s->mb_y == s->resync_mb_y+1){
2554 if(n==0 || n==4 || n==5)
2558 if (abs(a - b) < abs(b - c)) {
2560 *dir_ptr = 1; /* top */
2563 *dir_ptr = 0; /* left */
2565 /* we assume pred is positive */
2566 pred = FASTDIV((pred + (scale >> 1)), scale);
2573 if(s->error_recognition>=3){
2575 av_log(s->avctx, AV_LOG_ERROR, "dc<0 at %dx%d\n", s->mb_x, s->mb_y);
2578 if(level*scale > 2048 + scale){
2579 av_log(s->avctx, AV_LOG_ERROR, "dc overflow at %dx%d\n", s->mb_x, s->mb_y);
2588 else if(!(s->workaround_bugs&FF_BUG_DC_CLIP))
2598 * @param n block index (0-3 are luma, 4-5 are chroma)
2599 * @param dir the ac prediction direction
2601 void mpeg4_pred_ac(MpegEncContext * s, DCTELEM *block, int n,
2605 int16_t *ac_val, *ac_val1;
2606 int8_t * const qscale_table= s->current_picture.qscale_table;
2608 /* find prediction */
2609 ac_val = s->ac_val[0][0] + s->block_index[n] * 16;
2613 const int xy= s->mb_x-1 + s->mb_y*s->mb_stride;
2614 /* left prediction */
2617 if(s->mb_x==0 || s->qscale == qscale_table[xy] || n==1 || n==3){
2620 block[s->dsp.idct_permutation[i<<3]] += ac_val[i];
2623 /* different qscale, we must rescale */
2625 block[s->dsp.idct_permutation[i<<3]] += ROUNDED_DIV(ac_val[i]*qscale_table[xy], s->qscale);
2629 const int xy= s->mb_x + s->mb_y*s->mb_stride - s->mb_stride;
2630 /* top prediction */
2631 ac_val -= 16 * s->block_wrap[n];
2633 if(s->mb_y==0 || s->qscale == qscale_table[xy] || n==2 || n==3){
2636 block[s->dsp.idct_permutation[i]] += ac_val[i + 8];
2639 /* different qscale, we must rescale */
2641 block[s->dsp.idct_permutation[i]] += ROUNDED_DIV(ac_val[i + 8]*qscale_table[xy], s->qscale);
2648 ac_val1[i ] = block[s->dsp.idct_permutation[i<<3]];
2652 ac_val1[8 + i] = block[s->dsp.idct_permutation[i ]];
2659 * encodes the dc value.
2660 * @param n block index (0-3 are luma, 4-5 are chroma)
2662 static inline void mpeg4_encode_dc(PutBitContext * s, int level, int n)
2665 // if(level<-255 || level>255) printf("dc overflow\n");
2669 put_bits(s, uni_DCtab_lum_len[level], uni_DCtab_lum_bits[level]);
2672 put_bits(s, uni_DCtab_chrom_len[level], uni_DCtab_chrom_bits[level]);
2676 /* find number of bits */
2686 put_bits(&s->pb, DCtab_lum[size][1], DCtab_lum[size][0]);
2689 put_bits(&s->pb, DCtab_chrom[size][1], DCtab_chrom[size][0]);
2692 /* encode remaining bits */
2695 level = (-level) ^ ((1 << size) - 1);
2696 put_bits(&s->pb, size, level);
2698 put_bits(&s->pb, 1, 1);
2703 static inline int mpeg4_get_dc_length(int level, int n){
2705 return uni_DCtab_lum_len[level + 256];
2707 return uni_DCtab_chrom_len[level + 256];
2712 * encodes a 8x8 block
2713 * @param n block index (0-3 are luma, 4-5 are chroma)
2715 static inline void mpeg4_encode_block(MpegEncContext * s, DCTELEM * block, int n, int intra_dc,
2716 uint8_t *scan_table, PutBitContext *dc_pb, PutBitContext *ac_pb)
2718 int i, last_non_zero;
2719 #if 0 //variables for the outcommented version
2720 int code, sign, last;
2725 const int last_index = s->block_last_index[n];
2727 if (s->mb_intra) { //Note gcc (3.2.1 at least) will optimize this away
2728 /* mpeg4 based DC predictor */
2729 mpeg4_encode_dc(dc_pb, intra_dc, n);
2730 if(last_index<1) return;
2733 bits_tab= uni_mpeg4_intra_rl_bits;
2734 len_tab = uni_mpeg4_intra_rl_len;
2736 if(last_index<0) return;
2739 bits_tab= uni_mpeg4_inter_rl_bits;
2740 len_tab = uni_mpeg4_inter_rl_len;
2744 last_non_zero = i - 1;
2746 for (; i < last_index; i++) {
2747 int level = block[ scan_table[i] ];
2749 int run = i - last_non_zero - 1;
2751 if((level&(~127)) == 0){
2752 const int index= UNI_MPEG4_ENC_INDEX(0, run, level);
2753 put_bits(ac_pb, len_tab[index], bits_tab[index]);
2755 put_bits(ac_pb, 7+2+1+6+1+12+1, (3<<23)+(3<<21)+(0<<20)+(run<<14)+(1<<13)+(((level-64)&0xfff)<<1)+1);
2760 /*if(i<=last_index)*/{
2761 int level = block[ scan_table[i] ];
2762 int run = i - last_non_zero - 1;
2764 if((level&(~127)) == 0){
2765 const int index= UNI_MPEG4_ENC_INDEX(1, run, level);
2766 put_bits(ac_pb, len_tab[index], bits_tab[index]);
2768 put_bits(ac_pb, 7+2+1+6+1+12+1, (3<<23)+(3<<21)+(1<<20)+(run<<14)+(1<<13)+(((level-64)&0xfff)<<1)+1);
2772 for (; i <= last_index; i++) {
2773 const int slevel = block[ scan_table[i] ];
2776 int run = i - last_non_zero - 1;
2777 last = (i == last_index);
2784 code = get_rl_index(rl, last, run, level);
2785 put_bits(ac_pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
2786 if (code == rl->n) {
2788 level1 = level - rl->max_level[last][run];
2791 code = get_rl_index(rl, last, run, level1);
2792 if (code == rl->n) {
2794 put_bits(ac_pb, 1, 1);
2795 if (level > MAX_LEVEL)
2797 run1 = run - rl->max_run[last][level] - 1;
2800 code = get_rl_index(rl, last, run1, level);
2801 if (code == rl->n) {
2804 put_bits(ac_pb, 1, 1);
2805 put_bits(ac_pb, 1, last);
2806 put_bits(ac_pb, 6, run);
2807 put_bits(ac_pb, 1, 1);
2808 put_sbits(ac_pb, 12, slevel);
2809 put_bits(ac_pb, 1, 1);
2812 put_bits(ac_pb, 1, 0);
2813 put_bits(ac_pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
2814 put_bits(ac_pb, 1, sign);
2818 put_bits(ac_pb, 1, 0);
2819 put_bits(ac_pb, rl->table_vlc[code][1], rl->table_vlc[code][0]);
2820 put_bits(ac_pb, 1, sign);
2823 put_bits(ac_pb, 1, sign);
2831 static int mpeg4_get_block_length(MpegEncContext * s, DCTELEM * block, int n, int intra_dc,
2832 uint8_t *scan_table)
2834 int i, last_non_zero;
2837 const int last_index = s->block_last_index[n];
2840 if (s->mb_intra) { //Note gcc (3.2.1 at least) will optimize this away
2841 /* mpeg4 based DC predictor */
2842 len += mpeg4_get_dc_length(intra_dc, n);
2843 if(last_index<1) return len;
2846 len_tab = uni_mpeg4_intra_rl_len;
2848 if(last_index<0) return 0;
2851 len_tab = uni_mpeg4_inter_rl_len;
2855 last_non_zero = i - 1;
2856 for (; i < last_index; i++) {
2857 int level = block[ scan_table[i] ];
2859 int run = i - last_non_zero - 1;
2861 if((level&(~127)) == 0){
2862 const int index= UNI_MPEG4_ENC_INDEX(0, run, level);
2863 len += len_tab[index];
2865 len += 7+2+1+6+1+12+1;
2870 /*if(i<=last_index)*/{
2871 int level = block[ scan_table[i] ];
2872 int run = i - last_non_zero - 1;
2874 if((level&(~127)) == 0){
2875 const int index= UNI_MPEG4_ENC_INDEX(1, run, level);
2876 len += len_tab[index];
2878 len += 7+2+1+6+1+12+1;
2888 /***********************************************/
2891 static VLC intra_MCBPC_vlc;
2892 static VLC inter_MCBPC_vlc;
2893 static VLC cbpy_vlc;
2895 static VLC dc_lum, dc_chrom;
2896 static VLC sprite_trajectory;
2897 static VLC mb_type_b_vlc;
2898 static VLC h263_mbtype_b_vlc;
2899 static VLC cbpc_b_vlc;
2903 /* XXX: find a better solution to handle static init */
2904 void h263_decode_init_vlc(MpegEncContext *s)
2906 static int done = 0;
2911 INIT_VLC_STATIC(&intra_MCBPC_vlc, INTRA_MCBPC_VLC_BITS, 9,
2912 intra_MCBPC_bits, 1, 1,
2913 intra_MCBPC_code, 1, 1, 72);
2914 INIT_VLC_STATIC(&inter_MCBPC_vlc, INTER_MCBPC_VLC_BITS, 28,
2915 inter_MCBPC_bits, 1, 1,
2916 inter_MCBPC_code, 1, 1, 198);
2917 INIT_VLC_STATIC(&cbpy_vlc, CBPY_VLC_BITS, 16,
2918 &cbpy_tab[0][1], 2, 1,
2919 &cbpy_tab[0][0], 2, 1, 64);
2920 INIT_VLC_STATIC(&mv_vlc, MV_VLC_BITS, 33,
2922 &mvtab[0][0], 2, 1, 538);
2923 init_rl(&rl_inter, static_rl_table_store[0]);
2924 init_rl(&rl_intra, static_rl_table_store[1]);
2925 init_rl(&rvlc_rl_inter, static_rl_table_store[3]);
2926 init_rl(&rvlc_rl_intra, static_rl_table_store[4]);
2927 init_rl(&rl_intra_aic, static_rl_table_store[2]);
2928 INIT_VLC_RL(rl_inter, 554);
2929 INIT_VLC_RL(rl_intra, 554);
2930 INIT_VLC_RL(rvlc_rl_inter, 1072);
2931 INIT_VLC_RL(rvlc_rl_intra, 1072);
2932 INIT_VLC_RL(rl_intra_aic, 554);
2933 INIT_VLC_STATIC(&dc_lum, DC_VLC_BITS, 10 /* 13 */,
2934 &DCtab_lum[0][1], 2, 1,
2935 &DCtab_lum[0][0], 2, 1, 512);
2936 INIT_VLC_STATIC(&dc_chrom, DC_VLC_BITS, 10 /* 13 */,
2937 &DCtab_chrom[0][1], 2, 1,
2938 &DCtab_chrom[0][0], 2, 1, 512);
2939 INIT_VLC_STATIC(&sprite_trajectory, SPRITE_TRAJ_VLC_BITS, 15,
2940 &sprite_trajectory_tab[0][1], 4, 2,
2941 &sprite_trajectory_tab[0][0], 4, 2, 128);
2942 INIT_VLC_STATIC(&mb_type_b_vlc, MB_TYPE_B_VLC_BITS, 4,
2943 &mb_type_b_tab[0][1], 2, 1,
2944 &mb_type_b_tab[0][0], 2, 1, 16);
2945 INIT_VLC_STATIC(&h263_mbtype_b_vlc, H263_MBTYPE_B_VLC_BITS, 15,
2946 &h263_mbtype_b_tab[0][1], 2, 1,
2947 &h263_mbtype_b_tab[0][0], 2, 1, 80);
2948 INIT_VLC_STATIC(&cbpc_b_vlc, CBPC_B_VLC_BITS, 4,
2949 &cbpc_b_tab[0][1], 2, 1,
2950 &cbpc_b_tab[0][0], 2, 1, 8);
2955 * Get the GOB height based on picture height.
2957 int ff_h263_get_gob_height(MpegEncContext *s){
2958 if (s->height <= 400)
2960 else if (s->height <= 800)
2966 int ff_h263_decode_mba(MpegEncContext *s)
2971 if(s->mb_num-1 <= ff_mba_max[i]) break;
2973 mb_pos= get_bits(&s->gb, ff_mba_length[i]);
2974 s->mb_x= mb_pos % s->mb_width;
2975 s->mb_y= mb_pos / s->mb_width;
2980 void ff_h263_encode_mba(MpegEncContext *s)
2985 if(s->mb_num-1 <= ff_mba_max[i]) break;
2987 mb_pos= s->mb_x + s->mb_width*s->mb_y;
2988 put_bits(&s->pb, ff_mba_length[i], mb_pos);
2992 * decodes the group of blocks header or slice header.
2993 * @return <0 if an error occurred
2995 static int h263_decode_gob_header(MpegEncContext *s)
2997 unsigned int val, gfid, gob_number;
3000 /* Check for GOB Start Code */
3001 val = show_bits(&s->gb, 16);
3005 /* We have a GBSC probably with GSTUFF */
3006 skip_bits(&s->gb, 16); /* Drop the zeros */
3007 left= s->gb.size_in_bits - get_bits_count(&s->gb);
3008 //MN: we must check the bits left or we might end in a infinite loop (or segfault)
3009 for(;left>13; left--){
3010 if(get_bits1(&s->gb)) break; /* Seek the '1' bit */
3015 if(s->h263_slice_structured){
3016 if(get_bits1(&s->gb)==0)
3019 ff_h263_decode_mba(s);
3021 if(s->mb_num > 1583)
3022 if(get_bits1(&s->gb)==0)
3025 s->qscale = get_bits(&s->gb, 5); /* SQUANT */
3026 if(get_bits1(&s->gb)==0)
3028 gfid = get_bits(&s->gb, 2); /* GFID */
3030 gob_number = get_bits(&s->gb, 5); /* GN */
3032 s->mb_y= s->gob_index* gob_number;
3033 gfid = get_bits(&s->gb, 2); /* GFID */
3034 s->qscale = get_bits(&s->gb, 5); /* GQUANT */
3037 if(s->mb_y >= s->mb_height)
3046 static inline void memsetw(short *tab, int val, int n)
3055 void ff_mpeg4_init_partitions(MpegEncContext *s)
3057 uint8_t *start= pbBufPtr(&s->pb);
3058 uint8_t *end= s->pb.buf_end;
3059 int size= end - start;
3060 int pb_size = (((long)start + size/3)&(~3)) - (long)start;
3061 int tex_size= (size - 2*pb_size)&(~3);
3063 set_put_bits_buffer_size(&s->pb, pb_size);
3064 init_put_bits(&s->tex_pb, start + pb_size , tex_size);
3065 init_put_bits(&s->pb2 , start + pb_size + tex_size, pb_size);
3068 void ff_mpeg4_merge_partitions(MpegEncContext *s)
3070 const int pb2_len = put_bits_count(&s->pb2 );
3071 const int tex_pb_len= put_bits_count(&s->tex_pb);
3072 const int bits= put_bits_count(&s->pb);
3074 if(s->pict_type==FF_I_TYPE){
3075 put_bits(&s->pb, 19, DC_MARKER);
3076 s->misc_bits+=19 + pb2_len + bits - s->last_bits;
3077 s->i_tex_bits+= tex_pb_len;
3079 put_bits(&s->pb, 17, MOTION_MARKER);
3080 s->misc_bits+=17 + pb2_len;
3081 s->mv_bits+= bits - s->last_bits;
3082 s->p_tex_bits+= tex_pb_len;
3085 flush_put_bits(&s->pb2);
3086 flush_put_bits(&s->tex_pb);
3088 set_put_bits_buffer_size(&s->pb, s->pb2.buf_end - s->pb.buf);
3089 ff_copy_bits(&s->pb, s->pb2.buf , pb2_len);
3090 ff_copy_bits(&s->pb, s->tex_pb.buf, tex_pb_len);
3091 s->last_bits= put_bits_count(&s->pb);
3094 #endif //CONFIG_ENCODERS
3096 int ff_mpeg4_get_video_packet_prefix_length(MpegEncContext *s){
3097 switch(s->pict_type){
3102 return s->f_code+15;
3104 return FFMAX3(s->f_code, s->b_code, 2) + 15;
3112 void ff_mpeg4_encode_video_packet_header(MpegEncContext *s)
3114 int mb_num_bits= av_log2(s->mb_num - 1) + 1;
3116 put_bits(&s->pb, ff_mpeg4_get_video_packet_prefix_length(s), 0);
3117 put_bits(&s->pb, 1, 1);
3119 put_bits(&s->pb, mb_num_bits, s->mb_x + s->mb_y*s->mb_width);
3120 put_bits(&s->pb, s->quant_precision, s->qscale);
3121 put_bits(&s->pb, 1, 0); /* no HEC */
3124 #endif //CONFIG_ENCODERS
3127 * check if the next stuff is a resync marker or the end.
3130 static inline int mpeg4_is_resync(MpegEncContext *s){
3131 int bits_count= get_bits_count(&s->gb);
3132 int v= show_bits(&s->gb, 16);
3134 if(s->workaround_bugs&FF_BUG_NO_PADDING){
3139 if(s->pict_type==FF_B_TYPE || (v>>(8-s->pict_type)!=1) || s->partitioned_frame)
3141 skip_bits(&s->gb, 8+s->pict_type);
3142 bits_count+= 8+s->pict_type;
3143 v= show_bits(&s->gb, 16);
3146 if(bits_count + 8 >= s->gb.size_in_bits){
3148 v|= 0x7F >> (7-(bits_count&7));
3153 if(v == ff_mpeg4_resync_prefix[bits_count&7]){
3155 GetBitContext gb= s->gb;
3157 skip_bits(&s->gb, 1);
3158 align_get_bits(&s->gb);
3160 for(len=0; len<32; len++){
3161 if(get_bits1(&s->gb)) break;
3166 if(len>=ff_mpeg4_get_video_packet_prefix_length(s))
3174 * decodes the next video packet.
3175 * @return <0 if something went wrong
3177 static int mpeg4_decode_video_packet_header(MpegEncContext *s)
3179 int mb_num_bits= av_log2(s->mb_num - 1) + 1;
3180 int header_extension=0, mb_num, len;
3182 /* is there enough space left for a video packet + header */
3183 if( get_bits_count(&s->gb) > s->gb.size_in_bits-20) return -1;
3185 for(len=0; len<32; len++){
3186 if(get_bits1(&s->gb)) break;
3189 if(len!=ff_mpeg4_get_video_packet_prefix_length(s)){
3190 av_log(s->avctx, AV_LOG_ERROR, "marker does not match f_code\n");
3194 if(s->shape != RECT_SHAPE){
3195 header_extension= get_bits1(&s->gb);
3196 //FIXME more stuff here
3199 mb_num= get_bits(&s->gb, mb_num_bits);
3200 if(mb_num>=s->mb_num){
3201 av_log(s->avctx, AV_LOG_ERROR, "illegal mb_num in video packet (%d %d) \n", mb_num, s->mb_num);
3204 if(s->pict_type == FF_B_TYPE){
3205 while(s->next_picture.mbskip_table[ s->mb_index2xy[ mb_num ] ]) mb_num++;
3206 if(mb_num >= s->mb_num) return -1; // slice contains just skipped MBs which where already decoded
3209 s->mb_x= mb_num % s->mb_width;
3210 s->mb_y= mb_num / s->mb_width;
3212 if(s->shape != BIN_ONLY_SHAPE){
3213 int qscale= get_bits(&s->gb, s->quant_precision);
3215 s->chroma_qscale=s->qscale= qscale;
3218 if(s->shape == RECT_SHAPE){
3219 header_extension= get_bits1(&s->gb);
3221 if(header_extension){
3225 while (get_bits1(&s->gb) != 0)
3228 check_marker(&s->gb, "before time_increment in video packed header");
3229 time_increment= get_bits(&s->gb, s->time_increment_bits);
3230 check_marker(&s->gb, "before vop_coding_type in video packed header");
3232 skip_bits(&s->gb, 2); /* vop coding type */
3233 //FIXME not rect stuff here
3235 if(s->shape != BIN_ONLY_SHAPE){
3236 skip_bits(&s->gb, 3); /* intra dc vlc threshold */
3237 //FIXME don't just ignore everything
3238 if(s->pict_type == FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE){
3239 mpeg4_decode_sprite_trajectory(s, &s->gb);
3240 av_log(s->avctx, AV_LOG_ERROR, "untested\n");
3243 //FIXME reduced res stuff here
3245 if (s->pict_type != FF_I_TYPE) {
3246 int f_code = get_bits(&s->gb, 3); /* fcode_for */
3248 av_log(s->avctx, AV_LOG_ERROR, "Error, video packet header damaged (f_code=0)\n");
3251 if (s->pict_type == FF_B_TYPE) {
3252 int b_code = get_bits(&s->gb, 3);
3254 av_log(s->avctx, AV_LOG_ERROR, "Error, video packet header damaged (b_code=0)\n");
3259 //FIXME new-pred stuff
3261 //printf("parse ok %d %d %d %d\n", mb_num, s->mb_x + s->mb_y*s->mb_width, get_bits_count(gb), get_bits_count(&s->gb));
3266 void ff_mpeg4_clean_buffers(MpegEncContext *s)
3268 int c_wrap, c_xy, l_wrap, l_xy;
3270 l_wrap= s->b8_stride;
3271 l_xy= (2*s->mb_y-1)*l_wrap + s->mb_x*2 - 1;
3272 c_wrap= s->mb_stride;
3273 c_xy= (s->mb_y-1)*c_wrap + s->mb_x - 1;
3277 memsetw(s->dc_val[0] + l_xy, 1024, l_wrap*2+1);
3278 memsetw(s->dc_val[1] + c_xy, 1024, c_wrap+1);
3279 memsetw(s->dc_val[2] + c_xy, 1024, c_wrap+1);
3283 memset(s->ac_val[0] + l_xy, 0, (l_wrap*2+1)*16*sizeof(int16_t));
3284 memset(s->ac_val[1] + c_xy, 0, (c_wrap +1)*16*sizeof(int16_t));
3285 memset(s->ac_val[2] + c_xy, 0, (c_wrap +1)*16*sizeof(int16_t));
3288 // we can't clear the MVs as they might be needed by a b frame
3289 // memset(s->motion_val + l_xy, 0, (l_wrap*2+1)*2*sizeof(int16_t));
3290 // memset(s->motion_val, 0, 2*sizeof(int16_t)*(2 + s->mb_width*2)*(2 + s->mb_height*2));
3291 s->last_mv[0][0][0]=
3292 s->last_mv[0][0][1]=
3293 s->last_mv[1][0][0]=
3294 s->last_mv[1][0][1]= 0;
3298 * decodes the group of blocks / video packet header.
3299 * @return bit position of the resync_marker, or <0 if none was found
3301 int ff_h263_resync(MpegEncContext *s){
3304 if(s->codec_id==CODEC_ID_MPEG4){
3306 align_get_bits(&s->gb);
3309 if(show_bits(&s->gb, 16)==0){
3310 pos= get_bits_count(&s->gb);
3311 if(s->codec_id==CODEC_ID_MPEG4)
3312 ret= mpeg4_decode_video_packet_header(s);
3314 ret= h263_decode_gob_header(s);
3318 //OK, it's not where it is supposed to be ...
3319 s->gb= s->last_resync_gb;
3320 align_get_bits(&s->gb);
3321 left= s->gb.size_in_bits - get_bits_count(&s->gb);
3323 for(;left>16+1+5+5; left-=8){
3324 if(show_bits(&s->gb, 16)==0){
3325 GetBitContext bak= s->gb;
3327 pos= get_bits_count(&s->gb);
3328 if(s->codec_id==CODEC_ID_MPEG4)
3329 ret= mpeg4_decode_video_packet_header(s);
3331 ret= h263_decode_gob_header(s);
3337 skip_bits(&s->gb, 8);
3344 * gets the average motion vector for a GMC MB.
3345 * @param n either 0 for the x component or 1 for y
3346 * @returns the average MV for a GMC MB
3348 static inline int get_amv(MpegEncContext *s, int n){
3349 int x, y, mb_v, sum, dx, dy, shift;
3350 int len = 1 << (s->f_code + 4);
3351 const int a= s->sprite_warping_accuracy;
3353 if(s->workaround_bugs & FF_BUG_AMV)
3354 len >>= s->quarter_sample;
3356 if(s->real_sprite_warping_points==1){
3357 if(s->divx_version==500 && s->divx_build==413)
3358 sum= s->sprite_offset[0][n] / (1<<(a - s->quarter_sample));
3360 sum= RSHIFT(s->sprite_offset[0][n]<<s->quarter_sample, a);
3362 dx= s->sprite_delta[n][0];
3363 dy= s->sprite_delta[n][1];
3364 shift= s->sprite_shift[0];
3365 if(n) dy -= 1<<(shift + a + 1);
3366 else dx -= 1<<(shift + a + 1);
3367 mb_v= s->sprite_offset[0][n] + dx*s->mb_x*16 + dy*s->mb_y*16;
3370 for(y=0; y<16; y++){
3374 //XXX FIXME optimize
3375 for(x=0; x<16; x++){
3380 sum= RSHIFT(sum, a+8-s->quarter_sample);
3383 if (sum < -len) sum= -len;
3384 else if (sum >= len) sum= len-1;
3390 * decodes first partition.
3391 * @return number of MBs decoded or <0 if an error occurred
3393 static int mpeg4_decode_partition_a(MpegEncContext *s){
3395 static const int8_t quant_tab[4] = { -1, -2, 1, 2 };
3397 /* decode first partition */
3399 s->first_slice_line=1;
3400 for(; s->mb_y<s->mb_height; s->mb_y++){
3401 ff_init_block_index(s);
3402 for(; s->mb_x<s->mb_width; s->mb_x++){
3403 const int xy= s->mb_x + s->mb_y*s->mb_stride;
3408 ff_update_block_index(s);
3409 if(s->mb_x == s->resync_mb_x && s->mb_y == s->resync_mb_y+1)
3410 s->first_slice_line=0;
3412 if(s->pict_type==FF_I_TYPE){
3416 if(show_bits_long(&s->gb, 19)==DC_MARKER){
3420 cbpc = get_vlc2(&s->gb, intra_MCBPC_vlc.table, INTRA_MCBPC_VLC_BITS, 2);
3422 av_log(s->avctx, AV_LOG_ERROR, "cbpc corrupted at %d %d\n", s->mb_x, s->mb_y);
3427 s->cbp_table[xy]= cbpc & 3;
3428 s->current_picture.mb_type[xy]= MB_TYPE_INTRA;
3432 ff_set_qscale(s, s->qscale + quant_tab[get_bits(&s->gb, 2)]);
3434 s->current_picture.qscale_table[xy]= s->qscale;
3436 s->mbintra_table[xy]= 1;
3439 int dc= mpeg4_decode_dc(s, i, &dc_pred_dir);
3441 av_log(s->avctx, AV_LOG_ERROR, "DC corrupted at %d %d\n", s->mb_x, s->mb_y);
3445 if(dc_pred_dir) dir|=1;
3447 s->pred_dir_table[xy]= dir;
3448 }else{ /* P/S_TYPE */
3449 int mx, my, pred_x, pred_y, bits;
3450 int16_t * const mot_val= s->current_picture.motion_val[0][s->block_index[0]];
3451 const int stride= s->b8_stride*2;
3454 bits= show_bits(&s->gb, 17);
3455 if(bits==MOTION_MARKER){
3461 if(s->pict_type==FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE){
3462 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_16x16 | MB_TYPE_GMC | MB_TYPE_L0;
3466 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_16x16 | MB_TYPE_L0;
3469 mot_val[0 ]= mot_val[2 ]=
3470 mot_val[0+stride]= mot_val[2+stride]= mx;
3471 mot_val[1 ]= mot_val[3 ]=
3472 mot_val[1+stride]= mot_val[3+stride]= my;
3474 if(s->mbintra_table[xy])
3475 ff_clean_intra_table_entries(s);
3479 cbpc = get_vlc2(&s->gb, inter_MCBPC_vlc.table, INTER_MCBPC_VLC_BITS, 2);
3481 av_log(s->avctx, AV_LOG_ERROR, "cbpc corrupted at %d %d\n", s->mb_x, s->mb_y);
3487 s->cbp_table[xy]= cbpc&(8+3); //8 is dquant
3489 s->mb_intra = ((cbpc & 4) != 0);
3492 s->current_picture.mb_type[xy]= MB_TYPE_INTRA;
3493 s->mbintra_table[xy]= 1;
3494 mot_val[0 ]= mot_val[2 ]=
3495 mot_val[0+stride]= mot_val[2+stride]= 0;
3496 mot_val[1 ]= mot_val[3 ]=
3497 mot_val[1+stride]= mot_val[3+stride]= 0;
3499 if(s->mbintra_table[xy])
3500 ff_clean_intra_table_entries(s);
3502 if(s->pict_type==FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE && (cbpc & 16) == 0)
3503 s->mcsel= get_bits1(&s->gb);
3506 if ((cbpc & 16) == 0) {
3507 /* 16x16 motion prediction */
3509 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
3511 mx = h263_decode_motion(s, pred_x, s->f_code);
3515 my = h263_decode_motion(s, pred_y, s->f_code);
3518 s->current_picture.mb_type[xy]= MB_TYPE_16x16 | MB_TYPE_L0;
3522 s->current_picture.mb_type[xy]= MB_TYPE_16x16 | MB_TYPE_GMC | MB_TYPE_L0;
3525 mot_val[0 ]= mot_val[2 ] =
3526 mot_val[0+stride]= mot_val[2+stride]= mx;
3527 mot_val[1 ]= mot_val[3 ]=
3528 mot_val[1+stride]= mot_val[3+stride]= my;
3531 s->current_picture.mb_type[xy]= MB_TYPE_8x8 | MB_TYPE_L0;
3533 int16_t *mot_val= h263_pred_motion(s, i, 0, &pred_x, &pred_y);
3534 mx = h263_decode_motion(s, pred_x, s->f_code);
3538 my = h263_decode_motion(s, pred_y, s->f_code);
3555 * decode second partition.
3556 * @return <0 if an error occurred
3558 static int mpeg4_decode_partition_b(MpegEncContext *s, int mb_count){
3560 static const int8_t quant_tab[4] = { -1, -2, 1, 2 };
3562 s->mb_x= s->resync_mb_x;
3563 s->first_slice_line=1;
3564 for(s->mb_y= s->resync_mb_y; mb_num < mb_count; s->mb_y++){
3565 ff_init_block_index(s);
3566 for(; mb_num < mb_count && s->mb_x<s->mb_width; s->mb_x++){
3567 const int xy= s->mb_x + s->mb_y*s->mb_stride;
3570 ff_update_block_index(s);
3571 if(s->mb_x == s->resync_mb_x && s->mb_y == s->resync_mb_y+1)
3572 s->first_slice_line=0;
3574 if(s->pict_type==FF_I_TYPE){
3575 int ac_pred= get_bits1(&s->gb);
3576 int cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
3578 av_log(s->avctx, AV_LOG_ERROR, "cbpy corrupted at %d %d\n", s->mb_x, s->mb_y);
3582 s->cbp_table[xy]|= cbpy<<2;
3583 s->current_picture.mb_type[xy] |= ac_pred*MB_TYPE_ACPRED;
3584 }else{ /* P || S_TYPE */
3585 if(IS_INTRA(s->current_picture.mb_type[xy])){
3587 int ac_pred = get_bits1(&s->gb);
3588 int cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
3591 av_log(s->avctx, AV_LOG_ERROR, "I cbpy corrupted at %d %d\n", s->mb_x, s->mb_y);
3595 if(s->cbp_table[xy] & 8) {
3596 ff_set_qscale(s, s->qscale + quant_tab[get_bits(&s->gb, 2)]);
3598 s->current_picture.qscale_table[xy]= s->qscale;
3602 int dc= mpeg4_decode_dc(s, i, &dc_pred_dir);
3604 av_log(s->avctx, AV_LOG_ERROR, "DC corrupted at %d %d\n", s->mb_x, s->mb_y);
3608 if(dc_pred_dir) dir|=1;
3610 s->cbp_table[xy]&= 3; //remove dquant
3611 s->cbp_table[xy]|= cbpy<<2;
3612 s->current_picture.mb_type[xy] |= ac_pred*MB_TYPE_ACPRED;
3613 s->pred_dir_table[xy]= dir;
3614 }else if(IS_SKIP(s->current_picture.mb_type[xy])){
3615 s->current_picture.qscale_table[xy]= s->qscale;
3616 s->cbp_table[xy]= 0;
3618 int cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
3621 av_log(s->avctx, AV_LOG_ERROR, "P cbpy corrupted at %d %d\n", s->mb_x, s->mb_y);
3625 if(s->cbp_table[xy] & 8) {
3626 ff_set_qscale(s, s->qscale + quant_tab[get_bits(&s->gb, 2)]);
3628 s->current_picture.qscale_table[xy]= s->qscale;
3630 s->cbp_table[xy]&= 3; //remove dquant
3631 s->cbp_table[xy]|= (cbpy^0xf)<<2;
3635 if(mb_num >= mb_count) return 0;
3642 * decodes the first & second partition
3643 * @return <0 if error (and sets error type in the error_status_table)
3645 int ff_mpeg4_decode_partitions(MpegEncContext *s)
3648 const int part_a_error= s->pict_type==FF_I_TYPE ? (DC_ERROR|MV_ERROR) : MV_ERROR;
3649 const int part_a_end = s->pict_type==FF_I_TYPE ? (DC_END |MV_END) : MV_END;
3651 mb_num= mpeg4_decode_partition_a(s);
3653 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, part_a_error);
3657 if(s->resync_mb_x + s->resync_mb_y*s->mb_width + mb_num > s->mb_num){
3658 av_log(s->avctx, AV_LOG_ERROR, "slice below monitor ...\n");
3659 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, part_a_error);
3663 s->mb_num_left= mb_num;
3665 if(s->pict_type==FF_I_TYPE){
3666 while(show_bits(&s->gb, 9) == 1)
3667 skip_bits(&s->gb, 9);
3668 if(get_bits_long(&s->gb, 19)!=DC_MARKER){
3669 av_log(s->avctx, AV_LOG_ERROR, "marker missing after first I partition at %d %d\n", s->mb_x, s->mb_y);
3673 while(show_bits(&s->gb, 10) == 1)
3674 skip_bits(&s->gb, 10);
3675 if(get_bits(&s->gb, 17)!=MOTION_MARKER){
3676 av_log(s->avctx, AV_LOG_ERROR, "marker missing after first P partition at %d %d\n", s->mb_x, s->mb_y);
3680 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, part_a_end);
3682 if( mpeg4_decode_partition_b(s, mb_num) < 0){
3683 if(s->pict_type==FF_P_TYPE)
3684 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, DC_ERROR);
3687 if(s->pict_type==FF_P_TYPE)
3688 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, DC_END);
3695 * decode partition C of one MB.
3696 * @return <0 if an error occurred
3698 static int mpeg4_decode_partitioned_mb(MpegEncContext *s, DCTELEM block[6][64])
3701 const int xy= s->mb_x + s->mb_y*s->mb_stride;
3703 mb_type= s->current_picture.mb_type[xy];
3704 cbp = s->cbp_table[xy];
3706 s->use_intra_dc_vlc= s->qscale < s->intra_dc_threshold;
3708 if(s->current_picture.qscale_table[xy] != s->qscale){
3709 ff_set_qscale(s, s->current_picture.qscale_table[xy] );
3712 if (s->pict_type == FF_P_TYPE || s->pict_type==FF_S_TYPE) {
3715 s->mv[0][i][0] = s->current_picture.motion_val[0][ s->block_index[i] ][0];
3716 s->mv[0][i][1] = s->current_picture.motion_val[0][ s->block_index[i] ][1];
3718 s->mb_intra = IS_INTRA(mb_type);
3720 if (IS_SKIP(mb_type)) {
3723 s->block_last_index[i] = -1;
3724 s->mv_dir = MV_DIR_FORWARD;
3725 s->mv_type = MV_TYPE_16X16;
3726 if(s->pict_type==FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE){
3733 }else if(s->mb_intra){
3734 s->ac_pred = IS_ACPRED(s->current_picture.mb_type[xy]);
3735 }else if(!s->mb_intra){
3736 // s->mcsel= 0; //FIXME do we need to init that
3738 s->mv_dir = MV_DIR_FORWARD;
3739 if (IS_8X8(mb_type)) {
3740 s->mv_type = MV_TYPE_8X8;
3742 s->mv_type = MV_TYPE_16X16;
3745 } else { /* I-Frame */
3747 s->ac_pred = IS_ACPRED(s->current_picture.mb_type[xy]);
3750 if (!IS_SKIP(mb_type)) {
3752 s->dsp.clear_blocks(s->block[0]);
3753 /* decode each block */
3754 for (i = 0; i < 6; i++) {
3755 if(mpeg4_decode_block(s, block[i], i, cbp&32, s->mb_intra, s->rvlc) < 0){
3756 av_log(s->avctx, AV_LOG_ERROR, "texture corrupted at %d %d %d\n", s->mb_x, s->mb_y, s->mb_intra);
3763 /* per-MB end of slice check */
3765 if(--s->mb_num_left <= 0){
3766 //printf("%06X %d\n", show_bits(&s->gb, 24), s->gb.size_in_bits - get_bits_count(&s->gb));
3767 if(mpeg4_is_resync(s))
3772 if(mpeg4_is_resync(s)){
3773 const int delta= s->mb_x + 1 == s->mb_width ? 2 : 1;
3774 if(s->cbp_table[xy+delta])
3782 * read the next MVs for OBMC. yes this is a ugly hack, feel free to send a patch :)
3784 static void preview_obmc(MpegEncContext *s){
3785 GetBitContext gb= s->gb;
3787 int cbpc, i, pred_x, pred_y, mx, my;
3789 const int xy= s->mb_x + 1 + s->mb_y * s->mb_stride;
3790 const int stride= s->b8_stride*2;
3793 s->block_index[i]+= 2;
3795 s->block_index[i]+= 1;
3798 assert(s->pict_type == FF_P_TYPE);
3801 if (get_bits1(&s->gb)) {
3803 mot_val = s->current_picture.motion_val[0][ s->block_index[0] ];
3804 mot_val[0 ]= mot_val[2 ]=
3805 mot_val[0+stride]= mot_val[2+stride]= 0;
3806 mot_val[1 ]= mot_val[3 ]=
3807 mot_val[1+stride]= mot_val[3+stride]= 0;
3809 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_16x16 | MB_TYPE_L0;
3812 cbpc = get_vlc2(&s->gb, inter_MCBPC_vlc.table, INTER_MCBPC_VLC_BITS, 2);
3816 s->current_picture.mb_type[xy]= MB_TYPE_INTRA;
3818 get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
3820 if(s->modified_quant){
3821 if(get_bits1(&s->gb)) skip_bits(&s->gb, 1);
3822 else skip_bits(&s->gb, 5);
3824 skip_bits(&s->gb, 2);
3827 if ((cbpc & 16) == 0) {
3828 s->current_picture.mb_type[xy]= MB_TYPE_16x16 | MB_TYPE_L0;
3829 /* 16x16 motion prediction */
3830 mot_val= h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
3832 mx = h263p_decode_umotion(s, pred_x);
3834 mx = h263_decode_motion(s, pred_x, 1);
3837 my = h263p_decode_umotion(s, pred_y);
3839 my = h263_decode_motion(s, pred_y, 1);
3841 mot_val[0 ]= mot_val[2 ]=
3842 mot_val[0+stride]= mot_val[2+stride]= mx;
3843 mot_val[1 ]= mot_val[3 ]=
3844 mot_val[1+stride]= mot_val[3+stride]= my;
3846 s->current_picture.mb_type[xy]= MB_TYPE_8x8 | MB_TYPE_L0;
3848 mot_val = h263_pred_motion(s, i, 0, &pred_x, &pred_y);
3850 mx = h263p_decode_umotion(s, pred_x);
3852 mx = h263_decode_motion(s, pred_x, 1);
3855 my = h263p_decode_umotion(s, pred_y);
3857 my = h263_decode_motion(s, pred_y, 1);
3858 if (s->umvplus && (mx - pred_x) == 1 && (my - pred_y) == 1)
3859 skip_bits1(&s->gb); /* Bit stuffing to prevent PSC */
3868 s->block_index[i]-= 2;
3870 s->block_index[i]-= 1;
3876 static void h263_decode_dquant(MpegEncContext *s){
3877 static const int8_t quant_tab[4] = { -1, -2, 1, 2 };
3879 if(s->modified_quant){
3880 if(get_bits1(&s->gb))
3881 s->qscale= modified_quant_tab[get_bits1(&s->gb)][ s->qscale ];
3883 s->qscale= get_bits(&s->gb, 5);
3885 s->qscale += quant_tab[get_bits(&s->gb, 2)];
3886 ff_set_qscale(s, s->qscale);
3889 int ff_h263_decode_mb(MpegEncContext *s,
3890 DCTELEM block[6][64])
3892 int cbpc, cbpy, i, cbp, pred_x, pred_y, mx, my, dquant;
3894 const int xy= s->mb_x + s->mb_y * s->mb_stride;
3896 assert(!s->h263_pred);
3898 if (s->pict_type == FF_P_TYPE) {
3900 if (get_bits1(&s->gb)) {
3904 s->block_last_index[i] = -1;
3905 s->mv_dir = MV_DIR_FORWARD;
3906 s->mv_type = MV_TYPE_16X16;
3907 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_16x16 | MB_TYPE_L0;
3910 s->mb_skipped = !(s->obmc | s->loop_filter);
3913 cbpc = get_vlc2(&s->gb, inter_MCBPC_vlc.table, INTER_MCBPC_VLC_BITS, 2);
3914 //fprintf(stderr, "\tCBPC: %d", cbpc);
3916 av_log(s->avctx, AV_LOG_ERROR, "cbpc damaged at %d %d\n", s->mb_x, s->mb_y);
3921 s->dsp.clear_blocks(s->block[0]);
3924 s->mb_intra = ((cbpc & 4) != 0);
3925 if (s->mb_intra) goto intra;
3927 cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
3929 if(s->alt_inter_vlc==0 || (cbpc & 3)!=3)
3932 cbp = (cbpc & 3) | (cbpy << 2);
3934 h263_decode_dquant(s);
3937 s->mv_dir = MV_DIR_FORWARD;
3938 if ((cbpc & 16) == 0) {
3939 s->current_picture.mb_type[xy]= MB_TYPE_16x16 | MB_TYPE_L0;
3940 /* 16x16 motion prediction */
3941 s->mv_type = MV_TYPE_16X16;
3942 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
3944 mx = h263p_decode_umotion(s, pred_x);
3946 mx = h263_decode_motion(s, pred_x, 1);
3952 my = h263p_decode_umotion(s, pred_y);
3954 my = h263_decode_motion(s, pred_y, 1);
3958 s->mv[0][0][0] = mx;
3959 s->mv[0][0][1] = my;
3961 if (s->umvplus && (mx - pred_x) == 1 && (my - pred_y) == 1)
3962 skip_bits1(&s->gb); /* Bit stuffing to prevent PSC */
3964 s->current_picture.mb_type[xy]= MB_TYPE_8x8 | MB_TYPE_L0;
3965 s->mv_type = MV_TYPE_8X8;
3967 mot_val = h263_pred_motion(s, i, 0, &pred_x, &pred_y);
3969 mx = h263p_decode_umotion(s, pred_x);
3971 mx = h263_decode_motion(s, pred_x, 1);
3976 my = h263p_decode_umotion(s, pred_y);
3978 my = h263_decode_motion(s, pred_y, 1);
3981 s->mv[0][i][0] = mx;
3982 s->mv[0][i][1] = my;
3983 if (s->umvplus && (mx - pred_x) == 1 && (my - pred_y) == 1)
3984 skip_bits1(&s->gb); /* Bit stuffing to prevent PSC */
3990 /* decode each block */
3991 for (i = 0; i < 6; i++) {
3992 if (h263_decode_block(s, block[i], i, cbp&32) < 0)
3998 if(s->pict_type == FF_P_TYPE && s->mb_x+1<s->mb_width && s->mb_num_left != 1)
4001 } else if(s->pict_type==FF_B_TYPE) {
4003 const int stride= s->b8_stride;
4004 int16_t *mot_val0 = s->current_picture.motion_val[0][ 2*(s->mb_x + s->mb_y*stride) ];
4005 int16_t *mot_val1 = s->current_picture.motion_val[1][ 2*(s->mb_x + s->mb_y*stride) ];
4006 // const int mv_xy= s->mb_x + 1 + s->mb_y * s->mb_stride;
4009 mot_val0[0 ]= mot_val0[2 ]= mot_val0[0+2*stride]= mot_val0[2+2*stride]=
4010 mot_val0[1 ]= mot_val0[3 ]= mot_val0[1+2*stride]= mot_val0[3+2*stride]=
4011 mot_val1[0 ]= mot_val1[2 ]= mot_val1[0+2*stride]= mot_val1[2+2*stride]=
4012 mot_val1[1 ]= mot_val1[3 ]= mot_val1[1+2*stride]= mot_val1[3+2*stride]= 0;
4015 mb_type= get_vlc2(&s->gb, h263_mbtype_b_vlc.table, H263_MBTYPE_B_VLC_BITS, 2);
4017 av_log(s->avctx, AV_LOG_ERROR, "b mb_type damaged at %d %d\n", s->mb_x, s->mb_y);
4021 mb_type= h263_mb_type_b_map[ mb_type ];
4024 s->mb_intra = IS_INTRA(mb_type);
4025 if(HAS_CBP(mb_type)){
4026 s->dsp.clear_blocks(s->block[0]);
4027 cbpc = get_vlc2(&s->gb, cbpc_b_vlc.table, CBPC_B_VLC_BITS, 1);
4029 dquant = IS_QUANT(mb_type);
4033 cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
4036 av_log(s->avctx, AV_LOG_ERROR, "b cbpy damaged at %d %d\n", s->mb_x, s->mb_y);
4040 if(s->alt_inter_vlc==0 || (cbpc & 3)!=3)
4043 cbp = (cbpc & 3) | (cbpy << 2);
4047 assert(!s->mb_intra);
4049 if(IS_QUANT(mb_type)){
4050 h263_decode_dquant(s);
4053 if(IS_DIRECT(mb_type)){
4054 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
4055 mb_type |= ff_mpeg4_set_direct_mv(s, 0, 0);
4058 s->mv_type= MV_TYPE_16X16;
4061 if(USES_LIST(mb_type, 0)){
4062 int16_t *mot_val= h263_pred_motion(s, 0, 0, &mx, &my);
4063 s->mv_dir = MV_DIR_FORWARD;
4065 mx = h263_decode_motion(s, mx, 1);
4066 my = h263_decode_motion(s, my, 1);
4068 s->mv[0][0][0] = mx;
4069 s->mv[0][0][1] = my;
4070 mot_val[0 ]= mot_val[2 ]= mot_val[0+2*stride]= mot_val[2+2*stride]= mx;
4071 mot_val[1 ]= mot_val[3 ]= mot_val[1+2*stride]= mot_val[3+2*stride]= my;
4074 if(USES_LIST(mb_type, 1)){
4075 int16_t *mot_val= h263_pred_motion(s, 0, 1, &mx, &my);
4076 s->mv_dir |= MV_DIR_BACKWARD;
4078 mx = h263_decode_motion(s, mx, 1);
4079 my = h263_decode_motion(s, my, 1);
4081 s->mv[1][0][0] = mx;
4082 s->mv[1][0][1] = my;
4083 mot_val[0 ]= mot_val[2 ]= mot_val[0+2*stride]= mot_val[2+2*stride]= mx;
4084 mot_val[1 ]= mot_val[3 ]= mot_val[1+2*stride]= mot_val[3+2*stride]= my;
4088 s->current_picture.mb_type[xy]= mb_type;
4090 /* decode each block */
4091 for (i = 0; i < 6; i++) {
4092 if (h263_decode_block(s, block[i], i, cbp&32) < 0)
4096 } else { /* I-Frame */
4098 cbpc = get_vlc2(&s->gb, intra_MCBPC_vlc.table, INTRA_MCBPC_VLC_BITS, 2);
4100 av_log(s->avctx, AV_LOG_ERROR, "I cbpc damaged at %d %d\n", s->mb_x, s->mb_y);
4105 s->dsp.clear_blocks(s->block[0]);
4110 s->current_picture.mb_type[xy]= MB_TYPE_INTRA;
4112 s->ac_pred = get_bits1(&s->gb);
4114 s->current_picture.mb_type[xy]= MB_TYPE_INTRA | MB_TYPE_ACPRED;
4116 s->h263_aic_dir = get_bits1(&s->gb);
4121 cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
4123 av_log(s->avctx, AV_LOG_ERROR, "I cbpy damaged at %d %d\n", s->mb_x, s->mb_y);
4126 cbp = (cbpc & 3) | (cbpy << 2);
4128 h263_decode_dquant(s);
4131 /* decode each block */
4132 for (i = 0; i < 6; i++) {
4133 if (h263_decode_block(s, block[i], i, cbp&32) < 0)
4140 /* per-MB end of slice check */
4142 int v= show_bits(&s->gb, 16);
4144 if(get_bits_count(&s->gb) + 16 > s->gb.size_in_bits){
4145 v>>= get_bits_count(&s->gb) + 16 - s->gb.size_in_bits;
4155 int ff_mpeg4_decode_mb(MpegEncContext *s,
4156 DCTELEM block[6][64])
4158 int cbpc, cbpy, i, cbp, pred_x, pred_y, mx, my, dquant;
4160 static int8_t quant_tab[4] = { -1, -2, 1, 2 };
4161 const int xy= s->mb_x + s->mb_y * s->mb_stride;
4163 assert(s->h263_pred);
4165 if (s->pict_type == FF_P_TYPE || s->pict_type==FF_S_TYPE) {
4167 if (get_bits1(&s->gb)) {
4171 s->block_last_index[i] = -1;
4172 s->mv_dir = MV_DIR_FORWARD;
4173 s->mv_type = MV_TYPE_16X16;
4174 if(s->pict_type==FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE){
4175 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_GMC | MB_TYPE_16x16 | MB_TYPE_L0;
4177 s->mv[0][0][0]= get_amv(s, 0);
4178 s->mv[0][0][1]= get_amv(s, 1);
4182 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_16x16 | MB_TYPE_L0;
4190 cbpc = get_vlc2(&s->gb, inter_MCBPC_vlc.table, INTER_MCBPC_VLC_BITS, 2);
4191 //fprintf(stderr, "\tCBPC: %d", cbpc);
4193 av_log(s->avctx, AV_LOG_ERROR, "cbpc damaged at %d %d\n", s->mb_x, s->mb_y);
4198 s->dsp.clear_blocks(s->block[0]);
4200 s->mb_intra = ((cbpc & 4) != 0);
4201 if (s->mb_intra) goto intra;
4203 if(s->pict_type==FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE && (cbpc & 16) == 0)
4204 s->mcsel= get_bits1(&s->gb);
4206 cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1) ^ 0x0F;
4208 cbp = (cbpc & 3) | (cbpy << 2);
4210 ff_set_qscale(s, s->qscale + quant_tab[get_bits(&s->gb, 2)]);
4212 if((!s->progressive_sequence) && (cbp || (s->workaround_bugs&FF_BUG_XVID_ILACE)))
4213 s->interlaced_dct= get_bits1(&s->gb);
4215 s->mv_dir = MV_DIR_FORWARD;
4216 if ((cbpc & 16) == 0) {
4218 s->current_picture.mb_type[xy]= MB_TYPE_GMC | MB_TYPE_16x16 | MB_TYPE_L0;
4219 /* 16x16 global motion prediction */
4220 s->mv_type = MV_TYPE_16X16;
4223 s->mv[0][0][0] = mx;
4224 s->mv[0][0][1] = my;
4225 }else if((!s->progressive_sequence) && get_bits1(&s->gb)){
4226 s->current_picture.mb_type[xy]= MB_TYPE_16x8 | MB_TYPE_L0 | MB_TYPE_INTERLACED;
4227 /* 16x8 field motion prediction */
4228 s->mv_type= MV_TYPE_FIELD;
4230 s->field_select[0][0]= get_bits1(&s->gb);
4231 s->field_select[0][1]= get_bits1(&s->gb);
4233 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
4236 mx = h263_decode_motion(s, pred_x, s->f_code);
4240 my = h263_decode_motion(s, pred_y/2, s->f_code);
4244 s->mv[0][i][0] = mx;
4245 s->mv[0][i][1] = my;
4248 s->current_picture.mb_type[xy]= MB_TYPE_16x16 | MB_TYPE_L0;
4249 /* 16x16 motion prediction */
4250 s->mv_type = MV_TYPE_16X16;
4251 h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
4252 mx = h263_decode_motion(s, pred_x, s->f_code);
4257 my = h263_decode_motion(s, pred_y, s->f_code);
4261 s->mv[0][0][0] = mx;
4262 s->mv[0][0][1] = my;
4265 s->current_picture.mb_type[xy]= MB_TYPE_8x8 | MB_TYPE_L0;
4266 s->mv_type = MV_TYPE_8X8;
4268 mot_val = h263_pred_motion(s, i, 0, &pred_x, &pred_y);
4269 mx = h263_decode_motion(s, pred_x, s->f_code);
4273 my = h263_decode_motion(s, pred_y, s->f_code);
4276 s->mv[0][i][0] = mx;
4277 s->mv[0][i][1] = my;
4282 } else if(s->pict_type==FF_B_TYPE) {
4283 int modb1; // first bit of modb
4284 int modb2; // second bit of modb
4287 s->mb_intra = 0; //B-frames never contain intra blocks
4288 s->mcsel=0; // ... true gmc blocks
4292 s->last_mv[i][0][0]=
4293 s->last_mv[i][0][1]=
4294 s->last_mv[i][1][0]=
4295 s->last_mv[i][1][1]= 0;
4299 /* if we skipped it in the future P Frame than skip it now too */
4300 s->mb_skipped= s->next_picture.mbskip_table[s->mb_y * s->mb_stride + s->mb_x]; // Note, skiptab=0 if last was GMC
4305 s->block_last_index[i] = -1;
4307 s->mv_dir = MV_DIR_FORWARD;
4308 s->mv_type = MV_TYPE_16X16;
4313 s->current_picture.mb_type[xy]= MB_TYPE_SKIP | MB_TYPE_16x16 | MB_TYPE_L0;
4317 modb1= get_bits1(&s->gb);
4319 mb_type= MB_TYPE_DIRECT2 | MB_TYPE_SKIP | MB_TYPE_L0L1; //like MB_TYPE_B_DIRECT but no vectors coded
4322 modb2= get_bits1(&s->gb);
4323 mb_type= get_vlc2(&s->gb, mb_type_b_vlc.table, MB_TYPE_B_VLC_BITS, 1);
4325 av_log(s->avctx, AV_LOG_ERROR, "illegal MB_type\n");
4328 mb_type= mb_type_b_map[ mb_type ];
4331 s->dsp.clear_blocks(s->block[0]);
4332 cbp= get_bits(&s->gb, 6);
4335 if ((!IS_DIRECT(mb_type)) && cbp) {
4336 if(get_bits1(&s->gb)){
4337 ff_set_qscale(s, s->qscale + get_bits1(&s->gb)*4 - 2);
4341 if(!s->progressive_sequence){
4343 s->interlaced_dct= get_bits1(&s->gb);
4345 if(!IS_DIRECT(mb_type) && get_bits1(&s->gb)){
4346 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
4347 mb_type &= ~MB_TYPE_16x16;
4349 if(USES_LIST(mb_type, 0)){
4350 s->field_select[0][0]= get_bits1(&s->gb);
4351 s->field_select[0][1]= get_bits1(&s->gb);
4353 if(USES_LIST(mb_type, 1)){
4354 s->field_select[1][0]= get_bits1(&s->gb);
4355 s->field_select[1][1]= get_bits1(&s->gb);
4361 if((mb_type & (MB_TYPE_DIRECT2|MB_TYPE_INTERLACED)) == 0){
4362 s->mv_type= MV_TYPE_16X16;
4364 if(USES_LIST(mb_type, 0)){
4365 s->mv_dir = MV_DIR_FORWARD;
4367 mx = h263_decode_motion(s, s->last_mv[0][0][0], s->f_code);
4368 my = h263_decode_motion(s, s->last_mv[0][0][1], s->f_code);
4369 s->last_mv[0][1][0]= s->last_mv[0][0][0]= s->mv[0][0][0] = mx;
4370 s->last_mv[0][1][1]= s->last_mv[0][0][1]= s->mv[0][0][1] = my;
4373 if(USES_LIST(mb_type, 1)){
4374 s->mv_dir |= MV_DIR_BACKWARD;
4376 mx = h263_decode_motion(s, s->last_mv[1][0][0], s->b_code);
4377 my = h263_decode_motion(s, s->last_mv[1][0][1], s->b_code);
4378 s->last_mv[1][1][0]= s->last_mv[1][0][0]= s->mv[1][0][0] = mx;
4379 s->last_mv[1][1][1]= s->last_mv[1][0][1]= s->mv[1][0][1] = my;
4381 }else if(!IS_DIRECT(mb_type)){
4382 s->mv_type= MV_TYPE_FIELD;
4384 if(USES_LIST(mb_type, 0)){
4385 s->mv_dir = MV_DIR_FORWARD;
4388 mx = h263_decode_motion(s, s->last_mv[0][i][0] , s->f_code);
4389 my = h263_decode_motion(s, s->last_mv[0][i][1]/2, s->f_code);
4390 s->last_mv[0][i][0]= s->mv[0][i][0] = mx;
4391 s->last_mv[0][i][1]= (s->mv[0][i][1] = my)*2;
4395 if(USES_LIST(mb_type, 1)){
4396 s->mv_dir |= MV_DIR_BACKWARD;
4399 mx = h263_decode_motion(s, s->last_mv[1][i][0] , s->b_code);
4400 my = h263_decode_motion(s, s->last_mv[1][i][1]/2, s->b_code);
4401 s->last_mv[1][i][0]= s->mv[1][i][0] = mx;
4402 s->last_mv[1][i][1]= (s->mv[1][i][1] = my)*2;
4408 if(IS_DIRECT(mb_type)){
4409 if(IS_SKIP(mb_type))
4412 mx = h263_decode_motion(s, 0, 1);
4413 my = h263_decode_motion(s, 0, 1);
4416 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
4417 mb_type |= ff_mpeg4_set_direct_mv(s, mx, my);
4419 s->current_picture.mb_type[xy]= mb_type;
4420 } else { /* I-Frame */
4422 cbpc = get_vlc2(&s->gb, intra_MCBPC_vlc.table, INTRA_MCBPC_VLC_BITS, 2);
4424 av_log(s->avctx, AV_LOG_ERROR, "I cbpc damaged at %d %d\n", s->mb_x, s->mb_y);
4432 s->ac_pred = get_bits1(&s->gb);
4434 s->current_picture.mb_type[xy]= MB_TYPE_INTRA | MB_TYPE_ACPRED;
4436 s->current_picture.mb_type[xy]= MB_TYPE_INTRA;
4438 cbpy = get_vlc2(&s->gb, cbpy_vlc.table, CBPY_VLC_BITS, 1);
4440 av_log(s->avctx, AV_LOG_ERROR, "I cbpy damaged at %d %d\n", s->mb_x, s->mb_y);
4443 cbp = (cbpc & 3) | (cbpy << 2);
4445 s->use_intra_dc_vlc= s->qscale < s->intra_dc_threshold;
4448 ff_set_qscale(s, s->qscale + quant_tab[get_bits(&s->gb, 2)]);
4451 if(!s->progressive_sequence)
4452 s->interlaced_dct= get_bits1(&s->gb);
4454 s->dsp.clear_blocks(s->block[0]);
4455 /* decode each block */
4456 for (i = 0; i < 6; i++) {
4457 if (mpeg4_decode_block(s, block[i], i, cbp&32, 1, 0) < 0)
4464 /* decode each block */
4465 for (i = 0; i < 6; i++) {
4466 if (mpeg4_decode_block(s, block[i], i, cbp&32, 0, 0) < 0)
4472 /* per-MB end of slice check */
4473 if(s->codec_id==CODEC_ID_MPEG4){
4474 if(mpeg4_is_resync(s)){
4475 const int delta= s->mb_x + 1 == s->mb_width ? 2 : 1;
4476 if(s->pict_type==FF_B_TYPE && s->next_picture.mbskip_table[xy + delta])
4485 static int h263_decode_motion(MpegEncContext * s, int pred, int f_code)
4487 int code, val, sign, shift, l;
4488 code = get_vlc2(&s->gb, mv_vlc.table, MV_VLC_BITS, 2);
4495 sign = get_bits1(&s->gb);
4499 val = (val - 1) << shift;
4500 val |= get_bits(&s->gb, shift);
4507 /* modulo decoding */
4508 if (!s->h263_long_vectors) {
4509 l = INT_BIT - 5 - f_code;
4512 /* horrible h263 long vector mode */
4513 if (pred < -31 && val < -63)
4515 if (pred > 32 && val > 63)
4522 /* Decodes RVLC of H.263+ UMV */
4523 static int h263p_decode_umotion(MpegEncContext * s, int pred)
4527 if (get_bits1(&s->gb)) /* Motion difference = 0 */
4530 code = 2 + get_bits1(&s->gb);
4532 while (get_bits1(&s->gb))
4535 code += get_bits1(&s->gb);
4540 code = (sign) ? (pred - code) : (pred + code);
4542 av_log( s->avctx, AV_LOG_DEBUG,"H.263+ UMV Motion = %d\n", code);
4548 static int h263_decode_block(MpegEncContext * s, DCTELEM * block,
4551 int code, level, i, j, last, run;
4552 RLTable *rl = &rl_inter;
4553 const uint8_t *scan_table;
4554 GetBitContext gb= s->gb;
4556 scan_table = s->intra_scantable.permutated;
4557 if (s->h263_aic && s->mb_intra) {
4561 if (s->h263_aic_dir)
4562 scan_table = s->intra_v_scantable.permutated; /* left */
4564 scan_table = s->intra_h_scantable.permutated; /* top */
4566 } else if (s->mb_intra) {
4568 if(s->codec_id == CODEC_ID_RV10){
4569 #if CONFIG_RV10_DECODER
4570 if (s->rv10_version == 3 && s->pict_type == FF_I_TYPE) {
4571 int component, diff;
4572 component = (n <= 3 ? 0 : n - 4 + 1);
4573 level = s->last_dc[component];
4574 if (s->rv10_first_dc_coded[component]) {
4575 diff = rv_decode_dc(s, n);
4579 level = level & 0xff; /* handle wrap round */
4580 s->last_dc[component] = level;
4582 s->rv10_first_dc_coded[component] = 1;
4585 level = get_bits(&s->gb, 8);
4591 level = get_bits(&s->gb, 8);
4592 if((level&0x7F) == 0){
4593 av_log(s->avctx, AV_LOG_ERROR, "illegal dc %d at %d %d\n", level, s->mb_x, s->mb_y);
4594 if(s->error_recognition >= FF_ER_COMPLIANT)
4606 if (s->mb_intra && s->h263_aic)
4608 s->block_last_index[n] = i - 1;
4613 code = get_vlc2(&s->gb, rl->vlc.table, TEX_VLC_BITS, 2);
4615 av_log(s->avctx, AV_LOG_ERROR, "illegal ac vlc code at %dx%d\n", s->mb_x, s->mb_y);
4618 if (code == rl->n) {
4620 if (s->h263_flv > 1) {
4621 int is11 = get_bits1(&s->gb);
4622 last = get_bits1(&s->gb);
4623 run = get_bits(&s->gb, 6);
4625 level = get_sbits(&s->gb, 11);
4627 level = get_sbits(&s->gb, 7);
4630 last = get_bits1(&s->gb);
4631 run = get_bits(&s->gb, 6);
4632 level = (int8_t)get_bits(&s->gb, 8);
4634 if (s->codec_id == CODEC_ID_RV10) {
4635 /* XXX: should patch encoder too */
4636 level = get_sbits(&s->gb, 12);
4638 level = get_bits(&s->gb, 5);
4639 level |= get_sbits(&s->gb, 6)<<5;
4644 run = rl->table_run[code];
4645 level = rl->table_level[code];
4646 last = code >= rl->last;
4647 if (get_bits1(&s->gb))
4652 if(s->alt_inter_vlc && rl == &rl_inter && !s->mb_intra){
4653 //Looks like a hack but no, it's the way it is supposed to work ...
4657 s->dsp.clear_block(block);
4660 av_log(s->avctx, AV_LOG_ERROR, "run overflow at %dx%d i:%d\n", s->mb_x, s->mb_y, s->mb_intra);
4670 if (s->mb_intra && s->h263_aic) {
4671 h263_pred_acdc(s, block, n);
4674 s->block_last_index[n] = i;
4679 * decodes the dc value.
4680 * @param n block index (0-3 are luma, 4-5 are chroma)
4681 * @param dir_ptr the prediction direction will be stored here
4682 * @return the quantized dc
4684 static inline int mpeg4_decode_dc(MpegEncContext * s, int n, int *dir_ptr)
4689 code = get_vlc2(&s->gb, dc_lum.table, DC_VLC_BITS, 1);
4691 code = get_vlc2(&s->gb, dc_chrom.table, DC_VLC_BITS, 1);
4692 if (code < 0 || code > 9 /* && s->nbit<9 */){
4693 av_log(s->avctx, AV_LOG_ERROR, "illegal dc vlc\n");
4701 level= 2*get_bits1(&s->gb)-1;
4703 if(get_bits1(&s->gb))
4704 level = get_bits(&s->gb, code-1) + (1<<(code-1));
4706 level = -get_bits(&s->gb, code-1) - (1<<(code-1));
4709 level = get_xbits(&s->gb, code);
4713 if(get_bits1(&s->gb)==0){ /* marker */
4714 if(s->error_recognition>=2){
4715 av_log(s->avctx, AV_LOG_ERROR, "dc marker bit missing\n");
4722 return ff_mpeg4_pred_dc(s, n, level, dir_ptr, 0);
4727 * @return <0 if an error occurred
4729 static inline int mpeg4_decode_block(MpegEncContext * s, DCTELEM * block,
4730 int n, int coded, int intra, int rvlc)
4732 int level, i, last, run;
4735 RL_VLC_ELEM * rl_vlc;
4736 const uint8_t * scan_table;
4739 //Note intra & rvlc should be optimized away if this is inlined
4742 if(s->use_intra_dc_vlc){
4744 if(s->partitioned_frame){
4745 level = s->dc_val[0][ s->block_index[n] ];
4746 if(n<4) level= FASTDIV((level + (s->y_dc_scale>>1)), s->y_dc_scale);
4747 else level= FASTDIV((level + (s->c_dc_scale>>1)), s->c_dc_scale);
4748 dc_pred_dir= (s->pred_dir_table[s->mb_x + s->mb_y*s->mb_stride]<<n)&32;
4750 level = mpeg4_decode_dc(s, n, &dc_pred_dir);
4758 ff_mpeg4_pred_dc(s, n, 0, &dc_pred_dir, 0);
4764 rl = &rvlc_rl_intra;
4765 rl_vlc = rvlc_rl_intra.rl_vlc[0];
4768 rl_vlc = rl_intra.rl_vlc[0];
4771 if (dc_pred_dir == 0)
4772 scan_table = s->intra_v_scantable.permutated; /* left */
4774 scan_table = s->intra_h_scantable.permutated; /* top */
4776 scan_table = s->intra_scantable.permutated;
4783 s->block_last_index[n] = i;
4786 if(rvlc) rl = &rvlc_rl_inter;
4787 else rl = &rl_inter;
4789 scan_table = s->intra_scantable.permutated;
4795 rl_vlc = rvlc_rl_inter.rl_vlc[0];
4797 rl_vlc = rl_inter.rl_vlc[0];
4800 qmul = s->qscale << 1;
4801 qadd = (s->qscale - 1) | 1;
4803 rl_vlc = rvlc_rl_inter.rl_vlc[s->qscale];
4805 rl_vlc = rl_inter.rl_vlc[s->qscale];
4810 OPEN_READER(re, &s->gb);
4812 UPDATE_CACHE(re, &s->gb);
4813 GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 0);
4817 if(SHOW_UBITS(re, &s->gb, 1)==0){
4818 av_log(s->avctx, AV_LOG_ERROR, "1. marker bit missing in rvlc esc\n");
4820 }; SKIP_CACHE(re, &s->gb, 1);
4822 last= SHOW_UBITS(re, &s->gb, 1); SKIP_CACHE(re, &s->gb, 1);
4823 run= SHOW_UBITS(re, &s->gb, 6); LAST_SKIP_CACHE(re, &s->gb, 6);
4824 SKIP_COUNTER(re, &s->gb, 1+1+6);
4825 UPDATE_CACHE(re, &s->gb);
4827 if(SHOW_UBITS(re, &s->gb, 1)==0){
4828 av_log(s->avctx, AV_LOG_ERROR, "2. marker bit missing in rvlc esc\n");
4830 }; SKIP_CACHE(re, &s->gb, 1);
4832 level= SHOW_UBITS(re, &s->gb, 11); SKIP_CACHE(re, &s->gb, 11);
4834 if(SHOW_UBITS(re, &s->gb, 5)!=0x10){
4835 av_log(s->avctx, AV_LOG_ERROR, "reverse esc missing\n");
4837 }; SKIP_CACHE(re, &s->gb, 5);
4839 level= level * qmul + qadd;
4840 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1); LAST_SKIP_CACHE(re, &s->gb, 1);
4841 SKIP_COUNTER(re, &s->gb, 1+11+5+1);
4847 cache= GET_CACHE(re, &s->gb);
4850 cache ^= 0xC0000000;
4852 if (cache&0x80000000) {
4853 if (cache&0x40000000) {
4855 SKIP_CACHE(re, &s->gb, 2);
4856 last= SHOW_UBITS(re, &s->gb, 1); SKIP_CACHE(re, &s->gb, 1);
4857 run= SHOW_UBITS(re, &s->gb, 6); LAST_SKIP_CACHE(re, &s->gb, 6);
4858 SKIP_COUNTER(re, &s->gb, 2+1+6);
4859 UPDATE_CACHE(re, &s->gb);
4862 level= SHOW_SBITS(re, &s->gb, 12); LAST_SKIP_BITS(re, &s->gb, 12);
4864 if(SHOW_UBITS(re, &s->gb, 1)==0){
4865 av_log(s->avctx, AV_LOG_ERROR, "1. marker bit missing in 3. esc\n");
4867 }; SKIP_CACHE(re, &s->gb, 1);
4869 level= SHOW_SBITS(re, &s->gb, 12); SKIP_CACHE(re, &s->gb, 12);
4871 if(SHOW_UBITS(re, &s->gb, 1)==0){
4872 av_log(s->avctx, AV_LOG_ERROR, "2. marker bit missing in 3. esc\n");
4874 }; LAST_SKIP_CACHE(re, &s->gb, 1);
4876 SKIP_COUNTER(re, &s->gb, 1+12+1);
4880 if(s->error_recognition >= FF_ER_COMPLIANT){
4881 const int abs_level= FFABS(level);
4882 if(abs_level<=MAX_LEVEL && run<=MAX_RUN){
4883 const int run1= run - rl->max_run[last][abs_level] - 1;
4884 if(abs_level <= rl->max_level[last][run]){
4885 av_log(s->avctx, AV_LOG_ERROR, "illegal 3. esc, vlc encoding possible\n");
4888 if(s->error_recognition > FF_ER_COMPLIANT){
4889 if(abs_level <= rl->max_level[last][run]*2){
4890 av_log(s->avctx, AV_LOG_ERROR, "illegal 3. esc, esc 1 encoding possible\n");
4893 if(run1 >= 0 && abs_level <= rl->max_level[last][run1]){
4894 av_log(s->avctx, AV_LOG_ERROR, "illegal 3. esc, esc 2 encoding possible\n");
4901 if (level>0) level= level * qmul + qadd;
4902 else level= level * qmul - qadd;
4904 if((unsigned)(level + 2048) > 4095){
4905 if(s->error_recognition > FF_ER_COMPLIANT){
4906 if(level > 2560 || level<-2560){
4907 av_log(s->avctx, AV_LOG_ERROR, "|level| overflow in 3. esc, qp=%d\n", s->qscale);
4911 level= level<0 ? -2048 : 2047;
4918 #if MIN_CACHE_BITS < 20
4919 LAST_SKIP_BITS(re, &s->gb, 2);
4920 UPDATE_CACHE(re, &s->gb);
4922 SKIP_BITS(re, &s->gb, 2);
4924 GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 1);
4925 i+= run + rl->max_run[run>>7][level/qmul] +1; //FIXME opt indexing
4926 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
4927 LAST_SKIP_BITS(re, &s->gb, 1);
4931 #if MIN_CACHE_BITS < 19
4932 LAST_SKIP_BITS(re, &s->gb, 1);
4933 UPDATE_CACHE(re, &s->gb);
4935 SKIP_BITS(re, &s->gb, 1);
4937 GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 1);
4939 level = level + rl->max_level[run>>7][(run-1)&63] * qmul;//FIXME opt indexing
4940 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
4941 LAST_SKIP_BITS(re, &s->gb, 1);
4946 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
4947 LAST_SKIP_BITS(re, &s->gb, 1);
4952 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
4956 block[scan_table[i]] = level;
4960 block[scan_table[i]] = level;
4962 CLOSE_READER(re, &s->gb);
4966 if(!s->use_intra_dc_vlc){
4967 block[0] = ff_mpeg4_pred_dc(s, n, block[0], &dc_pred_dir, 0);
4969 i -= i>>31; //if(i == -1) i=0;
4972 mpeg4_pred_ac(s, block, n, dc_pred_dir);
4974 i = 63; /* XXX: not optimal */
4977 s->block_last_index[n] = i;
4981 /* most is hardcoded. should extend to handle all h263 streams */
4982 int h263_decode_picture_header(MpegEncContext *s)
4984 int format, width, height, i;
4987 align_get_bits(&s->gb);
4989 startcode= get_bits(&s->gb, 22-8);
4991 for(i= s->gb.size_in_bits - get_bits_count(&s->gb); i>24; i-=8) {
4992 startcode = ((startcode << 8) | get_bits(&s->gb, 8)) & 0x003FFFFF;
4994 if(startcode == 0x20)
4998 if (startcode != 0x20) {
4999 av_log(s->avctx, AV_LOG_ERROR, "Bad picture start code\n");
5002 /* temporal reference */
5003 i = get_bits(&s->gb, 8); /* picture timestamp */
5004 if( (s->picture_number&~0xFF)+i < s->picture_number)
5006 s->current_picture_ptr->pts=
5007 s->picture_number= (s->picture_number&~0xFF) + i;
5009 /* PTYPE starts here */
5010 if (get_bits1(&s->gb) != 1) {
5012 av_log(s->avctx, AV_LOG_ERROR, "Bad marker\n");
5015 if (get_bits1(&s->gb) != 0) {
5016 av_log(s->avctx, AV_LOG_ERROR, "Bad H263 id\n");
5017 return -1; /* h263 id */
5019 skip_bits1(&s->gb); /* split screen off */
5020 skip_bits1(&s->gb); /* camera off */
5021 skip_bits1(&s->gb); /* freeze picture release off */
5023 format = get_bits(&s->gb, 3);
5028 7 extended PTYPE (PLUSPTYPE)
5031 if (format != 7 && format != 6) {
5034 width = h263_format[format][0];
5035 height = h263_format[format][1];
5039 s->pict_type = FF_I_TYPE + get_bits1(&s->gb);
5041 s->h263_long_vectors = get_bits1(&s->gb);
5043 if (get_bits1(&s->gb) != 0) {
5044 av_log(s->avctx, AV_LOG_ERROR, "H263 SAC not supported\n");
5045 return -1; /* SAC: off */
5047 s->obmc= get_bits1(&s->gb); /* Advanced prediction mode */
5048 s->unrestricted_mv = s->h263_long_vectors || s->obmc;
5050 if (get_bits1(&s->gb) != 0) {
5051 av_log(s->avctx, AV_LOG_ERROR, "H263 PB frame not supported\n");
5052 return -1; /* not PB frame */
5054 s->chroma_qscale= s->qscale = get_bits(&s->gb, 5);
5055 skip_bits1(&s->gb); /* Continuous Presence Multipoint mode: off */
5059 s->avctx->sample_aspect_ratio= (AVRational){12,11};
5060 s->avctx->time_base= (AVRational){1001, 30000};
5066 ufep = get_bits(&s->gb, 3); /* Update Full Extended PTYPE */
5068 /* ufep other than 0 and 1 are reserved */
5071 format = get_bits(&s->gb, 3);
5072 dprintf(s->avctx, "ufep=1, format: %d\n", format);
5073 s->custom_pcf= get_bits1(&s->gb);
5074 s->umvplus = get_bits1(&s->gb); /* Unrestricted Motion Vector */
5075 if (get_bits1(&s->gb) != 0) {
5076 av_log(s->avctx, AV_LOG_ERROR, "Syntax-based Arithmetic Coding (SAC) not supported\n");
5078 s->obmc= get_bits1(&s->gb); /* Advanced prediction mode */
5079 s->h263_aic = get_bits1(&s->gb); /* Advanced Intra Coding (AIC) */
5080 s->loop_filter= get_bits1(&s->gb);
5081 s->unrestricted_mv = s->umvplus || s->obmc || s->loop_filter;
5083 s->h263_slice_structured= get_bits1(&s->gb);
5084 if (get_bits1(&s->gb) != 0) {
5085 av_log(s->avctx, AV_LOG_ERROR, "Reference Picture Selection not supported\n");
5087 if (get_bits1(&s->gb) != 0) {
5088 av_log(s->avctx, AV_LOG_ERROR, "Independent Segment Decoding not supported\n");
5090 s->alt_inter_vlc= get_bits1(&s->gb);
5091 s->modified_quant= get_bits1(&s->gb);
5092 if(s->modified_quant)
5093 s->chroma_qscale_table= ff_h263_chroma_qscale_table;
5095 skip_bits(&s->gb, 1); /* Prevent start code emulation */
5097 skip_bits(&s->gb, 3); /* Reserved */
5098 } else if (ufep != 0) {
5099 av_log(s->avctx, AV_LOG_ERROR, "Bad UFEP type (%d)\n", ufep);
5104 s->pict_type = get_bits(&s->gb, 3);
5105 switch(s->pict_type){
5106 case 0: s->pict_type= FF_I_TYPE;break;
5107 case 1: s->pict_type= FF_P_TYPE;break;
5108 case 3: s->pict_type= FF_B_TYPE;break;
5109 case 7: s->pict_type= FF_I_TYPE;break; //ZYGO
5113 skip_bits(&s->gb, 2);
5114 s->no_rounding = get_bits1(&s->gb);
5115 skip_bits(&s->gb, 4);
5117 /* Get the picture dimensions */
5120 /* Custom Picture Format (CPFMT) */
5121 s->aspect_ratio_info = get_bits(&s->gb, 4);
5122 dprintf(s->avctx, "aspect: %d\n", s->aspect_ratio_info);
5127 3 - 10:11 (525-type 4:3)
5128 4 - 16:11 (CIF 16:9)
5129 5 - 40:33 (525-type 16:9)
5132 width = (get_bits(&s->gb, 9) + 1) * 4;
5134 height = get_bits(&s->gb, 9) * 4;
5135 dprintf(s->avctx, "\nH.263+ Custom picture: %dx%d\n",width,height);
5136 if (s->aspect_ratio_info == FF_ASPECT_EXTENDED) {
5137 /* aspected dimensions */
5138 s->avctx->sample_aspect_ratio.num= get_bits(&s->gb, 8);
5139 s->avctx->sample_aspect_ratio.den= get_bits(&s->gb, 8);
5141 s->avctx->sample_aspect_ratio= pixel_aspect[s->aspect_ratio_info];
5144 width = h263_format[format][0];
5145 height = h263_format[format][1];
5146 s->avctx->sample_aspect_ratio= (AVRational){12,11};
5148 if ((width == 0) || (height == 0))
5155 s->avctx->time_base.den= 1800000;
5156 s->avctx->time_base.num= 1000 + get_bits1(&s->gb);
5157 s->avctx->time_base.num*= get_bits(&s->gb, 7);
5158 if(s->avctx->time_base.num == 0){
5159 av_log(s, AV_LOG_ERROR, "zero framerate\n");
5162 gcd= av_gcd(s->avctx->time_base.den, s->avctx->time_base.num);
5163 s->avctx->time_base.den /= gcd;
5164 s->avctx->time_base.num /= gcd;
5165 // av_log(s->avctx, AV_LOG_DEBUG, "%d/%d\n", s->avctx->time_base.den, s->avctx->time_base.num);
5167 s->avctx->time_base= (AVRational){1001, 30000};
5172 skip_bits(&s->gb, 2); //extended Temporal reference
5177 if(get_bits1(&s->gb)==0) /* Unlimited Unrestricted Motion Vectors Indicator (UUI) */
5180 if(s->h263_slice_structured){
5181 if (get_bits1(&s->gb) != 0) {
5182 av_log(s->avctx, AV_LOG_ERROR, "rectangular slices not supported\n");
5184 if (get_bits1(&s->gb) != 0) {
5185 av_log(s->avctx, AV_LOG_ERROR, "unordered slices not supported\n");
5190 s->qscale = get_bits(&s->gb, 5);
5193 s->mb_width = (s->width + 15) / 16;
5194 s->mb_height = (s->height + 15) / 16;
5195 s->mb_num = s->mb_width * s->mb_height;
5198 while (get_bits1(&s->gb) != 0) {
5199 skip_bits(&s->gb, 8);
5202 if(s->h263_slice_structured){
5203 if (get_bits1(&s->gb) != 1) {
5204 av_log(s->avctx, AV_LOG_ERROR, "SEPB1 marker missing\n");
5208 ff_h263_decode_mba(s);
5210 if (get_bits1(&s->gb) != 1) {
5211 av_log(s->avctx, AV_LOG_ERROR, "SEPB2 marker missing\n");
5218 s->y_dc_scale_table=
5219 s->c_dc_scale_table= ff_aic_dc_scale_table;
5221 s->y_dc_scale_table=
5222 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
5225 if(s->avctx->debug&FF_DEBUG_PICT_INFO){
5229 if (s->pict_type == FF_I_TYPE && s->codec_tag == AV_RL32("ZYGO")){
5231 for(i=0; i<85; i++) av_log(s->avctx, AV_LOG_DEBUG, "%d", get_bits1(&s->gb));
5232 av_log(s->avctx, AV_LOG_DEBUG, "\n");
5233 for(i=0; i<13; i++){
5235 int v= get_bits(&s->gb, 8);
5236 v |= get_sbits(&s->gb, 8)<<8;
5237 av_log(s->avctx, AV_LOG_DEBUG, " %5d", v);
5239 av_log(s->avctx, AV_LOG_DEBUG, "\n");
5241 for(i=0; i<50; i++) av_log(s->avctx, AV_LOG_DEBUG, "%d", get_bits1(&s->gb));
5248 static void mpeg4_decode_sprite_trajectory(MpegEncContext * s, GetBitContext *gb)
5251 int a= 2<<s->sprite_warping_accuracy;
5252 int rho= 3-s->sprite_warping_accuracy;
5254 const int vop_ref[4][2]= {{0,0}, {s->width,0}, {0, s->height}, {s->width, s->height}}; // only true for rectangle shapes
5255 int d[4][2]={{0,0}, {0,0}, {0,0}, {0,0}};
5256 int sprite_ref[4][2];
5257 int virtual_ref[2][2];
5259 int alpha=0, beta=0;
5264 for(i=0; i<s->num_sprite_warping_points; i++){
5268 length= get_vlc2(gb, sprite_trajectory.table, SPRITE_TRAJ_VLC_BITS, 3);
5270 x= get_xbits(gb, length);
5272 if(!(s->divx_version==500 && s->divx_build==413)) skip_bits1(gb); /* marker bit */
5274 length= get_vlc2(gb, sprite_trajectory.table, SPRITE_TRAJ_VLC_BITS, 3);
5276 y=get_xbits(gb, length);
5278 skip_bits1(gb); /* marker bit */
5279 //printf("%d %d %d %d\n", x, y, i, s->sprite_warping_accuracy);
5284 while((1<<alpha)<w) alpha++;
5285 while((1<<beta )<h) beta++; // there seems to be a typo in the mpeg4 std for the definition of w' and h'
5289 // Note, the 4th point isn't used for GMC
5290 if(s->divx_version==500 && s->divx_build==413){
5291 sprite_ref[0][0]= a*vop_ref[0][0] + d[0][0];
5292 sprite_ref[0][1]= a*vop_ref[0][1] + d[0][1];
5293 sprite_ref[1][0]= a*vop_ref[1][0] + d[0][0] + d[1][0];
5294 sprite_ref[1][1]= a*vop_ref[1][1] + d[0][1] + d[1][1];
5295 sprite_ref[2][0]= a*vop_ref[2][0] + d[0][0] + d[2][0];
5296 sprite_ref[2][1]= a*vop_ref[2][1] + d[0][1] + d[2][1];
5298 sprite_ref[0][0]= (a>>1)*(2*vop_ref[0][0] + d[0][0]);
5299 sprite_ref[0][1]= (a>>1)*(2*vop_ref[0][1] + d[0][1]);
5300 sprite_ref[1][0]= (a>>1)*(2*vop_ref[1][0] + d[0][0] + d[1][0]);
5301 sprite_ref[1][1]= (a>>1)*(2*vop_ref[1][1] + d[0][1] + d[1][1]);
5302 sprite_ref[2][0]= (a>>1)*(2*vop_ref[2][0] + d[0][0] + d[2][0]);
5303 sprite_ref[2][1]= (a>>1)*(2*vop_ref[2][1] + d[0][1] + d[2][1]);
5305 /* sprite_ref[3][0]= (a>>1)*(2*vop_ref[3][0] + d[0][0] + d[1][0] + d[2][0] + d[3][0]);
5306 sprite_ref[3][1]= (a>>1)*(2*vop_ref[3][1] + d[0][1] + d[1][1] + d[2][1] + d[3][1]); */
5308 // this is mostly identical to the mpeg4 std (and is totally unreadable because of that ...)
5309 // perhaps it should be reordered to be more readable ...
5310 // the idea behind this virtual_ref mess is to be able to use shifts later per pixel instead of divides
5311 // so the distance between points is converted from w&h based to w2&h2 based which are of the 2^x form
5312 virtual_ref[0][0]= 16*(vop_ref[0][0] + w2)
5313 + ROUNDED_DIV(((w - w2)*(r*sprite_ref[0][0] - 16*vop_ref[0][0]) + w2*(r*sprite_ref[1][0] - 16*vop_ref[1][0])),w);
5314 virtual_ref[0][1]= 16*vop_ref[0][1]
5315 + ROUNDED_DIV(((w - w2)*(r*sprite_ref[0][1] - 16*vop_ref[0][1]) + w2*(r*sprite_ref[1][1] - 16*vop_ref[1][1])),w);
5316 virtual_ref[1][0]= 16*vop_ref[0][0]
5317 + ROUNDED_DIV(((h - h2)*(r*sprite_ref[0][0] - 16*vop_ref[0][0]) + h2*(r*sprite_ref[2][0] - 16*vop_ref[2][0])),h);
5318 virtual_ref[1][1]= 16*(vop_ref[0][1] + h2)
5319 + ROUNDED_DIV(((h - h2)*(r*sprite_ref[0][1] - 16*vop_ref[0][1]) + h2*(r*sprite_ref[2][1] - 16*vop_ref[2][1])),h);
5321 switch(s->num_sprite_warping_points)
5324 s->sprite_offset[0][0]= 0;
5325 s->sprite_offset[0][1]= 0;
5326 s->sprite_offset[1][0]= 0;
5327 s->sprite_offset[1][1]= 0;
5328 s->sprite_delta[0][0]= a;
5329 s->sprite_delta[0][1]= 0;
5330 s->sprite_delta[1][0]= 0;
5331 s->sprite_delta[1][1]= a;
5332 s->sprite_shift[0]= 0;
5333 s->sprite_shift[1]= 0;
5336 s->sprite_offset[0][0]= sprite_ref[0][0] - a*vop_ref[0][0];
5337 s->sprite_offset[0][1]= sprite_ref[0][1] - a*vop_ref[0][1];
5338 s->sprite_offset[1][0]= ((sprite_ref[0][0]>>1)|(sprite_ref[0][0]&1)) - a*(vop_ref[0][0]/2);
5339 s->sprite_offset[1][1]= ((sprite_ref[0][1]>>1)|(sprite_ref[0][1]&1)) - a*(vop_ref[0][1]/2);
5340 s->sprite_delta[0][0]= a;
5341 s->sprite_delta[0][1]= 0;
5342 s->sprite_delta[1][0]= 0;
5343 s->sprite_delta[1][1]= a;
5344 s->sprite_shift[0]= 0;
5345 s->sprite_shift[1]= 0;
5348 s->sprite_offset[0][0]= (sprite_ref[0][0]<<(alpha+rho))
5349 + (-r*sprite_ref[0][0] + virtual_ref[0][0])*(-vop_ref[0][0])
5350 + ( r*sprite_ref[0][1] - virtual_ref[0][1])*(-vop_ref[0][1])
5351 + (1<<(alpha+rho-1));
5352 s->sprite_offset[0][1]= (sprite_ref[0][1]<<(alpha+rho))
5353 + (-r*sprite_ref[0][1] + virtual_ref[0][1])*(-vop_ref[0][0])
5354 + (-r*sprite_ref[0][0] + virtual_ref[0][0])*(-vop_ref[0][1])
5355 + (1<<(alpha+rho-1));
5356 s->sprite_offset[1][0]= ( (-r*sprite_ref[0][0] + virtual_ref[0][0])*(-2*vop_ref[0][0] + 1)
5357 +( r*sprite_ref[0][1] - virtual_ref[0][1])*(-2*vop_ref[0][1] + 1)
5358 +2*w2*r*sprite_ref[0][0]
5360 + (1<<(alpha+rho+1)));
5361 s->sprite_offset[1][1]= ( (-r*sprite_ref[0][1] + virtual_ref[0][1])*(-2*vop_ref[0][0] + 1)
5362 +(-r*sprite_ref[0][0] + virtual_ref[0][0])*(-2*vop_ref[0][1] + 1)
5363 +2*w2*r*sprite_ref[0][1]
5365 + (1<<(alpha+rho+1)));
5366 s->sprite_delta[0][0]= (-r*sprite_ref[0][0] + virtual_ref[0][0]);
5367 s->sprite_delta[0][1]= (+r*sprite_ref[0][1] - virtual_ref[0][1]);
5368 s->sprite_delta[1][0]= (-r*sprite_ref[0][1] + virtual_ref[0][1]);
5369 s->sprite_delta[1][1]= (-r*sprite_ref[0][0] + virtual_ref[0][0]);
5371 s->sprite_shift[0]= alpha+rho;
5372 s->sprite_shift[1]= alpha+rho+2;
5375 min_ab= FFMIN(alpha, beta);
5378 s->sprite_offset[0][0]= (sprite_ref[0][0]<<(alpha+beta+rho-min_ab))
5379 + (-r*sprite_ref[0][0] + virtual_ref[0][0])*h3*(-vop_ref[0][0])
5380 + (-r*sprite_ref[0][0] + virtual_ref[1][0])*w3*(-vop_ref[0][1])
5381 + (1<<(alpha+beta+rho-min_ab-1));
5382 s->sprite_offset[0][1]= (sprite_ref[0][1]<<(alpha+beta+rho-min_ab))
5383 + (-r*sprite_ref[0][1] + virtual_ref[0][1])*h3*(-vop_ref[0][0])
5384 + (-r*sprite_ref[0][1] + virtual_ref[1][1])*w3*(-vop_ref[0][1])
5385 + (1<<(alpha+beta+rho-min_ab-1));
5386 s->sprite_offset[1][0]= (-r*sprite_ref[0][0] + virtual_ref[0][0])*h3*(-2*vop_ref[0][0] + 1)
5387 + (-r*sprite_ref[0][0] + virtual_ref[1][0])*w3*(-2*vop_ref[0][1] + 1)
5388 + 2*w2*h3*r*sprite_ref[0][0]
5390 + (1<<(alpha+beta+rho-min_ab+1));
5391 s->sprite_offset[1][1]= (-r*sprite_ref[0][1] + virtual_ref[0][1])*h3*(-2*vop_ref[0][0] + 1)
5392 + (-r*sprite_ref[0][1] + virtual_ref[1][1])*w3*(-2*vop_ref[0][1] + 1)
5393 + 2*w2*h3*r*sprite_ref[0][1]
5395 + (1<<(alpha+beta+rho-min_ab+1));
5396 s->sprite_delta[0][0]= (-r*sprite_ref[0][0] + virtual_ref[0][0])*h3;
5397 s->sprite_delta[0][1]= (-r*sprite_ref[0][0] + virtual_ref[1][0])*w3;
5398 s->sprite_delta[1][0]= (-r*sprite_ref[0][1] + virtual_ref[0][1])*h3;
5399 s->sprite_delta[1][1]= (-r*sprite_ref[0][1] + virtual_ref[1][1])*w3;
5401 s->sprite_shift[0]= alpha + beta + rho - min_ab;
5402 s->sprite_shift[1]= alpha + beta + rho - min_ab + 2;
5405 /* try to simplify the situation */
5406 if( s->sprite_delta[0][0] == a<<s->sprite_shift[0]
5407 && s->sprite_delta[0][1] == 0
5408 && s->sprite_delta[1][0] == 0
5409 && s->sprite_delta[1][1] == a<<s->sprite_shift[0])
5411 s->sprite_offset[0][0]>>=s->sprite_shift[0];
5412 s->sprite_offset[0][1]>>=s->sprite_shift[0];
5413 s->sprite_offset[1][0]>>=s->sprite_shift[1];
5414 s->sprite_offset[1][1]>>=s->sprite_shift[1];
5415 s->sprite_delta[0][0]= a;
5416 s->sprite_delta[0][1]= 0;
5417 s->sprite_delta[1][0]= 0;
5418 s->sprite_delta[1][1]= a;
5419 s->sprite_shift[0]= 0;
5420 s->sprite_shift[1]= 0;
5421 s->real_sprite_warping_points=1;
5424 int shift_y= 16 - s->sprite_shift[0];
5425 int shift_c= 16 - s->sprite_shift[1];
5426 //printf("shifts %d %d\n", shift_y, shift_c);
5428 s->sprite_offset[0][i]<<= shift_y;
5429 s->sprite_offset[1][i]<<= shift_c;
5430 s->sprite_delta[0][i]<<= shift_y;
5431 s->sprite_delta[1][i]<<= shift_y;
5432 s->sprite_shift[i]= 16;
5434 s->real_sprite_warping_points= s->num_sprite_warping_points;
5437 printf("vop:%d:%d %d:%d %d:%d, sprite:%d:%d %d:%d %d:%d, virtual: %d:%d %d:%d\n",
5438 vop_ref[0][0], vop_ref[0][1],
5439 vop_ref[1][0], vop_ref[1][1],
5440 vop_ref[2][0], vop_ref[2][1],
5441 sprite_ref[0][0], sprite_ref[0][1],
5442 sprite_ref[1][0], sprite_ref[1][1],
5443 sprite_ref[2][0], sprite_ref[2][1],
5444 virtual_ref[0][0], virtual_ref[0][1],
5445 virtual_ref[1][0], virtual_ref[1][1]
5448 printf("offset: %d:%d , delta: %d %d %d %d, shift %d\n",
5449 s->sprite_offset[0][0], s->sprite_offset[0][1],
5450 s->sprite_delta[0][0], s->sprite_delta[0][1],
5451 s->sprite_delta[1][0], s->sprite_delta[1][1],
5457 static int mpeg4_decode_gop_header(MpegEncContext * s, GetBitContext *gb){
5458 int hours, minutes, seconds;
5460 hours= get_bits(gb, 5);
5461 minutes= get_bits(gb, 6);
5463 seconds= get_bits(gb, 6);
5465 s->time_base= seconds + 60*(minutes + 60*hours);
5473 static int decode_vol_header(MpegEncContext *s, GetBitContext *gb){
5474 int width, height, vo_ver_id;
5477 skip_bits(gb, 1); /* random access */
5478 s->vo_type= get_bits(gb, 8);
5479 if (get_bits1(gb) != 0) { /* is_ol_id */
5480 vo_ver_id = get_bits(gb, 4); /* vo_ver_id */
5481 skip_bits(gb, 3); /* vo_priority */
5485 //printf("vo type:%d\n",s->vo_type);
5486 s->aspect_ratio_info= get_bits(gb, 4);
5487 if(s->aspect_ratio_info == FF_ASPECT_EXTENDED){
5488 s->avctx->sample_aspect_ratio.num= get_bits(gb, 8); // par_width
5489 s->avctx->sample_aspect_ratio.den= get_bits(gb, 8); // par_height
5491 s->avctx->sample_aspect_ratio= pixel_aspect[s->aspect_ratio_info];
5494 if ((s->vol_control_parameters=get_bits1(gb))) { /* vol control parameter */
5495 int chroma_format= get_bits(gb, 2);
5496 if(chroma_format!=CHROMA_420){
5497 av_log(s->avctx, AV_LOG_ERROR, "illegal chroma format\n");
5499 s->low_delay= get_bits1(gb);
5500 if(get_bits1(gb)){ /* vbv parameters */
5501 get_bits(gb, 15); /* first_half_bitrate */
5502 skip_bits1(gb); /* marker */
5503 get_bits(gb, 15); /* latter_half_bitrate */
5504 skip_bits1(gb); /* marker */
5505 get_bits(gb, 15); /* first_half_vbv_buffer_size */
5506 skip_bits1(gb); /* marker */
5507 get_bits(gb, 3); /* latter_half_vbv_buffer_size */
5508 get_bits(gb, 11); /* first_half_vbv_occupancy */
5509 skip_bits1(gb); /* marker */
5510 get_bits(gb, 15); /* latter_half_vbv_occupancy */
5511 skip_bits1(gb); /* marker */
5514 // set low delay flag only once the smartest? low delay detection won't be overriden
5515 if(s->picture_number==0)
5519 s->shape = get_bits(gb, 2); /* vol shape */
5520 if(s->shape != RECT_SHAPE) av_log(s->avctx, AV_LOG_ERROR, "only rectangular vol supported\n");
5521 if(s->shape == GRAY_SHAPE && vo_ver_id != 1){
5522 av_log(s->avctx, AV_LOG_ERROR, "Gray shape not supported\n");
5523 skip_bits(gb, 4); //video_object_layer_shape_extension
5526 check_marker(gb, "before time_increment_resolution");
5528 s->avctx->time_base.den = get_bits(gb, 16);
5529 if(!s->avctx->time_base.den){
5530 av_log(s->avctx, AV_LOG_ERROR, "time_base.den==0\n");
5534 s->time_increment_bits = av_log2(s->avctx->time_base.den - 1) + 1;
5535 if (s->time_increment_bits < 1)
5536 s->time_increment_bits = 1;
5538 check_marker(gb, "before fixed_vop_rate");
5540 if (get_bits1(gb) != 0) { /* fixed_vop_rate */
5541 s->avctx->time_base.num = get_bits(gb, s->time_increment_bits);
5543 s->avctx->time_base.num = 1;
5547 if (s->shape != BIN_ONLY_SHAPE) {
5548 if (s->shape == RECT_SHAPE) {
5549 skip_bits1(gb); /* marker */
5550 width = get_bits(gb, 13);
5551 skip_bits1(gb); /* marker */
5552 height = get_bits(gb, 13);
5553 skip_bits1(gb); /* marker */
5554 if(width && height && !(s->width && s->codec_tag == AV_RL32("MP4S"))){ /* they should be non zero but who knows ... */
5557 // printf("width/height: %d %d\n", width, height);
5561 s->progressive_sequence=
5562 s->progressive_frame= get_bits1(gb)^1;
5563 s->interlaced_dct=0;
5564 if(!get_bits1(gb) && (s->avctx->debug & FF_DEBUG_PICT_INFO))
5565 av_log(s->avctx, AV_LOG_INFO, "MPEG4 OBMC not supported (very likely buggy encoder)\n"); /* OBMC Disable */
5566 if (vo_ver_id == 1) {
5567 s->vol_sprite_usage = get_bits1(gb); /* vol_sprite_usage */
5569 s->vol_sprite_usage = get_bits(gb, 2); /* vol_sprite_usage */
5571 if(s->vol_sprite_usage==STATIC_SPRITE) av_log(s->avctx, AV_LOG_ERROR, "Static Sprites not supported\n");
5572 if(s->vol_sprite_usage==STATIC_SPRITE || s->vol_sprite_usage==GMC_SPRITE){
5573 if(s->vol_sprite_usage==STATIC_SPRITE){
5574 s->sprite_width = get_bits(gb, 13);
5575 skip_bits1(gb); /* marker */
5576 s->sprite_height= get_bits(gb, 13);
5577 skip_bits1(gb); /* marker */
5578 s->sprite_left = get_bits(gb, 13);
5579 skip_bits1(gb); /* marker */
5580 s->sprite_top = get_bits(gb, 13);
5581 skip_bits1(gb); /* marker */
5583 s->num_sprite_warping_points= get_bits(gb, 6);
5584 if(s->num_sprite_warping_points > 3){
5585 av_log(s->avctx, AV_LOG_ERROR, "%d sprite_warping_points\n", s->num_sprite_warping_points);
5586 s->num_sprite_warping_points= 0;
5589 s->sprite_warping_accuracy = get_bits(gb, 2);
5590 s->sprite_brightness_change= get_bits1(gb);
5591 if(s->vol_sprite_usage==STATIC_SPRITE)
5592 s->low_latency_sprite= get_bits1(gb);
5594 // FIXME sadct disable bit if verid!=1 && shape not rect
5596 if (get_bits1(gb) == 1) { /* not_8_bit */
5597 s->quant_precision = get_bits(gb, 4); /* quant_precision */
5598 if(get_bits(gb, 4)!=8) av_log(s->avctx, AV_LOG_ERROR, "N-bit not supported\n"); /* bits_per_pixel */
5599 if(s->quant_precision!=5) av_log(s->avctx, AV_LOG_ERROR, "quant precision %d\n", s->quant_precision);
5601 s->quant_precision = 5;
5604 // FIXME a bunch of grayscale shape things
5606 if((s->mpeg_quant=get_bits1(gb))){ /* vol_quant_type */
5609 /* load default matrixes */
5610 for(i=0; i<64; i++){
5611 int j= s->dsp.idct_permutation[i];
5612 v= ff_mpeg4_default_intra_matrix[i];
5613 s->intra_matrix[j]= v;
5614 s->chroma_intra_matrix[j]= v;
5616 v= ff_mpeg4_default_non_intra_matrix[i];
5617 s->inter_matrix[j]= v;
5618 s->chroma_inter_matrix[j]= v;
5621 /* load custom intra matrix */
5624 for(i=0; i<64; i++){
5630 j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
5631 s->intra_matrix[j]= v;
5632 s->chroma_intra_matrix[j]= v;
5635 /* replicate last value */
5637 int j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
5638 s->intra_matrix[j]= last;
5639 s->chroma_intra_matrix[j]= last;
5643 /* load custom non intra matrix */
5646 for(i=0; i<64; i++){
5652 j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
5653 s->inter_matrix[j]= v;
5654 s->chroma_inter_matrix[j]= v;
5657 /* replicate last value */
5659 int j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
5660 s->inter_matrix[j]= last;
5661 s->chroma_inter_matrix[j]= last;
5665 // FIXME a bunch of grayscale shape things
5669 s->quarter_sample= get_bits1(gb);
5670 else s->quarter_sample=0;
5673 int pos= get_bits_count(gb);
5674 int estimation_method= get_bits(gb, 2);
5675 if(estimation_method<2){
5677 s->cplx_estimation_trash_i += 8*get_bits1(gb); //opaque
5678 s->cplx_estimation_trash_i += 8*get_bits1(gb); //transparent
5679 s->cplx_estimation_trash_i += 8*get_bits1(gb); //intra_cae
5680 s->cplx_estimation_trash_i += 8*get_bits1(gb); //inter_cae
5681 s->cplx_estimation_trash_i += 8*get_bits1(gb); //no_update
5682 s->cplx_estimation_trash_i += 8*get_bits1(gb); //upampling
5685 s->cplx_estimation_trash_i += 8*get_bits1(gb); //intra_blocks
5686 s->cplx_estimation_trash_p += 8*get_bits1(gb); //inter_blocks
5687 s->cplx_estimation_trash_p += 8*get_bits1(gb); //inter4v_blocks
5688 s->cplx_estimation_trash_i += 8*get_bits1(gb); //not coded blocks
5690 if(!check_marker(gb, "in complexity estimation part 1")){
5691 skip_bits_long(gb, pos - get_bits_count(gb));
5695 s->cplx_estimation_trash_i += 8*get_bits1(gb); //dct_coeffs
5696 s->cplx_estimation_trash_i += 8*get_bits1(gb); //dct_lines
5697 s->cplx_estimation_trash_i += 8*get_bits1(gb); //vlc_syms
5698 s->cplx_estimation_trash_i += 4*get_bits1(gb); //vlc_bits
5701 s->cplx_estimation_trash_p += 8*get_bits1(gb); //apm
5702 s->cplx_estimation_trash_p += 8*get_bits1(gb); //npm
5703 s->cplx_estimation_trash_b += 8*get_bits1(gb); //interpolate_mc_q
5704 s->cplx_estimation_trash_p += 8*get_bits1(gb); //forwback_mc_q
5705 s->cplx_estimation_trash_p += 8*get_bits1(gb); //halfpel2
5706 s->cplx_estimation_trash_p += 8*get_bits1(gb); //halfpel4
5708 if(!check_marker(gb, "in complexity estimation part 2")){
5709 skip_bits_long(gb, pos - get_bits_count(gb));
5712 if(estimation_method==1){
5713 s->cplx_estimation_trash_i += 8*get_bits1(gb); //sadct
5714 s->cplx_estimation_trash_p += 8*get_bits1(gb); //qpel
5717 av_log(s->avctx, AV_LOG_ERROR, "Invalid Complexity estimation method %d\n", estimation_method);
5720 s->cplx_estimation_trash_i=
5721 s->cplx_estimation_trash_p=
5722 s->cplx_estimation_trash_b= 0;
5725 s->resync_marker= !get_bits1(gb); /* resync_marker_disabled */
5727 s->data_partitioning= get_bits1(gb);
5728 if(s->data_partitioning){
5729 s->rvlc= get_bits1(gb);
5732 if(vo_ver_id != 1) {
5733 s->new_pred= get_bits1(gb);
5735 av_log(s->avctx, AV_LOG_ERROR, "new pred not supported\n");
5736 skip_bits(gb, 2); /* requested upstream message type */
5737 skip_bits1(gb); /* newpred segment type */
5739 s->reduced_res_vop= get_bits1(gb);
5740 if(s->reduced_res_vop) av_log(s->avctx, AV_LOG_ERROR, "reduced resolution VOP not supported\n");
5744 s->reduced_res_vop= 0;
5747 s->scalability= get_bits1(gb);
5749 if (s->scalability) {
5750 GetBitContext bak= *gb;
5752 int ref_layer_sampling_dir;
5753 int h_sampling_factor_n;
5754 int h_sampling_factor_m;
5755 int v_sampling_factor_n;
5756 int v_sampling_factor_m;
5758 s->hierachy_type= get_bits1(gb);
5759 ref_layer_id= get_bits(gb, 4);
5760 ref_layer_sampling_dir= get_bits1(gb);
5761 h_sampling_factor_n= get_bits(gb, 5);
5762 h_sampling_factor_m= get_bits(gb, 5);
5763 v_sampling_factor_n= get_bits(gb, 5);
5764 v_sampling_factor_m= get_bits(gb, 5);
5765 s->enhancement_type= get_bits1(gb);
5767 if( h_sampling_factor_n==0 || h_sampling_factor_m==0
5768 || v_sampling_factor_n==0 || v_sampling_factor_m==0){
5770 // fprintf(stderr, "illegal scalability header (VERY broken encoder), trying to workaround\n");
5775 av_log(s->avctx, AV_LOG_ERROR, "scalability not supported\n");
5777 // bin shape stuff FIXME
5784 * decodes the user data stuff in the header.
5785 * Also initializes divx/xvid/lavc_version/build.
5787 static int decode_user_data(MpegEncContext *s, GetBitContext *gb){
5791 int ver = 0, build = 0, ver2 = 0, ver3 = 0;
5794 for(i=0; i<255 && get_bits_count(gb) < gb->size_in_bits; i++){
5795 if(show_bits(gb, 23) == 0) break;
5796 buf[i]= get_bits(gb, 8);
5800 /* divx detection */
5801 e=sscanf(buf, "DivX%dBuild%d%c", &ver, &build, &last);
5803 e=sscanf(buf, "DivX%db%d%c", &ver, &build, &last);
5805 s->divx_version= ver;
5806 s->divx_build= build;
5807 s->divx_packed= e==3 && last=='p';
5808 if(s->divx_packed && !s->showed_packed_warning) {
5809 av_log(s->avctx, AV_LOG_WARNING, "Invalid and inefficient vfw-avi packed B frames detected\n");
5810 s->showed_packed_warning=1;
5814 /* ffmpeg detection */
5815 e=sscanf(buf, "FFmpe%*[^b]b%d", &build)+3;
5817 e=sscanf(buf, "FFmpeg v%d.%d.%d / libavcodec build: %d", &ver, &ver2, &ver3, &build);
5819 e=sscanf(buf, "Lavc%d.%d.%d", &ver, &ver2, &ver3)+1;
5821 build= (ver<<16) + (ver2<<8) + ver3;
5824 if(strcmp(buf, "ffmpeg")==0){
5825 s->lavc_build= 4600;
5829 s->lavc_build= build;
5832 /* Xvid detection */
5833 e=sscanf(buf, "XviD%d", &build);
5835 s->xvid_build= build;
5838 //printf("User Data: %s\n", buf);
5842 static int decode_vop_header(MpegEncContext *s, GetBitContext *gb){
5843 int time_incr, time_increment;
5845 s->pict_type = get_bits(gb, 2) + FF_I_TYPE; /* pict type: I = 0 , P = 1 */
5846 if(s->pict_type==FF_B_TYPE && s->low_delay && s->vol_control_parameters==0 && !(s->flags & CODEC_FLAG_LOW_DELAY)){
5847 av_log(s->avctx, AV_LOG_ERROR, "low_delay flag incorrectly, clearing it\n");
5851 s->partitioned_frame= s->data_partitioning && s->pict_type!=FF_B_TYPE;
5852 if(s->partitioned_frame)
5853 s->decode_mb= mpeg4_decode_partitioned_mb;
5855 s->decode_mb= ff_mpeg4_decode_mb;
5858 while (get_bits1(gb) != 0)
5861 check_marker(gb, "before time_increment");
5863 if(s->time_increment_bits==0 || !(show_bits(gb, s->time_increment_bits+1)&1)){
5864 av_log(s->avctx, AV_LOG_ERROR, "hmm, seems the headers are not complete, trying to guess time_increment_bits\n");
5866 for(s->time_increment_bits=1 ;s->time_increment_bits<16; s->time_increment_bits++){
5867 if(show_bits(gb, s->time_increment_bits+1)&1) break;
5870 av_log(s->avctx, AV_LOG_ERROR, "my guess is %d bits ;)\n",s->time_increment_bits);
5873 if(IS_3IV1) time_increment= get_bits1(gb); //FIXME investigate further
5874 else time_increment= get_bits(gb, s->time_increment_bits);
5876 // printf("%d %X\n", s->time_increment_bits, time_increment);
5877 //av_log(s->avctx, AV_LOG_DEBUG, " type:%d modulo_time_base:%d increment:%d t_frame %d\n", s->pict_type, time_incr, time_increment, s->t_frame);
5878 if(s->pict_type!=FF_B_TYPE){
5879 s->last_time_base= s->time_base;
5880 s->time_base+= time_incr;
5881 s->time= s->time_base*s->avctx->time_base.den + time_increment;
5882 if(s->workaround_bugs&FF_BUG_UMP4){
5883 if(s->time < s->last_non_b_time){
5884 // fprintf(stderr, "header is not mpeg4 compatible, broken encoder, trying to workaround\n");
5886 s->time+= s->avctx->time_base.den;
5889 s->pp_time= s->time - s->last_non_b_time;
5890 s->last_non_b_time= s->time;
5892 s->time= (s->last_time_base + time_incr)*s->avctx->time_base.den + time_increment;
5893 s->pb_time= s->pp_time - (s->last_non_b_time - s->time);
5894 if(s->pp_time <=s->pb_time || s->pp_time <= s->pp_time - s->pb_time || s->pp_time<=0){
5895 // printf("messed up order, maybe after seeking? skipping current b frame\n");
5896 return FRAME_SKIPPED;
5898 ff_mpeg4_init_direct_mv(s);
5900 if(s->t_frame==0) s->t_frame= s->pb_time;
5901 if(s->t_frame==0) s->t_frame=1; // 1/0 protection
5902 s->pp_field_time= ( ROUNDED_DIV(s->last_non_b_time, s->t_frame)
5903 - ROUNDED_DIV(s->last_non_b_time - s->pp_time, s->t_frame))*2;
5904 s->pb_field_time= ( ROUNDED_DIV(s->time, s->t_frame)
5905 - ROUNDED_DIV(s->last_non_b_time - s->pp_time, s->t_frame))*2;
5906 if(!s->progressive_sequence){
5907 if(s->pp_field_time <= s->pb_field_time || s->pb_field_time <= 1)
5908 return FRAME_SKIPPED;
5911 //av_log(s->avctx, AV_LOG_DEBUG, "last nonb %"PRId64" last_base %d time %"PRId64" pp %d pb %d t %d ppf %d pbf %d\n", s->last_non_b_time, s->last_time_base, s->time, s->pp_time, s->pb_time, s->t_frame, s->pp_field_time, s->pb_field_time);
5913 if(s->avctx->time_base.num)
5914 s->current_picture_ptr->pts= (s->time + s->avctx->time_base.num/2) / s->avctx->time_base.num;
5916 s->current_picture_ptr->pts= AV_NOPTS_VALUE;
5917 if(s->avctx->debug&FF_DEBUG_PTS)
5918 av_log(s->avctx, AV_LOG_DEBUG, "MPEG4 PTS: %"PRId64"\n", s->current_picture_ptr->pts);
5920 check_marker(gb, "before vop_coded");
5923 if (get_bits1(gb) != 1){
5924 if(s->avctx->debug&FF_DEBUG_PICT_INFO)
5925 av_log(s->avctx, AV_LOG_ERROR, "vop not coded\n");
5926 return FRAME_SKIPPED;
5928 //printf("time %d %d %d || %"PRId64" %"PRId64" %"PRId64"\n", s->time_increment_bits, s->avctx->time_base.den, s->time_base,
5929 //s->time, s->last_non_b_time, s->last_non_b_time - s->pp_time);
5930 if (s->shape != BIN_ONLY_SHAPE && ( s->pict_type == FF_P_TYPE
5931 || (s->pict_type == FF_S_TYPE && s->vol_sprite_usage==GMC_SPRITE))) {
5932 /* rounding type for motion estimation */
5933 s->no_rounding = get_bits1(gb);
5937 //FIXME reduced res stuff
5939 if (s->shape != RECT_SHAPE) {
5940 if (s->vol_sprite_usage != 1 || s->pict_type != FF_I_TYPE) {
5941 int width, height, hor_spat_ref, ver_spat_ref;
5943 width = get_bits(gb, 13);
5944 skip_bits1(gb); /* marker */
5945 height = get_bits(gb, 13);
5946 skip_bits1(gb); /* marker */
5947 hor_spat_ref = get_bits(gb, 13); /* hor_spat_ref */
5948 skip_bits1(gb); /* marker */
5949 ver_spat_ref = get_bits(gb, 13); /* ver_spat_ref */
5951 skip_bits1(gb); /* change_CR_disable */
5953 if (get_bits1(gb) != 0) {
5954 skip_bits(gb, 8); /* constant_alpha_value */
5957 //FIXME complexity estimation stuff
5959 if (s->shape != BIN_ONLY_SHAPE) {
5960 skip_bits_long(gb, s->cplx_estimation_trash_i);
5961 if(s->pict_type != FF_I_TYPE)
5962 skip_bits_long(gb, s->cplx_estimation_trash_p);
5963 if(s->pict_type == FF_B_TYPE)
5964 skip_bits_long(gb, s->cplx_estimation_trash_b);
5966 s->intra_dc_threshold= mpeg4_dc_threshold[ get_bits(gb, 3) ];
5967 if(!s->progressive_sequence){
5968 s->top_field_first= get_bits1(gb);
5969 s->alternate_scan= get_bits1(gb);
5971 s->alternate_scan= 0;
5974 if(s->alternate_scan){
5975 ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable , ff_alternate_vertical_scan);
5976 ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable , ff_alternate_vertical_scan);
5977 ff_init_scantable(s->dsp.idct_permutation, &s->intra_h_scantable, ff_alternate_vertical_scan);
5978 ff_init_scantable(s->dsp.idct_permutation, &s->intra_v_scantable, ff_alternate_vertical_scan);
5980 ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable , ff_zigzag_direct);
5981 ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable , ff_zigzag_direct);
5982 ff_init_scantable(s->dsp.idct_permutation, &s->intra_h_scantable, ff_alternate_horizontal_scan);
5983 ff_init_scantable(s->dsp.idct_permutation, &s->intra_v_scantable, ff_alternate_vertical_scan);
5986 if(s->pict_type == FF_S_TYPE && (s->vol_sprite_usage==STATIC_SPRITE || s->vol_sprite_usage==GMC_SPRITE)){
5987 mpeg4_decode_sprite_trajectory(s, gb);
5988 if(s->sprite_brightness_change) av_log(s->avctx, AV_LOG_ERROR, "sprite_brightness_change not supported\n");
5989 if(s->vol_sprite_usage==STATIC_SPRITE) av_log(s->avctx, AV_LOG_ERROR, "static sprite not supported\n");
5992 if (s->shape != BIN_ONLY_SHAPE) {
5993 s->chroma_qscale= s->qscale = get_bits(gb, s->quant_precision);
5995 av_log(s->avctx, AV_LOG_ERROR, "Error, header damaged or not MPEG4 header (qscale=0)\n");
5996 return -1; // makes no sense to continue, as there is nothing left from the image then
5999 if (s->pict_type != FF_I_TYPE) {
6000 s->f_code = get_bits(gb, 3); /* fcode_for */
6002 av_log(s->avctx, AV_LOG_ERROR, "Error, header damaged or not MPEG4 header (f_code=0)\n");
6003 return -1; // makes no sense to continue, as the MV decoding will break very quickly
6008 if (s->pict_type == FF_B_TYPE) {
6009 s->b_code = get_bits(gb, 3);
6013 if(s->avctx->debug&FF_DEBUG_PICT_INFO){
6014 av_log(s->avctx, AV_LOG_DEBUG, "qp:%d fc:%d,%d %s size:%d pro:%d alt:%d top:%d %spel part:%d resync:%d w:%d a:%d rnd:%d vot:%d%s dc:%d ce:%d/%d/%d\n",
6015 s->qscale, s->f_code, s->b_code,
6016 s->pict_type == FF_I_TYPE ? "I" : (s->pict_type == FF_P_TYPE ? "P" : (s->pict_type == FF_B_TYPE ? "B" : "S")),
6017 gb->size_in_bits,s->progressive_sequence, s->alternate_scan, s->top_field_first,
6018 s->quarter_sample ? "q" : "h", s->data_partitioning, s->resync_marker, s->num_sprite_warping_points,
6019 s->sprite_warping_accuracy, 1-s->no_rounding, s->vo_type, s->vol_control_parameters ? " VOLC" : " ", s->intra_dc_threshold, s->cplx_estimation_trash_i, s->cplx_estimation_trash_p, s->cplx_estimation_trash_b);
6022 if(!s->scalability){
6023 if (s->shape!=RECT_SHAPE && s->pict_type!=FF_I_TYPE) {
6024 skip_bits1(gb); // vop shape coding type
6027 if(s->enhancement_type){
6028 int load_backward_shape= get_bits1(gb);
6029 if(load_backward_shape){
6030 av_log(s->avctx, AV_LOG_ERROR, "load backward shape isn't supported\n");
6033 skip_bits(gb, 2); //ref_select_code
6036 /* detect buggy encoders which don't set the low_delay flag (divx4/xvid/opendivx)*/
6037 // note we cannot detect divx5 without b-frames easily (although it's buggy too)
6038 if(s->vo_type==0 && s->vol_control_parameters==0 && s->divx_version==0 && s->picture_number==0){
6039 av_log(s->avctx, AV_LOG_ERROR, "looks like this file was encoded with (divx4/(old)xvid/opendivx) -> forcing low_delay flag\n");
6043 s->picture_number++; // better than pic number==0 always ;)
6045 s->y_dc_scale_table= ff_mpeg4_y_dc_scale_table; //FIXME add short header support
6046 s->c_dc_scale_table= ff_mpeg4_c_dc_scale_table;
6048 if(s->workaround_bugs&FF_BUG_EDGE){
6049 s->h_edge_pos= s->width;
6050 s->v_edge_pos= s->height;
6056 * decode mpeg4 headers
6057 * @return <0 if no VOP found (or a damaged one)
6058 * FRAME_SKIPPED if a not coded VOP is found
6059 * 0 if a VOP is found
6061 int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb)
6065 /* search next start code */
6068 if(s->codec_tag == AV_RL32("WV1F") && show_bits(gb, 24) == 0x575630){
6070 if(get_bits(gb, 8) == 0xF0)
6076 if(get_bits_count(gb) >= gb->size_in_bits){
6077 if(gb->size_in_bits==8 && (s->divx_version || s->xvid_build)){
6078 av_log(s->avctx, AV_LOG_ERROR, "frame skip %d\n", gb->size_in_bits);
6079 return FRAME_SKIPPED; //divx bug
6081 return -1; //end of stream
6084 /* use the bits after the test */
6085 v = get_bits(gb, 8);
6086 startcode = ((startcode << 8) | v) & 0xffffffff;
6088 if((startcode&0xFFFFFF00) != 0x100)
6089 continue; //no startcode
6091 if(s->avctx->debug&FF_DEBUG_STARTCODE){
6092 av_log(s->avctx, AV_LOG_DEBUG, "startcode: %3X ", startcode);
6093 if (startcode<=0x11F) av_log(s->avctx, AV_LOG_DEBUG, "Video Object Start");
6094 else if(startcode<=0x12F) av_log(s->avctx, AV_LOG_DEBUG, "Video Object Layer Start");
6095 else if(startcode<=0x13F) av_log(s->avctx, AV_LOG_DEBUG, "Reserved");
6096 else if(startcode<=0x15F) av_log(s->avctx, AV_LOG_DEBUG, "FGS bp start");
6097 else if(startcode<=0x1AF) av_log(s->avctx, AV_LOG_DEBUG, "Reserved");
6098 else if(startcode==0x1B0) av_log(s->avctx, AV_LOG_DEBUG, "Visual Object Seq Start");
6099 else if(startcode==0x1B1) av_log(s->avctx, AV_LOG_DEBUG, "Visual Object Seq End");
6100 else if(startcode==0x1B2) av_log(s->avctx, AV_LOG_DEBUG, "User Data");
6101 else if(startcode==0x1B3) av_log(s->avctx, AV_LOG_DEBUG, "Group of VOP start");
6102 else if(startcode==0x1B4) av_log(s->avctx, AV_LOG_DEBUG, "Video Session Error");
6103 else if(startcode==0x1B5) av_log(s->avctx, AV_LOG_DEBUG, "Visual Object Start");
6104 else if(startcode==0x1B6) av_log(s->avctx, AV_LOG_DEBUG, "Video Object Plane start");
6105 else if(startcode==0x1B7) av_log(s->avctx, AV_LOG_DEBUG, "slice start");
6106 else if(startcode==0x1B8) av_log(s->avctx, AV_LOG_DEBUG, "extension start");
6107 else if(startcode==0x1B9) av_log(s->avctx, AV_LOG_DEBUG, "fgs start");
6108 else if(startcode==0x1BA) av_log(s->avctx, AV_LOG_DEBUG, "FBA Object start");
6109 else if(startcode==0x1BB) av_log(s->avctx, AV_LOG_DEBUG, "FBA Object Plane start");
6110 else if(startcode==0x1BC) av_log(s->avctx, AV_LOG_DEBUG, "Mesh Object start");
6111 else if(startcode==0x1BD) av_log(s->avctx, AV_LOG_DEBUG, "Mesh Object Plane start");
6112 else if(startcode==0x1BE) av_log(s->avctx, AV_LOG_DEBUG, "Still Texture Object start");
6113 else if(startcode==0x1BF) av_log(s->avctx, AV_LOG_DEBUG, "Texture Spatial Layer start");
6114 else if(startcode==0x1C0) av_log(s->avctx, AV_LOG_DEBUG, "Texture SNR Layer start");
6115 else if(startcode==0x1C1) av_log(s->avctx, AV_LOG_DEBUG, "Texture Tile start");
6116 else if(startcode==0x1C2) av_log(s->avctx, AV_LOG_DEBUG, "Texture Shape Layer start");
6117 else if(startcode==0x1C3) av_log(s->avctx, AV_LOG_DEBUG, "stuffing start");
6118 else if(startcode<=0x1C5) av_log(s->avctx, AV_LOG_DEBUG, "reserved");
6119 else if(startcode<=0x1FF) av_log(s->avctx, AV_LOG_DEBUG, "System start");
6120 av_log(s->avctx, AV_LOG_DEBUG, " at %d\n", get_bits_count(gb));
6123 if(startcode >= 0x120 && startcode <= 0x12F){
6124 if(decode_vol_header(s, gb) < 0)
6127 else if(startcode == USER_DATA_STARTCODE){
6128 decode_user_data(s, gb);
6130 else if(startcode == GOP_STARTCODE){
6131 mpeg4_decode_gop_header(s, gb);
6133 else if(startcode == VOP_STARTCODE){
6141 if(s->flags& CODEC_FLAG_LOW_DELAY)
6143 s->avctx->has_b_frames= !s->low_delay;
6144 return decode_vop_header(s, gb);
6147 /* don't understand why they choose a different header ! */
6148 int intel_h263_decode_picture_header(MpegEncContext *s)
6152 /* picture header */
6153 if (get_bits_long(&s->gb, 22) != 0x20) {
6154 av_log(s->avctx, AV_LOG_ERROR, "Bad picture start code\n");
6157 s->picture_number = get_bits(&s->gb, 8); /* picture timestamp */
6159 if (get_bits1(&s->gb) != 1) {
6160 av_log(s->avctx, AV_LOG_ERROR, "Bad marker\n");
6161 return -1; /* marker */
6163 if (get_bits1(&s->gb) != 0) {
6164 av_log(s->avctx, AV_LOG_ERROR, "Bad H263 id\n");
6165 return -1; /* h263 id */
6167 skip_bits1(&s->gb); /* split screen off */
6168 skip_bits1(&s->gb); /* camera off */
6169 skip_bits1(&s->gb); /* freeze picture release off */
6171 format = get_bits(&s->gb, 3);
6173 av_log(s->avctx, AV_LOG_ERROR, "Intel H263 free format not supported\n");
6178 s->pict_type = FF_I_TYPE + get_bits1(&s->gb);
6180 s->unrestricted_mv = get_bits1(&s->gb);
6181 s->h263_long_vectors = s->unrestricted_mv;
6183 if (get_bits1(&s->gb) != 0) {
6184 av_log(s->avctx, AV_LOG_ERROR, "SAC not supported\n");
6185 return -1; /* SAC: off */
6187 s->obmc= get_bits1(&s->gb);
6188 if (get_bits1(&s->gb) != 0) {
6189 av_log(s->avctx, AV_LOG_ERROR, "PB frame mode no supported\n");
6190 return -1; /* PB frame mode */
6193 /* skip unknown header garbage */
6194 skip_bits(&s->gb, 41);
6196 s->chroma_qscale= s->qscale = get_bits(&s->gb, 5);
6197 skip_bits1(&s->gb); /* Continuous Presence Multipoint mode: off */
6200 while (get_bits1(&s->gb) != 0) {
6201 skip_bits(&s->gb, 8);
6205 s->y_dc_scale_table=
6206 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
6208 if(s->avctx->debug&FF_DEBUG_PICT_INFO)
6214 int flv_h263_decode_picture_header(MpegEncContext *s)
6216 int format, width, height;
6218 /* picture header */
6219 if (get_bits_long(&s->gb, 17) != 1) {
6220 av_log(s->avctx, AV_LOG_ERROR, "Bad picture start code\n");
6223 format = get_bits(&s->gb, 5);
6224 if (format != 0 && format != 1) {
6225 av_log(s->avctx, AV_LOG_ERROR, "Bad picture format\n");
6228 s->h263_flv = format+1;
6229 s->picture_number = get_bits(&s->gb, 8); /* picture timestamp */
6230 format = get_bits(&s->gb, 3);
6233 width = get_bits(&s->gb, 8);
6234 height = get_bits(&s->gb, 8);
6237 width = get_bits(&s->gb, 16);
6238 height = get_bits(&s->gb, 16);
6264 if(avcodec_check_dimensions(s->avctx, width, height))
6269 s->pict_type = FF_I_TYPE + get_bits(&s->gb, 2);
6270 s->dropable= s->pict_type > FF_P_TYPE;
6272 s->pict_type = FF_P_TYPE;
6274 skip_bits1(&s->gb); /* deblocking flag */
6275 s->chroma_qscale= s->qscale = get_bits(&s->gb, 5);
6279 s->unrestricted_mv = 1;
6280 s->h263_long_vectors = 0;
6283 while (get_bits1(&s->gb) != 0) {
6284 skip_bits(&s->gb, 8);
6288 if(s->avctx->debug & FF_DEBUG_PICT_INFO){
6289 av_log(s->avctx, AV_LOG_DEBUG, "%c esc_type:%d, qp:%d num:%d\n",
6290 s->dropable ? 'D' : av_get_pict_type_char(s->pict_type), s->h263_flv-1, s->qscale, s->picture_number);
6293 s->y_dc_scale_table=
6294 s->c_dc_scale_table= ff_mpeg1_dc_scale_table;