/* Copyright (c) 1992 AT&T - All rights reserved. */
#include <libc.h>
#include <libg.h>
#include <stdio.h>
#include "libgint.h"

#define COMPRESSMOUSE

#define Cursor xCursor
#define Font xFont
#define Event xEvent

#include <X11/Intrinsic.h>
#include <X11/StringDefs.h>
#include <X11/Shell.h>
#include "Gwin.h"

#ifndef XtSpecificationRelease
#define R3
#define XtAppInitialize(a,b,c,d,e,f,g,h,i) XtInitialize(0,b,c,d,e,f)
#define XtConvertAndStore(a,b,c,d,e) (XtConvert(a,b,c,d,e),1)
#define XtAppPending(a) XtPending()
#define XtAppProcessEvent(a,b) XtProcessEvent(b)
#define XtAppAddTimeOut(a,b,c,d) XtAddTimeOut(b,c,d)
#define XtAppAddInput(a,b,c,d,e) XtAddInput(b,c,d,e)
#define XtPointer caddr_t
#endif

#undef Cursor
#undef Font
#undef Event

/* libg globals */
Bitmap	screen;
Font	*font;

/* implementation globals */
Display		*_dpy;
Widget		_toplevel;
unsigned long	_fgpixel, _bgpixel;
XColor		_fgcolor, _bgcolor;
int		_ld2d[6] = { 1, 2, 4, 8, 16, 24 };
unsigned long	_ld2dmask[6] = { 0x1, 0x3, 0xF, 0xFF, 0xFFFF, 0x00FFFFFF };
Colormap	_libg_cmap;
int		_cmap_installed;

/* xbinit implementation globals */
#ifndef R3
static XtAppContext app;
#endif
static Widget widg;
static int exposed = 0;
static Atom wm_take_focus;
static Mouse lastmouse;

typedef struct Ebuf {
	struct Ebuf	*next;
	int		n;
	unsigned char	buf[2];
} Ebuf;

typedef struct Esrc {
	int	inuse;
	int	size;
	int	count;
	Ebuf	*head;
	Ebuf	*tail;
} Esrc;

#define	MAXINPUT	1024		/* number of queued input events */
#define MAXSRC 		10

static Esrc	esrc[MAXSRC];
static int	nsrc;


static int einitcalled = 0;
static int Smouse = -1;
static int Skeyboard = -1;
static int Stimer = -1;


static void	reshaped(int, int, int, int);
static void	gotchar(int);
static void	gotmouse(Gwinmouse *);
static int	log2(int);
static void	pixtocolor(Pixel, XColor *);
static Subfont	*XFontStructtoSubfont(XFontStruct *);
static Ebuf	*ebread(Esrc *);
static Ebuf	*ebadd(Esrc *);
static void	focinit(Widget);
static void	wmproto(Widget, XEvent *, String *, Cardinal *);
static void	waitevent(void);

static Errfunc	onerr;

String _fallbacks[] = {
	"*gwin.width: 400",
	"*gwin.height: 400",
	NULL
};

#ifndef R3
static char *shelltrans = 
	"<ClientMessage> WM_PROTOCOLS : WMProtocolAction()";
static XtActionsRec wmpactions[] = {
	{"WMProtocolAction", wmproto}
};
#endif

	/* too many X options */
static XrmOptionDescRec optable[] = {
	{"-p9fn",	"*p9font",	XrmoptionSepArg,        (caddr_t)NULL},
	{"-p9font",	"*p9font",	XrmoptionSepArg,        (caddr_t)NULL},
};

