#include "ibm.h"
#include "io.h"
#include "pic.h"
#include "pit.h"
#include "video.h"

PIC_t pic,pic2;

int intclear;
int keywaiting=0;
int pic_intpending;

void pic_updatepending()
{
	if (AT/* || romset == ROM_XI8088*/)
	{
		if ((pic2.pend&~pic2.mask)&~pic2.mask2)
			pic.pend |= (1 << 2);
		else
			pic.pend &= ~(1 << 2);
		pic_intpending = (pic.pend & ~pic.mask) & ~pic.mask2;
		if (!((pic.mask | pic.mask2) & (1 << 2)))
			pic_intpending |= ((pic2.pend&~pic2.mask)&~pic2.mask2);
	}
	else
		pic_intpending = (pic.pend & ~pic.mask) & ~pic.mask2;

/*        pclog("pic_intpending = %i  %02X %02X %02X %02X\n", pic_intpending, pic.ins, pic.pend, pic.mask, pic.mask2);
	pclog("                    %02X %02X %02X %02X %i %i\n", pic2.ins, pic2.pend, pic2.mask, pic2.mask2, ((pic.mask | pic.mask2) & (1 << 2)), ((pic2.pend&~pic2.mask)&~pic2.mask2));*/
}


void pic_reset()
{
	pic.icw=0;
	pic.mask=0xFF;
	pic.mask2=0;
	pic.pend=pic.ins=0;
	pic.vector=8;
	pic.read=1;
	pic2.icw=0;
	pic2.mask=0xFF;
	pic.mask2=0;
	pic2.pend=pic2.ins=0;
	pic_intpending = 0;
	pic.level_sensitive = 0;
	pic2.level_sensitive = 0;
}

void pic_update_mask(uint8_t *mask, uint8_t ins)
{
	int c;
	*mask = 0;
	for (c = 0; c < 8; c++)
	{
		if (ins & (1 << c))
		{
			*mask = 0xff << c;
			return;
		}
	}
}

static void pic_autoeoi()
{
	int c;

	for (c=0;c<8;c++)
	{
		if (pic.ins&(1<<c))
		{
			pic.ins&=~(1<<c);
			pic_update_mask(&pic.mask2, pic.ins);

			if ((AT/* || romset == ROM_XI8088*/) && c == 2 && (pic2.pend&~pic2.mask)&~pic2.mask2)
				pic.pend |= (1 << 2);

			pic_updatepending();
			return;
		}
	}
}

void pic_write(uint16_t addr, uint8_t val, void *priv)
{
	int c;
//        pclog("Write PIC %04X %02X %04X(%06X):%04X\n",addr,val,CS,cs,pc);
	if (addr&1)
	{
		switch (pic.icw)
		{
			case 0: /*OCW1*/
//                        printf("Write mask %02X %04X:%04X\n",val,CS,pc);
			pic.mask=val;
			pic_updatepending();
			break;
			case 1: /*ICW2*/
			pic.vector=val&0xF8;
//                        printf("PIC vector now %02X\n",pic.vector);
	   //             output=1;
			if (pic.icw1&2) pic.icw=3;
			else            pic.icw=2;
			break;
			case 2: /*ICW3*/
			if (pic.icw1&1) pic.icw=3;
			else            pic.icw=0;
			break;
			case 3: /*ICW4*/
			pic.icw4 = val;
//                        pclog("ICW4 = %02x\n", val);
			pic.icw=0;
			break;
		}
	}
	else
	{
		if (val&16) /*ICW1*/
		{
			pic.mask = 0;
			pic.mask2=0;
			pic.icw=1;
			pic.icw1=val;
			pic.ins = 0;
			pic_updatepending();
		}
		else if (!(val&8)) /*OCW2*/
		{
//                        printf("Clear ints - %02X %02X\n",pic.ins,val);
			if ((val&0xE0)==0x60)
			{
//                                pclog("Specific EOI - %02X %i\n",pic.ins,1<<(val&7));
				pic.ins&=~(1<<(val&7));
				pic_update_mask(&pic.mask2, pic.ins);
				if (val == 2 && (pic2.pend&~pic2.mask)&~pic2.mask2)
					pic.pend |= (1 << 2);
//                                pic.pend&=(1<<(val&7));
//                                if ((val&7)==1) pollkeywaiting();
				pic_updatepending();
			}
			else
			{
				for (c=0;c<8;c++)
				{
					if (pic.ins&(1<<c))
					{
						pic.ins&=~(1<<c);
						pic_update_mask(&pic.mask2, pic.ins);

						if (c == 2 && (pic2.pend&~pic2.mask)&~pic2.mask2)
							pic.pend |= (1 << 2);

						if (c==1 && keywaiting)
						{
							intclear&=~1;
//                                                        pollkeywaiting();
						}
						pic_updatepending();
//                                                pclog("Generic EOI - Cleared int %i\n",c);
						return;
					}
				}
			}
		}
		else               /*OCW3*/
		{
		       // if (val&4) fatal("PIC1 write OCW3 4 %02X\n",val);
			if (val&2) pic.read=(val&1);
			if (val&0x40) { } //fatal("PIC 1 write OCW3 40 %02X\n",val);
		}
	}
}

