diff options
Diffstat (limited to 'apps/metadata/smaf.c')
-rw-r--r-- | apps/metadata/smaf.c | 369 |
1 files changed, 369 insertions, 0 deletions
diff --git a/apps/metadata/smaf.c b/apps/metadata/smaf.c new file mode 100644 index 0000000000..f7d825c96a --- /dev/null +++ b/apps/metadata/smaf.c | |||
@@ -0,0 +1,369 @@ | |||
1 | /*************************************************************************** | ||
2 | * __________ __ ___. | ||
3 | * Open \______ \ ____ ____ | | _\_ |__ _______ ___ | ||
4 | * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / | ||
5 | * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < | ||
6 | * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ | ||
7 | * \/ \/ \/ \/ \/ | ||
8 | * $Id$ | ||
9 | * | ||
10 | * Copyright (C) 2010 Yoshihisa Uchida | ||
11 | * | ||
12 | * This program is free software; you can redistribute it and/or | ||
13 | * modify it under the terms of the GNU General Public License | ||
14 | * as published by the Free Software Foundation; either version 2 | ||
15 | * of the License, or (at your option) any later version. | ||
16 | * | ||
17 | * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY | ||
18 | * KIND, either express or implied. | ||
19 | * | ||
20 | ****************************************************************************/ | ||
21 | #include <stdio.h> | ||
22 | #include <string.h> | ||
23 | #include <stdlib.h> | ||
24 | #include <ctype.h> | ||
25 | #include <inttypes.h> | ||
26 | |||
27 | #include "system.h" | ||
28 | #include "metadata.h" | ||
29 | #include "metadata_common.h" | ||
30 | #include "metadata_parsers.h" | ||
31 | #include "rbunicode.h" | ||
32 | #include "logf.h" | ||
33 | |||
34 | static int basebits[4] = { 4, 8, 12, 16 }; | ||
35 | |||
36 | static int frequency[5] = { 4000, 8000, 11025, 22050, 44100 }; | ||
37 | |||
38 | static int support_codepages[7] = { | ||
39 | SJIS, ISO_8859_1, -1, GB_2312, BIG_5, -1, -1, | ||
40 | }; | ||
41 | |||
42 | /* extra codepage */ | ||
43 | #define UCS_2 (NUM_CODEPAGES + 1) | ||
44 | #define UTF_16 (NUM_CODEPAGES + 2) | ||
45 | |||
46 | /* support id3 tag */ | ||
47 | #define TAG_TITLE (('S'<<8)|'T') | ||
48 | #define TAG_ARTIST (('A'<<8)|'N') | ||
49 | #define TAG_COMPOSER (('S'<<8)|'W') | ||
50 | |||
51 | static unsigned char smafbuf[1024]; | ||
52 | |||
53 | static bool read_datachunk(unsigned char *src, int size, unsigned short tag, | ||
54 | int codepage, unsigned char *dst) | ||
55 | { | ||
56 | int datasize = 0; | ||
57 | unsigned char *utf8; | ||
58 | |||
59 | while(size > datasize + 4) | ||
60 | { | ||
61 | datasize = (src[2] << 8) | src[3]; | ||
62 | if (tag == ((src[0] << 8) | src[1])) | ||
63 | { | ||
64 | src += 4; | ||
65 | if (codepage < NUM_CODEPAGES) | ||
66 | utf8 = iso_decode(src, dst, codepage, datasize); | ||
67 | else /* codepage == UTF_16, UCS_2 */ | ||
68 | utf8 = utf16BEdecode(src, dst, datasize); | ||
69 | *utf8 = '\0'; | ||
70 | |||
71 | return true; | ||
72 | } | ||
73 | src += (datasize + 4); | ||
74 | } | ||
75 | return false; | ||
76 | } | ||
77 | |||
78 | static bool read_option(unsigned char *src, int size, unsigned short tag, | ||
79 | int codepage, unsigned char *dst) | ||
80 | { | ||
81 | int datasize = 0; | ||
82 | unsigned char *endsrc = src + size; | ||
83 | unsigned char *utf8; | ||
84 | |||
85 | while(src < endsrc) | ||
86 | { | ||
87 | utf8 = src; | ||
88 | src += 3; | ||
89 | datasize = 0; | ||
90 | while (*src != ',' || *(src-1) == '\\') | ||
91 | { | ||
92 | datasize++; | ||
93 | src++; | ||
94 | } | ||
95 | if (tag == ((utf8[0] << 8) | utf8[1]) && utf8[2] == ':') | ||
96 | { | ||
97 | utf8 += 3; | ||
98 | if (codepage < NUM_CODEPAGES) | ||
99 | utf8 = iso_decode(utf8, dst, codepage, datasize); | ||
100 | else /* codepage == UTF_16, UCS_2 */ | ||
101 | utf8 = utf16BEdecode(utf8, dst, datasize); | ||
102 | *utf8 = '\0'; | ||
103 | |||
104 | return true; | ||
105 | } | ||
106 | src++; | ||
107 | } | ||
108 | return false; | ||
109 | } | ||
110 | |||
111 | static int convert_smaf_audio_frequency(unsigned int freq) | ||
112 | { | ||
113 | if (freq > 4) | ||
114 | return 0; | ||
115 | return frequency[freq]; | ||
116 | } | ||
117 | |||
118 | static int convert_smaf_audio_basebit(unsigned int basebit) | ||
119 | { | ||
120 | if (basebit > 4) | ||
121 | return 0; | ||
122 | return basebits[basebit]; | ||
123 | } | ||
124 | |||
125 | static int convert_smaf_codetype(unsigned int codetype) | ||
126 | { | ||
127 | if (codetype < 7) | ||
128 | return support_codepages[codetype]; | ||
129 | else if (codetype < 0x20) | ||
130 | return -1; | ||
131 | else if (codetype == 0x20) | ||
132 | return UCS_2; | ||
133 | else if (codetype == 0x23) | ||
134 | return UTF_8; | ||
135 | else if (codetype == 0x24) | ||
136 | return UTF_16; | ||
137 | else if (codetype == 0xff) | ||
138 | return ISO_8859_1; | ||
139 | return -1; | ||
140 | } | ||
141 | |||
142 | static bool get_smaf_metadata_audio_track(struct mp3entry *id3, | ||
143 | unsigned char* buf, unsigned char *endbuf) | ||
144 | { | ||
145 | int bitspersample; | ||
146 | int channels; | ||
147 | int chunksize; | ||
148 | long numbytes; | ||
149 | unsigned long totalsamples; | ||
150 | |||
151 | channels = ((buf[10] & 0x80) >> 7) + 1; | ||
152 | bitspersample = convert_smaf_audio_basebit(buf[11] >> 4); | ||
153 | if (bitspersample == 0) | ||
154 | { | ||
155 | DEBUGF("metada error: smaf unsupport basebit %d\n", buf[11] >> 4); | ||
156 | return false; | ||
157 | } | ||
158 | id3->frequency = convert_smaf_audio_frequency(buf[10] & 0x0f); | ||
159 | |||
160 | buf += 14; | ||
161 | while (buf < endbuf) | ||
162 | { | ||
163 | chunksize = get_long_be(buf + 4) + 8; | ||
164 | if (memcmp(buf, "Awa", 3) == 0) | ||
165 | { | ||
166 | numbytes = get_long_be(buf + 4) - 3; | ||
167 | totalsamples = (numbytes << 3) / (bitspersample * channels); | ||
168 | |||
169 | /* Calculate track length (in ms) and estimate the bitrate (in kbit/s) */ | ||
170 | id3->length = ((int64_t)totalsamples * 1000LL) / id3->frequency; | ||
171 | |||
172 | return true; | ||
173 | } | ||
174 | buf += chunksize; | ||
175 | } | ||
176 | DEBUGF("metada error: smaf does not include pcm audio data\n"); | ||
177 | return false; | ||
178 | } | ||
179 | |||
180 | static bool get_smaf_metadata_score_track(struct mp3entry *id3, | ||
181 | unsigned char* buf, unsigned char *endbuf) | ||
182 | { | ||
183 | int bitspersample; | ||
184 | int channels; | ||
185 | int chunksize; | ||
186 | long numbytes; | ||
187 | unsigned long totalsamples; | ||
188 | |||
189 | /* | ||
190 | * skip to the next chunk. | ||
191 | * MA-2/MA-3/MA-5: padding 16 bytes | ||
192 | * MA-7: padding 32 bytes | ||
193 | */ | ||
194 | if (buf[3] < 7) | ||
195 | buf += 28; | ||
196 | else | ||
197 | buf += 44; | ||
198 | |||
199 | while (buf + 10 < endbuf) | ||
200 | { | ||
201 | chunksize = get_long_be(buf + 4) + 8; | ||
202 | if (memcmp(buf, "Mtsp", 4) == 0) | ||
203 | { | ||
204 | buf += 8; | ||
205 | if (memcmp(buf, "Mwa", 3) != 0) | ||
206 | { | ||
207 | DEBUGF("metada error: smaf unsupport format: %c%c%c%c\n", | ||
208 | buf[0], buf[1], buf[2], buf[3]); | ||
209 | return false; | ||
210 | } | ||
211 | |||
212 | channels = ((buf[8] & 0x80) >> 7) + 1; | ||
213 | bitspersample = convert_smaf_audio_basebit(buf[8] & 0x0f); | ||
214 | if (bitspersample == 0) | ||
215 | { | ||
216 | DEBUGF("metada error: smaf unsupport basebit %d\n", buf[8] & 0x0f); | ||
217 | return false; | ||
218 | } | ||
219 | |||
220 | numbytes = get_long_be(buf + 4) - 3; | ||
221 | totalsamples = numbytes * 8 / (bitspersample * channels); | ||
222 | |||
223 | id3->frequency = (buf[9] << 8) | buf[10]; | ||
224 | |||
225 | /* Calculate track length (in ms) and estimate the bitrate (in kbit/s) */ | ||
226 | id3->length = ((int64_t) totalsamples * 1000) / id3->frequency; | ||
227 | |||
228 | return true; | ||
229 | } | ||
230 | buf += chunksize; | ||
231 | } | ||
232 | DEBUGF("metada error: smaf does not include pcm audio data\n"); | ||
233 | return false; | ||
234 | } | ||
235 | |||
236 | bool get_smaf_metadata(int fd, struct mp3entry* id3) | ||
237 | { | ||
238 | /* Use the trackname part of the id3 structure as a temporary buffer */ | ||
239 | unsigned char* buf = (unsigned char *)id3->path; | ||
240 | unsigned char *endbuf = smafbuf + 1024; | ||
241 | int i; | ||
242 | int contents_size; | ||
243 | int codepage = ISO_8859_1; | ||
244 | |||
245 | id3->title = NULL; | ||
246 | id3->artist = NULL; | ||
247 | id3->composer = NULL; | ||
248 | |||
249 | id3->vbr = false; /* All SMAF files are CBR */ | ||
250 | id3->filesize = filesize(fd); | ||
251 | |||
252 | /* get RIFF chunk header */ | ||
253 | if ((lseek(fd, 0, SEEK_SET) < 0) || (read(fd, buf, 21) < 21)) | ||
254 | { | ||
255 | return false; | ||
256 | } | ||
257 | |||
258 | if ((memcmp(buf, "MMMD", 4) != 0) || (memcmp(&buf[8], "CNTI", 4) != 0)) | ||
259 | { | ||
260 | DEBUGF("metada error: does not smaf format\n"); | ||
261 | return false; | ||
262 | } | ||
263 | |||
264 | contents_size = get_long_be(buf + 12); | ||
265 | if (contents_size < 5) | ||
266 | { | ||
267 | DEBUGF("metada error: CNTI chunk size is small %d\n", contents_size); | ||
268 | return false; | ||
269 | } | ||
270 | |||
271 | contents_size -= 5; | ||
272 | i = contents_size; | ||
273 | if (i == 0) | ||
274 | { | ||
275 | read(fd, buf, 16); | ||
276 | if (memcmp(buf, "OPDA", 4) != 0) | ||
277 | { | ||
278 | DEBUGF("metada error: smaf does not include OPDA chunk\n"); | ||
279 | return false; | ||
280 | } | ||
281 | contents_size = get_long_be(buf + 4) - 8; | ||
282 | |||
283 | if (memcmp(buf + 8, "Dch", 3) != 0) | ||
284 | { | ||
285 | DEBUGF("metada error: smaf does not include Dch chunk\n"); | ||
286 | return false; | ||
287 | } | ||
288 | codepage = convert_smaf_codetype(buf[11]); | ||
289 | if (codepage < 0) | ||
290 | { | ||
291 | DEBUGF("metada error: smaf unsupport codetype: %d\n", buf[11]); | ||
292 | return false; | ||
293 | } | ||
294 | |||
295 | i = get_long_be(buf + 12); | ||
296 | |||
297 | if (i > MAX_PATH) | ||
298 | { | ||
299 | DEBUGF("metada warning: smaf contents size is big %d\n", i); | ||
300 | i = MAX_PATH; | ||
301 | } | ||
302 | if (read(fd, buf, i) < i) | ||
303 | return false; | ||
304 | |||
305 | /* title */ | ||
306 | if (read_datachunk(buf, i, TAG_TITLE, codepage, id3->id3v1buf[0])) | ||
307 | id3->title = id3->id3v1buf[0]; | ||
308 | |||
309 | /* artist */ | ||
310 | if (read_datachunk(buf, i, TAG_ARTIST, codepage, id3->id3v1buf[1])) | ||
311 | id3->artist = id3->id3v1buf[1]; | ||
312 | |||
313 | /* composer */ | ||
314 | if (read_datachunk(buf, i, TAG_COMPOSER, codepage, id3->id3v1buf[2])) | ||
315 | id3->composer = id3->id3v1buf[2]; | ||
316 | } | ||
317 | else | ||
318 | { | ||
319 | codepage = convert_smaf_codetype(buf[14]); | ||
320 | if (codepage < 0) | ||
321 | { | ||
322 | DEBUGF("metada error: smaf unsupport codetype: %d\n", buf[11]); | ||
323 | return false; | ||
324 | } | ||
325 | |||
326 | if (i > MAX_PATH) | ||
327 | { | ||
328 | DEBUGF("metada warning: smaf contents size is big %d\n", i); | ||
329 | i = MAX_PATH; | ||
330 | } | ||
331 | if (read(fd, buf, i) < i) | ||
332 | return false; | ||
333 | |||
334 | /* title */ | ||
335 | if (read_option(buf, i, TAG_TITLE, codepage, id3->id3v1buf[0])) | ||
336 | id3->title = id3->id3v1buf[0]; | ||
337 | |||
338 | /* artist */ | ||
339 | if (read_option(buf, i, TAG_ARTIST, codepage, id3->id3v1buf[1])) | ||
340 | id3->artist = id3->id3v1buf[1]; | ||
341 | |||
342 | /* composer */ | ||
343 | if (read_option(buf, i, TAG_COMPOSER, codepage, id3->id3v1buf[2])) | ||
344 | id3->composer = id3->id3v1buf[2]; | ||
345 | } | ||
346 | |||
347 | if (contents_size > i) | ||
348 | lseek(fd, contents_size - i, SEEK_CUR); | ||
349 | |||
350 | /* assume the SMAF pcm data position is less than 1024 bytes */ | ||
351 | if (read(fd, smafbuf, 1024) < 1024) | ||
352 | return false; | ||
353 | |||
354 | buf = smafbuf; | ||
355 | while (buf + 8 < endbuf) | ||
356 | { | ||
357 | i = get_long_be(buf + 4) + 8; | ||
358 | |||
359 | if (memcmp(buf, "ATR", 3) == 0) | ||
360 | return get_smaf_metadata_audio_track(id3, buf, endbuf); | ||
361 | else if (memcmp(buf, "MTR", 3) == 0) | ||
362 | return get_smaf_metadata_score_track(id3, buf, endbuf); | ||
363 | |||
364 | buf += i; | ||
365 | } | ||
366 | |||
367 | DEBUGF("metada error: smaf does not include track chunk\n"); | ||
368 | return false; | ||
369 | } | ||