void
xtbinit(Errfunc f, char *class, int *pargc, char **argv, char **fallbacks)
{
	int n;
	unsigned int depth;
	XFontStruct *xf;
	Subfont *subfont;
	String fontname;
	Arg args[10];
	char *p;
	XSetWindowAttributes attr;
	int compose;

	if(!class && argv[0]){
		p = strrchr(argv[0], '/');
		if(p)
			class = XtNewString(p+1);
		else
			class = XtNewString(argv[0]);
		if(class[0] >= 'a' && class[0] <= 'z')
			class[0] += 'A' - 'a';
	}
	onerr = f;
	if (!fallbacks)
		fallbacks = _fallbacks;
	n = 0;
	XtSetArg(args[n], XtNinput, TRUE);		n++;
	_toplevel = XtAppInitialize(&app, class,
			optable, sizeof(optable)/sizeof(optable[0]),
			pargc, argv, fallbacks, args, n);

	n = 0;
	XtSetArg(args[n], XtNreshaped, reshaped);	n++;
	XtSetArg(args[n], XtNgotchar, gotchar);		n++;
	XtSetArg(args[n], XtNgotmouse, gotmouse);	n++;
	widg = XtCreateManagedWidget("gwin", gwinWidgetClass, _toplevel, args, n);

	n = 0;
	XtSetArg(args[n], XtNforeground, &_fgpixel);	n++;
	XtSetArg(args[n], XtNbackground, &_bgpixel);	n++;
	XtSetArg(args[n], XtNdepth, &depth);		n++;
	XtSetArg(args[n], XtNfont, &xf);		n++;
	XtSetArg(args[n], XtNp9font, &fontname);	n++;
	XtSetArg(args[n], XtNcomposeMod, &compose);	n++;
	XtGetValues(widg, args, n);

	if (compose < 0 || compose > 5) {
		n = 0;
		XtSetArg(args[n], XtNcomposeMod, 0);	n++;
		XtSetValues(widg, args, n);
	}

	_dpy = XtDisplay(widg);
	screen.id = 0;
	XtRealizeWidget(_toplevel);
	pixtocolor(_fgpixel, &_fgcolor);
	pixtocolor(_bgpixel, &_bgcolor);
	screen.id = (int) XtWindow(widg);
	screen.ldepth = log2(depth);
	screen.flag = SCR;
	if(_fgpixel != 0)
		screen.flag |= BL1;
	if(depth == 1)
		screen.flag |= DP1;
	font = 0;
	subfont = 0;
	if (fontname) {
		font = rdfontfile(fontname, screen.ldepth);
		if (!font || charwidth(font, (Rune) ' ') == 0) {
			subfont = getsubfont(fontname);
			if (!subfont)
				subfont = XFontStructtoSubfont(xf);
			font = mkfont(subfont);
		}
	} else {
		subfont = XFontStructtoSubfont(xf);
		font = mkfont(subfont);
	}
	/* leave screen rect at all zeros until reshaped() sets it */
	while(!exposed) {
		XFlush(_dpy);
		XtAppProcessEvent(app, XtIMXEvent);
	}
	XFlush(_dpy);
	focinit(_toplevel);
}

static void
focinit(Widget w)
{
#ifndef R3
	XrmValue src, dst;

	src.addr = "WM_TAKE_FOCUS";
	src.size = strlen((char *)src.addr)+1;
	dst.addr = (XtPointer) &wm_take_focus;
	dst.size = sizeof(Atom);
	XtConvertAndStore(w, XtRString, &src, XtRAtom, &dst);
	XSetWMProtocols(XtDisplay(w), XtWindow(w), &wm_take_focus, 1);
	XtAppAddActions(app, wmpactions, XtNumber(wmpactions));
	XtAugmentTranslations(w, XtParseTranslationTable(shelltrans));
#endif
}

#ifndef R3
static void
wmproto(Widget w, XEvent *e , String *p, Cardinal *np)
{
	Time t;

	if(e->type == ClientMessage &&
          (Atom)(e->xclient.data.l[0]) == wm_take_focus) {
		t = (Time) e->xclient.data.l[1];
		XtCallAcceptFocus(widg, &t);
	}
}
#endif

static void
reshaped(int minx, int miny, int maxx, int maxy)
{
	Ebuf *eb;
	Mouse m;

	screen.r = Rect(minx, miny, maxx, maxy);
	screen.clipr = screen.r;
	if (screen.id) {
		exposed = 1;
		ereshaped(screen.r);
	}
	if(einitcalled){
		/*
		 * Cause a mouse event, so programs like sam
		 * will get out of eread and REALLY do the reshape
		 */
		eb = ebadd(&esrc[Smouse]);
		if (eb == 0)
			berror("eballoc can't malloc");
		memcpy((void*)eb->buf, (void*)&lastmouse, sizeof lastmouse);
		esrc[Smouse].count++;
	}
}

