annotate ffmpeg/libavcodec/indeo4.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 * Indeo Video Interactive v4 compatible decoder
yading@10 3 * Copyright (c) 2009-2011 Maxim Poliakovski
yading@10 4 *
yading@10 5 * This file is part of Libav.
yading@10 6 *
yading@10 7 * Libav is free software; you can redistribute it and/or
yading@10 8 * modify it under the terms of the GNU Lesser General Public
yading@10 9 * License as published by the Free Software Foundation; either
yading@10 10 * version 2.1 of the License, or (at your option) any later version.
yading@10 11 *
yading@10 12 * Libav is distributed in the hope that it will be useful,
yading@10 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
yading@10 15 * Lesser General Public License for more details.
yading@10 16 *
yading@10 17 * You should have received a copy of the GNU Lesser General Public
yading@10 18 * License along with Libav; if not, write to the Free Software
yading@10 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10 20 */
yading@10 21
yading@10 22 /**
yading@10 23 * @file
yading@10 24 * Indeo Video Interactive version 4 decoder
yading@10 25 *
yading@10 26 * Indeo 4 data is usually transported within .avi or .mov files.
yading@10 27 * Known FOURCCs: 'IV41'
yading@10 28 */
yading@10 29
yading@10 30 #define BITSTREAM_READER_LE
yading@10 31 #include "avcodec.h"
yading@10 32 #include "get_bits.h"
yading@10 33 #include "ivi_dsp.h"
yading@10 34 #include "ivi_common.h"
yading@10 35 #include "indeo4data.h"
yading@10 36
yading@10 37 /**
yading@10 38 * Indeo 4 frame types.
yading@10 39 */
yading@10 40 enum {
yading@10 41 FRAMETYPE_INTRA = 0,
yading@10 42 FRAMETYPE_INTRA1 = 1, ///< intra frame with slightly different bitstream coding
yading@10 43 FRAMETYPE_INTER = 2, ///< non-droppable P-frame
yading@10 44 FRAMETYPE_BIDIR = 3, ///< bidirectional frame
yading@10 45 FRAMETYPE_INTER_NOREF = 4, ///< droppable P-frame
yading@10 46 FRAMETYPE_NULL_FIRST = 5, ///< empty frame with no data
yading@10 47 FRAMETYPE_NULL_LAST = 6 ///< empty frame with no data
yading@10 48 };
yading@10 49
yading@10 50 #define IVI4_PIC_SIZE_ESC 7
yading@10 51
yading@10 52
yading@10 53 static const struct {
yading@10 54 InvTransformPtr *inv_trans;
yading@10 55 DCTransformPtr *dc_trans;
yading@10 56 int is_2d_trans;
yading@10 57 } transforms[18] = {
yading@10 58 { ff_ivi_inverse_haar_8x8, ff_ivi_dc_haar_2d, 1 },
yading@10 59 { NULL, NULL, 0 }, /* inverse Haar 8x1 */
yading@10 60 { NULL, NULL, 0 }, /* inverse Haar 1x8 */
yading@10 61 { ff_ivi_put_pixels_8x8, ff_ivi_put_dc_pixel_8x8, 1 },
yading@10 62 { ff_ivi_inverse_slant_8x8, ff_ivi_dc_slant_2d, 1 },
yading@10 63 { ff_ivi_row_slant8, ff_ivi_dc_row_slant, 1 },
yading@10 64 { ff_ivi_col_slant8, ff_ivi_dc_col_slant, 1 },
yading@10 65 { NULL, NULL, 0 }, /* inverse DCT 8x8 */
yading@10 66 { NULL, NULL, 0 }, /* inverse DCT 8x1 */
yading@10 67 { NULL, NULL, 0 }, /* inverse DCT 1x8 */
yading@10 68 { NULL, NULL, 0 }, /* inverse Haar 4x4 */
yading@10 69 { ff_ivi_inverse_slant_4x4, ff_ivi_dc_slant_2d, 1 },
yading@10 70 { NULL, NULL, 0 }, /* no transform 4x4 */
yading@10 71 { NULL, NULL, 0 }, /* inverse Haar 1x4 */
yading@10 72 { NULL, NULL, 0 }, /* inverse Haar 4x1 */
yading@10 73 { NULL, NULL, 0 }, /* inverse slant 1x4 */
yading@10 74 { NULL, NULL, 0 }, /* inverse slant 4x1 */
yading@10 75 { NULL, NULL, 0 }, /* inverse DCT 4x4 */
yading@10 76 };
yading@10 77
yading@10 78 /**
yading@10 79 * Decode subdivision of a plane.
yading@10 80 * This is a simplified version that checks for two supported subdivisions:
yading@10 81 * - 1 wavelet band per plane, size factor 1:1, code pattern: 3
yading@10 82 * - 4 wavelet bands per plane, size factor 1:4, code pattern: 2,3,3,3,3
yading@10 83 * Anything else is either unsupported or corrupt.
yading@10 84 *
yading@10 85 * @param[in,out] gb the GetBit context
yading@10 86 * @return number of wavelet bands or 0 on error
yading@10 87 */
yading@10 88 static int decode_plane_subdivision(GetBitContext *gb)
yading@10 89 {
yading@10 90 int i;
yading@10 91
yading@10 92 switch (get_bits(gb, 2)) {
yading@10 93 case 3:
yading@10 94 return 1;
yading@10 95 case 2:
yading@10 96 for (i = 0; i < 4; i++)
yading@10 97 if (get_bits(gb, 2) != 3)
yading@10 98 return 0;
yading@10 99 return 4;
yading@10 100 default:
yading@10 101 return 0;
yading@10 102 }
yading@10 103 }
yading@10 104
yading@10 105 static inline int scale_tile_size(int def_size, int size_factor)
yading@10 106 {
yading@10 107 return size_factor == 15 ? def_size : (size_factor + 1) << 5;
yading@10 108 }
yading@10 109
yading@10 110 /**
yading@10 111 * Decode Indeo 4 picture header.
yading@10 112 *
yading@10 113 * @param[in,out] ctx pointer to the decoder context
yading@10 114 * @param[in] avctx pointer to the AVCodecContext
yading@10 115 * @return result code: 0 = OK, negative number = error
yading@10 116 */
yading@10 117 static int decode_pic_hdr(IVI45DecContext *ctx, AVCodecContext *avctx)
yading@10 118 {
yading@10 119 int pic_size_indx, i, p;
yading@10 120 IVIPicConfig pic_conf;
yading@10 121
yading@10 122 if (get_bits(&ctx->gb, 18) != 0x3FFF8) {
yading@10 123 av_log(avctx, AV_LOG_ERROR, "Invalid picture start code!\n");
yading@10 124 return AVERROR_INVALIDDATA;
yading@10 125 }
yading@10 126
yading@10 127 ctx->prev_frame_type = ctx->frame_type;
yading@10 128 ctx->frame_type = get_bits(&ctx->gb, 3);
yading@10 129 if (ctx->frame_type == 7) {
yading@10 130 av_log(avctx, AV_LOG_ERROR, "Invalid frame type: %d\n", ctx->frame_type);
yading@10 131 return AVERROR_INVALIDDATA;
yading@10 132 }
yading@10 133
yading@10 134 #if IVI4_STREAM_ANALYSER
yading@10 135 if (ctx->frame_type == FRAMETYPE_BIDIR)
yading@10 136 ctx->has_b_frames = 1;
yading@10 137 #endif
yading@10 138
yading@10 139 ctx->transp_status = get_bits1(&ctx->gb);
yading@10 140 #if IVI4_STREAM_ANALYSER
yading@10 141 if (ctx->transp_status) {
yading@10 142 ctx->has_transp = 1;
yading@10 143 }
yading@10 144 #endif
yading@10 145
yading@10 146 /* unknown bit: Mac decoder ignores this bit, XANIM returns error */
yading@10 147 if (get_bits1(&ctx->gb)) {
yading@10 148 av_log(avctx, AV_LOG_ERROR, "Sync bit is set!\n");
yading@10 149 return AVERROR_INVALIDDATA;
yading@10 150 }
yading@10 151
yading@10 152 ctx->data_size = get_bits1(&ctx->gb) ? get_bits(&ctx->gb, 24) : 0;
yading@10 153
yading@10 154 /* null frames don't contain anything else so we just return */
yading@10 155 if (ctx->frame_type >= FRAMETYPE_NULL_FIRST) {
yading@10 156 av_dlog(avctx, "Null frame encountered!\n");
yading@10 157 return 0;
yading@10 158 }
yading@10 159
yading@10 160 /* Check key lock status. If enabled - ignore lock word. */
yading@10 161 /* Usually we have to prompt the user for the password, but */
yading@10 162 /* we don't do that because Indeo 4 videos can be decoded anyway */
yading@10 163 if (get_bits1(&ctx->gb)) {
yading@10 164 skip_bits_long(&ctx->gb, 32);
yading@10 165 av_dlog(avctx, "Password-protected clip!\n");
yading@10 166 }
yading@10 167
yading@10 168 pic_size_indx = get_bits(&ctx->gb, 3);
yading@10 169 if (pic_size_indx == IVI4_PIC_SIZE_ESC) {
yading@10 170 pic_conf.pic_height = get_bits(&ctx->gb, 16);
yading@10 171 pic_conf.pic_width = get_bits(&ctx->gb, 16);
yading@10 172 } else {
yading@10 173 pic_conf.pic_height = ivi4_common_pic_sizes[pic_size_indx * 2 + 1];
yading@10 174 pic_conf.pic_width = ivi4_common_pic_sizes[pic_size_indx * 2 ];
yading@10 175 }
yading@10 176
yading@10 177 /* Decode tile dimensions. */
yading@10 178 if (get_bits1(&ctx->gb)) {
yading@10 179 pic_conf.tile_height = scale_tile_size(pic_conf.pic_height, get_bits(&ctx->gb, 4));
yading@10 180 pic_conf.tile_width = scale_tile_size(pic_conf.pic_width, get_bits(&ctx->gb, 4));
yading@10 181 #if IVI4_STREAM_ANALYSER
yading@10 182 ctx->uses_tiling = 1;
yading@10 183 #endif
yading@10 184 } else {
yading@10 185 pic_conf.tile_height = pic_conf.pic_height;
yading@10 186 pic_conf.tile_width = pic_conf.pic_width;
yading@10 187 }
yading@10 188
yading@10 189 /* Decode chroma subsampling. We support only 4:4 aka YVU9. */
yading@10 190 if (get_bits(&ctx->gb, 2)) {
yading@10 191 av_log(avctx, AV_LOG_ERROR, "Only YVU9 picture format is supported!\n");
yading@10 192 return AVERROR_INVALIDDATA;
yading@10 193 }
yading@10 194 pic_conf.chroma_height = (pic_conf.pic_height + 3) >> 2;
yading@10 195 pic_conf.chroma_width = (pic_conf.pic_width + 3) >> 2;
yading@10 196
yading@10 197 /* decode subdivision of the planes */
yading@10 198 pic_conf.luma_bands = decode_plane_subdivision(&ctx->gb);
yading@10 199 pic_conf.chroma_bands = 0;
yading@10 200 if (pic_conf.luma_bands)
yading@10 201 pic_conf.chroma_bands = decode_plane_subdivision(&ctx->gb);
yading@10 202 ctx->is_scalable = pic_conf.luma_bands != 1 || pic_conf.chroma_bands != 1;
yading@10 203 if (ctx->is_scalable && (pic_conf.luma_bands != 4 || pic_conf.chroma_bands != 1)) {
yading@10 204 av_log(avctx, AV_LOG_ERROR, "Scalability: unsupported subdivision! Luma bands: %d, chroma bands: %d\n",
yading@10 205 pic_conf.luma_bands, pic_conf.chroma_bands);
yading@10 206 return AVERROR_INVALIDDATA;
yading@10 207 }
yading@10 208
yading@10 209 /* check if picture layout was changed and reallocate buffers */
yading@10 210 if (ivi_pic_config_cmp(&pic_conf, &ctx->pic_conf)) {
yading@10 211 if (ff_ivi_init_planes(ctx->planes, &pic_conf)) {
yading@10 212 av_log(avctx, AV_LOG_ERROR, "Couldn't reallocate color planes!\n");
yading@10 213 return AVERROR(ENOMEM);
yading@10 214 }
yading@10 215
yading@10 216 ctx->pic_conf = pic_conf;
yading@10 217
yading@10 218 /* set default macroblock/block dimensions */
yading@10 219 for (p = 0; p <= 2; p++) {
yading@10 220 for (i = 0; i < (!p ? pic_conf.luma_bands : pic_conf.chroma_bands); i++) {
yading@10 221 ctx->planes[p].bands[i].mb_size = !p ? (!ctx->is_scalable ? 16 : 8) : 4;
yading@10 222 ctx->planes[p].bands[i].blk_size = !p ? 8 : 4;
yading@10 223 }
yading@10 224 }
yading@10 225
yading@10 226 if (ff_ivi_init_tiles(ctx->planes, ctx->pic_conf.tile_width,
yading@10 227 ctx->pic_conf.tile_height)) {
yading@10 228 av_log(avctx, AV_LOG_ERROR,
yading@10 229 "Couldn't reallocate internal structures!\n");
yading@10 230 return AVERROR(ENOMEM);
yading@10 231 }
yading@10 232 }
yading@10 233
yading@10 234 ctx->frame_num = get_bits1(&ctx->gb) ? get_bits(&ctx->gb, 20) : 0;
yading@10 235
yading@10 236 /* skip decTimeEst field if present */
yading@10 237 if (get_bits1(&ctx->gb))
yading@10 238 skip_bits(&ctx->gb, 8);
yading@10 239
yading@10 240 /* decode macroblock and block huffman codebooks */
yading@10 241 if (ff_ivi_dec_huff_desc(&ctx->gb, get_bits1(&ctx->gb), IVI_MB_HUFF, &ctx->mb_vlc, avctx) ||
yading@10 242 ff_ivi_dec_huff_desc(&ctx->gb, get_bits1(&ctx->gb), IVI_BLK_HUFF, &ctx->blk_vlc, avctx))
yading@10 243 return AVERROR_INVALIDDATA;
yading@10 244
yading@10 245 ctx->rvmap_sel = get_bits1(&ctx->gb) ? get_bits(&ctx->gb, 3) : 8;
yading@10 246
yading@10 247 ctx->in_imf = get_bits1(&ctx->gb);
yading@10 248 ctx->in_q = get_bits1(&ctx->gb);
yading@10 249
yading@10 250 ctx->pic_glob_quant = get_bits(&ctx->gb, 5);
yading@10 251
yading@10 252 /* TODO: ignore this parameter if unused */
yading@10 253 ctx->unknown1 = get_bits1(&ctx->gb) ? get_bits(&ctx->gb, 3) : 0;
yading@10 254
yading@10 255 ctx->checksum = get_bits1(&ctx->gb) ? get_bits(&ctx->gb, 16) : 0;
yading@10 256
yading@10 257 /* skip picture header extension if any */
yading@10 258 while (get_bits1(&ctx->gb)) {
yading@10 259 av_dlog(avctx, "Pic hdr extension encountered!\n");
yading@10 260 skip_bits(&ctx->gb, 8);
yading@10 261 }
yading@10 262
yading@10 263 if (get_bits1(&ctx->gb)) {
yading@10 264 av_log(avctx, AV_LOG_ERROR, "Bad blocks bits encountered!\n");
yading@10 265 }
yading@10 266
yading@10 267 align_get_bits(&ctx->gb);
yading@10 268
yading@10 269 return 0;
yading@10 270 }
yading@10 271
yading@10 272
yading@10 273 /**
yading@10 274 * Decode Indeo 4 band header.
yading@10 275 *
yading@10 276 * @param[in,out] ctx pointer to the decoder context
yading@10 277 * @param[in,out] band pointer to the band descriptor
yading@10 278 * @param[in] avctx pointer to the AVCodecContext
yading@10 279 * @return result code: 0 = OK, negative number = error
yading@10 280 */
yading@10 281 static int decode_band_hdr(IVI45DecContext *ctx, IVIBandDesc *band,
yading@10 282 AVCodecContext *avctx)
yading@10 283 {
yading@10 284 int plane, band_num, indx, transform_id, scan_indx;
yading@10 285 int i;
yading@10 286 int quant_mat;
yading@10 287
yading@10 288 plane = get_bits(&ctx->gb, 2);
yading@10 289 band_num = get_bits(&ctx->gb, 4);
yading@10 290 if (band->plane != plane || band->band_num != band_num) {
yading@10 291 av_log(avctx, AV_LOG_ERROR, "Invalid band header sequence!\n");
yading@10 292 return AVERROR_INVALIDDATA;
yading@10 293 }
yading@10 294
yading@10 295 band->is_empty = get_bits1(&ctx->gb);
yading@10 296 if (!band->is_empty) {
yading@10 297 /* skip header size
yading@10 298 * If header size is not given, header size is 4 bytes. */
yading@10 299 if (get_bits1(&ctx->gb))
yading@10 300 skip_bits(&ctx->gb, 16);
yading@10 301
yading@10 302 band->is_halfpel = get_bits(&ctx->gb, 2);
yading@10 303 if (band->is_halfpel >= 2) {
yading@10 304 av_log(avctx, AV_LOG_ERROR, "Invalid/unsupported mv resolution: %d!\n",
yading@10 305 band->is_halfpel);
yading@10 306 return AVERROR_INVALIDDATA;
yading@10 307 }
yading@10 308 #if IVI4_STREAM_ANALYSER
yading@10 309 if (!band->is_halfpel)
yading@10 310 ctx->uses_fullpel = 1;
yading@10 311 #endif
yading@10 312
yading@10 313 band->checksum_present = get_bits1(&ctx->gb);
yading@10 314 if (band->checksum_present)
yading@10 315 band->checksum = get_bits(&ctx->gb, 16);
yading@10 316
yading@10 317 indx = get_bits(&ctx->gb, 2);
yading@10 318 if (indx == 3) {
yading@10 319 av_log(avctx, AV_LOG_ERROR, "Invalid block size!\n");
yading@10 320 return AVERROR_INVALIDDATA;
yading@10 321 }
yading@10 322 band->mb_size = 16 >> indx;
yading@10 323 band->blk_size = 8 >> (indx >> 1);
yading@10 324
yading@10 325 band->inherit_mv = get_bits1(&ctx->gb);
yading@10 326 band->inherit_qdelta = get_bits1(&ctx->gb);
yading@10 327
yading@10 328 band->glob_quant = get_bits(&ctx->gb, 5);
yading@10 329
yading@10 330 if (!get_bits1(&ctx->gb) || ctx->frame_type == FRAMETYPE_INTRA) {
yading@10 331 transform_id = get_bits(&ctx->gb, 5);
yading@10 332 if (transform_id >= FF_ARRAY_ELEMS(transforms) ||
yading@10 333 !transforms[transform_id].inv_trans) {
yading@10 334 avpriv_request_sample(avctx, "Transform %d", transform_id);
yading@10 335 return AVERROR_PATCHWELCOME;
yading@10 336 }
yading@10 337 if ((transform_id >= 7 && transform_id <= 9) ||
yading@10 338 transform_id == 17) {
yading@10 339 avpriv_request_sample(avctx, "DCT transform");
yading@10 340 return AVERROR_PATCHWELCOME;
yading@10 341 }
yading@10 342
yading@10 343 if (transform_id < 10 && band->blk_size < 8) {
yading@10 344 av_log(avctx, AV_LOG_ERROR, "wrong transform size!\n");
yading@10 345 return AVERROR_INVALIDDATA;
yading@10 346 }
yading@10 347 #if IVI4_STREAM_ANALYSER
yading@10 348 if ((transform_id >= 0 && transform_id <= 2) || transform_id == 10)
yading@10 349 ctx->uses_haar = 1;
yading@10 350 #endif
yading@10 351
yading@10 352 band->inv_transform = transforms[transform_id].inv_trans;
yading@10 353 band->dc_transform = transforms[transform_id].dc_trans;
yading@10 354 band->is_2d_trans = transforms[transform_id].is_2d_trans;
yading@10 355 band->transform_size= (transform_id < 10) ? 8 : 4;
yading@10 356
yading@10 357 scan_indx = get_bits(&ctx->gb, 4);
yading@10 358 if ((scan_indx>4 && scan_indx<10) != (band->blk_size==4)) {
yading@10 359 av_log(avctx, AV_LOG_ERROR, "mismatching scan table!\n");
yading@10 360 return AVERROR_INVALIDDATA;
yading@10 361 }
yading@10 362 if (scan_indx == 15) {
yading@10 363 av_log(avctx, AV_LOG_ERROR, "Custom scan pattern encountered!\n");
yading@10 364 return AVERROR_INVALIDDATA;
yading@10 365 }
yading@10 366 band->scan = scan_index_to_tab[scan_indx];
yading@10 367 band->scan_size = band->blk_size;
yading@10 368
yading@10 369 quant_mat = get_bits(&ctx->gb, 5);
yading@10 370 if (quant_mat == 31) {
yading@10 371 av_log(avctx, AV_LOG_ERROR, "Custom quant matrix encountered!\n");
yading@10 372 return AVERROR_INVALIDDATA;
yading@10 373 }
yading@10 374 if (quant_mat > 21) {
yading@10 375 av_log(avctx, AV_LOG_ERROR, "Invalid quant matrix encountered!\n");
yading@10 376 return AVERROR_INVALIDDATA;
yading@10 377 }
yading@10 378 band->quant_mat = quant_mat;
yading@10 379 }
yading@10 380 if (quant_index_to_tab[band->quant_mat] > 4 && band->blk_size == 4) {
yading@10 381 av_log(avctx, AV_LOG_ERROR, "Invalid quant matrix for 4x4 block encountered!\n");
yading@10 382 band->quant_mat = 0;
yading@10 383 return AVERROR_INVALIDDATA;
yading@10 384 }
yading@10 385 if (band->scan_size != band->blk_size) {
yading@10 386 av_log(avctx, AV_LOG_ERROR, "mismatching scan table!\n");
yading@10 387 return AVERROR_INVALIDDATA;
yading@10 388 }
yading@10 389 if (band->transform_size == 8 && band->blk_size < 8) {
yading@10 390 av_log(avctx, AV_LOG_ERROR, "mismatching transform_size!\n");
yading@10 391 return AVERROR_INVALIDDATA;
yading@10 392 }
yading@10 393
yading@10 394 /* decode block huffman codebook */
yading@10 395 if (ff_ivi_dec_huff_desc(&ctx->gb, get_bits1(&ctx->gb), IVI_BLK_HUFF,
yading@10 396 &band->blk_vlc, avctx))
yading@10 397 return AVERROR_INVALIDDATA;
yading@10 398
yading@10 399 /* select appropriate rvmap table for this band */
yading@10 400 band->rvmap_sel = get_bits1(&ctx->gb) ? get_bits(&ctx->gb, 3) : 8;
yading@10 401
yading@10 402 /* decode rvmap probability corrections if any */
yading@10 403 band->num_corr = 0; /* there is no corrections */
yading@10 404 if (get_bits1(&ctx->gb)) {
yading@10 405 band->num_corr = get_bits(&ctx->gb, 8); /* get number of correction pairs */
yading@10 406 if (band->num_corr > 61) {
yading@10 407 av_log(avctx, AV_LOG_ERROR, "Too many corrections: %d\n",
yading@10 408 band->num_corr);
yading@10 409 return AVERROR_INVALIDDATA;
yading@10 410 }
yading@10 411
yading@10 412 /* read correction pairs */
yading@10 413 for (i = 0; i < band->num_corr * 2; i++)
yading@10 414 band->corr[i] = get_bits(&ctx->gb, 8);
yading@10 415 }
yading@10 416 }
yading@10 417
yading@10 418 if (band->blk_size == 8) {
yading@10 419 band->intra_base = &ivi4_quant_8x8_intra[quant_index_to_tab[band->quant_mat]][0];
yading@10 420 band->inter_base = &ivi4_quant_8x8_inter[quant_index_to_tab[band->quant_mat]][0];
yading@10 421 } else {
yading@10 422 band->intra_base = &ivi4_quant_4x4_intra[quant_index_to_tab[band->quant_mat]][0];
yading@10 423 band->inter_base = &ivi4_quant_4x4_inter[quant_index_to_tab[band->quant_mat]][0];
yading@10 424 }
yading@10 425
yading@10 426 /* Indeo 4 doesn't use scale tables */
yading@10 427 band->intra_scale = NULL;
yading@10 428 band->inter_scale = NULL;
yading@10 429
yading@10 430 align_get_bits(&ctx->gb);
yading@10 431
yading@10 432 if (!band->scan) {
yading@10 433 av_log(avctx, AV_LOG_ERROR, "band->scan not set\n");
yading@10 434 return AVERROR_INVALIDDATA;
yading@10 435 }
yading@10 436
yading@10 437 return 0;
yading@10 438 }
yading@10 439
yading@10 440
yading@10 441 /**
yading@10 442 * Decode information (block type, cbp, quant delta, motion vector)
yading@10 443 * for all macroblocks in the current tile.
yading@10 444 *
yading@10 445 * @param[in,out] ctx pointer to the decoder context
yading@10 446 * @param[in,out] band pointer to the band descriptor
yading@10 447 * @param[in,out] tile pointer to the tile descriptor
yading@10 448 * @param[in] avctx pointer to the AVCodecContext
yading@10 449 * @return result code: 0 = OK, negative number = error
yading@10 450 */
yading@10 451 static int decode_mb_info(IVI45DecContext *ctx, IVIBandDesc *band,
yading@10 452 IVITile *tile, AVCodecContext *avctx)
yading@10 453 {
yading@10 454 int x, y, mv_x, mv_y, mv_delta, offs, mb_offset, blks_per_mb,
yading@10 455 mv_scale, mb_type_bits, s;
yading@10 456 IVIMbInfo *mb, *ref_mb;
yading@10 457 int row_offset = band->mb_size * band->pitch;
yading@10 458
yading@10 459 mb = tile->mbs;
yading@10 460 ref_mb = tile->ref_mbs;
yading@10 461 offs = tile->ypos * band->pitch + tile->xpos;
yading@10 462
yading@10 463 blks_per_mb = band->mb_size != band->blk_size ? 4 : 1;
yading@10 464 mb_type_bits = ctx->frame_type == FRAMETYPE_BIDIR ? 2 : 1;
yading@10 465
yading@10 466 /* scale factor for motion vectors */
yading@10 467 mv_scale = (ctx->planes[0].bands[0].mb_size >> 3) - (band->mb_size >> 3);
yading@10 468 mv_x = mv_y = 0;
yading@10 469
yading@10 470 if (((tile->width + band->mb_size-1)/band->mb_size) * ((tile->height + band->mb_size-1)/band->mb_size) != tile->num_MBs) {
yading@10 471 av_log(avctx, AV_LOG_ERROR, "num_MBs mismatch %d %d %d %d\n", tile->width, tile->height, band->mb_size, tile->num_MBs);
yading@10 472 return -1;
yading@10 473 }
yading@10 474
yading@10 475 for (y = tile->ypos; y < tile->ypos + tile->height; y += band->mb_size) {
yading@10 476 mb_offset = offs;
yading@10 477
yading@10 478 for (x = tile->xpos; x < tile->xpos + tile->width; x += band->mb_size) {
yading@10 479 mb->xpos = x;
yading@10 480 mb->ypos = y;
yading@10 481 mb->buf_offs = mb_offset;
yading@10 482
yading@10 483 if (get_bits1(&ctx->gb)) {
yading@10 484 if (ctx->frame_type == FRAMETYPE_INTRA) {
yading@10 485 av_log(avctx, AV_LOG_ERROR, "Empty macroblock in an INTRA picture!\n");
yading@10 486 return AVERROR_INVALIDDATA;
yading@10 487 }
yading@10 488 mb->type = 1; /* empty macroblocks are always INTER */
yading@10 489 mb->cbp = 0; /* all blocks are empty */
yading@10 490
yading@10 491 mb->q_delta = 0;
yading@10 492 if (!band->plane && !band->band_num && ctx->in_q) {
yading@10 493 mb->q_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
yading@10 494 IVI_VLC_BITS, 1);
yading@10 495 mb->q_delta = IVI_TOSIGNED(mb->q_delta);
yading@10 496 }
yading@10 497
yading@10 498 mb->mv_x = mb->mv_y = 0; /* no motion vector coded */
yading@10 499 if (band->inherit_mv && ref_mb) {
yading@10 500 /* motion vector inheritance */
yading@10 501 if (mv_scale) {
yading@10 502 mb->mv_x = ivi_scale_mv(ref_mb->mv_x, mv_scale);
yading@10 503 mb->mv_y = ivi_scale_mv(ref_mb->mv_y, mv_scale);
yading@10 504 } else {
yading@10 505 mb->mv_x = ref_mb->mv_x;
yading@10 506 mb->mv_y = ref_mb->mv_y;
yading@10 507 }
yading@10 508 }
yading@10 509 } else {
yading@10 510 if (band->inherit_mv && ref_mb) {
yading@10 511 mb->type = ref_mb->type; /* copy mb_type from corresponding reference mb */
yading@10 512 } else if (ctx->frame_type == FRAMETYPE_INTRA ||
yading@10 513 ctx->frame_type == FRAMETYPE_INTRA1) {
yading@10 514 mb->type = 0; /* mb_type is always INTRA for intra-frames */
yading@10 515 } else {
yading@10 516 mb->type = get_bits(&ctx->gb, mb_type_bits);
yading@10 517 }
yading@10 518
yading@10 519 mb->cbp = get_bits(&ctx->gb, blks_per_mb);
yading@10 520
yading@10 521 mb->q_delta = 0;
yading@10 522 if (band->inherit_qdelta) {
yading@10 523 if (ref_mb) mb->q_delta = ref_mb->q_delta;
yading@10 524 } else if (mb->cbp || (!band->plane && !band->band_num &&
yading@10 525 ctx->in_q)) {
yading@10 526 mb->q_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
yading@10 527 IVI_VLC_BITS, 1);
yading@10 528 mb->q_delta = IVI_TOSIGNED(mb->q_delta);
yading@10 529 }
yading@10 530
yading@10 531 if (!mb->type) {
yading@10 532 mb->mv_x = mb->mv_y = 0; /* there is no motion vector in intra-macroblocks */
yading@10 533 } else {
yading@10 534 if (band->inherit_mv && ref_mb) {
yading@10 535 /* motion vector inheritance */
yading@10 536 if (mv_scale) {
yading@10 537 mb->mv_x = ivi_scale_mv(ref_mb->mv_x, mv_scale);
yading@10 538 mb->mv_y = ivi_scale_mv(ref_mb->mv_y, mv_scale);
yading@10 539 } else {
yading@10 540 mb->mv_x = ref_mb->mv_x;
yading@10 541 mb->mv_y = ref_mb->mv_y;
yading@10 542 }
yading@10 543 } else {
yading@10 544 /* decode motion vector deltas */
yading@10 545 mv_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
yading@10 546 IVI_VLC_BITS, 1);
yading@10 547 mv_y += IVI_TOSIGNED(mv_delta);
yading@10 548 mv_delta = get_vlc2(&ctx->gb, ctx->mb_vlc.tab->table,
yading@10 549 IVI_VLC_BITS, 1);
yading@10 550 mv_x += IVI_TOSIGNED(mv_delta);
yading@10 551 mb->mv_x = mv_x;
yading@10 552 mb->mv_y = mv_y;
yading@10 553 }
yading@10 554 }
yading@10 555 }
yading@10 556
yading@10 557 s= band->is_halfpel;
yading@10 558 if (mb->type)
yading@10 559 if ( x + (mb->mv_x >>s) + (y+ (mb->mv_y >>s))*band->pitch < 0 ||
yading@10 560 x + ((mb->mv_x+s)>>s) + band->mb_size - 1
yading@10 561 + (y+band->mb_size - 1 +((mb->mv_y+s)>>s))*band->pitch > band->bufsize -1) {
yading@10 562 av_log(avctx, AV_LOG_ERROR, "motion vector %d %d outside reference\n", x*s + mb->mv_x, y*s + mb->mv_y);
yading@10 563 return AVERROR_INVALIDDATA;
yading@10 564 }
yading@10 565
yading@10 566 mb++;
yading@10 567 if (ref_mb)
yading@10 568 ref_mb++;
yading@10 569 mb_offset += band->mb_size;
yading@10 570 }
yading@10 571
yading@10 572 offs += row_offset;
yading@10 573 }
yading@10 574
yading@10 575 align_get_bits(&ctx->gb);
yading@10 576
yading@10 577 return 0;
yading@10 578 }
yading@10 579
yading@10 580
yading@10 581 /**
yading@10 582 * Rearrange decoding and reference buffers.
yading@10 583 *
yading@10 584 * @param[in,out] ctx pointer to the decoder context
yading@10 585 */
yading@10 586 static void switch_buffers(IVI45DecContext *ctx)
yading@10 587 {
yading@10 588 switch (ctx->prev_frame_type) {
yading@10 589 case FRAMETYPE_INTRA:
yading@10 590 case FRAMETYPE_INTRA1:
yading@10 591 case FRAMETYPE_INTER:
yading@10 592 ctx->buf_switch ^= 1;
yading@10 593 ctx->dst_buf = ctx->buf_switch;
yading@10 594 ctx->ref_buf = ctx->buf_switch ^ 1;
yading@10 595 break;
yading@10 596 case FRAMETYPE_INTER_NOREF:
yading@10 597 break;
yading@10 598 }
yading@10 599
yading@10 600 switch (ctx->frame_type) {
yading@10 601 case FRAMETYPE_INTRA:
yading@10 602 case FRAMETYPE_INTRA1:
yading@10 603 ctx->buf_switch = 0;
yading@10 604 /* FALLTHROUGH */
yading@10 605 case FRAMETYPE_INTER:
yading@10 606 ctx->dst_buf = ctx->buf_switch;
yading@10 607 ctx->ref_buf = ctx->buf_switch ^ 1;
yading@10 608 break;
yading@10 609 case FRAMETYPE_INTER_NOREF:
yading@10 610 case FRAMETYPE_NULL_FIRST:
yading@10 611 case FRAMETYPE_NULL_LAST:
yading@10 612 break;
yading@10 613 }
yading@10 614 }
yading@10 615
yading@10 616
yading@10 617 static int is_nonnull_frame(IVI45DecContext *ctx)
yading@10 618 {
yading@10 619 return ctx->frame_type < FRAMETYPE_NULL_FIRST;
yading@10 620 }
yading@10 621
yading@10 622
yading@10 623 static av_cold int decode_init(AVCodecContext *avctx)
yading@10 624 {
yading@10 625 IVI45DecContext *ctx = avctx->priv_data;
yading@10 626
yading@10 627 ff_ivi_init_static_vlc();
yading@10 628
yading@10 629 /* copy rvmap tables in our context so we can apply changes to them */
yading@10 630 memcpy(ctx->rvmap_tabs, ff_ivi_rvmap_tabs, sizeof(ff_ivi_rvmap_tabs));
yading@10 631
yading@10 632 /* Force allocation of the internal buffers */
yading@10 633 /* during picture header decoding. */
yading@10 634 ctx->pic_conf.pic_width = 0;
yading@10 635 ctx->pic_conf.pic_height = 0;
yading@10 636
yading@10 637 avctx->pix_fmt = AV_PIX_FMT_YUV410P;
yading@10 638
yading@10 639 ctx->decode_pic_hdr = decode_pic_hdr;
yading@10 640 ctx->decode_band_hdr = decode_band_hdr;
yading@10 641 ctx->decode_mb_info = decode_mb_info;
yading@10 642 ctx->switch_buffers = switch_buffers;
yading@10 643 ctx->is_nonnull_frame = is_nonnull_frame;
yading@10 644
yading@10 645 return 0;
yading@10 646 }
yading@10 647
yading@10 648
yading@10 649 AVCodec ff_indeo4_decoder = {
yading@10 650 .name = "indeo4",
yading@10 651 .type = AVMEDIA_TYPE_VIDEO,
yading@10 652 .id = AV_CODEC_ID_INDEO4,
yading@10 653 .priv_data_size = sizeof(IVI45DecContext),
yading@10 654 .init = decode_init,
yading@10 655 .close = ff_ivi_decode_close,
yading@10 656 .decode = ff_ivi_decode_frame,
yading@10 657 .long_name = NULL_IF_CONFIG_SMALL("Intel Indeo Video Interactive 4"),
yading@10 658 .capabilities = CODEC_CAP_DR1,
yading@10 659 };