mirror of https://github.com/FFmpeg/FFmpeg.git
Signed-off-by: Paul B Mahol <onemda@gmail.com> Signed-off-by: Diego Biurrun <diego@biurrun.de>pull/272/head
parent
89d9869d24
commit
95a8a03a19
9 changed files with 650 additions and 1 deletions
@ -0,0 +1,636 @@ |
|||||||
|
/*
|
||||||
|
* FM Screen Capture Codec decoder |
||||||
|
* |
||||||
|
* Copyright (c) 2017 Paul B Mahol |
||||||
|
* |
||||||
|
* This file is part of Libav. |
||||||
|
* |
||||||
|
* Libav 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. |
||||||
|
* |
||||||
|
* Libav 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 Libav; 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 "avcodec.h" |
||||||
|
#include "bytestream.h" |
||||||
|
#include "internal.h" |
||||||
|
|
||||||
|
#define BLOCK_HEIGHT 112u |
||||||
|
#define BLOCK_WIDTH 84u |
||||||
|
|
||||||
|
typedef struct InterBlock { |
||||||
|
int w, h; |
||||||
|
int size; |
||||||
|
int xor; |
||||||
|
} InterBlock; |
||||||
|
|
||||||
|
typedef struct FMVCContext { |
||||||
|
GetByteContext gb; |
||||||
|
PutByteContext pb; |
||||||
|
uint8_t *buffer; |
||||||
|
size_t buffer_size; |
||||||
|
uint8_t *pbuffer; |
||||||
|
size_t pbuffer_size; |
||||||
|
ptrdiff_t stride; |
||||||
|
int bpp; |
||||||
|
int yb, xb; |
||||||
|
InterBlock *blocks; |
||||||
|
unsigned nb_blocks; |
||||||
|
} FMVCContext; |
||||||
|
|
||||||
|
static int decode_type2(GetByteContext *gb, PutByteContext *pb) |
||||||
|
{ |
||||||
|
unsigned repeat = 0, first = 1, opcode; |
||||||
|
int i, len, pos; |
||||||
|
|
||||||
|
while (bytestream2_get_bytes_left(gb) > 0) { |
||||||
|
GetByteContext gbc; |
||||||
|
|
||||||
|
while (bytestream2_get_bytes_left(gb) > 0) { |
||||||
|
if (first) { |
||||||
|
first = 0; |
||||||
|
if (bytestream2_peek_byte(gb) > 17) { |
||||||
|
len = bytestream2_get_byte(gb) - 17; |
||||||
|
if (len < 4) { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
continue; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
if (opcode < 0x10) { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
pos = - (opcode >> 2) - 4 * bytestream2_get_byte(gb) - 2049; |
||||||
|
|
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
|
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
len = opcode & 3; |
||||||
|
if (!len) { |
||||||
|
repeat = 1; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
} |
||||||
|
continue; |
||||||
|
} |
||||||
|
} |
||||||
|
repeat = 0; |
||||||
|
} |
||||||
|
repeat = 1; |
||||||
|
} |
||||||
|
if (repeat) { |
||||||
|
repeat = 0; |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
if (opcode < 0x10) { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
if (!opcode) { |
||||||
|
if (!bytestream2_peek_byte(gb)) { |
||||||
|
do { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
opcode += 255; |
||||||
|
} while (!bytestream2_peek_byte(gb) && bytestream2_get_bytes_left(gb) > 0); |
||||||
|
} |
||||||
|
opcode += bytestream2_get_byte(gb) + 15; |
||||||
|
} |
||||||
|
bytestream2_put_le32(pb, bytestream2_get_le32(gb)); |
||||||
|
for (i = opcode - 1; i > 0; --i) |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
if (opcode < 0x10) { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
pos = - (opcode >> 2) - 4 * bytestream2_get_byte(gb) - 2049; |
||||||
|
|
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
|
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
len = opcode & 3; |
||||||
|
if (!len) { |
||||||
|
repeat = 1; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
} |
||||||
|
continue; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if (opcode >= 0x40) { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
pos = - ((opcode >> 2) & 7) - 1 - 8 * bytestream2_get_byte(gb); |
||||||
|
len = (opcode >> 5) - 1; |
||||||
|
|
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
|
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
|
||||||
|
len = opcode & 3; |
||||||
|
|
||||||
|
if (!len) { |
||||||
|
repeat = 1; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
} |
||||||
|
continue; |
||||||
|
} else if (opcode < 0x20) { |
||||||
|
break; |
||||||
|
} |
||||||
|
len = opcode & 0x1F; |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
if (!len) { |
||||||
|
if (!bytestream2_peek_byte(gb)) { |
||||||
|
do { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
len += 255; |
||||||
|
} while (!bytestream2_peek_byte(gb) && bytestream2_get_bytes_left(gb) > 0); |
||||||
|
} |
||||||
|
len += bytestream2_get_byte(gb) + 31; |
||||||
|
} |
||||||
|
i = bytestream2_get_le16(gb); |
||||||
|
pos = - (i >> 2) - 1; |
||||||
|
|
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
|
||||||
|
if (len < 6 || bytestream2_tell_p(pb) - bytestream2_tell(&gbc) < 4) { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
} else { |
||||||
|
bytestream2_put_le32(pb, bytestream2_get_le32(&gbc)); |
||||||
|
for (len = len - 2; len; --len) |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
} |
||||||
|
len = i & 3; |
||||||
|
if (!len) { |
||||||
|
repeat = 1; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
} |
||||||
|
} |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
if (opcode < 0x10) { |
||||||
|
pos = -(opcode >> 2) - 1 - 4 * bytestream2_get_byte(gb); |
||||||
|
|
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
|
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
len = opcode & 3; |
||||||
|
if (!len) { |
||||||
|
repeat = 1; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
} |
||||||
|
continue; |
||||||
|
} |
||||||
|
len = opcode & 7; |
||||||
|
if (!len) { |
||||||
|
if (!bytestream2_peek_byte(gb)) { |
||||||
|
do { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
len += 255; |
||||||
|
} while (!bytestream2_peek_byte(gb) && bytestream2_get_bytes_left(gb) > 0); |
||||||
|
} |
||||||
|
len += bytestream2_get_byte(gb) + 7; |
||||||
|
} |
||||||
|
i = bytestream2_get_le16(gb); |
||||||
|
pos = bytestream2_tell_p(pb) - 2048 * (opcode & 8); |
||||||
|
pos = pos - (i >> 2); |
||||||
|
if (pos == bytestream2_tell_p(pb)) |
||||||
|
break; |
||||||
|
|
||||||
|
pos = pos - 0x4000; |
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, pos, SEEK_SET); |
||||||
|
|
||||||
|
if (len < 6 || bytestream2_tell_p(pb) - bytestream2_tell(&gbc) < 4) { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
} else { |
||||||
|
bytestream2_put_le32(pb, bytestream2_get_le32(&gbc)); |
||||||
|
for (len = len - 2; len; --len) |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
} |
||||||
|
|
||||||
|
len = i & 3; |
||||||
|
if (!len) { |
||||||
|
repeat = 1; |
||||||
|
} else { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--len; |
||||||
|
} while (len); |
||||||
|
opcode = bytestream2_peek_byte(gb); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int decode_type1(GetByteContext *gb, PutByteContext *pb) |
||||||
|
{ |
||||||
|
unsigned opcode, len; |
||||||
|
int high = 0; |
||||||
|
int i, pos; |
||||||
|
|
||||||
|
while (bytestream2_get_bytes_left(gb) > 0) { |
||||||
|
GetByteContext gbc; |
||||||
|
|
||||||
|
while (bytestream2_get_bytes_left(gb) > 0) { |
||||||
|
while (bytestream2_get_bytes_left(gb) > 0) { |
||||||
|
opcode = bytestream2_get_byte(gb); |
||||||
|
high = opcode >= 0x20; |
||||||
|
if (high) |
||||||
|
break; |
||||||
|
if (opcode) |
||||||
|
break; |
||||||
|
opcode = bytestream2_get_byte(gb); |
||||||
|
if (opcode < 0xF8) { |
||||||
|
opcode += 32; |
||||||
|
break; |
||||||
|
} |
||||||
|
i = opcode - 0xF8; |
||||||
|
if (i) { |
||||||
|
len = 256; |
||||||
|
do { |
||||||
|
len *= 2; |
||||||
|
--i; |
||||||
|
} while (i); |
||||||
|
} else { |
||||||
|
len = 280; |
||||||
|
} |
||||||
|
do { |
||||||
|
bytestream2_put_le32(pb, bytestream2_get_le32(gb)); |
||||||
|
bytestream2_put_le32(pb, bytestream2_get_le32(gb)); |
||||||
|
len -= 8; |
||||||
|
} while (len && bytestream2_get_bytes_left(gb) > 0); |
||||||
|
} |
||||||
|
|
||||||
|
if (!high) { |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
--opcode; |
||||||
|
} while (opcode && bytestream2_get_bytes_left(gb) > 0); |
||||||
|
|
||||||
|
while (bytestream2_get_bytes_left(gb) > 0) { |
||||||
|
GetByteContext gbc; |
||||||
|
|
||||||
|
opcode = bytestream2_get_byte(gb); |
||||||
|
if (opcode >= 0x20) |
||||||
|
break; |
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
|
||||||
|
pos = -(opcode | 32 * bytestream2_get_byte(gb)) - 1; |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(gb)); |
||||||
|
} |
||||||
|
} |
||||||
|
high = 0; |
||||||
|
if (opcode < 0x40) |
||||||
|
break; |
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
pos = (-((opcode & 0x1F) | 32 * bytestream2_get_byte(gb)) - 1); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos, SEEK_SET); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
len = (opcode >> 5) - 1; |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
--len; |
||||||
|
} while (len && bytestream2_get_bytes_left(&gbc) > 0); |
||||||
|
} |
||||||
|
len = opcode & 0x1F; |
||||||
|
if (!len) { |
||||||
|
if (!bytestream2_peek_byte(gb)) { |
||||||
|
do { |
||||||
|
bytestream2_skip(gb, 1); |
||||||
|
len += 255; |
||||||
|
} while (!bytestream2_peek_byte(gb) && bytestream2_get_bytes_left(gb) > 0); |
||||||
|
} |
||||||
|
len += bytestream2_get_byte(gb) + 31; |
||||||
|
} |
||||||
|
pos = -bytestream2_get_byte(gb); |
||||||
|
bytestream2_init(&gbc, pb->buffer_start, pb->buffer_end - pb->buffer_start); |
||||||
|
bytestream2_seek(&gbc, bytestream2_tell_p(pb) + pos - (bytestream2_get_byte(gb) << 8), SEEK_SET); |
||||||
|
if (bytestream2_tell_p(pb) == bytestream2_tell(&gbc)) |
||||||
|
break; |
||||||
|
if (len < 5 || bytestream2_tell_p(pb) - bytestream2_tell(&gbc) < 4) { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
} else { |
||||||
|
bytestream2_put_le32(pb, bytestream2_get_le32(&gbc)); |
||||||
|
len--; |
||||||
|
} |
||||||
|
do { |
||||||
|
bytestream2_put_byte(pb, bytestream2_get_byte(&gbc)); |
||||||
|
len--; |
||||||
|
} while (len && bytestream2_get_bytes_left(&gbc) > 0); |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int decode_frame(AVCodecContext *avctx, void *data, |
||||||
|
int *got_frame, AVPacket *avpkt) |
||||||
|
{ |
||||||
|
FMVCContext *s = avctx->priv_data; |
||||||
|
GetByteContext *gb = &s->gb; |
||||||
|
PutByteContext *pb = &s->pb; |
||||||
|
AVFrame *frame = data; |
||||||
|
int ret, y, x; |
||||||
|
|
||||||
|
if ((ret = ff_get_buffer(avctx, frame, 0)) < 0) |
||||||
|
return ret; |
||||||
|
|
||||||
|
bytestream2_init(gb, avpkt->data, avpkt->size); |
||||||
|
bytestream2_skip(gb, 2); |
||||||
|
|
||||||
|
frame->key_frame = !!bytestream2_get_le16(gb); |
||||||
|
frame->pict_type = frame->key_frame ? AV_PICTURE_TYPE_I : AV_PICTURE_TYPE_P; |
||||||
|
|
||||||
|
if (frame->key_frame) { |
||||||
|
const uint8_t *src; |
||||||
|
unsigned type, size; |
||||||
|
uint8_t *dst; |
||||||
|
|
||||||
|
type = bytestream2_get_le16(gb); |
||||||
|
size = bytestream2_get_le16(gb); |
||||||
|
if (size > bytestream2_get_bytes_left(gb)) |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
|
||||||
|
bytestream2_init_writer(pb, s->buffer, s->buffer_size); |
||||||
|
if (type == 1) { |
||||||
|
decode_type1(gb, pb); |
||||||
|
} else if (type == 2){ |
||||||
|
decode_type2(gb, pb); |
||||||
|
} else { |
||||||
|
avpriv_report_missing_feature(avctx, "Compression type %d", type); |
||||||
|
return AVERROR_PATCHWELCOME; |
||||||
|
} |
||||||
|
|
||||||
|
src = s->buffer; |
||||||
|
dst = frame->data[0] + (avctx->height - 1) * frame->linesize[0]; |
||||||
|
for (y = 0; y < avctx->height; y++) { |
||||||
|
memcpy(dst, src, avctx->width * s->bpp); |
||||||
|
dst -= frame->linesize[0]; |
||||||
|
src += avctx->width * s->bpp; |
||||||
|
} |
||||||
|
} else { |
||||||
|
unsigned block, nb_blocks; |
||||||
|
int type, k, l; |
||||||
|
uint8_t *ssrc, *ddst; |
||||||
|
const uint32_t *src; |
||||||
|
uint32_t *dst; |
||||||
|
|
||||||
|
for (block = 0; block < s->nb_blocks; block++) |
||||||
|
s->blocks[block].xor = 0; |
||||||
|
|
||||||
|
nb_blocks = bytestream2_get_le16(gb); |
||||||
|
if (nb_blocks > s->nb_blocks) |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
|
||||||
|
bytestream2_init_writer(pb, s->pbuffer, s->pbuffer_size); |
||||||
|
|
||||||
|
type = bytestream2_get_le16(gb); |
||||||
|
for (block = 0; block < nb_blocks; block++) { |
||||||
|
unsigned size, offset; |
||||||
|
int start = 0; |
||||||
|
|
||||||
|
offset = bytestream2_get_le16(gb); |
||||||
|
if (offset > s->nb_blocks) |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
|
||||||
|
size = bytestream2_get_le16(gb); |
||||||
|
if (size > bytestream2_get_bytes_left(gb)) |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
|
||||||
|
start = bytestream2_tell_p(pb); |
||||||
|
if (type == 1) { |
||||||
|
decode_type1(gb, pb); |
||||||
|
} else if (type == 2){ |
||||||
|
decode_type2(gb, pb); |
||||||
|
} else { |
||||||
|
avpriv_report_missing_feature(avctx, "Compression type %d", type); |
||||||
|
return AVERROR_PATCHWELCOME; |
||||||
|
} |
||||||
|
|
||||||
|
if (s->blocks[offset].size * 4 != bytestream2_tell_p(pb) - start) |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
|
||||||
|
s->blocks[offset].xor = 1; |
||||||
|
} |
||||||
|
|
||||||
|
src = (const uint32_t *)s->pbuffer; |
||||||
|
dst = (uint32_t *)s->buffer; |
||||||
|
|
||||||
|
for (block = 0, y = 0; y < s->yb; y++) { |
||||||
|
int block_h = s->blocks[block].h; |
||||||
|
uint32_t *rect = dst; |
||||||
|
|
||||||
|
for (x = 0; x < s->xb; x++) { |
||||||
|
int block_w = s->blocks[block].w; |
||||||
|
uint32_t *row = dst; |
||||||
|
|
||||||
|
block_h = s->blocks[block].h; |
||||||
|
if (s->blocks[block].xor) { |
||||||
|
for (k = 0; k < block_h; k++) { |
||||||
|
uint32_t *column = dst; |
||||||
|
for (l = 0; l < block_w; l++) |
||||||
|
*dst++ ^= *src++; |
||||||
|
dst = &column[s->stride]; |
||||||
|
} |
||||||
|
} |
||||||
|
dst = &row[block_w]; |
||||||
|
++block; |
||||||
|
} |
||||||
|
dst = &rect[block_h * s->stride]; |
||||||
|
} |
||||||
|
|
||||||
|
ssrc = s->buffer; |
||||||
|
ddst = frame->data[0] + (avctx->height - 1) * frame->linesize[0]; |
||||||
|
for (y = 0; y < avctx->height; y++) { |
||||||
|
memcpy(ddst, ssrc, avctx->width * s->bpp); |
||||||
|
ddst -= frame->linesize[0]; |
||||||
|
ssrc += avctx->width * s->bpp; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
*got_frame = 1; |
||||||
|
|
||||||
|
return avpkt->size; |
||||||
|
} |
||||||
|
|
||||||
|
static av_cold int decode_init(AVCodecContext *avctx) |
||||||
|
{ |
||||||
|
FMVCContext *s = avctx->priv_data; |
||||||
|
int i, j, m, block = 0, h = BLOCK_HEIGHT, w = BLOCK_WIDTH; |
||||||
|
|
||||||
|
switch (avctx->bits_per_coded_sample) { |
||||||
|
case 16: |
||||||
|
avctx->pix_fmt = AV_PIX_FMT_RGB555; |
||||||
|
break; |
||||||
|
case 24: |
||||||
|
avctx->pix_fmt = AV_PIX_FMT_BGR24; |
||||||
|
break; |
||||||
|
case 32: |
||||||
|
avctx->pix_fmt = AV_PIX_FMT_BGRA; |
||||||
|
break; |
||||||
|
default: |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Unsupported bitdepth %i\n", |
||||||
|
avctx->bits_per_coded_sample); |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
} |
||||||
|
|
||||||
|
s->stride = (avctx->width * avctx->bits_per_coded_sample + 31) / 32; |
||||||
|
s->xb = s->stride / BLOCK_WIDTH; |
||||||
|
m = s->stride % BLOCK_WIDTH; |
||||||
|
if (m) { |
||||||
|
if (m < 37) { |
||||||
|
w = m + BLOCK_WIDTH; |
||||||
|
} else { |
||||||
|
w = m; |
||||||
|
s->xb++; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
s->yb = avctx->height / BLOCK_HEIGHT; |
||||||
|
m = avctx->height % BLOCK_HEIGHT; |
||||||
|
if (m) { |
||||||
|
if (m < 49) { |
||||||
|
h = m + BLOCK_HEIGHT; |
||||||
|
} else { |
||||||
|
h = m; |
||||||
|
s->yb++; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
s->nb_blocks = s->xb * s->yb; |
||||||
|
if (!s->nb_blocks) |
||||||
|
return AVERROR_INVALIDDATA; |
||||||
|
s->blocks = av_mallocz(s->nb_blocks * sizeof(*s->blocks)); |
||||||
|
if (!s->blocks) |
||||||
|
return AVERROR(ENOMEM); |
||||||
|
|
||||||
|
for (i = 0; i < s->yb; i++) { |
||||||
|
for (j = 0; j < s->xb; j++) { |
||||||
|
if (i != (s->yb - 1) || j != (s->xb - 1)) { |
||||||
|
if (i == s->yb - 1) { |
||||||
|
s->blocks[block].w = BLOCK_WIDTH; |
||||||
|
s->blocks[block].h = h; |
||||||
|
s->blocks[block].size = BLOCK_WIDTH * h; |
||||||
|
} else if (j == s->xb - 1) { |
||||||
|
s->blocks[block].w = w; |
||||||
|
s->blocks[block].h = BLOCK_HEIGHT; |
||||||
|
s->blocks[block].size = BLOCK_HEIGHT * w; |
||||||
|
} else { |
||||||
|
s->blocks[block].w = BLOCK_WIDTH; |
||||||
|
s->blocks[block].h = BLOCK_HEIGHT; |
||||||
|
s->blocks[block].size = BLOCK_WIDTH * BLOCK_HEIGHT; |
||||||
|
} |
||||||
|
} else { |
||||||
|
s->blocks[block].w = w; |
||||||
|
s->blocks[block].h = h; |
||||||
|
s->blocks[block].size = w * h; |
||||||
|
} |
||||||
|
block++; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
s->bpp = avctx->bits_per_coded_sample >> 3; |
||||||
|
s->buffer_size = avctx->width * avctx->height * 4; |
||||||
|
s->pbuffer_size = avctx->width * avctx->height * 4; |
||||||
|
s->buffer = av_malloc(s->buffer_size); |
||||||
|
s->pbuffer = av_malloc(s->pbuffer_size); |
||||||
|
if (!s->buffer || !s->pbuffer) |
||||||
|
return AVERROR(ENOMEM); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static av_cold int decode_close(AVCodecContext *avctx) |
||||||
|
{ |
||||||
|
FMVCContext *s = avctx->priv_data; |
||||||
|
|
||||||
|
av_freep(&s->buffer); |
||||||
|
av_freep(&s->pbuffer); |
||||||
|
av_freep(&s->blocks); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
AVCodec ff_fmvc_decoder = { |
||||||
|
.name = "fmvc", |
||||||
|
.long_name = NULL_IF_CONFIG_SMALL("FM Screen Capture Codec"), |
||||||
|
.type = AVMEDIA_TYPE_VIDEO, |
||||||
|
.id = AV_CODEC_ID_FMVC, |
||||||
|
.priv_data_size = sizeof(FMVCContext), |
||||||
|
.init = decode_init, |
||||||
|
.close = decode_close, |
||||||
|
.decode = decode_frame, |
||||||
|
.capabilities = AV_CODEC_CAP_DR1, |
||||||
|
.caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | |
||||||
|
FF_CODEC_CAP_INIT_CLEANUP, |
||||||
|
}; |
Loading…
Reference in new issue