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

Back up before ISMIR
author Yading Song <yading.song@eecs.qmul.ac.uk>
date Thu, 31 Oct 2013 13:17:06 +0000
parents 6840f77b83aa
children
rev   line source
yading@10 1 /*
yading@10 2 * H.263 decoder
yading@10 3 * Copyright (c) 2001 Fabrice Bellard
yading@10 4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
yading@10 5 *
yading@10 6 * This file is part of FFmpeg.
yading@10 7 *
yading@10 8 * FFmpeg is free software; you can redistribute it and/or
yading@10 9 * modify it under the terms of the GNU Lesser General Public
yading@10 10 * License as published by the Free Software Foundation; either
yading@10 11 * version 2.1 of the License, or (at your option) any later version.
yading@10 12 *
yading@10 13 * FFmpeg is distributed in the hope that it will be useful,
yading@10 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
yading@10 16 * Lesser General Public License for more details.
yading@10 17 *
yading@10 18 * You should have received a copy of the GNU Lesser General Public
yading@10 19 * License along with FFmpeg; if not, write to the Free Software
yading@10 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10 21 */
yading@10 22
yading@10 23 /**
yading@10 24 * @file
yading@10 25 * H.263 decoder.
yading@10 26 */
yading@10 27
yading@10 28 #define UNCHECKED_BITSTREAM_READER 1
yading@10 29
yading@10 30 #include "libavutil/cpu.h"
yading@10 31 #include "internal.h"
yading@10 32 #include "avcodec.h"
yading@10 33 #include "error_resilience.h"
yading@10 34 #include "mpegvideo.h"
yading@10 35 #include "h263.h"
yading@10 36 #include "h263_parser.h"
yading@10 37 #include "mpeg4video_parser.h"
yading@10 38 #include "msmpeg4.h"
yading@10 39 #include "vdpau_internal.h"
yading@10 40 #include "thread.h"
yading@10 41 #include "flv.h"
yading@10 42 #include "mpeg4video.h"
yading@10 43
yading@10 44 //#define DEBUG
yading@10 45 //#define PRINT_FRAME_TIME
yading@10 46
yading@10 47 av_cold int ff_h263_decode_init(AVCodecContext *avctx)
yading@10 48 {
yading@10 49 MpegEncContext *s = avctx->priv_data;
yading@10 50 int ret;
yading@10 51
yading@10 52 s->avctx = avctx;
yading@10 53 s->out_format = FMT_H263;
yading@10 54
yading@10 55 s->width = avctx->coded_width;
yading@10 56 s->height = avctx->coded_height;
yading@10 57 s->workaround_bugs= avctx->workaround_bugs;
yading@10 58
yading@10 59 // set defaults
yading@10 60 ff_MPV_decode_defaults(s);
yading@10 61 s->quant_precision=5;
yading@10 62 s->decode_mb= ff_h263_decode_mb;
yading@10 63 s->low_delay= 1;
yading@10 64 if (avctx->codec->id == AV_CODEC_ID_MSS2)
yading@10 65 avctx->pix_fmt = AV_PIX_FMT_YUV420P;
yading@10 66 else
yading@10 67 avctx->pix_fmt = avctx->get_format(avctx, avctx->codec->pix_fmts);
yading@10 68 s->unrestricted_mv= 1;
yading@10 69
yading@10 70 /* select sub codec */
yading@10 71 switch(avctx->codec->id) {
yading@10 72 case AV_CODEC_ID_H263:
yading@10 73 case AV_CODEC_ID_H263P:
yading@10 74 s->unrestricted_mv= 0;
yading@10 75 avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
yading@10 76 break;
yading@10 77 case AV_CODEC_ID_MPEG4:
yading@10 78 break;
yading@10 79 case AV_CODEC_ID_MSMPEG4V1:
yading@10 80 s->h263_pred = 1;
yading@10 81 s->msmpeg4_version=1;
yading@10 82 break;
yading@10 83 case AV_CODEC_ID_MSMPEG4V2:
yading@10 84 s->h263_pred = 1;
yading@10 85 s->msmpeg4_version=2;
yading@10 86 break;
yading@10 87 case AV_CODEC_ID_MSMPEG4V3:
yading@10 88 s->h263_pred = 1;
yading@10 89 s->msmpeg4_version=3;
yading@10 90 break;
yading@10 91 case AV_CODEC_ID_WMV1:
yading@10 92 s->h263_pred = 1;
yading@10 93 s->msmpeg4_version=4;
yading@10 94 break;
yading@10 95 case AV_CODEC_ID_WMV2:
yading@10 96 s->h263_pred = 1;
yading@10 97 s->msmpeg4_version=5;
yading@10 98 break;
yading@10 99 case AV_CODEC_ID_VC1:
yading@10 100 case AV_CODEC_ID_WMV3:
yading@10 101 case AV_CODEC_ID_VC1IMAGE:
yading@10 102 case AV_CODEC_ID_WMV3IMAGE:
yading@10 103 case AV_CODEC_ID_MSS2:
yading@10 104 s->h263_pred = 1;
yading@10 105 s->msmpeg4_version=6;
yading@10 106 avctx->chroma_sample_location = AVCHROMA_LOC_LEFT;
yading@10 107 break;
yading@10 108 case AV_CODEC_ID_H263I:
yading@10 109 break;
yading@10 110 case AV_CODEC_ID_FLV1:
yading@10 111 s->h263_flv = 1;
yading@10 112 break;
yading@10 113 default:
yading@10 114 return AVERROR(EINVAL);
yading@10 115 }
yading@10 116 s->codec_id= avctx->codec->id;
yading@10 117 avctx->hwaccel= ff_find_hwaccel(avctx->codec->id, avctx->pix_fmt);
yading@10 118
yading@10 119 /* for h263, we allocate the images after having read the header */
yading@10 120 if (avctx->codec->id != AV_CODEC_ID_H263 && avctx->codec->id != AV_CODEC_ID_H263P && avctx->codec->id != AV_CODEC_ID_MPEG4)
yading@10 121 if ((ret = ff_MPV_common_init(s)) < 0)
yading@10 122 return ret;
yading@10 123
yading@10 124 ff_h263_decode_init_vlc();
yading@10 125
yading@10 126 return 0;
yading@10 127 }
yading@10 128
yading@10 129 av_cold int ff_h263_decode_end(AVCodecContext *avctx)
yading@10 130 {
yading@10 131 MpegEncContext *s = avctx->priv_data;
yading@10 132
yading@10 133 ff_MPV_common_end(s);
yading@10 134 return 0;
yading@10 135 }
yading@10 136
yading@10 137 /**
yading@10 138 * Return the number of bytes consumed for building the current frame.
yading@10 139 */
yading@10 140 static int get_consumed_bytes(MpegEncContext *s, int buf_size){
yading@10 141 int pos= (get_bits_count(&s->gb)+7)>>3;
yading@10 142
yading@10 143 if(s->divx_packed || s->avctx->hwaccel){
yading@10 144 //we would have to scan through the whole buf to handle the weird reordering ...
yading@10 145 return buf_size;
yading@10 146 }else if(s->flags&CODEC_FLAG_TRUNCATED){
yading@10 147 pos -= s->parse_context.last_index;
yading@10 148 if(pos<0) pos=0; // padding is not really read so this might be -1
yading@10 149 return pos;
yading@10 150 }else{
yading@10 151 if(pos==0) pos=1; //avoid infinite loops (i doubt that is needed but ...)
yading@10 152 if(pos+10>buf_size) pos=buf_size; // oops ;)
yading@10 153
yading@10 154 return pos;
yading@10 155 }
yading@10 156 }
yading@10 157
yading@10 158 static int decode_slice(MpegEncContext *s){
yading@10 159 const int part_mask= s->partitioned_frame ? (ER_AC_END|ER_AC_ERROR) : 0x7F;
yading@10 160 const int mb_size= 16>>s->avctx->lowres;
yading@10 161 int ret;
yading@10 162
yading@10 163 s->last_resync_gb= s->gb;
yading@10 164 s->first_slice_line= 1;
yading@10 165
yading@10 166 s->resync_mb_x= s->mb_x;
yading@10 167 s->resync_mb_y= s->mb_y;
yading@10 168
yading@10 169 ff_set_qscale(s, s->qscale);
yading@10 170
yading@10 171 if (s->avctx->hwaccel) {
yading@10 172 const uint8_t *start= s->gb.buffer + get_bits_count(&s->gb)/8;
yading@10 173 const uint8_t *end = ff_h263_find_resync_marker(s, start + 1, s->gb.buffer_end);
yading@10 174 skip_bits_long(&s->gb, 8*(end - start));
yading@10 175 return s->avctx->hwaccel->decode_slice(s->avctx, start, end - start);
yading@10 176 }
yading@10 177
yading@10 178 if(s->partitioned_frame){
yading@10 179 const int qscale= s->qscale;
yading@10 180
yading@10 181 if(CONFIG_MPEG4_DECODER && s->codec_id==AV_CODEC_ID_MPEG4){
yading@10 182 if ((ret = ff_mpeg4_decode_partitions(s)) < 0)
yading@10 183 return ret;
yading@10 184 }
yading@10 185
yading@10 186 /* restore variables which were modified */
yading@10 187 s->first_slice_line=1;
yading@10 188 s->mb_x= s->resync_mb_x;
yading@10 189 s->mb_y= s->resync_mb_y;
yading@10 190 ff_set_qscale(s, qscale);
yading@10 191 }
yading@10 192
yading@10 193 for(; s->mb_y < s->mb_height; s->mb_y++) {
yading@10 194 /* per-row end of slice checks */
yading@10 195 if(s->msmpeg4_version){
yading@10 196 if(s->resync_mb_y + s->slice_height == s->mb_y){
yading@10 197 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, ER_MB_END);
yading@10 198
yading@10 199 return 0;
yading@10 200 }
yading@10 201 }
yading@10 202
yading@10 203 if(s->msmpeg4_version==1){
yading@10 204 s->last_dc[0]=
yading@10 205 s->last_dc[1]=
yading@10 206 s->last_dc[2]= 128;
yading@10 207 }
yading@10 208
yading@10 209 ff_init_block_index(s);
yading@10 210 for(; s->mb_x < s->mb_width; s->mb_x++) {
yading@10 211 int ret;
yading@10 212
yading@10 213 ff_update_block_index(s);
yading@10 214
yading@10 215 if(s->resync_mb_x == s->mb_x && s->resync_mb_y+1 == s->mb_y){
yading@10 216 s->first_slice_line=0;
yading@10 217 }
yading@10 218
yading@10 219 /* DCT & quantize */
yading@10 220
yading@10 221 s->mv_dir = MV_DIR_FORWARD;
yading@10 222 s->mv_type = MV_TYPE_16X16;
yading@10 223 // s->mb_skipped = 0;
yading@10 224 av_dlog(s, "%d %d %06X\n",
yading@10 225 ret, get_bits_count(&s->gb), show_bits(&s->gb, 24));
yading@10 226 ret= s->decode_mb(s, s->block);
yading@10 227
yading@10 228 if (s->pict_type!=AV_PICTURE_TYPE_B)
yading@10 229 ff_h263_update_motion_val(s);
yading@10 230
yading@10 231 if(ret<0){
yading@10 232 const int xy= s->mb_x + s->mb_y*s->mb_stride;
yading@10 233 if(ret==SLICE_END){
yading@10 234 ff_MPV_decode_mb(s, s->block);
yading@10 235 if(s->loop_filter)
yading@10 236 ff_h263_loop_filter(s);
yading@10 237
yading@10 238 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, ER_MB_END&part_mask);
yading@10 239
yading@10 240 s->padding_bug_score--;
yading@10 241
yading@10 242 if(++s->mb_x >= s->mb_width){
yading@10 243 s->mb_x=0;
yading@10 244 ff_mpeg_draw_horiz_band(s, s->mb_y*mb_size, mb_size);
yading@10 245 ff_MPV_report_decode_progress(s);
yading@10 246 s->mb_y++;
yading@10 247 }
yading@10 248 return 0;
yading@10 249 }else if(ret==SLICE_NOEND){
yading@10 250 av_log(s->avctx, AV_LOG_ERROR, "Slice mismatch at MB: %d\n", xy);
yading@10 251 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y, s->mb_x+1, s->mb_y, ER_MB_END&part_mask);
yading@10 252 return AVERROR_INVALIDDATA;
yading@10 253 }
yading@10 254 av_log(s->avctx, AV_LOG_ERROR, "Error at MB: %d\n", xy);
yading@10 255 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, ER_MB_ERROR&part_mask);
yading@10 256
yading@10 257 return AVERROR_INVALIDDATA;
yading@10 258 }
yading@10 259
yading@10 260 ff_MPV_decode_mb(s, s->block);
yading@10 261 if(s->loop_filter)
yading@10 262 ff_h263_loop_filter(s);
yading@10 263 }
yading@10 264
yading@10 265 ff_mpeg_draw_horiz_band(s, s->mb_y*mb_size, mb_size);
yading@10 266 ff_MPV_report_decode_progress(s);
yading@10 267
yading@10 268 s->mb_x= 0;
yading@10 269 }
yading@10 270
yading@10 271 av_assert1(s->mb_x==0 && s->mb_y==s->mb_height);
yading@10 272
yading@10 273 if(s->codec_id==AV_CODEC_ID_MPEG4
yading@10 274 && (s->workaround_bugs&FF_BUG_AUTODETECT)
yading@10 275 && get_bits_left(&s->gb) >= 48
yading@10 276 && show_bits(&s->gb, 24)==0x4010
yading@10 277 && !s->data_partitioning)
yading@10 278 s->padding_bug_score+=32;
yading@10 279
yading@10 280 /* try to detect the padding bug */
yading@10 281 if( s->codec_id==AV_CODEC_ID_MPEG4
yading@10 282 && (s->workaround_bugs&FF_BUG_AUTODETECT)
yading@10 283 && get_bits_left(&s->gb) >=0
yading@10 284 && get_bits_left(&s->gb) < 137
yading@10 285 // && !s->resync_marker
yading@10 286 && !s->data_partitioning){
yading@10 287
yading@10 288 const int bits_count= get_bits_count(&s->gb);
yading@10 289 const int bits_left = s->gb.size_in_bits - bits_count;
yading@10 290
yading@10 291 if(bits_left==0){
yading@10 292 s->padding_bug_score+=16;
yading@10 293 } else if(bits_left != 1){
yading@10 294 int v= show_bits(&s->gb, 8);
yading@10 295 v|= 0x7F >> (7-(bits_count&7));
yading@10 296
yading@10 297 if(v==0x7F && bits_left<=8)
yading@10 298 s->padding_bug_score--;
yading@10 299 else if(v==0x7F && ((get_bits_count(&s->gb)+8)&8) && bits_left<=16)
yading@10 300 s->padding_bug_score+= 4;
yading@10 301 else
yading@10 302 s->padding_bug_score++;
yading@10 303 }
yading@10 304 }
yading@10 305
yading@10 306 if(s->workaround_bugs&FF_BUG_AUTODETECT){
yading@10 307 if(s->padding_bug_score > -2 && !s->data_partitioning /*&& (s->divx_version>=0 || !s->resync_marker)*/)
yading@10 308 s->workaround_bugs |= FF_BUG_NO_PADDING;
yading@10 309 else
yading@10 310 s->workaround_bugs &= ~FF_BUG_NO_PADDING;
yading@10 311 }
yading@10 312
yading@10 313 // handle formats which don't have unique end markers
yading@10 314 if(s->msmpeg4_version || (s->workaround_bugs&FF_BUG_NO_PADDING)){ //FIXME perhaps solve this more cleanly
yading@10 315 int left= get_bits_left(&s->gb);
yading@10 316 int max_extra=7;
yading@10 317
yading@10 318 /* no markers in M$ crap */
yading@10 319 if(s->msmpeg4_version && s->pict_type==AV_PICTURE_TYPE_I)
yading@10 320 max_extra+= 17;
yading@10 321
yading@10 322 /* buggy padding but the frame should still end approximately at the bitstream end */
yading@10 323 if((s->workaround_bugs&FF_BUG_NO_PADDING) && (s->err_recognition&(AV_EF_BUFFER|AV_EF_AGGRESSIVE)))
yading@10 324 max_extra+= 48;
yading@10 325 else if((s->workaround_bugs&FF_BUG_NO_PADDING))
yading@10 326 max_extra+= 256*256*256*64;
yading@10 327
yading@10 328 if(left>max_extra){
yading@10 329 av_log(s->avctx, AV_LOG_ERROR, "discarding %d junk bits at end, next would be %X\n", left, show_bits(&s->gb, 24));
yading@10 330 }
yading@10 331 else if(left<0){
yading@10 332 av_log(s->avctx, AV_LOG_ERROR, "overreading %d bits\n", -left);
yading@10 333 }else
yading@10 334 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, ER_MB_END);
yading@10 335
yading@10 336 return 0;
yading@10 337 }
yading@10 338
yading@10 339 av_log(s->avctx, AV_LOG_ERROR, "slice end not reached but screenspace end (%d left %06X, score= %d)\n",
yading@10 340 get_bits_left(&s->gb),
yading@10 341 show_bits(&s->gb, 24), s->padding_bug_score);
yading@10 342
yading@10 343 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, ER_MB_END&part_mask);
yading@10 344
yading@10 345 return AVERROR_INVALIDDATA;
yading@10 346 }
yading@10 347
yading@10 348 int ff_h263_decode_frame(AVCodecContext *avctx,
yading@10 349 void *data, int *got_frame,
yading@10 350 AVPacket *avpkt)
yading@10 351 {
yading@10 352 const uint8_t *buf = avpkt->data;
yading@10 353 int buf_size = avpkt->size;
yading@10 354 MpegEncContext *s = avctx->priv_data;
yading@10 355 int ret;
yading@10 356 AVFrame *pict = data;
yading@10 357
yading@10 358 #ifdef PRINT_FRAME_TIME
yading@10 359 uint64_t time= rdtsc();
yading@10 360 #endif
yading@10 361 s->flags= avctx->flags;
yading@10 362 s->flags2= avctx->flags2;
yading@10 363
yading@10 364 /* no supplementary picture */
yading@10 365 if (buf_size == 0) {
yading@10 366 /* special case for last picture */
yading@10 367 if (s->low_delay==0 && s->next_picture_ptr) {
yading@10 368 if ((ret = av_frame_ref(pict, &s->next_picture_ptr->f)) < 0)
yading@10 369 return ret;
yading@10 370 s->next_picture_ptr= NULL;
yading@10 371
yading@10 372 *got_frame = 1;
yading@10 373 }
yading@10 374
yading@10 375 return 0;
yading@10 376 }
yading@10 377
yading@10 378 if(s->flags&CODEC_FLAG_TRUNCATED){
yading@10 379 int next;
yading@10 380
yading@10 381 if(CONFIG_MPEG4_DECODER && s->codec_id==AV_CODEC_ID_MPEG4){
yading@10 382 next= ff_mpeg4_find_frame_end(&s->parse_context, buf, buf_size);
yading@10 383 }else if(CONFIG_H263_DECODER && s->codec_id==AV_CODEC_ID_H263){
yading@10 384 next= ff_h263_find_frame_end(&s->parse_context, buf, buf_size);
yading@10 385 }else if(CONFIG_H263P_DECODER && s->codec_id==AV_CODEC_ID_H263P){
yading@10 386 next= ff_h263_find_frame_end(&s->parse_context, buf, buf_size);
yading@10 387 }else{
yading@10 388 av_log(s->avctx, AV_LOG_ERROR, "this codec does not support truncated bitstreams\n");
yading@10 389 return AVERROR(EINVAL);
yading@10 390 }
yading@10 391
yading@10 392 if( ff_combine_frame(&s->parse_context, next, (const uint8_t **)&buf, &buf_size) < 0 )
yading@10 393 return buf_size;
yading@10 394 }
yading@10 395
yading@10 396
yading@10 397 retry:
yading@10 398 if(s->divx_packed && s->bitstream_buffer_size){
yading@10 399 int i;
yading@10 400 for(i=0; i<buf_size-3; i++){
yading@10 401 if(buf[i]==0 && buf[i+1]==0 && buf[i+2]==1){
yading@10 402 if(buf[i+3]==0xB0){
yading@10 403 av_log(s->avctx, AV_LOG_WARNING, "Discarding excessive bitstream in packed xvid\n");
yading@10 404 s->bitstream_buffer_size=0;
yading@10 405 }
yading@10 406 break;
yading@10 407 }
yading@10 408 }
yading@10 409 }
yading@10 410
yading@10 411 if(s->bitstream_buffer_size && (s->divx_packed || buf_size<20)){ //divx 5.01+/xvid frame reorder
yading@10 412 init_get_bits(&s->gb, s->bitstream_buffer, s->bitstream_buffer_size*8);
yading@10 413 }else
yading@10 414 init_get_bits(&s->gb, buf, buf_size*8);
yading@10 415 s->bitstream_buffer_size=0;
yading@10 416
yading@10 417 if (!s->context_initialized) {
yading@10 418 if ((ret = ff_MPV_common_init(s)) < 0) //we need the idct permutaton for reading a custom matrix
yading@10 419 return ret;
yading@10 420 }
yading@10 421
yading@10 422 /* We need to set current_picture_ptr before reading the header,
yading@10 423 * otherwise we cannot store anyting in there */
yading@10 424 if (s->current_picture_ptr == NULL || s->current_picture_ptr->f.data[0]) {
yading@10 425 int i= ff_find_unused_picture(s, 0);
yading@10 426 if (i < 0)
yading@10 427 return i;
yading@10 428 s->current_picture_ptr= &s->picture[i];
yading@10 429 }
yading@10 430
yading@10 431 /* let's go :-) */
yading@10 432 if (CONFIG_WMV2_DECODER && s->msmpeg4_version==5) {
yading@10 433 ret= ff_wmv2_decode_picture_header(s);
yading@10 434 } else if (CONFIG_MSMPEG4_DECODER && s->msmpeg4_version) {
yading@10 435 ret = ff_msmpeg4_decode_picture_header(s);
yading@10 436 } else if (CONFIG_MPEG4_DECODER && s->h263_pred) {
yading@10 437 if(s->avctx->extradata_size && s->picture_number==0){
yading@10 438 GetBitContext gb;
yading@10 439
yading@10 440 init_get_bits(&gb, s->avctx->extradata, s->avctx->extradata_size*8);
yading@10 441 ret = ff_mpeg4_decode_picture_header(s, &gb);
yading@10 442 }
yading@10 443 ret = ff_mpeg4_decode_picture_header(s, &s->gb);
yading@10 444 } else if (CONFIG_H263I_DECODER && s->codec_id == AV_CODEC_ID_H263I) {
yading@10 445 ret = ff_intel_h263_decode_picture_header(s);
yading@10 446 } else if (CONFIG_FLV_DECODER && s->h263_flv) {
yading@10 447 ret = ff_flv_decode_picture_header(s);
yading@10 448 } else {
yading@10 449 ret = ff_h263_decode_picture_header(s);
yading@10 450 }
yading@10 451
yading@10 452 if (ret < 0 || ret==FRAME_SKIPPED) {
yading@10 453 if ( s->width != avctx->coded_width
yading@10 454 || s->height != avctx->coded_height) {
yading@10 455 av_log(s->avctx, AV_LOG_WARNING, "Reverting picture dimensions change due to header decoding failure\n");
yading@10 456 s->width = avctx->coded_width;
yading@10 457 s->height= avctx->coded_height;
yading@10 458 }
yading@10 459 }
yading@10 460 if(ret==FRAME_SKIPPED) return get_consumed_bytes(s, buf_size);
yading@10 461
yading@10 462 /* skip if the header was thrashed */
yading@10 463 if (ret < 0){
yading@10 464 av_log(s->avctx, AV_LOG_ERROR, "header damaged\n");
yading@10 465 return ret;
yading@10 466 }
yading@10 467
yading@10 468 avctx->has_b_frames= !s->low_delay;
yading@10 469
yading@10 470 if(s->xvid_build==-1 && s->divx_version==-1 && s->lavc_build==-1){
yading@10 471 if(s->stream_codec_tag == AV_RL32("XVID") ||
yading@10 472 s->codec_tag == AV_RL32("XVID") || s->codec_tag == AV_RL32("XVIX") ||
yading@10 473 s->codec_tag == AV_RL32("RMP4") || s->codec_tag == AV_RL32("ZMP4") ||
yading@10 474 s->codec_tag == AV_RL32("SIPP")
yading@10 475 )
yading@10 476 s->xvid_build= 0;
yading@10 477 #if 0
yading@10 478 if(s->codec_tag == AV_RL32("DIVX") && s->vo_type==0 && s->vol_control_parameters==1
yading@10 479 && s->padding_bug_score > 0 && s->low_delay) // XVID with modified fourcc
yading@10 480 s->xvid_build= 0;
yading@10 481 #endif
yading@10 482 }
yading@10 483
yading@10 484 if(s->xvid_build==-1 && s->divx_version==-1 && s->lavc_build==-1){
yading@10 485 if(s->codec_tag == AV_RL32("DIVX") && s->vo_type==0 && s->vol_control_parameters==0)
yading@10 486 s->divx_version= 400; //divx 4
yading@10 487 }
yading@10 488
yading@10 489 if(s->xvid_build>=0 && s->divx_version>=0){
yading@10 490 s->divx_version=
yading@10 491 s->divx_build= -1;
yading@10 492 }
yading@10 493
yading@10 494 if(s->workaround_bugs&FF_BUG_AUTODETECT){
yading@10 495 if(s->codec_tag == AV_RL32("XVIX"))
yading@10 496 s->workaround_bugs|= FF_BUG_XVID_ILACE;
yading@10 497
yading@10 498 if(s->codec_tag == AV_RL32("UMP4")){
yading@10 499 s->workaround_bugs|= FF_BUG_UMP4;
yading@10 500 }
yading@10 501
yading@10 502 if(s->divx_version>=500 && s->divx_build<1814){
yading@10 503 s->workaround_bugs|= FF_BUG_QPEL_CHROMA;
yading@10 504 }
yading@10 505
yading@10 506 if(s->divx_version>502 && s->divx_build<1814){
yading@10 507 s->workaround_bugs|= FF_BUG_QPEL_CHROMA2;
yading@10 508 }
yading@10 509
yading@10 510 if(s->xvid_build<=3U)
yading@10 511 s->padding_bug_score= 256*256*256*64;
yading@10 512
yading@10 513 if(s->xvid_build<=1U)
yading@10 514 s->workaround_bugs|= FF_BUG_QPEL_CHROMA;
yading@10 515
yading@10 516 if(s->xvid_build<=12U)
yading@10 517 s->workaround_bugs|= FF_BUG_EDGE;
yading@10 518
yading@10 519 if(s->xvid_build<=32U)
yading@10 520 s->workaround_bugs|= FF_BUG_DC_CLIP;
yading@10 521
yading@10 522 #define SET_QPEL_FUNC(postfix1, postfix2) \
yading@10 523 s->dsp.put_ ## postfix1 = ff_put_ ## postfix2;\
yading@10 524 s->dsp.put_no_rnd_ ## postfix1 = ff_put_no_rnd_ ## postfix2;\
yading@10 525 s->dsp.avg_ ## postfix1 = ff_avg_ ## postfix2;
yading@10 526
yading@10 527 if(s->lavc_build<4653U)
yading@10 528 s->workaround_bugs|= FF_BUG_STD_QPEL;
yading@10 529
yading@10 530 if(s->lavc_build<4655U)
yading@10 531 s->workaround_bugs|= FF_BUG_DIRECT_BLOCKSIZE;
yading@10 532
yading@10 533 if(s->lavc_build<4670U){
yading@10 534 s->workaround_bugs|= FF_BUG_EDGE;
yading@10 535 }
yading@10 536
yading@10 537 if(s->lavc_build<=4712U)
yading@10 538 s->workaround_bugs|= FF_BUG_DC_CLIP;
yading@10 539
yading@10 540 if(s->divx_version>=0)
yading@10 541 s->workaround_bugs|= FF_BUG_DIRECT_BLOCKSIZE;
yading@10 542 if(s->divx_version==501 && s->divx_build==20020416)
yading@10 543 s->padding_bug_score= 256*256*256*64;
yading@10 544
yading@10 545 if(s->divx_version<500U){
yading@10 546 s->workaround_bugs|= FF_BUG_EDGE;
yading@10 547 }
yading@10 548
yading@10 549 if(s->divx_version>=0)
yading@10 550 s->workaround_bugs|= FF_BUG_HPEL_CHROMA;
yading@10 551 #if 0
yading@10 552 if(s->divx_version==500)
yading@10 553 s->padding_bug_score= 256*256*256*64;
yading@10 554
yading@10 555 /* very ugly XVID padding bug detection FIXME/XXX solve this differently
yading@10 556 * Let us hope this at least works.
yading@10 557 */
yading@10 558 if( s->resync_marker==0 && s->data_partitioning==0 && s->divx_version==-1
yading@10 559 && s->codec_id==AV_CODEC_ID_MPEG4 && s->vo_type==0)
yading@10 560 s->workaround_bugs|= FF_BUG_NO_PADDING;
yading@10 561
yading@10 562 if(s->lavc_build<4609U) //FIXME not sure about the version num but a 4609 file seems ok
yading@10 563 s->workaround_bugs|= FF_BUG_NO_PADDING;
yading@10 564 #endif
yading@10 565 }
yading@10 566
yading@10 567 if(s->workaround_bugs& FF_BUG_STD_QPEL){
yading@10 568 SET_QPEL_FUNC(qpel_pixels_tab[0][ 5], qpel16_mc11_old_c)
yading@10 569 SET_QPEL_FUNC(qpel_pixels_tab[0][ 7], qpel16_mc31_old_c)
yading@10 570 SET_QPEL_FUNC(qpel_pixels_tab[0][ 9], qpel16_mc12_old_c)
yading@10 571 SET_QPEL_FUNC(qpel_pixels_tab[0][11], qpel16_mc32_old_c)
yading@10 572 SET_QPEL_FUNC(qpel_pixels_tab[0][13], qpel16_mc13_old_c)
yading@10 573 SET_QPEL_FUNC(qpel_pixels_tab[0][15], qpel16_mc33_old_c)
yading@10 574
yading@10 575 SET_QPEL_FUNC(qpel_pixels_tab[1][ 5], qpel8_mc11_old_c)
yading@10 576 SET_QPEL_FUNC(qpel_pixels_tab[1][ 7], qpel8_mc31_old_c)
yading@10 577 SET_QPEL_FUNC(qpel_pixels_tab[1][ 9], qpel8_mc12_old_c)
yading@10 578 SET_QPEL_FUNC(qpel_pixels_tab[1][11], qpel8_mc32_old_c)
yading@10 579 SET_QPEL_FUNC(qpel_pixels_tab[1][13], qpel8_mc13_old_c)
yading@10 580 SET_QPEL_FUNC(qpel_pixels_tab[1][15], qpel8_mc33_old_c)
yading@10 581 }
yading@10 582
yading@10 583 if(avctx->debug & FF_DEBUG_BUGS)
yading@10 584 av_log(s->avctx, AV_LOG_DEBUG, "bugs: %X lavc_build:%d xvid_build:%d divx_version:%d divx_build:%d %s\n",
yading@10 585 s->workaround_bugs, s->lavc_build, s->xvid_build, s->divx_version, s->divx_build,
yading@10 586 s->divx_packed ? "p" : "");
yading@10 587
yading@10 588 #if HAVE_MMX
yading@10 589 if (s->codec_id == AV_CODEC_ID_MPEG4 && s->xvid_build>=0 && avctx->idct_algo == FF_IDCT_AUTO && (av_get_cpu_flags() & AV_CPU_FLAG_MMX)) {
yading@10 590 avctx->idct_algo= FF_IDCT_XVIDMMX;
yading@10 591 ff_dct_common_init(s);
yading@10 592 goto retry;
yading@10 593 }
yading@10 594 #endif
yading@10 595
yading@10 596 /* After H263 & mpeg4 header decode we have the height, width,*/
yading@10 597 /* and other parameters. So then we could init the picture */
yading@10 598 /* FIXME: By the way H263 decoder is evolving it should have */
yading@10 599 /* an H263EncContext */
yading@10 600
yading@10 601 if ((!avctx->coded_width || !avctx->coded_height) && 0) {
yading@10 602 ParseContext pc= s->parse_context; //FIXME move these demuxng hack to avformat
yading@10 603
yading@10 604 s->parse_context.buffer=0;
yading@10 605 ff_MPV_common_end(s);
yading@10 606 s->parse_context= pc;
yading@10 607 avcodec_set_dimensions(avctx, s->width, s->height);
yading@10 608
yading@10 609 goto retry;
yading@10 610 }
yading@10 611
yading@10 612 if (s->width != avctx->coded_width ||
yading@10 613 s->height != avctx->coded_height ||
yading@10 614 s->context_reinit) {
yading@10 615 /* H.263 could change picture size any time */
yading@10 616 s->context_reinit = 0;
yading@10 617
yading@10 618 avcodec_set_dimensions(avctx, s->width, s->height);
yading@10 619
yading@10 620 if ((ret = ff_MPV_common_frame_size_change(s)))
yading@10 621 return ret;
yading@10 622 }
yading@10 623
yading@10 624 if((s->codec_id==AV_CODEC_ID_H263 || s->codec_id==AV_CODEC_ID_H263P || s->codec_id == AV_CODEC_ID_H263I))
yading@10 625 s->gob_index = ff_h263_get_gob_height(s);
yading@10 626
yading@10 627 // for skipping the frame
yading@10 628 s->current_picture.f.pict_type = s->pict_type;
yading@10 629 s->current_picture.f.key_frame = s->pict_type == AV_PICTURE_TYPE_I;
yading@10 630
yading@10 631 /* skip B-frames if we don't have reference frames */
yading@10 632 if (s->last_picture_ptr == NULL &&
yading@10 633 (s->pict_type == AV_PICTURE_TYPE_B || s->droppable))
yading@10 634 return get_consumed_bytes(s, buf_size);
yading@10 635 if( (avctx->skip_frame >= AVDISCARD_NONREF && s->pict_type==AV_PICTURE_TYPE_B)
yading@10 636 || (avctx->skip_frame >= AVDISCARD_NONKEY && s->pict_type!=AV_PICTURE_TYPE_I)
yading@10 637 || avctx->skip_frame >= AVDISCARD_ALL)
yading@10 638 return get_consumed_bytes(s, buf_size);
yading@10 639
yading@10 640 if(s->next_p_frame_damaged){
yading@10 641 if(s->pict_type==AV_PICTURE_TYPE_B)
yading@10 642 return get_consumed_bytes(s, buf_size);
yading@10 643 else
yading@10 644 s->next_p_frame_damaged=0;
yading@10 645 }
yading@10 646
yading@10 647 if((!s->no_rounding) || s->pict_type==AV_PICTURE_TYPE_B){
yading@10 648 s->me.qpel_put= s->dsp.put_qpel_pixels_tab;
yading@10 649 s->me.qpel_avg= s->dsp.avg_qpel_pixels_tab;
yading@10 650 }else{
yading@10 651 s->me.qpel_put= s->dsp.put_no_rnd_qpel_pixels_tab;
yading@10 652 s->me.qpel_avg= s->dsp.avg_qpel_pixels_tab;
yading@10 653 }
yading@10 654
yading@10 655 if ((ret = ff_MPV_frame_start(s, avctx)) < 0)
yading@10 656 return ret;
yading@10 657
yading@10 658 if (!s->divx_packed && !avctx->hwaccel)
yading@10 659 ff_thread_finish_setup(avctx);
yading@10 660
yading@10 661 if (CONFIG_MPEG4_VDPAU_DECODER && (s->avctx->codec->capabilities & CODEC_CAP_HWACCEL_VDPAU)) {
yading@10 662 ff_vdpau_mpeg4_decode_picture(s, s->gb.buffer, s->gb.buffer_end - s->gb.buffer);
yading@10 663 goto frame_end;
yading@10 664 }
yading@10 665
yading@10 666 if (avctx->hwaccel) {
yading@10 667 if ((ret = avctx->hwaccel->start_frame(avctx, s->gb.buffer, s->gb.buffer_end - s->gb.buffer)) < 0)
yading@10 668 return ret;
yading@10 669 }
yading@10 670
yading@10 671 ff_mpeg_er_frame_start(s);
yading@10 672
yading@10 673 //the second part of the wmv2 header contains the MB skip bits which are stored in current_picture->mb_type
yading@10 674 //which is not available before ff_MPV_frame_start()
yading@10 675 if (CONFIG_WMV2_DECODER && s->msmpeg4_version==5){
yading@10 676 ret = ff_wmv2_decode_secondary_picture_header(s);
yading@10 677 if(ret<0) return ret;
yading@10 678 if(ret==1) goto intrax8_decoded;
yading@10 679 }
yading@10 680
yading@10 681 /* decode each macroblock */
yading@10 682 s->mb_x=0;
yading@10 683 s->mb_y=0;
yading@10 684
yading@10 685 ret = decode_slice(s);
yading@10 686 while(s->mb_y<s->mb_height){
yading@10 687 if(s->msmpeg4_version){
yading@10 688 if(s->slice_height==0 || s->mb_x!=0 || (s->mb_y%s->slice_height)!=0 || get_bits_left(&s->gb)<0)
yading@10 689 break;
yading@10 690 }else{
yading@10 691 int prev_x=s->mb_x, prev_y=s->mb_y;
yading@10 692 if(ff_h263_resync(s)<0)
yading@10 693 break;
yading@10 694 if (prev_y * s->mb_width + prev_x < s->mb_y * s->mb_width + s->mb_x)
yading@10 695 s->er.error_occurred = 1;
yading@10 696 }
yading@10 697
yading@10 698 if(s->msmpeg4_version<4 && s->h263_pred)
yading@10 699 ff_mpeg4_clean_buffers(s);
yading@10 700
yading@10 701 if (decode_slice(s) < 0) ret = AVERROR_INVALIDDATA;
yading@10 702 }
yading@10 703
yading@10 704 if (s->msmpeg4_version && s->msmpeg4_version<4 && s->pict_type==AV_PICTURE_TYPE_I)
yading@10 705 if(!CONFIG_MSMPEG4_DECODER || ff_msmpeg4_decode_ext_header(s, buf_size) < 0){
yading@10 706 s->er.error_status_table[s->mb_num - 1] = ER_MB_ERROR;
yading@10 707 }
yading@10 708
yading@10 709 av_assert1(s->bitstream_buffer_size==0);
yading@10 710 frame_end:
yading@10 711 /* divx 5.01+ bitstream reorder stuff */
yading@10 712 if(s->codec_id==AV_CODEC_ID_MPEG4 && s->divx_packed){
yading@10 713 int current_pos= s->gb.buffer == s->bitstream_buffer ? 0 : (get_bits_count(&s->gb)>>3);
yading@10 714 int startcode_found=0;
yading@10 715
yading@10 716 if(buf_size - current_pos > 7){
yading@10 717 int i;
yading@10 718 for(i=current_pos; i<buf_size-4; i++){
yading@10 719 if(buf[i]==0 && buf[i+1]==0 && buf[i+2]==1 && buf[i+3]==0xB6){
yading@10 720 startcode_found=!(buf[i+4]&0x40);
yading@10 721 break;
yading@10 722 }
yading@10 723 }
yading@10 724 }
yading@10 725
yading@10 726 if(startcode_found){
yading@10 727 av_fast_malloc(
yading@10 728 &s->bitstream_buffer,
yading@10 729 &s->allocated_bitstream_buffer_size,
yading@10 730 buf_size - current_pos + FF_INPUT_BUFFER_PADDING_SIZE);
yading@10 731 if (!s->bitstream_buffer)
yading@10 732 return AVERROR(ENOMEM);
yading@10 733 memcpy(s->bitstream_buffer, buf + current_pos, buf_size - current_pos);
yading@10 734 s->bitstream_buffer_size= buf_size - current_pos;
yading@10 735 }
yading@10 736 }
yading@10 737
yading@10 738 intrax8_decoded:
yading@10 739 ff_er_frame_end(&s->er);
yading@10 740
yading@10 741 if (avctx->hwaccel) {
yading@10 742 if ((ret = avctx->hwaccel->end_frame(avctx)) < 0)
yading@10 743 return ret;
yading@10 744 }
yading@10 745
yading@10 746 ff_MPV_frame_end(s);
yading@10 747
yading@10 748 if (!s->divx_packed && avctx->hwaccel)
yading@10 749 ff_thread_finish_setup(avctx);
yading@10 750
yading@10 751 av_assert1(s->current_picture.f.pict_type == s->current_picture_ptr->f.pict_type);
yading@10 752 av_assert1(s->current_picture.f.pict_type == s->pict_type);
yading@10 753 if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
yading@10 754 if ((ret = av_frame_ref(pict, &s->current_picture_ptr->f)) < 0)
yading@10 755 return ret;
yading@10 756 ff_print_debug_info(s, s->current_picture_ptr, pict);
yading@10 757 ff_mpv_export_qp_table(s, pict, s->current_picture_ptr, FF_QSCALE_TYPE_MPEG1);
yading@10 758 } else if (s->last_picture_ptr != NULL) {
yading@10 759 if ((ret = av_frame_ref(pict, &s->last_picture_ptr->f)) < 0)
yading@10 760 return ret;
yading@10 761 ff_print_debug_info(s, s->last_picture_ptr, pict);
yading@10 762 ff_mpv_export_qp_table(s, pict, s->last_picture_ptr, FF_QSCALE_TYPE_MPEG1);
yading@10 763 }
yading@10 764
yading@10 765 if(s->last_picture_ptr || s->low_delay){
yading@10 766 *got_frame = 1;
yading@10 767 }
yading@10 768
yading@10 769 #ifdef PRINT_FRAME_TIME
yading@10 770 av_log(avctx, AV_LOG_DEBUG, "%"PRId64"\n", rdtsc()-time);
yading@10 771 #endif
yading@10 772
yading@10 773 return (ret && (avctx->err_recognition & AV_EF_EXPLODE))?ret:get_consumed_bytes(s, buf_size);
yading@10 774 }
yading@10 775
yading@10 776 const enum AVPixelFormat ff_h263_hwaccel_pixfmt_list_420[] = {
yading@10 777 #if CONFIG_VAAPI
yading@10 778 AV_PIX_FMT_VAAPI_VLD,
yading@10 779 #endif
yading@10 780 #if CONFIG_VDPAU
yading@10 781 AV_PIX_FMT_VDPAU,
yading@10 782 #endif
yading@10 783 AV_PIX_FMT_YUV420P,
yading@10 784 AV_PIX_FMT_NONE
yading@10 785 };
yading@10 786
yading@10 787 AVCodec ff_h263_decoder = {
yading@10 788 .name = "h263",
yading@10 789 .type = AVMEDIA_TYPE_VIDEO,
yading@10 790 .id = AV_CODEC_ID_H263,
yading@10 791 .priv_data_size = sizeof(MpegEncContext),
yading@10 792 .init = ff_h263_decode_init,
yading@10 793 .close = ff_h263_decode_end,
yading@10 794 .decode = ff_h263_decode_frame,
yading@10 795 .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
yading@10 796 CODEC_CAP_TRUNCATED | CODEC_CAP_DELAY,
yading@10 797 .flush = ff_mpeg_flush,
yading@10 798 .max_lowres = 3,
yading@10 799 .long_name = NULL_IF_CONFIG_SMALL("H.263 / H.263-1996, H.263+ / H.263-1998 / H.263 version 2"),
yading@10 800 .pix_fmts = ff_h263_hwaccel_pixfmt_list_420,
yading@10 801 };
yading@10 802
yading@10 803 AVCodec ff_h263p_decoder = {
yading@10 804 .name = "h263p",
yading@10 805 .type = AVMEDIA_TYPE_VIDEO,
yading@10 806 .id = AV_CODEC_ID_H263P,
yading@10 807 .priv_data_size = sizeof(MpegEncContext),
yading@10 808 .init = ff_h263_decode_init,
yading@10 809 .close = ff_h263_decode_end,
yading@10 810 .decode = ff_h263_decode_frame,
yading@10 811 .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
yading@10 812 CODEC_CAP_TRUNCATED | CODEC_CAP_DELAY,
yading@10 813 .flush = ff_mpeg_flush,
yading@10 814 .max_lowres = 3,
yading@10 815 .long_name = NULL_IF_CONFIG_SMALL("H.263 / H.263-1996, H.263+ / H.263-1998 / H.263 version 2"),
yading@10 816 .pix_fmts = ff_h263_hwaccel_pixfmt_list_420,
yading@10 817 };