Mercurial > libavformat.hg
annotate mp3.c @ 2843:7a6b7595f933 libavformat
cosmetics: indentation
author | aurel |
---|---|
date | Thu, 20 Dec 2007 00:26:53 +0000 |
parents | 51e0a502392f |
children | 6eaceb561ae9 |
rev | line source |
---|---|
885 | 1 /* |
1415
3b00fb8ef8e4
replace coder/decoder file description in libavformat by muxer/demuxer
aurel
parents:
1358
diff
changeset
|
2 * MP3 muxer and demuxer |
234 | 3 * Copyright (c) 2003 Fabrice Bellard. |
4 * | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1321
diff
changeset
|
5 * This file is part of FFmpeg. |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1321
diff
changeset
|
6 * |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1321
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
234 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1321
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
234 | 11 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1321
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
234 | 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 | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1321
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
896
edbe5c3717f9
Update licensing information: The FSF changed postal address.
diego
parents:
885
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
234 | 20 */ |
21 #include "avformat.h" | |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
22 #include "mpegaudio.h" |
2189 | 23 #include "avstring.h" |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
24 #include "mpegaudiodecheader.h" |
234 | 25 |
2096 | 26 #define ID3v2_HEADER_SIZE 10 |
27 #define ID3v1_TAG_SIZE 128 | |
234 | 28 |
2096 | 29 #define ID3v1_GENRE_MAX 125 |
234 | 30 |
2096 | 31 static const char *id3v1_genre_str[ID3v1_GENRE_MAX + 1] = { |
234 | 32 [0] = "Blues", |
33 [1] = "Classic Rock", | |
34 [2] = "Country", | |
35 [3] = "Dance", | |
36 [4] = "Disco", | |
37 [5] = "Funk", | |
38 [6] = "Grunge", | |
39 [7] = "Hip-Hop", | |
40 [8] = "Jazz", | |
41 [9] = "Metal", | |
42 [10] = "New Age", | |
43 [11] = "Oldies", | |
44 [12] = "Other", | |
45 [13] = "Pop", | |
46 [14] = "R&B", | |
47 [15] = "Rap", | |
48 [16] = "Reggae", | |
49 [17] = "Rock", | |
50 [18] = "Techno", | |
51 [19] = "Industrial", | |
52 [20] = "Alternative", | |
53 [21] = "Ska", | |
54 [22] = "Death Metal", | |
55 [23] = "Pranks", | |
56 [24] = "Soundtrack", | |
57 [25] = "Euro-Techno", | |
58 [26] = "Ambient", | |
59 [27] = "Trip-Hop", | |
60 [28] = "Vocal", | |
61 [29] = "Jazz+Funk", | |
62 [30] = "Fusion", | |
63 [31] = "Trance", | |
64 [32] = "Classical", | |
65 [33] = "Instrumental", | |
66 [34] = "Acid", | |
67 [35] = "House", | |
68 [36] = "Game", | |
69 [37] = "Sound Clip", | |
70 [38] = "Gospel", | |
71 [39] = "Noise", | |
72 [40] = "AlternRock", | |
73 [41] = "Bass", | |
74 [42] = "Soul", | |
75 [43] = "Punk", | |
76 [44] = "Space", | |
77 [45] = "Meditative", | |
78 [46] = "Instrumental Pop", | |
79 [47] = "Instrumental Rock", | |
80 [48] = "Ethnic", | |
81 [49] = "Gothic", | |
82 [50] = "Darkwave", | |
83 [51] = "Techno-Industrial", | |
84 [52] = "Electronic", | |
85 [53] = "Pop-Folk", | |
86 [54] = "Eurodance", | |
87 [55] = "Dream", | |
88 [56] = "Southern Rock", | |
89 [57] = "Comedy", | |
90 [58] = "Cult", | |
91 [59] = "Gangsta", | |
92 [60] = "Top 40", | |
93 [61] = "Christian Rap", | |
94 [62] = "Pop/Funk", | |
95 [63] = "Jungle", | |
96 [64] = "Native American", | |
97 [65] = "Cabaret", | |
98 [66] = "New Wave", | |
99 [67] = "Psychadelic", | |
100 [68] = "Rave", | |
101 [69] = "Showtunes", | |
102 [70] = "Trailer", | |
103 [71] = "Lo-Fi", | |
104 [72] = "Tribal", | |
105 [73] = "Acid Punk", | |
106 [74] = "Acid Jazz", | |
107 [75] = "Polka", | |
108 [76] = "Retro", | |
109 [77] = "Musical", | |
110 [78] = "Rock & Roll", | |
111 [79] = "Hard Rock", | |
112 [80] = "Folk", | |
113 [81] = "Folk-Rock", | |
114 [82] = "National Folk", | |
115 [83] = "Swing", | |
116 [84] = "Fast Fusion", | |
117 [85] = "Bebob", | |
118 [86] = "Latin", | |
119 [87] = "Revival", | |
120 [88] = "Celtic", | |
121 [89] = "Bluegrass", | |
122 [90] = "Avantgarde", | |
123 [91] = "Gothic Rock", | |
124 [92] = "Progressive Rock", | |
125 [93] = "Psychedelic Rock", | |
126 [94] = "Symphonic Rock", | |
127 [95] = "Slow Rock", | |
128 [96] = "Big Band", | |
129 [97] = "Chorus", | |
130 [98] = "Easy Listening", | |
131 [99] = "Acoustic", | |
132 [100] = "Humour", | |
133 [101] = "Speech", | |
134 [102] = "Chanson", | |
135 [103] = "Opera", | |
136 [104] = "Chamber Music", | |
137 [105] = "Sonata", | |
138 [106] = "Symphony", | |
139 [107] = "Booty Bass", | |
140 [108] = "Primus", | |
141 [109] = "Porn Groove", | |
142 [110] = "Satire", | |
143 [111] = "Slow Jam", | |
144 [112] = "Club", | |
145 [113] = "Tango", | |
146 [114] = "Samba", | |
147 [115] = "Folklore", | |
148 [116] = "Ballad", | |
149 [117] = "Power Ballad", | |
150 [118] = "Rhythmic Soul", | |
151 [119] = "Freestyle", | |
152 [120] = "Duet", | |
153 [121] = "Punk Rock", | |
154 [122] = "Drum Solo", | |
155 [123] = "A capella", | |
156 [124] = "Euro-House", | |
157 [125] = "Dance Hall", | |
158 }; | |
159 | |
2096 | 160 /* buf must be ID3v2_HEADER_SIZE byte long */ |
161 static int id3v2_match(const uint8_t *buf) | |
234 | 162 { |
163 return (buf[0] == 'I' && | |
164 buf[1] == 'D' && | |
165 buf[2] == '3' && | |
166 buf[3] != 0xff && | |
167 buf[4] != 0xff && | |
168 (buf[6] & 0x80) == 0 && | |
169 (buf[7] & 0x80) == 0 && | |
170 (buf[8] & 0x80) == 0 && | |
171 (buf[9] & 0x80) == 0); | |
172 } | |
173 | |
2097 | 174 static unsigned int id3v2_get_size(ByteIOContext *s, int len) |
175 { | |
176 int v=0; | |
177 while(len--) | |
178 v= (v<<7) + (get_byte(s)&0x7F); | |
179 return v; | |
180 } | |
181 | |
182 static void id3v2_read_ttag(AVFormatContext *s, int taglen, char *dst, int dstlen) | |
183 { | |
184 char *q; | |
185 int len; | |
186 | |
187 if(taglen < 1) | |
188 return; | |
189 | |
190 taglen--; /* account for encoding type byte */ | |
191 dstlen--; /* Leave space for zero terminator */ | |
192 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
193 switch(get_byte(s->pb)) { /* encoding type */ |
2097 | 194 |
195 case 0: /* ISO-8859-1 (0 - 255 maps directly into unicode) */ | |
196 q = dst; | |
197 while(taglen--) { | |
198 uint8_t tmp; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
199 PUT_UTF8(get_byte(s->pb), tmp, if (q - dst < dstlen - 1) *q++ = tmp;) |
2097 | 200 } |
201 *q = '\0'; | |
202 break; | |
203 | |
204 case 3: /* UTF-8 */ | |
205 len = FFMIN(taglen, dstlen); | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
206 get_buffer(s->pb, dst, len); |
2097 | 207 dst[len] = 0; |
208 break; | |
209 } | |
210 } | |
211 | |
212 /** | |
213 * ID3v2 parser | |
214 * | |
215 * Handles ID3v2.2, 2.3 and 2.4. | |
216 * | |
217 */ | |
218 | |
219 static void id3v2_parse(AVFormatContext *s, int len, uint8_t version, uint8_t flags) | |
220 { | |
221 int isv34, tlen; | |
222 uint32_t tag; | |
223 offset_t next; | |
224 char tmp[16]; | |
225 int taghdrlen; | |
226 const char *reason; | |
227 | |
228 switch(version) { | |
229 case 2: | |
230 if(flags & 0x40) { | |
231 reason = "compression"; | |
232 goto error; | |
233 } | |
234 isv34 = 0; | |
235 taghdrlen = 6; | |
236 break; | |
237 | |
2231
7ad682f38b9a
* Getting rid of the use of GCC language extensions
romansh
parents:
2222
diff
changeset
|
238 case 3: |
7ad682f38b9a
* Getting rid of the use of GCC language extensions
romansh
parents:
2222
diff
changeset
|
239 case 4: |
2097 | 240 isv34 = 1; |
241 taghdrlen = 10; | |
242 break; | |
243 | |
244 default: | |
245 reason = "version"; | |
246 goto error; | |
247 } | |
248 | |
249 if(flags & 0x80) { | |
250 reason = "unsynchronization"; | |
251 goto error; | |
252 } | |
253 | |
254 if(isv34 && flags & 0x40) /* Extended header present, just skip over it */ | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
255 url_fskip(s->pb, id3v2_get_size(s->pb, 4)); |
2097 | 256 |
257 while(len >= taghdrlen) { | |
258 if(isv34) { | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
259 tag = get_be32(s->pb); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
260 tlen = id3v2_get_size(s->pb, 4); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
261 get_be16(s->pb); /* flags */ |
2097 | 262 } else { |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
263 tag = get_be24(s->pb); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
264 tlen = id3v2_get_size(s->pb, 3); |
2097 | 265 } |
266 len -= taghdrlen + tlen; | |
267 | |
268 if(len < 0) | |
269 break; | |
270 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
271 next = url_ftell(s->pb) + tlen; |
2097 | 272 |
273 switch(tag) { | |
274 case MKBETAG('T', 'I', 'T', '2'): | |
275 case MKBETAG(0, 'T', 'T', '2'): | |
276 id3v2_read_ttag(s, tlen, s->title, sizeof(s->title)); | |
277 break; | |
278 case MKBETAG('T', 'P', 'E', '1'): | |
279 case MKBETAG(0, 'T', 'P', '1'): | |
280 id3v2_read_ttag(s, tlen, s->author, sizeof(s->author)); | |
281 break; | |
282 case MKBETAG('T', 'A', 'L', 'B'): | |
283 case MKBETAG(0, 'T', 'A', 'L'): | |
284 id3v2_read_ttag(s, tlen, s->album, sizeof(s->album)); | |
285 break; | |
286 case MKBETAG('T', 'C', 'O', 'N'): | |
287 case MKBETAG(0, 'T', 'C', 'O'): | |
288 id3v2_read_ttag(s, tlen, s->genre, sizeof(s->genre)); | |
289 break; | |
290 case MKBETAG('T', 'C', 'O', 'P'): | |
291 case MKBETAG(0, 'T', 'C', 'R'): | |
292 id3v2_read_ttag(s, tlen, s->copyright, sizeof(s->copyright)); | |
293 break; | |
294 case MKBETAG('T', 'R', 'C', 'K'): | |
295 case MKBETAG(0, 'T', 'R', 'K'): | |
296 id3v2_read_ttag(s, tlen, tmp, sizeof(tmp)); | |
297 s->track = atoi(tmp); | |
298 break; | |
299 case 0: | |
300 /* padding, skip to end */ | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
301 url_fskip(s->pb, len); |
2097 | 302 len = 0; |
303 continue; | |
304 } | |
305 /* Skip to end of tag */ | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
306 url_fseek(s->pb, next, SEEK_SET); |
2097 | 307 } |
308 | |
309 if(version == 4 && flags & 0x10) /* Footer preset, always 10 bytes, skip over it */ | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
310 url_fskip(s->pb, 10); |
2097 | 311 return; |
312 | |
313 error: | |
314 av_log(s, AV_LOG_INFO, "ID3v2.%d tag skipped, cannot handle %s\n", version, reason); | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
315 url_fskip(s->pb, len); |
2097 | 316 } |
317 | |
2096 | 318 static void id3v1_get_string(char *str, int str_size, |
319 const uint8_t *buf, int buf_size) | |
234 | 320 { |
321 int i, c; | |
322 char *q; | |
323 | |
324 q = str; | |
325 for(i = 0; i < buf_size; i++) { | |
326 c = buf[i]; | |
327 if (c == '\0') | |
328 break; | |
329 if ((q - str) >= str_size - 1) | |
330 break; | |
331 *q++ = c; | |
332 } | |
333 *q = '\0'; | |
334 } | |
335 | |
2096 | 336 /* 'buf' must be ID3v1_TAG_SIZE byte long */ |
337 static int id3v1_parse_tag(AVFormatContext *s, const uint8_t *buf) | |
234 | 338 { |
339 char str[5]; | |
340 int genre; | |
885 | 341 |
234 | 342 if (!(buf[0] == 'T' && |
343 buf[1] == 'A' && | |
344 buf[2] == 'G')) | |
345 return -1; | |
2096 | 346 id3v1_get_string(s->title, sizeof(s->title), buf + 3, 30); |
347 id3v1_get_string(s->author, sizeof(s->author), buf + 33, 30); | |
348 id3v1_get_string(s->album, sizeof(s->album), buf + 63, 30); | |
349 id3v1_get_string(str, sizeof(str), buf + 93, 4); | |
234 | 350 s->year = atoi(str); |
2096 | 351 id3v1_get_string(s->comment, sizeof(s->comment), buf + 97, 30); |
234 | 352 if (buf[125] == 0 && buf[126] != 0) |
353 s->track = buf[126]; | |
354 genre = buf[127]; | |
2096 | 355 if (genre <= ID3v1_GENRE_MAX) |
2189 | 356 av_strlcpy(s->genre, id3v1_genre_str[genre], sizeof(s->genre)); |
234 | 357 return 0; |
358 } | |
359 | |
2096 | 360 static void id3v1_create_tag(AVFormatContext *s, uint8_t *buf) |
234 | 361 { |
362 int v, i; | |
363 | |
2096 | 364 memset(buf, 0, ID3v1_TAG_SIZE); /* fail safe */ |
234 | 365 buf[0] = 'T'; |
366 buf[1] = 'A'; | |
367 buf[2] = 'G'; | |
368 strncpy(buf + 3, s->title, 30); | |
369 strncpy(buf + 33, s->author, 30); | |
370 strncpy(buf + 63, s->album, 30); | |
371 v = s->year; | |
372 if (v > 0) { | |
373 for(i = 0;i < 4; i++) { | |
374 buf[96 - i] = '0' + (v % 10); | |
375 v = v / 10; | |
376 } | |
377 } | |
378 strncpy(buf + 97, s->comment, 30); | |
379 if (s->track != 0) { | |
380 buf[125] = 0; | |
381 buf[126] = s->track; | |
382 } | |
2096 | 383 for(i = 0; i <= ID3v1_GENRE_MAX; i++) { |
384 if (!strcasecmp(s->genre, id3v1_genre_str[i])) { | |
234 | 385 buf[127] = i; |
386 break; | |
387 } | |
388 } | |
389 } | |
390 | |
391 /* mp3 read */ | |
1107 | 392 |
393 static int mp3_read_probe(AVProbeData *p) | |
394 { | |
2234
2bc002540a9b
kill uninitialised variable warning in mp3_read_probe()
mru
parents:
2231
diff
changeset
|
395 int max_frames, first_frames = 0; |
1433
dababce8f69e
dont set the sampling rate just because 1 mp3 packet header says so (fixes playback speed on some old mencoder generated avis which where then dumped to mp3)
michael
parents:
1415
diff
changeset
|
396 int fsize, frames, sample_rate; |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
397 uint32_t header; |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
398 uint8_t *buf, *buf2, *end; |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
399 AVCodecContext avctx; |
1107 | 400 |
2096 | 401 if(id3v2_match(p->buf)) |
402 return AVPROBE_SCORE_MAX/2+1; // this must be less than mpeg-ps because some retards put id3v2 tags before mpeg-ps files | |
1107 | 403 |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
404 max_frames = 0; |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
405 buf = p->buf; |
2796 | 406 end = buf + p->buf_size - sizeof(uint32_t); |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
407 |
2797 | 408 for(; buf < end; buf= buf2+1) { |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
409 buf2 = buf; |
1107 | 410 |
1312 | 411 for(frames = 0; buf2 < end; frames++) { |
2222 | 412 header = AV_RB32(buf2); |
2090 | 413 fsize = ff_mpa_decode_header(&avctx, header, &sample_rate); |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
414 if(fsize < 0) |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
415 break; |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
416 buf2 += fsize; |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
417 } |
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
418 max_frames = FFMAX(max_frames, frames); |
1321
9eeb01383e30
reduce scores if the mp3 frames dont start from the begin of the file (fixes flv deteted as mp3 issues)
michael
parents:
1312
diff
changeset
|
419 if(buf == p->buf) |
9eeb01383e30
reduce scores if the mp3 frames dont start from the begin of the file (fixes flv deteted as mp3 issues)
michael
parents:
1312
diff
changeset
|
420 first_frames= frames; |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
421 } |
1321
9eeb01383e30
reduce scores if the mp3 frames dont start from the begin of the file (fixes flv deteted as mp3 issues)
michael
parents:
1312
diff
changeset
|
422 if (first_frames>=3) return AVPROBE_SCORE_MAX/2+1; |
2798
51e0a502392f
return a slightly larger score if we find more than 500 valid mp3 frames in a row
michael
parents:
2797
diff
changeset
|
423 else if(max_frames>500)return AVPROBE_SCORE_MAX/2; |
1321
9eeb01383e30
reduce scores if the mp3 frames dont start from the begin of the file (fixes flv deteted as mp3 issues)
michael
parents:
1312
diff
changeset
|
424 else if(max_frames>=3) return AVPROBE_SCORE_MAX/4; |
9eeb01383e30
reduce scores if the mp3 frames dont start from the begin of the file (fixes flv deteted as mp3 issues)
michael
parents:
1312
diff
changeset
|
425 else if(max_frames>=1) return 1; |
1308
866d43ed0a67
allow ffmpeg to read mp3s beginning with partial frames
gpoirier
parents:
1169
diff
changeset
|
426 else return 0; |
1107 | 427 } |
428 | |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
429 /** |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
430 * Try to find Xing/Info/VBRI tags and compute duration from info therein |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
431 */ |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
432 static void mp3_parse_vbr_tags(AVFormatContext *s, AVStream *st, offset_t base) |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
433 { |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
434 uint32_t v, spf; |
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
435 int frames = -1; /* Total number of frames in file */ |
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
436 const offset_t xing_offtbl[2][2] = {{32, 17}, {17,9}}; |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
437 MPADecodeContext c; |
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
438 |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
439 v = get_be32(s->pb); |
2703
2c2da3011d6e
make sure the mpeg audio header is valid before passing it to ff_mpegaudio_decode_header()
andoma
parents:
2667
diff
changeset
|
440 if(ff_mpa_check_header(v) < 0) |
2c2da3011d6e
make sure the mpeg audio header is valid before passing it to ff_mpegaudio_decode_header()
andoma
parents:
2667
diff
changeset
|
441 return; |
2c2da3011d6e
make sure the mpeg audio header is valid before passing it to ff_mpegaudio_decode_header()
andoma
parents:
2667
diff
changeset
|
442 |
2c2da3011d6e
make sure the mpeg audio header is valid before passing it to ff_mpegaudio_decode_header()
andoma
parents:
2667
diff
changeset
|
443 ff_mpegaudio_decode_header(&c, v); |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
444 if(c.layer != 3) |
2664 | 445 return; |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
446 |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
447 /* Check for Xing / Info tag */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
448 url_fseek(s->pb, xing_offtbl[c.lsf == 1][c.nb_channels == 1], SEEK_CUR); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
449 v = get_be32(s->pb); |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
450 if(v == MKBETAG('X', 'i', 'n', 'g') || v == MKBETAG('I', 'n', 'f', 'o')) { |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
451 v = get_be32(s->pb); |
2664 | 452 if(v & 0x1) |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
453 frames = get_be32(s->pb); |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
454 } |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
455 |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
456 /* Check for VBRI tag (always 32 bytes after end of mpegaudio header) */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
457 url_fseek(s->pb, base + 4 + 32, SEEK_SET); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
458 v = get_be32(s->pb); |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
459 if(v == MKBETAG('V', 'B', 'R', 'I')) { |
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
460 /* Check tag version */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
461 if(get_be16(s->pb) == 1) { |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
462 /* skip delay, quality and total bytes */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
463 url_fseek(s->pb, 8, SEEK_CUR); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
464 frames = get_be32(s->pb); |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
465 } |
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
466 } |
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
467 |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
468 if(frames < 0) |
2664 | 469 return; |
2663
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
470 |
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
471 spf = c.lsf ? 576 : 1152; /* Samples per frame, layer 3 */ |
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
472 st->duration = av_rescale_q(frames, (AVRational){spf, c.sample_rate}, |
be889462edfc
cosmetical change of mp3_parse_xing() to prepare for upcoming VBRI tag support
andoma
parents:
2662
diff
changeset
|
473 st->time_base); |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
474 } |
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
475 |
234 | 476 static int mp3_read_header(AVFormatContext *s, |
477 AVFormatParameters *ap) | |
478 { | |
479 AVStream *st; | |
2096 | 480 uint8_t buf[ID3v1_TAG_SIZE]; |
234 | 481 int len, ret, filesize; |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
482 offset_t off; |
234 | 483 |
484 st = av_new_stream(s, 0); | |
485 if (!st) | |
2273
7eb456c4ed8a
Replace all occurrences of AVERROR_NOMEM with AVERROR(ENOMEM).
takis
parents:
2234
diff
changeset
|
486 return AVERROR(ENOMEM); |
234 | 487 |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
814
diff
changeset
|
488 st->codec->codec_type = CODEC_TYPE_AUDIO; |
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
814
diff
changeset
|
489 st->codec->codec_id = CODEC_ID_MP3; |
2023 | 490 st->need_parsing = AVSTREAM_PARSE_FULL; |
2622 | 491 st->start_time = 0; |
885 | 492 |
234 | 493 /* try to get the TAG */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
494 if (!url_is_streamed(s->pb)) { |
234 | 495 /* XXX: change that */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
496 filesize = url_fsize(s->pb); |
234 | 497 if (filesize > 128) { |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
498 url_fseek(s->pb, filesize - 128, SEEK_SET); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
499 ret = get_buffer(s->pb, buf, ID3v1_TAG_SIZE); |
2096 | 500 if (ret == ID3v1_TAG_SIZE) { |
501 id3v1_parse_tag(s, buf); | |
234 | 502 } |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
503 url_fseek(s->pb, 0, SEEK_SET); |
234 | 504 } |
505 } | |
506 | |
2096 | 507 /* if ID3v2 header found, skip it */ |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
508 ret = get_buffer(s->pb, buf, ID3v2_HEADER_SIZE); |
2096 | 509 if (ret != ID3v2_HEADER_SIZE) |
234 | 510 return -1; |
2096 | 511 if (id3v2_match(buf)) { |
2097 | 512 /* parse ID3v2 header */ |
234 | 513 len = ((buf[6] & 0x7f) << 21) | |
514 ((buf[7] & 0x7f) << 14) | | |
515 ((buf[8] & 0x7f) << 7) | | |
516 (buf[9] & 0x7f); | |
2097 | 517 id3v2_parse(s, len, buf[3], buf[5]); |
234 | 518 } else { |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
519 url_fseek(s->pb, 0, SEEK_SET); |
234 | 520 } |
521 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
522 off = url_ftell(s->pb); |
2667
e896ac505ec6
add support for reading duration from VBRI-tag in mp3 files
andoma
parents:
2664
diff
changeset
|
523 mp3_parse_vbr_tags(s, st, off); |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
524 url_fseek(s->pb, off, SEEK_SET); |
2662
08b1ac852321
add support for reading duration from Xing-tag in mp3 files
andoma
parents:
2622
diff
changeset
|
525 |
234 | 526 /* the parameters will be extracted from the compressed bitstream */ |
527 return 0; | |
528 } | |
529 | |
530 #define MP3_PACKET_SIZE 1024 | |
531 | |
532 static int mp3_read_packet(AVFormatContext *s, AVPacket *pkt) | |
533 { | |
534 int ret, size; | |
535 // AVStream *st = s->streams[0]; | |
885 | 536 |
234 | 537 size= MP3_PACKET_SIZE; |
538 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
539 ret= av_get_packet(s->pb, pkt, size); |
234 | 540 |
541 pkt->stream_index = 0; | |
542 if (ret <= 0) { | |
2274
b21c2af60bc9
Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents:
2273
diff
changeset
|
543 return AVERROR(EIO); |
234 | 544 } |
545 /* note: we need to modify the packet size here to handle the last | |
546 packet */ | |
547 pkt->size = ret; | |
548 return ret; | |
549 } | |
550 | |
551 static int mp3_read_close(AVFormatContext *s) | |
552 { | |
553 return 0; | |
554 } | |
555 | |
858
66cc656ea404
Replace CONFIG_ENCODERS/CONFIG_DECODERS with CONFIG_MUXERS/CONFIG_DEMUXERS
diego
parents:
820
diff
changeset
|
556 #ifdef CONFIG_MUXERS |
234 | 557 /* simple formats */ |
2098 | 558 |
559 static void id3v2_put_size(AVFormatContext *s, int size) | |
560 { | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
561 put_byte(s->pb, size >> 21 & 0x7f); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
562 put_byte(s->pb, size >> 14 & 0x7f); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
563 put_byte(s->pb, size >> 7 & 0x7f); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
564 put_byte(s->pb, size & 0x7f); |
2098 | 565 } |
566 | |
567 static void id3v2_put_ttag(AVFormatContext *s, char *string, uint32_t tag) | |
568 { | |
569 int len = strlen(string); | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
570 put_be32(s->pb, tag); |
2098 | 571 id3v2_put_size(s, len + 1); |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
572 put_be16(s->pb, 0); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
573 put_byte(s->pb, 3); /* UTF-8 */ |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
574 put_buffer(s->pb, string, len); |
2098 | 575 } |
576 | |
577 | |
578 /** | |
579 * Write an ID3v2.4 header at beginning of stream | |
580 */ | |
581 | |
234 | 582 static int mp3_write_header(struct AVFormatContext *s) |
583 { | |
2098 | 584 int totlen = 0; |
585 char tracktxt[10]; | |
2735 | 586 char yeartxt[10]; |
2098 | 587 |
588 if(s->track) | |
2746 | 589 snprintf(tracktxt, sizeof(tracktxt), "%d", s->track); |
2735 | 590 if(s->year) |
2746 | 591 snprintf( yeartxt, sizeof(yeartxt) , "%d", s->year ); |
2098 | 592 |
593 if(s->title[0]) totlen += 11 + strlen(s->title); | |
594 if(s->author[0]) totlen += 11 + strlen(s->author); | |
595 if(s->album[0]) totlen += 11 + strlen(s->album); | |
596 if(s->genre[0]) totlen += 11 + strlen(s->genre); | |
597 if(s->copyright[0]) totlen += 11 + strlen(s->copyright); | |
598 if(s->track) totlen += 11 + strlen(tracktxt); | |
2735 | 599 if(s->year) totlen += 11 + strlen(yeartxt); |
2098 | 600 if(!(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) |
601 totlen += strlen(LIBAVFORMAT_IDENT) + 11; | |
602 | |
603 if(totlen == 0) | |
604 return 0; | |
605 | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
606 put_be32(s->pb, MKBETAG('I', 'D', '3', 0x04)); /* ID3v2.4 */ |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
607 put_byte(s->pb, 0); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
608 put_byte(s->pb, 0); /* flags */ |
2098 | 609 |
610 id3v2_put_size(s, totlen); | |
611 | |
612 if(s->title[0]) id3v2_put_ttag(s, s->title, MKBETAG('T', 'I', 'T', '2')); | |
613 if(s->author[0]) id3v2_put_ttag(s, s->author, MKBETAG('T', 'P', 'E', '1')); | |
614 if(s->album[0]) id3v2_put_ttag(s, s->album, MKBETAG('T', 'A', 'L', 'B')); | |
615 if(s->genre[0]) id3v2_put_ttag(s, s->genre, MKBETAG('T', 'C', 'O', 'N')); | |
616 if(s->copyright[0]) id3v2_put_ttag(s, s->copyright, MKBETAG('T', 'C', 'O', 'P')); | |
617 if(s->track) id3v2_put_ttag(s, tracktxt, MKBETAG('T', 'R', 'C', 'K')); | |
2735 | 618 if(s->year) id3v2_put_ttag(s, yeartxt, MKBETAG('T', 'Y', 'E', 'R')); |
2098 | 619 if(!(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) |
620 id3v2_put_ttag(s, LIBAVFORMAT_IDENT, MKBETAG('T', 'E', 'N', 'C')); | |
234 | 621 return 0; |
622 } | |
623 | |
468 | 624 static int mp3_write_packet(struct AVFormatContext *s, AVPacket *pkt) |
234 | 625 { |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
626 put_buffer(s->pb, pkt->data, pkt->size); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
627 put_flush_packet(s->pb); |
234 | 628 return 0; |
629 } | |
630 | |
631 static int mp3_write_trailer(struct AVFormatContext *s) | |
632 { | |
2096 | 633 uint8_t buf[ID3v1_TAG_SIZE]; |
234 | 634 |
2096 | 635 /* write the id3v1 tag */ |
234 | 636 if (s->title[0] != '\0') { |
2096 | 637 id3v1_create_tag(s, buf); |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
638 put_buffer(s->pb, buf, ID3v1_TAG_SIZE); |
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2746
diff
changeset
|
639 put_flush_packet(s->pb); |
234 | 640 } |
641 return 0; | |
642 } | |
858
66cc656ea404
Replace CONFIG_ENCODERS/CONFIG_DECODERS with CONFIG_MUXERS/CONFIG_DEMUXERS
diego
parents:
820
diff
changeset
|
643 #endif //CONFIG_MUXERS |
234 | 644 |
1169 | 645 #ifdef CONFIG_MP3_DEMUXER |
1167 | 646 AVInputFormat mp3_demuxer = { |
234 | 647 "mp3", |
648 "MPEG audio", | |
649 0, | |
1107 | 650 mp3_read_probe, |
234 | 651 mp3_read_header, |
652 mp3_read_packet, | |
653 mp3_read_close, | |
1756 | 654 .flags= AVFMT_GENERIC_INDEX, |
814 | 655 .extensions = "mp2,mp3,m2a", /* XXX: use probe */ |
234 | 656 }; |
1169 | 657 #endif |
658 #ifdef CONFIG_MP2_MUXER | |
1167 | 659 AVOutputFormat mp2_muxer = { |
234 | 660 "mp2", |
661 "MPEG audio layer 2", | |
662 "audio/x-mpeg", | |
1622 | 663 #ifdef CONFIG_LIBMP3LAME |
814 | 664 "mp2,m2a", |
234 | 665 #else |
814 | 666 "mp2,mp3,m2a", |
234 | 667 #endif |
668 0, | |
669 CODEC_ID_MP2, | |
670 0, | |
2306
d045cbcf37a3
do not write id3v2 header when writing .mp2 files
bcoudurier
parents:
2274
diff
changeset
|
671 NULL, |
234 | 672 mp3_write_packet, |
673 mp3_write_trailer, | |
674 }; | |
1169 | 675 #endif |
676 #ifdef CONFIG_MP3_MUXER | |
1167 | 677 AVOutputFormat mp3_muxer = { |
234 | 678 "mp3", |
679 "MPEG audio layer 3", | |
680 "audio/x-mpeg", | |
681 "mp3", | |
682 0, | |
683 CODEC_ID_MP3, | |
684 0, | |
685 mp3_write_header, | |
686 mp3_write_packet, | |
687 mp3_write_trailer, | |
688 }; | |
689 #endif |