//////////////////////////////////////////////////////////////////////////
// Mapper115 CartSaint : Yuu Yuu Hakusho Final                          //
//           JusticePao(?)                                              //
//////////////////////////////////////////////////////////////////////////
using static VirtualNes.Core.CPU;
using static VirtualNes.MMU;
using BYTE = System.Byte;
using INT = System.Int32;


namespace VirtualNes.Core
{
    public class Mapper115 : Mapper
    {
        BYTE[] reg = new byte[8];
        BYTE prg0, prg1, prg2, prg3;
        BYTE prg0L, prg1L;
        BYTE chr0, chr1, chr2, chr3, chr4, chr5, chr6, chr7;

        BYTE irq_enable;
        BYTE irq_counter;
        BYTE irq_latch;

        BYTE ExPrgSwitch;
        BYTE ExChrSwitch;
        public Mapper115(NES parent) : base(parent)
        {
        }

        public override void Reset()

        {
            for (INT i = 0; i < 8; i++)
            {
                reg[i] = 0x00;
            }

            prg0 = prg0L = 0;
            prg1 = prg1L = 1;
            prg2 = (byte)(PROM_8K_SIZE - 2);
            prg3 = (byte)(PROM_8K_SIZE - 1);

            ExPrgSwitch = 0;
            ExChrSwitch = 0;

            SetBank_CPU();

            if (VROM_1K_SIZE != 0)
            {
                chr0 = 0;
                chr1 = 1;
                chr2 = 2;
                chr3 = 3;
                chr4 = 4;
                chr5 = 5;
                chr6 = 6;
                chr7 = 7;
                SetBank_PPU();
            }
            else
            {
                chr0 = chr2 = chr4 = chr5 = chr6 = chr7 = 0;
                chr1 = chr3 = 1;
            }

            irq_enable = 0; // Disable
            irq_counter = 0;
            irq_latch = 0;
        }


        //void Mapper115::WriteLow(WORD addr, BYTE data)
        public override void WriteLow(ushort addr, byte data)
        {
            switch (addr)
            {
                case 0x6000:
                    ExPrgSwitch = data; //data
                    SetBank_CPU();
                    break;
                case 0x6001:
                    ExChrSwitch = (byte)(data & 0x1);
                    SetBank_PPU();
                    break;
            }
            //Mapper::WriteLow(addr, data);
            base.WriteLow(addr, data);
        }


        //void Mapper115::Write(WORD addr, BYTE data)
        public override void Write(ushort addr, byte data)
        {
            switch (addr & 0xE001)
            {
                case 0x8000:
                    reg[0] = data;
                    SetBank_CPU();
                    SetBank_PPU();
                    break;
                case 0x8001:
                    reg[1] = data;
                    switch (reg[0] & 0x07)
                    {
                        case 0x00:
                            chr0 = (byte)(data & 0xFE);
                            chr1 = (byte)(chr0 + 1);
                            SetBank_PPU();
                            break;
                        case 0x01:
                            chr2 = (byte)(data & 0xFE);
                            chr3 = (byte)(chr2 + 1);
                            SetBank_PPU();
                            break;
                        case 0x02:
                            chr4 = data;
                            SetBank_PPU();
                            break;
                        case 0x03:
                            chr5 = data;
                            SetBank_PPU();
                            break;
                        case 0x04:
                            chr6 = data;
                            SetBank_PPU();
                            break;
                        case 0x05:
                            chr7 = data;
                            SetBank_PPU();
                            break;
                        case 0x06:
                            prg0 = prg0L = data;
                            SetBank_CPU();
                            break;
                        case 0x07:
                            prg1 = prg1L = data;
                            SetBank_CPU();
                            break;
                    }
                    break;
                case 0xA000:
                    reg[2] = data;
                    if (!nes.rom.Is4SCREEN())
                    {
                        if ((data & 0x01) != 0) SetVRAM_Mirror(VRAM_HMIRROR);
                        else SetVRAM_Mirror(VRAM_VMIRROR);
                    }
                    break;
                case 0xA001:
                    reg[3] = data;
                    break;
                case 0xC000:
                    reg[4] = data;
                    irq_counter = data;
                    irq_enable = 0xFF;
                    break;
                case 0xC001:
                    reg[5] = data;
                    irq_latch = data;
                    break;
                case 0xE000:
                    reg[6] = data;
                    irq_enable = 0;
                    nes.cpu.ClrIRQ(IRQ_MAPPER);
                    break;
                case 0xE001:
                    reg[7] = data;
                    irq_enable = 0xFF;
                    break;
            }
        }

