ffmpeg/libavformat/argo_cvg.c

418 lines
12 KiB
C
Raw Normal View History

/*
* Argonaut Games CVG (de)muxer
*
* Copyright (C) 2021 Zane van Iperen (zane@zanevaniperen.com)
*
* 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 "config_components.h"
#include "libavutil/avstring.h"
#include "libavutil/channel_layout.h"
#include "avformat.h"
#include "internal.h"
#include "mux.h"
#include "libavutil/opt.h"
#include "libavutil/intreadwrite.h"
/*
* .CVG files are essentially PSX ADPCM wrapped with a size and checksum.
* Found in the PSX versions of the game.
*/
#define ARGO_CVG_HEADER_SIZE 12
#define ARGO_CVG_BLOCK_ALIGN 0x10
#define ARGO_CVG_NB_BLOCKS 32
#define ARGO_CVG_SAMPLES_PER_BLOCK 28
typedef struct ArgoCVGHeader {
uint32_t size; /*< File size -8 (this + trailing checksum) */
uint32_t loop; /*< Loop flag. */
uint32_t reverb; /*< Reverb flag. */
} ArgoCVGHeader;
typedef struct ArgoCVGOverride {
const char *name;
ArgoCVGHeader header;
uint32_t checksum;
int sample_rate;
} ArgoCVGOverride;
typedef struct ArgoCVGDemuxContext {
ArgoCVGHeader header;
uint32_t checksum;
uint32_t num_blocks;
uint32_t blocks_read;
} ArgoCVGDemuxContext;
typedef struct ArgoCVGMuxContext {
const AVClass *class;
int skip_rate_check;
int loop;
int reverb;
uint32_t checksum;
size_t size;
} ArgoCVGMuxContext;
#if CONFIG_ARGO_CVG_DEMUXER
/* "Special" files that are played at a different rate. */
static const ArgoCVGOverride overrides[] = {
{ "CRYS.CVG", { 23592, 0, 1 }, 2495499, 88200 }, /* Beta */
{ "REDCRY88.CVG", { 38280, 0, 1 }, 4134848, 88200 }, /* Beta */
{ "DANLOOP1.CVG", { 54744, 1, 0 }, 5684641, 37800 }, /* Beta */
{ "PICKUP88.CVG", { 12904, 0, 1 }, 1348091, 48000 }, /* Beta */
{ "SELECT1.CVG", { 5080, 0, 1 }, 549987, 44100 }, /* Beta */
};
static int argo_cvg_probe(const AVProbeData *p)
{
ArgoCVGHeader cvg;
/*
* It's almost impossible to detect these files based
* on the header alone. File extension is (unfortunately)
* the best way forward.
*/
if (!av_match_ext(p->filename, "cvg"))
return 0;
if (p->buf_size < ARGO_CVG_HEADER_SIZE)
return 0;
cvg.size = AV_RL32(p->buf + 0);
cvg.loop = AV_RL32(p->buf + 4);
cvg.reverb = AV_RL32(p->buf + 8);
if (cvg.size < 8)
return 0;
if (cvg.loop != 0 && cvg.loop != 1)
return 0;
if (cvg.reverb != 0 && cvg.reverb != 1)
return 0;
return AVPROBE_SCORE_MAX / 4 + 1;
}
static int argo_cvg_read_checksum(AVIOContext *pb, const ArgoCVGHeader *cvg, uint32_t *checksum)
{
int ret;
uint8_t buf[4];
if (!(pb->seekable & AVIO_SEEKABLE_NORMAL)) {
*checksum = 0;
return 0;
}
if ((ret = avio_seek(pb, cvg->size + 4, SEEK_SET)) < 0)
return ret;
/* NB: Not using avio_rl32() because no error checking. */
if ((ret = avio_read(pb, buf, sizeof(buf))) < 0)
return ret;
else if (ret != sizeof(buf))
return AVERROR(EIO);
if ((ret = avio_seek(pb, ARGO_CVG_HEADER_SIZE, SEEK_SET)) < 0)
return ret;
*checksum = AV_RL32(buf);
return 0;
}
static int argo_cvg_read_header(AVFormatContext *s)
{
int ret;
AVStream *st;
AVCodecParameters *par;
uint8_t buf[ARGO_CVG_HEADER_SIZE];
const char *filename = av_basename(s->url);
ArgoCVGDemuxContext *ctx = s->priv_data;
if (!(st = avformat_new_stream(s, NULL)))
return AVERROR(ENOMEM);
if ((ret = avio_read(s->pb, buf, ARGO_CVG_HEADER_SIZE)) < 0)
return ret;
else if (ret != ARGO_CVG_HEADER_SIZE)
return AVERROR(EIO);
ctx->header.size = AV_RL32(buf + 0);
ctx->header.loop = AV_RL32(buf + 4);
ctx->header.reverb = AV_RL32(buf + 8);
if (ctx->header.size < 8)
return AVERROR_INVALIDDATA;
if ((ret = argo_cvg_read_checksum(s->pb, &ctx->header, &ctx->checksum)) < 0)
return ret;
if ((ret = av_dict_set_int(&st->metadata, "loop", ctx->header.loop, 0)) < 0)
return ret;
if ((ret = av_dict_set_int(&st->metadata, "reverb", ctx->header.reverb, 0)) < 0)
return ret;
if ((ret = av_dict_set_int(&st->metadata, "checksum", ctx->checksum, 0)) < 0)
return ret;
par = st->codecpar;
par->codec_type = AVMEDIA_TYPE_AUDIO;
par->codec_id = AV_CODEC_ID_ADPCM_PSX;
par->sample_rate = 22050;
for (size_t i = 0; i < FF_ARRAY_ELEMS(overrides); i++) {
const ArgoCVGOverride *ovr = overrides + i;
if (ovr->header.size != ctx->header.size ||
ovr->header.loop != ctx->header.loop ||
ovr->header.reverb != ctx->header.reverb ||
ovr->checksum != ctx->checksum ||
av_strcasecmp(filename, ovr->name) != 0)
continue;
av_log(s, AV_LOG_TRACE, "found override, name = %s\n", ovr->name);
par->sample_rate = ovr->sample_rate;
break;
}
par->ch_layout = (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO;
par->bits_per_coded_sample = 4;
par->block_align = ARGO_CVG_BLOCK_ALIGN;
par->bit_rate = par->sample_rate * par->bits_per_coded_sample;
ctx->num_blocks = (ctx->header.size - 8) / ARGO_CVG_BLOCK_ALIGN;
av_log(s, AV_LOG_TRACE, "num blocks = %u\n", ctx->num_blocks);
avpriv_set_pts_info(st, 64, 1, par->sample_rate);
st->start_time = 0;
st->duration = ctx->num_blocks * ARGO_CVG_SAMPLES_PER_BLOCK;
st->nb_frames = ctx->num_blocks;
return 0;
}
static int argo_cvg_read_packet(AVFormatContext *s, AVPacket *pkt)
{
int ret;
AVStream *st = s->streams[0];
ArgoCVGDemuxContext *ctx = s->priv_data;
if (ctx->blocks_read >= ctx->num_blocks)
return AVERROR_EOF;
ret = av_get_packet(s->pb, pkt, st->codecpar->block_align *
FFMIN(ARGO_CVG_NB_BLOCKS, ctx->num_blocks - ctx->blocks_read));
if (ret < 0)
return ret;
if (ret % st->codecpar->block_align != 0)
return AVERROR_INVALIDDATA;
pkt->stream_index = 0;
pkt->duration = ARGO_CVG_SAMPLES_PER_BLOCK * (ret / st->codecpar->block_align);
pkt->pts = ctx->blocks_read * ARGO_CVG_SAMPLES_PER_BLOCK;
pkt->flags &= ~AV_PKT_FLAG_CORRUPT;
ctx->blocks_read += ret / st->codecpar->block_align;
return 0;
}
static int argo_cvg_seek(AVFormatContext *s, int stream_index,
int64_t pts, int flags)
{
int64_t ret;
ArgoCVGDemuxContext *ctx = s->priv_data;
if (pts != 0 || stream_index != 0)
return AVERROR(EINVAL);
if ((ret = avio_seek(s->pb, ARGO_CVG_HEADER_SIZE, SEEK_SET)) < 0)
return ret;
ctx->blocks_read = 0;
return 0;
}
const AVInputFormat ff_argo_cvg_demuxer = {
.name = "argo_cvg",
.long_name = NULL_IF_CONFIG_SMALL("Argonaut Games CVG"),
.priv_data_size = sizeof(ArgoCVGDemuxContext),
.read_probe = argo_cvg_probe,
.read_header = argo_cvg_read_header,
.read_packet = argo_cvg_read_packet,
.read_seek = argo_cvg_seek,
};
#endif
#if CONFIG_ARGO_CVG_MUXER
static int argo_cvg_write_init(AVFormatContext *s)
{
ArgoCVGMuxContext *ctx = s->priv_data;
const AVCodecParameters *par;
if (s->nb_streams != 1) {
av_log(s, AV_LOG_ERROR, "CVG files have exactly one stream\n");
return AVERROR(EINVAL);
}
par = s->streams[0]->codecpar;
if (par->codec_id != AV_CODEC_ID_ADPCM_PSX) {
av_log(s, AV_LOG_ERROR, "%s codec not supported\n",
avcodec_get_name(par->codec_id));
return AVERROR(EINVAL);
}
if (par->ch_layout.nb_channels != 1) {
av_log(s, AV_LOG_ERROR, "CVG files only support 1 channel\n");
return AVERROR(EINVAL);
}
if (par->block_align != ARGO_CVG_BLOCK_ALIGN)
return AVERROR(EINVAL);
if (!ctx->skip_rate_check && par->sample_rate != 22050) {
av_log(s, AV_LOG_ERROR, "Sample rate must be 22050\n");
return AVERROR(EINVAL);
}
if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL)) {
av_log(s, AV_LOG_ERROR, "Stream not seekable, unable to write output file\n");
return AVERROR(EINVAL);
}
return 0;
}
static int argo_cvg_write_header(AVFormatContext *s)
{
ArgoCVGMuxContext *ctx = s->priv_data;
avio_wl32(s->pb, 0); /* Size, fixed later. */
avio_wl32(s->pb, !!ctx->loop);
avio_wl32(s->pb, !!ctx->reverb);
ctx->checksum = !!ctx->loop + !!ctx->reverb;
ctx->size = 8;
return 0;
}
static int argo_cvg_write_packet(AVFormatContext *s, AVPacket *pkt)
{
ArgoCVGMuxContext *ctx = s->priv_data;
AVCodecParameters *par = s->streams[0]->codecpar;
if (pkt->size % par->block_align != 0)
return AVERROR_INVALIDDATA;
avio_write(s->pb, pkt->data, pkt->size);
ctx->size += pkt->size;
if (ctx->size > UINT32_MAX)
return AVERROR_INVALIDDATA;
for (int i = 0; i < pkt->size; i++)
ctx->checksum += pkt->data[i];
return 0;
}
static int argo_cvg_write_trailer(AVFormatContext *s)
{
ArgoCVGMuxContext *ctx = s->priv_data;
int64_t ret;
ctx->checksum += (ctx->size & 255)
+ ((ctx->size>> 8) & 255)
+ ((ctx->size>>16) & 255)
+ (ctx->size>>24);
av_log(s, AV_LOG_TRACE, "size = %zu\n", ctx->size);
av_log(s, AV_LOG_TRACE, "checksum = %u\n", ctx->checksum);
avio_wl32(s->pb, ctx->checksum);
if ((ret = avio_seek(s->pb, 0, SEEK_SET)) < 0)
return ret;
avio_wl32(s->pb, (uint32_t)ctx->size);
return 0;
}
static const AVOption argo_cvg_options[] = {
{
.name = "skip_rate_check",
.help = "skip sample rate check",
.offset = offsetof(ArgoCVGMuxContext, skip_rate_check),
.type = AV_OPT_TYPE_BOOL,
.default_val = {.i64 = 0},
.min = 0,
.max = 1,
.flags = AV_OPT_FLAG_ENCODING_PARAM
},
{
.name = "loop",
.help = "set loop flag",
.offset = offsetof(ArgoCVGMuxContext, loop),
.type = AV_OPT_TYPE_BOOL,
.default_val = {.i64 = 0},
.min = 0,
.max = 1,
.flags = AV_OPT_FLAG_ENCODING_PARAM
},
{
.name = "reverb",
.help = "set reverb flag",
.offset = offsetof(ArgoCVGMuxContext, reverb),
.type = AV_OPT_TYPE_BOOL,
.default_val = {.i64 = 1},
.min = 0,
.max = 1,
.flags = AV_OPT_FLAG_ENCODING_PARAM
},
{ NULL }
};
static const AVClass argo_cvg_muxer_class = {
.class_name = "argo_cvg_muxer",
.item_name = av_default_item_name,
.option = argo_cvg_options,
.version = LIBAVUTIL_VERSION_INT
};
const FFOutputFormat ff_argo_cvg_muxer = {
.p.name = "argo_cvg",
.p.long_name = NULL_IF_CONFIG_SMALL("Argonaut Games CVG"),
.p.extensions = "cvg",
.p.audio_codec = AV_CODEC_ID_ADPCM_PSX,
.p.video_codec = AV_CODEC_ID_NONE,
.p.priv_class = &argo_cvg_muxer_class,
.init = argo_cvg_write_init,
.write_header = argo_cvg_write_header,
.write_packet = argo_cvg_write_packet,
.write_trailer = argo_cvg_write_trailer,
.priv_data_size = sizeof(ArgoCVGMuxContext),
};
#endif