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