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