Merged revisions 338995 via svnmerge from
[asterisk/asterisk.git] / addons / format_mp3.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Anthony Minessale <anthmct@yahoo.com>
5  *
6  * Derived from other asterisk sound formats by
7  * Mark Spencer <markster@linux-support.net>
8  *
9  * Thanks to mpglib from http://www.mpg123.org/
10  * and Chris Stenton [jacs@gnome.co.uk]
11  * for coding the ability to play stereo and non-8khz files
12  
13  * See http://www.asterisk.org for more information about
14  * the Asterisk project. Please do not directly contact
15  * any of the maintainers of this project for assistance;
16  * the project provides a web site, mailing lists and IRC
17  * channels for your use.
18  *
19  * This program is free software, distributed under the terms of
20  * the GNU General Public License Version 2. See the LICENSE file
21  * at the top of the source tree.
22  */
23
24 /*!
25  * \file
26  * \brief MP3 Format Handler
27  * \ingroup formats
28  */
29
30 /*** MODULEINFO
31         <defaultenabled>no</defaultenabled>
32         <support_level>extended</support_level>
33  ***/
34
35 #include "asterisk.h"
36
37 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
38
39 #include "mp3/mpg123.h"
40 #include "mp3/mpglib.h"
41
42 #include "asterisk/module.h"
43 #include "asterisk/mod_format.h"
44 #include "asterisk/logger.h"
45
46 #define MP3_BUFLEN 320
47 #define MP3_SCACHE 16384
48 #define MP3_DCACHE 8192
49
50 struct mp3_private {
51         char waste[AST_FRIENDLY_OFFSET];        /* Buffer for sending frames, etc */
52         char empty;                             /* Empty character */
53         int lasttimeout;
54         int maxlen;
55         struct timeval last;
56         struct mpstr mp;
57         char sbuf[MP3_SCACHE];
58         char dbuf[MP3_DCACHE];
59         int buflen;
60         int sbuflen;
61         int dbuflen;
62         int dbufoffset;
63         int sbufoffset;
64         int lastseek;
65         int offset;
66         long seek;
67 };
68
69 static const char name[] = "mp3";
70
71 #define BLOCKSIZE 160
72 #define OUTSCALE 4096
73
74 #define GAIN -4         /* 2^GAIN is the multiple to increase the volume by */
75
76 #if __BYTE_ORDER == __LITTLE_ENDIAN
77 #define htoll(b) (b)
78 #define htols(b) (b)
79 #define ltohl(b) (b)
80 #define ltohs(b) (b)
81 #else
82 #if __BYTE_ORDER == __BIG_ENDIAN
83 #define htoll(b)  \
84           (((((b)      ) & 0xFF) << 24) | \
85                ((((b) >>  8) & 0xFF) << 16) | \
86                    ((((b) >> 16) & 0xFF) <<  8) | \
87                    ((((b) >> 24) & 0xFF)      ))
88 #define htols(b) \
89           (((((b)      ) & 0xFF) << 8) | \
90                    ((((b) >> 8) & 0xFF)      ))
91 #define ltohl(b) htoll(b)
92 #define ltohs(b) htols(b)
93 #else
94 #error "Endianess not defined"
95 #endif
96 #endif
97
98
99 static int mp3_open(struct ast_filestream *s)
100 {
101         struct mp3_private *p = s->_private;
102         InitMP3(&p->mp, OUTSCALE);
103         return 0;
104 }
105
106
107 static void mp3_close(struct ast_filestream *s)
108 {
109         struct mp3_private *p = s->_private;
110         
111         ExitMP3(&p->mp);
112         return;
113 }
114
115 static int mp3_squeue(struct ast_filestream *s) 
116 {
117         struct mp3_private *p = s->_private;
118         int res=0;
119         
120         p->lastseek = ftell(s->f);
121         p->sbuflen = fread(p->sbuf, 1, MP3_SCACHE, s->f);
122         if(p->sbuflen < 0) {
123                 ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", p->sbuflen, strerror(errno));
124                 return -1;
125         }
126         res = decodeMP3(&p->mp,p->sbuf,p->sbuflen,p->dbuf,MP3_DCACHE,&p->dbuflen);
127         if(res != MP3_OK)
128                 return -1;
129         p->sbuflen -= p->dbuflen;
130         p->dbufoffset = 0;
131         return 0;
132 }
133
134 static int mp3_dqueue(struct ast_filestream *s) 
135 {
136         struct mp3_private *p = s->_private;
137         int res=0;
138         
139         if((res = decodeMP3(&p->mp,NULL,0,p->dbuf,MP3_DCACHE,&p->dbuflen)) == MP3_OK) {
140                 p->sbuflen -= p->dbuflen;
141                 p->dbufoffset = 0;
142         }
143         return res;
144 }
145
146 static int mp3_queue(struct ast_filestream *s)
147 {
148         struct mp3_private *p = s->_private;
149         int res = 0, bytes = 0;
150         
151         if(p->seek) {
152                 ExitMP3(&p->mp);
153                 InitMP3(&p->mp, OUTSCALE);
154                 fseek(s->f, 0, SEEK_SET);
155                 p->sbuflen = p->dbuflen = p->offset = 0;
156                 while(p->offset < p->seek) {
157                         if(mp3_squeue(s))
158                                 return -1;
159                         while(p->offset < p->seek && ((res = mp3_dqueue(s))) == MP3_OK) {
160                                 for(bytes = 0 ; bytes < p->dbuflen ; bytes++) {
161                                         p->dbufoffset++;
162                                         p->offset++;
163                                         if(p->offset >= p->seek)
164                                                 break;
165                                 }
166                         }
167                         if(res == MP3_ERR)
168                                 return -1;
169                 }
170                 
171                 p->seek = 0;
172                 return 0;
173         }
174         if(p->dbuflen == 0) {
175                 if(p->sbuflen) {
176                         res = mp3_dqueue(s);
177                         if(res == MP3_ERR)
178                                 return -1;
179                 }
180                 if(! p->sbuflen || res != MP3_OK) {
181                         if(mp3_squeue(s))
182                                 return -1;
183                 }
184                 
185         }
186
187         return 0;
188 }
189
190 static struct ast_frame *mp3_read(struct ast_filestream *s, int *whennext)
191 {
192
193         struct mp3_private *p = s->_private;
194         int delay =0;
195         int save=0;
196
197         /* Send a frame from the file to the appropriate channel */
198
199         if(mp3_queue(s))
200                 return NULL;
201
202         if(p->dbuflen) {
203                 for(p->buflen=0; p->buflen < MP3_BUFLEN && p->buflen < p->dbuflen; p->buflen++) {
204                         s->buf[p->buflen + AST_FRIENDLY_OFFSET] = p->dbuf[p->buflen+p->dbufoffset];
205                         p->sbufoffset++;
206                 }
207                 p->dbufoffset += p->buflen;
208                 p->dbuflen -= p->buflen;
209
210                 if(p->buflen < MP3_BUFLEN) {
211                         if(mp3_queue(s))
212                                 return NULL;
213
214                         for(save = p->buflen; p->buflen < MP3_BUFLEN; p->buflen++) {
215                                 s->buf[p->buflen + AST_FRIENDLY_OFFSET] = p->dbuf[(p->buflen-save)+p->dbufoffset];
216                                 p->sbufoffset++;
217                         }
218                         p->dbufoffset += (MP3_BUFLEN - save);
219                         p->dbuflen -= (MP3_BUFLEN - save);
220
221                 } 
222
223         }
224         
225         p->offset += p->buflen;
226         delay = p->buflen/2;
227         s->fr.frametype = AST_FRAME_VOICE;
228         ast_format_set(&s->fr.subclass.format, AST_FORMAT_SLINEAR, 0);
229         AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, p->buflen);
230         s->fr.mallocd = 0;
231         s->fr.samples = delay;
232         *whennext = delay;
233         return &s->fr;
234 }
235
236
237 static int mp3_write(struct ast_filestream *fs, struct ast_frame *f)
238 {
239         ast_log(LOG_ERROR,"I Can't write MP3 only read them.\n");
240         return -1;
241
242 }
243
244
245 static int mp3_seek(struct ast_filestream *s, off_t sample_offset, int whence)
246 {
247         struct mp3_private *p = s->_private;
248         off_t min,max,cur;
249         long offset=0,samples;
250         samples = sample_offset * 2;
251
252         min = 0;
253         fseek(s->f, 0, SEEK_END);
254         max = ftell(s->f) * 100;
255         cur = p->offset;
256
257         if (whence == SEEK_SET)
258                 offset = samples + min;
259         else if (whence == SEEK_CUR || whence == SEEK_FORCECUR)
260                 offset = samples + cur;
261         else if (whence == SEEK_END)
262                 offset = max - samples;
263         if (whence != SEEK_FORCECUR) {
264                 offset = (offset > max)?max:offset;
265         }
266
267         p->seek = offset;
268         return fseek(s->f, offset, SEEK_SET);
269         
270 }
271
272 static int mp3_rewrite(struct ast_filestream *s, const char *comment) 
273 {
274         ast_log(LOG_ERROR,"I Can't write MP3 only read them.\n");
275         return -1;
276 }
277
278 static int mp3_trunc(struct ast_filestream *s) 
279 {
280
281         ast_log(LOG_ERROR,"I Can't write MP3 only read them.\n");
282         return -1;
283 }
284
285 static off_t mp3_tell(struct ast_filestream *s)
286 {
287         struct mp3_private *p = s->_private;
288         
289         return p->offset/2;
290 }
291
292 static char *mp3_getcomment(struct ast_filestream *s)
293 {
294         return NULL;
295 }
296
297 static struct ast_format_def mp3_f = {
298         .name = "mp3",
299         .exts = "mp3",
300         .open = mp3_open,
301         .write = mp3_write,
302         .rewrite = mp3_rewrite,
303         .seek = mp3_seek,
304         .trunc = mp3_trunc,
305         .tell = mp3_tell,
306         .read = mp3_read,
307         .close = mp3_close,
308         .getcomment = mp3_getcomment,
309         .buf_size = MP3_BUFLEN + AST_FRIENDLY_OFFSET,
310         .desc_size = sizeof(struct mp3_private),
311 };
312
313
314 static int load_module(void)
315 {
316         ast_format_set(&mp3_f.format, AST_FORMAT_SLINEAR, 0);
317         InitMP3Constants();
318         return ast_format_def_register(&mp3_f);
319 }
320
321 static int unload_module(void)
322 {
323         return ast_format_def_unregister(name);
324 }       
325
326 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "MP3 format [Any rate but 8000hz mono is optimal]");