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

/**
23
 * @file
Kostya Shishkov's avatar
Kostya Shishkov committed
24 25 26 27 28 29
 * Indeo Video Interactive version 5 decoder
 *
 * Indeo5 data is usually transported within .avi or .mov files.
 * Known FOURCCs: 'IV50'
 */

30
#define BITSTREAM_READER_LE
Kostya Shishkov's avatar
Kostya Shishkov committed
31 32
#include "avcodec.h"
#include "get_bits.h"
33
#include "dsputil.h"
Kostya Shishkov's avatar
Kostya Shishkov committed
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
#include "ivi_dsp.h"
#include "ivi_common.h"
#include "indeo5data.h"

/**
 *  Indeo5 frame types.
 */
enum {
    FRAMETYPE_INTRA       = 0,
    FRAMETYPE_INTER       = 1,  ///< non-droppable P-frame
    FRAMETYPE_INTER_SCAL  = 2,  ///< droppable P-frame used in the scalability mode
    FRAMETYPE_INTER_NOREF = 3,  ///< droppable P-frame
    FRAMETYPE_NULL        = 4   ///< empty frame with no data
};

#define IVI5_PIC_SIZE_ESC       15

/**
52
 *  Decode Indeo5 GOP (Group of pictures) header.
Kostya Shishkov's avatar
Kostya Shishkov committed
53 54 55
 *  This header is present in key frames only.
 *  It defines parameters for all frames in a GOP.
 *
56 57
 *  @param[in,out] ctx    ptr to the decoder context
 *  @param[in]     avctx  ptr to the AVCodecContext
Kostya Shishkov's avatar
Kostya Shishkov committed
58 59
 *  @return         result code: 0 = OK, -1 = error
 */