uint8_t pic_read(uint16_t addr, void *priv)
{
	if (addr&1) { /*pclog("Read PIC mask %02X\n",pic.mask);*/ return pic.mask; }
	if (pic.read) { /*pclog("Read PIC ins %02X\n",pic.ins);*/ return pic.ins | (pic2.ins ? 4 : 0); }
//        pclog("Read PIC pend %02X %08X\n",pic.pend,EDX);
	return pic.pend;
}

void pic_init()
{
	io_sethandler(0x0020, 0x0002, pic_read, NULL, NULL, pic_write, NULL, NULL, NULL);
}

static void pic2_autoeoi()
{
	int c;

	for (c = 0; c < 8; c++)
	{
		if (pic2.ins & (1 << c))
		{
			pic2.ins &= ~(1 << c);
			pic_update_mask(&pic2.mask2, pic2.ins);

			pic_updatepending();
			return;
		}
	}
}

void pic2_write(uint16_t addr, uint8_t val, void *priv)
{
	int c;
//        pclog("Write PIC2 %04X %02X %04X:%04X %i\n",addr,val,CS,pc,ins);
	if (addr&1)
	{
		switch (pic2.icw)
		{
			case 0: /*OCW1*/
//                        printf("PIC2 Write mask %02X %04X:%04X\n",val,CS,pc);
			pic2.mask=val;
			pic_updatepending();
			break;
			case 1: /*ICW2*/
			pic2.vector=val&0xF8;
//                        pclog("PIC2 vector %02X\n", val & 0xf8);
			if (pic2.icw1&2) pic2.icw=3;
			else            pic2.icw=2;
			break;
			case 2: /*ICW3*/
			if (pic2.icw1&1) pic2.icw=3;
			else            pic2.icw=0;
			break;
			case 3: /*ICW4*/
			pic2.icw4 = val;
			pic2.icw=0;
			break;
		}
	}
	else
	{
		if (val&16) /*ICW1*/
		{
			pic2.mask = 0;
			pic2.mask2=0;
			pic2.icw=1;
			pic2.icw1=val;
			pic2.ins = 0;
			pic_updatepending();
		}
		else if (!(val&8)) /*OCW2*/
		{
			if ((val&0xE0)==0x60)
			{
				pic2.ins&=~(1<<(val&7));
				pic_update_mask(&pic2.mask2, pic2.ins);

				pic_updatepending();
			}
			else
			{
				for (c=0;c<8;c++)
				{
					if (pic2.ins&(1<<c))
					{
						pic2.ins &= ~(1<<c);
						pic_update_mask(&pic2.mask2, pic2.ins);

						pic_updatepending();
						return;
					}
				}
			}
		}
		else               /*OCW3*/
		{
			if (val&2) pic2.read=(val&1);
		}
	}
}

uint8_t pic2_read(uint16_t addr, void *priv)
{
	if (addr&1) { /*pclog("Read PIC2 mask %02X %04X:%08X\n",pic2.mask,CS,pc);*/ return pic2.mask; }
	if (pic2.read) { /*pclog("Read PIC2 ins %02X %04X:%08X\n",pic2.ins,CS,pc);*/ return pic2.ins; }
	/*pclog("Read PIC2 pend %02X %04X:%08X\n",pic2.pend,CS,pc);*/
	return pic2.pend;
}

void pic2_init()
{
	io_sethandler(0x00a0, 0x0002, pic2_read, NULL, NULL, pic2_write, NULL, NULL, NULL);
}

static uint8_t pic_elcrx_read(uint16_t addr, void *p)
{
	uint8_t temp = 0xff;

	switch (addr)
	{
		case 0x4d0:
		temp = pic.level_sensitive;
		break;
		case 0x4d1:
		temp = pic2.level_sensitive;
		break;
	}

	return temp;
}