static void
gotchar(int c)
{
	Ebuf *eb;

	if(!einitcalled || Skeyboard == -1)
		return;
	eb = ebadd(&esrc[Skeyboard]);
	if (eb == 0)
		berror("eballoc can't malloc");
	BPSHORT(eb->buf, (unsigned short)(c & 0xffff));
	esrc[Skeyboard].count++;
}

static void
gotmouse(Gwinmouse *gm)
{
	Ebuf *eb;
	Mouse m;

	if(!einitcalled || Smouse == -1)
		return;
	m.buttons = gm->buttons;
	m.xy.x = gm->xy.x;
	m.xy.y = gm->xy.y;
	m.msec = gm->msec;
	lastmouse = m;
	eb = ebadd(&esrc[Smouse]);
	if (eb == 0)
		berror("eballoc can't malloc");
	memcpy((void*)eb->buf, (void*)&m, sizeof m);
	esrc[Smouse].count++;
}

static void
gotinput(XtPointer cldata, int *pfd, XtInputId *id)
{
	Ebuf *eb, *lasttail, *newe;
	Esrc *es;
	int n;

	if(!einitcalled)
		return;
	es = (Esrc *)cldata;
	if (es->count >= MAXINPUT)
		return;
	lasttail = es->tail;
	eb = ebadd(es);
	if (eb == 0)
		return;
	if(es->size){
		n = read(*pfd, (char *)eb->buf, es->size);
		if (n < 0)
			n = 0;
		if(n < es->size) {
			newe = realloc(eb, sizeof(Ebuf)+n);
			newe->n = n;
			if (es->head == eb)
				es->head = newe;
			else
				lasttail->next = newe;
			es->tail = newe;
		}
	}
	es->count++;
}

static void
gottimeout(XtPointer cldata, XtIntervalId *id)
{
	if(!einitcalled || Stimer == -1)
		return;
	/*
	 * Don't queue up timeouts, because there's
	 * too big a danger that they might pile up
	 * too quickly.
	 */
	esrc[Stimer].head = (Ebuf *)1;
	esrc[Stimer].count = 1;
	XtAppAddTimeOut(app, (long)cldata, gottimeout, cldata);
}

static int
log2(int n)
{
	int i, v;

	for(i=0, v=1; i < 6; i++, v<<=1)
		if(n <= v)
			break;
	return i;
}

static void
pixtocolor(Pixel p, XColor *pc)
{
#ifdef R3
	Colormap cmap;
	Arg args[2];
	int n;

	n = 0;
	XtSetArg(args[n], XtNcolormap, &cmap);	n++;
	XtGetValues(_toplevel, args, n);
	pc->pixel = p;
	XQueryColor(_dpy, cmap, pc);
#else
	XrmValue xvf, xvt;

	xvf.size = sizeof(Pixel);
	xvf.addr = (XtPointer)&p;
	xvt.size = sizeof(XColor);
	xvt.addr = (XtPointer)pc;
	if(!XtConvertAndStore(_toplevel, XtRPixel, &xvf, XtRColor, &xvt))
		pc->pixel = p;	/* maybe that's enough */
#endif
}

unsigned long
rgbpix(Bitmap *b, RGB col)
{
	XColor c;
	Colormap cmap;
	Arg args[2];
	int n, depth, dr, dg, db;
	RGB map[256], *m;
	unsigned long d, max, pixel;

	if (!_cmap_installed) {
		n = 0;
		XtSetArg(args[n], XtNcolormap, &cmap);	n++;
		XtGetValues(_toplevel, args, n);
		c.red = col.red>>16;
		c.green = col.green>>16;
		c.blue = col.blue>>16;
		c.flags = DoRed|DoGreen|DoBlue;
		if(XAllocColor(_dpy, cmap, &c))
			return (unsigned long)(c.pixel);
	}
	depth = _ld2d[screen.ldepth];
	rdcolmap(&screen, map);
	max = -1;
	for (n = 0, m = map; n < (1 << depth); n++, m++)
	{
		dr = m->red - col.red;
		dg = m->green - col.green;
		db = m->blue - col.blue;
		d = dr*dr+dg*dg+db*db;
		if (d < max || max == -1)
		{
			max = d;
			pixel = n;
		}
	}
	return pixel;
}

