#include <string.h>

#include "ibm.h"
#include "io.h"
#include "keyboard_at.h"
#include "mem.h"
#include "pci.h"
#include "x86.h"

#include "i430fx.h"

static uint8_t card_i430fx[256];

static void i430fx_map(uint32_t addr, uint32_t size, int state) {
	switch (state & 3) {
	case 0:mem_set_mem_state(addr, size, MEM_READ_EXTERNAL | MEM_WRITE_EXTERNAL);
		break;
	case 1:mem_set_mem_state(addr, size, MEM_READ_INTERNAL | MEM_WRITE_EXTERNAL);
		break;
	case 2:mem_set_mem_state(addr, size, MEM_READ_EXTERNAL | MEM_WRITE_INTERNAL);
		break;
	case 3:mem_set_mem_state(addr, size, MEM_READ_INTERNAL | MEM_WRITE_INTERNAL);
		break;
	}
	flushmmucache_nopc();
}

void i430fx_write(int func, int addr, uint8_t val, void *priv) {
	if (func)
		return;

	if (addr >= 0x10 && addr < 0x4f)
		return;

	switch (addr) {
	case 0x00:
	case 0x01:
	case 0x02:
	case 0x03:
	case 0x08:
	case 0x09:
	case 0x0a:
	case 0x0b:
	case 0x0c:
	case 0x0e:return;

	case 0x04: /*Command register*/
		val &= 0x02;
		val |= 0x04;
		break;
	case 0x05:val = 0;
		break;

	case 0x06: /*Status*/
		val = 0;
		break;
	case 0x07:val = 0x02;
		break;

	case 0x59: /*PAM0*/
		if ((card_i430fx[0x59] ^ val) & 0xf0) {
			i430fx_map(0xf0000, 0x10000, val >> 4);
		}
		pclog("i430fx_write : PAM0 write %02X\n", val);
		break;
	case 0x5a: /*PAM1*/
		if ((card_i430fx[0x5a] ^ val) & 0x0f)
			i430fx_map(0xc0000, 0x04000, val & 0xf);
		if ((card_i430fx[0x5a] ^ val) & 0xf0)
			i430fx_map(0xc4000, 0x04000, val >> 4);
		break;
	case 0x5b: /*PAM2*/
		if ((card_i430fx[0x5b] ^ val) & 0x0f)
			i430fx_map(0xc8000, 0x04000, val & 0xf);
		if ((card_i430fx[0x5b] ^ val) & 0xf0)
			i430fx_map(0xcc000, 0x04000, val >> 4);
		break;
	case 0x5c: /*PAM3*/
		if ((card_i430fx[0x5c] ^ val) & 0x0f)
			i430fx_map(0xd0000, 0x04000, val & 0xf);
		if ((card_i430fx[0x5c] ^ val) & 0xf0)
			i430fx_map(0xd4000, 0x04000, val >> 4);
		break;
	case 0x5d: /*PAM4*/
		if ((card_i430fx[0x5d] ^ val) & 0x0f)
			i430fx_map(0xd8000, 0x04000, val & 0xf);
		if ((card_i430fx[0x5d] ^ val) & 0xf0)
			i430fx_map(0xdc000, 0x04000, val >> 4);
		break;
	case 0x5e: /*PAM5*/
		if ((card_i430fx[0x5e] ^ val) & 0x0f)
			i430fx_map(0xe0000, 0x04000, val & 0xf);
		if ((card_i430fx[0x5e] ^ val) & 0xf0)
			i430fx_map(0xe4000, 0x04000, val >> 4);
		pclog("i430fx_write : PAM5 write %02X\n", val);
		break;
	case 0x5f: /*PAM6*/
		if ((card_i430fx[0x5f] ^ val) & 0x0f)
			i430fx_map(0xe8000, 0x04000, val & 0xf);
		if ((card_i430fx[0x5f] ^ val) & 0xf0)
			i430fx_map(0xec000, 0x04000, val >> 4);
		pclog("i430fx_write : PAM6 write %02X\n", val);
		break;

	case 0x72: /*SMRAM*/
		pclog("Write SMRAM %02x\n", val);
		val = (val & 0x78) | 2; /*SMRAM always at A0000-BFFFF*/
		val |= (card_i430fx[0x72] & 0x10); /*D_LCK can not be cleared by software*/
		if (val & 0x10) /*D_LCK locks D_OPEN and G_SMRAME*/
		{
			val &= ~0x48; /*D_OPEN is forced to 0, G_SMRAME is read only*/
			val |= (card_i430fx[0x72] & 0x08);
		}
		if ((card_i430fx[0x72] ^ val) & 0x40) {
			if (val & 0x40) /*SMRAM enabled*/
			{
				pclog("Enable SMRAM\n");
				mem_set_mem_state(0xa0000, 0x20000, MEM_READ_INTERNAL | MEM_WRITE_INTERNAL);
			} else {
				pclog("Disable SMRAM\n");
				mem_set_mem_state(0xa0000, 0x20000, MEM_READ_EXTERNAL | MEM_WRITE_EXTERNAL);
			}
		}
		break;
	}

	card_i430fx[addr] = val;
}

uint8_t i430fx_read(int func, int addr, void *priv) {
	if (func)
		return 0xff;

	return card_i430fx[addr];
}

static void i430fx_smram_enable(void) {
	if (card_i430fx[0x72] & 8)
		mem_set_mem_state(0xa0000, 0x20000, MEM_READ_INTERNAL | MEM_WRITE_INTERNAL);
}
static void i430fx_smram_disable(void) {
	if (card_i430fx[0x72] & 8)
		mem_set_mem_state(0xa0000, 0x20000, MEM_READ_EXTERNAL | MEM_WRITE_EXTERNAL);
}

void i430fx_init() {
	pci_add_specific(0, i430fx_read, i430fx_write, NULL);

	memset(card_i430fx, 0, 256);
	card_i430fx[0x00] = 0x86;
	card_i430fx[0x01] = 0x80; /*Intel*/
	card_i430fx[0x02] = 0x2d;
	card_i430fx[0x03] = 0x12; /*SB82437FX-66*/
	card_i430fx[0x04] = 0x06;
	card_i430fx[0x05] = 0x00;
	card_i430fx[0x06] = 0x00;
	card_i430fx[0x07] = 0x82;
	card_i430fx[0x08] = 0x00; /*A0 stepping*/
	card_i430fx[0x09] = 0x00;
	card_i430fx[0x0a] = 0x00;
	card_i430fx[0x0b] = 0x06;
	card_i430fx[0x52] = 0x40; /*256kb PLB cache*/
//        card_i430fx[0x53] = 0x14;
//        card_i430fx[0x56] = 0x52; /*DRAM control*/
	card_i430fx[0x57] = 0x01;
	card_i430fx[0x60] = card_i430fx[0x61] = card_i430fx[0x62] = card_i430fx[0x63] = card_i430fx[0x64] = 0x02;
//        card_i430fx[0x67] = 0x11;
//        card_i430fx[0x69] = 0x03;
//        card_i430fx[0x70] = 0x20;
	card_i430fx[0x72] = 0x02;
//        card_i430fx[0x74] = 0x0e;
//        card_i430fx[0x78] = 0x23;

	smram_enable = i430fx_smram_enable;
	smram_disable = i430fx_smram_disable;
}

void i430fx_reset() {
	i430fx_write(0, 0x59, 0xf, NULL); /*Should reset all PCI devices, but just set PAM0 to point to ROM for now*/
}
