libavcodec/imm4.c - third_party/ffmpeg - Git at Google

 /*
  * 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,
 };
	/*
	* 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,
	};