/* * Infinity IMM4 decoder * * Copyright (c) 2018 Paul B Mahol * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include "libavutil/thread.h" #include "avcodec.h" #include "bswapdsp.h" #include "copy_block.h" #include "get_bits.h" #include "idctdsp.h" #include "internal.h" typedef struct IMM4Context { BswapDSPContext bdsp; GetBitContext gb; AVFrame *prev_frame; uint8_t *bitstream; int bitstream_size; int factor; unsigned lo; unsigned hi; ScanTable intra_scantable; DECLARE_ALIGNED(32, int16_t, block)[6][64]; IDCTDSPContext idsp; } IMM4Context; static const uint8_t intra_cb[] = { 24, 18, 12 }; static const uint8_t inter_cb[] = { 30, 20, 15 }; static const uint8_t cbplo_symbols[] = { 3, 4, 19, 20, 35, 36, 51, 52 }; static const uint8_t cbplo_bits[] = { 1, 4, 3, 6, 3, 6, 3, 6 }; static const uint8_t cbplo_codes[] = { 1, 1, 1, 1, 2, 2, 3, 3 }; static const uint8_t cbphi_bits[] = { 4, 5, 5, 4, 5, 4, 6, 4, 5, 6, 4, 4, 4, 4, 4, 2 }; static const uint8_t cbphi_codes[] = { 3, 5, 4, 9, 3, 7, 2, 11, 2, 3, 5, 10, 4, 8, 6, 3 }; static const uint8_t blktype_symbols[] = { 0, 1, 2, 3, 4, 16, 17, 18, 19, 20, 32, 33, 34, 35, 48, 50, 51, 52 }; static const uint8_t blktype_bits[] = { 1, 3, 3, 5, 6, 4, 7, 7, 8, 9, 4, 7, 7, 8, 6, 8, 7, 9 }; static const uint8_t blktype_codes[] = { 1, 3, 2, 3, 4, 3, 7, 5, 4, 4, 2, 6, 4, 3, 5, 5, 3, 2 }; static const uint16_t block_symbols[] = { 0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7, 0x8, 0x9, 0xA, 0xB, 0xC, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x101, 0x102, 0x103, 0x104, 0x181, 0x182, 0x183, 0x201, 0x202, 0x203, 0x281, 0x282, 0x283, 0x301, 0x302, 0x303, 0x381, 0x382, 0x401, 0x402, 0x481, 0x482, 0x501, 0x502, 0x581, 0x601, 0x681, 0x701, 0x781, 0x801, 0x881, 0x901, 0x981, 0xA01, 0xA81, 0xB01, 0xB81, 0xC01, 0xC81, 0xD01, 0x4001, 0x4002, 0x4003, 0x4081, 0x4082, 0x4101, 0x4181, 0x4201, 0x4281, 0x4301, 0x4381, 0x4401, 0x4481, 0x4501, 0x4581, 0x4601, 0x4681, 0x4701, 0x4781, 0x4801, 0x4881, 0x4901, 0x4981, 0x4A01, 0x4A81, 0x4B01, 0x4B81, 0x4C01, 0x4C81, 0x4D01, 0x4D81, 0x4E01, 0x4E81, 0x4F01, 0x4F81, 0x5001, 0x5081, 0x5101, 0x5181, 0x5201, 0x5281, 0x5301, 0x5381, 0x5401 }; static const uint8_t block_bits[] = { 7, 2, 4, 6, 7, 8, 9, 9, 10, 10, 11, 11, 11, 3, 6, 8, 10, 11, 12, 4, 8, 10, 12, 5, 9, 10, 5, 9, 12, 5, 10, 12, 6, 10, 12, 6, 10, 6, 10, 6, 10, 7, 12, 7, 7, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9, 11, 11, 12, 12, 4, 9, 11, 6, 11, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9, 10, 10, 10, 10, 11, 11, 11, 11, 12, 12, 12, 12, 12, 12, 12, 12 }; static const uint8_t block_codes[] = { 3, 2, 15, 21, 23, 31, 37, 36, 33, 32, 7, 6, 32, 6, 20, 30, 15, 33, 80, 14, 29, 14, 81, 13, 35, 13, 12, 34, 82, 11, 12, 83, 19, 11, 84, 18, 10, 17, 9, 16, 8, 22, 85, 21, 20, 28, 27, 33, 32, 31, 30, 29, 28, 27, 26, 34, 35, 86, 87, 7, 25, 5, 15, 4, 14, 13, 12, 19, 18, 17, 16, 26, 25, 24, 23, 22, 21, 20, 19, 24, 23, 22, 21, 20, 19, 18, 17, 7, 6, 5, 4, 36, 37, 38, 39, 88, 89, 90, 91, 92, 93, 94, 95 }; static VLC cbplo_tab; static VLC cbphi_tab; static VLC blktype_tab; static VLC block_tab; static int get_cbphi(GetBitContext *gb, int x) { int value; value = get_vlc2(gb, cbphi_tab.table, cbphi_tab.bits, 1); if (value < 0) return AVERROR_INVALIDDATA; return x ? value : 15 - value; } static int decode_block(AVCodecContext *avctx, GetBitContext *gb, int block, int factor, int flag, int offset, int flag2) { IMM4Context *s = avctx->priv_data; const uint8_t *scantable = s->intra_scantable.permutated; int i, last, len, factor2; for (i = !flag; i < 64; i++) { int value; value = get_vlc2(gb, block_tab.table, block_tab.bits, 1); if (value < 0) return AVERROR_INVALIDDATA; if (value == 0) { last = get_bits1(gb); len = get_bits(gb, 6); factor2 = get_sbits(gb, 8); } else { factor2 = value & 0x7F; last = (value >> 14) & 1; len = (value >> 7) & 0x3F; if (get_bits1(gb)) factor2 = -factor2; } i += len; if (i >= 64) break; s->block[block][scantable[i]] = offset * (factor2 < 0 ? -1 : 1) + factor * factor2; if (last) break; } if (s->hi == 2 && flag2 && block < 4) { if (flag) s->block[block][scantable[0]] *= 2; s->block[block][scantable[1]] *= 2; s->block[block][scantable[8]] *= 2; s->block[block][scantable[16]] *= 2; } return 0; } static int decode_blocks(AVCodecContext *avctx, GetBitContext *gb, unsigned cbp, int flag, int offset, unsigned flag2) { IMM4Context *s = avctx->priv_data; const uint8_t *scantable = s->intra_scantable.permutated; int ret, i; memset(s->block, 0, sizeof(s->block)); for (i = 0; i < 6; i++) { if (!flag) { int x = get_bits(gb, 8); if (x == 255) x = 128; x *= 8; s->block[i][scantable[0]] = x; } if (cbp & (1 << (5 - i))) { ret = decode_block(avctx, gb, i, s->factor, flag, offset, flag2); if (ret < 0) return ret; } } return 0; } static int decode_intra(AVCodecContext *avctx, GetBitContext *gb, AVFrame *frame) { IMM4Context *s = avctx->priv_data; int ret, x, y, offset = 0; if (s->hi == 0) { if (s->lo > 2) return AVERROR_INVALIDDATA; s->factor = intra_cb[s->lo]; } else { s->factor = s->lo * 2; } if (s->hi) { offset = s->factor; offset >>= 1; if (!(offset & 1)) offset--; } for (y = 0; y < avctx->height; y += 16) { for (x = 0; x < avctx->width; x += 16) { unsigned flag, cbphi, cbplo; cbplo = get_vlc2(gb, cbplo_tab.table, cbplo_tab.bits, 1) >> 4; flag = get_bits1(gb); cbphi = get_cbphi(gb, 1); ret = decode_blocks(avctx, gb, cbplo | (cbphi << 2), 0, offset, flag); if (ret < 0) return ret; s->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x, frame->linesize[0], s->block[0]); s->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x + 8, frame->linesize[0], s->block[1]); s->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x, frame->linesize[0], s->block[2]); s->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8, frame->linesize[0], s->block[3]); s->idsp.idct_put(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1), frame->linesize[1], s->block[4]); s->idsp.idct_put(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1), frame->linesize[2], s->block[5]); } } return 0; } static int decode_inter(AVCodecContext *avctx, GetBitContext *gb, AVFrame *frame, AVFrame *prev) { IMM4Context *s = avctx->priv_data; int ret, x, y, offset = 0; if (s->hi == 0) { if (s->lo > 2) return AVERROR_INVALIDDATA; s->factor = inter_cb[s->lo]; } else { s->factor = s->lo * 2; } if (s->hi) { offset = s->factor; offset >>= 1; if (!(offset & 1)) offset--; } for (y = 0; y < avctx->height; y += 16) { for (x = 0; x < avctx->width; x += 16) { int reverse, intra_block, value; unsigned cbphi, cbplo, flag2 = 0; if (get_bits1(gb)) { copy_block16(frame->data[0] + y * frame->linesize[0] + x, prev->data[0] + y * prev->linesize[0] + x, frame->linesize[0], prev->linesize[0], 16); copy_block8(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1), prev->data[1] + (y >> 1) * prev->linesize[1] + (x >> 1), frame->linesize[1], prev->linesize[1], 8); copy_block8(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1), prev->data[2] + (y >> 1) * prev->linesize[2] + (x >> 1), frame->linesize[2], prev->linesize[2], 8); continue; } value = get_vlc2(gb, blktype_tab.table, blktype_tab.bits, 1); if (value < 0) return AVERROR_INVALIDDATA; intra_block = value & 0x07; reverse = intra_block == 3; if (reverse) flag2 = get_bits1(gb); cbplo = value >> 4; cbphi = get_cbphi(gb, reverse); if (intra_block) { ret = decode_blocks(avctx, gb, cbplo | (cbphi << 2), 0, offset, flag2); if (ret < 0) return ret; s->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x, frame->linesize[0], s->block[0]); s->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x + 8, frame->linesize[0], s->block[1]); s->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x, frame->linesize[0], s->block[2]); s->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8, frame->linesize[0], s->block[3]); s->idsp.idct_put(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1), frame->linesize[1], s->block[4]); s->idsp.idct_put(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1), frame->linesize[2], s->block[5]); } else { flag2 = get_bits1(gb); skip_bits1(gb); ret = decode_blocks(avctx, gb, cbplo | (cbphi << 2), 1, offset, flag2); if (ret < 0) return ret; copy_block16(frame->data[0] + y * frame->linesize[0] + x, prev->data[0] + y * prev->linesize[0] + x, frame->linesize[0], prev->linesize[0], 16); copy_block8(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1), prev->data[1] + (y >> 1) * prev->linesize[1] + (x >> 1), frame->linesize[1], prev->linesize[1], 8); copy_block8(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1), prev->data[2] + (y >> 1) * prev->linesize[2] + (x >> 1), frame->linesize[2], prev->linesize[2], 8); s->idsp.idct_add(frame->data[0] + y * frame->linesize[0] + x, frame->linesize[0], s->block[0]); s->idsp.idct_add(frame->data[0] + y * frame->linesize[0] + x + 8, frame->linesize[0], s->block[1]); s->idsp.idct_add(frame->data[0] + (y + 8) * frame->linesize[0] + x, frame->linesize[0], s->block[2]); s->idsp.idct_add(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8, frame->linesize[0], s->block[3]); s->idsp.idct_add(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1), frame->linesize[1], s->block[4]); s->idsp.idct_add(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1), frame->linesize[2], s->block[5]); } } } return 0; } static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt) { IMM4Context *s = avctx->priv_data; GetBitContext *gb = &s->gb; AVFrame *frame = data; int width, height; unsigned type; int ret, scaled; if (avpkt->size <= 32) return AVERROR_INVALIDDATA; av_fast_padded_malloc(&s->bitstream, &s->bitstream_size, FFALIGN(avpkt->size, 4)); if (!s->bitstream) return AVERROR(ENOMEM); s->bdsp.bswap_buf((uint32_t *)s->bitstream, (uint32_t *)avpkt->data, (avpkt->size + 3) >> 2); if ((ret = init_get_bits8(gb, s->bitstream, FFALIGN(avpkt->size, 4))) < 0) return ret; avctx->pix_fmt = AV_PIX_FMT_YUV420P; avctx->color_range = AVCOL_RANGE_JPEG; width = avctx->width; height = avctx->height; scaled = avpkt->data[8]; if (scaled < 2) { int mode = avpkt->data[10]; switch (mode) { case 1: width = 352; height = 240; break; case 2: width = 704; height = 240; break; case 4: width = 480; height = 704; break; case 17: width = 352; height = 288; break; case 18: width = 704; height = 288; break; default: width = 704; height = 576; break; } } skip_bits_long(gb, 24 * 8); type = get_bits_long(gb, 32); s->hi = get_bits(gb, 16); s->lo = get_bits(gb, 16); switch (type) { case 0x19781977: frame->key_frame = 1; frame->pict_type = AV_PICTURE_TYPE_I; break; case 0x12250926: frame->key_frame = 0; frame->pict_type = AV_PICTURE_TYPE_P; break; default: avpriv_request_sample(avctx, "type %X", type); return AVERROR_PATCHWELCOME; } if (avctx->width != width || avctx->height != height) { if (!frame->key_frame) { av_log(avctx, AV_LOG_ERROR, "Frame size change is unsupported.\n"); return AVERROR_INVALIDDATA; } av_frame_unref(s->prev_frame); } ret = ff_set_dimensions(avctx, width, height); if (ret < 0) return ret; if ((ret = ff_get_buffer(avctx, frame, frame->key_frame ? AV_GET_BUFFER_FLAG_REF : 0)) < 0) return ret; if (frame->key_frame) { ret = decode_intra(avctx, gb, frame); if (ret < 0) return ret; av_frame_unref(s->prev_frame); if ((ret = av_frame_ref(s->prev_frame, frame)) < 0) return ret; } else { if (!s->prev_frame->data[0]) { av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n"); return AVERROR_INVALIDDATA; } ret = decode_inter(avctx, gb, frame, s->prev_frame); if (ret < 0) return ret; } *got_frame = 1; return avpkt->size; } static av_cold void imm4_init_static_data(void) { INIT_VLC_SPARSE_STATIC(&cbplo_tab, 9, FF_ARRAY_ELEMS(cbplo_bits), cbplo_bits, 1, 1, cbplo_codes, 1, 1, cbplo_symbols, 1, 1, 512); INIT_VLC_SPARSE_STATIC(&cbphi_tab, 6, FF_ARRAY_ELEMS(cbphi_bits), cbphi_bits, 1, 1, cbphi_codes, 1, 1, NULL, 0, 0, 64); INIT_VLC_SPARSE_STATIC(&blktype_tab, 9, FF_ARRAY_ELEMS(blktype_bits), blktype_bits, 1, 1, blktype_codes, 1, 1, blktype_symbols, 1, 1, 512); INIT_VLC_SPARSE_STATIC(&block_tab, 12, FF_ARRAY_ELEMS(block_bits), block_bits, 1, 1, block_codes, 1, 1, block_symbols, 2, 2, 4096); } static av_cold int decode_init(AVCodecContext *avctx) { static AVOnce init_static_once = AV_ONCE_INIT; IMM4Context *s = avctx->priv_data; uint8_t table[64]; for (int i = 0; i < 64; i++) table[i] = i; ff_bswapdsp_init(&s->bdsp); ff_idctdsp_init(&s->idsp, avctx); ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, table); s->prev_frame = av_frame_alloc(); if (!s->prev_frame) return AVERROR(ENOMEM); ff_thread_once(&init_static_once, imm4_init_static_data); return 0; } static void decode_flush(AVCodecContext *avctx) { IMM4Context *s = avctx->priv_data; av_frame_unref(s->prev_frame); } static av_cold int decode_close(AVCodecContext *avctx) { IMM4Context *s = avctx->priv_data; av_frame_free(&s->prev_frame); av_freep(&s->bitstream); s->bitstream_size = 0; return 0; } AVCodec ff_imm4_decoder = { .name = "imm4", .long_name = NULL_IF_CONFIG_SMALL("Infinity IMM4"), .type = AVMEDIA_TYPE_VIDEO, .id = AV_CODEC_ID_IMM4, .priv_data_size = sizeof(IMM4Context), .init = decode_init, .close = decode_close, .decode = decode_frame, .flush = decode_flush, .capabilities = AV_CODEC_CAP_DR1, .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP, };