use intreadwrite functions in vocdec
[ffmpeg.git] / libavformat / vocdec.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 "voc.h"
23
24
25
26
27 static int voc_probe(AVProbeData *p)
28 {
29     int version, check;
30
31     if (memcmp(p->buf, voc_magic, sizeof(voc_magic) - 1))
32         return 0;
33     version = AV_RL16(p->buf + 22);
34     check = AV_RL16(p->buf + 24);
35     if (~version + 0x1234 != check)
36         return 10;
37
38     return AVPROBE_SCORE_MAX;
39 }
40
41 static int voc_read_header(AVFormatContext *s, AVFormatParameters *ap)
42 {
43     voc_dec_context_t *voc = s->priv_data;
44     ByteIOContext *pb = &s->pb;
45     int header_size;
46     AVStream *st;
47
48     url_fskip(pb, 20);
49     header_size = get_le16(pb) - 22;
50     if (header_size != 4) {
51         av_log(s, AV_LOG_ERROR, "unknown header size: %d\n", header_size);
52         return AVERROR_NOTSUPP;
53     }
54     url_fskip(pb, header_size);
55     st = av_new_stream(s, 0);
56     if (!st)
57         return AVERROR_NOMEM;
58     st->codec->codec_type = CODEC_TYPE_AUDIO;
59
60     voc->remaining_size = 0;
61     return 0;
62 }
63
64 int
65 voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
66 {
67     voc_dec_context_t *voc = s->priv_data;
68     AVCodecContext *dec = st->codec;
69     ByteIOContext *pb = &s->pb;
70     voc_type_t type;
71     int size;
72     int sample_rate = 0;
73     int channels = 1;
74
75     while (!voc->remaining_size) {
76         type = get_byte(pb);
77         if (type == VOC_TYPE_EOF)
78             return AVERROR_IO;
79         voc->remaining_size = get_le24(pb);
80         max_size -= 4;
81
82         switch (type) {
83         case VOC_TYPE_VOICE_DATA:
84             dec->sample_rate = 1000000 / (256 - get_byte(pb));
85             if (sample_rate)
86                 dec->sample_rate = sample_rate;
87             dec->channels = channels;
88             dec->codec_id = codec_get_id(voc_codec_tags, get_byte(pb));
89             dec->bits_per_sample = av_get_bits_per_sample(dec->codec_id);
90             voc->remaining_size -= 2;
91             max_size -= 2;
92             channels = 1;
93             break;
94
95         case VOC_TYPE_VOICE_DATA_CONT:
96             break;
97
98         case VOC_TYPE_EXTENDED:
99             sample_rate = get_le16(pb);
100             get_byte(pb);
101             channels = get_byte(pb) + 1;
102             sample_rate = 256000000 / (channels * (65536 - sample_rate));
103             voc->remaining_size = 0;
104             max_size -= 4;
105             break;
106
107         case VOC_TYPE_NEW_VOICE_DATA:
108             dec->sample_rate = get_le32(pb);
109             dec->bits_per_sample = get_byte(pb);
110             dec->channels = get_byte(pb);
111             dec->codec_id = codec_get_id(voc_codec_tags, get_le16(pb));
112             url_fskip(pb, 4);
113             voc->remaining_size -= 12;
114             max_size -= 12;
115             break;
116
117         default:
118             url_fskip(pb, voc->remaining_size);
119             max_size -= voc->remaining_size;
120             voc->remaining_size = 0;
121             break;
122         }
123     }
124
125     dec->bit_rate = dec->sample_rate * dec->bits_per_sample;
126
127     if (max_size <= 0)
128         max_size = 2048;
129     size = FFMIN(voc->remaining_size, max_size);
130     voc->remaining_size -= size;
131     return av_get_packet(pb, pkt, size);
132 }
133
134 static int voc_read_packet(AVFormatContext *s, AVPacket *pkt)
135 {
136     return voc_get_packet(s, pkt, s->streams[0], 0);
137 }
138
139 static int voc_read_close(AVFormatContext *s)
140 {
141     return 0;
142 }
143
144 AVInputFormat voc_demuxer = {
145     "voc",
146     "Creative Voice File format",
147     sizeof(voc_dec_context_t),
148     voc_probe,
149     voc_read_header,
150     voc_read_packet,
151     voc_read_close,
152     .codec_tag=(const AVCodecTag*[]){voc_codec_tags, 0},
153 };