xref: /third_party/ffmpeg/libavcodec/speedhq.c (revision cabdff1a)
1cabdff1aSopenharmony_ci/*
2cabdff1aSopenharmony_ci * NewTek SpeedHQ codec
3cabdff1aSopenharmony_ci * Copyright 2017 Steinar H. Gunderson
4cabdff1aSopenharmony_ci *
5cabdff1aSopenharmony_ci * This file is part of FFmpeg.
6cabdff1aSopenharmony_ci *
7cabdff1aSopenharmony_ci * FFmpeg is free software; you can redistribute it and/or
8cabdff1aSopenharmony_ci * modify it under the terms of the GNU Lesser General Public
9cabdff1aSopenharmony_ci * License as published by the Free Software Foundation; either
10cabdff1aSopenharmony_ci * version 2.1 of the License, or (at your option) any later version.
11cabdff1aSopenharmony_ci *
12cabdff1aSopenharmony_ci * FFmpeg is distributed in the hope that it will be useful,
13cabdff1aSopenharmony_ci * but WITHOUT ANY WARRANTY; without even the implied warranty of
14cabdff1aSopenharmony_ci * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15cabdff1aSopenharmony_ci * Lesser General Public License for more details.
16cabdff1aSopenharmony_ci *
17cabdff1aSopenharmony_ci * You should have received a copy of the GNU Lesser General Public
18cabdff1aSopenharmony_ci * License along with FFmpeg; if not, write to the Free Software
19cabdff1aSopenharmony_ci * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20cabdff1aSopenharmony_ci */
21cabdff1aSopenharmony_ci
22cabdff1aSopenharmony_ci/**
23cabdff1aSopenharmony_ci * @file
24cabdff1aSopenharmony_ci * NewTek SpeedHQ decoder.
25cabdff1aSopenharmony_ci */
26cabdff1aSopenharmony_ci
27cabdff1aSopenharmony_ci#define BITSTREAM_READER_LE
28cabdff1aSopenharmony_ci
29cabdff1aSopenharmony_ci#include "config.h"
30cabdff1aSopenharmony_ci#include "config_components.h"
31cabdff1aSopenharmony_ci#include "libavutil/attributes.h"
32cabdff1aSopenharmony_ci#include "libavutil/mem_internal.h"
33cabdff1aSopenharmony_ci
34cabdff1aSopenharmony_ci#include "avcodec.h"
35cabdff1aSopenharmony_ci#include "blockdsp.h"
36cabdff1aSopenharmony_ci#include "codec_internal.h"
37cabdff1aSopenharmony_ci#include "get_bits.h"
38cabdff1aSopenharmony_ci#include "idctdsp.h"
39cabdff1aSopenharmony_ci#include "internal.h"
40cabdff1aSopenharmony_ci#include "libavutil/thread.h"
41cabdff1aSopenharmony_ci#include "mathops.h"
42cabdff1aSopenharmony_ci#include "mpeg12dec.h"
43cabdff1aSopenharmony_ci#include "mpeg12data.h"
44cabdff1aSopenharmony_ci#include "mpeg12vlc.h"
45cabdff1aSopenharmony_ci
46cabdff1aSopenharmony_ci#define MAX_INDEX (64 - 1)
47cabdff1aSopenharmony_ci
48cabdff1aSopenharmony_ci/*
49cabdff1aSopenharmony_ci * 5 bits makes for very small tables, with no more than two lookups needed
50cabdff1aSopenharmony_ci * for the longest (10-bit) codes.
51cabdff1aSopenharmony_ci */
52cabdff1aSopenharmony_ci#define ALPHA_VLC_BITS 5
53cabdff1aSopenharmony_ci
54cabdff1aSopenharmony_citypedef struct SHQContext {
55cabdff1aSopenharmony_ci    AVCodecContext *avctx;
56cabdff1aSopenharmony_ci    BlockDSPContext bdsp;
57cabdff1aSopenharmony_ci    IDCTDSPContext idsp;
58cabdff1aSopenharmony_ci    ScanTable intra_scantable;
59cabdff1aSopenharmony_ci    int quant_matrix[64];
60cabdff1aSopenharmony_ci    enum { SHQ_SUBSAMPLING_420, SHQ_SUBSAMPLING_422, SHQ_SUBSAMPLING_444 }
61cabdff1aSopenharmony_ci        subsampling;
62cabdff1aSopenharmony_ci    enum { SHQ_NO_ALPHA, SHQ_RLE_ALPHA, SHQ_DCT_ALPHA } alpha_type;
63cabdff1aSopenharmony_ci} SHQContext;
64cabdff1aSopenharmony_ci
65cabdff1aSopenharmony_ci
66cabdff1aSopenharmony_ci/* AC codes: Very similar but not identical to MPEG-2. */
67cabdff1aSopenharmony_cistatic const uint16_t speedhq_vlc[123][2] = {
68cabdff1aSopenharmony_ci    {0x0001,  2}, {0x0003,  3}, {0x000E,  4}, {0x0007,  5},
69cabdff1aSopenharmony_ci    {0x0017,  5}, {0x0028,  6}, {0x0008,  6}, {0x006F,  7},
70cabdff1aSopenharmony_ci    {0x001F,  7}, {0x00C4,  8}, {0x0044,  8}, {0x005F,  8},
71cabdff1aSopenharmony_ci    {0x00DF,  8}, {0x007F,  8}, {0x00FF,  8}, {0x3E00, 14},
72cabdff1aSopenharmony_ci    {0x1E00, 14}, {0x2E00, 14}, {0x0E00, 14}, {0x3600, 14},
73cabdff1aSopenharmony_ci    {0x1600, 14}, {0x2600, 14}, {0x0600, 14}, {0x3A00, 14},
74cabdff1aSopenharmony_ci    {0x1A00, 14}, {0x2A00, 14}, {0x0A00, 14}, {0x3200, 14},
75cabdff1aSopenharmony_ci    {0x1200, 14}, {0x2200, 14}, {0x0200, 14}, {0x0C00, 15},
76cabdff1aSopenharmony_ci    {0x7400, 15}, {0x3400, 15}, {0x5400, 15}, {0x1400, 15},
77cabdff1aSopenharmony_ci    {0x6400, 15}, {0x2400, 15}, {0x4400, 15}, {0x0400, 15},
78cabdff1aSopenharmony_ci    {0x0002,  3}, {0x000C,  5}, {0x004F,  7}, {0x00E4,  8},
79cabdff1aSopenharmony_ci    {0x0004,  8}, {0x0D00, 13}, {0x1500, 13}, {0x7C00, 15},
80cabdff1aSopenharmony_ci    {0x3C00, 15}, {0x5C00, 15}, {0x1C00, 15}, {0x6C00, 15},
81cabdff1aSopenharmony_ci    {0x2C00, 15}, {0x4C00, 15}, {0xC800, 16}, {0x4800, 16},
82cabdff1aSopenharmony_ci    {0x8800, 16}, {0x0800, 16}, {0x0300, 13}, {0x1D00, 13},
83cabdff1aSopenharmony_ci    {0x0014,  5}, {0x0070,  7}, {0x003F,  8}, {0x00C0, 10},
84cabdff1aSopenharmony_ci    {0x0500, 13}, {0x0180, 12}, {0x0280, 12}, {0x0C80, 12},
85cabdff1aSopenharmony_ci    {0x0080, 12}, {0x0B00, 13}, {0x1300, 13}, {0x001C,  5},
86cabdff1aSopenharmony_ci    {0x0064,  8}, {0x0380, 12}, {0x1900, 13}, {0x0D80, 12},
87cabdff1aSopenharmony_ci    {0x0018,  6}, {0x00BF,  8}, {0x0480, 12}, {0x0B80, 12},
88cabdff1aSopenharmony_ci    {0x0038,  6}, {0x0040,  9}, {0x0900, 13}, {0x0030,  7},
89cabdff1aSopenharmony_ci    {0x0780, 12}, {0x2800, 16}, {0x0010,  7}, {0x0A80, 12},
90cabdff1aSopenharmony_ci    {0x0050,  7}, {0x0880, 12}, {0x000F,  7}, {0x1100, 13},
91cabdff1aSopenharmony_ci    {0x002F,  7}, {0x0100, 13}, {0x0084,  8}, {0x5800, 16},
92cabdff1aSopenharmony_ci    {0x00A4,  8}, {0x9800, 16}, {0x0024,  8}, {0x1800, 16},
93cabdff1aSopenharmony_ci    {0x0140,  9}, {0xE800, 16}, {0x01C0,  9}, {0x6800, 16},
94cabdff1aSopenharmony_ci    {0x02C0, 10}, {0xA800, 16}, {0x0F80, 12}, {0x0580, 12},
95cabdff1aSopenharmony_ci    {0x0980, 12}, {0x0E80, 12}, {0x0680, 12}, {0x1F00, 13},
96cabdff1aSopenharmony_ci    {0x0F00, 13}, {0x1700, 13}, {0x0700, 13}, {0x1B00, 13},
97cabdff1aSopenharmony_ci    {0xF800, 16}, {0x7800, 16}, {0xB800, 16}, {0x3800, 16},
98cabdff1aSopenharmony_ci    {0xD800, 16},
99cabdff1aSopenharmony_ci    {0x0020,  6}, /* escape */
100cabdff1aSopenharmony_ci    {0x0006,  4}  /* EOB */
101cabdff1aSopenharmony_ci};
102cabdff1aSopenharmony_ci
103cabdff1aSopenharmony_cistatic const uint8_t speedhq_level[121] = {
104cabdff1aSopenharmony_ci     1,  2,  3,  4,  5,  6,  7,  8,
105cabdff1aSopenharmony_ci     9, 10, 11, 12, 13, 14, 15, 16,
106cabdff1aSopenharmony_ci    17, 18, 19, 20, 21, 22, 23, 24,
107cabdff1aSopenharmony_ci    25, 26, 27, 28, 29, 30, 31, 32,
108cabdff1aSopenharmony_ci    33, 34, 35, 36, 37, 38, 39, 40,
109cabdff1aSopenharmony_ci     1,  2,  3,  4,  5,  6,  7,  8,
110cabdff1aSopenharmony_ci     9, 10, 11, 12, 13, 14, 15, 16,
111cabdff1aSopenharmony_ci    17, 18, 19, 20,  1,  2,  3,  4,
112cabdff1aSopenharmony_ci     5,  6,  7,  8,  9, 10, 11,  1,
113cabdff1aSopenharmony_ci     2,  3,  4,  5,  1,  2,  3,  4,
114cabdff1aSopenharmony_ci     1,  2,  3,  1,  2,  3,  1,  2,
115cabdff1aSopenharmony_ci     1,  2,  1,  2,  1,  2,  1,  2,
116cabdff1aSopenharmony_ci     1,  2,  1,  2,  1,  2,  1,  2,
117cabdff1aSopenharmony_ci     1,  2,  1,  1,  1,  1,  1,  1,
118cabdff1aSopenharmony_ci     1,  1,  1,  1,  1,  1,  1,  1,
119cabdff1aSopenharmony_ci     1,
120cabdff1aSopenharmony_ci};
121cabdff1aSopenharmony_ci
122cabdff1aSopenharmony_cistatic const uint8_t speedhq_run[121] = {
123cabdff1aSopenharmony_ci     0,  0,  0,  0,  0,  0,  0,  0,
124cabdff1aSopenharmony_ci     0,  0,  0,  0,  0,  0,  0,  0,
125cabdff1aSopenharmony_ci     0,  0,  0,  0,  0,  0,  0,  0,
126cabdff1aSopenharmony_ci     0,  0,  0,  0,  0,  0,  0,  0,
127cabdff1aSopenharmony_ci     0,  0,  0,  0,  0,  0,  0,  0,
128cabdff1aSopenharmony_ci     1,  1,  1,  1,  1,  1,  1,  1,
129cabdff1aSopenharmony_ci     1,  1,  1,  1,  1,  1,  1,  1,
130cabdff1aSopenharmony_ci     1,  1,  1,  1,  2,  2,  2,  2,
131cabdff1aSopenharmony_ci     2,  2,  2,  2,  2,  2,  2,  3,
132cabdff1aSopenharmony_ci     3,  3,  3,  3,  4,  4,  4,  4,
133cabdff1aSopenharmony_ci     5,  5,  5,  6,  6,  6,  7,  7,
134cabdff1aSopenharmony_ci     8,  8,  9,  9, 10, 10, 11, 11,
135cabdff1aSopenharmony_ci    12, 12, 13, 13, 14, 14, 15, 15,
136cabdff1aSopenharmony_ci    16, 16, 17, 18, 19, 20, 21, 22,
137cabdff1aSopenharmony_ci    23, 24, 25, 26, 27, 28, 29, 30,
138cabdff1aSopenharmony_ci    31,
139cabdff1aSopenharmony_ci};
140cabdff1aSopenharmony_ci
141cabdff1aSopenharmony_ciRLTable ff_rl_speedhq = {
142cabdff1aSopenharmony_ci    121,
143cabdff1aSopenharmony_ci    121,
144cabdff1aSopenharmony_ci    speedhq_vlc,
145cabdff1aSopenharmony_ci    speedhq_run,
146cabdff1aSopenharmony_ci    speedhq_level,
147cabdff1aSopenharmony_ci};
148cabdff1aSopenharmony_ci
149cabdff1aSopenharmony_ci#if CONFIG_SPEEDHQ_DECODER
150cabdff1aSopenharmony_ci/* NOTE: The first element is always 16, unscaled. */
151cabdff1aSopenharmony_cistatic const uint8_t unscaled_quant_matrix[64] = {
152cabdff1aSopenharmony_ci    16, 16, 19, 22, 26, 27, 29, 34,
153cabdff1aSopenharmony_ci    16, 16, 22, 24, 27, 29, 34, 37,
154cabdff1aSopenharmony_ci    19, 22, 26, 27, 29, 34, 34, 38,
155cabdff1aSopenharmony_ci    22, 22, 26, 27, 29, 34, 37, 40,
156cabdff1aSopenharmony_ci    22, 26, 27, 29, 32, 35, 40, 48,
157cabdff1aSopenharmony_ci    26, 27, 29, 32, 35, 40, 48, 58,
158cabdff1aSopenharmony_ci    26, 27, 29, 34, 38, 46, 56, 69,
159cabdff1aSopenharmony_ci    27, 29, 35, 38, 46, 56, 69, 83
160cabdff1aSopenharmony_ci};
161cabdff1aSopenharmony_ci
162cabdff1aSopenharmony_cistatic VLC dc_lum_vlc_le;
163cabdff1aSopenharmony_cistatic VLC dc_chroma_vlc_le;
164cabdff1aSopenharmony_cistatic VLC dc_alpha_run_vlc_le;
165cabdff1aSopenharmony_cistatic VLC dc_alpha_level_vlc_le;
166cabdff1aSopenharmony_ci
167cabdff1aSopenharmony_cistatic inline int decode_dc_le(GetBitContext *gb, int component)
168cabdff1aSopenharmony_ci{
169cabdff1aSopenharmony_ci    int code, diff;
170cabdff1aSopenharmony_ci
171cabdff1aSopenharmony_ci    if (component == 0 || component == 3) {
172cabdff1aSopenharmony_ci        code = get_vlc2(gb, dc_lum_vlc_le.table, DC_VLC_BITS, 2);
173cabdff1aSopenharmony_ci    } else {
174cabdff1aSopenharmony_ci        code = get_vlc2(gb, dc_chroma_vlc_le.table, DC_VLC_BITS, 2);
175cabdff1aSopenharmony_ci    }
176cabdff1aSopenharmony_ci    if (!code) {
177cabdff1aSopenharmony_ci        diff = 0;
178cabdff1aSopenharmony_ci    } else {
179cabdff1aSopenharmony_ci        diff = get_xbits_le(gb, code);
180cabdff1aSopenharmony_ci    }
181cabdff1aSopenharmony_ci    return diff;
182cabdff1aSopenharmony_ci}
183cabdff1aSopenharmony_ci
184cabdff1aSopenharmony_cistatic inline int decode_alpha_block(const SHQContext *s, GetBitContext *gb, uint8_t last_alpha[16], uint8_t *dest, int linesize)
185cabdff1aSopenharmony_ci{
186cabdff1aSopenharmony_ci    uint8_t block[128];
187cabdff1aSopenharmony_ci    int i = 0, x, y;
188cabdff1aSopenharmony_ci
189cabdff1aSopenharmony_ci    memset(block, 0, sizeof(block));
190cabdff1aSopenharmony_ci
191cabdff1aSopenharmony_ci    {
192cabdff1aSopenharmony_ci        OPEN_READER(re, gb);
193cabdff1aSopenharmony_ci
194cabdff1aSopenharmony_ci        for ( ;; ) {
195cabdff1aSopenharmony_ci            int run, level;
196cabdff1aSopenharmony_ci
197cabdff1aSopenharmony_ci            UPDATE_CACHE_LE(re, gb);
198cabdff1aSopenharmony_ci            GET_VLC(run, re, gb, dc_alpha_run_vlc_le.table, ALPHA_VLC_BITS, 2);
199cabdff1aSopenharmony_ci
200cabdff1aSopenharmony_ci            if (run < 0) break;
201cabdff1aSopenharmony_ci            i += run;
202cabdff1aSopenharmony_ci            if (i >= 128)
203cabdff1aSopenharmony_ci                return AVERROR_INVALIDDATA;
204cabdff1aSopenharmony_ci
205cabdff1aSopenharmony_ci            UPDATE_CACHE_LE(re, gb);
206cabdff1aSopenharmony_ci            GET_VLC(level, re, gb, dc_alpha_level_vlc_le.table, ALPHA_VLC_BITS, 2);
207cabdff1aSopenharmony_ci            block[i++] = level;
208cabdff1aSopenharmony_ci        }
209cabdff1aSopenharmony_ci
210cabdff1aSopenharmony_ci        CLOSE_READER(re, gb);
211cabdff1aSopenharmony_ci    }
212cabdff1aSopenharmony_ci
213cabdff1aSopenharmony_ci    for (y = 0; y < 8; y++) {
214cabdff1aSopenharmony_ci        for (x = 0; x < 16; x++) {
215cabdff1aSopenharmony_ci            last_alpha[x] -= block[y * 16 + x];
216cabdff1aSopenharmony_ci        }
217cabdff1aSopenharmony_ci        memcpy(dest, last_alpha, 16);
218cabdff1aSopenharmony_ci        dest += linesize;
219cabdff1aSopenharmony_ci    }
220cabdff1aSopenharmony_ci
221cabdff1aSopenharmony_ci    return 0;
222cabdff1aSopenharmony_ci}
223cabdff1aSopenharmony_ci
224cabdff1aSopenharmony_cistatic inline int decode_dct_block(const SHQContext *s, GetBitContext *gb, int last_dc[4], int component, uint8_t *dest, int linesize)
225cabdff1aSopenharmony_ci{
226cabdff1aSopenharmony_ci    const int *quant_matrix = s->quant_matrix;
227cabdff1aSopenharmony_ci    const uint8_t *scantable = s->intra_scantable.permutated;
228cabdff1aSopenharmony_ci    LOCAL_ALIGNED_32(int16_t, block, [64]);
229cabdff1aSopenharmony_ci    int dc_offset;
230cabdff1aSopenharmony_ci
231cabdff1aSopenharmony_ci    s->bdsp.clear_block(block);
232cabdff1aSopenharmony_ci
233cabdff1aSopenharmony_ci    dc_offset = decode_dc_le(gb, component);
234cabdff1aSopenharmony_ci    last_dc[component] -= dc_offset;  /* Note: Opposite of most codecs. */
235cabdff1aSopenharmony_ci    block[scantable[0]] = last_dc[component];  /* quant_matrix[0] is always 16. */
236cabdff1aSopenharmony_ci
237cabdff1aSopenharmony_ci    /* Read AC coefficients. */
238cabdff1aSopenharmony_ci    {
239cabdff1aSopenharmony_ci        int i = 0;
240cabdff1aSopenharmony_ci        OPEN_READER(re, gb);
241cabdff1aSopenharmony_ci        for ( ;; ) {
242cabdff1aSopenharmony_ci            int level, run;
243cabdff1aSopenharmony_ci            UPDATE_CACHE_LE(re, gb);
244cabdff1aSopenharmony_ci            GET_RL_VLC(level, run, re, gb, ff_rl_speedhq.rl_vlc[0],
245cabdff1aSopenharmony_ci                       TEX_VLC_BITS, 2, 0);
246cabdff1aSopenharmony_ci            if (level == 127) {
247cabdff1aSopenharmony_ci                break;
248cabdff1aSopenharmony_ci            } else if (level) {
249cabdff1aSopenharmony_ci                i += run;
250cabdff1aSopenharmony_ci                if (i > MAX_INDEX)
251cabdff1aSopenharmony_ci                    return AVERROR_INVALIDDATA;
252cabdff1aSopenharmony_ci                /* If next bit is 1, level = -level */
253cabdff1aSopenharmony_ci                level = (level ^ SHOW_SBITS(re, gb, 1)) -
254cabdff1aSopenharmony_ci                        SHOW_SBITS(re, gb, 1);
255cabdff1aSopenharmony_ci                LAST_SKIP_BITS(re, gb, 1);
256cabdff1aSopenharmony_ci            } else {
257cabdff1aSopenharmony_ci                /* Escape. */
258cabdff1aSopenharmony_ci#if MIN_CACHE_BITS < 6 + 6 + 12
259cabdff1aSopenharmony_ci#error MIN_CACHE_BITS is too small for the escape code, add UPDATE_CACHE
260cabdff1aSopenharmony_ci#endif
261cabdff1aSopenharmony_ci                run = SHOW_UBITS(re, gb, 6) + 1;
262cabdff1aSopenharmony_ci                SKIP_BITS(re, gb, 6);
263cabdff1aSopenharmony_ci                level = SHOW_UBITS(re, gb, 12) - 2048;
264cabdff1aSopenharmony_ci                LAST_SKIP_BITS(re, gb, 12);
265cabdff1aSopenharmony_ci
266cabdff1aSopenharmony_ci                i += run;
267cabdff1aSopenharmony_ci                if (i > MAX_INDEX)
268cabdff1aSopenharmony_ci                    return AVERROR_INVALIDDATA;
269cabdff1aSopenharmony_ci            }
270cabdff1aSopenharmony_ci
271cabdff1aSopenharmony_ci            block[scantable[i]] = (level * quant_matrix[i]) >> 4;
272cabdff1aSopenharmony_ci        }
273cabdff1aSopenharmony_ci        CLOSE_READER(re, gb);
274cabdff1aSopenharmony_ci    }
275cabdff1aSopenharmony_ci
276cabdff1aSopenharmony_ci    s->idsp.idct_put(dest, linesize, block);
277cabdff1aSopenharmony_ci
278cabdff1aSopenharmony_ci    return 0;
279cabdff1aSopenharmony_ci}
280cabdff1aSopenharmony_ci
281cabdff1aSopenharmony_cistatic int decode_speedhq_border(const SHQContext *s, GetBitContext *gb, AVFrame *frame, int field_number, int line_stride)
282cabdff1aSopenharmony_ci{
283cabdff1aSopenharmony_ci    int linesize_y  = frame->linesize[0] * line_stride;
284cabdff1aSopenharmony_ci    int linesize_cb = frame->linesize[1] * line_stride;
285cabdff1aSopenharmony_ci    int linesize_cr = frame->linesize[2] * line_stride;
286cabdff1aSopenharmony_ci    int linesize_a;
287cabdff1aSopenharmony_ci    int ret;
288cabdff1aSopenharmony_ci
289cabdff1aSopenharmony_ci    if (s->alpha_type != SHQ_NO_ALPHA)
290cabdff1aSopenharmony_ci        linesize_a = frame->linesize[3] * line_stride;
291cabdff1aSopenharmony_ci
292cabdff1aSopenharmony_ci    for (int y = 0; y < frame->height; y += 16 * line_stride) {
293cabdff1aSopenharmony_ci        int last_dc[4] = { 1024, 1024, 1024, 1024 };
294cabdff1aSopenharmony_ci        uint8_t *dest_y, *dest_cb, *dest_cr, *dest_a;
295cabdff1aSopenharmony_ci        uint8_t last_alpha[16];
296cabdff1aSopenharmony_ci        int x = frame->width - 8;
297cabdff1aSopenharmony_ci
298cabdff1aSopenharmony_ci        dest_y = frame->data[0] + frame->linesize[0] * (y + field_number) + x;
299cabdff1aSopenharmony_ci        if (s->subsampling == SHQ_SUBSAMPLING_420) {
300cabdff1aSopenharmony_ci            dest_cb = frame->data[1] + frame->linesize[1] * (y/2 + field_number) + x / 2;
301cabdff1aSopenharmony_ci            dest_cr = frame->data[2] + frame->linesize[2] * (y/2 + field_number) + x / 2;
302cabdff1aSopenharmony_ci        } else {
303cabdff1aSopenharmony_ci            av_assert2(s->subsampling == SHQ_SUBSAMPLING_422);
304cabdff1aSopenharmony_ci            dest_cb = frame->data[1] + frame->linesize[1] * (y + field_number) + x / 2;
305cabdff1aSopenharmony_ci            dest_cr = frame->data[2] + frame->linesize[2] * (y + field_number) + x / 2;
306cabdff1aSopenharmony_ci        }
307cabdff1aSopenharmony_ci        if (s->alpha_type != SHQ_NO_ALPHA) {
308cabdff1aSopenharmony_ci            memset(last_alpha, 255, sizeof(last_alpha));
309cabdff1aSopenharmony_ci            dest_a = frame->data[3] + frame->linesize[3] * (y + field_number) + x;
310cabdff1aSopenharmony_ci        }
311cabdff1aSopenharmony_ci
312cabdff1aSopenharmony_ci        if ((ret = decode_dct_block(s, gb, last_dc, 0, dest_y, linesize_y)) < 0)
313cabdff1aSopenharmony_ci            return ret;
314cabdff1aSopenharmony_ci        if ((ret = decode_dct_block(s, gb, last_dc, 0, dest_y + 8, linesize_y)) < 0)
315cabdff1aSopenharmony_ci            return ret;
316cabdff1aSopenharmony_ci        if ((ret = decode_dct_block(s, gb, last_dc, 0, dest_y + 8 * linesize_y, linesize_y)) < 0)
317cabdff1aSopenharmony_ci            return ret;
318cabdff1aSopenharmony_ci        if ((ret = decode_dct_block(s, gb, last_dc, 0, dest_y + 8 * linesize_y + 8, linesize_y)) < 0)
319cabdff1aSopenharmony_ci            return ret;
320cabdff1aSopenharmony_ci        if ((ret = decode_dct_block(s, gb, last_dc, 1, dest_cb, linesize_cb)) < 0)
321cabdff1aSopenharmony_ci            return ret;
322cabdff1aSopenharmony_ci        if ((ret = decode_dct_block(s, gb, last_dc, 2, dest_cr, linesize_cr)) < 0)
323cabdff1aSopenharmony_ci            return ret;
324cabdff1aSopenharmony_ci
325cabdff1aSopenharmony_ci        if (s->subsampling != SHQ_SUBSAMPLING_420) {
326cabdff1aSopenharmony_ci            if ((ret = decode_dct_block(s, gb, last_dc, 1, dest_cb + 8 * linesize_cb, linesize_cb)) < 0)
327cabdff1aSopenharmony_ci                return ret;
328cabdff1aSopenharmony_ci            if ((ret = decode_dct_block(s, gb, last_dc, 2, dest_cr + 8 * linesize_cr, linesize_cr)) < 0)
329cabdff1aSopenharmony_ci                return ret;
330cabdff1aSopenharmony_ci        }
331cabdff1aSopenharmony_ci
332cabdff1aSopenharmony_ci        if (s->alpha_type == SHQ_RLE_ALPHA) {
333cabdff1aSopenharmony_ci            /* Alpha coded using 16x8 RLE blocks. */
334cabdff1aSopenharmony_ci            if ((ret = decode_alpha_block(s, gb, last_alpha, dest_a, linesize_a)) < 0)
335cabdff1aSopenharmony_ci                return ret;
336cabdff1aSopenharmony_ci            if ((ret = decode_alpha_block(s, gb, last_alpha, dest_a + 8 * linesize_a, linesize_a)) < 0)
337cabdff1aSopenharmony_ci                return ret;
338cabdff1aSopenharmony_ci        } else if (s->alpha_type == SHQ_DCT_ALPHA) {
339cabdff1aSopenharmony_ci            /* Alpha encoded exactly like luma. */
340cabdff1aSopenharmony_ci            if ((ret = decode_dct_block(s, gb, last_dc, 3, dest_a, linesize_a)) < 0)
341cabdff1aSopenharmony_ci                return ret;
342cabdff1aSopenharmony_ci            if ((ret = decode_dct_block(s, gb, last_dc, 3, dest_a + 8, linesize_a)) < 0)
343cabdff1aSopenharmony_ci                return ret;
344cabdff1aSopenharmony_ci            if ((ret = decode_dct_block(s, gb, last_dc, 3, dest_a + 8 * linesize_a, linesize_a)) < 0)
345cabdff1aSopenharmony_ci                return ret;
346cabdff1aSopenharmony_ci            if ((ret = decode_dct_block(s, gb, last_dc, 3, dest_a + 8 * linesize_a + 8, linesize_a)) < 0)
347cabdff1aSopenharmony_ci                return ret;
348cabdff1aSopenharmony_ci        }
349cabdff1aSopenharmony_ci    }
350cabdff1aSopenharmony_ci
351cabdff1aSopenharmony_ci    return 0;
352cabdff1aSopenharmony_ci}
353cabdff1aSopenharmony_ci
354cabdff1aSopenharmony_cistatic int decode_speedhq_field(const SHQContext *s, const uint8_t *buf, int buf_size, AVFrame *frame, int field_number, int start, int end, int line_stride)
355cabdff1aSopenharmony_ci{
356cabdff1aSopenharmony_ci    int ret, slice_number, slice_offsets[5];
357cabdff1aSopenharmony_ci    int linesize_y  = frame->linesize[0] * line_stride;
358cabdff1aSopenharmony_ci    int linesize_cb = frame->linesize[1] * line_stride;
359cabdff1aSopenharmony_ci    int linesize_cr = frame->linesize[2] * line_stride;
360cabdff1aSopenharmony_ci    int linesize_a;
361cabdff1aSopenharmony_ci    GetBitContext gb;
362cabdff1aSopenharmony_ci
363cabdff1aSopenharmony_ci    if (s->alpha_type != SHQ_NO_ALPHA)
364cabdff1aSopenharmony_ci        linesize_a = frame->linesize[3] * line_stride;
365cabdff1aSopenharmony_ci
366cabdff1aSopenharmony_ci    if (end < start || end - start < 3 || end > buf_size)
367cabdff1aSopenharmony_ci        return AVERROR_INVALIDDATA;
368cabdff1aSopenharmony_ci
369cabdff1aSopenharmony_ci    slice_offsets[0] = start;
370cabdff1aSopenharmony_ci    slice_offsets[4] = end;
371cabdff1aSopenharmony_ci    for (slice_number = 1; slice_number < 4; slice_number++) {
372cabdff1aSopenharmony_ci        uint32_t last_offset, slice_len;
373cabdff1aSopenharmony_ci
374cabdff1aSopenharmony_ci        last_offset = slice_offsets[slice_number - 1];
375cabdff1aSopenharmony_ci        slice_len = AV_RL24(buf + last_offset);
376cabdff1aSopenharmony_ci        slice_offsets[slice_number] = last_offset + slice_len;
377cabdff1aSopenharmony_ci
378cabdff1aSopenharmony_ci        if (slice_len < 3 || slice_offsets[slice_number] > end - 3)
379cabdff1aSopenharmony_ci            return AVERROR_INVALIDDATA;
380cabdff1aSopenharmony_ci    }
381cabdff1aSopenharmony_ci
382cabdff1aSopenharmony_ci    for (slice_number = 0; slice_number < 4; slice_number++) {
383cabdff1aSopenharmony_ci        uint32_t slice_begin, slice_end;
384cabdff1aSopenharmony_ci        int x, y;
385cabdff1aSopenharmony_ci
386cabdff1aSopenharmony_ci        slice_begin = slice_offsets[slice_number];
387cabdff1aSopenharmony_ci        slice_end = slice_offsets[slice_number + 1];
388cabdff1aSopenharmony_ci
389cabdff1aSopenharmony_ci        if ((ret = init_get_bits8(&gb, buf + slice_begin + 3, slice_end - slice_begin - 3)) < 0)
390cabdff1aSopenharmony_ci            return ret;
391cabdff1aSopenharmony_ci
392cabdff1aSopenharmony_ci        for (y = slice_number * 16 * line_stride; y < frame->height; y += line_stride * 64) {
393cabdff1aSopenharmony_ci            uint8_t *dest_y, *dest_cb, *dest_cr, *dest_a;
394cabdff1aSopenharmony_ci            int last_dc[4] = { 1024, 1024, 1024, 1024 };
395cabdff1aSopenharmony_ci            uint8_t last_alpha[16];
396cabdff1aSopenharmony_ci
397cabdff1aSopenharmony_ci            memset(last_alpha, 255, sizeof(last_alpha));
398cabdff1aSopenharmony_ci
399cabdff1aSopenharmony_ci            dest_y = frame->data[0] + frame->linesize[0] * (y + field_number);
400cabdff1aSopenharmony_ci            if (s->subsampling == SHQ_SUBSAMPLING_420) {
401cabdff1aSopenharmony_ci                dest_cb = frame->data[1] + frame->linesize[1] * (y/2 + field_number);
402cabdff1aSopenharmony_ci                dest_cr = frame->data[2] + frame->linesize[2] * (y/2 + field_number);
403cabdff1aSopenharmony_ci            } else {
404cabdff1aSopenharmony_ci                dest_cb = frame->data[1] + frame->linesize[1] * (y + field_number);
405cabdff1aSopenharmony_ci                dest_cr = frame->data[2] + frame->linesize[2] * (y + field_number);
406cabdff1aSopenharmony_ci            }
407cabdff1aSopenharmony_ci            if (s->alpha_type != SHQ_NO_ALPHA) {
408cabdff1aSopenharmony_ci                dest_a = frame->data[3] + frame->linesize[3] * (y + field_number);
409cabdff1aSopenharmony_ci            }
410cabdff1aSopenharmony_ci
411cabdff1aSopenharmony_ci            for (x = 0; x < frame->width - 8 * (s->subsampling != SHQ_SUBSAMPLING_444); x += 16) {
412cabdff1aSopenharmony_ci                /* Decode the four luma blocks. */
413cabdff1aSopenharmony_ci                if ((ret = decode_dct_block(s, &gb, last_dc, 0, dest_y, linesize_y)) < 0)
414cabdff1aSopenharmony_ci                    return ret;
415cabdff1aSopenharmony_ci                if ((ret = decode_dct_block(s, &gb, last_dc, 0, dest_y + 8, linesize_y)) < 0)
416cabdff1aSopenharmony_ci                    return ret;
417cabdff1aSopenharmony_ci                if ((ret = decode_dct_block(s, &gb, last_dc, 0, dest_y + 8 * linesize_y, linesize_y)) < 0)
418cabdff1aSopenharmony_ci                    return ret;
419cabdff1aSopenharmony_ci                if ((ret = decode_dct_block(s, &gb, last_dc, 0, dest_y + 8 * linesize_y + 8, linesize_y)) < 0)
420cabdff1aSopenharmony_ci                    return ret;
421cabdff1aSopenharmony_ci
422cabdff1aSopenharmony_ci                /*
423cabdff1aSopenharmony_ci                 * Decode the first chroma block. For 4:2:0, this is the only one;
424cabdff1aSopenharmony_ci                 * for 4:2:2, it's the top block; for 4:4:4, it's the top-left block.
425cabdff1aSopenharmony_ci                 */
426cabdff1aSopenharmony_ci                if ((ret = decode_dct_block(s, &gb, last_dc, 1, dest_cb, linesize_cb)) < 0)
427cabdff1aSopenharmony_ci                    return ret;
428cabdff1aSopenharmony_ci                if ((ret = decode_dct_block(s, &gb, last_dc, 2, dest_cr, linesize_cr)) < 0)
429cabdff1aSopenharmony_ci                    return ret;
430cabdff1aSopenharmony_ci
431cabdff1aSopenharmony_ci                if (s->subsampling != SHQ_SUBSAMPLING_420) {
432cabdff1aSopenharmony_ci                    /* For 4:2:2, this is the bottom block; for 4:4:4, it's the bottom-left block. */
433cabdff1aSopenharmony_ci                    if ((ret = decode_dct_block(s, &gb, last_dc, 1, dest_cb + 8 * linesize_cb, linesize_cb)) < 0)
434cabdff1aSopenharmony_ci                        return ret;
435cabdff1aSopenharmony_ci                    if ((ret = decode_dct_block(s, &gb, last_dc, 2, dest_cr + 8 * linesize_cr, linesize_cr)) < 0)
436cabdff1aSopenharmony_ci                        return ret;
437cabdff1aSopenharmony_ci
438cabdff1aSopenharmony_ci                    if (s->subsampling == SHQ_SUBSAMPLING_444) {
439cabdff1aSopenharmony_ci                        /* Top-right and bottom-right blocks. */
440cabdff1aSopenharmony_ci                        if ((ret = decode_dct_block(s, &gb, last_dc, 1, dest_cb + 8, linesize_cb)) < 0)
441cabdff1aSopenharmony_ci                            return ret;
442cabdff1aSopenharmony_ci                        if ((ret = decode_dct_block(s, &gb, last_dc, 2, dest_cr + 8, linesize_cr)) < 0)
443cabdff1aSopenharmony_ci                            return ret;
444cabdff1aSopenharmony_ci                        if ((ret = decode_dct_block(s, &gb, last_dc, 1, dest_cb + 8 * linesize_cb + 8, linesize_cb)) < 0)
445cabdff1aSopenharmony_ci                            return ret;
446cabdff1aSopenharmony_ci                        if ((ret = decode_dct_block(s, &gb, last_dc, 2, dest_cr + 8 * linesize_cr + 8, linesize_cr)) < 0)
447cabdff1aSopenharmony_ci                            return ret;
448cabdff1aSopenharmony_ci
449cabdff1aSopenharmony_ci                        dest_cb += 8;
450cabdff1aSopenharmony_ci                        dest_cr += 8;
451cabdff1aSopenharmony_ci                    }
452cabdff1aSopenharmony_ci                }
453cabdff1aSopenharmony_ci                dest_y += 16;
454cabdff1aSopenharmony_ci                dest_cb += 8;
455cabdff1aSopenharmony_ci                dest_cr += 8;
456cabdff1aSopenharmony_ci
457cabdff1aSopenharmony_ci                if (s->alpha_type == SHQ_RLE_ALPHA) {
458cabdff1aSopenharmony_ci                    /* Alpha coded using 16x8 RLE blocks. */
459cabdff1aSopenharmony_ci                    if ((ret = decode_alpha_block(s, &gb, last_alpha, dest_a, linesize_a)) < 0)
460cabdff1aSopenharmony_ci                        return ret;
461cabdff1aSopenharmony_ci                    if ((ret = decode_alpha_block(s, &gb, last_alpha, dest_a + 8 * linesize_a, linesize_a)) < 0)
462cabdff1aSopenharmony_ci                        return ret;
463cabdff1aSopenharmony_ci                    dest_a += 16;
464cabdff1aSopenharmony_ci                } else if (s->alpha_type == SHQ_DCT_ALPHA) {
465cabdff1aSopenharmony_ci                    /* Alpha encoded exactly like luma. */
466cabdff1aSopenharmony_ci                    if ((ret = decode_dct_block(s, &gb, last_dc, 3, dest_a, linesize_a)) < 0)
467cabdff1aSopenharmony_ci                        return ret;
468cabdff1aSopenharmony_ci                    if ((ret = decode_dct_block(s, &gb, last_dc, 3, dest_a + 8, linesize_a)) < 0)
469cabdff1aSopenharmony_ci                        return ret;
470cabdff1aSopenharmony_ci                    if ((ret = decode_dct_block(s, &gb, last_dc, 3, dest_a + 8 * linesize_a, linesize_a)) < 0)
471cabdff1aSopenharmony_ci                        return ret;
472cabdff1aSopenharmony_ci                    if ((ret = decode_dct_block(s, &gb, last_dc, 3, dest_a + 8 * linesize_a + 8, linesize_a)) < 0)
473cabdff1aSopenharmony_ci                        return ret;
474cabdff1aSopenharmony_ci                    dest_a += 16;
475cabdff1aSopenharmony_ci                }
476cabdff1aSopenharmony_ci            }
477cabdff1aSopenharmony_ci        }
478cabdff1aSopenharmony_ci    }
479cabdff1aSopenharmony_ci
480cabdff1aSopenharmony_ci    if (s->subsampling != SHQ_SUBSAMPLING_444 && (frame->width & 15))
481cabdff1aSopenharmony_ci        return decode_speedhq_border(s, &gb, frame, field_number, line_stride);
482cabdff1aSopenharmony_ci
483cabdff1aSopenharmony_ci    return 0;
484cabdff1aSopenharmony_ci}
485cabdff1aSopenharmony_ci
486cabdff1aSopenharmony_cistatic void compute_quant_matrix(int *output, int qscale)
487cabdff1aSopenharmony_ci{
488cabdff1aSopenharmony_ci    int i;
489cabdff1aSopenharmony_ci    for (i = 0; i < 64; i++) output[i] = unscaled_quant_matrix[ff_zigzag_direct[i]] * qscale;
490cabdff1aSopenharmony_ci}
491cabdff1aSopenharmony_ci
492cabdff1aSopenharmony_cistatic int speedhq_decode_frame(AVCodecContext *avctx, AVFrame *frame,
493cabdff1aSopenharmony_ci                                int *got_frame, AVPacket *avpkt)
494cabdff1aSopenharmony_ci{
495cabdff1aSopenharmony_ci    SHQContext * const s = avctx->priv_data;
496cabdff1aSopenharmony_ci    const uint8_t *buf   = avpkt->data;
497cabdff1aSopenharmony_ci    int buf_size         = avpkt->size;
498cabdff1aSopenharmony_ci    uint8_t quality;
499cabdff1aSopenharmony_ci    uint32_t second_field_offset;
500cabdff1aSopenharmony_ci    int ret;
501cabdff1aSopenharmony_ci
502cabdff1aSopenharmony_ci    if (buf_size < 4 || avctx->width < 8 || avctx->width % 8 != 0)
503cabdff1aSopenharmony_ci        return AVERROR_INVALIDDATA;
504cabdff1aSopenharmony_ci    if (buf_size < avctx->width*avctx->height / 64 / 4)
505cabdff1aSopenharmony_ci        return AVERROR_INVALIDDATA;
506cabdff1aSopenharmony_ci
507cabdff1aSopenharmony_ci    quality = buf[0];
508cabdff1aSopenharmony_ci    if (quality >= 100) {
509cabdff1aSopenharmony_ci        return AVERROR_INVALIDDATA;
510cabdff1aSopenharmony_ci    }
511cabdff1aSopenharmony_ci
512cabdff1aSopenharmony_ci    compute_quant_matrix(s->quant_matrix, 100 - quality);
513cabdff1aSopenharmony_ci
514cabdff1aSopenharmony_ci    second_field_offset = AV_RL24(buf + 1);
515cabdff1aSopenharmony_ci    if (second_field_offset >= buf_size - 3) {
516cabdff1aSopenharmony_ci        return AVERROR_INVALIDDATA;
517cabdff1aSopenharmony_ci    }
518cabdff1aSopenharmony_ci
519cabdff1aSopenharmony_ci    avctx->coded_width = FFALIGN(avctx->width, 16);
520cabdff1aSopenharmony_ci    avctx->coded_height = FFALIGN(avctx->height, 16);
521cabdff1aSopenharmony_ci
522cabdff1aSopenharmony_ci    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0) {
523cabdff1aSopenharmony_ci        return ret;
524cabdff1aSopenharmony_ci    }
525cabdff1aSopenharmony_ci    frame->key_frame = 1;
526cabdff1aSopenharmony_ci
527cabdff1aSopenharmony_ci    if (second_field_offset == 4 || second_field_offset == (buf_size-4)) {
528cabdff1aSopenharmony_ci        /*
529cabdff1aSopenharmony_ci         * Overlapping first and second fields is used to signal
530cabdff1aSopenharmony_ci         * encoding only a single field. In this case, "height"
531cabdff1aSopenharmony_ci         * is ambiguous; it could mean either the height of the
532cabdff1aSopenharmony_ci         * frame as a whole, or of the field. The former would make
533cabdff1aSopenharmony_ci         * more sense for compatibility with legacy decoders,
534cabdff1aSopenharmony_ci         * but this matches the convention used in NDI, which is
535cabdff1aSopenharmony_ci         * the primary user of this trick.
536cabdff1aSopenharmony_ci         */
537cabdff1aSopenharmony_ci        if ((ret = decode_speedhq_field(s, buf, buf_size, frame, 0, 4, buf_size, 1)) < 0)
538cabdff1aSopenharmony_ci            return ret;
539cabdff1aSopenharmony_ci    } else {
540cabdff1aSopenharmony_ci        if ((ret = decode_speedhq_field(s, buf, buf_size, frame, 0, 4, second_field_offset, 2)) < 0)
541cabdff1aSopenharmony_ci            return ret;
542cabdff1aSopenharmony_ci        if ((ret = decode_speedhq_field(s, buf, buf_size, frame, 1, second_field_offset, buf_size, 2)) < 0)
543cabdff1aSopenharmony_ci            return ret;
544cabdff1aSopenharmony_ci    }
545cabdff1aSopenharmony_ci
546cabdff1aSopenharmony_ci    *got_frame = 1;
547cabdff1aSopenharmony_ci    return buf_size;
548cabdff1aSopenharmony_ci}
549cabdff1aSopenharmony_ci
550cabdff1aSopenharmony_ci/*
551cabdff1aSopenharmony_ci * Alpha VLC. Run and level are independently coded, and would be
552cabdff1aSopenharmony_ci * outside the default limits for MAX_RUN/MAX_LEVEL, so we don't
553cabdff1aSopenharmony_ci * bother with combining them into one table.
554cabdff1aSopenharmony_ci */
555cabdff1aSopenharmony_cistatic av_cold void compute_alpha_vlcs(void)
556cabdff1aSopenharmony_ci{
557cabdff1aSopenharmony_ci    uint16_t run_code[134], level_code[266];
558cabdff1aSopenharmony_ci    uint8_t run_bits[134], level_bits[266];
559cabdff1aSopenharmony_ci    int16_t run_symbols[134], level_symbols[266];
560cabdff1aSopenharmony_ci    int entry, i, sign;
561cabdff1aSopenharmony_ci
562cabdff1aSopenharmony_ci    /* Initialize VLC for alpha run. */
563cabdff1aSopenharmony_ci    entry = 0;
564cabdff1aSopenharmony_ci
565cabdff1aSopenharmony_ci    /* 0 -> 0. */
566cabdff1aSopenharmony_ci    run_code[entry] = 0;
567cabdff1aSopenharmony_ci    run_bits[entry] = 1;
568cabdff1aSopenharmony_ci    run_symbols[entry] = 0;
569cabdff1aSopenharmony_ci    ++entry;
570cabdff1aSopenharmony_ci
571cabdff1aSopenharmony_ci    /* 10xx -> xx plus 1. */
572cabdff1aSopenharmony_ci    for (i = 0; i < 4; ++i) {
573cabdff1aSopenharmony_ci        run_code[entry] = (i << 2) | 1;
574cabdff1aSopenharmony_ci        run_bits[entry] = 4;
575cabdff1aSopenharmony_ci        run_symbols[entry] = i + 1;
576cabdff1aSopenharmony_ci        ++entry;
577cabdff1aSopenharmony_ci    }
578cabdff1aSopenharmony_ci
579cabdff1aSopenharmony_ci    /* 111xxxxxxx -> xxxxxxx. */
580cabdff1aSopenharmony_ci    for (i = 0; i < 128; ++i) {
581cabdff1aSopenharmony_ci        run_code[entry] = (i << 3) | 7;
582cabdff1aSopenharmony_ci        run_bits[entry] = 10;
583cabdff1aSopenharmony_ci        run_symbols[entry] = i;
584cabdff1aSopenharmony_ci        ++entry;
585cabdff1aSopenharmony_ci    }
586cabdff1aSopenharmony_ci
587cabdff1aSopenharmony_ci    /* 110 -> EOB. */
588cabdff1aSopenharmony_ci    run_code[entry] = 3;
589cabdff1aSopenharmony_ci    run_bits[entry] = 3;
590cabdff1aSopenharmony_ci    run_symbols[entry] = -1;
591cabdff1aSopenharmony_ci    ++entry;
592cabdff1aSopenharmony_ci
593cabdff1aSopenharmony_ci    av_assert0(entry == FF_ARRAY_ELEMS(run_code));
594cabdff1aSopenharmony_ci
595cabdff1aSopenharmony_ci    INIT_LE_VLC_SPARSE_STATIC(&dc_alpha_run_vlc_le, ALPHA_VLC_BITS,
596cabdff1aSopenharmony_ci                              FF_ARRAY_ELEMS(run_code),
597cabdff1aSopenharmony_ci                              run_bits, 1, 1,
598cabdff1aSopenharmony_ci                              run_code, 2, 2,
599cabdff1aSopenharmony_ci                              run_symbols, 2, 2, 160);
600cabdff1aSopenharmony_ci
601cabdff1aSopenharmony_ci    /* Initialize VLC for alpha level. */
602cabdff1aSopenharmony_ci    entry = 0;
603cabdff1aSopenharmony_ci
604cabdff1aSopenharmony_ci    for (sign = 0; sign <= 1; ++sign) {
605cabdff1aSopenharmony_ci        /* 1s -> -1 or +1 (depending on sign bit). */
606cabdff1aSopenharmony_ci        level_code[entry] = (sign << 1) | 1;
607cabdff1aSopenharmony_ci        level_bits[entry] = 2;
608cabdff1aSopenharmony_ci        level_symbols[entry] = sign ? -1 : 1;
609cabdff1aSopenharmony_ci        ++entry;
610cabdff1aSopenharmony_ci
611cabdff1aSopenharmony_ci        /* 01sxx -> xx plus 2 (2..5 or -2..-5, depending on sign bit). */
612cabdff1aSopenharmony_ci        for (i = 0; i < 4; ++i) {
613cabdff1aSopenharmony_ci            level_code[entry] = (i << 3) | (sign << 2) | 2;
614cabdff1aSopenharmony_ci            level_bits[entry] = 5;
615cabdff1aSopenharmony_ci            level_symbols[entry] = sign ? -(i + 2) : (i + 2);
616cabdff1aSopenharmony_ci            ++entry;
617cabdff1aSopenharmony_ci        }
618cabdff1aSopenharmony_ci    }
619cabdff1aSopenharmony_ci
620cabdff1aSopenharmony_ci    /*
621cabdff1aSopenharmony_ci     * 00xxxxxxxx -> xxxxxxxx, in two's complement. There are many codes
622cabdff1aSopenharmony_ci     * here that would better be encoded in other ways (e.g. 0 would be
623cabdff1aSopenharmony_ci     * encoded by increasing run, and +/- 1 would be encoded with a
624cabdff1aSopenharmony_ci     * shorter code), but it doesn't hurt to allow everything.
625cabdff1aSopenharmony_ci     */
626cabdff1aSopenharmony_ci    for (i = 0; i < 256; ++i) {
627cabdff1aSopenharmony_ci        level_code[entry] = i << 2;
628cabdff1aSopenharmony_ci        level_bits[entry] = 10;
629cabdff1aSopenharmony_ci        level_symbols[entry] = i;
630cabdff1aSopenharmony_ci        ++entry;
631cabdff1aSopenharmony_ci    }
632cabdff1aSopenharmony_ci
633cabdff1aSopenharmony_ci    av_assert0(entry == FF_ARRAY_ELEMS(level_code));
634cabdff1aSopenharmony_ci
635cabdff1aSopenharmony_ci    INIT_LE_VLC_SPARSE_STATIC(&dc_alpha_level_vlc_le, ALPHA_VLC_BITS,
636cabdff1aSopenharmony_ci                              FF_ARRAY_ELEMS(level_code),
637cabdff1aSopenharmony_ci                              level_bits, 1, 1,
638cabdff1aSopenharmony_ci                              level_code, 2, 2,
639cabdff1aSopenharmony_ci                              level_symbols, 2, 2, 288);
640cabdff1aSopenharmony_ci}
641cabdff1aSopenharmony_ci
642cabdff1aSopenharmony_cistatic av_cold void speedhq_static_init(void)
643cabdff1aSopenharmony_ci{
644cabdff1aSopenharmony_ci    /* Exactly the same as MPEG-2, except for a little-endian reader. */
645cabdff1aSopenharmony_ci    INIT_CUSTOM_VLC_STATIC(&dc_lum_vlc_le, DC_VLC_BITS, 12,
646cabdff1aSopenharmony_ci                           ff_mpeg12_vlc_dc_lum_bits, 1, 1,
647cabdff1aSopenharmony_ci                           ff_mpeg12_vlc_dc_lum_code, 2, 2,
648cabdff1aSopenharmony_ci                           INIT_VLC_OUTPUT_LE, 512);
649cabdff1aSopenharmony_ci    INIT_CUSTOM_VLC_STATIC(&dc_chroma_vlc_le, DC_VLC_BITS, 12,
650cabdff1aSopenharmony_ci                           ff_mpeg12_vlc_dc_chroma_bits, 1, 1,
651cabdff1aSopenharmony_ci                           ff_mpeg12_vlc_dc_chroma_code, 2, 2,
652cabdff1aSopenharmony_ci                           INIT_VLC_OUTPUT_LE, 514);
653cabdff1aSopenharmony_ci
654cabdff1aSopenharmony_ci    INIT_2D_VLC_RL(ff_rl_speedhq, 674, INIT_VLC_LE);
655cabdff1aSopenharmony_ci
656cabdff1aSopenharmony_ci    compute_alpha_vlcs();
657cabdff1aSopenharmony_ci}
658cabdff1aSopenharmony_ci
659cabdff1aSopenharmony_cistatic av_cold int speedhq_decode_init(AVCodecContext *avctx)
660cabdff1aSopenharmony_ci{
661cabdff1aSopenharmony_ci    int ret;
662cabdff1aSopenharmony_ci    static AVOnce init_once = AV_ONCE_INIT;
663cabdff1aSopenharmony_ci    SHQContext * const s = avctx->priv_data;
664cabdff1aSopenharmony_ci
665cabdff1aSopenharmony_ci    s->avctx = avctx;
666cabdff1aSopenharmony_ci
667cabdff1aSopenharmony_ci    ret = ff_thread_once(&init_once, speedhq_static_init);
668cabdff1aSopenharmony_ci    if (ret)
669cabdff1aSopenharmony_ci        return AVERROR_UNKNOWN;
670cabdff1aSopenharmony_ci
671cabdff1aSopenharmony_ci    ff_blockdsp_init(&s->bdsp, avctx);
672cabdff1aSopenharmony_ci    ff_idctdsp_init(&s->idsp, avctx);
673cabdff1aSopenharmony_ci    ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_zigzag_direct);
674cabdff1aSopenharmony_ci
675cabdff1aSopenharmony_ci    switch (avctx->codec_tag) {
676cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '0'):
677cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_420;
678cabdff1aSopenharmony_ci        s->alpha_type = SHQ_NO_ALPHA;
679cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUV420P;
680cabdff1aSopenharmony_ci        break;
681cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '1'):
682cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_420;
683cabdff1aSopenharmony_ci        s->alpha_type = SHQ_RLE_ALPHA;
684cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUVA420P;
685cabdff1aSopenharmony_ci        break;
686cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '2'):
687cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_422;
688cabdff1aSopenharmony_ci        s->alpha_type = SHQ_NO_ALPHA;
689cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUV422P;
690cabdff1aSopenharmony_ci        break;
691cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '3'):
692cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_422;
693cabdff1aSopenharmony_ci        s->alpha_type = SHQ_RLE_ALPHA;
694cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUVA422P;
695cabdff1aSopenharmony_ci        break;
696cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '4'):
697cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_444;
698cabdff1aSopenharmony_ci        s->alpha_type = SHQ_NO_ALPHA;
699cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUV444P;
700cabdff1aSopenharmony_ci        break;
701cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '5'):
702cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_444;
703cabdff1aSopenharmony_ci        s->alpha_type = SHQ_RLE_ALPHA;
704cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUVA444P;
705cabdff1aSopenharmony_ci        break;
706cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '7'):
707cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_422;
708cabdff1aSopenharmony_ci        s->alpha_type = SHQ_DCT_ALPHA;
709cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUVA422P;
710cabdff1aSopenharmony_ci        break;
711cabdff1aSopenharmony_ci    case MKTAG('S', 'H', 'Q', '9'):
712cabdff1aSopenharmony_ci        s->subsampling = SHQ_SUBSAMPLING_444;
713cabdff1aSopenharmony_ci        s->alpha_type = SHQ_DCT_ALPHA;
714cabdff1aSopenharmony_ci        avctx->pix_fmt = AV_PIX_FMT_YUVA444P;
715cabdff1aSopenharmony_ci        break;
716cabdff1aSopenharmony_ci    default:
717cabdff1aSopenharmony_ci        av_log(avctx, AV_LOG_ERROR, "Unknown NewTek SpeedHQ FOURCC provided (%08X)\n",
718cabdff1aSopenharmony_ci               avctx->codec_tag);
719cabdff1aSopenharmony_ci        return AVERROR_INVALIDDATA;
720cabdff1aSopenharmony_ci    }
721cabdff1aSopenharmony_ci
722cabdff1aSopenharmony_ci    /* This matches what NDI's RGB -> Y'CbCr 4:2:2 converter uses. */
723cabdff1aSopenharmony_ci    avctx->colorspace = AVCOL_SPC_BT470BG;
724cabdff1aSopenharmony_ci    avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
725cabdff1aSopenharmony_ci
726cabdff1aSopenharmony_ci    return 0;
727cabdff1aSopenharmony_ci}
728cabdff1aSopenharmony_ci
729cabdff1aSopenharmony_ciconst FFCodec ff_speedhq_decoder = {
730cabdff1aSopenharmony_ci    .p.name         = "speedhq",
731cabdff1aSopenharmony_ci    .p.long_name    = NULL_IF_CONFIG_SMALL("NewTek SpeedHQ"),
732cabdff1aSopenharmony_ci    .p.type         = AVMEDIA_TYPE_VIDEO,
733cabdff1aSopenharmony_ci    .p.id           = AV_CODEC_ID_SPEEDHQ,
734cabdff1aSopenharmony_ci    .priv_data_size = sizeof(SHQContext),
735cabdff1aSopenharmony_ci    .init           = speedhq_decode_init,
736cabdff1aSopenharmony_ci    FF_CODEC_DECODE_CB(speedhq_decode_frame),
737cabdff1aSopenharmony_ci    .p.capabilities = AV_CODEC_CAP_DR1,
738cabdff1aSopenharmony_ci    .caps_internal  = FF_CODEC_CAP_INIT_THREADSAFE,
739cabdff1aSopenharmony_ci};
740cabdff1aSopenharmony_ci#endif /* CONFIG_SPEEDHQ_DECODER */
741