1 /*
2  * FWSE demuxer
3  * Copyright (c) 2020 Paul B Mahol
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/channel_layout.h"
23 #include "libavutil/intreadwrite.h"
24 #include "avformat.h"
25 #include "internal.h"
26 #include "pcm.h"
27 
fwse_probe(const AVProbeData *p)28 static int fwse_probe(const AVProbeData *p)
29 {
30     if (AV_RL32(p->buf) != MKTAG('F','W','S','E'))
31         return 0;
32     if (AV_RL32(p->buf+4) != 2 && AV_RL32(p->buf+4) != 3)
33         return 0;
34     if (AV_RL32(p->buf+16) != 1 && AV_RL32(p->buf+16) != 2)
35         return 0;
36 
37     return AVPROBE_SCORE_MAX / 4 * 3;
38 }
39 
fwse_read_header(AVFormatContext *s)40 static int fwse_read_header(AVFormatContext *s)
41 {
42     unsigned start_offset, version;
43     int channels;
44     AVIOContext *pb = s->pb;
45     AVCodecParameters *par;
46     AVStream *st;
47 
48     avio_skip(pb, 4);
49     version = avio_rl32(pb);
50     if (version != 2 && version != 3)
51         return AVERROR_INVALIDDATA;
52     avio_skip(pb, 4);
53     start_offset = avio_rl32(pb);
54 
55     st = avformat_new_stream(s, NULL);
56     if (!st)
57         return AVERROR(ENOMEM);
58 
59     par              = st->codecpar;
60     par->codec_type  = AVMEDIA_TYPE_AUDIO;
61     par->codec_id    = AV_CODEC_ID_ADPCM_IMA_MTF;
62     par->format      = AV_SAMPLE_FMT_S16;
63     channels         = avio_rl32(pb);
64     if (channels != 1 && channels != 2)
65         return AVERROR_INVALIDDATA;
66     av_channel_layout_default(&par->ch_layout, channels);
67     st->duration = avio_rl32(pb);
68     par->sample_rate = avio_rl32(pb);
69     if (par->sample_rate <= 0 || par->sample_rate > INT_MAX)
70         return AVERROR_INVALIDDATA;
71 
72     par->block_align = 1;
73     avio_skip(pb, start_offset - avio_tell(pb));
74 
75     avpriv_set_pts_info(st, 64, 1, par->sample_rate);
76 
77     return 0;
78 }
79 
80 const AVInputFormat ff_fwse_demuxer = {
81     .name           = "fwse",
82     .long_name      = NULL_IF_CONFIG_SMALL("Capcom's MT Framework sound"),
83     .read_probe     = fwse_probe,
84     .read_header    = fwse_read_header,
85     .read_packet    = ff_pcm_read_packet,
86     .extensions     = "fwse",
87 };
88