| /* |
| * FSB demuxer |
| * Copyright (c) 2015 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/avassert.h" |
| #include "libavutil/intreadwrite.h" |
| #include "avformat.h" |
| #include "avio.h" |
| #include "internal.h" |
| |
| static int fsb_probe(AVProbeData *p) |
| { |
| if (memcmp(p->buf, "FSB", 3) || p->buf[3] - '0' < 1 || p->buf[3] - '0' > 5) |
| return 0; |
| if (AV_RL32(p->buf + 4) != 1) |
| return 0; |
| return AVPROBE_SCORE_MAX; |
| } |
| |
| static int fsb_read_header(AVFormatContext *s) |
| { |
| AVIOContext *pb = s->pb; |
| unsigned format, version, c; |
| int64_t offset; |
| AVCodecParameters *par; |
| AVStream *st = avformat_new_stream(s, NULL); |
| |
| avio_skip(pb, 3); // "FSB" |
| version = avio_r8(pb) - '0'; |
| if (version != 4 && version != 3) { |
| avpriv_request_sample(s, "version %d", version); |
| return AVERROR_PATCHWELCOME; |
| } |
| |
| avio_skip(pb, 4); |
| |
| if (!st) |
| return AVERROR(ENOMEM); |
| par = st->codecpar; |
| par->codec_type = AVMEDIA_TYPE_AUDIO; |
| par->codec_tag = 0; |
| |
| if (version == 3) { |
| offset = avio_rl32(pb) + 0x18; |
| avio_skip(pb, 44); |
| st->duration = avio_rl32(pb); |
| avio_skip(pb, 12); |
| format = avio_rl32(pb); |
| par->sample_rate = avio_rl32(pb); |
| if (par->sample_rate <= 0) |
| return AVERROR_INVALIDDATA; |
| avio_skip(pb, 6); |
| par->channels = avio_rl16(pb); |
| if (!par->channels) |
| return AVERROR_INVALIDDATA; |
| |
| if (format & 0x00000100) { |
| par->codec_id = AV_CODEC_ID_PCM_S16LE; |
| par->block_align = 4096 * par->channels; |
| } else if (format & 0x00400000) { |
| par->bits_per_coded_sample = 4; |
| par->codec_id = AV_CODEC_ID_ADPCM_IMA_WAV; |
| par->block_align = 36 * par->channels; |
| } else if (format & 0x00800000) { |
| par->codec_id = AV_CODEC_ID_ADPCM_PSX; |
| par->block_align = 16 * par->channels; |
| } else if (format & 0x02000000) { |
| par->codec_id = AV_CODEC_ID_ADPCM_THP; |
| par->block_align = 8 * par->channels; |
| if (par->channels > INT_MAX / 32) |
| return AVERROR_INVALIDDATA; |
| ff_alloc_extradata(par, 32 * par->channels); |
| if (!par->extradata) |
| return AVERROR(ENOMEM); |
| avio_seek(pb, 0x68, SEEK_SET); |
| for (c = 0; c < par->channels; c++) { |
| avio_read(pb, par->extradata + 32 * c, 32); |
| avio_skip(pb, 14); |
| } |
| } else { |
| avpriv_request_sample(s, "format 0x%X", format); |
| return AVERROR_PATCHWELCOME; |
| } |
| } else if (version == 4) { |
| offset = avio_rl32(pb) + 0x30; |
| avio_skip(pb, 80); |
| st->duration = avio_rl32(pb); |
| |
| format = avio_rb32(pb); |
| switch(format) { |
| case 0x40001001: |
| case 0x00001005: |
| case 0x40001081: |
| case 0x40200001: |
| par->codec_id = AV_CODEC_ID_XMA2; |
| break; |
| case 0x40000802: |
| par->codec_id = AV_CODEC_ID_ADPCM_THP; |
| break; |
| default: |
| avpriv_request_sample(s, "format 0x%X", format); |
| return AVERROR_PATCHWELCOME; |
| } |
| |
| par->sample_rate = avio_rl32(pb); |
| if (par->sample_rate <= 0) |
| return AVERROR_INVALIDDATA; |
| avio_skip(pb, 6); |
| |
| par->channels = avio_rl16(pb); |
| if (!par->channels) |
| return AVERROR_INVALIDDATA; |
| |
| switch (par->codec_id) { |
| case AV_CODEC_ID_XMA2: |
| ff_alloc_extradata(par, 34); |
| if (!par->extradata) |
| return AVERROR(ENOMEM); |
| memset(par->extradata, 0, 34); |
| par->block_align = 2048; |
| break; |
| case AV_CODEC_ID_ADPCM_THP: |
| if (par->channels > INT_MAX / 32) |
| return AVERROR_INVALIDDATA; |
| ff_alloc_extradata(par, 32 * par->channels); |
| if (!par->extradata) |
| return AVERROR(ENOMEM); |
| avio_seek(pb, 0x80, SEEK_SET); |
| for (c = 0; c < par->channels; c++) { |
| avio_read(pb, par->extradata + 32 * c, 32); |
| avio_skip(pb, 14); |
| } |
| par->block_align = 8 * par->channels; |
| break; |
| } |
| } else { |
| av_assert0(0); |
| } |
| |
| avio_skip(pb, offset - avio_tell(pb)); |
| s->internal->data_offset = avio_tell(pb); |
| |
| avpriv_set_pts_info(st, 64, 1, par->sample_rate); |
| |
| return 0; |
| } |
| |
| static int fsb_read_packet(AVFormatContext *s, AVPacket *pkt) |
| { |
| AVCodecParameters *par = s->streams[0]->codecpar; |
| int64_t pos; |
| int ret; |
| |
| if (avio_feof(s->pb)) |
| return AVERROR_EOF; |
| |
| pos = avio_tell(s->pb); |
| if (par->codec_id == AV_CODEC_ID_ADPCM_THP && |
| par->channels > 1) { |
| int i, ch; |
| |
| ret = av_new_packet(pkt, par->block_align); |
| if (ret < 0) |
| return ret; |
| for (i = 0; i < 4; i++) { |
| for (ch = 0; ch < par->channels; ch++) { |
| pkt->data[ch * 8 + i * 2 + 0] = avio_r8(s->pb); |
| pkt->data[ch * 8 + i * 2 + 1] = avio_r8(s->pb); |
| } |
| } |
| ret = 0; |
| } else { |
| ret = av_get_packet(s->pb, pkt, par->block_align); |
| } |
| |
| if (par->codec_id == AV_CODEC_ID_XMA2 && pkt->size >= 1) |
| pkt->duration = (pkt->data[0] >> 2) * 512; |
| |
| pkt->pos = pos; |
| pkt->stream_index = 0; |
| |
| return ret; |
| } |
| |
| AVInputFormat ff_fsb_demuxer = { |
| .name = "fsb", |
| .long_name = NULL_IF_CONFIG_SMALL("FMOD Sample Bank"), |
| .read_probe = fsb_probe, |
| .read_header = fsb_read_header, |
| .read_packet = fsb_read_packet, |
| .extensions = "fsb", |
| .flags = AVFMT_GENERIC_INDEX, |
| }; |