60
static int decode_gop_header(IVI45DecContext *ctx, AVCodecContext *avctx)
Kostya Shishkov's avatar
Kostya Shishkov committed
61
{
62
    int             result, i, p, tile_size, pic_size_indx, mb_size, blk_size, is_scalable;
63
    int             quant_mat, blk_size_changed = 0;
Kostya Shishkov's avatar
Kostya Shishkov committed
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
    IVIBandDesc     *band, *band1, *band2;
    IVIPicConfig    pic_conf;

    ctx->gop_flags = get_bits(&ctx->gb, 8);

    ctx->gop_hdr_size = (ctx->gop_flags & 1) ? get_bits(&ctx->gb, 16) : 0;

    if (ctx->gop_flags & IVI5_IS_PROTECTED)
        ctx->lock_word = get_bits_long(&ctx->gb, 32);

    tile_size = (ctx->gop_flags & 0x40) ? 64 << get_bits(&ctx->gb, 2) : 0;
    if (tile_size > 256) {
        av_log(avctx, AV_LOG_ERROR, "Invalid tile size: %d\n", tile_size);
        return -1;
    }

    /* decode number of wavelet bands */
    /* num_levels * 3 + 1 */
    pic_conf.luma_bands   = get_bits(&ctx->gb, 2) * 3 + 1;
    pic_conf.chroma_bands = get_bits1(&ctx->gb)   * 3 + 1;
84 85
    is_scalable = pic_conf.luma_bands != 1 || pic_conf.chroma_bands != 1;
    if (is_scalable && (pic_conf.luma_bands != 4 || pic_conf.chroma_bands != 1)) {
Kostya Shishkov's avatar
Kostya Shishkov committed
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
        av_log(avctx, AV_LOG_ERROR, "Scalability: unsupported subdivision! Luma bands: %d, chroma bands: %d\n",
               pic_conf.luma_bands, pic_conf.chroma_bands);
        return -1;
    }

    pic_size_indx = get_bits(&ctx->gb, 4);
    if (pic_size_indx == IVI5_PIC_SIZE_ESC) {
        pic_conf.pic_height = get_bits(&ctx->gb, 13);
        pic_conf.pic_width  = get_bits(&ctx->gb, 13);
    } else {
        pic_conf.pic_height = ivi5_common_pic_sizes[pic_size_indx * 2 + 1] << 2;
        pic_conf.pic_width  = ivi5_common_pic_sizes[pic_size_indx * 2    ] << 2;
    }

    if (ctx->gop_flags & 2) {
        av_log(avctx, AV_LOG_ERROR, "YV12 picture format not supported!\n");
        return -1;
    }

    pic_conf.chroma_height = (pic_conf.pic_height + 3) >> 2;
    pic_conf.chroma_width  = (pic_conf.pic_width  + 3) >> 2;

    if (!tile_size) {
        pic_conf.tile_height = pic_conf.pic_height;
        pic_conf.tile_width  = pic_conf.pic_width;
    } else {
        pic_conf.tile_height = pic_conf.tile_width = tile_size;
    }

    /* check if picture layout was changed and reallocate buffers */
    if (ivi_pic_config_cmp(&pic_conf, &ctx->pic_conf)) {
        result = ff_ivi_init_planes(ctx->planes, &pic_conf);
        if (result) {
            av_log(avctx, AV_LOG_ERROR, "Couldn't reallocate color planes!\n");
            return -1;
        }
        ctx->pic_conf = pic_conf;
123
        ctx->is_scalable = is_scalable;
Kostya Shishkov's avatar
Kostya Shishkov committed
124 125 126 127 128 129 130 131 132 133 134 135 136
        blk_size_changed = 1; /* force reallocation of the internal structures */
    }

    for (p = 0; p <= 1; p++) {
        for (i = 0; i < (!p ? pic_conf.luma_bands : pic_conf.chroma_bands); i++) {
            band = &ctx->planes[p].bands[i];

            band->is_halfpel = get_bits1(&ctx->gb);

            mb_size  = get_bits1(&ctx->gb);
            blk_size = 8 >> get_bits1(&ctx->gb);
            mb_size  = blk_size << !mb_size;

137 138 139 140 141
            if (p==0 && blk_size==4) {
                av_log(avctx, AV_LOG_ERROR, "4x4 luma blocks are unsupported!\n");
                return AVERROR_PATCHWELCOME;
            }

Kostya Shishkov's avatar
Kostya Shishkov committed
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157
            blk_size_changed = mb_size != band->mb_size || blk_size != band->blk_size;
            if (blk_size_changed) {
                band->mb_size  = mb_size;
                band->blk_size = blk_size;
            }

            if (get_bits1(&ctx->gb)) {
                av_log(avctx, AV_LOG_ERROR, "Extended transform info encountered!\n");
                return -1;
            }

            /* select transform function and scan pattern according to plane and band number */
            switch ((p << 2) + i) {
            case 0:
                band->inv_transform = ff_ivi_inverse_slant_8x8;
                band->dc_transform  = ff_ivi_dc_slant_2d;
158
                band->scan          = ff_zigzag_direct;
159
                band->transform_size= 8;
Kostya Shishkov's avatar
Kostya Shishkov committed
160 161 162 163 164
                break;

            case 1:
                band->inv_transform = ff_ivi_row_slant8;
                band->dc_transform  = ff_ivi_dc_row_slant;
165
                band->scan          = ff_ivi_vertical_scan_8x8;
166
                band->transform_size= 8;
Kostya Shishkov's avatar
Kostya Shishkov committed
167 168 169 170 171
                break;

            case 2:
                band->inv_transform = ff_ivi_col_slant8;
                band->dc_transform  = ff_ivi_dc_col_slant;
172
                band->scan          = ff_ivi_horizontal_scan_8x8;
173
                band->transform_size= 8;
Kostya Shishkov's avatar
Kostya Shishkov committed
174 175 176 177 178
                break;

            case 3:
                band->inv_transform = ff_ivi_put_pixels_8x8;
                band->dc_transform  = ff_ivi_put_dc_pixel_8x8;
179
                band->scan          = ff_ivi_horizontal_scan_8x8;
180
                band->transform_size= 8;
Kostya Shishkov's avatar
Kostya Shishkov committed
181 182 183 184 185
                break;

            case 4:
                band->inv_transform = ff_ivi_inverse_slant_4x4;
                band->dc_transform  = ff_ivi_dc_slant_2d;
186
                band->scan          = ff_ivi_direct_scan_4x4;
187
                band->transform_size= 4;
Kostya Shishkov's avatar
Kostya Shishkov committed
188 189 190 191 192 193 194 195
                break;
            }

            band->is_2d_trans = band->inv_transform == ff_ivi_inverse_slant_8x8 ||
                                band->inv_transform == ff_ivi_inverse_slant_4x4;

            /* select dequant matrix according to plane and band number */
            if (!p) {
196
                quant_mat = (pic_conf.luma_bands > 1) ? i+1 : 0;
Kostya Shishkov's avatar
Kostya Shishkov committed
197
            } else {
198 199 200 201
                quant_mat = 5;
            }

            if (band->blk_size == 8) {
202 203 204 205
                if(quant_mat >= 5){
                    av_log(avctx, AV_LOG_ERROR, "quant_mat %d too large!\n", quant_mat);
                    return -1;
                }
206 207 208 209 210 211 212 213 214
                band->intra_base  = &ivi5_base_quant_8x8_intra[quant_mat][0];
                band->inter_base  = &ivi5_base_quant_8x8_inter[quant_mat][0];
                band->intra_scale = &ivi5_scale_quant_8x8_intra[quant_mat][0];
                band->inter_scale = &ivi5_scale_quant_8x8_inter[quant_mat][0];
            } else {
                band->intra_base  = ivi5_base_quant_4x4_intra;
                band->inter_base  = ivi5_base_quant_4x4_inter;
                band->intra_scale = ivi5_scale_quant_4x4_intra;
                band->inter_scale = ivi5_scale_quant_4x4_inter;
Kostya Shishkov's avatar
Kostya Shishkov committed
215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
            }

            if (get_bits(&ctx->gb, 2)) {
                av_log(avctx, AV_LOG_ERROR, "End marker missing!\n");
                return -1;
            }
        }
    }

    /* copy chroma parameters into the 2nd chroma plane */
    for (i = 0; i < pic_conf.chroma_bands; i++) {
        band1 = &ctx->planes[1].bands[i];
        band2 = &ctx->planes[2].bands[i];

        band2->width         = band1->width;
        band2->height        = band1->height;
        band2->mb_size       = band1->mb_size;
        band2->blk_size      = band1->blk_size;
        band2->is_halfpel    = band1->is_halfpel;
234 235 236 237
        band2->intra_base    = band1->intra_base;
        band2->inter_base    = band1->inter_base;
        band2->intra_scale   = band1->intra_scale;
        band2->inter_scale   = band1->inter_scale;
Kostya Shishkov's avatar
Kostya Shishkov committed
238 239 240 241
        band2->scan          = band1->scan;
        band2->inv_transform = band1->inv_transform;
        band2->dc_transform  = band1->dc_transform;
        band2->is_2d_trans   = band1->is_2d_trans;
242
        band2->transform_size= band1->transform_size;
Kostya Shishkov's avatar
Kostya Shishkov committed
243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283
    }

    /* reallocate internal structures if needed */
    if (blk_size_changed) {
        result = ff_ivi_init_tiles(ctx->planes, pic_conf.tile_width,
                                   pic_conf.tile_height);
        if (result) {
            av_log(avctx, AV_LOG_ERROR,
                   "Couldn't reallocate internal structures!\n");
            return -1;
        }
    }

    if (ctx->gop_flags & 8) {
        if (get_bits(&ctx->gb, 3)) {
            av_log(avctx, AV_LOG_ERROR, "Alignment bits are not zero!\n");
            return -1;
        }

        if (get_bits1(&ctx->gb))
            skip_bits_long(&ctx->gb, 24); /* skip transparency fill color */
    }

    align_get_bits(&ctx->gb);

    skip_bits(&ctx->gb, 23); /* FIXME: unknown meaning */

    /* skip GOP extension if any */
    if (get_bits1(&ctx->gb)) {
        do {
            i = get_bits(&ctx->gb, 16);
        } while (i & 0x8000);
    }

    align_get_bits(&ctx->gb);

    return 0;
}