void
rdcolmap(Bitmap *b, RGB *map)
{
	XColor cols[256];
	int i, n, depth;
	Colormap cmap;
	Arg args[2];

	if (_cmap_installed) {
		cmap = _libg_cmap;
	} else {
		i = 0;
		XtSetArg(args[i], XtNcolormap, &cmap);	i++;
		XtGetValues(_toplevel, args, i);
	}

	depth = _ld2d[screen.ldepth];
	n = 1 << depth;
	if (depth == 1) {
		map[0].red = map[0].green = map[0].blue = ~0;
		map[1].red = map[1].green = map[1].blue = 0;
	}
	else {
		if (n > 256) {
			berror("rdcolmap bitmap too deep");
			return;
		}
		for (i = 0; i < n; i++)
			cols[i].pixel = i;
		XQueryColors(_dpy, cmap, cols, n);
		for (i = 0; i < n; i++) {
			map[i].red = (cols[i].red << 16) | cols[i].red;
			map[i].green = (cols[i].green << 16) | cols[i].green;
			map[i].blue = (cols[i].blue << 16) | cols[i].blue;
		}
	}
}

void
wrcolmap(Bitmap *b, RGB *map)
{
	int i, n, depth;
	Screen *scr;
	XColor cols[256];
	Arg args[2];
	XVisualInfo vi;
	Window w;

	scr = XtScreen(_toplevel);
	depth = _ld2d[screen.ldepth];
	n = 1 << depth;
	if (n > 256) {
		berror("wrcolmap bitmap too deep");
		return;
	} else if (depth > 1) {
		for (i = 0; i < n; i++) {
			cols[i].red = map[i].red >> 16;
			cols[i].green = map[i].green >> 16;
			cols[i].blue = map[i].blue >> 16;
			cols[i].pixel = i;
			cols[i].flags = DoRed|DoGreen|DoBlue;
		}
		if (!XMatchVisualInfo(_dpy, XScreenNumberOfScreen(scr),
					depth, PseudoColor, &vi)) {
			berror("wrcolmap can't get visual");
			return;
		}
		w = XtWindow(_toplevel);
		_libg_cmap = XCreateColormap(_dpy, w, vi.visual, AllocAll);
		XStoreColors(_dpy, _libg_cmap, cols, n);

		i = 0;
		XtSetArg(args[i], XtNcolormap, _libg_cmap);	i++;
		XtSetValues(_toplevel, args, i);
		_cmap_installed = 1;
	}
}

Subfont *
getsubfont(char *s)
{
	XFontStruct *fp;

	if(!s)
		return 0;
	fp = XLoadQueryFont(_dpy, s);
	if(!fp)
		return 0;
	return XFontStructtoSubfont(fp);
}

static Subfont *
XFontStructtoSubfont(XFontStruct *fp)
{
	XCharStruct *cp;
	Subfont *f;
	int min, max;
	int i;

	if(!fp)
		berror("no font");
	f = (Subfont *)malloc(sizeof(Subfont));
	if(!f)
		berror("XFontStructtoSubfont malloc");
        min = fp->min_byte1;
        max = fp->max_byte1;
	f->minrow = min;
	f->mincol = fp->min_char_or_byte2;
	f->width = fp->max_char_or_byte2-fp->min_char_or_byte2+1;
	f->n = f->width;
	f->minchar = 0;
	f->maxchar = fp->max_char_or_byte2;
	if (min || max) {
		f->maxchar |= (max<<8);
		f->n *= (max-min+1);
	}
	f->id = fp->fid;
	f->height = fp->max_bounds.ascent + fp->max_bounds.descent;
	f->ascent = fp->max_bounds.ascent;
	f->info = (Fontchar *)malloc((f->n+1)*sizeof(Fontchar));
	if(!f->info)
		berror("getsubfont malloc");
	memset((void*)f->info, 0, (f->n+1)*sizeof(Fontchar));
	for(i = 0; i < f->n; i++){
		if(fp->per_char)
			cp = fp->per_char + i;
		else
			cp = &fp->max_bounds;
		f->info[i].left = cp->lbearing;
		f->info[i].cwidth = cp->rbearing - cp->lbearing;
		f->info[i].width = cp->width;
		f->info[i].top = 0;
		f->info[i].bottom = f->height;
	}
	XFreeFontInfo(0, fp, 0);
	return f;
}

