/*
 *  Sword keyboard -- USAR SPIcoder + Fujitsu Keyboard
 *
 * BUG: the SSP/SPI support should be broken out (and done properly!)
 */

#include "u.h"
#include "../port/lib.h"
#include "mem.h"
#include "dat.h"
#include "io.h"
#include "fns.h"
#include "keyboard.h"

static void	fujkbdputc(Queue*, int);

Rune kbtab[] = 
{	
[0x00]	No,   LAlt,   No,   No,   No,   No,   No,   No, 
[0x08]	No,  96, '\\', '\t', 'z', 'a', 'x',   No, 
[0x10]	No,   No,   LShift,   No,   No,   No,   No,   No, 
[0x18]	No,   LCtrl, No,   No,   No,   No,   No,   No, 
[0x20]	No,   Meta,   No,   No,   No,   No,   No,   No, 
[0x28]	No, Esc,Del, 'q',  Caps, 's', 'c', '3', 
[0x30]	No, '1',   No, 'w',   No, 'd', 'v', '4', 
[0x38]	No, '2', 't', 'e',   No, 'f', 'b', '5', 
[0x40]	No, '9', 'y', 'r', 'k', 'g', 'n', '6', 
[0x48]	No, '0', 'u', 'o', 'l', 'h', 'm', '7', 
[0x50]	No, '-', 'i', 'p', ';', 'j', ',', '8', 
[0x58]	No, '=','\n', '[','\'', '/', '.', Latin, 
[0x60]	No,   No,   RShift,   No,   No,   No,   No,   No, 
[0x68]	No,  '\b',  Down, ']',  Up,  Left, ' ',  Right, 
[0x70]	No,   No,   No,   No,   No,   No,   No,   No, 
[0x78]	No,   No,   No,   No,   No,   No,   No,   No
};

Rune kbtabshift[] =
{	
[0x00]	No,   LAlt,   No,   No,   No,   No,   No,   No, 
[0x08]	No, '~', '|', BackTab, 'Z', 'A', 'X',   No, 
[0x10]	No,   No,   LShift,   No,   No,   No,   No,   No, 
[0x18]	No,   LCtrl, No,   No,   No,   No,   No,   No, 
[0x20]	No,   Meta,   No,   No,   No,   No,   No,   No, 
[0x28]	No, Esc,Del, 'Q', Caps, 'S', 'C', '#', 
[0x30]	No, '!',   No, 'W',   No, 'D', 'V', '$', 
[0x38]	No, '@', 'T', 'E',   No, 'F', 'B', '%', 
[0x40]	No, '(', 'Y', 'R', 'K', 'G', 'N', '^', 
[0x48]	No, ')', 'U', 'O', 'L', 'H', 'M', '&', 
[0x50]	No, '_', 'I', 'P', ':', 'J', '<', '*', 
[0x58]	No, '+','\n', '{', '"', '?', '>', Latin, 
[0x60]	No,   No,   RShift,   No,   No,   No,   No,   No, 
[0x68]	No,  '\b',  Down, '}',  Up,  Left, ' ',  Right, 
[0x70]	No,   No,   No,   No,   No,   No,   No,   No, 
[0x78]	No,   No,   No,   No,   No,   No,   No,   No
};

Rune kbtabmeta[] =
{	
[0x00]	No, LAlt, No, No, No, No, No, No,
[0x08]	No, No, No, No, No, No, No, No,
[0x10]	No, No, LShift, No, No, No, No, No,
[0x18]	No, LCtrl, No, No, No, No, No, No,
[0x20]	No, Meta, No, No, No, No, No, No,
[0x28]	No, No, Del, No, Caps, No, No, No,
[0x30]	No, No, No, No, No, No, No, No,
[0x38]	No, No, No, No, No, No, No, No,
[0x40]	No, No, No, No, No, No, No, No,
[0x48]	No, No, No, No, No, No, No, No,
[0x50]	No, Num, No, No, No, No, No, No,
[0x58]	No, No, No, No, No, No, No, Latin,
[0x60]	No, No, RShift, No, No, No, No, No,
[0x68]	No, No, Pgdown, No, Pgup, Home, No, End,
[0x70]	No, No, No, No, No, No, No, No,
[0x78]	No, No, No, No, No, No, No, No,
};