/**
284
 *  Skip a header extension.
Kostya Shishkov's avatar
Kostya Shishkov committed
285
 *
286
 *  @param[in,out]  gb  the GetBit context
Kostya Shishkov's avatar
Kostya Shishkov committed
287 288 289 290 291 292 293 294 295 296 297 298 299
 */
static inline void skip_hdr_extension(GetBitContext *gb)
{
    int i, len;

    do {
        len = get_bits(gb, 8);
        for (i = 0; i < len; i++) skip_bits(gb, 8);
    } while(len);
}


/**
300
 *  Decode Indeo5 picture header.
Kostya Shishkov's avatar
Kostya Shishkov committed
301
 *
302 303
 *  @param[in,out]  ctx    ptr to the decoder context
 *  @param[in]      avctx  ptr to the AVCodecContext
Kostya Shishkov's avatar
Kostya Shishkov committed
304 305
 *  @return         result code: 0 = OK, -1 = error
 */
306
static int decode_pic_hdr(IVI45DecContext *ctx, AVCodecContext *avctx)
Kostya Shishkov's avatar
Kostya Shishkov committed
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322
{
    if (get_bits(&ctx->gb, 5) != 0x1F) {
        av_log(avctx, AV_LOG_ERROR, "Invalid picture start code!\n");
        return -1;
    }

    ctx->prev_frame_type = ctx->frame_type;
    ctx->frame_type      = get_bits(&ctx->gb, 3);
    if (ctx->frame_type >= 5) {
        av_log(avctx, AV_LOG_ERROR, "Invalid frame type: %d \n", ctx->frame_type);
        return -1;
    }

    ctx->frame_num = get_bits(&ctx->gb, 8);

    if (ctx->frame_type == FRAMETYPE_INTRA) {
323
        ctx->gop_invalid = 1;
324 325 326 327
        if (decode_gop_header(ctx, avctx)) {
            av_log(avctx, AV_LOG_ERROR, "Invalid GOP header, skipping frames.\n");
            return AVERROR_INVALIDDATA;
        }
328
        ctx->gop_invalid = 0;
Kostya Shishkov's avatar
Kostya Shishkov committed
329 330
    }

331 332 333 334 335 336
    if (ctx->frame_type == FRAMETYPE_INTER_SCAL && !ctx->is_scalable) {
        av_log(avctx, AV_LOG_ERROR, "Scalable inter frame in non scaleable stream\n");
        ctx->frame_type = FRAMETYPE_INTER;
        return AVERROR_INVALIDDATA;
    }

Kostya Shishkov's avatar
Kostya Shishkov committed
337 338 339 340 341 342 343 344 345 346 347 348
    if (ctx->frame_type != FRAMETYPE_NULL) {
        ctx->frame_flags = get_bits(&ctx->gb, 8);

        ctx->pic_hdr_size = (ctx->frame_flags & 1) ? get_bits_long(&ctx->gb, 24) : 0;

        ctx->checksum = (ctx->frame_flags & 0x10) ? get_bits(&ctx->gb, 16) : 0;

        /* skip unknown extension if any */
        if (ctx->frame_flags & 0x20)
            skip_hdr_extension(&ctx->gb); /* XXX: untested */

        /* decode macroblock huffman codebook */
349 350
        if (ff_ivi_dec_huff_desc(&ctx->gb, ctx->frame_flags & 0x40, IVI_MB_HUFF, &ctx->mb_vlc, avctx))
            return -1;
Kostya Shishkov's avatar
Kostya Shishkov committed
351 352 353 354 355 356 357 358 359 360 361

        skip_bits(&ctx->gb, 3); /* FIXME: unknown meaning! */
    }

    align_get_bits(&ctx->gb);

    return 0;
}


