#include <u.h>
#include <libc.h>
#include <thread.h>
#include "../eui.h"
#include "dat.h"
#include "fns.h"

u16int ram[32768], vram[32768];
u16int cram[64], vsram[40];
u32int cramc[64];
u8int zram[8192];
u8int reg[32];
u8int ctl[15];

u8int dma;
u8int vdplatch;
u16int vdpaddr, vdpdata;

u8int yma1, yma2;

u8int z80bus = RESET;
u16int z80bank;

//#define vramdebug(a, s, a1, a2, a3) if((a & ~1) == 0xe7a0) print(s, a1, a2, a3);
#define vramdebug(a, s, a1, a2, a3)

u8int
regread(u16int a)
{
	u16int v;

	switch(a | 1){
	case 0x0001: return 0xa0;
	case 0x0003:
		v = ~(keys & 0xffff);
		if((ctl[0] & 0x40) == 0)
			v >>= 8;
		return ctl[0] & 0xc0 | v & 0x3f;
	case 0x0005:
	case 0x0007:
		return ctl[a-3>>1] & 0xc0 | 0x3f;
	case 0x0009: case 0x000b: case 0x000d:
		return ctl[a-3>>1];
	case 0x1101:
		return (~z80bus & BUSACK) >> 1;
	}
	sysfatal("read from 0xa1%.4ux (pc=%#.6ux)", a, curpc);
	return 0;
}

void
regwrite(u16int a, u16int v)
{
	switch(a | 1){
	case 0x0003: case 0x0005: case 0x0007:
	case 0x0009: case 0x000b: case 0x000d:
		ctl[a-3>>1] = v;
		return;
	case 0x1101:
		z80bus = z80bus & ~BUSREQ | v & BUSREQ;
		return;
	case 0x1201:
		if((v & 1) == 0){
			z80bus |= RESET;
			z80bus &= ~BUSACK;
		}else
			z80bus &= ~RESET;
		return;
	case 0x30f1:
		if((v & 1) != 0)
			sramctl |= SRAMEN;
		else
			sramctl &= ~SRAMEN;
		return;
	case 0x30f3: case 0x30f5: case 0x30f7: case 0x30f9: case 0x30fb:
		return;
	}
	sysfatal("write to 0xa1%.4x (pc=%#.6ux)", a, curpc);
}

void
vdpwrite(u16int v)
{
	u8int a;

	if((vdplatch & 0x80) == 0){
		if((v & 0xc000) == 0x8000){
			a = v >> 8 & 0x1f;
			reg[a] = v & 0xff;
			if(a == 0x0c)
				vdpmode();
			vdplatch = 0;
			return;
		}
		vdplatch = vdplatch & 0xfc | v >> 14 | 0x80;
		vdpaddr = vdpaddr & 0xc000 | v & 0x3fff;
	}else{
		vdplatch = vdplatch & 0x03 | v >> 2 & 0x1c;
		vdpaddr = vdpaddr & 0x3fff | v << 14 & 0xc000;
		if((v & 0x80) != 0 && (reg[MODE2] & DMAEN) != 0){
			dma = reg[23] >> 6 & 3;
			if(dma == 0)
				dma++;
		}
	}
}

void
cramwrite(u16int a, u16int v)
{
	u32int w;

	cram[a/2] = v;
	w = v << 12 & 0xe00000 | v << 8 & 0xe000 | v << 4 & 0xe0;
	cramc[a/2] = w;
}