        //void Mapper115::HSync(INT scanline)
        public override void HSync(int scanline)
        {
            if ((scanline >= 0 && scanline <= 239))
            {
                if (nes.ppu.IsDispON())
                {
                    if (irq_enable != 0)
                    {
                        if ((irq_counter--) == 0)
                        {
                            irq_counter = irq_latch;
                            //					nes.cpu.IRQ_NotPending();
                            nes.cpu.SetIRQ(IRQ_MAPPER);
                        }
                    }
                }
            }
        }


        void SetBank_CPU()
        {
            if ((ExPrgSwitch & 0x80) != 0)
            {
                prg0 = (byte)(((ExPrgSwitch << 1) & 0x1e));
                prg1 = (byte)(prg0 + 1);

                SetPROM_32K_Bank(prg0, prg1, prg0 + 2, prg1 + 2);
            }
            else
            {
                prg0 = prg0L;
                prg1 = prg1L;
                if ((reg[0] & 0x40) != 0)
                {
                    SetPROM_32K_Bank(PROM_8K_SIZE - 2, prg1, prg0, PROM_8K_SIZE - 1);
                }
                else
                {
                    SetPROM_32K_Bank(prg0, prg1, PROM_8K_SIZE - 2, PROM_8K_SIZE - 1);
                }
            }
        }

        void SetBank_PPU()
        {
            if (VROM_1K_SIZE != 0)
            {
                if ((reg[0] & 0x80) != 0)
                {
                    SetVROM_8K_Bank((ExChrSwitch << 8) + chr4, (ExChrSwitch << 8) + chr5,
                             (ExChrSwitch << 8) + chr6, (ExChrSwitch << 8) + chr7,
                             (ExChrSwitch << 8) + chr0, (ExChrSwitch << 8) + chr1,
                             (ExChrSwitch << 8) + chr2, (ExChrSwitch << 8) + chr3);
                }
                else
                {
                    SetVROM_8K_Bank((ExChrSwitch << 8) + chr0, (ExChrSwitch << 8) + chr1,
                             (ExChrSwitch << 8) + chr2, (ExChrSwitch << 8) + chr3,
                              (ExChrSwitch << 8) + chr4, (ExChrSwitch << 8) + chr5,
                              (ExChrSwitch << 8) + chr6, (ExChrSwitch << 8) + chr7);
                }
            }
        }

        //void Mapper115::SaveState(LPBYTE p)
        public override void SaveState(byte[] p)
        {
            for (byte i = 0; i < 8; i++)
            {
                p[i] = reg[i];
            }
            p[8] = prg0;
            p[9] = prg1;
            p[10] = prg2;
            p[11] = prg3;
            p[12] = chr0;
            p[13] = chr1;
            p[14] = chr2;
            p[15] = chr3;
            p[16] = chr4;
            p[17] = chr5;
            p[18] = chr6;
            p[19] = chr7;
            p[20] = irq_enable;
            p[21] = irq_counter;
            p[22] = irq_latch;
            p[23] = ExPrgSwitch;
            p[24] = prg0L;
            p[25] = prg1L;
            p[26] = ExChrSwitch;

        }

        //void Mapper115::LoadState(LPBYTE p)
        public override void LoadState(byte[] p)
        {
            for (byte i = 0; i < 8; i++)
            {
                reg[i] = p[i];
            }
            prg0 = p[8];
            prg1 = p[9];
            prg2 = p[10];
            prg3 = p[11];
            chr0 = p[12];
            chr1 = p[13];
            chr2 = p[14];
            chr3 = p[15];
            chr4 = p[16];
            chr5 = p[17];
            chr6 = p[18];
            chr7 = p[19];
            irq_enable = p[20];
            irq_counter = p[21];
            irq_latch = p[22];
            ExPrgSwitch = p[23];
            prg0L = p[24];
            prg1L = p[25];
            ExChrSwitch = p[26];
        }

        public override bool IsStateSave()
        {
            return true;
        }
    }
}