h263dec.c 24.1 KB
Newer Older
Fabrice Bellard's avatar
Fabrice Bellard committed
1
/*
Michael Niedermayer's avatar
Michael Niedermayer committed
2
 * H.263 decoder
3
 * Copyright (c) 2001 Fabrice Bellard
4
 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
Fabrice Bellard's avatar
Fabrice Bellard committed
5
 *
6
 * This file is part of Libav.
7
 *
8
 * Libav is free software; you can redistribute it and/or
Fabrice Bellard's avatar
Fabrice Bellard committed
9 10
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
11
 * version 2.1 of the License, or (at your option) any later version.
Fabrice Bellard's avatar
Fabrice Bellard committed
12
 *
13
 * Libav is distributed in the hope that it will be useful,
Fabrice Bellard's avatar
Fabrice Bellard committed
14
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Fabrice Bellard's avatar
Fabrice Bellard committed
15 16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
Fabrice Bellard's avatar
Fabrice Bellard committed
17
 *
Fabrice Bellard's avatar
Fabrice Bellard committed
18
 * You should have received a copy of the GNU Lesser General Public
19
 * License along with Libav; if not, write to the Free Software
20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
Fabrice Bellard's avatar
Fabrice Bellard committed
21
 */
22

Michael Niedermayer's avatar
Michael Niedermayer committed
23
/**
24
 * @file
Michael Niedermayer's avatar
Michael Niedermayer committed
25 26
 * H.263 decoder.
 */
27

28
#include "libavutil/cpu.h"
29
#include "internal.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
30
#include "avcodec.h"
31
#include "dsputil.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
32
#include "mpegvideo.h"
33
#include "h263.h"
34
#include "h263_parser.h"
35
#include "mpeg4video_parser.h"
36
#include "msmpeg4.h"
37
#include "vdpau_internal.h"
Michael Niedermayer's avatar
Michael Niedermayer committed
38
#include "flv.h"
39
#include "mpeg4video.h"
Fabrice Bellard's avatar
Fabrice Bellard committed
40 41

//#define DEBUG
42
//#define PRINT_FRAME_TIME
Fabrice Bellard's avatar
Fabrice Bellard committed
43

44
av_cold int ff_h263_decode_init(AVCodecContext *avctx)
Fabrice Bellard's avatar
Fabrice Bellard committed
45 46
{
    MpegEncContext *s = avctx->priv_data;
47

Fabrice Bellard's avatar
Fabrice Bellard committed
48
    s->avctx = avctx;
Fabrice Bellard's avatar
Fabrice Bellard committed
49 50
    s->out_format = FMT_H263;

51 52
    s->width  = avctx->coded_width;
    s->height = avctx->coded_height;
53
    s->workaround_bugs= avctx->workaround_bugs;
Fabrice Bellard's avatar
Fabrice Bellard committed
54

55
    // set defaults
Michael Niedermayer's avatar
Michael Niedermayer committed
56
    MPV_decode_defaults(s);
57 58
    s->quant_precision=5;
    s->decode_mb= ff_h263_decode_mb;
Michael Niedermayer's avatar
Michael Niedermayer committed
59
    s->low_delay= 1;
60
    avctx->pix_fmt= avctx->get_format(avctx, avctx->codec->pix_fmts);
61
    s->unrestricted_mv= 1;
62

Fabrice Bellard's avatar
Fabrice Bellard committed
63 64 65
    /* select sub codec */
    switch(avctx->codec->id) {
    case CODEC_ID_H263:
Michael Niedermayer's avatar
Michael Niedermayer committed
66
        s->unrestricted_mv= 0;
67
        avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
Fabrice Bellard's avatar
Fabrice Bellard committed
68
        break;
Fabrice Bellard's avatar
Fabrice Bellard committed
69
    case CODEC_ID_MPEG4:
Fabrice Bellard's avatar
Fabrice Bellard committed
70
        break;
71
    case CODEC_ID_MSMPEG4V1:
Fabrice Bellard's avatar
Fabrice Bellard committed
72
        s->h263_pred = 1;
73 74 75 76 77 78 79 80 81
        s->msmpeg4_version=1;
        break;
    case CODEC_ID_MSMPEG4V2:
        s->h263_pred = 1;
        s->msmpeg4_version=2;
        break;
    case CODEC_ID_MSMPEG4V3:
        s->h263_pred = 1;
        s->msmpeg4_version=3;
Fabrice Bellard's avatar
Fabrice Bellard committed
82
        break;
83 84 85 86
    case CODEC_ID_WMV1:
        s->h263_pred = 1;
        s->msmpeg4_version=4;
        break;
87 88 89 90
    case CODEC_ID_WMV2:
        s->h263_pred = 1;
        s->msmpeg4_version=5;
        break;
91
    case CODEC_ID_VC1:
anonymous's avatar
anonymous committed
92 93 94
    case CODEC_ID_WMV3:
        s->h263_pred = 1;
        s->msmpeg4_version=6;
95
        avctx->chroma_sample_location = AVCHROMA_LOC_LEFT;
anonymous's avatar
anonymous committed
96
        break;
Fabrice Bellard's avatar
Fabrice Bellard committed
97 98
    case CODEC_ID_H263I:
        break;
99 100 101
    case CODEC_ID_FLV1:
        s->h263_flv = 1;
        break;
Fabrice Bellard's avatar
Fabrice Bellard committed
102 103 104
    default:
        return -1;
    }
105
    s->codec_id= avctx->codec->id;
106
    avctx->hwaccel= ff_find_hwaccel(avctx->codec->id, avctx->pix_fmt);
107

Fabrice Bellard's avatar
Fabrice Bellard committed
108
    /* for h263, we allocate the images after having read the header */
109
    if (avctx->codec->id != CODEC_ID_H263 && avctx->codec->id != CODEC_ID_MPEG4)
110 111
        if (MPV_common_init(s) < 0)
            return -1;
Fabrice Bellard's avatar
Fabrice Bellard committed
112 113

        h263_decode_init_vlc(s);
114

Fabrice Bellard's avatar
Fabrice Bellard committed
115 116 117
    return 0;
}