u16int
memread(u32int a)
{
	u16int v;

	switch(a >> 21 & 7){
	case 0: case 1:
		if((sramctl & SRAMEN) != 0 && a >= sram0 && a <= sram1)
			switch(sramctl & ADDRMASK){
			case ADDREVEN: return sram[(a - sram0) >> 1] << 8;
			case ADDRODD: return sram[(a - sram0) >> 1];
			case ADDRBOTH: return sram[a - sram0] << 8 | sram[a - sram0 + 1];
			}
		return prg[(a % nprg) / 2];
	case 5:
		switch(a >> 16 & 0xff){
		case 0xa0:
			if((z80bus & BUSACK) != 0)
				v = z80read(a & 0x7fff);
			else
				v = 0;
			return v << 8 | v;
		case 0xa1:
			v = regread(a);
			return v << 8 | v;
		}
		goto invalid;
	case 6:
		if((a & 0xe700e0) != 0xc00000)
			goto invalid;
		switch(a & 30){
		case 0: case 2:
			vdplatch &= 0x7f;
			switch(vdplatch & 0xf){
			case 0:
				v = vram[vdpaddr/2];
				vdpaddr += reg[AUTOINC];
				break;
			case 4:
				v = vdpaddr & 0x7f;
				if(v < 80)
					v = vsram[v / 2];
				else
					v = 0;
				vdpaddr = (vdpaddr + reg[AUTOINC]) & 0x7f;
				break;
			case 8:
				v = cram[(vdpaddr & 0x7f) / 2];
				vdpaddr = (vdpaddr + reg[AUTOINC]) & 0x7f;
				break;
			default: v = 0;
			}
			return v;
		case 4: case 6:
			vdplatch &= 0x7f;
			v = vdpstat;
			if(dma != 0 && dma != 2)
				v |= STATDMA;
			if(vdpx >= 0xe4 || vdpx < 0x08)
				v |= STATHBL;
			return v;
		case 8: case 10: case 12: case 14:
			if((reg[MODE4] & WIDE) != 0)
				v = vdpx - (vdpx >= 360 ? 406 : 0);
			else
				v = vdpx - (vdpx >= 296 ? 342 : 0);
			if(intla)
				return vdpy - (vdpy >= 234 ? 5 : 0) << 8 & 0xfe00 | frame << 8 | v >> 1 & 0xff;
			return vdpy - (vdpy >= 234 ? 5 : 0) << 8 | v >> 1 & 0xff;
		default:
			goto invalid;
		}
	case 7: return ram[((u16int)a) / 2];
	default:
	invalid:
		sysfatal("read from %#.6ux (pc=%#.6ux)", a, curpc);
		return 0;
	}
}

void
memwrite(u32int a, u16int v, u16int m)
{
	u16int *p;
	u16int w;

	if(0 && (a & 0xe0fffe) == 0xe0df46)
		print("%x %x %x\n", curpc, v, m);
	switch((a >> 21) & 7){
	case 0: case 1:
		if((sramctl & SRAMEN) != 0 && a >= sram0 && a <= sram1){
			switch(sramctl & ADDRMASK){
			case ADDREVEN: sram[(a - sram0) >> 1] = v >> 8; break;
			case ADDRODD: sram[(a - sram0) >> 1] = v; break;
			case ADDRBOTH:
				if((m & 0xff00) == 0xff00)
					sram[a - sram0] = v >> 8;
				if((m & 0xff) == 0xff)
					sram[a + 1 - sram0] = v;
				break;
			}
			if(saveclock == 0)
				saveclock = SAVEFREQ;
			return;
		}
		goto invalid;
	case 5:
		switch(a >> 16 & 0xff){
		case 0xa0:
			if((z80bus & BUSACK) != 0)
				z80write(a & 0xffff, v >> 8);
			return;
		case 0xa1:
			regwrite(a, v >> 8);
			return;
		default:
			goto invalid;
		}
	case 6:
		if((a & 0xe700e0) != 0xc00000)
			goto invalid;
		switch(a & 30){
		case 0: case 2:
			if(dma == 2){
				dma = 4;
				vdpdata = v >> 8;
				vramdebug(vdpaddr, "vdp fill write val %x (pc = %x) %d\n", v & 0xff, curpc, 0);
				p = &vram[vdpaddr / 2];
				if((vdpaddr & 1) == 0)
					*p = *p & 0xff | v << 8;
				else
					*p = *p & 0xff00 | v & 0xff;
				return;
			}
			vdplatch &= 0x7f;
			switch(vdplatch & 0xf){
			case 1:
				if((vdpaddr & 1) != 0)
					v = v << 8 | v >> 8;
				p = &vram[vdpaddr / 2];
				vramdebug(vdpaddr, "vdp write val %x mask %x (pc = %x)\n", v, m, curpc);
				*p = *p & ~m | v & m;
				vdpaddr += reg[AUTOINC];
				return;
			case 3:
				cramwrite(vdpaddr & 0x7f, v);
				vdpaddr = (vdpaddr + reg[AUTOINC]) & 0x7f;
				return;
			case 5:
				w = vdpaddr & 0x7f;
				if(w < 80)
					vsram[w / 2] = v;
				vdpaddr = (vdpaddr + reg[AUTOINC]) & 0x7f;
				return;
			default:
				return;
			}
		case 4: case 6:
			vdpwrite(v);
			return;
		case 16: case 18: case 20: case 22:
			return;
		default:
			goto invalid;
		}
	case 7:
		p = &ram[((u16int)a) / 2];
		*p = *p & ~m | v & m;
		break;
	default:
	invalid:
		sysfatal("write to %#.6x (pc=%#.6x)", a, curpc);
	}
}

