apps/metadata/smaf.c

   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 }