118
av_cold int ff_h263_decode_end(AVCodecContext *avctx)
Fabrice Bellard's avatar
Fabrice Bellard committed
119 120 121 122 123 124 125
{
    MpegEncContext *s = avctx->priv_data;

    MPV_common_end(s);
    return 0;
}

Michael Niedermayer's avatar
Michael Niedermayer committed
126
/**
Diego Biurrun's avatar
Diego Biurrun committed
127
 * returns the number of bytes consumed for building the current frame
Michael Niedermayer's avatar
Michael Niedermayer committed
128 129 130
 */
static int get_consumed_bytes(MpegEncContext *s, int buf_size){
    int pos= (get_bits_count(&s->gb)+7)>>3;
131

132
    if(s->divx_packed || s->avctx->hwaccel){
Michael Niedermayer's avatar
Michael Niedermayer committed
133
        //we would have to scan through the whole buf to handle the weird reordering ...
134
        return buf_size;
135 136 137 138
    }else if(s->flags&CODEC_FLAG_TRUNCATED){
        pos -= s->parse_context.last_index;
        if(pos<0) pos=0; // padding is not really read so this might be -1
        return pos;
Michael Niedermayer's avatar
Michael Niedermayer committed
139
    }else{
Diego Biurrun's avatar
Diego Biurrun committed
140
        if(pos==0) pos=1; //avoid infinite loops (i doubt that is needed but ...)
141
        if(pos+10>buf_size) pos=buf_size; // oops ;)
Michael Niedermayer's avatar
Michael Niedermayer committed
142 143 144 145 146

        return pos;
    }
}

