voc: set bits_per_sample correctly (fix mplayer playback)
[ffmpeg.git] / libavformat / voc.c
1 /*
2  * Creative Voice File demuxer.
3  * Copyright (c) 2006  Aurelien Jacobs <aurel@gnuage.org>
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18  */
19
20 #include "avformat.h"
21 #include "avi.h"    /* for CodecTag */
22 #include "voc.h"
23
24
25 typedef enum voc_type {
26     VOC_TYPE_EOF              = 0x00,
27     VOC_TYPE_VOICE_DATA       = 0x01,
28     VOC_TYPE_VOICE_DATA_CONT  = 0x02,
29     VOC_TYPE_SILENCE          = 0x03,
30     VOC_TYPE_MARKER           = 0x04,
31     VOC_TYPE_ASCII            = 0x05,
32     VOC_TYPE_REPETITION_START = 0x06,
33     VOC_TYPE_REPETITION_END   = 0x07,
34     VOC_TYPE_EXTENDED         = 0x08,
35     VOC_TYPE_NEW_VOICE_DATA   = 0x09,
36 } voc_type_t;
37
38
39 static const int voc_max_pkt_size = 2048;
40 static const unsigned char voc_magic[] = "Creative Voice File\x1A";
41
42 static const CodecTag voc_codec_tags[] = {
43     {CODEC_ID_PCM_U8,        0x00},
44     {CODEC_ID_PCM_S16LE,     0x04},
45     {CODEC_ID_PCM_ALAW,      0x06},
46     {CODEC_ID_PCM_MULAW,     0x07},
47     {CODEC_ID_ADPCM_CT,    0x0200},
48     {0, 0},
49 };
50
51
52 #ifdef CONFIG_DEMUXERS
53
54 static int voc_probe(AVProbeData *p)
55 {
56     int version, check;
57
58     if (p->buf_size < 26)
59         return 0;
60     if (memcmp(p->buf, voc_magic, sizeof(voc_magic) - 1))
61         return 0;
62     version = p->buf[22] | (p->buf[23] << 8);
63     check = p->buf[24] | (p->buf[25] << 8);
64     if (~version + 0x1234 != check)
65         return 10;
66
67     return AVPROBE_SCORE_MAX;
68 }
69
70 static int voc_read_header(AVFormatContext *s, AVFormatParameters *ap)
71 {
72     voc_dec_context_t *voc = s->priv_data;
73     ByteIOContext *pb = &s->pb;
74     int header_size;
75     AVStream *st;
76
77     url_fskip(pb, 20);
78     header_size = get_le16(pb) - 22;
79     if (header_size != 4) {
80         av_log(s, AV_LOG_ERROR, "unkown header size: %d\n", header_size);
81         return AVERROR_NOTSUPP;
82     }
83     url_fskip(pb, header_size);
84     st = av_new_stream(s, 0);
85     if (!st)
86         return AVERROR_NOMEM;
87     st->codec->codec_type = CODEC_TYPE_AUDIO;
88
89     voc->remaining_size = 0;
90     return 0;
91 }
92
93 static int voc_get_bps(int codec_id)
94 {
95     switch (codec_id) {
96     case CODEC_ID_PCM_S16LE:
97         return 16;
98     case CODEC_ID_ADPCM_CT:
99         return 4;
100     default:
101         return 8;
102     }
103 }
104
105 int
106 voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
107 {
108     voc_dec_context_t *voc = s->priv_data;
109     AVCodecContext *dec = st->codec;
110     ByteIOContext *pb = &s->pb;
111     voc_type_t type;
112     int size;
113     int sample_rate = 0;
114     int channels = 1;
115
116     while (!voc->remaining_size) {
117         type = get_byte(pb);
118         if (type == VOC_TYPE_EOF)
119             return AVERROR_IO;
120         voc->remaining_size = get_le24(pb);
121         max_size -= 4;
122
123         switch (type) {
124         case VOC_TYPE_VOICE_DATA:
125             dec->sample_rate = 1000000 / (256 - get_byte(pb));
126             if (sample_rate)
127                 dec->sample_rate = sample_rate;
128             dec->channels = channels;
129             dec->codec_id = codec_get_id(voc_codec_tags, get_byte(pb));
130             dec->bits_per_sample = voc_get_bps(dec->codec_id);
131             voc->remaining_size -= 2;
132             max_size -= 2;
133             channels = 1;
134             break;
135
136         case VOC_TYPE_VOICE_DATA_CONT:
137             break;
138
139         case VOC_TYPE_EXTENDED:
140             sample_rate = get_le16(pb);
141             get_byte(pb);
142             channels = get_byte(pb) + 1;
143             sample_rate = 256000000 / (channels * (65536 - sample_rate));
144             voc->remaining_size = 0;
145             max_size -= 4;
146             break;
147
148         case VOC_TYPE_NEW_VOICE_DATA:
149             dec->sample_rate = get_le32(pb);
150             dec->bits_per_sample = get_byte(pb);
151             dec->channels = get_byte(pb);
152             dec->codec_id = codec_get_id(voc_codec_tags, get_le16(pb));
153             url_fskip(pb, 4);
154             voc->remaining_size -= 12;
155             max_size -= 12;
156             break;
157
158         default:
159             url_fskip(pb, voc->remaining_size);
160             max_size -= voc->remaining_size;
161             voc->remaining_size = 0;
162             break;
163         }
164     }
165
166     dec->bit_rate = dec->sample_rate * dec->bits_per_sample;
167
168     if (max_size <= 0)
169         max_size = voc_max_pkt_size;
170     size = FFMIN(voc->remaining_size, max_size);
171     voc->remaining_size -= size;
172     return av_get_packet(pb, pkt, size);
173 }
174
175 static int voc_read_packet(AVFormatContext *s, AVPacket *pkt)
176 {
177     return voc_get_packet(s, pkt, s->streams[0], 0);
178 }
179
180 static int voc_read_close(AVFormatContext *s)
181 {
182     return 0;
183 }
184
185 static AVInputFormat voc_iformat = {
186     "voc",
187     "Creative Voice File format",
188     sizeof(voc_dec_context_t),
189     voc_probe,
190     voc_read_header,
191     voc_read_packet,
192     voc_read_close,
193 };
194
195 #endif /* CONFIG_DEMUXERS */
196
197
198 #ifdef CONFIG_MUXERS
199
200 typedef struct voc_enc_context {
201     int param_written;
202 } voc_enc_context_t;
203
204 static int voc_write_header(AVFormatContext *s)
205 {
206     ByteIOContext *pb = &s->pb;
207     const int header_size = 26;
208     const int version = 0x0114;
209
210     if (s->nb_streams != 1
211         || s->streams[0]->codec->codec_type != CODEC_TYPE_AUDIO)
212         return AVERROR_NOTSUPP;
213
214     put_buffer(pb, voc_magic, sizeof(voc_magic) - 1);
215     put_le16(pb, header_size);
216     put_le16(pb, version);
217     put_le16(pb, ~version + 0x1234);
218
219     return 0;
220 }
221
222 static int voc_write_packet(AVFormatContext *s, AVPacket *pkt)
223 {
224     voc_enc_context_t *voc = s->priv_data;
225     AVCodecContext *enc = s->streams[0]->codec;
226     ByteIOContext *pb = &s->pb;
227
228     if (!voc->param_written) {
229         int format = codec_get_tag(voc_codec_tags, enc->codec_id);
230
231         if (format > 0xFF) {
232             put_byte(pb, VOC_TYPE_NEW_VOICE_DATA);
233             put_le24(pb, pkt->size + 12);
234             put_le32(pb, enc->sample_rate);
235             put_byte(pb, enc->bits_per_sample);
236             put_byte(pb, enc->channels);
237             put_le16(pb, format);
238             put_le32(pb, 0);
239         } else {
240             if (s->streams[0]->codec->channels > 1) {
241                 put_byte(pb, VOC_TYPE_EXTENDED);
242                 put_le24(pb, 4);
243                 put_le16(pb, 65536-256000000/(enc->sample_rate*enc->channels));
244                 put_byte(pb, format);
245                 put_byte(pb, enc->channels - 1);
246             }
247             put_byte(pb, VOC_TYPE_VOICE_DATA);
248             put_le24(pb, pkt->size + 2);
249             put_byte(pb, 256 - 1000000 / enc->sample_rate);
250             put_byte(pb, format);
251         }
252         voc->param_written = 1;
253     } else {
254         put_byte(pb, VOC_TYPE_VOICE_DATA_CONT);
255         put_le24(pb, pkt->size);
256     }
257
258     put_buffer(pb, pkt->data, pkt->size);
259     return 0;
260 }
261
262 static int voc_write_trailer(AVFormatContext *s)
263 {
264     put_byte(&s->pb, 0);
265     return 0;
266 }
267
268 static AVOutputFormat voc_oformat = {
269     "voc",
270     "Creative Voice File format",
271     "audio/x-voc",
272     "voc",
273     sizeof(voc_enc_context_t),
274     CODEC_ID_PCM_U8,
275     CODEC_ID_NONE,
276     voc_write_header,
277     voc_write_packet,
278     voc_write_trailer,
279 };
280
281 #endif /* CONFIG_MUXERS */
282
283
284 int voc_init(void)
285 {
286 #ifdef CONFIG_DEMUXERS
287     av_register_input_format(&voc_iformat);
288 #endif /* CONFIG_DEMUXERS */
289 #ifdef CONFIG_MUXERS
290     av_register_output_format(&voc_oformat);
291 #endif /* CONFIG_MUXERS */
292     return 0;
293 }