ref: d29b1fa9119ef89090971376b6c4b5378813efd7
dir: /waveform/waveform.c/
#include <u.h> #include <libc.h> #include <draw.h> #include <mouse.h> #include <keyboard.h> #include <thread.h> #define MIN(a,b) ((a)<=(b)?(a):(b)) #define MAX(a,b) ((a)>=(b)?(a):(b)) typedef struct Waveform Waveform; struct Waveform { float *samples; Image *image; float min; float max; int nchan; int nframes; int nvalid; int rate; }; static Waveform * wvform(int f, int nchan, int rate) { Waveform *w; int i, r, n, sz; w = calloc(1, sizeof(*w)); w->nchan = nchan; w->rate = rate; sz = 128*nchan; w->samples = malloc(sz*sizeof(float)); for (n = 0;;) { if (sz-n < 1) { sz *= 2; w->samples = realloc(w->samples, sz*sizeof(float)); } if ((r = read(f, w->samples+n, (sz-n)*sizeof(float))) < 0) { free(w->samples); free(w); return nil; } r /= sizeof(float); if (r == 0) break; if (n == 0) w->min = w->max = w->samples[0]; for (i = 0; i < r; i++, n++) { w->min = MIN(w->min, w->samples[n]); w->max = MAX(w->max, w->samples[n]); } } w->samples = realloc(w->samples, n*sizeof(float)); w->nframes = w->nvalid = n / nchan; return w; } static u64int * rotate9064(u64int *b, int h) { u64int *p; int i, j; u64int v, i7; p = calloc(1, 64*h/8); for (i = 0; i < h; i++) { i7 = 1ULL<<(i^7); v = b[i]; for (j = 0; j < 64; j++) { if (v & (1ULL<<(j^56))) p[j*h/64 + i/64] |= i7; } } for (i = 0; i < h; i++) p[i] = ~p[i]; return p; } static u64int * rotate90128(u64int *b, int h) { u64int *p; int i, j; u64int v, v2, i7; p = calloc(1, 128*h/8); for (i = 0; i < h; i++) { i7 = 1ULL<<(i^7); v = b[i*2+0]; v2 = b[i*2+1]; for (j = 1; j < 65; j++) { if (v & (1ULL<<(j^56))) p[(j+63)*h/64 + i/64] |= i7; if (v2 & (1ULL<<(j^56))) p[j*h/64 + i/64] |= i7; } } for (i = 0; i < 128*h/64; i++) p[i] = ~p[i]; return p; } static int wvimage64(Waveform *w, int offset, Rectangle r, float zoom) { float m, i; u64int *b, *p; int x, x2, y, incy, y2; offset = MIN(offset, w->nvalid); r = Rect(0, 0, 64, Dx(r)); if (badrect(r)) return -1; y = (Dy(r)+63) & ~63; b = calloc(1, 64*y); m = MAX(abs(w->min), abs(w->max)); m = m > 1.0f ? 31.0f/m : 31.0f; for (y = 0, i = offset; y < Dy(r) && i < w->nvalid;) { x = 31 + m*w->samples[(int)i]; i += zoom; x2 = i < w->nvalid ? 31 + m*w->samples[(int)i] : x; incy = (x2 + x)/2; y2 = y + 1; do { b[y+1] |= 1ULL<<(x^7); if (x == x2) break; else if (x < x2) x++; else if (x > x2) x--; else if (x == incy) y++; } while (1); y = y2; } y = (y+63) & ~63; p = rotate9064(b, y); free(b); r.max.x = y; r.max.y = 64; if (w->image == nil || !eqrect(r, w->image->r)) { freeimage(w->image); if ((w->image = allocimage(display, r, GREY1, 0, DNofill)) == nil) return -1; } if (loadimage(w->image, r, (void*)p, 64*y) < 0) fprint(2, "failed: %r\n"); free(p); return 0; } static int wvimage128(Waveform *w, int offset, Rectangle r, float zoom) { float m, i; u64int *b, *p; int x, x2, y, incy, y2; offset = MIN(offset, w->nvalid); r = Rect(0, 0, 128, Dx(r)); if (badrect(r)) return -1; y = (Dy(r)+63) & ~63; b = calloc(1, 128*y); m = MAX(abs(w->min), abs(w->max)); m = m > 1.0f ? 63.0f/m : 63.0f; for (y = 0, i = offset; y < Dy(r) && i < w->nvalid;) { x = m*w->samples[(int)i]; i += zoom; x2 = i < w->nvalid ? m*w->samples[(int)i] : x; incy = (x2 + x)/2; y2 = y + 1; do { if (x >= 0) b[y*2+1] |= 1ULL<<(x^7); else b[y*2+0] |= 1ULL<<(-x^56); if (x == x2) break; else if (x < x2) x++; else if (x > x2) x--; else if (x == incy) y++; } while (1); y = y2; } y = (y+63) & ~63; p = rotate90128(b, y); free(b); r.max.x = y; r.max.y = 128; if (w->image == nil || !eqrect(r, w->image->r)) { freeimage(w->image); if ((w->image = allocimage(display, r, GREY1, 0, DNofill)) == nil) return -1; } if (loadimage(w->image, r, (void*)p, 128*y) < 0) fprint(2, "failed: %r\n"); free(p); return 0; } static int offset; static float zoom; static void redraw(Waveform *w) { Rectangle r; lockdisplay(display); r = screen->r; r.min.y += Dy(r)/4; r.max.y -= Dy(r)/4; draw(screen, screen->r, display->white, nil, ZP); if (wvimage64(w, offset, r, zoom) == 0) draw(screen, r, w->image, nil, ZP); flushimage(display, 1); unlockdisplay(display); } static Waveform * wvbuffer(int nsamples) { Waveform *w; w = calloc(1, sizeof(*w)); w->samples = malloc(nsamples*sizeof(float)); w->min = -1.0f; w->max = 1.0f; w->nchan = 1; w->nframes = nsamples; w->rate = 44100; w->nvalid = 0; return w; } static void wvproc(void *w_) { int fi, fo, n, i; Waveform *w = w_; int c; fi = 0; fo = 1; for (c = 0;; c++) { if ((n = readn(fi, w->samples, sizeof(float)*w->nframes)) < 1) break; n /= sizeof(float); w->nvalid = n; for (i = 0; i < n; i++) { w->min = MIN(-1.0f, MIN(w->min, w->samples[i])); w->max = MAX(1.0f, MAX(w->max, w->samples[i])); } if (write(fo, w->samples, n*sizeof(float)) != n*sizeof(float)) break; redraw(w); } threadexits(nil); } void threadmain(int argc, char **argv) { Waveform *w; Mousectl *mctl; Keyboardctl *kctl; Rune key; int oldo; Mouse m; int oldb; Point oldp; Alt a[] = { { nil, &m, CHANRCV }, { nil, nil, CHANRCV }, { nil, &key, CHANRCV }, { nil, nil, CHANEND }, }; USED(argc); USED(argv); if ((w = wvform(0, 1, 44100)) == nil) sysfatal("%r"); /* if ((w = wvbuffer(1024)) == nil) sysfatal("%r"); */ if (initdraw(nil, nil, "daw/waveform") < 0) sysfatal("initdraw: %r"); display->locking = 1; unlockdisplay(display); if ((mctl = initmouse(nil, screen)) == nil) sysfatal("initmouse: %r"); if ((kctl = initkeyboard(nil)) == nil) sysfatal("initkeyboard: %r"); a[0].c = mctl->c; a[1].c = mctl->resizec; a[2].c = kctl->c; threadsetname("daw/waveform"); zoom = 1.0f; offset = 0; oldb = 0; oldo = 0; oldp = ZP; //proccreate(wvproc, w, mainstacksize); redraw(w); for (;;) { switch (alt(a)) { case 0: /* mouse */ if (m.buttons == 1) { if (oldb == 0) { oldp = m.xy; oldo = offset; } else if (oldb == 1) { offset = MAX(0, oldo + (oldp.x - m.xy.x)*zoom); redraw(w); } } oldb = m.buttons; break; case 1: /* resize */ getwindow(display, Refnone); redraw(w); break; case 2: /* keyboard */ switch (key) { case Kdel: goto end; case Kleft: offset = MAX(0, offset-MAX(8, 8*MAX(1, 1/zoom))); redraw(w); break; case Kright: offset = MIN(w->nframes-1, offset+MAX(8, 8*MAX(1, 1/zoom))); redraw(w); break; case '-': zoom *= 1.1f; if (zoom > 32.0f) zoom = 32.0f; redraw(w); break; case '+': zoom /= 1.1f; if (zoom < 0.01f) zoom = 0.01f; redraw(w); break; } break; } } end: threadexitsall(nil); }