yading@10: /* yading@10: * SVQ1 decoder yading@10: * ported to MPlayer by Arpi yading@10: * ported to libavcodec by Nick Kurshev yading@10: * yading@10: * Copyright (C) 2002 the xine project yading@10: * Copyright (C) 2002 the ffmpeg project yading@10: * yading@10: * SVQ1 Encoder (c) 2004 Mike Melanson yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * Sorenson Vector Quantizer #1 (SVQ1) video codec. yading@10: * For more information of the SVQ1 algorithm, visit: yading@10: * http://www.pcisys.net/~melanson/codecs/ yading@10: */ yading@10: yading@10: #include "avcodec.h" yading@10: #include "get_bits.h" yading@10: #include "hpeldsp.h" yading@10: #include "internal.h" yading@10: #include "mathops.h" yading@10: #include "svq1.h" yading@10: yading@10: #undef NDEBUG yading@10: #include yading@10: yading@10: extern const uint8_t ff_mvtab[33][2]; yading@10: yading@10: static VLC svq1_block_type; yading@10: static VLC svq1_motion_component; yading@10: static VLC svq1_intra_multistage[6]; yading@10: static VLC svq1_inter_multistage[6]; yading@10: static VLC svq1_intra_mean; yading@10: static VLC svq1_inter_mean; yading@10: yading@10: /* motion vector (prediction) */ yading@10: typedef struct svq1_pmv_s { yading@10: int x; yading@10: int y; yading@10: } svq1_pmv; yading@10: yading@10: typedef struct SVQ1Context { yading@10: HpelDSPContext hdsp; yading@10: GetBitContext gb; yading@10: AVFrame *prev; yading@10: int width; yading@10: int height; yading@10: int frame_code; yading@10: int nonref; // 1 if the current frame won't be referenced yading@10: } SVQ1Context; yading@10: yading@10: static const uint8_t string_table[256] = { yading@10: 0x00, 0xD5, 0x7F, 0xAA, 0xFE, 0x2B, 0x81, 0x54, yading@10: 0x29, 0xFC, 0x56, 0x83, 0xD7, 0x02, 0xA8, 0x7D, yading@10: 0x52, 0x87, 0x2D, 0xF8, 0xAC, 0x79, 0xD3, 0x06, yading@10: 0x7B, 0xAE, 0x04, 0xD1, 0x85, 0x50, 0xFA, 0x2F, yading@10: 0xA4, 0x71, 0xDB, 0x0E, 0x5A, 0x8F, 0x25, 0xF0, yading@10: 0x8D, 0x58, 0xF2, 0x27, 0x73, 0xA6, 0x0C, 0xD9, yading@10: 0xF6, 0x23, 0x89, 0x5C, 0x08, 0xDD, 0x77, 0xA2, yading@10: 0xDF, 0x0A, 0xA0, 0x75, 0x21, 0xF4, 0x5E, 0x8B, yading@10: 0x9D, 0x48, 0xE2, 0x37, 0x63, 0xB6, 0x1C, 0xC9, yading@10: 0xB4, 0x61, 0xCB, 0x1E, 0x4A, 0x9F, 0x35, 0xE0, yading@10: 0xCF, 0x1A, 0xB0, 0x65, 0x31, 0xE4, 0x4E, 0x9B, yading@10: 0xE6, 0x33, 0x99, 0x4C, 0x18, 0xCD, 0x67, 0xB2, yading@10: 0x39, 0xEC, 0x46, 0x93, 0xC7, 0x12, 0xB8, 0x6D, yading@10: 0x10, 0xC5, 0x6F, 0xBA, 0xEE, 0x3B, 0x91, 0x44, yading@10: 0x6B, 0xBE, 0x14, 0xC1, 0x95, 0x40, 0xEA, 0x3F, yading@10: 0x42, 0x97, 0x3D, 0xE8, 0xBC, 0x69, 0xC3, 0x16, yading@10: 0xEF, 0x3A, 0x90, 0x45, 0x11, 0xC4, 0x6E, 0xBB, yading@10: 0xC6, 0x13, 0xB9, 0x6C, 0x38, 0xED, 0x47, 0x92, yading@10: 0xBD, 0x68, 0xC2, 0x17, 0x43, 0x96, 0x3C, 0xE9, yading@10: 0x94, 0x41, 0xEB, 0x3E, 0x6A, 0xBF, 0x15, 0xC0, yading@10: 0x4B, 0x9E, 0x34, 0xE1, 0xB5, 0x60, 0xCA, 0x1F, yading@10: 0x62, 0xB7, 0x1D, 0xC8, 0x9C, 0x49, 0xE3, 0x36, yading@10: 0x19, 0xCC, 0x66, 0xB3, 0xE7, 0x32, 0x98, 0x4D, yading@10: 0x30, 0xE5, 0x4F, 0x9A, 0xCE, 0x1B, 0xB1, 0x64, yading@10: 0x72, 0xA7, 0x0D, 0xD8, 0x8C, 0x59, 0xF3, 0x26, yading@10: 0x5B, 0x8E, 0x24, 0xF1, 0xA5, 0x70, 0xDA, 0x0F, yading@10: 0x20, 0xF5, 0x5F, 0x8A, 0xDE, 0x0B, 0xA1, 0x74, yading@10: 0x09, 0xDC, 0x76, 0xA3, 0xF7, 0x22, 0x88, 0x5D, yading@10: 0xD6, 0x03, 0xA9, 0x7C, 0x28, 0xFD, 0x57, 0x82, yading@10: 0xFF, 0x2A, 0x80, 0x55, 0x01, 0xD4, 0x7E, 0xAB, yading@10: 0x84, 0x51, 0xFB, 0x2E, 0x7A, 0xAF, 0x05, 0xD0, yading@10: 0xAD, 0x78, 0xD2, 0x07, 0x53, 0x86, 0x2C, 0xF9 yading@10: }; yading@10: yading@10: #define SVQ1_PROCESS_VECTOR() \ yading@10: for (; level > 0; i++) { \ yading@10: /* process next depth */ \ yading@10: if (i == m) { \ yading@10: m = n; \ yading@10: if (--level == 0) \ yading@10: break; \ yading@10: } \ yading@10: /* divide block if next bit set */ \ yading@10: if (!get_bits1(bitbuf)) \ yading@10: break; \ yading@10: /* add child nodes */ \ yading@10: list[n++] = list[i]; \ yading@10: list[n++] = list[i] + (((level & 1) ? pitch : 1) << ((level >> 1) + 1));\ yading@10: } yading@10: yading@10: #define SVQ1_ADD_CODEBOOK() \ yading@10: /* add codebook entries to vector */ \ yading@10: for (j = 0; j < stages; j++) { \ yading@10: n3 = codebook[entries[j]] ^ 0x80808080; \ yading@10: n1 += (n3 & 0xFF00FF00) >> 8; \ yading@10: n2 += n3 & 0x00FF00FF; \ yading@10: } \ yading@10: \ yading@10: /* clip to [0..255] */ \ yading@10: if (n1 & 0xFF00FF00) { \ yading@10: n3 = (n1 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \ yading@10: n1 += 0x7F007F00; \ yading@10: n1 |= (~n1 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \ yading@10: n1 &= n3 & 0x00FF00FF; \ yading@10: } \ yading@10: \ yading@10: if (n2 & 0xFF00FF00) { \ yading@10: n3 = (n2 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \ yading@10: n2 += 0x7F007F00; \ yading@10: n2 |= (~n2 >> 15 & 0x00010001 | 0x01000100) - 0x00010001; \ yading@10: n2 &= n3 & 0x00FF00FF; \ yading@10: } yading@10: yading@10: #define SVQ1_CALC_CODEBOOK_ENTRIES(cbook) \ yading@10: codebook = (const uint32_t *)cbook[level]; \ yading@10: if (stages > 0) \ yading@10: bit_cache = get_bits(bitbuf, 4 * stages); \ yading@10: /* calculate codebook entries for this vector */ \ yading@10: for (j = 0; j < stages; j++) { \ yading@10: entries[j] = (((bit_cache >> (4 * (stages - j - 1))) & 0xF) + \ yading@10: 16 * j) << (level + 1); \ yading@10: } \ yading@10: mean -= stages * 128; \ yading@10: n4 = (mean << 16) + mean; yading@10: yading@10: static int svq1_decode_block_intra(GetBitContext *bitbuf, uint8_t *pixels, yading@10: int pitch) yading@10: { yading@10: uint32_t bit_cache; yading@10: uint8_t *list[63]; yading@10: uint32_t *dst; yading@10: const uint32_t *codebook; yading@10: int entries[6]; yading@10: int i, j, m, n; yading@10: int mean, stages; yading@10: unsigned x, y, width, height, level; yading@10: uint32_t n1, n2, n3, n4; yading@10: yading@10: /* initialize list for breadth first processing of vectors */ yading@10: list[0] = pixels; yading@10: yading@10: /* recursively process vector */ yading@10: for (i = 0, m = 1, n = 1, level = 5; i < n; i++) { yading@10: SVQ1_PROCESS_VECTOR(); yading@10: yading@10: /* destination address and vector size */ yading@10: dst = (uint32_t *)list[i]; yading@10: width = 1 << ((4 + level) / 2); yading@10: height = 1 << ((3 + level) / 2); yading@10: yading@10: /* get number of stages (-1 skips vector, 0 for mean only) */ yading@10: stages = get_vlc2(bitbuf, svq1_intra_multistage[level].table, 3, 3) - 1; yading@10: yading@10: if (stages == -1) { yading@10: for (y = 0; y < height; y++) yading@10: memset(&dst[y * (pitch / 4)], 0, width); yading@10: continue; /* skip vector */ yading@10: } yading@10: yading@10: if (stages > 0 && level >= 4) { yading@10: av_dlog(NULL, yading@10: "Error (svq1_decode_block_intra): invalid vector: stages=%i level=%i\n", yading@10: stages, level); yading@10: return AVERROR_INVALIDDATA; /* invalid vector */ yading@10: } yading@10: yading@10: mean = get_vlc2(bitbuf, svq1_intra_mean.table, 8, 3); yading@10: yading@10: if (stages == 0) { yading@10: for (y = 0; y < height; y++) yading@10: memset(&dst[y * (pitch / 4)], mean, width); yading@10: } else { yading@10: SVQ1_CALC_CODEBOOK_ENTRIES(ff_svq1_intra_codebooks); yading@10: yading@10: for (y = 0; y < height; y++) { yading@10: for (x = 0; x < width / 4; x++, codebook++) { yading@10: n1 = n4; yading@10: n2 = n4; yading@10: SVQ1_ADD_CODEBOOK() yading@10: /* store result */ yading@10: dst[x] = n1 << 8 | n2; yading@10: } yading@10: dst += pitch / 4; yading@10: } yading@10: } yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int svq1_decode_block_non_intra(GetBitContext *bitbuf, uint8_t *pixels, yading@10: int pitch) yading@10: { yading@10: uint32_t bit_cache; yading@10: uint8_t *list[63]; yading@10: uint32_t *dst; yading@10: const uint32_t *codebook; yading@10: int entries[6]; yading@10: int i, j, m, n; yading@10: int mean, stages; yading@10: int x, y, width, height, level; yading@10: uint32_t n1, n2, n3, n4; yading@10: yading@10: /* initialize list for breadth first processing of vectors */ yading@10: list[0] = pixels; yading@10: yading@10: /* recursively process vector */ yading@10: for (i = 0, m = 1, n = 1, level = 5; i < n; i++) { yading@10: SVQ1_PROCESS_VECTOR(); yading@10: yading@10: /* destination address and vector size */ yading@10: dst = (uint32_t *)list[i]; yading@10: width = 1 << ((4 + level) / 2); yading@10: height = 1 << ((3 + level) / 2); yading@10: yading@10: /* get number of stages (-1 skips vector, 0 for mean only) */ yading@10: stages = get_vlc2(bitbuf, svq1_inter_multistage[level].table, 3, 2) - 1; yading@10: yading@10: if (stages == -1) yading@10: continue; /* skip vector */ yading@10: yading@10: if ((stages > 0) && (level >= 4)) { yading@10: av_dlog(NULL, yading@10: "Error (svq1_decode_block_non_intra): invalid vector: stages=%i level=%i\n", yading@10: stages, level); yading@10: return AVERROR_INVALIDDATA; /* invalid vector */ yading@10: } yading@10: yading@10: mean = get_vlc2(bitbuf, svq1_inter_mean.table, 9, 3) - 256; yading@10: yading@10: SVQ1_CALC_CODEBOOK_ENTRIES(ff_svq1_inter_codebooks); yading@10: yading@10: for (y = 0; y < height; y++) { yading@10: for (x = 0; x < width / 4; x++, codebook++) { yading@10: n3 = dst[x]; yading@10: /* add mean value to vector */ yading@10: n1 = n4 + ((n3 & 0xFF00FF00) >> 8); yading@10: n2 = n4 + (n3 & 0x00FF00FF); yading@10: SVQ1_ADD_CODEBOOK() yading@10: /* store result */ yading@10: dst[x] = n1 << 8 | n2; yading@10: } yading@10: dst += pitch / 4; yading@10: } yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static int svq1_decode_motion_vector(GetBitContext *bitbuf, svq1_pmv *mv, yading@10: svq1_pmv **pmv) yading@10: { yading@10: int diff; yading@10: int i; yading@10: yading@10: for (i = 0; i < 2; i++) { yading@10: /* get motion code */ yading@10: diff = get_vlc2(bitbuf, svq1_motion_component.table, 7, 2); yading@10: if (diff < 0) yading@10: return AVERROR_INVALIDDATA; yading@10: else if (diff) { yading@10: if (get_bits1(bitbuf)) yading@10: diff = -diff; yading@10: } yading@10: yading@10: /* add median of motion vector predictors and clip result */ yading@10: if (i == 1) yading@10: mv->y = sign_extend(diff + mid_pred(pmv[0]->y, pmv[1]->y, pmv[2]->y), 6); yading@10: else yading@10: mv->x = sign_extend(diff + mid_pred(pmv[0]->x, pmv[1]->x, pmv[2]->x), 6); yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static void svq1_skip_block(uint8_t *current, uint8_t *previous, yading@10: int pitch, int x, int y) yading@10: { yading@10: uint8_t *src; yading@10: uint8_t *dst; yading@10: int i; yading@10: yading@10: src = &previous[x + y * pitch]; yading@10: dst = current; yading@10: yading@10: for (i = 0; i < 16; i++) { yading@10: memcpy(dst, src, 16); yading@10: src += pitch; yading@10: dst += pitch; yading@10: } yading@10: } yading@10: yading@10: static int svq1_motion_inter_block(HpelDSPContext *hdsp, GetBitContext *bitbuf, yading@10: uint8_t *current, uint8_t *previous, yading@10: int pitch, svq1_pmv *motion, int x, int y, yading@10: int width, int height) yading@10: { yading@10: uint8_t *src; yading@10: uint8_t *dst; yading@10: svq1_pmv mv; yading@10: svq1_pmv *pmv[3]; yading@10: int result; yading@10: yading@10: /* predict and decode motion vector */ yading@10: pmv[0] = &motion[0]; yading@10: if (y == 0) { yading@10: pmv[1] = yading@10: pmv[2] = pmv[0]; yading@10: } else { yading@10: pmv[1] = &motion[x / 8 + 2]; yading@10: pmv[2] = &motion[x / 8 + 4]; yading@10: } yading@10: yading@10: result = svq1_decode_motion_vector(bitbuf, &mv, pmv); yading@10: if (result) yading@10: return result; yading@10: yading@10: motion[0].x = yading@10: motion[x / 8 + 2].x = yading@10: motion[x / 8 + 3].x = mv.x; yading@10: motion[0].y = yading@10: motion[x / 8 + 2].y = yading@10: motion[x / 8 + 3].y = mv.y; yading@10: yading@10: mv.x = av_clip(mv.x, -2 * x, 2 * (width - x - 16)); yading@10: mv.y = av_clip(mv.y, -2 * y, 2 * (height - y - 16)); yading@10: yading@10: src = &previous[(x + (mv.x >> 1)) + (y + (mv.y >> 1)) * pitch]; yading@10: dst = current; yading@10: yading@10: hdsp->put_pixels_tab[0][(mv.y & 1) << 1 | (mv.x & 1)](dst, src, pitch, 16); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int svq1_motion_inter_4v_block(HpelDSPContext *hdsp, GetBitContext *bitbuf, yading@10: uint8_t *current, uint8_t *previous, yading@10: int pitch, svq1_pmv *motion, int x, int y, yading@10: int width, int height) yading@10: { yading@10: uint8_t *src; yading@10: uint8_t *dst; yading@10: svq1_pmv mv; yading@10: svq1_pmv *pmv[4]; yading@10: int i, result; yading@10: yading@10: /* predict and decode motion vector (0) */ yading@10: pmv[0] = &motion[0]; yading@10: if (y == 0) { yading@10: pmv[1] = yading@10: pmv[2] = pmv[0]; yading@10: } else { yading@10: pmv[1] = &motion[(x / 8) + 2]; yading@10: pmv[2] = &motion[(x / 8) + 4]; yading@10: } yading@10: yading@10: result = svq1_decode_motion_vector(bitbuf, &mv, pmv); yading@10: if (result) yading@10: return result; yading@10: yading@10: /* predict and decode motion vector (1) */ yading@10: pmv[0] = &mv; yading@10: if (y == 0) { yading@10: pmv[1] = yading@10: pmv[2] = pmv[0]; yading@10: } else { yading@10: pmv[1] = &motion[(x / 8) + 3]; yading@10: } yading@10: result = svq1_decode_motion_vector(bitbuf, &motion[0], pmv); yading@10: if (result) yading@10: return result; yading@10: yading@10: /* predict and decode motion vector (2) */ yading@10: pmv[1] = &motion[0]; yading@10: pmv[2] = &motion[(x / 8) + 1]; yading@10: yading@10: result = svq1_decode_motion_vector(bitbuf, &motion[(x / 8) + 2], pmv); yading@10: if (result) yading@10: return result; yading@10: yading@10: /* predict and decode motion vector (3) */ yading@10: pmv[2] = &motion[(x / 8) + 2]; yading@10: pmv[3] = &motion[(x / 8) + 3]; yading@10: yading@10: result = svq1_decode_motion_vector(bitbuf, pmv[3], pmv); yading@10: if (result) yading@10: return result; yading@10: yading@10: /* form predictions */ yading@10: for (i = 0; i < 4; i++) { yading@10: int mvx = pmv[i]->x + (i & 1) * 16; yading@10: int mvy = pmv[i]->y + (i >> 1) * 16; yading@10: yading@10: // FIXME: clipping or padding? yading@10: mvx = av_clip(mvx, -2 * x, 2 * (width - x - 8)); yading@10: mvy = av_clip(mvy, -2 * y, 2 * (height - y - 8)); yading@10: yading@10: src = &previous[(x + (mvx >> 1)) + (y + (mvy >> 1)) * pitch]; yading@10: dst = current; yading@10: yading@10: hdsp->put_pixels_tab[1][((mvy & 1) << 1) | (mvx & 1)](dst, src, pitch, 8); yading@10: yading@10: /* select next block */ yading@10: if (i & 1) yading@10: current += 8 * (pitch - 1); yading@10: else yading@10: current += 8; yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int svq1_decode_delta_block(AVCodecContext *avctx, HpelDSPContext *hdsp, yading@10: GetBitContext *bitbuf, yading@10: uint8_t *current, uint8_t *previous, yading@10: int pitch, svq1_pmv *motion, int x, int y, yading@10: int width, int height) yading@10: { yading@10: uint32_t block_type; yading@10: int result = 0; yading@10: yading@10: /* get block type */ yading@10: block_type = get_vlc2(bitbuf, svq1_block_type.table, 2, 2); yading@10: yading@10: /* reset motion vectors */ yading@10: if (block_type == SVQ1_BLOCK_SKIP || block_type == SVQ1_BLOCK_INTRA) { yading@10: motion[0].x = yading@10: motion[0].y = yading@10: motion[x / 8 + 2].x = yading@10: motion[x / 8 + 2].y = yading@10: motion[x / 8 + 3].x = yading@10: motion[x / 8 + 3].y = 0; yading@10: } yading@10: yading@10: switch (block_type) { yading@10: case SVQ1_BLOCK_SKIP: yading@10: svq1_skip_block(current, previous, pitch, x, y); yading@10: break; yading@10: yading@10: case SVQ1_BLOCK_INTER: yading@10: result = svq1_motion_inter_block(hdsp, bitbuf, current, previous, yading@10: pitch, motion, x, y, width, height); yading@10: yading@10: if (result != 0) { yading@10: av_dlog(avctx, "Error in svq1_motion_inter_block %i\n", result); yading@10: break; yading@10: } yading@10: result = svq1_decode_block_non_intra(bitbuf, current, pitch); yading@10: break; yading@10: yading@10: case SVQ1_BLOCK_INTER_4V: yading@10: result = svq1_motion_inter_4v_block(hdsp, bitbuf, current, previous, yading@10: pitch, motion, x, y, width, height); yading@10: yading@10: if (result != 0) { yading@10: av_dlog(avctx, "Error in svq1_motion_inter_4v_block %i\n", result); yading@10: break; yading@10: } yading@10: result = svq1_decode_block_non_intra(bitbuf, current, pitch); yading@10: break; yading@10: yading@10: case SVQ1_BLOCK_INTRA: yading@10: result = svq1_decode_block_intra(bitbuf, current, pitch); yading@10: break; yading@10: } yading@10: yading@10: return result; yading@10: } yading@10: yading@10: static void svq1_parse_string(GetBitContext *bitbuf, uint8_t *out) yading@10: { yading@10: uint8_t seed; yading@10: int i; yading@10: yading@10: out[0] = get_bits(bitbuf, 8); yading@10: seed = string_table[out[0]]; yading@10: yading@10: for (i = 1; i <= out[0]; i++) { yading@10: out[i] = get_bits(bitbuf, 8) ^ seed; yading@10: seed = string_table[out[i] ^ seed]; yading@10: } yading@10: } yading@10: yading@10: static int svq1_decode_frame_header(AVCodecContext *avctx, AVFrame *frame) yading@10: { yading@10: SVQ1Context *s = avctx->priv_data; yading@10: GetBitContext *bitbuf = &s->gb; yading@10: int frame_size_code; yading@10: int width = s->width; yading@10: int height = s->height; yading@10: yading@10: skip_bits(bitbuf, 8); /* temporal_reference */ yading@10: yading@10: /* frame type */ yading@10: s->nonref = 0; yading@10: switch (get_bits(bitbuf, 2)) { yading@10: case 0: yading@10: frame->pict_type = AV_PICTURE_TYPE_I; yading@10: break; yading@10: case 2: yading@10: s->nonref = 1; yading@10: case 1: yading@10: frame->pict_type = AV_PICTURE_TYPE_P; yading@10: break; yading@10: default: yading@10: av_log(avctx, AV_LOG_ERROR, "Invalid frame type.\n"); yading@10: return AVERROR_INVALIDDATA; yading@10: } yading@10: yading@10: if (frame->pict_type == AV_PICTURE_TYPE_I) { yading@10: /* unknown fields */ yading@10: if (s->frame_code == 0x50 || s->frame_code == 0x60) { yading@10: int csum = get_bits(bitbuf, 16); yading@10: yading@10: csum = ff_svq1_packet_checksum(bitbuf->buffer, yading@10: bitbuf->size_in_bits >> 3, yading@10: csum); yading@10: yading@10: av_dlog(avctx, "%s checksum (%02x) for packet data\n", yading@10: (csum == 0) ? "correct" : "incorrect", csum); yading@10: } yading@10: yading@10: if ((s->frame_code ^ 0x10) >= 0x50) { yading@10: uint8_t msg[256]; yading@10: yading@10: svq1_parse_string(bitbuf, msg); yading@10: yading@10: av_log(avctx, AV_LOG_INFO, yading@10: "embedded message: \"%s\"\n", (char *)msg); yading@10: } yading@10: yading@10: skip_bits(bitbuf, 2); yading@10: skip_bits(bitbuf, 2); yading@10: skip_bits1(bitbuf); yading@10: yading@10: /* load frame size */ yading@10: frame_size_code = get_bits(bitbuf, 3); yading@10: yading@10: if (frame_size_code == 7) { yading@10: /* load width, height (12 bits each) */ yading@10: width = get_bits(bitbuf, 12); yading@10: height = get_bits(bitbuf, 12); yading@10: yading@10: if (!width || !height) yading@10: return AVERROR_INVALIDDATA; yading@10: } else { yading@10: /* get width, height from table */ yading@10: width = ff_svq1_frame_size_table[frame_size_code][0]; yading@10: height = ff_svq1_frame_size_table[frame_size_code][1]; yading@10: } yading@10: } yading@10: yading@10: /* unknown fields */ yading@10: if (get_bits1(bitbuf)) { yading@10: skip_bits1(bitbuf); /* use packet checksum if (1) */ yading@10: skip_bits1(bitbuf); /* component checksums after image data if (1) */ yading@10: yading@10: if (get_bits(bitbuf, 2) != 0) yading@10: return AVERROR_INVALIDDATA; yading@10: } yading@10: yading@10: if (get_bits1(bitbuf)) { yading@10: skip_bits1(bitbuf); yading@10: skip_bits(bitbuf, 4); yading@10: skip_bits1(bitbuf); yading@10: skip_bits(bitbuf, 2); yading@10: yading@10: while (get_bits1(bitbuf)) yading@10: skip_bits(bitbuf, 8); yading@10: } yading@10: yading@10: s->width = width; yading@10: s->height = height; yading@10: return 0; yading@10: } yading@10: yading@10: static int svq1_decode_frame(AVCodecContext *avctx, void *data, yading@10: int *got_frame, AVPacket *avpkt) yading@10: { yading@10: const uint8_t *buf = avpkt->data; yading@10: int buf_size = avpkt->size; yading@10: SVQ1Context *s = avctx->priv_data; yading@10: AVFrame *cur = data; yading@10: uint8_t *current; yading@10: int result, i, x, y, width, height; yading@10: svq1_pmv *pmv; yading@10: yading@10: /* initialize bit buffer */ yading@10: init_get_bits(&s->gb, buf, buf_size * 8); yading@10: yading@10: /* decode frame header */ yading@10: s->frame_code = get_bits(&s->gb, 22); yading@10: yading@10: if ((s->frame_code & ~0x70) || !(s->frame_code & 0x60)) yading@10: return AVERROR_INVALIDDATA; yading@10: yading@10: /* swap some header bytes (why?) */ yading@10: if (s->frame_code != 0x20) { yading@10: uint32_t *src = (uint32_t *)(buf + 4); yading@10: yading@10: if (buf_size < 36) yading@10: return AVERROR_INVALIDDATA; yading@10: yading@10: for (i = 0; i < 4; i++) yading@10: src[i] = ((src[i] << 16) | (src[i] >> 16)) ^ src[7 - i]; yading@10: } yading@10: yading@10: result = svq1_decode_frame_header(avctx, cur); yading@10: if (result != 0) { yading@10: av_dlog(avctx, "Error in svq1_decode_frame_header %i\n", result); yading@10: return result; yading@10: } yading@10: avcodec_set_dimensions(avctx, s->width, s->height); yading@10: yading@10: if ((avctx->skip_frame >= AVDISCARD_NONREF && s->nonref) || yading@10: (avctx->skip_frame >= AVDISCARD_NONKEY && yading@10: cur->pict_type != AV_PICTURE_TYPE_I) || yading@10: avctx->skip_frame >= AVDISCARD_ALL) yading@10: return buf_size; yading@10: yading@10: result = ff_get_buffer(avctx, cur, s->nonref ? 0 : AV_GET_BUFFER_FLAG_REF); yading@10: if (result < 0) yading@10: return result; yading@10: yading@10: pmv = av_malloc((FFALIGN(s->width, 16) / 8 + 3) * sizeof(*pmv)); yading@10: if (!pmv) yading@10: return AVERROR(ENOMEM); yading@10: yading@10: /* decode y, u and v components */ yading@10: for (i = 0; i < 3; i++) { yading@10: int linesize = cur->linesize[i]; yading@10: if (i == 0) { yading@10: width = FFALIGN(s->width, 16); yading@10: height = FFALIGN(s->height, 16); yading@10: } else { yading@10: if (avctx->flags & CODEC_FLAG_GRAY) yading@10: break; yading@10: width = FFALIGN(s->width / 4, 16); yading@10: height = FFALIGN(s->height / 4, 16); yading@10: } yading@10: yading@10: current = cur->data[i]; yading@10: yading@10: if (cur->pict_type == AV_PICTURE_TYPE_I) { yading@10: /* keyframe */ yading@10: for (y = 0; y < height; y += 16) { yading@10: for (x = 0; x < width; x += 16) { yading@10: result = svq1_decode_block_intra(&s->gb, ¤t[x], yading@10: linesize); yading@10: if (result) { yading@10: av_log(avctx, AV_LOG_ERROR, yading@10: "Error in svq1_decode_block %i (keyframe)\n", yading@10: result); yading@10: goto err; yading@10: } yading@10: } yading@10: current += 16 * linesize; yading@10: } yading@10: } else { yading@10: /* delta frame */ yading@10: uint8_t *previous = s->prev->data[i]; yading@10: if (!previous || yading@10: s->prev->width != s->width || s->prev->height != s->height) { yading@10: av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n"); yading@10: result = AVERROR_INVALIDDATA; yading@10: goto err; yading@10: } yading@10: yading@10: memset(pmv, 0, ((width / 8) + 3) * sizeof(svq1_pmv)); yading@10: yading@10: for (y = 0; y < height; y += 16) { yading@10: for (x = 0; x < width; x += 16) { yading@10: result = svq1_decode_delta_block(avctx, &s->hdsp, yading@10: &s->gb, ¤t[x], yading@10: previous, linesize, yading@10: pmv, x, y, width, height); yading@10: if (result != 0) { yading@10: av_dlog(avctx, yading@10: "Error in svq1_decode_delta_block %i\n", yading@10: result); yading@10: goto err; yading@10: } yading@10: } yading@10: yading@10: pmv[0].x = yading@10: pmv[0].y = 0; yading@10: yading@10: current += 16 * linesize; yading@10: } yading@10: } yading@10: } yading@10: yading@10: if (!s->nonref) { yading@10: av_frame_unref(s->prev); yading@10: result = av_frame_ref(s->prev, cur); yading@10: if (result < 0) yading@10: goto err; yading@10: } yading@10: yading@10: *got_frame = 1; yading@10: result = buf_size; yading@10: yading@10: err: yading@10: av_free(pmv); yading@10: return result; yading@10: } yading@10: yading@10: static av_cold int svq1_decode_init(AVCodecContext *avctx) yading@10: { yading@10: SVQ1Context *s = avctx->priv_data; yading@10: int i; yading@10: int offset = 0; yading@10: yading@10: s->prev = avcodec_alloc_frame(); yading@10: if (!s->prev) yading@10: return AVERROR(ENOMEM); yading@10: yading@10: s->width = avctx->width + 3 & ~3; yading@10: s->height = avctx->height + 3 & ~3; yading@10: avctx->pix_fmt = AV_PIX_FMT_YUV410P; yading@10: yading@10: ff_hpeldsp_init(&s->hdsp, avctx->flags); yading@10: yading@10: INIT_VLC_STATIC(&svq1_block_type, 2, 4, yading@10: &ff_svq1_block_type_vlc[0][1], 2, 1, yading@10: &ff_svq1_block_type_vlc[0][0], 2, 1, 6); yading@10: yading@10: INIT_VLC_STATIC(&svq1_motion_component, 7, 33, yading@10: &ff_mvtab[0][1], 2, 1, yading@10: &ff_mvtab[0][0], 2, 1, 176); yading@10: yading@10: for (i = 0; i < 6; i++) { yading@10: static const uint8_t sizes[2][6] = { { 14, 10, 14, 18, 16, 18 }, yading@10: { 10, 10, 14, 14, 14, 16 } }; yading@10: static VLC_TYPE table[168][2]; yading@10: svq1_intra_multistage[i].table = &table[offset]; yading@10: svq1_intra_multistage[i].table_allocated = sizes[0][i]; yading@10: offset += sizes[0][i]; yading@10: init_vlc(&svq1_intra_multistage[i], 3, 8, yading@10: &ff_svq1_intra_multistage_vlc[i][0][1], 2, 1, yading@10: &ff_svq1_intra_multistage_vlc[i][0][0], 2, 1, yading@10: INIT_VLC_USE_NEW_STATIC); yading@10: svq1_inter_multistage[i].table = &table[offset]; yading@10: svq1_inter_multistage[i].table_allocated = sizes[1][i]; yading@10: offset += sizes[1][i]; yading@10: init_vlc(&svq1_inter_multistage[i], 3, 8, yading@10: &ff_svq1_inter_multistage_vlc[i][0][1], 2, 1, yading@10: &ff_svq1_inter_multistage_vlc[i][0][0], 2, 1, yading@10: INIT_VLC_USE_NEW_STATIC); yading@10: } yading@10: yading@10: INIT_VLC_STATIC(&svq1_intra_mean, 8, 256, yading@10: &ff_svq1_intra_mean_vlc[0][1], 4, 2, yading@10: &ff_svq1_intra_mean_vlc[0][0], 4, 2, 632); yading@10: yading@10: INIT_VLC_STATIC(&svq1_inter_mean, 9, 512, yading@10: &ff_svq1_inter_mean_vlc[0][1], 4, 2, yading@10: &ff_svq1_inter_mean_vlc[0][0], 4, 2, 1434); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static av_cold int svq1_decode_end(AVCodecContext *avctx) yading@10: { yading@10: SVQ1Context *s = avctx->priv_data; yading@10: yading@10: av_frame_free(&s->prev); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static void svq1_flush(AVCodecContext *avctx) yading@10: { yading@10: SVQ1Context *s = avctx->priv_data; yading@10: yading@10: av_frame_unref(s->prev); yading@10: } yading@10: yading@10: AVCodec ff_svq1_decoder = { yading@10: .name = "svq1", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_SVQ1, yading@10: .priv_data_size = sizeof(SVQ1Context), yading@10: .init = svq1_decode_init, yading@10: .close = svq1_decode_end, yading@10: .decode = svq1_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .flush = svq1_flush, yading@10: .pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUV410P, yading@10: AV_PIX_FMT_NONE }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("Sorenson Vector Quantizer 1 / Sorenson Video 1 / SVQ1"), yading@10: };