ref: b818f48f41895754dd9b470be407ba85cfea45bd
dir: /sys/src/cmd/audio/zuke/mkplist.c/
#include <u.h> #include <libc.h> #include <bio.h> #include <tags.h> #include <thread.h> #include "plist.h" #include "icy.h" typedef struct Aux Aux; enum { Maxname = 256+2, /* seems enough? */ Maxdepth = 16, /* max recursion depth */ }; #define MAX(a, b) (a > b ? a : b) struct Aux { Meta; Biobuf *f; int firstiscomposer; int keepfirstartist; }; int mainstacksize = 32768; static int simplesort; static int moddec; static Channel *cmeta; static Channel *cpath; static Meta **tracks; static int ntracks; static char *fmts[] = { [Fmp3] = "mp3", [Fogg] = "ogg", [Fflac] = "flac", [Fm4a] = "m4a", [Fopus] = "opus", [Fwav] = "wav", [Fit] = "mod", [Fxm] = "mod", [Fs3m] = "mod", [Fmod] = "mod", }; static void metathread(void *) { int max; Meta *m; max = 0; for(;;){ if((m = recvp(cmeta)) == nil) break; if(ntracks+1 > max){ max = max ? max*2 : 1024; tracks = realloc(tracks, sizeof(Meta*)*max); } tracks[ntracks++] = m; } threadexits(nil); } static void cb(Tagctx *ctx, int t, const char *k, const char *v, int offset, int size, Tagread f) { int i, iscomposer; Aux *aux; aux = ctx->aux; switch(t){ case Tartist: if(aux->numartist < Maxartist){ iscomposer = strcmp(k, "TCM") == 0 || strcmp(k, "TCOM") == 0; /* prefer lead performer/soloist, helps when TP2/TPE2 is the first one and is set to "VA" */ /* always put composer first, if available */ if(iscomposer || (!aux->keepfirstartist && (strcmp(k, "TP1") == 0 || strcmp(k, "TPE1") == 0))){ if(aux->numartist > 0) aux->artist[aux->numartist] = aux->artist[aux->numartist-1]; aux->artist[0] = strdup(v); aux->numartist++; aux->keepfirstartist = 1; aux->firstiscomposer = iscomposer; return; } for(i = 0; i < aux->numartist; i++){ if(cistrcmp(aux->artist[i], v) == 0) return; } aux->artist[aux->numartist++] = strdup(v); } break; case Talbum: if(aux->album == nil) aux->album = strdup(v); break; case Ttitle: if(aux->title == nil) aux->title = strdup(v); break; case Tdate: if(aux->date == nil) aux->date = strdup(v); break; case Ttrack: if(aux->track == nil) aux->track = strdup(v); break; case Timage: if(aux->imagefmt == nil){ aux->imagefmt = strdup(v); aux->imageoffset = offset; aux->imagesize = size; aux->imagereader = f != nil; } break; case Ttrackgain: aux->rgtrack = atof(v); if(strncmp(k, "R128_", 5) == 0) aux->rgtrack /= 256.0; break; case Talbumgain: aux->rgalbum = atof(v); if(strncmp(k, "R128_", 5) == 0) aux->rgalbum /= 256.0; break; } } static int ctxread(Tagctx *ctx, void *buf, int cnt) { return Bread(((Aux*)ctx->aux)->f, buf, cnt); } static int ctxseek(Tagctx *ctx, int offset, int whence) { return Bseek(((Aux*)ctx->aux)->f, offset, whence); } static uvlong modduration(char *path) { int f, pid, p[2], n; char t[1024], *s; if(!moddec) return 0; pipe(p); if((pid = rfork(RFPROC|RFFDG|RFNOTEG|RFCENVG|RFNOWAIT)) == 0){ dup(f = open(path, OREAD), 0); close(f); close(1); dup(p[1], 2); close(p[1]); close(p[0]); execl("/bin/audio/moddec", "moddec", "-r", "0", nil); sysfatal("execl: %r"); } close(p[1]); n = pid > 0 ? readn(p[0], t, sizeof(t)-1) : -1; close(p[0]); if(n > 0){ t[n] = 0; for(s = t; s != nil; s = strchr(s+1, '\n')){ if(*s == '\n') s++; if(strncmp(s, "duration: ", 10) == 0) return strtod(s+10, nil)*1000.0; } } return 0; } static Meta * scanfile(char *path) { char buf[4096], *s; Aux aux = {0}; Tagctx ctx = { .read = ctxread, .seek = ctxseek, .tag = cb, .buf = buf, .bufsz = sizeof(buf), .aux = &aux, }; int res; Meta *m; if((aux.f = Bopen(path, OREAD)) == nil){ fprint(2, "%s: %r\n", path); return nil; } res = tagsget(&ctx); Bterm(aux.f); if(ctx.format == Funknown) return nil; if(ctx.format >= nelem(fmts)) sysfatal("mkplist needs a rebuild with updated libtags"); m = malloc(sizeof(*m)); memmove(m, &aux.Meta, sizeof(*m)); if(res != 0) fprint(2, "%s: no tags\n", path); if(ctx.duration == 0){ if(ctx.format == Fit || ctx.format == Fxm || ctx.format == Fs3m || ctx.format == Fmod) ctx.duration = modduration(path); if(ctx.duration == 0) fprint(2, "%s: no duration\n", path); } m->duration = ctx.duration; if(m->title == nil){ if((s = utfrrune(path, '/')) == nil) s = path; m->title = strdup(s+1); } m->path = strdup(path); m->filefmt = fmts[ctx.format]; return m; } static void tagreadproc(void *cexit) { char *path; Meta *m; for(;;){ if(recv(cpath, &path) != 1) break; if((m = scanfile(path)) != nil) sendp(cmeta, m); free(path); } sendul(cexit, 0); threadexits(nil); } static int scan(char **dir, int depth) { char *path; Dir *buf, *d; long n; int dirfd, len; if((dirfd = open(*dir, OREAD)) < 0){ fprint(2, "scan: %r\n"); return -1; } len = strlen(*dir); if((*dir = realloc(*dir, len+1+Maxname)) == nil) sysfatal("scan: no memory"); path = *dir; path[len] = '/'; for(n = 0, buf = nil; n >= 0;){ if((n = dirread(dirfd, &buf)) < 0){ path[len] = 0; sendp(cpath, strdup(path)); break; } if(n == 0){ free(buf); break; } for(d = buf; n > 0; n--, d++){ if(strcmp(d->name, ".") == 0 || strcmp(d->name, "..") == 0) continue; path[len+1+Maxname-2] = 0; strncpy(&path[len+1], d->name, Maxname); if(path[len+1+Maxname-2] != 0) sysfatal("Maxname=%d was a bad choice", Maxname); if((d->mode & DMDIR) == 0){ sendp(cpath, strdup(path)); }else if(depth < Maxdepth){ /* recurse into the directory */ scan(dir, depth+1); path = *dir; }else{ fprint(2, "%s: too deep\n", path); } } free(buf); } close(dirfd); return 0; } static int cmpmeta(void *a_, void *b_) { Meta *a, *b; char *ae, *be; int i, x; a = *(Meta**)a_; b = *(Meta**)b_; if(simplesort) return cistrcmp(a->path, b->path); ae = utfrrune(a->path, '/'); be = utfrrune(b->path, '/'); if(ae != nil && be != nil && (x = cistrncmp(a->path, b->path, MAX(ae-a->path, be-b->path))) != 0) /* different path */ return x; /* same path, must be the same album/cd, but first check */ for(i = 0; i < a->numartist && i < b->numartist; i++){ if((x = cistrcmp(a->artist[i], b->artist[i])) != 0){ if(a->album != nil && b->album != nil && cistrcmp(a->album, b->album) != 0) return x; } } if(a->date != nil || b->date != nil){ if(a->date == nil && b->date != nil) return -1; if(a->date != nil && b->date == nil) return 1; if((x = atoi(a->date) - atoi(b->date)) != 0) return x; }else if(a->album != nil || b->album != nil){ if(a->album == nil && b->album != nil) return -1; if(a->album != nil && b->album == nil) return 1; if((x = cistrcmp(a->album, b->album)) != 0) return x; } if(a->track != nil || b->track != nil){ if(a->track == nil && b->track != nil) return -1; if(a->track != nil && b->track == nil) return 1; if((x = atoi(a->track) - atoi(b->track)) != 0) return x; } return cistrcmp(a->path, b->path); } static void usage(void) { fprint(2, "usage: %s [-s] directory/file/URL [...] > noise.plist\n", argv0); exits("usage"); } void threadmain(int argc, char **argv) { char *dir, *s, wd[4096]; Channel *cexit; int i, nproc; Biobuf out; Meta *m; ARGBEGIN{ case 's': simplesort++; break; default: usage(); }ARGEND if(argc < 1) usage(); if(getwd(wd, sizeof(wd)) == nil) sysfatal("%r"); moddec = access("/bin/audio/moddec", AEXEC) == 0; cmeta = chancreate(sizeof(Meta*), 0); cpath = chancreate(sizeof(char*), 32); cexit = chancreate(sizeof(ulong), 0); if((s = getenv("NPROC")) == nil) s = strdup("1"); if((nproc = atoi(s)-1) < 1) nproc = 1; free(s); for(i = 0; i < nproc; i++) proccreate(tagreadproc, cexit, 16384); threadcreate(metathread, nil, 4096); Binit(&out, 1, OWRITE); for(i = 0; i < argc; i++){ if(strncmp(argv[i], "http://", 7) == 0 || strncmp(argv[i], "https://", 8) == 0){ m = mallocz(sizeof(*m), 1); m->path = argv[i]; m->filefmt = ""; if(icyget(m, -1, nil) != 0){ fprint(2, "%s: %r\n", argv[i]); free(m); }else{ if(m->numartist == 0) m->artist[m->numartist++] = argv[i]; sendp(cmeta, m); } }else{ if(argv[i][0] == '/') dir = strdup(argv[i]); else dir = smprint("%s/%s", wd, argv[i]); cleanname(dir); scan(&dir, 0); free(dir); } } chanclose(cpath); for(i = 0; i < nproc; i++) recvul(cexit); chanclose(cmeta); qsort(tracks, ntracks, sizeof(Meta*), cmpmeta); for(i = 0; i < ntracks; i++){ if(tracks[i]->numartist < 1) fprint(2, "no artists: %s\n", tracks[i]->path); if(tracks[i]->title == nil) fprint(2, "no title: %s\n", tracks[i]->path); printmeta(&out, tracks[i]); } Bterm(&out); fprint(2, "found %d tagged tracks\n", ntracks); threadexitsall(nil); }