Mercurial > libavformat.hg
annotate mp3.c @ 294:6091b76cfc2a libavformat
added MPEG2TS support in RTP, SDP and RTSP - replaced fake RTP demux by a specific API
author | bellard |
---|---|
date | Wed, 29 Oct 2003 14:25:27 +0000 |
parents | a313e1080322 |
children | 5552d3761ec0 |
rev | line source |
---|---|
234 | 1 /* |
2 * MP3 encoder and decoder | |
3 * Copyright (c) 2003 Fabrice Bellard. | |
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
18 */ | |
19 #include "avformat.h" | |
20 | |
21 #define ID3_HEADER_SIZE 10 | |
22 #define ID3_TAG_SIZE 128 | |
23 | |
24 #define ID3_GENRE_MAX 125 | |
25 | |
26 static const char *id3_genre_str[ID3_GENRE_MAX + 1] = { | |
27 [0] = "Blues", | |
28 [1] = "Classic Rock", | |
29 [2] = "Country", | |
30 [3] = "Dance", | |
31 [4] = "Disco", | |
32 [5] = "Funk", | |
33 [6] = "Grunge", | |
34 [7] = "Hip-Hop", | |
35 [8] = "Jazz", | |
36 [9] = "Metal", | |
37 [10] = "New Age", | |
38 [11] = "Oldies", | |
39 [12] = "Other", | |
40 [13] = "Pop", | |
41 [14] = "R&B", | |
42 [15] = "Rap", | |
43 [16] = "Reggae", | |
44 [17] = "Rock", | |
45 [18] = "Techno", | |
46 [19] = "Industrial", | |
47 [20] = "Alternative", | |
48 [21] = "Ska", | |
49 [22] = "Death Metal", | |
50 [23] = "Pranks", | |
51 [24] = "Soundtrack", | |
52 [25] = "Euro-Techno", | |
53 [26] = "Ambient", | |
54 [27] = "Trip-Hop", | |
55 [28] = "Vocal", | |
56 [29] = "Jazz+Funk", | |
57 [30] = "Fusion", | |
58 [31] = "Trance", | |
59 [32] = "Classical", | |
60 [33] = "Instrumental", | |
61 [34] = "Acid", | |
62 [35] = "House", | |
63 [36] = "Game", | |
64 [37] = "Sound Clip", | |
65 [38] = "Gospel", | |
66 [39] = "Noise", | |
67 [40] = "AlternRock", | |
68 [41] = "Bass", | |
69 [42] = "Soul", | |
70 [43] = "Punk", | |
71 [44] = "Space", | |
72 [45] = "Meditative", | |
73 [46] = "Instrumental Pop", | |
74 [47] = "Instrumental Rock", | |
75 [48] = "Ethnic", | |
76 [49] = "Gothic", | |
77 [50] = "Darkwave", | |
78 [51] = "Techno-Industrial", | |
79 [52] = "Electronic", | |
80 [53] = "Pop-Folk", | |
81 [54] = "Eurodance", | |
82 [55] = "Dream", | |
83 [56] = "Southern Rock", | |
84 [57] = "Comedy", | |
85 [58] = "Cult", | |
86 [59] = "Gangsta", | |
87 [60] = "Top 40", | |
88 [61] = "Christian Rap", | |
89 [62] = "Pop/Funk", | |
90 [63] = "Jungle", | |
91 [64] = "Native American", | |
92 [65] = "Cabaret", | |
93 [66] = "New Wave", | |
94 [67] = "Psychadelic", | |
95 [68] = "Rave", | |
96 [69] = "Showtunes", | |
97 [70] = "Trailer", | |
98 [71] = "Lo-Fi", | |
99 [72] = "Tribal", | |
100 [73] = "Acid Punk", | |
101 [74] = "Acid Jazz", | |
102 [75] = "Polka", | |
103 [76] = "Retro", | |
104 [77] = "Musical", | |
105 [78] = "Rock & Roll", | |
106 [79] = "Hard Rock", | |
107 [80] = "Folk", | |
108 [81] = "Folk-Rock", | |
109 [82] = "National Folk", | |
110 [83] = "Swing", | |
111 [84] = "Fast Fusion", | |
112 [85] = "Bebob", | |
113 [86] = "Latin", | |
114 [87] = "Revival", | |
115 [88] = "Celtic", | |
116 [89] = "Bluegrass", | |
117 [90] = "Avantgarde", | |
118 [91] = "Gothic Rock", | |
119 [92] = "Progressive Rock", | |
120 [93] = "Psychedelic Rock", | |
121 [94] = "Symphonic Rock", | |
122 [95] = "Slow Rock", | |
123 [96] = "Big Band", | |
124 [97] = "Chorus", | |
125 [98] = "Easy Listening", | |
126 [99] = "Acoustic", | |
127 [100] = "Humour", | |
128 [101] = "Speech", | |
129 [102] = "Chanson", | |
130 [103] = "Opera", | |
131 [104] = "Chamber Music", | |
132 [105] = "Sonata", | |
133 [106] = "Symphony", | |
134 [107] = "Booty Bass", | |
135 [108] = "Primus", | |
136 [109] = "Porn Groove", | |
137 [110] = "Satire", | |
138 [111] = "Slow Jam", | |
139 [112] = "Club", | |
140 [113] = "Tango", | |
141 [114] = "Samba", | |
142 [115] = "Folklore", | |
143 [116] = "Ballad", | |
144 [117] = "Power Ballad", | |
145 [118] = "Rhythmic Soul", | |
146 [119] = "Freestyle", | |
147 [120] = "Duet", | |
148 [121] = "Punk Rock", | |
149 [122] = "Drum Solo", | |
150 [123] = "A capella", | |
151 [124] = "Euro-House", | |
152 [125] = "Dance Hall", | |
153 }; | |
154 | |
155 /* buf must be ID3_HEADER_SIZE byte long */ | |
156 static int id3_match(const uint8_t *buf) | |
157 { | |
158 return (buf[0] == 'I' && | |
159 buf[1] == 'D' && | |
160 buf[2] == '3' && | |
161 buf[3] != 0xff && | |
162 buf[4] != 0xff && | |
163 (buf[6] & 0x80) == 0 && | |
164 (buf[7] & 0x80) == 0 && | |
165 (buf[8] & 0x80) == 0 && | |
166 (buf[9] & 0x80) == 0); | |
167 } | |
168 | |
169 static void id3_get_string(char *str, int str_size, | |
170 const uint8_t *buf, int buf_size) | |
171 { | |
172 int i, c; | |
173 char *q; | |
174 | |
175 q = str; | |
176 for(i = 0; i < buf_size; i++) { | |
177 c = buf[i]; | |
178 if (c == '\0') | |
179 break; | |
180 if ((q - str) >= str_size - 1) | |
181 break; | |
182 *q++ = c; | |
183 } | |
184 *q = '\0'; | |
185 } | |
186 | |
187 /* 'buf' must be ID3_TAG_SIZE byte long */ | |
188 static int id3_parse_tag(AVFormatContext *s, const uint8_t *buf) | |
189 { | |
190 char str[5]; | |
191 int genre; | |
192 | |
193 if (!(buf[0] == 'T' && | |
194 buf[1] == 'A' && | |
195 buf[2] == 'G')) | |
196 return -1; | |
197 id3_get_string(s->title, sizeof(s->title), buf + 3, 30); | |
198 id3_get_string(s->author, sizeof(s->author), buf + 33, 30); | |
199 id3_get_string(s->album, sizeof(s->album), buf + 63, 30); | |
200 id3_get_string(str, sizeof(str), buf + 93, 4); | |
201 s->year = atoi(str); | |
202 id3_get_string(s->comment, sizeof(s->comment), buf + 97, 30); | |
203 if (buf[125] == 0 && buf[126] != 0) | |
204 s->track = buf[126]; | |
205 genre = buf[127]; | |
206 if (genre <= ID3_GENRE_MAX) | |
207 pstrcpy(s->genre, sizeof(s->genre), id3_genre_str[genre]); | |
208 return 0; | |
209 } | |
210 | |
211 static void id3_create_tag(AVFormatContext *s, uint8_t *buf) | |
212 { | |
213 int v, i; | |
214 | |
215 memset(buf, 0, ID3_TAG_SIZE); /* fail safe */ | |
216 buf[0] = 'T'; | |
217 buf[1] = 'A'; | |
218 buf[2] = 'G'; | |
219 strncpy(buf + 3, s->title, 30); | |
220 strncpy(buf + 33, s->author, 30); | |
221 strncpy(buf + 63, s->album, 30); | |
222 v = s->year; | |
223 if (v > 0) { | |
224 for(i = 0;i < 4; i++) { | |
225 buf[96 - i] = '0' + (v % 10); | |
226 v = v / 10; | |
227 } | |
228 } | |
229 strncpy(buf + 97, s->comment, 30); | |
230 if (s->track != 0) { | |
231 buf[125] = 0; | |
232 buf[126] = s->track; | |
233 } | |
234 for(i = 0; i <= ID3_GENRE_MAX; i++) { | |
235 if (!strcasecmp(s->genre, id3_genre_str[i])) { | |
236 buf[127] = i; | |
237 break; | |
238 } | |
239 } | |
240 } | |
241 | |
242 /* mp3 read */ | |
243 static int mp3_read_header(AVFormatContext *s, | |
244 AVFormatParameters *ap) | |
245 { | |
246 AVStream *st; | |
247 uint8_t buf[ID3_TAG_SIZE]; | |
248 int len, ret, filesize; | |
249 | |
250 st = av_new_stream(s, 0); | |
251 if (!st) | |
252 return AVERROR_NOMEM; | |
253 | |
254 st->codec.codec_type = CODEC_TYPE_AUDIO; | |
255 st->codec.codec_id = CODEC_ID_MP3; | |
256 | |
257 /* try to get the TAG */ | |
258 if (!url_is_streamed(&s->pb)) { | |
259 /* XXX: change that */ | |
260 filesize = url_filesize(url_fileno(&s->pb)); | |
261 if (filesize > 128) { | |
262 url_fseek(&s->pb, filesize - 128, SEEK_SET); | |
263 ret = get_buffer(&s->pb, buf, ID3_TAG_SIZE); | |
264 if (ret == ID3_TAG_SIZE) { | |
265 id3_parse_tag(s, buf); | |
266 } | |
267 url_fseek(&s->pb, 0, SEEK_SET); | |
268 } | |
269 } | |
270 | |
271 /* if ID3 header found, skip it */ | |
272 ret = get_buffer(&s->pb, buf, ID3_HEADER_SIZE); | |
273 if (ret != ID3_HEADER_SIZE) | |
274 return -1; | |
275 if (id3_match(buf)) { | |
276 /* skip ID3 header */ | |
277 len = ((buf[6] & 0x7f) << 21) | | |
278 ((buf[7] & 0x7f) << 14) | | |
279 ((buf[8] & 0x7f) << 7) | | |
280 (buf[9] & 0x7f); | |
281 url_fskip(&s->pb, len); | |
282 } else { | |
283 url_fseek(&s->pb, 0, SEEK_SET); | |
284 } | |
285 | |
286 /* the parameters will be extracted from the compressed bitstream */ | |
287 return 0; | |
288 } | |
289 | |
290 #define MP3_PACKET_SIZE 1024 | |
291 | |
292 static int mp3_read_packet(AVFormatContext *s, AVPacket *pkt) | |
293 { | |
294 int ret, size; | |
295 // AVStream *st = s->streams[0]; | |
296 | |
297 size= MP3_PACKET_SIZE; | |
298 | |
299 if (av_new_packet(pkt, size) < 0) | |
300 return -EIO; | |
301 | |
302 pkt->stream_index = 0; | |
303 ret = get_buffer(&s->pb, pkt->data, size); | |
304 if (ret <= 0) { | |
305 av_free_packet(pkt); | |
306 return -EIO; | |
307 } | |
308 /* note: we need to modify the packet size here to handle the last | |
309 packet */ | |
310 pkt->size = ret; | |
311 return ret; | |
312 } | |
313 | |
314 static int mp3_read_close(AVFormatContext *s) | |
315 { | |
316 return 0; | |
317 } | |
318 | |
277
a313e1080322
disable encoders where appropriate (patch courtesy of BERO
melanson
parents:
241
diff
changeset
|
319 #ifdef CONFIG_ENCODERS |
234 | 320 /* simple formats */ |
321 static int mp3_write_header(struct AVFormatContext *s) | |
322 { | |
323 return 0; | |
324 } | |
325 | |
326 static int mp3_write_packet(struct AVFormatContext *s, int stream_index, | |
241 | 327 const uint8_t *buf, int size, int64_t pts) |
234 | 328 { |
329 put_buffer(&s->pb, buf, size); | |
330 put_flush_packet(&s->pb); | |
331 return 0; | |
332 } | |
333 | |
334 static int mp3_write_trailer(struct AVFormatContext *s) | |
335 { | |
336 uint8_t buf[ID3_TAG_SIZE]; | |
337 | |
338 /* write the id3 header */ | |
339 if (s->title[0] != '\0') { | |
340 id3_create_tag(s, buf); | |
341 put_buffer(&s->pb, buf, ID3_TAG_SIZE); | |
342 put_flush_packet(&s->pb); | |
343 } | |
344 return 0; | |
345 } | |
277
a313e1080322
disable encoders where appropriate (patch courtesy of BERO
melanson
parents:
241
diff
changeset
|
346 #endif //CONFIG_ENCODERS |
234 | 347 |
348 AVInputFormat mp3_iformat = { | |
349 "mp3", | |
350 "MPEG audio", | |
351 0, | |
352 NULL, | |
353 mp3_read_header, | |
354 mp3_read_packet, | |
355 mp3_read_close, | |
356 .extensions = "mp2,mp3", /* XXX: use probe */ | |
357 }; | |
358 | |
277
a313e1080322
disable encoders where appropriate (patch courtesy of BERO
melanson
parents:
241
diff
changeset
|
359 #ifdef CONFIG_ENCODERS |
234 | 360 AVOutputFormat mp2_oformat = { |
361 "mp2", | |
362 "MPEG audio layer 2", | |
363 "audio/x-mpeg", | |
364 #ifdef CONFIG_MP3LAME | |
365 "mp2", | |
366 #else | |
367 "mp2,mp3", | |
368 #endif | |
369 0, | |
370 CODEC_ID_MP2, | |
371 0, | |
372 mp3_write_header, | |
373 mp3_write_packet, | |
374 mp3_write_trailer, | |
375 }; | |
376 | |
377 #ifdef CONFIG_MP3LAME | |
378 AVOutputFormat mp3_oformat = { | |
379 "mp3", | |
380 "MPEG audio layer 3", | |
381 "audio/x-mpeg", | |
382 "mp3", | |
383 0, | |
384 CODEC_ID_MP3, | |
385 0, | |
386 mp3_write_header, | |
387 mp3_write_packet, | |
388 mp3_write_trailer, | |
389 }; | |
390 #endif | |
277
a313e1080322
disable encoders where appropriate (patch courtesy of BERO
melanson
parents:
241
diff
changeset
|
391 #endif //CONFIG_ENCODERS |
234 | 392 |
393 int mp3_init(void) | |
394 { | |
395 av_register_input_format(&mp3_iformat); | |
277
a313e1080322
disable encoders where appropriate (patch courtesy of BERO
melanson
parents:
241
diff
changeset
|
396 #ifdef CONFIG_ENCODERS |
234 | 397 av_register_output_format(&mp2_oformat); |
398 #ifdef CONFIG_MP3LAME | |
399 av_register_output_format(&mp3_oformat); | |
400 #endif | |
277
a313e1080322
disable encoders where appropriate (patch courtesy of BERO
melanson
parents:
241
diff
changeset
|
401 #endif //CONFIG_ENCODERS |
234 | 402 return 0; |
403 } |