indeo5.c 22.7 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;
Kostya Shishkov's avatar
Kostya Shishkov committed
324 325
        if (decode_gop_header(ctx, avctx))
            return -1;
326
        ctx->gop_invalid = 0;
Kostya Shishkov's avatar
Kostya Shishkov committed
327 328
    }

329 330 331 332 333 334
    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
335 336 337 338 339 340 341 342 343 344 345 346
    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 */
347 348
        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
349 350 351 352 353 354 355 356 357 358 359

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

    align_get_bits(&ctx->gb);

    return 0;
}


/**
360
 *  Decode Indeo5 band header.
Kostya Shishkov's avatar
Kostya Shishkov committed
361
 *
362 363 364
 *  @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
365 366
 *  @return         result code: 0 = OK, -1 = error
 */
367
static int decode_band_hdr(IVI45DecContext *ctx, IVIBandDesc *band,
Kostya Shishkov's avatar
Kostya Shishkov committed
368 369
                           AVCodecContext *avctx)
{
370
    int         i;
Kostya Shishkov's avatar
Kostya Shishkov committed
371 372 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
    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 */
406 407
    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
408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427

    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;
}


/**
428
 *  Decode info (block type, cbp, quant delta, motion vector)
Kostya Shishkov's avatar
Kostya Shishkov committed
429 430
 *  for all macroblocks in the current tile.
 *
431 432 433 434
 *  @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
435 436
 *  @return         result code: 0 = OK, -1 = error
 */
437
static int decode_mb_info(IVI45DecContext *ctx, IVIBandDesc *band,
Kostya Shishkov's avatar
Kostya Shishkov committed
438 439 440
                          IVITile *tile, AVCodecContext *avctx)
{
    int         x, y, mv_x, mv_y, mv_delta, offs, mb_offset,
441
                mv_scale, blks_per_mb, s;
Kostya Shishkov's avatar
Kostya Shishkov committed
442 443 444 445 446 447 448
    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;

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

453 454 455 456 457 458
    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",
               tile->num_MBs, IVI_MBs_PER_TILE(tile->width, tile->height, band->mb_size));
        return AVERROR_INVALIDDATA;
    }

Kostya Shishkov's avatar
Kostya Shishkov committed
459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480
    /* 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)) {
481
                    mb->q_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
482 483 484 485 486
                                           IVI_VLC_BITS, 1);
                    mb->q_delta = IVI_TOSIGNED(mb->q_delta);
                }

                mb->mv_x = mb->mv_y = 0; /* no motion vector coded */
487
                if (band->inherit_mv && ref_mb){
Kostya Shishkov's avatar
Kostya Shishkov committed
488 489 490 491 492 493 494 495 496 497
                    /* 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 {
498
                if (band->inherit_mv && ref_mb) {
Kostya Shishkov's avatar
Kostya Shishkov committed
499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
                    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))) {
515
                        mb->q_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
516 517 518 519 520 521 522 523
                                               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 {
524
                    if (band->inherit_mv && ref_mb){
Kostya Shishkov's avatar
Kostya Shishkov committed
525 526 527 528 529 530 531 532 533 534
                        /* 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 */
535
                        mv_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
536 537
                                            IVI_VLC_BITS, 1);
                        mv_y += IVI_TOSIGNED(mv_delta);
538
                        mv_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
Kostya Shishkov's avatar
Kostya Shishkov committed
539 540 541 542 543 544 545 546
                                            IVI_VLC_BITS, 1);
                        mv_x += IVI_TOSIGNED(mv_delta);
                        mb->mv_x = mv_x;
                        mb->mv_y = mv_y;
                    }
                }
            }

547 548 549 550 551 552 553 554 555
            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
556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
            mb++;
            if (ref_mb)
                ref_mb++;
            mb_offset += band->mb_size;
        }

        offs += row_offset;
    }

    align_get_bits(&ctx->gb);

    return 0;
}


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

    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
608
    case FRAMETYPE_NULL:
609
        break;
Kostya Shishkov's avatar
Kostya Shishkov committed
610 611 612 613
    }
}


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


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

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

    /* 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;

644 645
    avcodec_get_frame_defaults(&ctx->frame);

Kostya Shishkov's avatar
Kostya Shishkov committed
646 647 648 649 650 651
    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;
    }

652 653 654
    ctx->buf_switch = 0;
    ctx->inter_scal = 0;

655 656 657 658 659
    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
660

661
    avctx->pix_fmt = PIX_FMT_YUV410P;
Kostya Shishkov's avatar
Kostya Shishkov committed
662 663 664 665

    return 0;
}

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