shithub: pplay

ref: 7fb266c8b5f73defd884c57ec1742d28ef5f6404
dir: /pplay.c/

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

int ifd;
uchar *pcmbuf;
vlong filesz, nsamp;
int file, stereo, zoom = 1;
vlong seekp, T, loops, loope;

static Keyboardctl *kc;
static Mousectl *mc;
static int cat;
static int afd = -1;

void *
emalloc(ulong n)
{
	void *p;

	if((p = mallocz(n, 1)) == nil)
		sysfatal("emalloc: %r");
	setmalloctag(p, getcallerpc(&n));
	return p;
}

static void
athread(void *)
{
	int n;
	uchar *p;

	p = pcmbuf;
	for(;;){
		if(afd < 0)
			return;
		n = seekp + Nchunk >= loope ? loope - seekp : Nchunk;
		if(!file)
			p = pcmbuf + seekp;
		else if(read(ifd, pcmbuf, n) != n)
			fprint(2, "read: %r\n");
		if(write(afd, p, n) != n){
			fprint(2, "athread: %r\n");
			break;
		}
		seekp += n;
		if(seekp >= loope)
			setpos(loops);
		update();
		yield();
	}
}

static void
toggleplay(void)
{
	static int play;

	if(play ^= 1){
		if((afd = cat ? 1 : open("/dev/audio", OWRITE)) < 0){
			fprint(2, "toggleplay: %r\n");
			play = 0;
			return;
		}
		if(threadcreate(athread, nil, mainstacksize) < 0)
			sysfatal("threadcreate: %r");
	}else{
		if(!cat)
			close(afd);
		afd = -1;
	}
}

static char *
prompt(Rune r)
{
	static char buf[512];

	snprint(buf, sizeof buf, "%C", r);
	if(enter(nil, buf, sizeof(buf)-UTFmax, mc, kc, _screen) < 0)
		return nil;
	return buf;
}

static void
reallocbuf(ulong n)
{
	if((pcmbuf = realloc(pcmbuf, n)) == nil)
		sysfatal("realloc: %r");
}

static void
initbuf(int fd)
{
	int n, sz;
	vlong ofs;
	Dir *d;

	reallocbuf(filesz += Nreadsz);
	ifd = fd;
	if(file){
		if((d = dirfstat(fd)) == nil)
			sysfatal("dirfstat: %r");
		filesz = d->length;
		nsamp = filesz / 4;
		free(d);
		return;
	}
	if((sz = iounit(fd)) == 0)
		sz = 8192;
	ofs = 0;
	while((n = read(fd, pcmbuf+ofs, sz)) > 0){
		ofs += n;
		if(ofs + sz >= filesz)
			reallocbuf(filesz += Nreadsz);
	}
	if(n < 0)
		sysfatal("read: %r");
	filesz = ofs;
	reallocbuf(filesz);
	nsamp = filesz / 4;
	close(fd);
}

static void
usage(void)
{
	fprint(2, "usage: %s [-cfs] [pcm]\n", argv0);
	threadexits("usage");
}

void
threadmain(int argc, char **argv)
{
	int fd;
	char *p;
	Mouse mo;
	Rune r;

	ARGBEGIN{
	case 'c': cat = 1; break;
	case 'f': file = 1; break;
	case 's': stereo = 1; break;
	default: usage();
	}ARGEND
	if((fd = *argv != nil ? open(*argv, OREAD) : 0) < 0)
		sysfatal("open: %r");
	initbuf(fd);
	initdrw();
	if((kc = initkeyboard(nil)) == nil)
		sysfatal("initkeyboard: %r");
	if((mc = initmouse(nil, screen)) == nil)
		sysfatal("initmouse: %r");
	mo.xy = ZP;
	Alt a[] = {
		{mc->resizec, nil, CHANRCV},
		{mc->c, &mc->Mouse, CHANRCV},
		{kc->c, &r, CHANRCV},
		{nil, nil, CHANEND}
	};
	initcmd();
	toggleplay();
	for(;;){
		switch(alt(a)){
		case 0:
			if(getwindow(display, Refnone) < 0)
				sysfatal("resize failed: %r");
			mo = mc->Mouse;
			redraw(1);
			break;
		case 1:
			if(eqpt(mo.xy, ZP))
				mo = mc->Mouse;
			switch(mc->buttons){
			case 1: setofs(mc->xy.x - screen->r.min.x); break;
			case 2: setloop(mc->xy.x - screen->r.min.x); break;
			case 4: setpan(mo.xy.x - mc->xy.x); break;
			case 8: setzoom(1, 1); break;
			case 16: setzoom(-1, 1); break;
			}
			mo = mc->Mouse;
			break;
		case 2:
			switch(r){
			case ' ': toggleplay(); break;
			case 'b': setpos(loops); break;
			case Kesc: loops = 0; loope = filesz; update(); break;
			case Kdel:
			case 'q': threadexitsall(nil);
			case 'z': setzoom(-zoom + 1, 0); break;
			case '-': setzoom(-1, 0); break;
			case '=': setzoom(1, 0); break;
			case '_': setzoom(-1, 1); break;
			case '+': setzoom(1, 1); break;
			default:
				if((p = prompt(r)) == nil || strlen(p) == 0)
					break;
				if(cmd(p) < 0)
					fprint(2, "cmd \"%s\" failed: %r\n", p);
				break;
			}
			break;
		}
	}
}