1 | /*
|
---|
2 | * Creative Voice File demuxer.
|
---|
3 | * Copyright (c) 2006 Aurelien Jacobs <aurel@gnuage.org>
|
---|
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 St, Fifth Floor, Boston, MA 02110-1301 USA
|
---|
20 | */
|
---|
21 |
|
---|
22 | #include "avformat.h"
|
---|
23 | #include "riff.h" /* for CodecTag */
|
---|
24 | #include "voc.h"
|
---|
25 |
|
---|
26 |
|
---|
27 | typedef enum voc_type {
|
---|
28 | VOC_TYPE_EOF = 0x00,
|
---|
29 | VOC_TYPE_VOICE_DATA = 0x01,
|
---|
30 | VOC_TYPE_VOICE_DATA_CONT = 0x02,
|
---|
31 | VOC_TYPE_SILENCE = 0x03,
|
---|
32 | VOC_TYPE_MARKER = 0x04,
|
---|
33 | VOC_TYPE_ASCII = 0x05,
|
---|
34 | VOC_TYPE_REPETITION_START = 0x06,
|
---|
35 | VOC_TYPE_REPETITION_END = 0x07,
|
---|
36 | VOC_TYPE_EXTENDED = 0x08,
|
---|
37 | VOC_TYPE_NEW_VOICE_DATA = 0x09,
|
---|
38 | } voc_type_t;
|
---|
39 |
|
---|
40 |
|
---|
41 | static const int voc_max_pkt_size = 2048;
|
---|
42 | static const unsigned char voc_magic[] = "Creative Voice File\x1A";
|
---|
43 |
|
---|
44 | static const CodecTag voc_codec_tags[] = {
|
---|
45 | {CODEC_ID_PCM_U8, 0x00},
|
---|
46 | {CODEC_ID_ADPCM_SBPRO_4, 0x01},
|
---|
47 | {CODEC_ID_ADPCM_SBPRO_3, 0x02},
|
---|
48 | {CODEC_ID_ADPCM_SBPRO_2, 0x03},
|
---|
49 | {CODEC_ID_PCM_S16LE, 0x04},
|
---|
50 | {CODEC_ID_PCM_ALAW, 0x06},
|
---|
51 | {CODEC_ID_PCM_MULAW, 0x07},
|
---|
52 | {CODEC_ID_ADPCM_CT, 0x0200},
|
---|
53 | {0, 0},
|
---|
54 | };
|
---|
55 |
|
---|
56 |
|
---|
57 | #ifdef CONFIG_DEMUXERS
|
---|
58 |
|
---|
59 | static int voc_probe(AVProbeData *p)
|
---|
60 | {
|
---|
61 | int version, check;
|
---|
62 |
|
---|
63 | if (p->buf_size < 26)
|
---|
64 | return 0;
|
---|
65 | if (memcmp(p->buf, voc_magic, sizeof(voc_magic) - 1))
|
---|
66 | return 0;
|
---|
67 | version = p->buf[22] | (p->buf[23] << 8);
|
---|
68 | check = p->buf[24] | (p->buf[25] << 8);
|
---|
69 | if (~version + 0x1234 != check)
|
---|
70 | return 10;
|
---|
71 |
|
---|
72 | return AVPROBE_SCORE_MAX;
|
---|
73 | }
|
---|
74 |
|
---|
75 | static int voc_read_header(AVFormatContext *s, AVFormatParameters *ap)
|
---|
76 | {
|
---|
77 | voc_dec_context_t *voc = s->priv_data;
|
---|
78 | ByteIOContext *pb = &s->pb;
|
---|
79 | int header_size;
|
---|
80 | AVStream *st;
|
---|
81 |
|
---|
82 | url_fskip(pb, 20);
|
---|
83 | header_size = get_le16(pb) - 22;
|
---|
84 | if (header_size != 4) {
|
---|
85 | av_log(s, AV_LOG_ERROR, "unkown header size: %d\n", header_size);
|
---|
86 | return AVERROR_NOTSUPP;
|
---|
87 | }
|
---|
88 | url_fskip(pb, header_size);
|
---|
89 | st = av_new_stream(s, 0);
|
---|
90 | if (!st)
|
---|
91 | return AVERROR_NOMEM;
|
---|
92 | st->codec->codec_type = CODEC_TYPE_AUDIO;
|
---|
93 |
|
---|
94 | voc->remaining_size = 0;
|
---|
95 | return 0;
|
---|
96 | }
|
---|
97 |
|
---|
98 | int
|
---|
99 | voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
|
---|
100 | {
|
---|
101 | voc_dec_context_t *voc = s->priv_data;
|
---|
102 | AVCodecContext *dec = st->codec;
|
---|
103 | ByteIOContext *pb = &s->pb;
|
---|
104 | voc_type_t type;
|
---|
105 | int size;
|
---|
106 | int sample_rate = 0;
|
---|
107 | int channels = 1;
|
---|
108 |
|
---|
109 | while (!voc->remaining_size) {
|
---|
110 | type = get_byte(pb);
|
---|
111 | if (type == VOC_TYPE_EOF)
|
---|
112 | return AVERROR_IO;
|
---|
113 | voc->remaining_size = get_le24(pb);
|
---|
114 | max_size -= 4;
|
---|
115 |
|
---|
116 | switch (type) {
|
---|
117 | case VOC_TYPE_VOICE_DATA:
|
---|
118 | dec->sample_rate = 1000000 / (256 - get_byte(pb));
|
---|
119 | if (sample_rate)
|
---|
120 | dec->sample_rate = sample_rate;
|
---|
121 | dec->channels = channels;
|
---|
122 | dec->codec_id = codec_get_id(voc_codec_tags, get_byte(pb));
|
---|
123 | dec->bits_per_sample = av_get_bits_per_sample(dec->codec_id);
|
---|
124 | voc->remaining_size -= 2;
|
---|
125 | max_size -= 2;
|
---|
126 | channels = 1;
|
---|
127 | break;
|
---|
128 |
|
---|
129 | case VOC_TYPE_VOICE_DATA_CONT:
|
---|
130 | break;
|
---|
131 |
|
---|
132 | case VOC_TYPE_EXTENDED:
|
---|
133 | sample_rate = get_le16(pb);
|
---|
134 | get_byte(pb);
|
---|
135 | channels = get_byte(pb) + 1;
|
---|
136 | sample_rate = 256000000 / (channels * (65536 - sample_rate));
|
---|
137 | voc->remaining_size = 0;
|
---|
138 | max_size -= 4;
|
---|
139 | break;
|
---|
140 |
|
---|
141 | case VOC_TYPE_NEW_VOICE_DATA:
|
---|
142 | dec->sample_rate = get_le32(pb);
|
---|
143 | dec->bits_per_sample = get_byte(pb);
|
---|
144 | dec->channels = get_byte(pb);
|
---|
145 | dec->codec_id = codec_get_id(voc_codec_tags, get_le16(pb));
|
---|
146 | url_fskip(pb, 4);
|
---|
147 | voc->remaining_size -= 12;
|
---|
148 | max_size -= 12;
|
---|
149 | break;
|
---|
150 |
|
---|
151 | default:
|
---|
152 | url_fskip(pb, voc->remaining_size);
|
---|
153 | max_size -= voc->remaining_size;
|
---|
154 | voc->remaining_size = 0;
|
---|
155 | break;
|
---|
156 | }
|
---|
157 | }
|
---|
158 |
|
---|
159 | dec->bit_rate = dec->sample_rate * dec->bits_per_sample;
|
---|
160 |
|
---|
161 | if (max_size <= 0)
|
---|
162 | max_size = voc_max_pkt_size;
|
---|
163 | size = FFMIN(voc->remaining_size, max_size);
|
---|
164 | voc->remaining_size -= size;
|
---|
165 | return av_get_packet(pb, pkt, size);
|
---|
166 | }
|
---|
167 |
|
---|
168 | static int voc_read_packet(AVFormatContext *s, AVPacket *pkt)
|
---|
169 | {
|
---|
170 | return voc_get_packet(s, pkt, s->streams[0], 0);
|
---|
171 | }
|
---|
172 |
|
---|
173 | static int voc_read_close(AVFormatContext *s)
|
---|
174 | {
|
---|
175 | return 0;
|
---|
176 | }
|
---|
177 |
|
---|
178 | AVInputFormat voc_demuxer = {
|
---|
179 | "voc",
|
---|
180 | "Creative Voice File format",
|
---|
181 | sizeof(voc_dec_context_t),
|
---|
182 | voc_probe,
|
---|
183 | voc_read_header,
|
---|
184 | voc_read_packet,
|
---|
185 | voc_read_close,
|
---|
186 | };
|
---|
187 |
|
---|
188 | #endif /* CONFIG_DEMUXERS */
|
---|
189 |
|
---|
190 |
|
---|
191 | #ifdef CONFIG_MUXERS
|
---|
192 |
|
---|
193 | typedef struct voc_enc_context {
|
---|
194 | int param_written;
|
---|
195 | } voc_enc_context_t;
|
---|
196 |
|
---|
197 | static int voc_write_header(AVFormatContext *s)
|
---|
198 | {
|
---|
199 | ByteIOContext *pb = &s->pb;
|
---|
200 | const int header_size = 26;
|
---|
201 | const int version = 0x0114;
|
---|
202 |
|
---|
203 | if (s->nb_streams != 1
|
---|
204 | || s->streams[0]->codec->codec_type != CODEC_TYPE_AUDIO)
|
---|
205 | return AVERROR_NOTSUPP;
|
---|
206 |
|
---|
207 | put_buffer(pb, voc_magic, sizeof(voc_magic) - 1);
|
---|
208 | put_le16(pb, header_size);
|
---|
209 | put_le16(pb, version);
|
---|
210 | put_le16(pb, ~version + 0x1234);
|
---|
211 |
|
---|
212 | return 0;
|
---|
213 | }
|
---|
214 |
|
---|
215 | static int voc_write_packet(AVFormatContext *s, AVPacket *pkt)
|
---|
216 | {
|
---|
217 | voc_enc_context_t *voc = s->priv_data;
|
---|
218 | AVCodecContext *enc = s->streams[0]->codec;
|
---|
219 | ByteIOContext *pb = &s->pb;
|
---|
220 |
|
---|
221 | if (!voc->param_written) {
|
---|
222 | int format = codec_get_tag(voc_codec_tags, enc->codec_id);
|
---|
223 |
|
---|
224 | if (format > 0xFF) {
|
---|
225 | put_byte(pb, VOC_TYPE_NEW_VOICE_DATA);
|
---|
226 | put_le24(pb, pkt->size + 12);
|
---|
227 | put_le32(pb, enc->sample_rate);
|
---|
228 | put_byte(pb, enc->bits_per_sample);
|
---|
229 | put_byte(pb, enc->channels);
|
---|
230 | put_le16(pb, format);
|
---|
231 | put_le32(pb, 0);
|
---|
232 | } else {
|
---|
233 | if (s->streams[0]->codec->channels > 1) {
|
---|
234 | put_byte(pb, VOC_TYPE_EXTENDED);
|
---|
235 | put_le24(pb, 4);
|
---|
236 | put_le16(pb, 65536-256000000/(enc->sample_rate*enc->channels));
|
---|
237 | put_byte(pb, format);
|
---|
238 | put_byte(pb, enc->channels - 1);
|
---|
239 | }
|
---|
240 | put_byte(pb, VOC_TYPE_VOICE_DATA);
|
---|
241 | put_le24(pb, pkt->size + 2);
|
---|
242 | put_byte(pb, 256 - 1000000 / enc->sample_rate);
|
---|
243 | put_byte(pb, format);
|
---|
244 | }
|
---|
245 | voc->param_written = 1;
|
---|
246 | } else {
|
---|
247 | put_byte(pb, VOC_TYPE_VOICE_DATA_CONT);
|
---|
248 | put_le24(pb, pkt->size);
|
---|
249 | }
|
---|
250 |
|
---|
251 | put_buffer(pb, pkt->data, pkt->size);
|
---|
252 | return 0;
|
---|
253 | }
|
---|
254 |
|
---|
255 | static int voc_write_trailer(AVFormatContext *s)
|
---|
256 | {
|
---|
257 | put_byte(&s->pb, 0);
|
---|
258 | return 0;
|
---|
259 | }
|
---|
260 |
|
---|
261 | AVOutputFormat voc_muxer = {
|
---|
262 | "voc",
|
---|
263 | "Creative Voice File format",
|
---|
264 | "audio/x-voc",
|
---|
265 | "voc",
|
---|
266 | sizeof(voc_enc_context_t),
|
---|
267 | CODEC_ID_PCM_U8,
|
---|
268 | CODEC_ID_NONE,
|
---|
269 | voc_write_header,
|
---|
270 | voc_write_packet,
|
---|
271 | voc_write_trailer,
|
---|
272 | };
|
---|
273 |
|
---|
274 | #endif /* CONFIG_MUXERS */
|
---|