#include "all.h"
#include "mem.h"

enum {
	Width		= 160,
	Height		= 25,

	Attr		= 7,		/* white on black */
};

#define BASE	((uchar*)(KZERO+0xB8000))

static int pos;
static int screeninitdone;
static Lock screenlock;

static uchar
cgaregr(int index)
{
	outb(0x3D4, index);
	return inb(0x3D4+1) & 0xFF;
}

static void
cgaregw(int index, int data)
{
	outb(0x3D4, index);
	outb(0x3D4+1, data);
}

static void
movecursor(void)
{
	cgaregw(0x0E, (pos/2>>8) & 0xFF);
	cgaregw(0x0F, pos/2 & 0xFF);
	BASE[pos+1] = Attr;
}

static void
cgascreenputc(int c)
{
	int i;

	if(c == '\r')
		return;
	if(c == '\n'){
		pos = pos/Width;
		pos = (pos+1)*Width;
	}
	else if(c == '\t'){
		i = 4 - ((pos/2)&3);
		while(i-->0)
			cgascreenputc(' ');
	}
	else if(c == '\b'){
		if(pos >= 2)
			pos -= 2;
		cgascreenputc(' ');
		pos -= 2;
	}
	else{
		BASE[pos++] = c;
		BASE[pos++] = Attr;
	}
	if(pos >= Width*Height){
		memmove(BASE, &BASE[Width], Width*(Height-1));
		memset(&BASE[Width*(Height-1)], 0, Width);
		pos = Width*(Height-1);
	}
	movecursor();
}

static void
screeninit(void)
{
	lock(&screenlock);
	if(screeninitdone == 0){
		pos = cgaregr(0x0E)<<8;
		pos |= cgaregr(0x0F);
		pos *= 2;
		screeninitdone = 1;
	}
	unlock(&screenlock);
}

void
cgaputs(char* s, int n)
{
	if(screeninitdone == 0)
		screeninit();
	while(n-- > 0)
		cgascreenputc(*s++);
}

void
cgaputc(int c)
{
	if(screeninitdone == 0)
		screeninit();
	cgascreenputc(c);
}