4b3f903e3b67ee6cc43a2eff68276cb1df5b13e6
[libav.git] / libavcodec / indeo4.c
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"
32 #include "bitstream.h"
33 #include "indeo4data.h"
34 #include "internal.h"
35 #include "ivi.h"
36 #include "ivi_dsp.h"
37
38 #define IVI4_PIC_SIZE_ESC 7
39
40
41 static const struct {
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 },
47 { ff_ivi_row_haar8, ff_ivi_dc_haar_2d, 0 },
48 { ff_ivi_col_haar8, ff_ivi_dc_haar_2d, 0 },
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 */
56 { ff_ivi_inverse_haar_4x4, ff_ivi_dc_haar_2d, 1 },
57 { ff_ivi_inverse_slant_4x4, ff_ivi_dc_slant_2d, 1 },
58 { NULL, NULL, 0 }, /* no transform 4x4 */
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 },
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 *
73 * @param[in,out] bc the Bitstream context
74 * @return number of wavelet bands or 0 on error
75 */
76 static int decode_plane_subdivision(BitstreamContext *bc)
77 {
78 int i;
79
80 switch (bitstream_read(bc, 2)) {
81 case 3:
82 return 1;
83 case 2:
84 for (i = 0; i < 4; i++)
85 if (bitstream_read(bc, 2) != 3)
86 return 0;
87 return 4;
88 default:
89 return 0;
90 }
91 }
92
93 static inline int scale_tile_size(int def_size, int size_factor)
94 {
95 return size_factor == 15 ? def_size : (size_factor + 1) << 5;
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 */
105 static int decode_pic_hdr(IVI45DecContext *ctx, AVCodecContext *avctx)
106 {
107 int pic_size_indx, i, p;
108 IVIPicConfig pic_conf;
109
110 if (bitstream_read(&ctx->bc, 18) != 0x3FFF8) {
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;
116 ctx->frame_type = bitstream_read(&ctx->bc, 3);
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
122 if (ctx->frame_type == IVI4_FRAMETYPE_BIDIR)
123 ctx->has_b_frames = 1;
124
125 ctx->has_transp = bitstream_read_bit(&ctx->bc);
126
127 /* unknown bit: Mac decoder ignores this bit, XANIM returns error */
128 if (bitstream_read_bit(&ctx->bc)) {
129 av_log(avctx, AV_LOG_ERROR, "Sync bit is set!\n");
130 return AVERROR_INVALIDDATA;
131 }
132
133 ctx->data_size = bitstream_read_bit(&ctx->bc) ? bitstream_read(&ctx->bc, 24) : 0;
134
135 /* null frames don't contain anything else so we just return */
136 if (ctx->frame_type >= IVI4_FRAMETYPE_NULL_FIRST) {
137 ff_dlog(avctx, "Null frame encountered!\n");
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 */
144 if (bitstream_read_bit(&ctx->bc)) {
145 bitstream_skip(&ctx->bc, 32);
146 ff_dlog(avctx, "Password-protected clip!\n");
147 }
148
149 pic_size_indx = bitstream_read(&ctx->bc, 3);
150 if (pic_size_indx == IVI4_PIC_SIZE_ESC) {
151 pic_conf.pic_height = bitstream_read(&ctx->bc, 16);
152 pic_conf.pic_width = bitstream_read(&ctx->bc, 16);
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. */
159 ctx->uses_tiling = bitstream_read_bit(&ctx->bc);
160 if (ctx->uses_tiling) {
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));
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. */
169 if (bitstream_read(&ctx->bc, 2)) {
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 */
177 pic_conf.luma_bands = decode_plane_subdivision(&ctx->bc);
178 if (pic_conf.luma_bands)
179 pic_conf.chroma_bands = decode_plane_subdivision(&ctx->bc);
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)) {
189 if (ff_ivi_init_planes(ctx->planes, &pic_conf, 1)) {
190 av_log(avctx, AV_LOG_ERROR, "Couldn't reallocate color planes!\n");
191 ctx->pic_conf.luma_bands = 0;
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
213 ctx->frame_num = bitstream_read_bit(&ctx->bc) ? bitstream_read(&ctx->bc, 20) : 0;
214
215 /* skip decTimeEst field if present */
216 if (bitstream_read_bit(&ctx->bc))
217 bitstream_skip(&ctx->bc, 8);
218
219 /* decode macroblock and block huffman codebooks */
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))
222 return AVERROR_INVALIDDATA;
223
224 ctx->rvmap_sel = bitstream_read_bit(&ctx->bc) ? bitstream_read(&ctx->bc, 3) : 8;
225
226 ctx->in_imf = bitstream_read_bit(&ctx->bc);
227 ctx->in_q = bitstream_read_bit(&ctx->bc);
228
229 ctx->pic_glob_quant = bitstream_read(&ctx->bc, 5);
230
231 /* TODO: ignore this parameter if unused */
232 ctx->unknown1 = bitstream_read_bit(&ctx->bc) ? bitstream_read(&ctx->bc, 3) : 0;
233
234 ctx->checksum = bitstream_read_bit(&ctx->bc) ? bitstream_read(&ctx->bc, 16) : 0;
235
236 /* skip picture header extension if any */
237 while (bitstream_read_bit(&ctx->bc)) {
238 ff_dlog(avctx, "Pic hdr extension encountered!\n");
239 bitstream_skip(&ctx->bc, 8);
240 }
241
242 if (bitstream_read_bit(&ctx->bc)) {
243 av_log(avctx, AV_LOG_ERROR, "Bad blocks bits encountered!\n");
244 }
245
246 bitstream_align(&ctx->bc);
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 */
260 static int decode_band_hdr(IVI45DecContext *ctx, IVIBandDesc *band,
261 AVCodecContext *avctx)
262 {
263 int plane, band_num, indx, transform_id, scan_indx;
264 int i;
265
266 plane = bitstream_read(&ctx->bc, 2);
267 band_num = bitstream_read(&ctx->bc, 4);
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
273 band->is_empty = bitstream_read_bit(&ctx->bc);
274 if (!band->is_empty) {
275 int old_blk_size = band->blk_size;
276 /* skip header size
277 * If header size is not given, header size is 4 bytes. */
278 if (bitstream_read_bit(&ctx->bc))
279 bitstream_skip(&ctx->bc, 16);
280
281 band->is_halfpel = bitstream_read(&ctx->bc, 2);
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 }
287 if (!band->is_halfpel)
288 ctx->uses_fullpel = 1;
289
290 band->checksum_present = bitstream_read_bit(&ctx->bc);
291 if (band->checksum_present)
292 band->checksum = bitstream_read(&ctx->bc, 16);
293
294 indx = bitstream_read(&ctx->bc, 2);
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
302 band->inherit_mv = bitstream_read_bit(&ctx->bc);
303 band->inherit_qdelta = bitstream_read_bit(&ctx->bc);
304
305 band->glob_quant = bitstream_read(&ctx->bc, 5);
306
307 if (!bitstream_read_bit(&ctx->bc) || ctx->frame_type == IVI4_FRAMETYPE_INTRA) {
308 transform_id = bitstream_read(&ctx->bc, 5);
309 if (transform_id >= FF_ARRAY_ELEMS(transforms) ||
310 !transforms[transform_id].inv_trans) {
311 avpriv_request_sample(avctx, "Transform %d", transform_id);
312 return AVERROR_PATCHWELCOME;
313 }
314 if ((transform_id >= 7 && transform_id <= 9) ||
315 transform_id == 17) {
316 avpriv_request_sample(avctx, "DCT transform");
317 return AVERROR_PATCHWELCOME;
318 }
319
320 if ((transform_id >= 0 && transform_id <= 2) || transform_id == 10)
321 ctx->uses_haar = 1;
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;
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;
333
334 scan_indx = bitstream_read(&ctx->bc, 4);
335 if (scan_indx == 15) {
336 av_log(avctx, AV_LOG_ERROR, "Custom scan pattern encountered!\n");
337 return AVERROR_INVALIDDATA;
338 }
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
345 band->scan = scan_index_to_tab[scan_indx];
346
347 band->quant_mat = bitstream_read(&ctx->bc, 5);
348 if (band->quant_mat >= FF_ARRAY_ELEMS(quant_index_to_tab)) {
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;
357 return AVERROR_INVALIDDATA;
358 }
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 }
366 if (band->quant_mat < 0) {
367 av_log(avctx, AV_LOG_ERROR, "Invalid quant_mat inherited\n");
368 return AVERROR_INVALIDDATA;
369 }
370 }
371
372 /* decode block huffman codebook */
373 if (!bitstream_read_bit(&ctx->bc))
374 band->blk_vlc.tab = ctx->blk_vlc.tab;
375 else
376 if (ff_ivi_dec_huff_desc(&ctx->bc, 1, IVI_BLK_HUFF,
377 &band->blk_vlc, avctx))
378 return AVERROR_INVALIDDATA;
379
380 /* select appropriate rvmap table for this band */
381 band->rvmap_sel = bitstream_read_bit(&ctx->bc) ? bitstream_read(&ctx->bc, 3) : 8;
382
383 /* decode rvmap probability corrections if any */
384 band->num_corr = 0; /* there is no corrections */
385 if (bitstream_read_bit(&ctx->bc)) {
386 band->num_corr = bitstream_read(&ctx->bc, 8); /* get number of correction pairs */
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++)
395 band->corr[i] = bitstream_read(&ctx->bc, 8);
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
411 bitstream_align(&ctx->bc);
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 */
427 static int decode_mb_info(IVI45DecContext *ctx, IVIBandDesc *band,
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;
440 mb_type_bits = ctx->frame_type == IVI4_FRAMETYPE_BIDIR ? 2 : 1;
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;
453 mb->b_mv_x =
454 mb->b_mv_y = 0;
455
456 if (bitstream_read_bit(&ctx->bc)) {
457 if (ctx->frame_type == IVI4_FRAMETYPE_INTRA) {
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) {
466 mb->q_delta = bitstream_read_vlc(&ctx->bc,
467 ctx->mb_vlc.tab->table,
468 IVI_VLC_BITS, 1);
469 mb->q_delta = IVI_TOSIGNED(mb->q_delta);
470 }
471
472 mb->mv_x = mb->mv_y = 0; /* no motion vector coded */
473 if (band->inherit_mv && ref_mb) {
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) {
485 /* copy mb_type from corresponding reference mb */
486 if (!ref_mb)
487 return AVERROR_INVALIDDATA;
488 mb->type = ref_mb->type;
489 } else if (ctx->frame_type == IVI4_FRAMETYPE_INTRA ||
490 ctx->frame_type == IVI4_FRAMETYPE_INTRA1) {
491 mb->type = 0; /* mb_type is always INTRA for intra-frames */
492 } else {
493 mb->type = bitstream_read(&ctx->bc, mb_type_bits);
494 }
495
496 mb->cbp = bitstream_read(&ctx->bc, blks_per_mb);
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)) {
503 mb->q_delta = bitstream_read_vlc(&ctx->bc,
504 ctx->mb_vlc.tab->table,
505 IVI_VLC_BITS, 1);
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) {
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 }
522 } else {
523 /* decode motion vector deltas */
524 mv_delta = bitstream_read_vlc(&ctx->bc,
525 ctx->mb_vlc.tab->table,
526 IVI_VLC_BITS, 1);
527 mv_y += IVI_TOSIGNED(mv_delta);
528 mv_delta = bitstream_read_vlc(&ctx->bc,
529 ctx->mb_vlc.tab->table,
530 IVI_VLC_BITS, 1);
531 mv_x += IVI_TOSIGNED(mv_delta);
532 mb->mv_x = mv_x;
533 mb->mv_y = mv_y;
534 if (mb->type == 3) {
535 mv_delta = bitstream_read_vlc(&ctx->bc,
536 ctx->mb_vlc.tab->table,
537 IVI_VLC_BITS, 1);
538 mv_y += IVI_TOSIGNED(mv_delta);
539 mv_delta = bitstream_read_vlc(&ctx->bc,
540 ctx->mb_vlc.tab->table,
541 IVI_VLC_BITS, 1);
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;
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
565 bitstream_align(&ctx->bc);
566
567 return 0;
568 }
569
570
571 /**
572 * Rearrange decoding and reference buffers.
573 *
574 * @param[in,out] ctx pointer to the decoder context
575 */
576 static void switch_buffers(IVI45DecContext *ctx)
577 {
578 int is_prev_ref = 0, is_ref = 0;
579
580 switch (ctx->prev_frame_type) {
581 case IVI4_FRAMETYPE_INTRA:
582 case IVI4_FRAMETYPE_INTRA1:
583 case IVI4_FRAMETYPE_INTER:
584 is_prev_ref = 1;
585 break;
586 }
587
588 switch (ctx->frame_type) {
589 case IVI4_FRAMETYPE_INTRA:
590 case IVI4_FRAMETYPE_INTRA1:
591 case IVI4_FRAMETYPE_INTER:
592 is_ref = 1;
593 break;
594 }
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 }
602 }
603
604
605 static int is_nonnull_frame(IVI45DecContext *ctx)
606 {
607 return ctx->frame_type < IVI4_FRAMETYPE_NULL_FIRST;
608 }
609
610
611 static av_cold int decode_init(AVCodecContext *avctx)
612 {
613 IVI45DecContext *ctx = avctx->priv_data;
614
615 ff_ivi_init_static_vlc();
616
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));
619
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
625 avctx->pix_fmt = AV_PIX_FMT_YUV410P;
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;
632
633 ctx->is_indeo4 = 1;
634 ctx->show_indeo4_info = 1;
635
636 ctx->dst_buf = 0;
637 ctx->ref_buf = 1;
638 ctx->b_ref_buf = 3; /* buffer 2 is used for scalability mode */
639 ctx->p_frame = av_frame_alloc();
640 if (!ctx->p_frame)
641 return AVERROR(ENOMEM);
642
643 return 0;
644 }
645
646
647 AVCodec ff_indeo4_decoder = {
648 .name = "indeo4",
649 .long_name = NULL_IF_CONFIG_SMALL("Intel Indeo Video Interactive 4"),
650 .type = AVMEDIA_TYPE_VIDEO,
651 .id = AV_CODEC_ID_INDEO4,
652 .priv_data_size = sizeof(IVI45DecContext),
653 .init = decode_init,
654 .close = ff_ivi_decode_close,
655 .decode = ff_ivi_decode_frame,
656 .capabilities = AV_CODEC_CAP_DR1,
657 };