draw.c

#include <u.h>
#include <libc.h>
#include <thread.h>
#include <draw.h>
#include "dat.h"
#include "fns.h"

/* FIXME: clean up, now that we're just using alt; ddot vs rdot, dot */

QLock lsync;
Channel *drawc;
int samptime;

enum{
	Cbg,
	Csamp,
	Cline,
	Cins,
	Cloop,
	Cchunk,
	Ctext,
	Ncol,
};
static Image *col[Ncol];
static Image *view;
static Rectangle statr;
static usize views, viewe, viewmax, linepos;
static int bgscalyl, bgscalyr;
static double bgscalf;
static Channel *sampc;
static usize T;
static int sampwidth = 1;	/* pixels per sample */
static double zoom = 1.0;
static int stalerender, tworking;
static int working;
static vlong slen;
static s16int *graph[2];
static Dot ddot;

#define Rrate	(1000.0 / 60.0)

static Image *
eallocimage(Rectangle r, int repl, ulong col)
{
	Image *i;

	if((i = allocimage(display, r, screen->chan, repl, col)) == nil)
		sysfatal("allocimage: %r");
	return i;
}

vlong
ss2view(int x)
{
	return (x - views) / (T * sampwidth);
}

vlong
view2ss(int x)
{
	return views + x * T * sampwidth & ~3;
}

static void
b2t(usize off, int *th, int *tm, int *ts, int *tμ)
{
	usize nsamp;

	nsamp = off / Sampsz;
	*ts = nsamp / Rate;
	*tm = *ts / 60;
	*th = *tm / 60;
	*tμ = 100 * (nsamp - *ts * Rate) / Rate;
	*ts %= 60;
	*tm %= 60;
}

int
τfmt(Fmt *fmt)
{
	int th, tm, ts, tμ;
	usize p;

	p = va_arg(fmt->args, usize);
	if(p > ddot.totalsz)
		return fmtstrcpy(fmt, "-∞");
	b2t(p, &th, &tm, &ts, &tμ);
	if(samptime)
		return fmtprint(fmt, "%zd", p / Sampsz);
	else
		return fmtprint(fmt, "%02d:%02d:%02d.%03d", th, tm, ts, tμ);
}

static int
renderpos(usize ss, Image *c, int justdraw)
{
	Rectangle r;

	if(ss <= views || ss >= viewe)
		return 0;
	r = view->r;
	r.min.x = ss2view(ss);
	r.max.x = r.min.x + 1;
	if(justdraw){
		r = rectaddpt(r, screen->r.min);
		draw(screen, r, c, nil, ZP);
	}else
		draw(view, r, c, nil, ZP);
	return 1;
}

static void
renderchunks(void)
{
	usize p, off;
	Chunk *c;

	c = p2c(views, &off, &ddot);
	for(p=views-off; p<viewe; p+=c->len, c=c->right){
		if(p == 0)
			continue;
		renderpos(p, col[Cchunk], 0);
	}
}

static void
rendermarks(void)
{
	renderchunks();
	renderpos(ddot.from, col[Cloop], 0);
	renderpos(ddot.to, col[Cloop], 0);
	if(ddot.off != ddot.from)
		renderpos(ddot.off, col[Cins], 0);
}

static void
rendersamples(void)
{
	s16int *l, *e, *r;
	Rectangle rx, rr;

	if(slen == 0)
		return;
	rr = view->r;
	draw(view, rr, col[Cbg], nil, ZP);
	if(Dx(rr) > slen / 2)
		rr.max.x = rr.min.x + slen / 2;
	rx = rr;
	for(l=graph[chan]+2*rx.min.x, e=l+2*Dx(rr); l<e; l+=2, rx.min.x++){
		rx.min.y = rr.min.y + bgscalyl - l[1] / bgscalf;
		rx.max.x = rx.min.x + sampwidth;
		rx.max.y = rr.min.y + bgscalyl - l[0] / bgscalf;
		draw(view, rx, col[Csamp], nil, ZP);
	}
	if(!stereo)
		return;
	rx = rectaddpt(rr, Pt(0, Dy(view->r)/2));
	for(r=graph[chan+1&1]+2*rx.min.x, e=r+2*Dx(rr); r<e; r+=2, rx.min.x++){
		rx.min.y = rr.min.y + bgscalyr - r[1] / bgscalf;
		rx.max.x = rx.min.x + sampwidth;
		rx.max.y = rr.min.y + bgscalyr - r[0] / bgscalf;
		draw(view, rx, col[Csamp], nil, ZP);
	}
}

static void
render(void)
{
	rendersamples();
	rendermarks();
}

