#include <lib9.h>
#include <image.h>
#include <memimage.h>

#include "xmem.h"

static int
_cloadmemimage(Memimage *i, Rectangle r, uchar *data, int ndata)
{
	int y, bpl, c, cnt, offs;
	uchar mem[NMEM], *memp, *omemp, *emem, *linep, *elinep, *u, *eu;

	bpl = bytesperline(r, i->ldepth);
	u = data;
	eu = data+ndata;
	memp = mem;
	emem = mem+NMEM;
	y = r.min.y;
	linep = byteaddr(i, Pt(r.min.x, y));
	elinep = linep+bpl;
	for(;;){
		if(linep == elinep){
			if(++y == r.max.y)
				break;
			linep = byteaddr(i, Pt(r.min.x, y));
			elinep = linep+bpl;
		}
		if(u == eu)		/* buffer too small */
			return -1;
		c = *u++;
		if(c >= 128){
			for(cnt=c-128+1; cnt!=0 ;--cnt){
				if(u == eu)		/* buffer too small */
					return -1;
				if(linep == elinep)	/* phase error */
					return -1;
				*linep++ = *u;
				*memp++ = *u++;
				if(memp == emem)
					memp = mem;
			}
		}
		else{
			if(u == eu)	/* short buffer */
				return -1;
			offs = *u++ + ((c&3)<<8)+1;
			if(memp-mem < offs)
				omemp = memp+(NMEM-offs);
			else
				omemp = memp-offs;
			for(cnt=(c>>2)+NMATCH; cnt!=0; --cnt){
				if(linep == elinep)	/* phase error */
					return -1;
				*linep++ = *omemp;
				*memp++ = *omemp++;
				if(omemp == emem)
					omemp = mem;
				if(memp == emem)
					memp = mem;
			}
		}
	}
	return u-data;
}

int
cloadmemimage(Memimage *i, Rectangle r, uchar *data, int ndata)
{
	XImage *x;
	Xmem *xm;
	int n;

	if(!rectinrect(r, i->r))
		return -1;
	
	xm = i->X;
	x = getXdata(i, Rpt(r.min, r.min));
	n = _cloadmemimage(i, r, data, ndata);

	if(x != nil){
		putXdata(i, x, r);
		XDestroyImage(x);
		i->base = xm->wordp;
	}
	xdirtied(i);
	return n;
}