147
static int decode_slice(MpegEncContext *s){
148
    const int part_mask= s->partitioned_frame ? (AC_END|AC_ERROR) : 0x7F;
Michael Niedermayer's avatar
Michael Niedermayer committed
149
    const int mb_size= 16>>s->avctx->lowres;
150 151
    s->last_resync_gb= s->gb;
    s->first_slice_line= 1;
152

153 154 155
    s->resync_mb_x= s->mb_x;
    s->resync_mb_y= s->mb_y;

Michael Niedermayer's avatar
cleanup  
Michael Niedermayer committed
156
    ff_set_qscale(s, s->qscale);
157

158 159 160 161 162 163
    if (s->avctx->hwaccel) {
        const uint8_t *start= s->gb.buffer + get_bits_count(&s->gb)/8;
        const uint8_t *end  = ff_h263_find_resync_marker(start + 1, s->gb.buffer_end);
        skip_bits_long(&s->gb, 8*(end - start));
        return s->avctx->hwaccel->decode_slice(s->avctx, start, end - start);
    }
164

165 166 167
    if(s->partitioned_frame){
        const int qscale= s->qscale;

168
        if(CONFIG_MPEG4_DECODER && s->codec_id==CODEC_ID_MPEG4){
169
            if(ff_mpeg4_decode_partitions(s) < 0)
170
                return -1;
171
        }
172

Fabrice Bellard's avatar
Fabrice Bellard committed
173
        /* restore variables which were modified */
174 175 176
        s->first_slice_line=1;
        s->mb_x= s->resync_mb_x;
        s->mb_y= s->resync_mb_y;
Michael Niedermayer's avatar
cleanup  
Michael Niedermayer committed
177
        ff_set_qscale(s, qscale);
178 179 180 181 182 183
    }

    for(; s->mb_y < s->mb_height; s->mb_y++) {
        /* per-row end of slice checks */
        if(s->msmpeg4_version){
            if(s->resync_mb_y + s->slice_height == s->mb_y){
184 185
                ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, AC_END|DC_END|MV_END);

186 187 188
                return 0;
            }
        }
189

190 191 192 193 194
        if(s->msmpeg4_version==1){
            s->last_dc[0]=
            s->last_dc[1]=
            s->last_dc[2]= 128;
        }
195

196 197 198 199 200 201 202
        ff_init_block_index(s);
        for(; s->mb_x < s->mb_width; s->mb_x++) {
            int ret;

            ff_update_block_index(s);

            if(s->resync_mb_x == s->mb_x && s->resync_mb_y+1 == s->mb_y){
203
                s->first_slice_line=0;
204 205 206
            }

            /* DCT & quantize */
207

208 209
            s->mv_dir = MV_DIR_FORWARD;
            s->mv_type = MV_TYPE_16X16;
210
//            s->mb_skipped = 0;
211 212 213
//printf("%d %d %06X\n", ret, get_bits_count(&s->gb), show_bits(&s->gb, 24));
            ret= s->decode_mb(s, s->block);

214
            if (s->pict_type!=AV_PICTURE_TYPE_B)
215 216
                ff_h263_update_motion_val(s);

217
            if(ret<0){
218
                const int xy= s->mb_x + s->mb_y*s->mb_stride;
219
                if(ret==SLICE_END){
220
                    MPV_decode_mb(s, s->block);
Michael Niedermayer's avatar
Michael Niedermayer committed
221 222
                    if(s->loop_filter)
                        ff_h263_loop_filter(s);
223

224
//printf("%d %d %d %06X\n", s->mb_x, s->mb_y, s->gb.size*8 - get_bits_count(&s->gb), show_bits(&s->gb, 24));
225
                    ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, (AC_END|DC_END|MV_END)&part_mask);
226 227

                    s->padding_bug_score--;
228

229 230
                    if(++s->mb_x >= s->mb_width){
                        s->mb_x=0;
Michael Niedermayer's avatar
Michael Niedermayer committed
231
                        ff_draw_horiz_band(s, s->mb_y*mb_size, mb_size);
232 233
                        s->mb_y++;
                    }
234
                    return 0;
235
                }else if(ret==SLICE_NOEND){
236
                    av_log(s->avctx, AV_LOG_ERROR, "Slice mismatch at MB: %d\n", xy);
237
                    ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x+1, s->mb_y, (AC_END|DC_END|MV_END)&part_mask);
238 239
                    return -1;
                }
240
                av_log(s->avctx, AV_LOG_ERROR, "Error at MB: %d\n", xy);
241
                ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, (AC_ERROR|DC_ERROR|MV_ERROR)&part_mask);
242

243 244
                return -1;
            }
245 246

            MPV_decode_mb(s, s->block);
Michael Niedermayer's avatar
Michael Niedermayer committed
247 248
            if(s->loop_filter)
                ff_h263_loop_filter(s);
249
        }
250

Michael Niedermayer's avatar
Michael Niedermayer committed
251
        ff_draw_horiz_band(s, s->mb_y*mb_size, mb_size);
252

253 254
        s->mb_x= 0;
    }
255

