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