#include <u.h>
#include <libc.h>
#include <draw.h>
#include <thread.h>
#include <mouse.h>
#include <cursor.h>

enum { ncolors = 600 };

double xmin = -2, xmax = 1;
double ymin = -1, ymax = 1;
Mousectl *mctl;
Rendez rend;
int stopdraw;
uchar *imagedata;
Image *image;
extern Cursor reading;
uchar colors[3 * (ncolors + 1)];

double
convx(Rectangle *r, int x)
{
	return (xmax - xmin) * (x - r->min.x) / (r->max.x - r->min.x) + xmin;
}

double
convy(Rectangle *r, int y)
{
	return (ymax - ymin) * (r->max.y - y) / (r->max.y - r->min.y) + ymin;
}

void
pixel(int x, int y)
{
	draw(screen, Rect(x, y, x + 1, y + 1), display->black, nil, ZP);
}

ulong
iterate(double x, double y)
{
	int i;
	double zx, zy, zx2, zy2, v;

	zx = zy = zx2 = zy2 = 0;
	for(i = 0; i < 100; i++){
		zy = 2 * zx * zy + y;
		zx = zx2 - zy2 + x;
		zx2 = zx * zx;
		zy2 = zy * zy;
		if(zx2 + zy2 >= 4){
			v = 2 + i - log(log(sqrt(zx2 + zy2)) / log(i + 2)) / 0.69314718;
			return (int)(v * 1000) % ncolors;
		}
	}
	return 0;
}

void
redrawproc(void *)
{
	int x, y;
	uchar *p, *q;

	qlock(rend.l);
	for(;;){
		setcursor(mctl, &reading);
		p = imagedata;
		for(y = screen->r.min.y; y < screen->r.max.y; y++)
			for(x = screen->r.min.x; x < screen->r.max.x; x++){
				if(stopdraw)
					goto check;
				q = colors + 3 * iterate(convx(&screen->r, x), convy(&screen->r, y));
				*p++ = *q++;
				*p++ = *q++;
				*p++ = *q;
			}
		if(stopdraw)
			goto check;
		lockdisplay(display);
		loadimage(image, image->r, imagedata, Dx(image->r) * Dy(image->r) * 3);
		draw(screen, screen->r, image, nil, screen->r.min);
		flushimage(display, 1);
		unlockdisplay(display);
	check:
		stopdraw = 0;
		setcursor(mctl, nil);
		rsleep(&rend);
		stopdraw = 0;
	}
}

void
zoom(void)
{
	Rectangle r;
	double xmin_, xmax_, ymin_, ymax_;

	r = getrect(3, mctl);
	if(r.min.x == 0 && r.min.y == 0 && r.max.x == 0 && r.max.y == 0)
		return;
	xmin_ = convx(&screen->r, r.min.x);
	xmax_ = convx(&screen->r, r.max.x);
	ymin_ = convy(&screen->r, r.max.y);
	ymax_ = convy(&screen->r, r.min.y);
	stopdraw = 1;
	qlock(rend.l);
	xmin = xmin_;
	xmax = xmax_;
	ymin = ymin_;
	ymax = ymax_;
	rwakeup(&rend);
	qunlock(rend.l);
}

char *menus[] = {
	"zoom in",
	"reset",
	"quit",
	nil,
};

Menu menu = {
	.item = menus
};

void
resizethread(void *)
{
	ulong l;

	for(;;){
		if(recv(mctl->resizec, &l) < 1)
			continue;
		stopdraw = 1;
		qlock(rend.l);
		if(getwindow(display, Refnone) < 0)
			sysfatal("getwindow: %r");
		freeimage(image);
		free(imagedata);
		image = allocimage(display, screen->r, RGB24, 0, DBlack);
		imagedata = malloc(3 * Dx(screen->r) * Dy(screen->r));
		rwakeup(&rend);
		qunlock(rend.l);
	}
}

void
initcolors(void)
{
	uchar *p;
	int h, x;

	for(p = colors + 3, h = 0; p < colors + nelem(colors); h++){
		x = 0xFF - abs(h % (ncolors / 3) - ncolors / 6) * 0xFF / (ncolors / 6);
		if(h < ncolors/6){
			*p++ = 0xFF;
			*p++ = x;
			*p++ = 0;
		}else if(h < ncolors/3){
			*p++ = x;
			*p++ = 0xFF;
			*p++ = 0;
		}else if(h < ncolors/2){
			*p++ = 0;
			*p++ = 0xFF;
			*p++ = x;
		}else if(h < 2*ncolors/3){
			*p++ = 0;
			*p++ = x;
			*p++ = 0xFF;
		}else if(h < 5*ncolors/6){
			*p++ = x;
			*p++ = 0;
			*p++ = 0xFF;
		}else{
			*p++ = 0xFF;
			*p++ = 0;
			*p++ = x;
		}
	}
}

void
threadmain()
{
	static QLock ql;
	int rc;

	if(initdraw(nil, nil, "mandelbrot") < 0)
		sysfatal("initdraw: %r");
	display->locking = 1;
	unlockdisplay(display);
	if((mctl = initmouse(nil, screen)) == nil)
		sysfatal("initmouse: %r");
	rend.l = &ql;
	image = allocimage(display, screen->r, RGB24, 0, DBlack);
	imagedata = malloc(3 * Dx(screen->r) * Dy(screen->r));
	initcolors();
	proccreate(redrawproc, nil, mainstacksize);
	threadcreate(resizethread, nil, mainstacksize);
	for(;;){
		readmouse(mctl);
		if(mctl->buttons & 4){
			lockdisplay(display);
			rc = menuhit(3, mctl, &menu, nil);
			unlockdisplay(display);
			switch(rc){
			case 0:
				zoom();
				break;
			case 1:
				stopdraw = 1;
				qlock(rend.l);
				xmin = -2;
				xmax = 1;
				ymin = -1;
				ymax = 1;
				rwakeup(&rend);
				qunlock(rend.l);
				break;
			case 2:
				threadexitsall(nil);
			}
		}
	}
}

Cursor reading = {
	{-1, -1},
	{0xff, 0x80, 0xff, 0x80, 0xff, 0x00, 0xfe, 0x00, 
	 0xff, 0x00, 0xff, 0x80, 0xff, 0xc0, 0xef, 0xe0, 
	 0xc7, 0xf0, 0x03, 0xf0, 0x01, 0xe0, 0x00, 0xc0, 
	 0x03, 0xff, 0x03, 0xff, 0x03, 0xff, 0x03, 0xff, },
	{0x00, 0x00, 0x7f, 0x00, 0x7e, 0x00, 0x7c, 0x00, 
	 0x7e, 0x00, 0x7f, 0x00, 0x6f, 0x80, 0x47, 0xc0, 
	 0x03, 0xe0, 0x01, 0xf0, 0x00, 0xe0, 0x00, 0x40, 
	 0x00, 0x00, 0x01, 0xb6, 0x01, 0xb6, 0x00, 0x00, }
};