/**
362
 *  Decode Indeo5 band header.
Kostya Shishkov's avatar
Kostya Shishkov committed
363
 *
364 365 366
 *  @param[in,out]  ctx    ptr to the decoder context
 *  @param[in,out]  band   ptr to the band descriptor
 *  @param[in]      avctx  ptr to the AVCodecContext
Kostya Shishkov's avatar
Kostya Shishkov committed
367 368
 *  @return         result code: 0 = OK, -1 = error
 */
369
static int decode_band_hdr(IVI45DecContext *ctx, IVIBandDesc *band,
Kostya Shishkov's avatar
Kostya Shishkov committed
370 371
                           AVCodecContext *avctx)
{
372
    int         i;
Kostya Shishkov's avatar
Kostya Shishkov committed
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407
    uint8_t     band_flags;

    band_flags = get_bits(&ctx->gb, 8);

    if (band_flags & 1) {
        band->is_empty = 1;
        return 0;
    }

    band->data_size = (ctx->frame_flags & 0x80) ? get_bits_long(&ctx->gb, 24) : 0;

    band->inherit_mv     = band_flags & 2;
    band->inherit_qdelta = band_flags & 8;
    band->qdelta_present = band_flags & 4;
    if (!band->qdelta_present) band->inherit_qdelta = 1;

    /* decode rvmap probability corrections if any */
    band->num_corr = 0; /* there are no corrections */
    if (band_flags & 0x10) {
        band->num_corr = get_bits(&ctx->gb, 8); /* get number of correction pairs */
        if (band->num_corr > 61) {
            av_log(avctx, AV_LOG_ERROR, "Too many corrections: %d\n",
                   band->num_corr);
            return -1;
        }

        /* read correction pairs */
        for (i = 0; i < band->num_corr * 2; i++)
            band->corr[i] = get_bits(&ctx->gb, 8);
    }

    /* select appropriate rvmap table for this band */
    band->rvmap_sel = (band_flags & 0x40) ? get_bits(&ctx->gb, 3) : 8;

    /* decode block huffman codebook */
408 409
    if (ff_ivi_dec_huff_desc(&ctx->gb, band_flags & 0x80, IVI_BLK_HUFF, &band->blk_vlc, avctx))
        return -1;
Kostya Shishkov's avatar
Kostya Shishkov committed
410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429

    band->checksum_present = get_bits1(&ctx->gb);
    if (band->checksum_present)
        band->checksum = get_bits(&ctx->gb, 16);

    band->glob_quant = get_bits(&ctx->gb, 5);

    /* skip unknown extension if any */
    if (band_flags & 0x20) { /* XXX: untested */
        align_get_bits(&ctx->gb);
        skip_hdr_extension(&ctx->gb);
    }

    align_get_bits(&ctx->gb);

    return 0;
}


