blob: 73ea7c7d54b8c46d9a3b4d7f43c1dcd85be4e59e
1 | /* |
2 | * ADS/SS2 demuxer |
3 | * Copyright (c) 2015 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 "avformat.h" |
24 | #include "internal.h" |
25 | |
26 | static int ads_probe(AVProbeData *p) |
27 | { |
28 | if (memcmp(p->buf, "SShd", 4) || |
29 | memcmp(p->buf+32, "SSbd", 4)) |
30 | return 0; |
31 | |
32 | return AVPROBE_SCORE_MAX / 3 * 2; |
33 | } |
34 | |
35 | static int ads_read_header(AVFormatContext *s) |
36 | { |
37 | int align, codec, size; |
38 | AVStream *st; |
39 | |
40 | st = avformat_new_stream(s, NULL); |
41 | if (!st) |
42 | return AVERROR(ENOMEM); |
43 | |
44 | avio_skip(s->pb, 8); |
45 | st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; |
46 | codec = avio_rl32(s->pb); |
47 | st->codecpar->sample_rate = avio_rl32(s->pb); |
48 | if (st->codecpar->sample_rate <= 0) |
49 | return AVERROR_INVALIDDATA; |
50 | st->codecpar->channels = avio_rl32(s->pb); |
51 | if (st->codecpar->channels <= 0) |
52 | return AVERROR_INVALIDDATA; |
53 | align = avio_rl32(s->pb); |
54 | if (align <= 0 || align > INT_MAX / st->codecpar->channels) |
55 | return AVERROR_INVALIDDATA; |
56 | |
57 | if (codec == 1) |
58 | st->codecpar->codec_id = AV_CODEC_ID_PCM_S16LE_PLANAR; |
59 | else |
60 | st->codecpar->codec_id = AV_CODEC_ID_ADPCM_PSX; |
61 | |
62 | st->codecpar->block_align = st->codecpar->channels * align; |
63 | avio_skip(s->pb, 12); |
64 | size = avio_rl32(s->pb); |
65 | if (st->codecpar->codec_id == AV_CODEC_ID_ADPCM_PSX) |
66 | st->duration = (size - 0x40) / 16 / st->codecpar->channels * 28; |
67 | avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate); |
68 | |
69 | return 0; |
70 | } |
71 | |
72 | static int ads_read_packet(AVFormatContext *s, AVPacket *pkt) |
73 | { |
74 | AVCodecParameters *par = s->streams[0]->codecpar; |
75 | int ret; |
76 | |
77 | ret = av_get_packet(s->pb, pkt, par->block_align); |
78 | pkt->stream_index = 0; |
79 | return ret; |
80 | } |
81 | |
82 | AVInputFormat ff_ads_demuxer = { |
83 | .name = "ads", |
84 | .long_name = NULL_IF_CONFIG_SMALL("Sony PS2 ADS"), |
85 | .read_probe = ads_probe, |
86 | .read_header = ads_read_header, |
87 | .read_packet = ads_read_packet, |
88 | .extensions = "ads,ss2", |
89 | }; |
90 |