Rune kbtabnuml[] =
{	
[0x00]	No,   LAlt,   No,   No,   No,   No,   No,   No, 
[0x08]	No,  96, 112, '\t', 'z', 'a', 'x',   No, 
[0x10]	No,   No,   LShift,   No,   No,   No,   No,   No, 
[0x18]	No,   LCtrl, No,   No,   No,   No,   No,   No, 
[0x20]	No,   Meta,   No,   No,   No,   No,   No,   No, 
[0x28]	No, Esc,Del, 'q',  Caps, 's', 'c', '3', 
[0x30]	No, '1',   No, 'w',   No, 'd', 'v', '4', 
[0x38]	No, '2', 't', 'e',   No, 'f', 'b', '5', 
[0x40]	No, '9', 'y', 'r', '2', 'g', 'n', '6', 
[0x48]	No, '*', '4', '6', '3', 'h', '0', '7', 
[0x50]	No, '-', '5', '-', '+', '1', ',', '8', 
[0x58]	No, '=','\n', '[','\'', '/', '.', Latin, 
[0x60]	No,   No,   RShift,   No,   No,   No,   No,   No, 
[0x68]	No,  '\b',  Down, ']',  Up,  Left, ' ',  Right, 
[0x70]	No,   No,   No,   No,   No,   No,   No,   No, 
[0x78]	No,   No,   No,   No,   No,   No,   No,   No
};

int pressed[NShifts];
int toggle[NShifts] = { [Caps-Shift]1, [Num-Shift]1, [Meta-Shift]1 };


static uchar
kbd_io(uchar c)
{
	uchar d;
	SspReg *r;

	r = SSPREG;
	r->sscr0 = 0;
	r->sscr0 = (3<<SSPCR0_V_SCR) | (1<<SSPCR0_V_SSE) | (0<<SSPCR0_V_FRF)
                | (7<<SSPCR0_V_DSS);
	while((r->sssr & SSPSR_M_RNE) != 0) {      /* drain recv data */
		d = r->ssdr;
		USED(d);
	}
	while((r->sssr & SSPSR_M_TNF) == 0)       /* check for empty */
		;
	r->ssdr = (c<<8) | c;     /* put char */
	while((r->sssr & SSPSR_M_RNE) == 0)       /* wait to get char back */
                ;
	d = r->ssdr;
	return d;
}

static void
kbdfujitsuintr(Ureg*, void*)
{
	int c;

	for (;;) {
		if(GPIOREG->gplr&(1<<25))
			break;	/* no new scancode */
		c = kbd_io(0);
		timer_devwait(&GPIOREG->gplr, 1<<25, 1<<25, MS2TMR(500));
		fujkbdputc(kbdq, c);
		GPIOREG->gedr = (1<<25);
	}
}

void
kbdinit(void)
{
	GpioReg *g;
	SspReg *r;

	g = GPIOREG;
	g->gpdr = (g->gpdr | (0xd << 10) | (1<<23)) & ~((0x2 << 10) | (1<<25));
	g->gafr = (g->gafr | (0xf << 10)) & ~((1<<23) | (1<<25));      
		/* gp10- 13 alt func */
	// g->gpdr = 0x0d807424;
	// g->gafr = 0x00003c00;
	g->gpsr = (1<<23);          /* assert kbctl wakeup pin */
	PPCREG->ppar |= (1<<18);    /* set alt func for spi interface */

	r = SSPREG;
	r->sscr0 = 0;
	r->sscr1 = 0;	/* no interrupts, no loopback */
	r->sssr = 0;	/* remove any rcv overrun errors */

	/* turn on SSP: */
	r->sscr0 = (3<<SSPCR0_V_SCR) | (1<<SSPCR0_V_SSE)
		| (0<<SSPCR0_V_FRF) | (7<<SSPCR0_V_DSS);

	g->grer &= ~(1<<25);
	intrenable(25, kbdfujitsuintr, 0, BusGPIO);
	g->gfer |= (1<<25);
	if (kbdq == nil) {
		kbdq = qopen(4*1024, 0, 0, 0);
		qnoblock(kbdq, 1);
	}
}

static void
fujkbdputc(Queue *kbdq, int sc)
{
	int keyup, c;

	keyup = sc & 0x80;
	c = sc & 0x7f;

	if (pressed[Meta-Shift])
		c = kbtabmeta[c];
	else if (pressed[Num-Shift])
		c = kbtabnuml[c];
	else if (pressed[LShift-Shift] || pressed[RShift-Shift])
		c = kbtabshift[c];
	else
		c = kbtab[c];

	if(c >= Shift && c <= Shift+NShifts) {
		pressed[c-Shift] = !keyup^(pressed[c-Shift]&toggle[c-Shift]);
		return;
	}
	if(keyup)
		return;
	if(c == (Rune)No){
		print("kbd #%ux\n", sc);
		return;
	}
	if(pressed[Caps-Shift])
		c = toupper(c);
	if(pressed[LCtrl-Shift] || pressed[RCtrl-Shift])
		c &= 0x1f;
	if(pressed[LAlt-Shift] || pressed[RAlt-Shift])
		c = APP|(c&0xff);
	kbdputc(kbdq, c);
}