/**
430
 *  Decode info (block type, cbp, quant delta, motion vector)
Kostya Shishkov's avatar
Kostya Shishkov committed
431 432
 *  for all macroblocks in the current tile.
 *
433 434 435 436
 *  @param[in,out]  ctx    ptr to the decoder context
 *  @param[in,out]  band   ptr to the band descriptor
 *  @param[in,out]  tile   ptr to the tile descriptor
 *  @param[in]      avctx  ptr to the AVCodecContext
Kostya Shishkov's avatar
Kostya Shishkov committed
437 438
 *  @return         result code: 0 = OK, -1 = error
 */
439
static int decode_mb_info(IVI45DecContext *ctx, IVIBandDesc *band,
Kostya Shishkov's avatar
Kostya Shishkov committed
440 441 442
                          IVITile *tile, AVCodecContext *avctx)
{
    int         x, y, mv_x, mv_y, mv_delta, offs, mb_offset,
443
                mv_scale, blks_per_mb, s;
Kostya Shishkov's avatar
Kostya Shishkov committed
444 445 446 447 448 449 450
    IVIMbInfo   *mb, *ref_mb;
    int         row_offset = band->mb_size * band->pitch;

    mb     = tile->mbs;
    ref_mb = tile->ref_mbs;
    offs   = tile->ypos * band->pitch + tile->xpos;

451 452 453 454
    if (!ref_mb &&
        ((band->qdelta_present && band->inherit_qdelta) || band->inherit_mv))
        return AVERROR_INVALIDDATA;

455 456
    if (tile->num_MBs != IVI_MBs_PER_TILE(tile->width, tile->height, band->mb_size)) {
        av_log(avctx, AV_LOG_ERROR, "Allocated tile size %d mismatches parameters %d\n",
457 458 459 460
               tile->num_MBs, IVI_MBs_PER_TILE(tile->width, tile->height, band->mb_size));
        return AVERROR_INVALIDDATA;
    }

Kostya Shishkov's avatar
Kostya Shishkov committed
461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482
    /* scale factor for motion vectors */
    mv_scale = (ctx->planes[0].bands[0].mb_size >> 3) - (band->mb_size >> 3);
    mv_x = mv_y = 0;

    for (y = tile->ypos; y < (tile->ypos + tile->height); y += band->mb_size) {
        mb_offset = offs;

        for (x = tile->xpos; x < (tile->xpos + tile->width); x += band->mb_size) {
            mb->xpos     = x;
            mb->ypos     = y;
            mb->buf_offs = mb_offset;

            if (get_bits1(&ctx->gb)) {
                if (ctx->frame_type == FRAMETYPE_INTRA) {
                    av_log(avctx, AV_LOG_ERROR, "Empty macroblock in an INTRA picture!\n");
                    return -1;
                }
                mb->type = 1; /* empty macroblocks are always INTER */
                mb->cbp  = 0; /* all blocks are empty */

                mb->q_delta = 0;
                if (!band->plane && !band->band_num && (ctx->frame_flags & 8)) {
483
                    mb->q_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
484 485 486 487 488
                                           IVI_VLC_BITS, 1);
                    mb->q_delta = IVI_TOSIGNED(mb->q_delta);
                }

                mb->mv_x = mb->mv_y = 0; /* no motion vector coded */
489
                if (band->inherit_mv && ref_mb){
Kostya Shishkov's avatar
Kostya Shishkov committed
490 491 492 493 494 495 496 497 498 499
                    /* motion vector inheritance */
                    if (mv_scale) {
                        mb->mv_x = ivi_scale_mv(ref_mb->mv_x, mv_scale);
                        mb->mv_y = ivi_scale_mv(ref_mb->mv_y, mv_scale);
                    } else {
                        mb->mv_x = ref_mb->mv_x;
                        mb->mv_y = ref_mb->mv_y;
                    }
                }
            } else {
500
                if (band->inherit_mv && ref_mb) {
Kostya Shishkov's avatar
Kostya Shishkov committed
501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516
                    mb->type = ref_mb->type; /* copy mb_type from corresponding reference mb */
                } else if (ctx->frame_type == FRAMETYPE_INTRA) {
                    mb->type = 0; /* mb_type is always INTRA for intra-frames */
                } else {
                    mb->type = get_bits1(&ctx->gb);
                }

                blks_per_mb = band->mb_size != band->blk_size ? 4 : 1;
                mb->cbp = get_bits(&ctx->gb, blks_per_mb);

                mb->q_delta = 0;
                if (band->qdelta_present) {
                    if (band->inherit_qdelta) {
                        if (ref_mb) mb->q_delta = ref_mb->q_delta;
                    } else if (mb->cbp || (!band->plane && !band->band_num &&
                                           (ctx->frame_flags & 8))) {
517
                        mb->q_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
518 519 520 521 522 523 524 525
                                               IVI_VLC_BITS, 1);
                        mb->q_delta = IVI_TOSIGNED(mb->q_delta);
                    }
                }

                if (!mb->type) {
                    mb->mv_x = mb->mv_y = 0; /* there is no motion vector in intra-macroblocks */
                } else {
526
                    if (band->inherit_mv && ref_mb){
Kostya Shishkov's avatar
Kostya Shishkov committed
527 528 529 530 531 532 533 534 535 536
                        /* motion vector inheritance */
                        if (mv_scale) {
                            mb->mv_x = ivi_scale_mv(ref_mb->mv_x, mv_scale);
                            mb->mv_y = ivi_scale_mv(ref_mb->mv_y, mv_scale);
                        } else {
                            mb->mv_x = ref_mb->mv_x;
                            mb->mv_y = ref_mb->mv_y;
                        }
                    } else {
                        /* decode motion vector deltas */
537
                        mv_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
538 539
                                            IVI_VLC_BITS, 1);
                        mv_y += IVI_TOSIGNED(mv_delta);
540
                        mv_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
541 542 543 544 545 546 547 548
                                            IVI_VLC_BITS, 1);
                        mv_x += IVI_TOSIGNED(mv_delta);
                        mb->mv_x = mv_x;
                        mb->mv_y = mv_y;
                    }
                }
            }

