From 7de4a16508cb9bd2d40855f7114dea14329ef62a Mon Sep 17 00:00:00 2001 From: Paul B Mahol Date: Tue, 27 Mar 2012 19:27:53 +0000 Subject: [PATCH] PAF demuxer and decoder Signed-off-by: Paul B Mahol --- Changelog | 1 + doc/general.texi | 4 + libavcodec/Makefile | 2 + libavcodec/allcodecs.c | 2 + libavcodec/avcodec.h | 2 + libavcodec/paf.c | 454 +++++++++++++++++++++++++++++++++++++++ libavcodec/paf.h | 28 +++ libavcodec/version.h | 2 +- libavformat/Makefile | 1 + libavformat/allformats.c | 1 + libavformat/paf.c | 257 ++++++++++++++++++++++ libavformat/version.h | 2 +- 12 files changed, 754 insertions(+), 2 deletions(-) create mode 100644 libavcodec/paf.c create mode 100644 libavcodec/paf.h create mode 100644 libavformat/paf.c diff --git a/Changelog b/Changelog index 169b86486b..6bd20dd73e 100644 --- a/Changelog +++ b/Changelog @@ -19,6 +19,7 @@ version next: - LucasArts SMUSH playback support - SAMI demuxer and decoder - RealText demuxer and decoder +- Heart Of Darkness PAF playback support version 0.11: diff --git a/doc/general.texi b/doc/general.texi index 4ca4945f8f..d1925c097b 100644 --- a/doc/general.texi +++ b/doc/general.texi @@ -127,6 +127,8 @@ library: @item American Laser Games MM @tab @tab X @tab Multimedia format used in games like Mad Dog McCree. @item 3GPP AMR @tab X @tab X +@item Amazing Studio Packed Animation File @tab @tab X + @tab Multimedia format used in game Heart Of Darkness. @item Apple HTTP Live Streaming @tab @tab X @item Artworx Data Format @tab @tab X @item ASF @tab X @tab X @@ -441,6 +443,7 @@ following image formats are supported: @item 8SVX fibonacci @tab @tab X @item A64 multicolor @tab X @tab @tab Creates video suitable to be played on a commodore 64 (multicolor mode). +@item Amazing Studio PAF Video @tab @tab X @item American Laser Games MM @tab @tab X @tab Used in games like Mad Dog McCree. @item AMV Video @tab X @tab X @@ -735,6 +738,7 @@ following image formats are supported: @tab encoding supported through external library libopencore-amrnb @item AMR-WB @tab E @tab X @tab encoding supported through external library libvo-amrwbenc +@item Amazing Studio PAF Audio @tab @tab X @item Apple lossless audio @tab X @tab X @tab QuickTime fourcc 'alac' @item Atrac 1 @tab @tab X diff --git a/libavcodec/Makefile b/libavcodec/Makefile index 3cc0124a93..d50a40e522 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -337,6 +337,8 @@ OBJS-$(CONFIG_NELLYMOSER_DECODER) += nellymoserdec.o nellymoser.o OBJS-$(CONFIG_NELLYMOSER_ENCODER) += nellymoserenc.o nellymoser.o \ audio_frame_queue.o OBJS-$(CONFIG_NUV_DECODER) += nuv.o rtjpeg.o +OBJS-$(CONFIG_PAF_VIDEO_DECODER) += paf.o +OBJS-$(CONFIG_PAF_AUDIO_DECODER) += paf.o OBJS-$(CONFIG_PAM_DECODER) += pnmdec.o pnm.o OBJS-$(CONFIG_PAM_ENCODER) += pamenc.o pnm.o OBJS-$(CONFIG_PBM_DECODER) += pnmdec.o pnm.o diff --git a/libavcodec/allcodecs.c b/libavcodec/allcodecs.c index e7c11039e1..ef75b08f42 100644 --- a/libavcodec/allcodecs.c +++ b/libavcodec/allcodecs.c @@ -176,6 +176,7 @@ void avcodec_register_all(void) REGISTER_DECODER (MSZH, mszh); REGISTER_DECODER (MXPEG, mxpeg); REGISTER_DECODER (NUV, nuv); + REGISTER_DECODER (PAF_VIDEO, paf_video); REGISTER_ENCDEC (PAM, pam); REGISTER_ENCDEC (PBM, pbm); REGISTER_ENCDEC (PCX, pcx); @@ -307,6 +308,7 @@ void avcodec_register_all(void) REGISTER_DECODER (MPC7, mpc7); REGISTER_DECODER (MPC8, mpc8); REGISTER_ENCDEC (NELLYMOSER, nellymoser); + REGISTER_DECODER (PAF_AUDIO, paf_audio); REGISTER_DECODER (QCELP, qcelp); REGISTER_DECODER (QDM2, qdm2); REGISTER_ENCDEC (RA_144, ra_144); diff --git a/libavcodec/avcodec.h b/libavcodec/avcodec.h index 9748ba62af..96ce223567 100644 --- a/libavcodec/avcodec.h +++ b/libavcodec/avcodec.h @@ -271,6 +271,7 @@ enum CodecID { CODEC_ID_V408 = MKBETAG('V','4','0','8'), CODEC_ID_YUV4 = MKBETAG('Y','U','V','4'), CODEC_ID_SANM = MKBETAG('S','A','N','M'), + CODEC_ID_PAF_VIDEO = MKBETAG('P','A','F','V'), /* various PCM "codecs" */ CODEC_ID_FIRST_AUDIO = 0x10000, ///< A dummy id pointing at the start of audio codecs @@ -416,6 +417,7 @@ enum CodecID { CODEC_ID_8SVX_RAW = MKBETAG('8','S','V','X'), CODEC_ID_SONIC = MKBETAG('S','O','N','C'), CODEC_ID_SONIC_LS = MKBETAG('S','O','N','L'), + CODEC_ID_PAF_AUDIO = MKBETAG('P','A','F','A'), /* subtitle codecs */ CODEC_ID_FIRST_SUBTITLE = 0x17000, ///< A dummy ID pointing at the start of subtitle codecs. diff --git a/libavcodec/paf.c b/libavcodec/paf.c new file mode 100644 index 0000000000..ff5db838bf --- /dev/null +++ b/libavcodec/paf.c @@ -0,0 +1,454 @@ +/* + * Packed Animation File video and audio decoder + * Copyright (c) 2012 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 "libavutil/intreadwrite.h" +#include "libavcodec/dsputil.h" +#include "libavcodec/paf.h" +#include "bytestream.h" +#include "avcodec.h" + +static const uint8_t block_sequences[16][8] = +{ + { 0, 0, 0, 0, 0, 0, 0, 0 }, + { 2, 0, 0, 0, 0, 0, 0, 0 }, + { 5, 7, 0, 0, 0, 0, 0, 0 }, + { 5, 0, 0, 0, 0, 0, 0, 0 }, + { 6, 0, 0, 0, 0, 0, 0, 0 }, + { 5, 7, 5, 7, 0, 0, 0, 0 }, + { 5, 7, 5, 0, 0, 0, 0, 0 }, + { 5, 7, 6, 0, 0, 0, 0, 0 }, + { 5, 5, 0, 0, 0, 0, 0, 0 }, + { 3, 0, 0, 0, 0, 0, 0, 0 }, + { 6, 6, 0, 0, 0, 0, 0, 0 }, + { 2, 4, 0, 0, 0, 0, 0, 0 }, + { 2, 4, 5, 7, 0, 0, 0, 0 }, + { 2, 4, 5, 0, 0, 0, 0, 0 }, + { 2, 4, 6, 0, 0, 0, 0, 0 }, + { 2, 4, 5, 7, 5, 7, 0, 0 } +}; + +typedef struct PAFVideoDecContext { + AVFrame pic; + GetByteContext gb; + + int current_frame; + uint8_t *frame[4]; + int frame_size; + int video_size; + + uint8_t *opcodes; +} PAFVideoDecContext; + +static av_cold int paf_vid_init(AVCodecContext *avctx) +{ + PAFVideoDecContext *c = avctx->priv_data; + int i; + + if (avctx->height & 3 || avctx->width & 3) { + av_log(avctx, AV_LOG_ERROR, "width and height must be multiplies of 4\n"); + return AVERROR_INVALIDDATA; + } + + avctx->pix_fmt = PIX_FMT_PAL8; + + avcodec_get_frame_defaults(&c->pic); + c->frame_size = FFALIGN(avctx->height, 256) * avctx->width; + c->video_size = avctx->height * avctx->width; + for (i = 0; i < 4; i++) { + c->frame[i] = av_mallocz(c->frame_size); + if (!c->frame[i]) + return AVERROR(ENOMEM); + } + + return 0; +} + +static int get_video_page_offset(AVCodecContext *avctx, uint8_t a, uint8_t b) +{ + int x, y; + + x = b & 0x7F; + y = ((a & 0x3F) << 1) | (b >> 7 & 1); + + return y * 2 * avctx->width + x * 2; +} + +static void copy4h(AVCodecContext *avctx, uint8_t *dst) +{ + PAFVideoDecContext *c = avctx->priv_data; + int i; + + for (i = 0; i < 4; i++) { + bytestream2_get_buffer(&c->gb, dst, 4); + dst += avctx->width; + } +} + +static void copy_color_mask(AVCodecContext *avctx, uint8_t mask, uint8_t *dst, uint8_t color) +{ + int i; + + for (i = 0; i < 4; i++) { + if ((mask >> 4) & (1 << (3 - i))) + dst[i] = color; + if ((mask & 15) & (1 << (3 - i))) + dst[avctx->width + i] = color; + } +} + +static void copy_src_mask(AVCodecContext *avctx, uint8_t mask, uint8_t *dst, const uint8_t *src) +{ + int i; + + for (i = 0; i < 4; i++) { + if ((mask >> 4) & (1 << (3 - i))) + dst[i] = src[i]; + if ((mask & 15) & (1 << (3 - i))) + dst[avctx->width + i] = src[avctx->width + i]; + } +} + +static int decode_0(AVCodecContext *avctx, uint8_t code, uint8_t *pkt) +{ + PAFVideoDecContext *c = avctx->priv_data; + uint32_t opcode_size, offset; + uint8_t *dst, *dend, mask = 0, color = 0, a, b, p; + const uint8_t *src, *send, *opcodes; + int i, j, x = 0; + + i = bytestream2_get_byte(&c->gb); + if (i) { + if (code & 0x10) { + int align; + + align = bytestream2_tell(&c->gb) & 3; + if (align) + bytestream2_skip(&c->gb, 4 - align); + } + do { + a = bytestream2_get_byte(&c->gb); + b = bytestream2_get_byte(&c->gb); + p = (a & 0xC0) >> 6; + dst = c->frame[p] + get_video_page_offset(avctx, a, b); + dend = c->frame[p] + c->frame_size; + offset = (b & 0x7F) * 2; + j = bytestream2_get_le16(&c->gb) + offset; + + do { + offset++; + if (dst + 3 * avctx->width + 4 > dend) + return AVERROR_INVALIDDATA; + copy4h(avctx, dst); + if ((offset & 0x3F) == 0) + dst += avctx->width * 3; + dst += 4; + } while (offset < j); + } while (--i); + } + + dst = c->frame[c->current_frame]; + do { + a = bytestream2_get_byte(&c->gb); + b = bytestream2_get_byte(&c->gb); + p = (a & 0xC0) >> 6; + src = c->frame[p] + get_video_page_offset(avctx, a, b); + send = c->frame[p] + c->frame_size; + if (src + 3 * avctx->width + 4 > send) + return AVERROR_INVALIDDATA; + copy_block4(dst, src, avctx->width, avctx->width, 4); + i++; + if ((i & 0x3F) == 0) + dst += avctx->width * 3; + dst += 4; + } while (i < c->video_size / 16); + + opcode_size = bytestream2_get_le16(&c->gb); + bytestream2_skip(&c->gb, 2); + + if (bytestream2_get_bytes_left(&c->gb) < opcode_size) + return AVERROR_INVALIDDATA; + + opcodes = pkt + bytestream2_tell(&c->gb); + bytestream2_skipu(&c->gb, opcode_size); + + dst = c->frame[c->current_frame]; + + for (i = 0; i < avctx->height; i += 4, dst += avctx->width * 3) { + for (j = 0; j < avctx->width; j += 4, dst += 4) { + int opcode, k = 0; + + if (x > opcode_size) + return AVERROR_INVALIDDATA; + if (j & 4) { + opcode = opcodes[x] & 15; + x++; + } else { + opcode = opcodes[x] >> 4; + } + + while (block_sequences[opcode][k]) { + + offset = avctx->width * 2; + code = block_sequences[opcode][k++]; + + switch (code) { + case 2: + offset = 0; + case 3: + color = bytestream2_get_byte(&c->gb); + case 4: + mask = bytestream2_get_byte(&c->gb); + copy_color_mask(avctx, mask, dst + offset, color); + break; + case 5: + offset = 0; + case 6: + a = bytestream2_get_byte(&c->gb); + b = bytestream2_get_byte(&c->gb); + p = (a & 0xC0) >> 6; + src = c->frame[p] + get_video_page_offset(avctx, a, b); + send = c->frame[p] + c->frame_size; + case 7: + if (src + offset + avctx->width + 4 > send) + return AVERROR_INVALIDDATA; + mask = bytestream2_get_byte(&c->gb); + copy_src_mask(avctx, mask, dst + offset, src + offset); + break; + } + } + } + } + + return 0; +} + +static int paf_vid_decode(AVCodecContext *avctx, void *data, + int *data_size, AVPacket *pkt) +{ + PAFVideoDecContext *c = avctx->priv_data; + uint8_t code, *dst, *src, *end; + int i, frame, ret; + + if (c->pic.data[0]) + avctx->release_buffer(avctx, &c->pic); + + c->pic.reference = 0; + if ((ret = avctx->get_buffer(avctx, &c->pic)) < 0) + return ret; + + bytestream2_init(&c->gb, pkt->data, pkt->size); + + code = bytestream2_get_byte(&c->gb); + if (code & 0x20) { + for (i = 0; i < 4; i++) + memset(c->frame[i], 0, c->frame_size); + + memset(c->pic.data[1], 0, AVPALETTE_SIZE); + c->current_frame = 0; + c->pic.key_frame = 1; + c->pic.pict_type = AV_PICTURE_TYPE_I; + } else { + c->pic.key_frame = 0; + c->pic.pict_type = AV_PICTURE_TYPE_P; + } + + if (code & 0x40) { + uint32_t *out = (uint32_t *)c->pic.data[1]; + int index, count; + + index = bytestream2_get_byte(&c->gb); + count = bytestream2_get_byte(&c->gb) + 1; + + if (index + count > AVPALETTE_SIZE) + return AVERROR_INVALIDDATA; + if (bytestream2_get_bytes_left(&c->gb) < 3 * AVPALETTE_SIZE) + return AVERROR_INVALIDDATA; + + out += index; + for (i = 0; i < count; i++) { + unsigned r, g, b; + + r = bytestream2_get_byteu(&c->gb); + r = r << 2 | r >> 4; + g = bytestream2_get_byteu(&c->gb); + g = g << 2 | g >> 4; + b = bytestream2_get_byteu(&c->gb); + b = b << 2 | b >> 4; + *out++ = 0xFFU << 24 | r << 16 | g << 8 | b; + } + c->pic.palette_has_changed = 1; + } + + switch (code & 0x0F) { + case 0: + if ((ret = decode_0(avctx, code, pkt->data)) < 0) + return ret; + break; + case 1: + dst = c->frame[c->current_frame]; + bytestream2_skip(&c->gb, 2); + if (bytestream2_get_bytes_left(&c->gb) < c->video_size) + return AVERROR_INVALIDDATA; + bytestream2_get_bufferu(&c->gb, dst, c->video_size); + break; + case 2: + frame = bytestream2_get_byte(&c->gb); + if (frame > 3) + return AVERROR_INVALIDDATA; + if (frame != c->current_frame) + memcpy(c->frame[c->current_frame], c->frame[frame], c->frame_size); + break; + case 4: + dst = c->frame[c->current_frame]; + end = dst + c->video_size; + + bytestream2_skip(&c->gb, 2); + + while (dst < end) { + int8_t code; + int count; + + if (bytestream2_get_bytes_left(&c->gb) < 2) + return AVERROR_INVALIDDATA; + + code = bytestream2_get_byteu(&c->gb); + count = FFABS(code) + 1; + + if (dst + count > end) + return AVERROR_INVALIDDATA; + if (code < 0) + memset(dst, bytestream2_get_byteu(&c->gb), count); + else + bytestream2_get_buffer(&c->gb, dst, count); + dst += count; + } + break; + default: + av_log_ask_for_sample(avctx, "unknown/invalid code\n"); + return AVERROR_INVALIDDATA; + } + + dst = c->pic.data[0]; + src = c->frame[c->current_frame]; + for (i = 0; i < avctx->height; i++) { + memcpy(dst, src, avctx->width); + dst += c->pic.linesize[0]; + src += avctx->width; + } + + c->current_frame = (c->current_frame + 1) & 3; + + *data_size = sizeof(AVFrame); + *(AVFrame *)data = c->pic; + + return pkt->size; +} + +static av_cold int paf_vid_close(AVCodecContext *avctx) +{ + PAFVideoDecContext *c = avctx->priv_data; + int i; + + if (c->pic.data[0]) + avctx->release_buffer(avctx, &c->pic); + + for (i = 0; i < 4; i++) + av_freep(&c->frame[i]); + + return 0; +} + +typedef struct PAFAudioDecContext { + AVFrame frame; +} PAFAudioDecContext; + +static av_cold int paf_aud_init(AVCodecContext *avctx) +{ + PAFAudioDecContext *c = avctx->priv_data; + + if (avctx->channels != 2) { + av_log(avctx, AV_LOG_ERROR, "invalid number of channels\n"); + return AVERROR_INVALIDDATA; + } + + avcodec_get_frame_defaults(&c->frame); + avctx->channel_layout = AV_CH_LAYOUT_STEREO; + avctx->coded_frame = &c->frame; + avctx->sample_fmt = AV_SAMPLE_FMT_S16; + + return 0; +} + +static int paf_aud_decode(AVCodecContext *avctx, void *data, + int *got_frame_ptr, AVPacket *pkt) +{ + PAFAudioDecContext *c = avctx->priv_data; + uint8_t *buf = pkt->data; + int16_t *output_samples; + const uint8_t *t; + int frames, ret, i, j, k; + + frames = pkt->size / PAF_SOUND_FRAME_SIZE; + if (frames < 1) + return AVERROR_INVALIDDATA; + + c->frame.nb_samples = PAF_SOUND_SAMPLES * frames; + if ((ret = avctx->get_buffer(avctx, &c->frame)) < 0) + return ret; + + output_samples = (int16_t *)c->frame.data[0]; + for (i = 0; i < frames; i++) { + t = buf + 256 * sizeof(uint16_t); + for (j = 0; j < PAF_SOUND_SAMPLES; j++) { + for (k = 0; k < 2; k++) { + *output_samples++ = AV_RL16(buf + *t++ * 2); + } + } + buf += PAF_SOUND_FRAME_SIZE; + } + + *got_frame_ptr = 1; + *(AVFrame *)data = c->frame; + + return pkt->size; +} + +AVCodec ff_paf_video_decoder = { + .name = "paf_video", + .type = AVMEDIA_TYPE_VIDEO, + .id = CODEC_ID_PAF_VIDEO, + .priv_data_size = sizeof(PAFVideoDecContext), + .init = paf_vid_init, + .close = paf_vid_close, + .decode = paf_vid_decode, + .long_name = NULL_IF_CONFIG_SMALL("Amazing Studio Packed Animation File Video"), +}; + +AVCodec ff_paf_audio_decoder = { + .name = "paf_audio", + .type = AVMEDIA_TYPE_AUDIO, + .id = CODEC_ID_PAF_AUDIO, + .priv_data_size = sizeof(PAFAudioDecContext), + .init = paf_aud_init, + .decode = paf_aud_decode, + .capabilities = CODEC_CAP_DR1, + .long_name = NULL_IF_CONFIG_SMALL("Amazing Studio Packed Animation File Audio"), +}; diff --git a/libavcodec/paf.h b/libavcodec/paf.h new file mode 100644 index 0000000000..ce8245f223 --- /dev/null +++ b/libavcodec/paf.h @@ -0,0 +1,28 @@ +/* + * Packed Animation File decoder/demuxer common code + * Copyright (c) 2012 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 + */ + +#ifndef AVCODEC_PAF_H +#define AVCODEC_PAF_H + +#define PAF_SOUND_SAMPLES 2205 +#define PAF_SOUND_FRAME_SIZE ((256 + PAF_SOUND_SAMPLES) * 2) + +#endif /* AVCODEC_PAF_H */ diff --git a/libavcodec/version.h b/libavcodec/version.h index 5cbd41bac8..64c1dc6b1b 100644 --- a/libavcodec/version.h +++ b/libavcodec/version.h @@ -27,7 +27,7 @@ */ #define LIBAVCODEC_VERSION_MAJOR 54 -#define LIBAVCODEC_VERSION_MINOR 32 +#define LIBAVCODEC_VERSION_MINOR 33 #define LIBAVCODEC_VERSION_MICRO 100 #define LIBAVCODEC_VERSION_INT AV_VERSION_INT(LIBAVCODEC_VERSION_MAJOR, \ diff --git a/libavformat/Makefile b/libavformat/Makefile index 9eaf28acae..54b26c1416 100644 --- a/libavformat/Makefile +++ b/libavformat/Makefile @@ -209,6 +209,7 @@ OBJS-$(CONFIG_OGG_MUXER) += oggenc.o \ vorbiscomment.o OBJS-$(CONFIG_OMA_DEMUXER) += omadec.o pcm.o oma.o OBJS-$(CONFIG_OMA_MUXER) += omaenc.o rawenc.o oma.o id3v2enc.o +OBJS-$(CONFIG_PAF_DEMUXER) += paf.o OBJS-$(CONFIG_PCM_ALAW_DEMUXER) += pcmdec.o pcm.o rawdec.o OBJS-$(CONFIG_PCM_ALAW_MUXER) += pcmenc.o rawenc.o OBJS-$(CONFIG_PCM_F32BE_DEMUXER) += pcmdec.o pcm.o rawdec.o diff --git a/libavformat/allformats.c b/libavformat/allformats.c index 231b0cbf95..7ea4ebb378 100644 --- a/libavformat/allformats.c +++ b/libavformat/allformats.c @@ -173,6 +173,7 @@ void av_register_all(void) REGISTER_DEMUXER (NUV, nuv); REGISTER_MUXDEMUX (OGG, ogg); REGISTER_MUXDEMUX (OMA, oma); + REGISTER_DEMUXER (PAF, paf); REGISTER_MUXDEMUX (PCM_ALAW, pcm_alaw); REGISTER_MUXDEMUX (PCM_MULAW, pcm_mulaw); REGISTER_MUXDEMUX (PCM_F64BE, pcm_f64be); diff --git a/libavformat/paf.c b/libavformat/paf.c new file mode 100644 index 0000000000..5261ecf1ef --- /dev/null +++ b/libavformat/paf.c @@ -0,0 +1,257 @@ +/* + * Packed Animation File demuxer + * Copyright (c) 2012 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 "libavcodec/paf.h" +#include "avformat.h" +#include "internal.h" + +#define MAGIC "Packed Animation File V1.0\n(c) 1992-96 Amazing Studio\x0a\x1a" + +typedef struct { + uint32_t buffer_size; + uint32_t frame_blks; + uint32_t nb_frames; + uint32_t start_offset; + uint32_t preload_count; + uint32_t max_video_blks; + uint32_t max_audio_blks; + + uint32_t current_frame; + uint32_t current_frame_count; + uint32_t current_frame_block; + + uint32_t *blocks_count_table; + uint32_t *frames_offset_table; + uint32_t *blocks_offset_table; + + uint8_t *video_frame; + int video_size; + + uint8_t *audio_frame; + uint8_t *temp_audio_frame; + int audio_size; + + int got_audio; +} PAFDemuxContext; + +static int read_probe(AVProbeData *p) +{ + if ((p->buf_size >= strlen(MAGIC)) && + !memcmp(p->buf, MAGIC, strlen(MAGIC))) + return AVPROBE_SCORE_MAX; + return 0; +} + +static int read_close(AVFormatContext *s) +{ + PAFDemuxContext *p = s->priv_data; + + av_freep(&p->blocks_count_table); + av_freep(&p->frames_offset_table); + av_freep(&p->blocks_offset_table); + av_freep(&p->video_frame); + av_freep(&p->audio_frame); + av_freep(&p->temp_audio_frame); + + return 0; +} + +static void read_table(AVFormatContext *s, uint32_t *table, uint32_t count) +{ + int i; + + for (i = 0; i < count; i++) + table[i] = avio_rl32(s->pb); + + avio_skip(s->pb, 4 * (FFALIGN(count, 512) - count)); +} + +static int read_header(AVFormatContext *s) +{ + PAFDemuxContext *p = s->priv_data; + AVIOContext *pb = s->pb; + AVStream *ast, *vst; + int ret = 0; + + avio_skip(pb, 132); + + vst = avformat_new_stream(s, 0); + if (!vst) + return AVERROR(ENOMEM); + + vst->start_time = 0; + vst->nb_frames = + vst->duration = + p->nb_frames = avio_rl32(pb); + avio_skip(pb, 4); + vst->codec->width = avio_rl32(pb); + vst->codec->height = avio_rl32(pb); + avio_skip(pb, 4); + vst->codec->codec_type = AVMEDIA_TYPE_VIDEO; + vst->codec->codec_tag = 0; + vst->codec->codec_id = CODEC_ID_PAF_VIDEO; + avpriv_set_pts_info(vst, 64, 1, 10); + + ast = avformat_new_stream(s, 0); + if (!ast) + return AVERROR(ENOMEM); + + ast->start_time = 0; + ast->codec->codec_type = AVMEDIA_TYPE_AUDIO; + ast->codec->codec_tag = 0; + ast->codec->codec_id = CODEC_ID_PAF_AUDIO; + ast->codec->channels = 2; + ast->codec->sample_rate = 22050; + avpriv_set_pts_info(ast, 64, 1, 22050); + + p->buffer_size = avio_rl32(pb); + p->preload_count = avio_rl32(pb); + p->frame_blks = avio_rl32(pb); + p->start_offset = avio_rl32(pb); + p->max_video_blks = avio_rl32(pb); + p->max_audio_blks = avio_rl32(pb); + if (p->buffer_size < 175 || + p->max_audio_blks < 2 || + p->max_video_blks < 1 || + p->frame_blks < 1 || + p->nb_frames < 1 || + p->preload_count < 1 || + p->buffer_size > 2048 || + p->max_video_blks > 2048 || + p->max_audio_blks > 2048 || + p->nb_frames > INT_MAX / sizeof(uint32_t) || + p->frame_blks > INT_MAX / sizeof(uint32_t)) + return AVERROR_INVALIDDATA; + + p->blocks_count_table = av_mallocz(p->nb_frames * sizeof(uint32_t)); + p->frames_offset_table = av_mallocz(p->nb_frames * sizeof(uint32_t)); + p->blocks_offset_table = av_mallocz(p->frame_blks * sizeof(uint32_t)); + + p->video_size = p->max_video_blks * p->buffer_size; + p->video_frame = av_mallocz(p->video_size); + + p->audio_size = p->max_audio_blks * p->buffer_size; + p->audio_frame = av_mallocz(p->audio_size); + p->temp_audio_frame = av_mallocz(p->audio_size); + + if (!p->blocks_count_table || + !p->frames_offset_table || + !p->blocks_offset_table || + !p->video_frame || + !p->audio_frame || + !p->temp_audio_frame) { + ret = AVERROR(ENOMEM); + goto fail; + } + + avio_seek(pb, p->buffer_size, SEEK_SET); + + read_table(s, p->blocks_count_table, p->nb_frames); + read_table(s, p->frames_offset_table, p->nb_frames); + read_table(s, p->blocks_offset_table, p->frame_blks); + + p->got_audio = 0; + p->current_frame = 0; + p->current_frame_block = 0; + + avio_seek(pb, p->start_offset, SEEK_SET); + + return 0; + +fail: + read_close(s); + + return ret; +} + +static int read_packet(AVFormatContext *s, AVPacket *pkt) +{ + PAFDemuxContext *p = s->priv_data; + AVIOContext *pb = s->pb; + uint32_t count, offset; + int size, i; + + if (p->current_frame >= p->nb_frames) + return AVERROR_EOF; + + if (url_feof(pb)) + return AVERROR_EOF; + + if (p->got_audio) { + if (av_new_packet(pkt, p->audio_size) < 0) + return AVERROR(ENOMEM); + + memcpy(pkt->data, p->temp_audio_frame, p->audio_size); + pkt->duration = PAF_SOUND_SAMPLES * p->audio_size / PAF_SOUND_FRAME_SIZE; + pkt->flags |= AV_PKT_FLAG_KEY; + pkt->stream_index = 1; + p->got_audio = 0; + return pkt->size; + } + + count = (p->current_frame == 0) ? p->preload_count : p->blocks_count_table[p->current_frame - 1]; + for (i = 0; i < count; i++) { + if (p->current_frame_block >= p->frame_blks) + return AVERROR_INVALIDDATA; + + offset = p->blocks_offset_table[p->current_frame_block] & ~(1 << 31); + if (p->blocks_offset_table[p->current_frame_block] & (1 << 31)) { + if (offset > p->audio_size - p->buffer_size) + return AVERROR_INVALIDDATA; + + avio_read(pb, p->audio_frame + offset, p->buffer_size); + if (offset == (p->max_audio_blks - 2) * p->buffer_size) { + memcpy(p->temp_audio_frame, p->audio_frame, p->audio_size); + p->got_audio = 1; + } + } else { + if (offset > p->video_size - p->buffer_size) + return AVERROR_INVALIDDATA; + + avio_read(pb, p->video_frame + offset, p->buffer_size); + } + p->current_frame_block++; + } + + size = p->video_size - p->frames_offset_table[p->current_frame]; + if (size < 0) + return AVERROR_INVALIDDATA; + + if (av_new_packet(pkt, size) < 0) + return AVERROR(ENOMEM); + + pkt->stream_index = 0; + pkt->duration = 1; + memcpy(pkt->data, p->video_frame + p->frames_offset_table[p->current_frame], size); + p->current_frame++; + + return pkt->size; +} + +AVInputFormat ff_paf_demuxer = { + .name = "paf", + .long_name = NULL_IF_CONFIG_SMALL("Amazing Studio Packed Animation File"), + .priv_data_size = sizeof(PAFDemuxContext), + .read_probe = read_probe, + .read_header = read_header, + .read_packet = read_packet, + .read_close = read_close, +}; diff --git a/libavformat/version.h b/libavformat/version.h index a48e0a4042..acf591aac6 100644 --- a/libavformat/version.h +++ b/libavformat/version.h @@ -30,7 +30,7 @@ #include "libavutil/avutil.h" #define LIBAVFORMAT_VERSION_MAJOR 54 -#define LIBAVFORMAT_VERSION_MINOR 14 +#define LIBAVFORMAT_VERSION_MINOR 15 #define LIBAVFORMAT_VERSION_MICRO 100 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \