/* $Id$ */
#include <Mw/Milsko.h>

#include "../../external/stb_ds.h"

#define OpenerSize 10
#define LineSpace 16

static void vscroll_changed(MwWidget handle, void* user, void* call) {
	MwTreeView tv = handle->parent->internal;

	(void)user;
	(void)call;

	tv->changed = 1;
}

static void recursion(MwWidget handle, MwTreeViewEntry* tree, MwLLColor base, MwLLColor text, MwPoint* p, int next, int shift, int* skip, int* shared){
	int i;
	MwPoint l[2];
	int skipped = 0;
	if((*skip) > 0){
		(*skip)--;
		skipped = 1;
		if(p->y == MwDefaultBorderWidth(handle)) p->y -= MwTextHeight(handle, "M");
	}else if((*shared) < (MwGetInteger(handle, MwNheight) / MwTextHeight(handle, "M") + 2)){
			MwRect r;
		p->x += shift;
		p->y += MwTextHeight(handle, "M") / 2;

		if(shift > 0){
			l[0] = *p;
			l[0].x -= LineSpace / 2;
			l[1] = *p;
			MwLLLine(handle->lowlevel, &l[0], text);
	
			l[0] = *p;
			l[0].x -= LineSpace / 2;
			l[1] = l[0];
			l[0].y -= MwTextHeight(handle, "M") / 2;
			if(next){
				l[1].y += MwTextHeight(handle, "M") / 2;
			}
			MwLLLine(handle->lowlevel, &l[0], text);
		}
		if(tree->tree != NULL){
			r.width = OpenerSize;
			r.height = OpenerSize;
			r.x = p->x - LineSpace + (LineSpace - r.width) / 2;
			r.y = p->y - MwTextHeight(handle, "M") / 2 + (MwTextHeight(handle, "M") - r.height) / 2;
			MwDrawWidgetBack(handle, &r, base, tree->opened, 1);
		}
		if(tree->pixmap != NULL){

			r.height = MwTextHeight(handle, "M");
			r.width = r.height * tree->pixmap->common.width / tree->pixmap->common.height;
			r.x = p->x;
			r.y = p->y - MwTextHeight(handle, "M") / 2;

			MwLLDrawPixmap(handle->lowlevel, &r, tree->pixmap);
		}
		p->x += MwGetInteger(handle->parent, MwNleftPadding);
		MwDrawText(handle, p, tree->label, 0, MwALIGNMENT_BEGINNING, text);
		p->x -= MwGetInteger(handle->parent, MwNleftPadding);

		p->y += MwTextHeight(handle, "M") / 2;
		p->x -= shift;

		(*shared)++;
	}
	if(!tree->opened) return;
	for(i = 0; i < arrlen(tree->tree); i++){
		l[0] = *p;
		l[0].x += shift + LineSpace / 2;
		l[0].y += MwTextHeight(handle, "M") - (MwTextHeight(handle, "M") - OpenerSize) / 2;

		recursion(handle, &tree->tree[i], base, text, p, i != (arrlen(tree->tree) - 1) ? 1 : 0, shift + LineSpace, skip, shared);

		l[1] = *p;
		l[1].x += shift + LineSpace / 2;

		if(skipped && p->y > l[0].y){
			skipped = 0;
		}
		if(!skipped && i != (arrlen(tree->tree) - 1)) MwLLLine(handle->lowlevel, &l[0], text);

	}
}

static void frame_draw(MwWidget handle) {
	MwRect r;
	MwTreeView tv = handle->parent->internal;
        MwLLColor base = MwParseColor(handle, MwGetText(handle, MwNbackground));
        MwLLColor text = MwParseColor(handle, MwGetText(handle, MwNforeground));
	MwPoint p;
	int shared = 0;
	int i;
	int skip = MwGetInteger(tv->vscroll, MwNvalue) * (MwGetInteger(tv->vscroll, MwNmaxValue) - MwGetInteger(tv->vscroll, MwNareaShown)) / MwGetInteger(tv->vscroll, MwNmaxValue);

        r.x      = 0;
        r.y      = 0;
        r.width  = MwGetInteger(handle, MwNwidth);
        r.height = MwGetInteger(handle, MwNheight);

	p.x = MwDefaultBorderWidth(handle);
	p.y = MwDefaultBorderWidth(handle);

	for(i = 0; i < arrlen(tv->tree); i++){
		if(shared > (r.height / MwTextHeight(handle, "M"))) break;
		recursion(handle, &tv->tree[i], base, text, &p, 0, 0, &skip, &shared);
	}

        MwDrawFrame(handle, &r, base, 1);

	MwLLFreeColor(text);
        MwLLFreeColor(base);
}

static int recursive_length(MwTreeViewEntry* e){
	int l = 0;
	int i;
	for(i = 0; i < arrlen(e); i++){
		if(e[i].opened && e[i].tree != NULL) l += recursive_length(e[i].tree);
		l++;
	}
	return l;
}

static void resize(MwWidget handle) {
	MwTreeView tv = handle->internal;
	int	   w  = MwGetInteger(handle, MwNwidth);
	int	   h  = MwGetInteger(handle, MwNheight);
	int	   ih = 0;

	if(tv->vscroll == NULL) {
		tv->vscroll = MwCreateWidget(MwScrollBarClass, "vscroll", handle, w - 16, 0, 16, h);
		MwAddUserHandler(tv->vscroll, MwNchangedHandler, vscroll_changed, NULL);
		MwSetInteger(tv->vscroll, MwNvalue, 7);
	} else {
		MwVaApply(tv->vscroll,
			  MwNx, w - 16,
			  MwNy, 0,
			  MwNwidth, 16,
			  MwNheight, h,
			  NULL);
	}

	if(tv->frame == NULL) {
		tv->frame	       = MwVaCreateWidget(MwFrameClass, "frame", handle, 0, 0, w - 16, h,
							  MwNhasBorder, 1,
							  MwNinverted, 1,
							  NULL);
		tv->frame->draw_inject = frame_draw;
	} else {
		MwVaApply(tv->frame,
			  MwNx, 0,
			  MwNy, 0,
			  MwNwidth, w - 16,
			  MwNheight, h,
			  NULL);
	}

	ih = recursive_length(tv->tree);
	if(ih == 0) ih = 1;

	MwVaApply(tv->vscroll,
		  MwNareaShown, h / MwTextHeight(handle, "M"),
		  MwNmaxValue, ih,
		  NULL);

	if(ih <= (h / MwTextHeight(handle, "M"))){
		MwLLShow(tv->vscroll->lowlevel, 0);
		MwSetInteger(tv->frame, MwNwidth, w);
	}else{
		MwLLShow(tv->vscroll->lowlevel, 1);
	}
}

static int create(MwWidget handle) {
	MwTreeView tv = malloc(sizeof(*tv));
	MwTreeViewEntry e;
	int i, j, k, l, c = 0;
	MwLLPixmap p = MwLoadIcon(handle, MwIconFile);
	memset(tv, 0, sizeof(*tv));

	for(i = 0; i < 10; i++){
		char str[32];

		sprintf(str, "hello %d", ++c);

		e.label = MwStringDupliacte(str);
		e.pixmap = p;
		e.tree = NULL;
		e.opened = 1;
		for(j = 0; j < 3; j++){
			MwTreeViewEntry e2;
			sprintf(str, "hello %d", ++c);
			e2.label = MwStringDupliacte(str);
			e2.pixmap = p;
			e2.tree = NULL;
			e2.opened = j == 1 ? 0 : 1;
			for(k = 0; k < 3; k++){
				MwTreeViewEntry e3;
				sprintf(str, "hello %d", ++c);
				e3.label = MwStringDupliacte(str);
				e3.pixmap = p;
				e3.tree = NULL;
				e3.opened = 1;
				for(l = 0; l < 3; l++){
					MwTreeViewEntry e4;
					sprintf(str, "hello %d", ++c);
					e4.label = MwStringDupliacte(str);
					e4.pixmap = p;
					e4.tree = NULL;
					e4.opened = 1;
					arrput(e3.tree, e4);
				}
				arrput(e2.tree, e3);
			}
			arrput(e.tree, e2);
		}
		arrput(tv->tree, e);
	}

	handle->internal = tv;

	MwSetDefault(handle);

	MwSetInteger(handle, MwNsingleClickSelectable, 0);
	MwSetInteger(handle, MwNleftPadding, 16);

	resize(handle);
	tv->changed = 0;

	return 0;
}

static void destroy(MwWidget handle) {
	free(handle->internal);
}

static void draw(MwWidget handle) {
	MwLLColor c = MwParseColor(handle, MwGetText(handle, MwNbackground));
	MwRect	  r;

	r.x	 = 0;
	r.y	 = 0;
	r.width	 = MwGetInteger(handle, MwNwidth);
	r.height = MwGetInteger(handle, MwNheight);
	MwDrawRect(handle, &r, c);

	MwLLFreeColor(c);
}

static void prop_change(MwWidget handle, const char* prop) {
	if(strcmp(prop, MwNwidth) == 0 || strcmp(prop, MwNheight) == 0) resize(handle);
}

static void tick(MwWidget handle) {
	MwTreeView tv = handle->internal;

	if(tv->changed) {
		tv->changed = 0;
		MwForceRender(tv->frame);
	}
}

MwClassRec MwTreeViewClassRec = {
    create,	 /* create */
    destroy,	 /* destroy */
    draw,	 /* draw */
    NULL,	 /* click */
    NULL,	 /* parent_resize */
    prop_change, /* prop_change */
    NULL,	 /* mouse_move */
    NULL,	 /* mouse_up */
    NULL,	 /* mouse_down */
    NULL,	 /* key */
    NULL,	 /* execute */
    tick,	 /* tick */
    NULL,
    NULL,
    NULL};
MwClass MwTreeViewClass = &MwTreeViewClassRec;