mirror of
https://github.com/pineappleEA/pineapple-src.git
synced 2024-12-09 01:38:25 -05:00
198 lines
5.8 KiB
C
Executable File
198 lines
5.8 KiB
C
Executable File
/*
|
|
* Simon & Schuster Interactive VAG (de)muxer
|
|
*
|
|
* Copyright (C) 2020 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 "avformat.h"
|
|
#include "internal.h"
|
|
#include "rawenc.h"
|
|
#include "libavutil/intreadwrite.h"
|
|
|
|
#define KVAG_TAG MKTAG('K', 'V', 'A', 'G')
|
|
#define KVAG_HEADER_SIZE 14
|
|
#define KVAG_MAX_READ_SIZE 4096
|
|
|
|
typedef struct KVAGHeader {
|
|
uint32_t magic;
|
|
uint32_t data_size;
|
|
uint32_t sample_rate;
|
|
uint16_t stereo;
|
|
} KVAGHeader;
|
|
|
|
#if CONFIG_KVAG_DEMUXER
|
|
static int kvag_probe(const AVProbeData *p)
|
|
{
|
|
if (AV_RL32(p->buf) != KVAG_TAG)
|
|
return 0;
|
|
|
|
return AVPROBE_SCORE_EXTENSION + 1;
|
|
}
|
|
|
|
static int kvag_read_header(AVFormatContext *s)
|
|
{
|
|
int ret;
|
|
AVStream *st;
|
|
KVAGHeader hdr;
|
|
AVCodecParameters *par;
|
|
uint8_t buf[KVAG_HEADER_SIZE];
|
|
|
|
if (!(st = avformat_new_stream(s, NULL)))
|
|
return AVERROR(ENOMEM);
|
|
|
|
if ((ret = avio_read(s->pb, buf, KVAG_HEADER_SIZE)) < 0)
|
|
return ret;
|
|
else if (ret != KVAG_HEADER_SIZE)
|
|
return AVERROR(EIO);
|
|
|
|
hdr.magic = AV_RL32(buf + 0);
|
|
hdr.data_size = AV_RL32(buf + 4);
|
|
hdr.sample_rate = AV_RL32(buf + 8);
|
|
hdr.stereo = AV_RL16(buf + 12);
|
|
|
|
par = st->codecpar;
|
|
par->codec_type = AVMEDIA_TYPE_AUDIO;
|
|
par->codec_id = AV_CODEC_ID_ADPCM_IMA_SSI;
|
|
par->format = AV_SAMPLE_FMT_S16;
|
|
|
|
if (hdr.stereo) {
|
|
par->channel_layout = AV_CH_LAYOUT_STEREO;
|
|
par->channels = 2;
|
|
} else {
|
|
par->channel_layout = AV_CH_LAYOUT_MONO;
|
|
par->channels = 1;
|
|
}
|
|
|
|
par->sample_rate = hdr.sample_rate;
|
|
par->bits_per_coded_sample = 4;
|
|
par->bits_per_raw_sample = 16;
|
|
par->block_align = 1;
|
|
par->bit_rate = par->channels *
|
|
par->sample_rate *
|
|
par->bits_per_coded_sample;
|
|
|
|
avpriv_set_pts_info(st, 64, 1, par->sample_rate);
|
|
st->start_time = 0;
|
|
st->duration = hdr.data_size *
|
|
(8 / par->bits_per_coded_sample) /
|
|
par->channels;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int kvag_read_packet(AVFormatContext *s, AVPacket *pkt)
|
|
{
|
|
int ret;
|
|
AVCodecParameters *par = s->streams[0]->codecpar;
|
|
|
|
if ((ret = av_get_packet(s->pb, pkt, KVAG_MAX_READ_SIZE)) < 0)
|
|
return ret;
|
|
|
|
pkt->flags &= ~AV_PKT_FLAG_CORRUPT;
|
|
pkt->stream_index = 0;
|
|
pkt->duration = ret * (8 / par->bits_per_coded_sample) / par->channels;
|
|
|
|
return 0;
|
|
}
|
|
|
|
AVInputFormat ff_kvag_demuxer = {
|
|
.name = "kvag",
|
|
.long_name = NULL_IF_CONFIG_SMALL("Simon & Schuster Interactive VAG"),
|
|
.read_probe = kvag_probe,
|
|
.read_header = kvag_read_header,
|
|
.read_packet = kvag_read_packet
|
|
};
|
|
#endif
|
|
|
|
#if CONFIG_KVAG_MUXER
|
|
static int kvag_write_init(AVFormatContext *s)
|
|
{
|
|
AVCodecParameters *par;
|
|
|
|
if (s->nb_streams != 1) {
|
|
av_log(s, AV_LOG_ERROR, "KVAG files have exactly one stream\n");
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
par = s->streams[0]->codecpar;
|
|
|
|
if (par->codec_id != AV_CODEC_ID_ADPCM_IMA_SSI) {
|
|
av_log(s, AV_LOG_ERROR, "%s codec not supported\n",
|
|
avcodec_get_name(par->codec_id));
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
if (par->channels > 2) {
|
|
av_log(s, AV_LOG_ERROR, "KVAG files only support up to 2 channels\n");
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL)) {
|
|
av_log(s, AV_LOG_WARNING, "Stream not seekable, unable to write output file\n");
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int kvag_write_header(AVFormatContext *s)
|
|
{
|
|
uint8_t buf[KVAG_HEADER_SIZE];
|
|
AVCodecParameters *par = s->streams[0]->codecpar;
|
|
|
|
AV_WL32(buf + 0, KVAG_TAG);
|
|
AV_WL32(buf + 4, 0); /* Data size, we fix this up later. */
|
|
AV_WL32(buf + 8, par->sample_rate);
|
|
AV_WL16(buf + 12, par->channels == 2);
|
|
|
|
avio_write(s->pb, buf, sizeof(buf));
|
|
return 0;
|
|
}
|
|
|
|
static int kvag_write_trailer(AVFormatContext *s)
|
|
{
|
|
int64_t file_size, data_size;
|
|
|
|
file_size = avio_tell(s->pb);
|
|
data_size = file_size - KVAG_HEADER_SIZE;
|
|
if (data_size < UINT32_MAX) {
|
|
avio_seek(s->pb, 4, SEEK_SET);
|
|
avio_wl32(s->pb, (uint32_t)data_size);
|
|
avio_seek(s->pb, file_size, SEEK_SET);
|
|
} else {
|
|
av_log(s, AV_LOG_WARNING,
|
|
"Filesize %"PRId64" invalid for KVAG, output file will be broken\n",
|
|
file_size);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
AVOutputFormat ff_kvag_muxer = {
|
|
.name = "kvag",
|
|
.long_name = NULL_IF_CONFIG_SMALL("Simon & Schuster Interactive VAG"),
|
|
.extensions = "vag",
|
|
.audio_codec = AV_CODEC_ID_ADPCM_IMA_SSI,
|
|
.video_codec = AV_CODEC_ID_NONE,
|
|
.init = kvag_write_init,
|
|
.write_header = kvag_write_header,
|
|
.write_packet = ff_raw_write_packet,
|
|
.write_trailer = kvag_write_trailer
|
|
};
|
|
#endif
|