256 257
    assert(s->mb_x==0 && s->mb_y==s->mb_height);

258 259 260 261 262 263 264
    if(s->codec_id==CODEC_ID_MPEG4
       && (s->workaround_bugs&FF_BUG_AUTODETECT)
       && get_bits_left(&s->gb) >= 48
       && show_bits(&s->gb, 24)==0x4010
       && !s->data_partitioning)
        s->padding_bug_score+=32;

265 266
    /* try to detect the padding bug */
    if(      s->codec_id==CODEC_ID_MPEG4
267
       &&   (s->workaround_bugs&FF_BUG_AUTODETECT)
268 269
       &&    get_bits_left(&s->gb) >=0
       &&    get_bits_left(&s->gb) < 48
270
//       &&   !s->resync_marker
271
       &&   !s->data_partitioning){
272

273
        const int bits_count= get_bits_count(&s->gb);
274
        const int bits_left = s->gb.size_in_bits - bits_count;
275

276 277
        if(bits_left==0){
            s->padding_bug_score+=16;
278
        } else if(bits_left != 1){
279 280
            int v= show_bits(&s->gb, 8);
            v|= 0x7F >> (7-(bits_count&7));
281

282
            if(v==0x7F && bits_left<=8)
283
                s->padding_bug_score--;
284 285
            else if(v==0x7F && ((get_bits_count(&s->gb)+8)&8) && bits_left<=16)
                s->padding_bug_score+= 4;
286
            else
287 288
                s->padding_bug_score++;
        }
289
    }
290

291
    if(s->workaround_bugs&FF_BUG_AUTODETECT){
292
        if(s->padding_bug_score > -2 && !s->data_partitioning /*&& (s->divx_version>=0 || !s->resync_marker)*/)
293 294 295 296
            s->workaround_bugs |=  FF_BUG_NO_PADDING;
        else
            s->workaround_bugs &= ~FF_BUG_NO_PADDING;
    }
297

Diego Biurrun's avatar
Diego Biurrun committed
298
    // handle formats which don't have unique end markers
299
    if(s->msmpeg4_version || (s->workaround_bugs&FF_BUG_NO_PADDING)){ //FIXME perhaps solve this more cleanly
300
        int left= get_bits_left(&s->gb);
301
        int max_extra=7;
302

303
        /* no markers in M$ crap */
304
        if(s->msmpeg4_version && s->pict_type==AV_PICTURE_TYPE_I)
305
            max_extra+= 17;
306

307
        /* buggy padding but the frame should still end approximately at the bitstream end */
308
        if((s->workaround_bugs&FF_BUG_NO_PADDING) && s->error_recognition>=3)
309
            max_extra+= 48;
310
        else if((s->workaround_bugs&FF_BUG_NO_PADDING))
311
            max_extra+= 256*256*256*64;
312

313
        if(left>max_extra){
314
            av_log(s->avctx, AV_LOG_ERROR, "discarding %d junk bits at end, next would be %X\n", left, show_bits(&s->gb, 24));
315 316
        }
        else if(left<0){
317
            av_log(s->avctx, AV_LOG_ERROR, "overreading %d bits\n", -left);
318
        }else
319
            ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, AC_END|DC_END|MV_END);
320

321 322
        return 0;
    }
323

324
    av_log(s->avctx, AV_LOG_ERROR, "slice end not reached but screenspace end (%d left %06X, score= %d)\n",
325
            get_bits_left(&s->gb),
326
            show_bits(&s->gb, 24), s->padding_bug_score);
327

328 329
    ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, (AC_END|DC_END|MV_END)&part_mask);

330 331 332
    return -1;
}

333
int ff_h263_decode_frame(AVCodecContext *avctx,
Fabrice Bellard's avatar
Fabrice Bellard committed
334
                             void *data, int *data_size,
335
                             AVPacket *avpkt)