549 550 551 552 553 554 555 556 557
            s= band->is_halfpel;
            if (mb->type)
            if ( x +  (mb->mv_x   >>s) +                 (y+               (mb->mv_y   >>s))*band->pitch < 0 ||
                 x + ((mb->mv_x+s)>>s) + band->mb_size - 1
                   + (y+band->mb_size - 1 +((mb->mv_y+s)>>s))*band->pitch > band->bufsize - 1) {
                av_log(avctx, AV_LOG_ERROR, "motion vector %d %d outside reference\n", x*s + mb->mv_x, y*s + mb->mv_y);
                return AVERROR_INVALIDDATA;
            }

Kostya Shishkov's avatar
Kostya Shishkov committed
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573
            mb++;
            if (ref_mb)
                ref_mb++;
            mb_offset += band->mb_size;
        }

        offs += row_offset;
    }

    align_get_bits(&ctx->gb);

    return 0;
}


/**
574
 *  Switch buffers.
Kostya Shishkov's avatar
Kostya Shishkov committed
575
 *
576
 *  @param[in,out] ctx  ptr to the decoder context
Kostya Shishkov's avatar
Kostya Shishkov committed
577
 */
578
static void switch_buffers(IVI45DecContext *ctx)
Kostya Shishkov's avatar
Kostya Shishkov committed
579
{
580
    switch (ctx->prev_frame_type) {
Kostya Shishkov's avatar
Kostya Shishkov committed
581 582
    case FRAMETYPE_INTRA:
    case FRAMETYPE_INTER:
583
        ctx->buf_switch ^= 1;
Kostya Shishkov's avatar
Kostya Shishkov committed
584 585 586 587
        ctx->dst_buf = ctx->buf_switch;
        ctx->ref_buf = ctx->buf_switch ^ 1;
        break;
    case FRAMETYPE_INTER_SCAL:
588 589 590
        if (!ctx->inter_scal) {
            ctx->ref2_buf   = 2;
            ctx->inter_scal = 1;
Kostya Shishkov's avatar
Kostya Shishkov committed
591
        }
592 593
        FFSWAP(int, ctx->dst_buf, ctx->ref2_buf);
        ctx->ref_buf = ctx->ref2_buf;
Kostya Shishkov's avatar
Kostya Shishkov committed
594 595 596
        break;
    case FRAMETYPE_INTER_NOREF:
        break;
597 598 599 600 601 602 603 604 605 606 607 608 609
    }

    switch (ctx->frame_type) {
    case FRAMETYPE_INTRA:
        ctx->buf_switch = 0;
        /* FALLTHROUGH */
    case FRAMETYPE_INTER:
        ctx->inter_scal = 0;
        ctx->dst_buf = ctx->buf_switch;
        ctx->ref_buf = ctx->buf_switch ^ 1;
        break;
    case FRAMETYPE_INTER_SCAL:
    case FRAMETYPE_INTER_NOREF:
Kostya Shishkov's avatar
Kostya Shishkov committed
610
    case FRAMETYPE_NULL:
611
        break;
Kostya Shishkov's avatar
Kostya Shishkov committed
612 613 614 615
    }
}