int
scrollfwdbut(void)
{
	Arg arg;
	Boolean v;
	String s;

	XtSetArg(arg, XtNscrollForwardR, &v);
	XtGetValues(widg, &arg, 1);
	return v ? 3 : 1;
}

void
einit(unsigned long keys)
{
	/*
	 * Make sure Smouse = log2(Emouse) and Skeyboard == log2(Ekeyboard)
	 */
	nsrc = 0;
	if(keys&Emouse){
		Smouse = 0;
		esrc[Smouse].inuse = 1;
		esrc[Smouse].size = sizeof(Mouse);
		esrc[Smouse].count = 0;
		nsrc = Smouse+1;
	}
	if(keys&Ekeyboard){
		Skeyboard = 1;
		esrc[Skeyboard].inuse = 1;
		esrc[Skeyboard].size = 1;
		esrc[Skeyboard].count = 0;
		if(Skeyboard >= nsrc)
			nsrc = Skeyboard+1;
	}
	einitcalled = 1;
}

unsigned long
estart(unsigned long key, int fd, int n)
{
	int i;

	if(fd < 0)
		berror("bad fd to estart");
	if(n <= 0 || n > EMAXMSG)
		n = EMAXMSG;
	for(i=0; i<MAXSRC; i++)
		if((key & ~(1<<i)) == 0 && !esrc[i].inuse){
			if(nsrc <= i)
				nsrc = i+1;
			esrc[i].inuse = 1;
			esrc[i].size = n;
			esrc[i].count = 0;
			XtAppAddInput(app, fd, (XtPointer)XtInputReadMask,
				gotinput, (XtPointer) &esrc[i]);
			return 1<<i;
		}
	return 0;
}

unsigned long
etimer(unsigned long key, long n)
{
	int i;

	if(Stimer != -1)
		berror("timer started twice");
	if(n <= 0)
		n = 1000;
	for(i=0; i<MAXSRC; i++)
		if((key & ~(1<<i)) == 0 && !esrc[i].inuse){
			if(nsrc <= i)
				nsrc = i+1;
			esrc[i].inuse = 1;
			esrc[i].size = 0;
			esrc[i].count = 0;
			XtAppAddTimeOut(app, n, gottimeout, (XtPointer)n);
			Stimer = i;
			return 1<<i;
		}
	return 0;
}

unsigned long
event(Event *e)
{
	return eread(~0L, e);
}

unsigned long
eread(unsigned long keys, Event *e)
{
	Ebuf *eb;
	int i;

	if(keys == 0)
		return 0;
		/* Give Priority to X events */
	if (XtAppPending(app) & XtIMXEvent)
		XtAppProcessEvent(app, XtIMXEvent);

	for(;;){
		for(i=0; i<nsrc; i++)
			if((keys & (1<<i)) && esrc[i].head){
				if(i == Smouse)
					e->mouse = emouse();
				else if(i == Skeyboard)
					e->kbdc = ekbd();
				else if(i == Stimer) {
					esrc[i].head = 0;
					esrc[i].count = 0;
				} else {
					eb = ebread(&esrc[i]);
					e->n = eb->n;
					if(e->n > 0)
						memcpy((void*)e->data, (void*)eb->buf, e->n);
					free(eb);
				}
				return 1<<i;
			}
		waitevent();
	}
}