Fabrice Bellard's avatar
Fabrice Bellard committed
336
{
337 338
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
Fabrice Bellard's avatar
Fabrice Bellard committed
339
    MpegEncContext *s = avctx->priv_data;
Fabrice Bellard's avatar
Fabrice Bellard committed
340
    int ret;
341 342
    AVFrame *pict = data;

343 344
#ifdef PRINT_FRAME_TIME
uint64_t time= rdtsc();
Fabrice Bellard's avatar
Fabrice Bellard committed
345
#endif
Michael Niedermayer's avatar
Michael Niedermayer committed
346
    s->flags= avctx->flags;
347
    s->flags2= avctx->flags2;
348

Michael Niedermayer's avatar
Michael Niedermayer committed
349
    /* no supplementary picture */
Fabrice Bellard's avatar
Fabrice Bellard committed
350
    if (buf_size == 0) {
Michael Niedermayer's avatar
Michael Niedermayer committed
351 352 353 354 355 356 357 358
        /* special case for last picture */
        if (s->low_delay==0 && s->next_picture_ptr) {
            *pict= *(AVFrame*)s->next_picture_ptr;
            s->next_picture_ptr= NULL;

            *data_size = sizeof(AVFrame);
        }

Fabrice Bellard's avatar
Fabrice Bellard committed
359 360
        return 0;
    }
Michael Niedermayer's avatar
cleanup  
Michael Niedermayer committed
361

362 363
    if(s->flags&CODEC_FLAG_TRUNCATED){
        int next;
364

365
        if(CONFIG_MPEG4_DECODER && s->codec_id==CODEC_ID_MPEG4){
366
            next= ff_mpeg4_find_frame_end(&s->parse_context, buf, buf_size);
367
        }else if(CONFIG_H263_DECODER && s->codec_id==CODEC_ID_H263){
368
            next= ff_h263_find_frame_end(&s->parse_context, buf, buf_size);
369
        }else{
370
            av_log(s->avctx, AV_LOG_ERROR, "this codec does not support truncated bitstreams\n");
371 372
            return -1;
        }
373

374
        if( ff_combine_frame(&s->parse_context, next, (const uint8_t **)&buf, &buf_size) < 0 )
375 376
            return buf_size;
    }
Fabrice Bellard's avatar
Fabrice Bellard committed
377

378

379
retry:
380

381
    if(s->bitstream_buffer_size && (s->divx_packed || buf_size<20)){ //divx 5.01+/xvid frame reorder
382
        init_get_bits(&s->gb, s->bitstream_buffer, s->bitstream_buffer_size*8);
Michael Niedermayer's avatar
Michael Niedermayer committed
383
    }else
384
        init_get_bits(&s->gb, buf, buf_size*8);
385
    s->bitstream_buffer_size=0;
Fabrice Bellard's avatar
Fabrice Bellard committed
386

387
    if (!s->context_initialized) {
388
        if (MPV_common_init(s) < 0) //we need the idct permutaton for reading a custom matrix
389 390
            return -1;
    }
391

Diego Biurrun's avatar
Diego Biurrun committed
392 393
    /* We need to set current_picture_ptr before reading the header,
     * otherwise we cannot store anyting in there */
394 395 396 397
    if(s->current_picture_ptr==NULL || s->current_picture_ptr->data[0]){
        int i= ff_find_unused_picture(s, 0);
        s->current_picture_ptr= &s->picture[i];
    }
398

Fabrice Bellard's avatar
Fabrice Bellard committed
399
    /* let's go :-) */
400
    if (CONFIG_WMV2_DECODER && s->msmpeg4_version==5) {
Michael Niedermayer's avatar
Michael Niedermayer committed
401
        ret= ff_wmv2_decode_picture_header(s);
402
    } else if (CONFIG_MSMPEG4_DECODER && s->msmpeg4_version) {
Fabrice Bellard's avatar
Fabrice Bellard committed
403
        ret = msmpeg4_decode_picture_header(s);
404
    } else if (CONFIG_MPEG4_DECODER && s->h263_pred) {
405 406
        if(s->avctx->extradata_size && s->picture_number==0){
            GetBitContext gb;
407

408
            init_get_bits(&gb, s->avctx->extradata, s->avctx->extradata_size*8);
409 410 411
            ret = ff_mpeg4_decode_picture_header(s, &gb);
        }
        ret = ff_mpeg4_decode_picture_header(s, &s->gb);
412
    } else if (CONFIG_H263I_DECODER && s->codec_id == CODEC_ID_H263I) {
413
        ret = ff_intel_h263_decode_picture_header(s);
Michael Niedermayer's avatar
Michael Niedermayer committed
414 415
    } else if (CONFIG_FLV_DECODER && s->h263_flv) {
        ret = ff_flv_decode_picture_header(s);
Fabrice Bellard's avatar
Fabrice Bellard committed
416 417
    } else {
        ret = h263_decode_picture_header(s);
418
    }
419

420
    if(ret==FRAME_SKIPPED) return get_consumed_bytes(s, buf_size);
421 422 423

    /* skip if the header was thrashed */
    if (ret < 0){
424
        av_log(s->avctx, AV_LOG_ERROR, "header damaged\n");
425 426
        return -1;
    }
427

Michael Niedermayer's avatar
Michael Niedermayer committed
428
    avctx->has_b_frames= !s->low_delay;
429

430
    if(s->xvid_build==-1 && s->divx_version==-1 && s->lavc_build==-1){
431 432
        if(s->stream_codec_tag == AV_RL32("XVID") ||
           s->codec_tag == AV_RL32("XVID") || s->codec_tag == AV_RL32("XVIX") ||
433 434 435
           s->codec_tag == AV_RL32("RMP4") ||
           s->codec_tag == AV_RL32("SIPP")
           )
436
            s->xvid_build= 0;
437
#if 0
438
        if(s->codec_tag == AV_RL32("DIVX") && s->vo_type==0 && s->vol_control_parameters==1
439
           && s->padding_bug_score > 0 && s->low_delay) // XVID with modified fourcc
440
            s->xvid_build= 0;
441 442
#endif
    }
443

444
    if(s->xvid_build==-1 && s->divx_version==-1 && s->lavc_build==-1){
445
        if(s->codec_tag == AV_RL32("DIVX") && s->vo_type==0 && s->vol_control_parameters==0)
446 447
            s->divx_version= 400; //divx 4
    }
448

449
    if(s->xvid_build>=0 && s->divx_version>=0){
450
        s->divx_version=
451
        s->divx_build= -1;
452
    }
453 454

    if(s->workaround_bugs&FF_BUG_AUTODETECT){
455
        if(s->codec_tag == AV_RL32("XVIX"))
456
            s->workaround_bugs|= FF_BUG_XVID_ILACE;
457

458
        if(s->codec_tag == AV_RL32("UMP4")){
459 460
            s->workaround_bugs|= FF_BUG_UMP4;
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
461

462
        if(s->divx_version>=500 && s->divx_build<1814){
Michael Niedermayer's avatar
Michael Niedermayer committed
463 464
            s->workaround_bugs|= FF_BUG_QPEL_CHROMA;
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
465

466
        if(s->divx_version>502 && s->divx_build<1814){
Michael Niedermayer's avatar
Michael Niedermayer committed
467 468 469
            s->workaround_bugs|= FF_BUG_QPEL_CHROMA2;
        }

470
        if(s->xvid_build<=3U)
471
            s->padding_bug_score= 256*256*256*64;
472

473
        if(s->xvid_build<=1U)
Michael Niedermayer's avatar
Michael Niedermayer committed
474 475
            s->workaround_bugs|= FF_BUG_QPEL_CHROMA;

476
        if(s->xvid_build<=12U)
477 478
            s->workaround_bugs|= FF_BUG_EDGE;

479
        if(s->xvid_build<=32U)
480 481
            s->workaround_bugs|= FF_BUG_DC_CLIP;

Michael Niedermayer's avatar
Michael Niedermayer committed
482 483 484 485 486
#define SET_QPEL_FUNC(postfix1, postfix2) \
    s->dsp.put_ ## postfix1 = ff_put_ ## postfix2;\
    s->dsp.put_no_rnd_ ## postfix1 = ff_put_no_rnd_ ## postfix2;\
    s->dsp.avg_ ## postfix1 = ff_avg_ ## postfix2;

487
        if(s->lavc_build<4653U)
Michael Niedermayer's avatar
Michael Niedermayer committed
488
            s->workaround_bugs|= FF_BUG_STD_QPEL;
489

490
        if(s->lavc_build<4655U)
491 492
            s->workaround_bugs|= FF_BUG_DIRECT_BLOCKSIZE;

493
        if(s->lavc_build<4670U){
494 495
            s->workaround_bugs|= FF_BUG_EDGE;
        }
496

497
        if(s->lavc_build<=4712U)
498
            s->workaround_bugs|= FF_BUG_DC_CLIP;
499

500
        if(s->divx_version>=0)
501
            s->workaround_bugs|= FF_BUG_DIRECT_BLOCKSIZE;
502
//printf("padding_bug_score: %d\n", s->padding_bug_score);
503 504
        if(s->divx_version==501 && s->divx_build==20020416)
            s->padding_bug_score= 256*256*256*64;
505

506
        if(s->divx_version<500U){
507 508
            s->workaround_bugs|= FF_BUG_EDGE;
        }
509

510
        if(s->divx_version>=0)
511
            s->workaround_bugs|= FF_BUG_HPEL_CHROMA;
512
#if 0
513
        if(s->divx_version==500)
514
            s->padding_bug_score= 256*256*256*64;
515 516

        /* very ugly XVID padding bug detection FIXME/XXX solve this differently
517
         * Let us hope this at least works.
518
         */
519
        if(   s->resync_marker==0 && s->data_partitioning==0 && s->divx_version==-1
520 521
           && s->codec_id==CODEC_ID_MPEG4 && s->vo_type==0)
            s->workaround_bugs|= FF_BUG_NO_PADDING;
522

523
        if(s->lavc_build<4609U) //FIXME not sure about the version num but a 4609 file seems ok
524
            s->workaround_bugs|= FF_BUG_NO_PADDING;
525
#endif
526
    }
527

Michael Niedermayer's avatar
Michael Niedermayer committed
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542
    if(s->workaround_bugs& FF_BUG_STD_QPEL){
        SET_QPEL_FUNC(qpel_pixels_tab[0][ 5], qpel16_mc11_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[0][ 7], qpel16_mc31_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[0][ 9], qpel16_mc12_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[0][11], qpel16_mc32_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[0][13], qpel16_mc13_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[0][15], qpel16_mc33_old_c)

        SET_QPEL_FUNC(qpel_pixels_tab[1][ 5], qpel8_mc11_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[1][ 7], qpel8_mc31_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[1][ 9], qpel8_mc12_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[1][11], qpel8_mc32_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[1][13], qpel8_mc13_old_c)
        SET_QPEL_FUNC(qpel_pixels_tab[1][15], qpel8_mc33_old_c)
    }
543

544
    if(avctx->debug & FF_DEBUG_BUGS)
545
        av_log(s->avctx, AV_LOG_DEBUG, "bugs: %X lavc_build:%d xvid_build:%d divx_version:%d divx_build:%d %s\n",
546 547
               s->workaround_bugs, s->lavc_build, s->xvid_build, s->divx_version, s->divx_build,
               s->divx_packed ? "p" : "");
548

549
#if HAVE_MMX
550
    if (s->codec_id == CODEC_ID_MPEG4 && s->xvid_build>=0 && avctx->idct_algo == FF_IDCT_AUTO && (av_get_cpu_flags() & AV_CPU_FLAG_MMX)) {
551
        avctx->idct_algo= FF_IDCT_XVIDMMX;
Michael Niedermayer's avatar
Michael Niedermayer committed
552
        avctx->coded_width= 0; // force reinit
553 554
//        dsputil_init(&s->dsp, avctx);
        s->picture_number=0;
555
    }
Alex Beregszaszi's avatar
Alex Beregszaszi committed
556 557
#endif

558
        /* After H263 & mpeg4 header decode we have the height, width,*/
559 560 561
        /* and other parameters. So then we could init the picture   */
        /* FIXME: By the way H263 decoder is evolving it should have */
        /* an H263EncContext                                         */
562 563

    if (   s->width  != avctx->coded_width
564
        || s->height != avctx->coded_height) {
565
        /* H.263 could change picture size any time */
566 567
        ParseContext pc= s->parse_context; //FIXME move these demuxng hack to avformat
        s->parse_context.buffer=0;
568
        MPV_common_end(s);
569
        s->parse_context= pc;
570
    }
571
    if (!s->context_initialized) {
572
        avcodec_set_dimensions(avctx, s->width, s->height);
573

574
        goto retry;
Fabrice Bellard's avatar
Fabrice Bellard committed
575
    }
576

577
    if((s->codec_id==CODEC_ID_H263 || s->codec_id==CODEC_ID_H263P || s->codec_id == CODEC_ID_H263I))
578
        s->gob_index = ff_h263_get_gob_height(s);
579

580
    // for skipping the frame
Michael Niedermayer's avatar
cleanup  
Michael Niedermayer committed
581
    s->current_picture.pict_type= s->pict_type;
582
    s->current_picture.key_frame= s->pict_type == AV_PICTURE_TYPE_I;
583

Diego Biurrun's avatar
Diego Biurrun committed
584
    /* skip B-frames if we don't have reference frames */
585 586 587
    if(s->last_picture_ptr==NULL && (s->pict_type==AV_PICTURE_TYPE_B || s->dropable)) return get_consumed_bytes(s, buf_size);
    if(   (avctx->skip_frame >= AVDISCARD_NONREF && s->pict_type==AV_PICTURE_TYPE_B)
       || (avctx->skip_frame >= AVDISCARD_NONKEY && s->pict_type!=AV_PICTURE_TYPE_I)
588
       ||  avctx->skip_frame >= AVDISCARD_ALL)
Michael Niedermayer's avatar
Michael Niedermayer committed
589
        return get_consumed_bytes(s, buf_size);
590

591
    if(s->next_p_frame_damaged){
592
        if(s->pict_type==AV_PICTURE_TYPE_B)
Michael Niedermayer's avatar
Michael Niedermayer committed
593
            return get_consumed_bytes(s, buf_size);
594 595 596 597
        else
            s->next_p_frame_damaged=0;
    }

598
    if((s->avctx->flags2 & CODEC_FLAG2_FAST) && s->pict_type==AV_PICTURE_TYPE_B){
599 600
        s->me.qpel_put= s->dsp.put_2tap_qpel_pixels_tab;
        s->me.qpel_avg= s->dsp.avg_2tap_qpel_pixels_tab;
601
    }else if((!s->no_rounding) || s->pict_type==AV_PICTURE_TYPE_B){
Loren Merritt's avatar
Loren Merritt committed
602
        s->me.qpel_put= s->dsp.put_qpel_pixels_tab;
603 604
        s->me.qpel_avg= s->dsp.avg_qpel_pixels_tab;
    }else{
Loren Merritt's avatar
Loren Merritt committed
605
        s->me.qpel_put= s->dsp.put_no_rnd_qpel_pixels_tab;
606 607 608
        s->me.qpel_avg= s->dsp.avg_qpel_pixels_tab;
    }

609 610
    if(MPV_frame_start(s, avctx) < 0)
        return -1;
Fabrice Bellard's avatar
Fabrice Bellard committed
611

612
    if (CONFIG_MPEG4_VDPAU_DECODER && (s->avctx->codec->capabilities & CODEC_CAP_HWACCEL_VDPAU)) {
613
        ff_vdpau_mpeg4_decode_picture(s, s->gb.buffer, s->gb.buffer_end - s->gb.buffer);
614 615 616
        goto frame_end;
    }

617
    if (avctx->hwaccel) {
618
        if (avctx->hwaccel->start_frame(avctx, s->gb.buffer, s->gb.buffer_end - s->gb.buffer) < 0)
619 620 621
            return -1;
    }

622
    ff_er_frame_start(s);
623

Michael Niedermayer's avatar
Michael Niedermayer committed
624
    //the second part of the wmv2 header contains the MB skip bits which are stored in current_picture->mb_type
Diego Biurrun's avatar
Diego Biurrun committed
625
    //which is not available before MPV_frame_start()
626
    if (CONFIG_WMV2_DECODER && s->msmpeg4_version==5){
627 628 629
        ret = ff_wmv2_decode_secondary_picture_header(s);
        if(ret<0) return ret;
        if(ret==1) goto intrax8_decoded;
Michael Niedermayer's avatar
Michael Niedermayer committed
630 631
    }

Fabrice Bellard's avatar
Fabrice Bellard committed
632
    /* decode each macroblock */
633
    s->mb_x=0;
634
    s->mb_y=0;
635

636
    decode_slice(s);
Michael Niedermayer's avatar
Michael Niedermayer committed
637
    while(s->mb_y<s->mb_height){
638
        if(s->msmpeg4_version){
Michael Niedermayer's avatar
1/0 fix  
Michael Niedermayer committed
639
            if(s->slice_height==0 || s->mb_x!=0 || (s->mb_y%s->slice_height)!=0 || get_bits_count(&s