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