static void
erasemark(usize ss)
{
	Rectangle r;

	ss = ss2view(ss);
	r = view->r;
	r.min.x = ss;
	r.max.x = r.min.x + 1;
	r = rectaddpt(r, screen->r.min);
	draw(screen, r, view, nil, subpt(r.min, screen->r.min));
}

static void
drawstat(void)
{
	char s[256], *b[3];
	Point p;

	draw(screen, statr, col[Cbg], nil, ZP);
	seprint(s, s+sizeof s, "%sT=%zd @ %τ",
		stereo ? "" : chan==0?"Left ":"Right ",
		T / Sampsz, ddot.cur);
	p = string(screen, statr.min, col[Ctext], ZP, font, s);
	if(bound == Bstart){
		b[0] = "[";
		b[1] = "] ";
		b[2] = " ";
	}else{
		b[0] = " ";
		b[1] = " [";
		b[2] = "]";
	}
	seprint(s, s+sizeof s, " %sfrom %τ%sto %τ%s",
		b[0], ddot.from, b[1], ddot.to, b[2]);
	p = string(screen, p, col[Cloop], ZP, font, s);
	if(ddot.off != ddot.from && ddot.off >= 0){
		seprint(s, s+sizeof s, " last %τ", ddot.off);
		p = string(screen, p, col[Cins], ZP, font, s);
	}
	statr.max.x = p.x;
}

void
paint(int what)
{
	ddot = dot;
	lockdisplay(display);
	if((what & Drawrender) != 0 || stalerender || working){
		if(!working)
			stalerender = 0;
		render();
		draw(screen, rectaddpt(view->r, screen->r.min), view, nil, ZP);
	}else
		erasemark(linepos);
	renderpos(ddot.cur, col[Cline], 1);
	linepos = ddot.cur;
	drawstat();
	flushimage(display, 1);
	unlockdisplay(display);
}

/* throttling of draw requests happens here */
void
refresh(int what)
{
	static int f;

	f |= what;
	if(nbsend(drawc, &f) != 0)
		f = 0;
}

static void
sampler(void *)
{
	int n, lmin, lmax, rmin, rmax;
	usize k;
	uchar *p, *e;
	s16int s, *l, *r, *le;
	vlong N;
	Dot d;

again:
	if(recv(sampc, &d) < 0)
		threadexits("recv: %r");
	tworking = ++working;
	stalerender++;
	N = (d.to - d.from) / (T * sampwidth);
	if(slen < 2*N){	/* min, max */
		graph[0] = erealloc(graph[0],
			2*N * sizeof *graph[0],
			slen * sizeof *graph[0]);
		graph[1] = erealloc(graph[1],
			2*N * sizeof *graph[1],
			slen * sizeof *graph[1]);
	}
	slen = 2*N;
	l = graph[0];
	r = graph[1];
	le = l + slen;
	while(l < le){
		if(sampc->n > 0)	/* start over */
			break;
		n = T * sampwidth;
		lmin = lmax = rmin = rmax = 0;
		while(n > 0){
			if((p = getslice(&d, n, &k)) == nil){
				if(k > 0)
					fprint(2, "getslice: %r\n");
				l = le;
				break;
			}
			d.cur += k;
			for(e=p+k; p<e; p+=Sampsz*sampwidth){
				s = (s16int)(p[1] << 8 | p[0]);
				if(s < lmin)
					lmin = s;
				else if(s > lmax)
					lmax = s;
				s = (s16int)(p[3] << 8 | p[2]);
				if(s < rmin)
					rmin = s;
				else if(s > rmax)
					rmax = s;
			}
			n -= k;
		}
		*l++ = lmin;
		*l++ = lmax;
		*r++ = rmin;
		*r++ = rmax;
	}
	working--;
	refresh(Drawrender);
	goto again;
}

static void
resetdraw(void)
{
	Rectangle viewr;

	viewr = rectsubpt(screen->r, screen->r.min);
	statr = screen->r;
	if(stereo){
		statr.min.y += (Dy(screen->r) - font->height) / 2 + 1;
		statr.max.y = statr.min.y + font->height;
	}else
		statr.min.y = screen->r.max.y - font->height;
	freeimage(view);
	view = eallocimage(viewr, 0, DNofill);
	bgscalyl = (viewr.max.y - font->height) / (1 * (stereo ? 4 : 2));
	bgscalyr = viewr.max.y - bgscalyl;
	bgscalf = 32767. / bgscalyl;
}

static void
resetview(int all)
{
	usize span;

	lockdisplay(display);
	T = (vlong)(ddot.totalsz / zoom / Dx(screen->r)) & ~3;
	if(T < Sampsz)
		T = Sampsz;
	span = Dx(screen->r) * T;
	viewmax = ddot.totalsz - span;
	if(views > viewmax)
		views = viewmax;
	viewe = views + span;
	if(all)
		resetdraw();
	unlockdisplay(display);
}