void
eflush(unsigned long keys)
{
	int i;
	Ebuf *eb, *enext;

	if(keys == 0)
		return;

	for(i=0; i<nsrc; i++)
		if((keys & (1<<i))){
			for (eb = esrc[i].head; eb; eb = enext) {
				enext = eb->next;
				free(eb);
			}
			esrc[i].count = 0;
			esrc[i].head = 0;
			esrc[i].tail = 0;
		}
}

Mouse
emouse(void)
{
	Mouse m;
	Ebuf *eb;

	if(!esrc[Smouse].inuse)
		berror("mouse events not selected");
	eb = ebread(&esrc[Smouse]);
	memcpy((void*)&m, (void*)eb->buf, sizeof(Mouse));
	free(eb);
	return m;
}

int
ekbd(void)
{
	Ebuf *eb;
	int c;

	if(!esrc[Skeyboard].inuse)
		berror("keyboard events not selected");
	eb = ebread(&esrc[Skeyboard]);
	c = BGSHORT(eb->buf);
	free(eb);
	return c;
}

int
ecanread(unsigned long keys)
{
	int i;

	for(;;){
		for(i=0; i<nsrc; i++){
			if((keys & (1<<i)) && esrc[i].head)
				return 1<<i;
		}
		if(XtAppPending(app))
			waitevent();
		else
			return 0;
	}
}

int
ecanmouse(void)
{
	if(Smouse == -1)
		berror("mouse events not selected");
	return ecanread(Emouse);
}

int
ecankbd(void)
{
	if(Skeyboard == -1)
		berror("keyboard events not selected");
	return ecanread(Ekeyboard);
}

static Ebuf*
ebread(Esrc *s)
{
	Ebuf *eb;

	while(s->head == 0)
		waitevent();
	eb = s->head;
#ifdef COMPRESSMOUSE
	if(s == &esrc[Smouse]) {
		while(eb->next) {
			s->head = eb->next;
			s->count--;
			free(eb);
			eb = s->head;
		}
	}
#endif
	s->head = s->head->next;
	if(s->head == 0) {
		s->tail = 0;
		s->count = 0;
	} else
		s->count--;
	return eb;
}

static Ebuf*
ebadd(Esrc *s)
{
	Ebuf *eb;
	int m;

	m = sizeof(Ebuf);
	if(s->size > 1)
		m += (s->size-1);	/* overestimate, because of alignment */
	eb = (Ebuf *)malloc(m);
	if(eb) {
		eb->next = 0;
		eb->n = s->size;
		if(s->tail){
			s->tail->next = eb;
			s->tail = eb;
		}else
			s->head = s->tail = eb;
	}
	return eb;
}

void
berror(char *s)
{
	if(onerr)
		(*onerr)(s);
	else{
		fprintf(stderr, "libg error: %s:\n", s);
		exit(1);
	}
}

void
bflush(void)
{
	while(XtAppPending(app) & XtIMXEvent)
		waitevent();
}

static void
waitevent(void)
{
	XFlush(_dpy);
	if (XtAppPending(app) & XtIMXEvent)
		XtAppProcessEvent(app, XtIMXEvent);
	else
		XtAppProcessEvent(app, XtIMAll);
}
		
int
snarfswap(char *s, int n, char **t)
{
	*t = GwinSelectionSwap(widg, s);
	if (*t)
		return strlen(*t);
	return 0;
}

int scrpix(int *w, int *h)
{
	if (w)
		*w = WidthOfScreen(XtScreen(_toplevel));
	if (h)
		*h = HeightOfScreen(XtScreen(_toplevel));
	return 1;
}

#ifdef DEBUG
/* for debugging */
printgc(char *msg, GC g)
{
	XGCValues v;

	XGetGCValues(_dpy, g, GCFunction|GCForeground|GCBackground|GCFont|
			GCTile|GCFillStyle|GCStipple, &v);
	fprintf(stderr, "%s: gc %x\n", msg, g);
	fprintf(stderr, "  fg %d bg %d func %d fillstyle %d font %x tile %x stipple %x\n",
		v.foreground, v.background, v.function, v.fill_style,
		v.font, v.tile, v.stipple);
}
#endif