Change license headers to say 'FFmpeg' instead of 'this program/this library'
[ffmpeg.git] / libavformat / voc.c
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 */