616 617 618 619 620 621
static int is_nonnull_frame(IVI45DecContext *ctx)
{
    return ctx->frame_type != FRAMETYPE_NULL;
}


Kostya Shishkov's avatar
Kostya Shishkov committed
622
/**
623
 *  Initialize Indeo5 decoder.
Kostya Shishkov's avatar
Kostya Shishkov committed
624 625 626
 */
static av_cold int decode_init(AVCodecContext *avctx)
{
627
    IVI45DecContext  *ctx = avctx->priv_data;
628
    int             result;
Kostya Shishkov's avatar
Kostya Shishkov committed
629

630
    ff_ivi_init_static_vlc();
Kostya Shishkov's avatar
Kostya Shishkov committed
631 632 633 634 635 636 637 638 639 640 641 642 643 644 645

    /* copy rvmap tables in our context so we can apply changes to them */
    memcpy(ctx->rvmap_tabs, ff_ivi_rvmap_tabs, sizeof(ff_ivi_rvmap_tabs));

    /* set the initial picture layout according to the basic profile:
       there is only one band per plane (no scalability), only one tile (no local decoding)
       and picture format = YVU9 */
    ctx->pic_conf.pic_width     = avctx->width;
    ctx->pic_conf.pic_height    = avctx->height;
    ctx->pic_conf.chroma_width  = (avctx->width  + 3) >> 2;
    ctx->pic_conf.chroma_height = (avctx->height + 3) >> 2;
    ctx->pic_conf.tile_width    = avctx->width;
    ctx->pic_conf.tile_height   = avctx->height;
    ctx->pic_conf.luma_bands    = ctx->pic_conf.chroma_bands = 1;

646 647
    avcodec_get_frame_defaults(&ctx->frame);

Kostya Shishkov's avatar
Kostya Shishkov committed
648 649 650 651 652 653
    result = ff_ivi_init_planes(ctx->planes, &ctx->pic_conf);
    if (result) {
        av_log(avctx, AV_LOG_ERROR, "Couldn't allocate color planes!\n");
        return -1;
    }

654 655 656
    ctx->buf_switch = 0;
    ctx->inter_scal = 0;

657 658 659 660 661
    ctx->decode_pic_hdr   = decode_pic_hdr;
    ctx->decode_band_hdr  = decode_band_hdr;
    ctx->decode_mb_info   = decode_mb_info;
    ctx->switch_buffers   = switch_buffers;
    ctx->is_nonnull_frame = is_nonnull_frame;
Kostya Shishkov's avatar
Kostya Shishkov committed
662

663
    avctx->pix_fmt = AV_PIX_FMT_YUV410P;
Kostya Shishkov's avatar
Kostya Shishkov committed
664 665 666 667

    return 0;
}

668
AVCodec ff_indeo5_decoder = {
Kostya Shishkov's avatar
Kostya Shishkov committed
669
    .name           = "indeo5",
670
    .type           = AVMEDIA_TYPE_VIDEO,
671
    .id             = AV_CODEC_ID_INDEO5,
672
    .priv_data_size = sizeof(IVI45DecContext),
Kostya Shishkov's avatar
Kostya Shishkov committed
673
    .init           = decode_init,
674 675
    .close          = ff_ivi_decode_close,
    .decode         = ff_ivi_decode_frame,
Kostya Shishkov's avatar
Kostya Shishkov committed
676
    .long_name      = NULL_IF_CONFIG_SMALL("Intel Indeo Video Interactive 5"),
677
    .capabilities   = CODEC_CAP_DR1,
Kostya Shishkov's avatar
Kostya Shishkov committed
678
};