blob: 3c7e8ac84b4f30f18dd554540099b5b05c571b86
1 | /* |
2 | * RAW PCM demuxers |
3 | * Copyright (c) 2002 Fabrice Bellard |
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 "avformat.h" |
23 | #include "internal.h" |
24 | #include "pcm.h" |
25 | #include "libavutil/log.h" |
26 | #include "libavutil/opt.h" |
27 | #include "libavutil/avassert.h" |
28 | |
29 | typedef struct PCMAudioDemuxerContext { |
30 | AVClass *class; |
31 | int sample_rate; |
32 | int channels; |
33 | } PCMAudioDemuxerContext; |
34 | |
35 | static int pcm_read_header(AVFormatContext *s) |
36 | { |
37 | PCMAudioDemuxerContext *s1 = s->priv_data; |
38 | AVStream *st; |
39 | uint8_t *mime_type = NULL; |
40 | |
41 | st = avformat_new_stream(s, NULL); |
42 | if (!st) |
43 | return AVERROR(ENOMEM); |
44 | |
45 | |
46 | st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; |
47 | st->codecpar->codec_id = s->iformat->raw_codec_id; |
48 | st->codecpar->sample_rate = s1->sample_rate; |
49 | st->codecpar->channels = s1->channels; |
50 | |
51 | av_opt_get(s->pb, "mime_type", AV_OPT_SEARCH_CHILDREN, &mime_type); |
52 | if (mime_type && s->iformat->mime_type) { |
53 | int rate = 0, channels = 0; |
54 | size_t len = strlen(s->iformat->mime_type); |
55 | if (!strncmp(s->iformat->mime_type, mime_type, len)) { |
56 | uint8_t *options = mime_type + len; |
57 | len = strlen(mime_type); |
58 | while (options < mime_type + len) { |
59 | options = strstr(options, ";"); |
60 | if (!options++) |
61 | break; |
62 | if (!rate) |
63 | sscanf(options, " rate=%d", &rate); |
64 | if (!channels) |
65 | sscanf(options, " channels=%d", &channels); |
66 | } |
67 | if (rate <= 0) { |
68 | av_log(s, AV_LOG_ERROR, |
69 | "Invalid sample_rate found in mime_type \"%s\"\n", |
70 | mime_type); |
71 | return AVERROR_INVALIDDATA; |
72 | } |
73 | st->codecpar->sample_rate = rate; |
74 | if (channels > 0) |
75 | st->codecpar->channels = channels; |
76 | } |
77 | } |
78 | |
79 | st->codecpar->bits_per_coded_sample = |
80 | av_get_bits_per_sample(st->codecpar->codec_id); |
81 | |
82 | av_assert0(st->codecpar->bits_per_coded_sample > 0); |
83 | |
84 | st->codecpar->block_align = |
85 | st->codecpar->bits_per_coded_sample * st->codecpar->channels / 8; |
86 | |
87 | avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate); |
88 | return 0; |
89 | } |
90 | |
91 | static const AVOption pcm_options[] = { |
92 | { "sample_rate", "", offsetof(PCMAudioDemuxerContext, sample_rate), AV_OPT_TYPE_INT, {.i64 = 44100}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM }, |
93 | { "channels", "", offsetof(PCMAudioDemuxerContext, channels), AV_OPT_TYPE_INT, {.i64 = 1}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM }, |
94 | { NULL }, |
95 | }; |
96 | |
97 | #define PCMDEF(name_, long_name_, ext, codec, ...) \ |
98 | static const AVClass name_ ## _demuxer_class = { \ |
99 | .class_name = #name_ " demuxer", \ |
100 | .item_name = av_default_item_name, \ |
101 | .option = pcm_options, \ |
102 | .version = LIBAVUTIL_VERSION_INT, \ |
103 | }; \ |
104 | AVInputFormat ff_pcm_ ## name_ ## _demuxer = { \ |
105 | .name = #name_, \ |
106 | .long_name = NULL_IF_CONFIG_SMALL(long_name_), \ |
107 | .priv_data_size = sizeof(PCMAudioDemuxerContext), \ |
108 | .read_header = pcm_read_header, \ |
109 | .read_packet = ff_pcm_read_packet, \ |
110 | .read_seek = ff_pcm_read_seek, \ |
111 | .flags = AVFMT_GENERIC_INDEX, \ |
112 | .extensions = ext, \ |
113 | .raw_codec_id = codec, \ |
114 | .priv_class = &name_ ## _demuxer_class, \ |
115 | __VA_ARGS__ \ |
116 | }; |
117 | |
118 | PCMDEF(f64be, "PCM 64-bit floating-point big-endian", |
119 | NULL, AV_CODEC_ID_PCM_F64BE) |
120 | |
121 | PCMDEF(f64le, "PCM 64-bit floating-point little-endian", |
122 | NULL, AV_CODEC_ID_PCM_F64LE) |
123 | |
124 | PCMDEF(f32be, "PCM 32-bit floating-point big-endian", |
125 | NULL, AV_CODEC_ID_PCM_F32BE) |
126 | |
127 | PCMDEF(f32le, "PCM 32-bit floating-point little-endian", |
128 | NULL, AV_CODEC_ID_PCM_F32LE) |
129 | |
130 | PCMDEF(s32be, "PCM signed 32-bit big-endian", |
131 | NULL, AV_CODEC_ID_PCM_S32BE) |
132 | |
133 | PCMDEF(s32le, "PCM signed 32-bit little-endian", |
134 | NULL, AV_CODEC_ID_PCM_S32LE) |
135 | |
136 | PCMDEF(s24be, "PCM signed 24-bit big-endian", |
137 | NULL, AV_CODEC_ID_PCM_S24BE) |
138 | |
139 | PCMDEF(s24le, "PCM signed 24-bit little-endian", |
140 | NULL, AV_CODEC_ID_PCM_S24LE) |
141 | |
142 | PCMDEF(s16be, "PCM signed 16-bit big-endian", |
143 | AV_NE("sw", NULL), AV_CODEC_ID_PCM_S16BE) |
144 | |
145 | PCMDEF(s16le, "PCM signed 16-bit little-endian", |
146 | AV_NE(NULL, "sw"), AV_CODEC_ID_PCM_S16LE, .mime_type = "audio/L16",) |
147 | |
148 | PCMDEF(s8, "PCM signed 8-bit", |
149 | "sb", AV_CODEC_ID_PCM_S8) |
150 | |
151 | PCMDEF(u32be, "PCM unsigned 32-bit big-endian", |
152 | NULL, AV_CODEC_ID_PCM_U32BE) |
153 | |
154 | PCMDEF(u32le, "PCM unsigned 32-bit little-endian", |
155 | NULL, AV_CODEC_ID_PCM_U32LE) |
156 | |
157 | PCMDEF(u24be, "PCM unsigned 24-bit big-endian", |
158 | NULL, AV_CODEC_ID_PCM_U24BE) |
159 | |
160 | PCMDEF(u24le, "PCM unsigned 24-bit little-endian", |
161 | NULL, AV_CODEC_ID_PCM_U24LE) |
162 | |
163 | PCMDEF(u16be, "PCM unsigned 16-bit big-endian", |
164 | AV_NE("uw", NULL), AV_CODEC_ID_PCM_U16BE) |
165 | |
166 | PCMDEF(u16le, "PCM unsigned 16-bit little-endian", |
167 | AV_NE(NULL, "uw"), AV_CODEC_ID_PCM_U16LE) |
168 | |
169 | PCMDEF(u8, "PCM unsigned 8-bit", |
170 | "ub", AV_CODEC_ID_PCM_U8) |
171 | |
172 | PCMDEF(alaw, "PCM A-law", |
173 | "al", AV_CODEC_ID_PCM_ALAW) |
174 | |
175 | PCMDEF(mulaw, "PCM mu-law", |
176 | "ul", AV_CODEC_ID_PCM_MULAW) |
177 | |
178 | static const AVOption sln_options[] = { |
179 | { "sample_rate", "", offsetof(PCMAudioDemuxerContext, sample_rate), AV_OPT_TYPE_INT, {.i64 = 8000}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM }, |
180 | { "channels", "", offsetof(PCMAudioDemuxerContext, channels), AV_OPT_TYPE_INT, {.i64 = 1}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM }, |
181 | { NULL }, |
182 | }; |
183 | |
184 | static const AVClass sln_demuxer_class = { |
185 | .class_name = "sln demuxer", |
186 | .item_name = av_default_item_name, |
187 | .option = sln_options, |
188 | .version = LIBAVUTIL_VERSION_INT, |
189 | }; |
190 | |
191 | AVInputFormat ff_sln_demuxer = { |
192 | .name = "sln", |
193 | .long_name = NULL_IF_CONFIG_SMALL("Asterisk raw pcm"), |
194 | .priv_data_size = sizeof(PCMAudioDemuxerContext), |
195 | .read_header = pcm_read_header, |
196 | .read_packet = ff_pcm_read_packet, |
197 | .read_seek = ff_pcm_read_seek, |
198 | .flags = AVFMT_GENERIC_INDEX, |
199 | .extensions = "sln", |
200 | .raw_codec_id = AV_CODEC_ID_PCM_S16LE, |
201 | .priv_class = &sln_demuxer_class, |
202 | }; |
203 |