ref: 8cac775c460cd32ffa82a756e6c7e6dc6d7fda51
dir: /draw.c/
#include <u.h>
#include <libc.h>
#include <thread.h>
#include <draw.h>
#include "dat.h"
#include "fns.h"
QLock lsync;
enum{
Cbg,
Csamp,
Cline,
Cloop,
Cchunk,
Ncol,
};
static Image *col[Ncol];
static Image *viewbg, *view;
static Rectangle liner;
static Point statp;
static usize views, viewe, viewmax;
static int bgscalyl, bgscalyr, bgscalf;
static Channel *drawc;
static usize T;
static uchar *sbuf;
static usize sbufsz;
static int sampwidth = 1;
static double zoom = 1.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;
}
static void
drawchunks(void)
{
int x;
usize p, off;
Chunk *c;
Rectangle r;
c = p2c(views, &off);
r = view->r;
for(p=views-off; p<viewe; p+=c->bufsz, c=c->right){
if(p == 0)
continue;
x = (p - views) / T;
if(x > Dx(view->r))
break;
r = view->r;
r.min.x += x;
r.max.x = r.min.x + 1;
draw(view, r, col[Cchunk], nil, ZP);
if(c->bufsz == 0)
break;
}
}
static void
drawsamps(void*)
{
int x, lmin, lmax, rmin, rmax;
usize n, m;
s16int s;
uchar *p, *e;
Rectangle l, r;
Dot d;
for(;;){
end:
recvul(drawc);
again:
if(sbufsz < T * sampwidth){
sbuf = erealloc(sbuf, T * sampwidth, sbufsz);
sbufsz = T * sampwidth;
}
lockdisplay(display);
draw(viewbg, viewbg->r, col[Cbg], nil, ZP);
unlockdisplay(display);
d.pos = views;
m = viewe - views;
x = 0;
while(m > 0){
qlock(&lsync);
if(nbrecvul(drawc) == 1){
qunlock(&lsync);
goto again;
}
n = m < T * sampwidth ? m : T * sampwidth;
p = getbuf(d, n, sbuf, &n);
qunlock(&lsync);
if(p == nil){
if(n > 0)
fprint(2, "getbuf: %r\n");
goto end;
}
d.pos += n;
e = p + n;
lmin = lmax = 0;
rmin = rmax = 0;
while(p < e){
s = (s16int)(p[1] << 8 | p[0]);
if(s < lmin)
lmin = s;
else if(s > lmax)
lmax = s;
if(stereo){
s = (s16int)(p[3] << 8 | p[2]);
if(s < rmin)
rmin = s;
else if(s > rmax)
rmax = s;
}
p += 4 * sampwidth;
}
m -= n;
l = Rect(x, bgscalyl - lmax / bgscalf,
x+sampwidth, bgscalyl - lmin / bgscalf);
r = Rect(x, bgscalyr - rmax / bgscalf,
x+sampwidth, bgscalyr - rmin / bgscalf);
lockdisplay(display);
draw(viewbg, l, col[Csamp], nil, ZP);
if(stereo)
draw(viewbg, r, col[Csamp], nil, ZP);
unlockdisplay(display);
x = (d.pos - views) / T;
}
}
}
static void
b2t(usize ofs, int *th, int *tm, int *ts, int *tμ)
{
usize nsamp;
nsamp = ofs / 4;
*ts = nsamp / 44100;
*tm = *ts / 60;
*th = *tm / 60;
*tμ = 100 * (nsamp - *ts * 44100) / 44100;
*ts %= 60;
}
static void
drawstat(void)
{
int th, tm, ts, tμ;
char s[256], *p;
b2t(dot.pos, &th, &tm, &ts, &tμ);
p = seprint(s, s+sizeof s, "T %zd @ %02d:%02d:%02d.%03d (%zd) ⋅ ",
T/4, th, tm, ts, tμ, dot.pos/4);
if(dot.from.pos > 0){
b2t(dot.from.pos, &th, &tm, &ts, &tμ);
p = seprint(p, s+sizeof s, "%02d:%02d:%02d.%03d (%zd) ↺ ",
th, tm, ts, tμ, dot.from.pos/4);
}else
p = seprint(p, s+sizeof s, "0 ↺ ");
if(dot.to.pos != totalsz){
b2t(dot.to.pos, &th, &tm, &ts, &tμ);
seprint(p, s+sizeof s, "%02d:%02d:%02d.%03d (%zd)",
th, tm, ts, tμ, dot.to.pos/4);
}else
seprint(p, s+sizeof s, "∞");
string(screen, statp, col[Cline], ZP, font, s);
}
static void
drawview(void)
{
int x;
usize left, right;
Rectangle r;
left = dot.from.pos;
draw(view, view->r, viewbg, nil, ZP);
if(left != 0 && left >= views){
x = (left - views) / T;
r = view->r;
r.min.x += x;
r.max.x = r.min.x + 1;
draw(view, r, col[Cloop], nil, ZP);
}
right = dot.to.pos;
if(right != totalsz){
x = (right - views) / T;
r = view->r;
r.min.x += x;
r.max.x = r.min.x + 1;
draw(view, r, col[Cloop], nil, ZP);
}
if(debug)
drawchunks();
}
void
update(void)
{
int x;
usize p;
p = dot.pos;
lockdisplay(display);
drawview();
x = screen->r.min.x + (p - views) / T;
draw(screen, screen->r, view, nil, ZP);
liner.min.x = x;
liner.max.x = x + 1;
if(p >= views)
draw(screen, liner, col[Cline], nil, ZP);
drawstat();
flushimage(display, 1);
unlockdisplay(display);
}
void
setzoom(int Δz, int mul)
{
double z;
if(!mul)
z = zoom + Δz;
else if(Δz < 0)
z = zoom / pow(2, -Δz);
else
z = zoom * pow(2, Δz);
if(z < 1.0 || z > (totalsz / Sampsz) / Dx(screen->r))
return;
zoom = z;
redraw(0);
}
int
zoominto(vlong from, vlong to)
{
if(dot.from.pos == 0 && dot.to.pos == totalsz){
fprint(2, "not a range\n");
return -1;
}
if(from < 0)
from = 0;
from &= ~3;
if(to >= totalsz)
to = totalsz;
to &= ~3;
if((to - from) / Sampsz < Dx(screen->r)){
werrstr("range too small");
return -1;
}
views = from;
viewe = to;
zoom = (double)totalsz / (to - from);
redraw(0);
return 0;
}
void
setpan(int Δx)
{
usize new;
Δx *= T;
if(zoom == 1)
return;
if(Δx < 0 && -Δx > views)
new = 0;
else if(views + Δx >= viewmax)
new = viewmax;
else
new = views + Δx;
if(new == views)
return;
views = new;
redraw(0);
}
void
setloop(vlong off)
{
off *= T;
off += views;
if(off < 0 || off > totalsz)
return;
if(off < dot.pos)
setrange(off, dot.to.pos);
else
setrange(dot.from.pos, off);
update();
}
static void
setcur(usize off)
{
if(off < dot.from.pos || off > dot.to.pos - Outsz)
return;
jump(off);
update();
}
void
setjump(usize off)
{
setcur(off);
}
void
setofs(usize ofs)
{
setcur(views + ofs * T);
}
static void
resetdraw(void)
{
int x;
Rectangle viewr, midr;
x = screen->r.min.x + (dot.pos - views) / T;
viewr = rectsubpt(screen->r, screen->r.min);
freeimage(viewbg);
freeimage(view);
viewbg = eallocimage(viewr, 0, DBlack);
view = eallocimage(viewr, 0, DBlack);
if(stereo){
midr = viewr;
midr.min.y = midr.max.y / 2;
midr.max.y = midr.min.y + 1;
draw(viewbg, midr, col[Csamp], nil, ZP);
statp = Pt(screen->r.min.x,
screen->r.min.y + (Dy(screen->r) - font->height) / 2 + 1);
}else
statp = Pt(screen->r.min.x, screen->r.max.y - font->height);
liner = screen->r;
liner.min.x = x;
liner.max.x = x + 1;
bgscalyl = viewbg->r.max.y / (stereo ? 4 : 2);
bgscalyr = viewbg->r.max.y - bgscalyl;
bgscalf = 32767 / bgscalyl;
}
void
redraw(int all)
{
usize span;
lockdisplay(display);
T = (vlong)(totalsz / zoom / Dx(screen->r)) & ~3;
if(T == 0)
T = 4;
span = Dx(screen->r) * T;
viewmax = totalsz - span;
if(views > viewmax)
views = viewmax;
viewe = views + span;
if(all)
resetdraw();
unlockdisplay(display);
nbsendul(drawc, 1);
update();
}
void
initdrw(void)
{
if(initdraw(nil, nil, "pplay") < 0)
sysfatal("initdraw: %r");
display->locking = 1;
unlockdisplay(display);
col[Cbg] = eallocimage(Rect(0,0,1,1), 1, DBlack);
col[Csamp] = eallocimage(Rect(0,0,1,1), 1, 0x440000FF);
col[Cline] = eallocimage(Rect(0,0,1,1), 1, 0x884400FF);
col[Cloop] = eallocimage(Rect(0,0,1,1), 1, 0x777777FF);
col[Cchunk] = eallocimage(Rect(0,0,1,1), 1, 0xBBBB00FF);
if((drawc = chancreate(sizeof(ulong), 4)) == nil)
sysfatal("chancreate: %r");
if(proccreate(drawsamps, nil, mainstacksize) < 0)
sysfatal("proccreate: %r");
redraw(1);
}