initial support for HVSC song length database

1 files changed, 217 insertions, 3 deletions

diff --git a/csid.cpp b/csid.cpp
index 17503b75..2752114a 100644
--- a/csid.cpp
+++ b/csid.cpp
@@ -7,13 +7,182 @@
 extern "C" {
 #include "codec.h"
 #include "playlist.h"
-}
 #include "csid.h"
+#include "md5.h"
+}
+
+#define MAX_SID_SONGS 65536
+#define MAX_SID_SUBSONGS 20
 
 static sidplay2 *sidplay;
 static ReSIDBuilder *resid;
 static SidTune *tune;
 extern int sdl_player_freq; // hack!
+static uint8_t sldb_digests[MAX_SID_SONGS][16];
+static float sldb_lengths[MAX_SID_SONGS][MAX_SID_SUBSONGS];
+static int sldb_size;
+static int sldb_loaded;
+static const char *sldb_fname = "/home/waker/hvsc/C64Music/DOCUMENTS/Songlengths.txt";
+
+static void sldb_load(const char *fname)
+{
+    if (sldb_loaded) {
+        return;
+    }
+    sldb_loaded = 1;
+    FILE *fp = fopen (fname, "r");
+    if (!fp) {
+        return;
+    }
+    char str[1024];
+
+    int line = 1;
+    if (fgets (str, 1024, fp) != str) {
+        goto fail; // eof
+    }
+    if (strncmp (str, "[Database]", 10)) {
+        goto fail; // bad format
+    }
+
+    while (fgets (str, 1024, fp) == str) {
+        line++;
+        if (str[0] == ';') {
+//            printf ("reading songlength for %s", str);
+            continue; // comment
+        }
+        // read/validate md5
+        const char *p = str;
+        uint8_t digest[16];
+        int sz = 0;
+        char byte[3];
+        while (*p && sz < 16) {
+            byte[0] = tolower (*p);
+            if (!((byte[0] >= '0' && byte[0] <= '9') || (byte[0] >= 'a' && byte[0] <= 'f'))) {
+                printf ("invalid byte 0 in md5: %c\n", byte[0]);
+                break;
+            }
+            p++;
+            if (!(*p)) {
+                break;
+            }
+            byte[1] = tolower (*p);
+            if (!((byte[1] >= '0' && byte[1] <= '9') || (byte[1] >= 'a' && byte[1] <= 'f'))) {
+                printf ("invalid byte 1 in md5: %c\n", byte[1]);
+                break;
+            }
+            byte[2] = 0;
+            p++;
+
+            // convert from ascii hex to uint8_t
+            if (byte[1] < 'a') {
+                digest[sz] = byte[1] - '0';
+            }
+            else {
+                digest[sz] = byte[1] - 'a' + 10;
+            }
+            if (byte[0] < 'a') {
+                digest[sz] |= (byte[0] - '0') << 4;
+            }
+            else {
+                digest[sz] |= (byte[0] - 'a' + 10) << 4;
+            }
+            sz++;
+        }
+        if (sz < 16) {
+            printf ("bad md5 (sz=%d, line=%d)\n", sz, line);
+            continue; // bad song md5
+        }
+//        else {
+//            printf ("digest: ");
+//            for (int j = 0; j < 16; j++) {
+//                printf ("%02x", (int)digest[j]);
+//            }
+//            printf ("\n");
+//            exit (0);
+//        }
+        memcpy (sldb_digests[sldb_size], digest, 16);
+        for (int k = 0; k < MAX_SID_SUBSONGS; k++) {
+            sldb_lengths[sldb_size][k] = -1;
+        }
+        sldb_size++;
+        // check '=' sign
+        if (*p != '=') {
+            continue; // no '=' sign
+        }
+        p++;
+        if (!(*p)) {
+            continue; // unexpected eol
+        }
+        int subsong = 0;
+        while (*p >= ' ') {
+            // read subsong lengths until eol
+            char timestamp[7]; // up to MMM:SS
+            sz = 0;
+            while (*p > ' ' && *p != '(' && sz < 7) {
+                timestamp[sz++] = *p;
+                p++;
+            }
+            if (sz < 4 || sz == 6 && *p > ' ' && *p != '(') {
+                break; // bad timestamp
+            }
+            timestamp[sz] = 0;
+            // check for unknown time
+            float time = -1;
+            if (!strcmp (timestamp, "-:--")) {
+                time = -1;
+            }
+            else {
+                // parse timestamp
+                const char *colon = strchr (timestamp, ':');
+                if (!colon) {
+                    break; // bad timestamp
+                }
+                // minute
+                char minute[4];
+                strncpy (minute, timestamp, colon-timestamp);
+                minute[colon-timestamp] = 0;
+                // second
+                char second[3];
+                strncpy (second, colon+1, 3);
+                //printf ("subsong %d, time %s:%s\n", subsong, minute, second);
+                time = atoi (minute) * 60 + atoi (second);
+                sldb_lengths[sldb_size-1][subsong] = time;
+                subsong++;
+            }
+
+            // prepare for next timestamp
+            if (*p == '(') {
+                // skip until next whitespace
+                while (*p > ' ') {
+                    p++;
+                }
+            }
+            if (*p < ' ') {
+                break; // eol
+            }
+            // skip white spaces
+            while (*p == ' ') {
+                p++;
+            }
+            if (*p < ' ') {
+                break; // eol
+            }
+        }
+    }
+
+fail:
+    fclose (fp);
+}
+
+static int
+sldb_find (const uint8_t *digest) {
+    for (int i = 0; i < sldb_size; i++) {
+        if (!memcmp (digest, sldb_digests[i], 16)) {
+            return i;
+        }
+    }
+    return -1;
+}
 
 extern "C" int
 csid_init (const char *fname, int track, float start, float end) {
@@ -23,6 +192,28 @@ csid_init (const char *fname, int track, float start, float end) {
     resid->filter (true);
     resid->sampling (sdl_player_freq);
     tune = new SidTune (fname);
+    // calc md5
+    uint8_t sig[16];
+    md5_t md5;
+    md5_init (&md5);
+    md5_process (&md5, (const char *)tune->cache.get () + tune->fileOffset, tune->getInfo ().c64dataLen);
+    uint16_t tmp;
+    tmp = tune->getInfo ().initAddr;
+    md5_process (&md5, &tmp, 2);
+    tmp = tune->getInfo ().playAddr;
+    md5_process (&md5, &tmp, 2);
+    tmp = tune->getInfo ().songs;
+    md5_process (&md5, &tmp, 2);
+    for (int s = 1; s <= tune->getInfo ().songs; s++)
+    {
+        tune->selectSong (s);
+        md5_process (&md5, &tune->getInfo ().songSpeed, sizeof (tune->getInfo ().songSpeed));
+    }
+    if (tune->getInfo ().clockSpeed == SIDTUNE_CLOCK_NTSC) {
+        md5_process (&md5, &tune->getInfo ().clockSpeed, sizeof (tune->getInfo ().clockSpeed));
+    }
+    md5_finish (&md5, sig);
+
     tune->selectSong (track+1);
     csid.info.channels = tune->isStereo () ? 2 : 1;
     sid2_config_t conf;
@@ -36,8 +227,23 @@ csid_init (const char *fname, int track, float start, float end) {
     sidplay->load (tune);
     csid.info.bitsPerSample = 16;
     csid.info.samplesPerSecond = sdl_player_freq;
-    csid.info.duration = 120;
     csid.info.position = 0;
+    float length = 120;
+    sldb_load(sldb_fname);
+    if (sldb_loaded) {
+        int song = sldb_find (sig);
+        if (song >= 0 && sldb_lengths[song][track] >= 0) {
+            length = sldb_lengths[song][track];
+        }
+//        if (song < 0) {
+//            printf ("song %s not found in db, md5: ", fname);
+//            for (int j = 0; j < 16; j++) {
+//                printf ("%02x", (int)sig[j]);
+//            }
+//            printf ("\n");
+//        }
+    }
+    csid.info.duration = length;
 
     return 0;
 }
@@ -54,7 +260,11 @@ csid_free (void) {
 
 extern "C" int
 csid_read (char *bytes, int size) {
+    if (csid.info.position > csid.info.duration) {
+        return 0;
+    }
     int rd = sidplay->play (bytes, size/csid.info.channels);
+    csid.info.position += size/csid.info.channels/2 / (float)csid.info.samplesPerSecond;
     return rd * csid.info.channels;
 }
 
@@ -87,11 +297,13 @@ convstr (const char* str) {
 
 extern "C" int
 csid_add (const char *fname) {
+    sldb_load(sldb_fname);
     SidTune *tune;
     tune = new SidTune (fname);
     int tunes = tune->getInfo ().songs;
     for (int s = 0; s < tunes; s++) {
         if (tune->selectSong (s+1)) {
+            unsigned length = 120;
             playItem_t *it = (playItem_t*)malloc (sizeof (playItem_t));
             memset (it, 0, sizeof (playItem_t));
             it->codec = &csid;
@@ -104,9 +316,11 @@ csid_add (const char *fname) {
             int i = sidinfo.numberOfCommentStrings;
             if (i >= 1 && sidinfo.infoString[0] && sidinfo.infoString[0][0]) {
                 ps_add_meta (it, sidinfo.songs > 1 ? "album" : "title", convstr (sidinfo.infoString[0]));
+                //printf ("album/title: %s\n", convstr (sidinfo.infoString[0]));
             }
             if (i >= 2 && sidinfo.infoString[1] && sidinfo.infoString[1][0]) {
                 ps_add_meta (it, "artist", convstr (sidinfo.infoString[1]));
+                //printf ("artist: %s\n", convstr (sidinfo.infoString[1]));
             }
             if (i >= 3 && sidinfo.infoString[2] && sidinfo.infoString[2][0]) {
                 ps_add_meta (it, "copyright", convstr (sidinfo.infoString[2]));
@@ -128,7 +342,7 @@ csid_add (const char *fname) {
     return 0;
 }
 
-static const char * exts[]=
+static const char *exts[]=
 {
 	"sid",NULL
 };