static void pic_elcrx_write(uint16_t addr, uint8_t val, void *p)
{
//        pclog("pic_elcrx_write: addr=%04x val=%02x\n", addr, val);
	switch (addr)
	{
		case 0x4d0:
		pic.level_sensitive = val & 0xf8;
		break;
		case 0x4d1:
		pic2.level_sensitive = val & 0xde;
		break;
	}
}

void pic_init_elcrx()
{
	io_sethandler(0x04d0, 0x0002, pic_elcrx_read, NULL, NULL, pic_elcrx_write, NULL, NULL, NULL);
}

void clearpic()
{
	pic.pend=pic.ins=0;
	pic_updatepending();
//        pclog("Clear PIC\n");
}

int pic_current[16];

void picint(uint16_t num)
{
	if ((AT/* || romset == ROM_XI8088*/) && num == (1 << 2))
		num = 1 << 9;
//        pclog("picint : %04X\n", num);
	if ((AT/* || romset == ROM_XI8088*/) && num>0xFF)
	{
		pic2.pend|=(num>>8);
		if ((pic2.pend&~pic2.mask)&~pic2.mask2)
			pic.pend |= (1 << 2);
	}
	else if (num <= 0xff)
	{
		pic.pend|=num;
	}
//        pclog("picint : PEND now %02X %02X\n", pic.pend, pic2.pend);
	pic_updatepending();
}

void picintlevel(uint16_t num)
{
	int c = 0;
	while (!(num & (1 << c))) c++;
	if ((AT/* || romset == ROM_XI8088*/) && num == (1 << 2))
	{
		c = 9;
		num = 1 << 9;
	}
//        pclog("INTLEVEL %04X %i\n", num, c);
	if (!pic_current[c])
	{
		pic_current[c]=1;
		if (num>0xFF)
		{
			pic2.pend|=(num>>8);
		}
		else
		{
			pic.pend|=num;
		}
	}
	pic_updatepending();
}
void picintc(uint16_t num)
{
	int c = 0;
	if (!num)
		return;
	while (!(num & (1 << c))) c++;
	if ((AT/* || romset == ROM_XI8088*/) && num == (1 << 2))
	{
		c = 9;
		num = 1 << 9;
	}
//        pclog("INTC %04X %i\n", num, c);
	pic_current[c]=0;

	if ((AT/* || romset == ROM_XI8088*/) && num > 0xff)
	{
		pic2.pend &= ~(num >> 8);
		if (!((pic2.pend&~pic2.mask)&~pic2.mask2))
			pic.pend &= ~(1 << 2);
	}
	else if(num <= 0xff)
	{
		pic.pend&=~num;
	}
	pic_updatepending();
}

uint8_t picinterrupt()
{
	uint8_t temp=pic.pend&~pic.mask;
	int c;
	for (c = 0; c < 8; c++)
	{
		if ((AT/* || romset == ROM_XI8088*/) && (temp & (1 << 2)))
		{
			uint8_t temp2 = pic2.pend & ~pic2.mask;
			for (c = 0; c < 8; c++)
			{
				if (temp2 & (1 << c))
				{
					if (!(pic2.level_sensitive & (1 << c)))
						pic2.pend &= ~(1 << c);
					pic2.ins |= (1 << c);
					pic_update_mask(&pic2.mask2, pic2.ins);

					if (!(pic2.level_sensitive & (1 << c)))
						pic.pend &= ~(1 << c);
					pic.ins |= (1 << 2); /*Cascade IRQ*/
					pic_update_mask(&pic.mask2, pic.ins);

					pic_updatepending();

					if (pic2.icw4 & 0x02)
						pic2_autoeoi();

					return c+pic2.vector;
				}
			}
		}
		else if (temp & (1 << c))
		{
			if (!(pic.level_sensitive & (1 << c)))
				pic.pend &= ~(1 << c);
			pic.ins |= (1 << c);
			pic_update_mask(&pic.mask2, pic.ins);
			pic_updatepending();

			if (pic.icw4 & 0x02)
				pic_autoeoi();
			return c+pic.vector;
		}
	}
	return 0xFF;
}

void dumppic()
{
	pclog("PIC1 : MASK %02X PEND %02X INS %02X VECTOR %02X\n",pic.mask,pic.pend,pic.ins,pic.vector);
	pclog("PIC2 : MASK %02X PEND %02X INS %02X VECTOR %02X\n",pic2.mask,pic2.pend,pic2.ins,pic2.vector);
}