void
dmastep(void)
{
	u16int v, *p;
	u32int a;
	
	switch(dma){
	case 1:
		a = reg[DMASRC0] << 1 | reg[DMASRC1] << 9 | reg[DMASRC2] << 17;
		v = memread(a);
		if(++reg[DMASRC0] == 0)
			reg[DMASRC1]++;
		switch(vdplatch & 0x7){
		case 1:
			if((vdpaddr & 1) != 0)
				v = v >> 8 | v << 8;
			vramdebug(vdpaddr, "dma from 68K %x val %x (%d)\n", a, v, 0);
			vram[vdpaddr / 2] = v;
			break;
		case 3:
			if(vdpaddr > 0x7f)
				dma = 0;
			else
				cramwrite(vdpaddr, v);
			break;
		case 5:
			if(vdpaddr < 80)
				vsram[vdpaddr / 2] = v;
			break;
		}
		break;
	case 2:
		return;
	case 3:
		a = reg[DMASRC0] | reg[DMASRC1] << 8;
		v = vram[a / 2];
		if((a & 1) == 0)
			v = v >> 8;
		if(++reg[DMASRC0] == 0)
			reg[DMASRC1]++;
		vramdebug(vdpaddr, "dma copy from %x val %x (%d)\n", a, v, 0);
		p = &vram[vdpaddr / 2];
		if((vdpaddr & 1) != 0)
			*p = *p & 0xff00 | v & 0xff;
		else
			*p = *p & 0xff | v << 8;
		break;
	case 4:
		p = &vram[vdpaddr / 2];
		vramdebug(vdpaddr, "dma fill val %x (%d%d)\n", vdpdata, 0, 0);
		if((vdpaddr & 1) == 0)
			*p = *p & 0xff00 | vdpdata;
		else
			*p = *p & 0xff | vdpdata << 8;
		break;
	}
	vdpaddr += reg[AUTOINC];
	if(reg[DMACL]-- == 0)
		reg[DMACH]--;
	if((reg[DMACL] | reg[DMACH]) == 0)
		dma = 0;
}

u8int
z80read(u16int a)
{
	u16int v;

	switch(a >> 13){
	case 0:
	case 1:
		return zram[a & 0x1fff];
	case 2:
		return ymstat;
	case 3:
		if(a >= 0x7f00){
			v = memread(0xc00000 | a & 0x7e);
			if((a & 1) == 0)
				v >>= 8;
			return v;
		}
		sysfatal("z80 read from %#.4x (pc=%#.4x)", a, scurpc);
	default:
		v = memread(z80bank << 15 | a & 0x7ffe);
		if((a & 1) == 0)
			v >>= 8;
		return v;
	}
}

void
z80write(u16int a, u8int v)
{
	switch(a >> 13){
	case 0:
	case 1:
		zram[a & 0x1fff] = v;
		return;
	case 2:
		switch(a & 3){
		case 0: yma1 = v; return;
		case 1: ymwrite(yma1, v, 0); return;
		case 2: yma2 = v; return;
		case 3: ymwrite(yma2, v, 3); return;
		}
	case 3:
		if(a < 0x6100){
			z80bank = z80bank >> 1 | v << 8 & 0x100;
			return;
		}
		if(a >= 0x7f00){
			memwrite(0xc00000 | a & 0x7e, v | v << 8, (a & 1) != 0 ? 0xff : 0xff00);
			return;
		}
		sysfatal("z80 write to %#.4x (pc=%#.4x)", a, scurpc);
	default:
		memwrite(z80bank << 15 | a & 0x7ffe, v << 8 | v, (a & 1) != 0 ? 0xff : 0xff00);
	}
}

u8int
z80in(u8int)
{
	return 0xff;
}

void
z80out(u8int, u8int)
{
}

u32int irql[8] = {[6] INTVBL, [4] INTHOR};

int
intack(int l)
{
	switch(l){
	case 4:
		irq &= ~INTHOR;
		break;
	case 6:
		irq &= ~INTVBL;
		break;
	}
	return 24 + l;
}