void
redraw(int all)
{
	Dot d;

	ddot = dot;
	resetview(all);
	if(paused)
		refresh(Drawall);
		
	d = ddot;
	d.from = d.cur = views;
	d.to = viewe;
	nbsend(sampc, &d);
}

void
setzoom(int Δz, int x)
{
	double z, span, Δx;

	if(Δz < 0)
		z = zoom / pow(1.025, -Δz);
	else
		z = zoom * pow(1.025, Δz);
	if(z < 1.0)
		z = 1.0;
	else if(z > (ddot.totalsz / Sampsz) / Dx(screen->r))
		z = (ddot.totalsz / Sampsz) / Dx(screen->r);
	if(z == zoom)
		return;
	zoom = z;
	span = T;
	resetview(0);
	span -= T;
	span *= Dx(screen->r);
	Δx = ((double)x / Dx(screen->r)) * span / T;
	if(!setpan(Δx))
		redraw(0);
}

int
zoominto(vlong from, vlong to)
{
	if(from < 0)
		from = 0;
	from &= ~3;
	if(to >= ddot.totalsz)
		to = ddot.totalsz;
	to &= ~3;
	if((to - from) / Sampsz < Dx(screen->r)){
		werrstr("range too small");
		return -1;
	}
	views = from;
	viewe = to;
	zoom = (double)ddot.totalsz / (to - from);
	redraw(0);
	return 0;
}

int
setpan(int Δx)
{
	usize new;

	Δx *= T;
	if(zoom == 1.0)
		return 0;
	if(Δx < 0 && -Δx > views)
		new = 0;
	else if(views + Δx >= viewmax)
		new = viewmax;
	else
		new = views + Δx;
	if(new == views)
		return 0;
	views = new;
	redraw(0);
	return 1;
}

void
setpage(int d)
{
	setpan(d * (Dx(view->r) + 1));
}

void
setrange(usize from, usize to)
{
	from &= ~3;
	to &= ~3;
	if(from >= to)
		return;
	ddot.from = from;
	ddot.to = to;
	/* advance may desync and reset it again */
	if(ddot.cur < from || ddot.cur >= to)
		ddot.cur = from;
	ddot.off = -1;
	dot = ddot;
	stalerender = 1;
	refresh(Drawrender);
}

int
setjump(vlong off)
{
	off &= ~3;
	if(off < ddot.from || off > ddot.to - Sampsz){
		werrstr("cannot jump outside of loop bounds");
		return -1;
	}
	ddot.off = ddot.cur = off;
	dot = ddot;
	stalerender = 1;
	refresh(Drawrender);
	return 0;
}

int
setloop(vlong off)
{
	off &= ~3;
	if(off < 0 || off > ddot.totalsz){
		werrstr("invalid range");
		return -1;
	}
	if(bound == Bstart)
		setrange(off, ddot.to);
	else
		setrange(ddot.from, off);
	return 0;
}

void
initdrw(int fuckit)
{
	if(initdraw(nil, nil, "pplay") < 0)
		sysfatal("initdraw: %r");
	display->locking = 1;
	unlockdisplay(display);
	if(!fuckit){
		col[Cbg] = eallocimage(Rect(0,0,1,1), 1, 0xFFFFC0FF);
		col[Csamp] = eallocimage(Rect(0,0,1,1), 1, 0x3F3F20FF);
		col[Ctext] = display->black;
		col[Cline] = eallocimage(Rect(0,0,1,1), 1, DBlue);
		col[Cins] = eallocimage(Rect(0,0,1,1), 1, DPaleblue);
		col[Cloop] = eallocimage(Rect(0,0,1,1), 1, DBluegreen);
		col[Cchunk] = eallocimage(Rect(0,0,1,1), 1, DRed);
	}else{
		col[Cbg] = display->black;
		col[Csamp] = eallocimage(Rect(0,0,1,1), 1, 0x777777FF);
		col[Ctext] = eallocimage(Rect(0,0,1,1), 1, 0xBBBBBBFF);
		col[Cline] = eallocimage(Rect(0,0,1,1), 1, 0xEEA000FF);
		col[Cins] = eallocimage(Rect(0,0,1,1), 1, 0x509A9AFF);
		col[Cloop] = eallocimage(Rect(0,0,1,1), 1, 0x8888CCFF);
		col[Cchunk] = eallocimage(Rect(0,0,1,1), 1, 0xEE0000FF);
	}
	if((drawc = chancreate(sizeof(int), 8)) == nil
	|| (sampc = chancreate(sizeof(Dot), 2)) == nil)
		sysfatal("chancreate: %r");
	redraw(1);
	if(proccreate(sampler, nil, mainstacksize) < 0)
		sysfatal("proccreate: %r");
}