FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/mv30.c
Date: 2025-10-10 03:51:19
Exec Total Coverage
Lines: 0 410 0.0%
Functions: 0 18 0.0%
Branches: 0 142 0.0%

Line Branch Exec Source
1 /*
2 * MidiVid MV30 decoder
3 *
4 * Copyright (c) 2020 Paul B Mahol
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 #include <stddef.h>
24 #include <string.h>
25
26 #include "libavutil/attributes.h"
27 #include "libavutil/mem.h"
28 #include "libavutil/thread.h"
29
30 #include "avcodec.h"
31 #include "bytestream.h"
32 #include "codec_internal.h"
33 #include "copy_block.h"
34 #include "decode.h"
35 #include "mathops.h"
36 #include "blockdsp.h"
37 #include "get_bits.h"
38 #include "aandcttab.h"
39
40 #define CBP_VLC_BITS 9
41
42 typedef struct MV30Context {
43 GetBitContext gb;
44
45 int intra_quant;
46 int inter_quant;
47 int is_inter;
48 int mode_size;
49 int nb_mvectors;
50
51 int block[6][64];
52 int16_t *mvectors;
53 unsigned int mvectors_size;
54 int16_t *coeffs;
55 unsigned int coeffs_size;
56
57 int16_t intraq_tab[2][64];
58 int16_t interq_tab[2][64];
59
60 BlockDSPContext bdsp;
61 AVFrame *prev_frame;
62 } MV30Context;
63
64 static VLCElem cbp_tab[1 << CBP_VLC_BITS];
65
66 static const uint8_t luma_tab[] = {
67 12, 12, 15, 19, 25, 34, 40, 48,
68 12, 12, 18, 22, 27, 44, 47, 46,
69 17, 18, 21, 26, 35, 46, 52, 47,
70 18, 20, 24, 28, 40, 61, 59, 51,
71 20, 24, 32, 43, 50, 72, 72, 63,
72 25, 31, 42, 48, 58, 72, 81, 75,
73 38, 46, 54, 61, 71, 84, 88, 85,
74 50, 61, 65, 68, 79, 78, 86, 91,
75 };
76
77 static const uint8_t chroma_tab[] = {
78 12, 16, 24, 47, 99, 99, 99, 99,
79 16, 21, 26, 66, 99, 99, 99, 99,
80 24, 26, 56, 99, 99, 99, 99, 99,
81 47, 66, 99, 99, 99, 99, 99, 99,
82 99, 99, 99, 99, 99, 99, 99, 99,
83 99, 99, 99, 99, 99, 99, 99, 99,
84 99, 99, 99, 99, 99, 99, 99, 99,
85 99, 99, 99, 99, 99, 99, 99, 99,
86 };
87
88 static const uint8_t zigzag[] = {
89 0, 1, 8, 9, 16, 2, 3, 10,
90 17, 24, 32, 25, 18, 11, 4, 5,
91 12, 19, 26, 33, 40, 48, 41, 34,
92 27, 20, 13, 6, 7, 14, 21, 28,
93 35, 42, 49, 56, 57, 50, 43, 36,
94 29, 22, 15, 23, 30, 37, 44, 51,
95 58, 59, 52, 45, 38, 31, 39, 46,
96 53, 60, 61, 54, 47, 55, 62, 63,
97 };
98
99 static void get_qtable(int16_t *table, int quant, const uint8_t *quant_tab)
100 {
101 int factor = quant < 50 ? 5000 / FFMAX(quant, 1) : 200 - FFMIN(quant, 100) * 2;
102
103 for (int i = 0; i < 64; i++) {
104 table[i] = av_clip((quant_tab[i] * factor + 0x32) / 100, 1, 0x7fff);
105 table[i] = ((int)ff_aanscales[i] * (int)table[i] + 0x800) >> 12;
106 }
107 }
108
109 static inline void idct_1d(unsigned *blk, int step)
110 {
111 const unsigned t0 = blk[0 * step] + blk[4 * step];
112 const unsigned t1 = blk[0 * step] - blk[4 * step];
113 const unsigned t2 = blk[2 * step] + blk[6 * step];
114 const unsigned t3 = ((int)((blk[2 * step] - blk[6 * step]) * 362U) >> 8) - t2;
115 const unsigned t4 = t0 + t2;
116 const unsigned t5 = t0 - t2;
117 const unsigned t6 = t1 + t3;
118 const unsigned t7 = t1 - t3;
119 const unsigned t8 = blk[5 * step] + blk[3 * step];
120 const unsigned t9 = blk[5 * step] - blk[3 * step];
121 const unsigned tA = blk[1 * step] + blk[7 * step];
122 const unsigned tB = blk[1 * step] - blk[7 * step];
123 const unsigned tC = t8 + tA;
124 const unsigned tD = (int)((tB + t9) * 473U) >> 8;
125 const unsigned tE = (((int)(t9 * -669U) >> 8) - tC) + tD;
126 const unsigned tF = ((int)((tA - t8) * 362U) >> 8) - tE;
127 const unsigned t10 = (((int)(tB * 277U) >> 8) - tD) + tF;
128
129 blk[0 * step] = t4 + tC;
130 blk[1 * step] = t6 + tE;
131 blk[2 * step] = t7 + tF;
132 blk[3 * step] = t5 - t10;
133 blk[4 * step] = t5 + t10;
134 blk[5 * step] = t7 - tF;
135 blk[6 * step] = t6 - tE;
136 blk[7 * step] = t4 - tC;
137 }
138
139 static void idct_put(uint8_t *dst, int stride, int *block)
140 {
141 for (int i = 0; i < 8; i++) {
142 if ((block[0x08 + i] |
143 block[0x10 + i] |
144 block[0x18 + i] |
145 block[0x20 + i] |
146 block[0x28 + i] |
147 block[0x30 + i] |
148 block[0x38 + i]) == 0) {
149 block[0x08 + i] = block[i];
150 block[0x10 + i] = block[i];
151 block[0x18 + i] = block[i];
152 block[0x20 + i] = block[i];
153 block[0x28 + i] = block[i];
154 block[0x30 + i] = block[i];
155 block[0x38 + i] = block[i];
156 } else {
157 idct_1d(block + i, 8);
158 }
159 }
160
161 for (int i = 0; i < 8; i++) {
162 idct_1d(block, 1);
163 for (int j = 0; j < 8; j++)
164 dst[j] = av_clip_uint8((block[j] >> 5) + 128);
165 block += 8;
166 dst += stride;
167 }
168 }
169
170 static void idct_add(uint8_t *dst, int stride,
171 const uint8_t *src, int in_linesize, int *block)
172 {
173 for (int i = 0; i < 8; i++) {
174 if ((block[0x08 + i] |
175 block[0x10 + i] |
176 block[0x18 + i] |
177 block[0x20 + i] |
178 block[0x28 + i] |
179 block[0x30 + i] |
180 block[0x38 + i]) == 0) {
181 block[0x08 + i] = block[i];
182 block[0x10 + i] = block[i];
183 block[0x18 + i] = block[i];
184 block[0x20 + i] = block[i];
185 block[0x28 + i] = block[i];
186 block[0x30 + i] = block[i];
187 block[0x38 + i] = block[i];
188 } else {
189 idct_1d(block + i, 8);
190 }
191 }
192
193 for (int i = 0; i < 8; i++) {
194 idct_1d(block, 1);
195 for (int j = 0; j < 8; j++)
196 dst[j] = av_clip_uint8((block[j] >> 5) + src[j]);
197 block += 8;
198 dst += stride;
199 src += in_linesize;
200 }
201 }
202
203 static inline void idct2_1d(int *blk, int step)
204 {
205 const unsigned int t0 = blk[0 * step];
206 const unsigned int t1 = blk[1 * step];
207 const unsigned int t2 = (int)(t1 * 473U) >> 8;
208 const unsigned int t3 = t2 - t1;
209 const unsigned int t4 = ((int)(t1 * 362U) >> 8) - t3;
210 const unsigned int t5 = (((int)(t1 * 277U) >> 8) - t2) + t4;
211
212 blk[0 * step] = t1 + t0;
213 blk[1 * step] = t0 + t3;
214 blk[2 * step] = t4 + t0;
215 blk[3 * step] = t0 - t5;
216 blk[4 * step] = t5 + t0;
217 blk[5 * step] = t0 - t4;
218 blk[6 * step] = t0 - t3;
219 blk[7 * step] = t0 - t1;
220 }
221
222 static void idct2_put(uint8_t *dst, int stride, int *block)
223 {
224 for (int i = 0; i < 2; i++) {
225 if ((block[0x08 + i]) == 0) {
226 block[0x08 + i] = block[i];
227 block[0x10 + i] = block[i];
228 block[0x18 + i] = block[i];
229 block[0x20 + i] = block[i];
230 block[0x28 + i] = block[i];
231 block[0x30 + i] = block[i];
232 block[0x38 + i] = block[i];
233 } else {
234 idct2_1d(block + i, 8);
235 }
236 }
237
238 for (int i = 0; i < 8; i++) {
239 if (block[1] == 0) {
240 for (int j = 0; j < 8; j++)
241 dst[j] = av_clip_uint8((block[0] >> 5) + 128);
242 } else {
243 idct2_1d(block, 1);
244 for (int j = 0; j < 8; j++)
245 dst[j] = av_clip_uint8((block[j] >> 5) + 128);
246 }
247 block += 8;
248 dst += stride;
249 }
250 }
251
252 static void idct2_add(uint8_t *dst, int stride,
253 const uint8_t *src, int in_linesize,
254 int *block)
255 {
256 for (int i = 0; i < 2; i++) {
257 if ((block[0x08 + i]) == 0) {
258 block[0x08 + i] = block[i];
259 block[0x10 + i] = block[i];
260 block[0x18 + i] = block[i];
261 block[0x20 + i] = block[i];
262 block[0x28 + i] = block[i];
263 block[0x30 + i] = block[i];
264 block[0x38 + i] = block[i];
265 } else {
266 idct2_1d(block + i, 8);
267 }
268 }
269
270 for (int i = 0; i < 8; i++) {
271 if (block[1] == 0) {
272 for (int j = 0; j < 8; j++)
273 dst[j] = av_clip_uint8((block[0] >> 5) + src[j]);
274 } else {
275 idct2_1d(block, 1);
276 for (int j = 0; j < 8; j++)
277 dst[j] = av_clip_uint8((block[j] >> 5) + src[j]);
278 }
279 block += 8;
280 dst += stride;
281 src += in_linesize;
282 }
283 }
284
285 static void update_inter_block(uint8_t *dst, int stride,
286 const uint8_t *src, int in_linesize,
287 int block)
288 {
289 for (int i = 0; i < 8; i++) {
290 for (int j = 0; j < 8; j++)
291 dst[j] = av_clip_uint8(block + src[j]);
292 dst += stride;
293 src += in_linesize;
294 }
295 }
296
297 static int decode_intra_block(AVCodecContext *avctx, int mode,
298 GetByteContext *gbyte, int16_t *qtab,
299 int *block, int *pfill,
300 uint8_t *dst, int linesize)
301 {
302 MV30Context *s = avctx->priv_data;
303 int fill;
304
305 switch (mode) {
306 case 0:
307 s->bdsp.fill_block_tab[1](dst, 128, linesize, 8);
308 break;
309 case 1:
310 fill = sign_extend(bytestream2_get_ne16(gbyte), 16);
311 pfill[0] += fill;
312 block[0] = ((int)((unsigned)pfill[0] * qtab[0]) >> 5) + 128;
313 s->bdsp.fill_block_tab[1](dst, block[0], linesize, 8);
314 break;
315 case 2:
316 memset(block, 0, sizeof(*block) * 64);
317 fill = sign_extend(bytestream2_get_ne16(gbyte), 16);
318 pfill[0] += fill;
319 block[0] = (unsigned)pfill[0] * qtab[0];
320 block[1] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[1];
321 block[8] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[8];
322 block[9] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[9];
323 idct2_put(dst, linesize, block);
324 break;
325 case 3:
326 fill = sign_extend(bytestream2_get_ne16(gbyte), 16);
327 pfill[0] += fill;
328 block[0] = (unsigned)pfill[0] * qtab[0];
329 for (int i = 1; i < 64; i++)
330 block[zigzag[i]] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[zigzag[i]];
331 idct_put(dst, linesize, block);
332 break;
333 }
334
335 return 0;
336 }
337
338 static int decode_inter_block(AVCodecContext *avctx, int mode,
339 GetByteContext *gbyte, int16_t *qtab,
340 int *block, int *pfill,
341 uint8_t *dst, int linesize,
342 const uint8_t *src, int in_linesize)
343 {
344 int fill;
345
346 switch (mode) {
347 case 0:
348 copy_block8(dst, src, linesize, in_linesize, 8);
349 break;
350 case 1:
351 fill = sign_extend(bytestream2_get_ne16(gbyte), 16);
352 pfill[0] += fill;
353 block[0] = (int)((unsigned)pfill[0] * qtab[0]) >> 5;
354 update_inter_block(dst, linesize, src, in_linesize, block[0]);
355 break;
356 case 2:
357 memset(block, 0, sizeof(*block) * 64);
358 fill = sign_extend(bytestream2_get_ne16(gbyte), 16);
359 pfill[0] += fill;
360 block[0] = (unsigned)pfill[0] * qtab[0];
361 block[1] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[1];
362 block[8] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[8];
363 block[9] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[9];
364 idct2_add(dst, linesize, src, in_linesize, block);
365 break;
366 case 3:
367 fill = sign_extend(bytestream2_get_ne16(gbyte), 16);
368 pfill[0] += fill;
369 block[0] = (unsigned)pfill[0] * qtab[0];
370 for (int i = 1; i < 64; i++)
371 block[zigzag[i]] = sign_extend(bytestream2_get_ne16(gbyte), 16) * qtab[zigzag[i]];
372 idct_add(dst, linesize, src, in_linesize, block);
373 break;
374 }
375
376 return 0;
377 }
378
379 static int decode_coeffs(GetBitContext *gb, int16_t *coeffs, int nb_codes)
380 {
381 memset(coeffs, 0, nb_codes * sizeof(*coeffs));
382
383 for (int i = 0; i < nb_codes;) {
384 int value = get_vlc2(gb, cbp_tab, CBP_VLC_BITS, 1);
385
386 if (value > 0) {
387 int x = get_bits(gb, value);
388
389 if (x < (1 << value) / 2) {
390 x = (1 << (value - 1)) + (x & ((1 << value) - 1 >> 1));
391 } else {
392 x = -(1 << (value - 1)) - (x & ((1 << value) - 1 >> 1));
393 }
394 coeffs[i++] = x;
395 } else {
396 int flag = get_bits1(gb);
397
398 i += get_bits(gb, 3 + flag * 3) + 1 + flag * 8;
399 }
400 }
401
402 return 0;
403 }
404
405 static int decode_intra(AVCodecContext *avctx, GetBitContext *gb, AVFrame *frame)
406 {
407 MV30Context *s = avctx->priv_data;
408 GetBitContext mgb;
409 uint8_t *dst[6];
410 int linesize[6];
411 int ret;
412
413 mgb = *gb;
414 if (get_bits_left(gb) < s->mode_size * 8)
415 return AVERROR_INVALIDDATA;
416
417 skip_bits_long(gb, s->mode_size * 8);
418
419 linesize[0] = frame->linesize[0];
420 linesize[1] = frame->linesize[0];
421 linesize[2] = frame->linesize[0];
422 linesize[3] = frame->linesize[0];
423 linesize[4] = frame->linesize[1];
424 linesize[5] = frame->linesize[2];
425
426 for (int y = 0; y < avctx->height; y += 16) {
427 GetByteContext gbyte;
428 int pfill[3][1] = { {0} };
429 int nb_codes = get_bits(gb, 16);
430
431 av_fast_padded_malloc(&s->coeffs, &s->coeffs_size, nb_codes * sizeof(*s->coeffs));
432 if (!s->coeffs)
433 return AVERROR(ENOMEM);
434 ret = decode_coeffs(gb, s->coeffs, nb_codes);
435 if (ret < 0)
436 return ret;
437
438 bytestream2_init(&gbyte, (uint8_t *)s->coeffs, nb_codes * sizeof(*s->coeffs));
439
440 for (int x = 0; x < avctx->width; x += 16) {
441 dst[0] = frame->data[0] + linesize[0] * y + x;
442 dst[1] = frame->data[0] + linesize[0] * y + x + 8;
443 dst[2] = frame->data[0] + linesize[0] * (y + 8) + x;
444 dst[3] = frame->data[0] + linesize[0] * (y + 8) + x + 8;
445 dst[4] = frame->data[1] + linesize[4] * (y >> 1) + (x >> 1);
446 dst[5] = frame->data[2] + linesize[5] * (y >> 1) + (x >> 1);
447
448 for (int b = 0; b < 6; b++) {
449 int mode = get_bits_le(&mgb, 2);
450
451 ret = decode_intra_block(avctx, mode, &gbyte, s->intraq_tab[b >= 4],
452 s->block[b],
453 pfill[(b >= 4) + (b >= 5)],
454 dst[b], linesize[b]);
455 if (ret < 0)
456 return ret;
457 }
458 }
459 }
460
461 return 0;
462 }
463
464 static int decode_inter(AVCodecContext *avctx, GetBitContext *gb,
465 AVFrame *frame, AVFrame *prev)
466 {
467 MV30Context *s = avctx->priv_data;
468 GetBitContext mask;
469 GetBitContext mgb;
470 GetByteContext mv;
471 const int mask_size = ((avctx->height >> 4) * (avctx->width >> 4) * 2 + 7) / 8;
472 uint8_t *dst[6], *src[6];
473 int in_linesize[6];
474 int linesize[6];
475 int ret, cnt = 0;
476 int flags = 0;
477
478 in_linesize[0] = prev->linesize[0];
479 in_linesize[1] = prev->linesize[0];
480 in_linesize[2] = prev->linesize[0];
481 in_linesize[3] = prev->linesize[0];
482 in_linesize[4] = prev->linesize[1];
483 in_linesize[5] = prev->linesize[2];
484
485 linesize[0] = frame->linesize[0];
486 linesize[1] = frame->linesize[0];
487 linesize[2] = frame->linesize[0];
488 linesize[3] = frame->linesize[0];
489 linesize[4] = frame->linesize[1];
490 linesize[5] = frame->linesize[2];
491
492 av_fast_padded_malloc(&s->mvectors, &s->mvectors_size, 2 * s->nb_mvectors * sizeof(*s->mvectors));
493 if (!s->mvectors) {
494 ret = AVERROR(ENOMEM);
495 goto fail;
496 }
497
498 mask = *gb;
499 skip_bits_long(gb, mask_size * 8);
500 mgb = *gb;
501 skip_bits_long(gb, s->mode_size * 8);
502
503 ret = decode_coeffs(gb, s->mvectors, 2 * s->nb_mvectors);
504 if (ret < 0)
505 goto fail;
506
507 bytestream2_init(&mv, (uint8_t *)s->mvectors, 2 * s->nb_mvectors * sizeof(*s->mvectors));
508
509 for (int y = 0; y < avctx->height; y += 16) {
510 GetByteContext gbyte;
511 int pfill[3][1] = { {0} };
512 int nb_codes = get_bits(gb, 16);
513
514 skip_bits(gb, 8);
515 if (get_bits_left(gb) < 0) {
516 ret = AVERROR_INVALIDDATA;
517 goto fail;
518 }
519
520 av_fast_padded_malloc(&s->coeffs, &s->coeffs_size, nb_codes * sizeof(*s->coeffs));
521 if (!s->coeffs) {
522 ret = AVERROR(ENOMEM);
523 goto fail;
524 }
525
526 ret = decode_coeffs(gb, s->coeffs, nb_codes);
527 if (ret < 0)
528 goto fail;
529
530 bytestream2_init(&gbyte, (uint8_t *)s->coeffs, nb_codes * sizeof(*s->coeffs));
531
532 for (int x = 0; x < avctx->width; x += 16) {
533 if (cnt >= 4)
534 cnt = 0;
535 if (cnt == 0) {
536 if (get_bits_left(&mask) < 8) {
537 ret = AVERROR_INVALIDDATA;
538 goto fail;
539 }
540 flags = get_bits(&mask, 8);
541 }
542
543 dst[0] = frame->data[0] + linesize[0] * y + x;
544 dst[1] = frame->data[0] + linesize[0] * y + x + 8;
545 dst[2] = frame->data[0] + linesize[0] * (y + 8) + x;
546 dst[3] = frame->data[0] + linesize[0] * (y + 8) + x + 8;
547 dst[4] = frame->data[1] + linesize[4] * (y >> 1) + (x >> 1);
548 dst[5] = frame->data[2] + linesize[5] * (y >> 1) + (x >> 1);
549
550 if ((flags >> (cnt)) & 1) {
551 int mv_x = sign_extend(bytestream2_get_ne16(&mv), 16);
552 int mv_y = sign_extend(bytestream2_get_ne16(&mv), 16);
553
554 int px = x + mv_x;
555 int py = y + mv_y;
556
557 if (px < 0 || px > FFALIGN(avctx->width , 16) - 16 ||
558 py < 0 || py > FFALIGN(avctx->height, 16) - 16)
559 return AVERROR_INVALIDDATA;
560
561 src[0] = prev->data[0] + in_linesize[0] * py + px;
562 src[1] = prev->data[0] + in_linesize[0] * py + px + 8;
563 src[2] = prev->data[0] + in_linesize[0] * (py + 8) + px;
564 src[3] = prev->data[0] + in_linesize[0] * (py + 8) + px + 8;
565 src[4] = prev->data[1] + in_linesize[4] * (py >> 1) + (px >> 1);
566 src[5] = prev->data[2] + in_linesize[5] * (py >> 1) + (px >> 1);
567
568 if ((flags >> (cnt + 4)) & 1) {
569 for (int b = 0; b < 6; b++)
570 copy_block8(dst[b], src[b], linesize[b], in_linesize[b], 8);
571 } else {
572 for (int b = 0; b < 6; b++) {
573 int mode = get_bits_le(&mgb, 2);
574
575 ret = decode_inter_block(avctx, mode, &gbyte, s->interq_tab[b >= 4],
576 s->block[b],
577 pfill[(b >= 4) + (b >= 5)],
578 dst[b], linesize[b],
579 src[b], in_linesize[b]);
580 if (ret < 0)
581 goto fail;
582 }
583 }
584 } else {
585 for (int b = 0; b < 6; b++) {
586 int mode = get_bits_le(&mgb, 2);
587
588 ret = decode_intra_block(avctx, mode, &gbyte, s->intraq_tab[b >= 4],
589 s->block[b],
590 pfill[(b >= 4) + (b >= 5)],
591 dst[b], linesize[b]);
592 if (ret < 0)
593 goto fail;
594 }
595 }
596
597 cnt++;
598 }
599 }
600
601 fail:
602 return ret;
603 }
604
605 static int decode_frame(AVCodecContext *avctx, AVFrame *frame,
606 int *got_frame, AVPacket *avpkt)
607 {
608 MV30Context *s = avctx->priv_data;
609 GetBitContext *gb = &s->gb;
610 int ret;
611
612 if ((ret = init_get_bits8(gb, avpkt->data, avpkt->size)) < 0)
613 return ret;
614
615 if ((ret = ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF)) < 0)
616 return ret;
617
618 s->intra_quant = get_bits(gb, 8);
619 s->inter_quant = s->intra_quant + get_sbits(gb, 8);
620 s->is_inter = get_bits_le(gb, 16);
621 s->mode_size = get_bits_le(gb, 16);
622 if (s->is_inter)
623 s->nb_mvectors = get_bits_le(gb, 16);
624
625 get_qtable(s->intraq_tab[0], s->intra_quant, luma_tab);
626 get_qtable(s->intraq_tab[1], s->intra_quant, chroma_tab);
627
628 if (s->is_inter == 0) {
629 frame->flags |= AV_FRAME_FLAG_KEY;
630 ret = decode_intra(avctx, gb, frame);
631 if (ret < 0)
632 return ret;
633 } else {
634 get_qtable(s->interq_tab[0], s->inter_quant, luma_tab);
635 get_qtable(s->interq_tab[1], s->inter_quant, chroma_tab);
636
637 if (!s->prev_frame->data[0]) {
638 av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n");
639 return AVERROR_INVALIDDATA;
640 }
641
642 frame->flags &= ~AV_FRAME_FLAG_KEY;
643 ret = decode_inter(avctx, gb, frame, s->prev_frame);
644 if (ret < 0)
645 return ret;
646 }
647
648 if ((ret = av_frame_replace(s->prev_frame, frame)) < 0)
649 return ret;
650
651 *got_frame = 1;
652
653 return avpkt->size;
654 }
655
656 static const uint8_t cbp_bits[] = {
657 2, 2, 3, 3, 3, 4, 5, 6, 7, 8, 9, 9,
658 };
659
660 static av_cold void init_static_data(void)
661 {
662 VLC_INIT_STATIC_TABLE_FROM_LENGTHS(cbp_tab, CBP_VLC_BITS,
663 FF_ARRAY_ELEMS(cbp_bits),
664 cbp_bits, 1, NULL, 0, 0, 0, 0);
665 }
666
667 static av_cold int decode_init(AVCodecContext *avctx)
668 {
669 MV30Context *s = avctx->priv_data;
670 static AVOnce init_static_once = AV_ONCE_INIT;
671
672 avctx->pix_fmt = AV_PIX_FMT_YUV420P;
673 avctx->color_range = AVCOL_RANGE_JPEG;
674
675 ff_blockdsp_init(&s->bdsp);
676
677 s->prev_frame = av_frame_alloc();
678 if (!s->prev_frame)
679 return AVERROR(ENOMEM);
680
681 ff_thread_once(&init_static_once, init_static_data);
682
683 return 0;
684 }
685
686 static av_cold void decode_flush(AVCodecContext *avctx)
687 {
688 MV30Context *s = avctx->priv_data;
689
690 av_frame_unref(s->prev_frame);
691 }
692
693 static av_cold int decode_close(AVCodecContext *avctx)
694 {
695 MV30Context *s = avctx->priv_data;
696
697 av_frame_free(&s->prev_frame);
698 av_freep(&s->coeffs);
699 s->coeffs_size = 0;
700 av_freep(&s->mvectors);
701 s->mvectors_size = 0;
702
703 return 0;
704 }
705
706 const FFCodec ff_mv30_decoder = {
707 .p.name = "mv30",
708 CODEC_LONG_NAME("MidiVid 3.0"),
709 .p.type = AVMEDIA_TYPE_VIDEO,
710 .p.id = AV_CODEC_ID_MV30,
711 .priv_data_size = sizeof(MV30Context),
712 .init = decode_init,
713 .close = decode_close,
714 FF_CODEC_DECODE_CB(decode_frame),
715 .flush = decode_flush,
716 .p.capabilities = AV_CODEC_CAP_DR1,
717 .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
718 };
719