using System;
using System.IO;
using BizHawk.Emulation.CPUs.H6280;

namespace BizHawk.Emulation.Consoles.TurboGrafx
{
    // ------------------------------------------------------
    // HuC6202 Video Priority Controller
    // ------------------------------------------------------
    // Responsible for merging VDC1 and VDC2 data on the SuperGrafx.
    // Pretty much all documentation on the SuperGrafx courtesy of Charles MacDonald.

    // VPC Frame timing has not been updated with new stuff. I expect Madoo Granzort to be fixed once the timing is updated.
    // However, 1) The new frame timing is not working 100% yet, 2) I haven't decided if I want to continue having separate
    // rendering in the VPC and the VDC.

    public sealed class VPC : IVideoProvider
    {
        public VDC VDC1;
        public VDC VDC2;
        public VCE VCE;
        private HuC6280 CPU;

        public byte[] Registers = {0x11, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00};

        public int Window1Width { get { return ((Registers[3] & 3) << 8) | Registers[2]; } }
        public int Window2Width { get { return ((Registers[5] & 3) << 8) | Registers[4]; } }
        public int PriorityModeSlot0 { get { return Registers[0] & 0x0F; } }
        public int PriorityModeSlot1 { get { return (Registers[0] >> 4) & 0x0F; } }
        public int PriorityModeSlot2 { get { return Registers[1] & 0x0F; } }
        public int PriorityModeSlot3 { get { return (Registers[1] >> 4) & 0x0F; } }

        public VPC(VDC vdc1, VDC vdc2, VCE vce, HuC6280 cpu)
        {
            VDC1 = vdc1;
            VDC2 = vdc2;
            VCE = vce;
            CPU = cpu;

            // latch initial video buffer
            FrameBuffer = vdc1.GetVideoBuffer();
            FrameWidth = vdc1.BufferWidth;
            FrameHeight = vdc1.BufferHeight;
        }

        public byte ReadVPC(int port)
        {
            port &= 0x0F;
            switch (port)
            {
                case 0x08: return Registers[0];
                case 0x09: return Registers[1];
                case 0x0A: return Registers[2];
                case 0x0B: return Registers[3];
                case 0x0C: return Registers[4];
                case 0x0D: return Registers[5];
                case 0x0E: return Registers[6];
                case 0x0F: return 0;
                default:   return 0xFF;
            }
        }

        public void WriteVPC(int port, byte value)
        {
            port &= 0x0F;
            switch (port)
            {
                case 0x08: Registers[0] = value; break;
                case 0x09: Registers[1] = value; break;
                case 0x0A: Registers[2] = value; break;
                case 0x0B: Registers[3] = value; break;
                case 0x0C: Registers[4] = value; break;
                case 0x0D: Registers[5] = value; break;
                case 0x0E: 
                    // CPU Store Immediate VDC Select
                    CPU.WriteVDC = (value & 1) == 0 ? (Action<int,byte>) VDC1.WriteVDC : VDC2.WriteVDC;
                    Registers[6] = value; 
                    break;
            }
        }

        public void SaveStateBinary(BinaryWriter writer)
        {
            writer.Write(Registers);
        }

        public void LoadStateBinary(BinaryReader reader)
        {
            Registers = reader.ReadBytes(7);
        }

        public void SaveStateText(TextWriter writer)
        {
            writer.WriteLine("[VPC]");
            writer.Write("Registers ");
            Registers.SaveAsHex(writer);
            writer.WriteLine("[/VPC]\n");
        }

        public void LoadStateText(TextReader reader)
        {
            while (true)
            {
                string[] args = reader.ReadLine().Split(' ');
                if (args[0].Trim() == "") continue;
                if (args[0] == "[/VPC]") break;
                if (args[0] == "Registers")
                    Registers.ReadFromHex(args[1]);
                else
                    Console.WriteLine("Skipping unrecognized identifier " + args[0]);
            }
        }

        // We use a single priority mode for the whole frame.
        // No commercial SGX games really use the 'window' features AFAIK.
        // And there are no homebrew SGX games I know of.

        private const int BXR = 7;
        private const int BYR = 8;

        private int EffectivePriorityMode = 0;

        private int FrameHeight;
        private int FrameWidth;
        private int[] FrameBuffer;

        private byte[] PriorityBuffer = new byte[512];
        private byte[] InterSpritePriorityBuffer = new byte[512];

        public void ExecFrame()
        {
            // Determine the effective priority mode.
            if (Window1Width < 0x40 && Window2Width < 0x40)
                EffectivePriorityMode = PriorityModeSlot3 >> 2;
            else if (Window2Width > 512)
                EffectivePriorityMode = PriorityModeSlot1 >> 2;
            else
            {
                Console.WriteLine("Unsupported VPC window settings");
                EffectivePriorityMode = 0;
            }

            // Latch frame dimensions and framebuffer, for purely dumb reasons
            FrameWidth = VDC1.BufferWidth;
            FrameHeight = VDC1.BufferHeight;
            FrameBuffer = VDC1.GetVideoBuffer();

            for (int ScanLine = 0; ScanLine < 262; ScanLine++)
            {
                VDC1.ScanLine = ScanLine;
                VDC2.ScanLine = ScanLine;

                if ((ScanLine + 64) == (VDC1.Registers[6] & 0x3FF))
                {
                    if (VDC1.RasterCompareInterruptEnabled)
                    {
                        VDC1.StatusByte |= VDC.StatusRasterCompare;
                        CPU.IRQ1Assert = true;
                    }
                }

                if ((ScanLine + 64) == (VDC2.Registers[6] & 0x3FF))
                {
                    if (VDC2.RasterCompareInterruptEnabled)
                    {
                        VDC2.StatusByte |= VDC.StatusRasterCompare;
                        CPU.IRQ1Assert = true;
                    }
                }

                if (ScanLine == 240 && VDC1.VBlankInterruptEnabled)
                {
                    VDC1.StatusByte |= VDC.StatusVerticalBlanking;
                    CPU.IRQ1Assert = true;
                }

                if (ScanLine == 240 && VDC2.VBlankInterruptEnabled)
                {
                    VDC2.StatusByte |= VDC.StatusVerticalBlanking;
                    CPU.IRQ1Assert = true;
                }

                CPU.Execute(455);

                if (ScanLine < FrameHeight)
                    RenderScanLine();
            }
        }

        private void RenderScanLine()
        {
            if (VDC1.ScanLine == 0)
            {
                VDC1.BackgroundY = VDC1.Registers[BYR];
                VDC2.BackgroundY = VDC2.Registers[BYR];
            }

            InitializeScanLine(VDC1.ScanLine);

            switch (EffectivePriorityMode)
            {
                case 0:
                    RenderBackgroundScanline(VDC1, 12);
                    RenderBackgroundScanline(VDC2, 2);
                    RenderSpritesScanline(VDC1, 11, 14);
                    RenderSpritesScanline(VDC2, 1, 3);
                    break;
                case 1:
                    RenderBackgroundScanline(VDC1, 12);
                    RenderBackgroundScanline(VDC2, 2);
                    RenderSpritesScanline(VDC1, 11, 14);
                    RenderSpritesScanline(VDC2, 1, 13);
                    break;
            }

            if (VDC1.ScanLine == FrameHeight - 1)
            {
                VDC1.UpdateSpriteAttributeTable();
                VDC2.UpdateSpriteAttributeTable();
            }
        }

        private void InitializeScanLine(int scanline)
        {
            // Clear priority buffer
            Array.Clear(PriorityBuffer, 0, FrameWidth);

            // Initialize scanline to background color
            for (int i = 0; i < FrameWidth; i++)
                FrameBuffer[(scanline * FrameWidth) + i] = VCE.Palette[0];
        }

        private void RenderBackgroundScanline(VDC vdc, byte priority)
        {
            if (vdc.BackgroundEnabled == false)
            {
                vdc.BackgroundY++;
                vdc.BackgroundY &= 0x01FF;
                return;
            }

            int vertLine = vdc.BackgroundY;
            vertLine %= vdc.BatHeight * 8;
            int yTile = (vertLine / 8);
            int yOfs = vertLine % 8;
            vdc.BackgroundY++;
            vdc.BackgroundY &= 0x01FF;

            int xScroll = vdc.Registers[BXR] & 0x3FF;
            for (int x = 0; x < FrameWidth; x++)
            {
                if (PriorityBuffer[x] >= priority) continue;
                int xTile = ((x + xScroll) / 8) % vdc.BatWidth;
                int xOfs = (x + xScroll) & 7;
                int tileNo = vdc.VRAM[(ushort)(((yTile * vdc.BatWidth) + xTile))] & 2047;
                int paletteNo = vdc.VRAM[(ushort)(((yTile * vdc.BatWidth) + xTile))] >> 12;
                int paletteBase = paletteNo * 16;

                byte c = vdc.PatternBuffer[(tileNo * 64) + (yOfs * 8) + xOfs];
                if (c != 0)
                {
                    FrameBuffer[(vdc.ScanLine * FrameWidth) + x] = VCE.Palette[paletteBase + c];
                    PriorityBuffer[x] = priority;
                }
            }
        }

        private static byte[] heightTable = { 16, 32, 64, 64 };

        private void RenderSpritesScanline(VDC vdc, byte lowPriority, byte highPriority)
        {
            if (vdc.SpritesEnabled == false)
                return;

            // clear inter-sprite priority buffer
            Array.Clear(InterSpritePriorityBuffer, 0, FrameWidth);

            for (int i = 0; i < 64; i++)
            {
                int y = (vdc.SpriteAttributeTable[(i * 4) + 0] & 1023) - 64;
                int x = (vdc.SpriteAttributeTable[(i * 4) + 1] & 1023) - 32;
                ushort flags = vdc.SpriteAttributeTable[(i * 4) + 3];
                int height = heightTable[(flags >> 12) & 3];

                if (y + height <= vdc.ScanLine || y > vdc.ScanLine)
                    continue;

                int patternNo = (((vdc.SpriteAttributeTable[(i * 4) + 2]) >> 1) & 0x1FF);
                int paletteBase = 256 + ((flags & 15) * 16);
                int width = (flags & 0x100) == 0 ? 16 : 32;
                bool priority = (flags & 0x80) != 0;
                bool hflip = (flags & 0x0800) != 0;
                bool vflip = (flags & 0x8000) != 0;

                if (width == 32)
                    patternNo &= 0x1FE;

                int yofs;
                if (vflip == false)
                {
                    yofs = (vdc.ScanLine - y) & 15;
                    if (height == 32)
                    {
                        patternNo &= 0x1FD;
                        if (vdc.ScanLine - y >= 16)
                        {
                            y += 16;
                            patternNo += 2;
                        }
                    }
                    else if (height == 64)
                    {
                        patternNo &= 0x1F9;
                        if (vdc.ScanLine - y >= 48)
                        {
                            y += 48;
                            patternNo += 6;
                        }
                        else if (vdc.ScanLine - y >= 32)
                        {
                            y += 32;
                            patternNo += 4;
                        }
                        else if (vdc.ScanLine - y >= 16)
                        {
                            y += 16;
                            patternNo += 2;
                        }
                    }
                }
                else // vflip == true
                {
                    yofs = 15 - ((vdc.ScanLine - y) & 15);
                    if (height == 32)
                    {
                        patternNo &= 0x1FD;
                        if (vdc.ScanLine - y < 16)
                        {
                            y += 16;
                            patternNo += 2;
                        }
                    }
                    else if (height == 64)
                    {
                        patternNo &= 0x1F9;
                        if (vdc.ScanLine - y < 16)
                        {
                            y += 48;
                            patternNo += 6;
                        }
                        else if (vdc.ScanLine - y < 32)
                        {
                            y += 32;
                            patternNo += 4;
                        }
                        else if (vdc.ScanLine - y < 48)
                        {
                            y += 16;
                            patternNo += 2;
                        }
                    }
                }
                if (hflip == false)
                {
                    if (x + width > 0 && y + height > 0)
                    {
                        for (int xs = x >= 0 ? x : 0; xs < x + 16 && xs >= 0 && xs < FrameWidth; xs++)
                        {
                            byte pixel = vdc.SpriteBuffer[(patternNo * 256) + (yofs * 16) + (xs - x)];
                            if (pixel != 0 && InterSpritePriorityBuffer[xs] == 0)
                            {
                                InterSpritePriorityBuffer[xs] = 1;
                                byte myPriority = priority ? highPriority : lowPriority;
                                if (PriorityBuffer[xs] < myPriority)
                                {
                                    FrameBuffer[(vdc.ScanLine * FrameWidth) + xs] = VCE.Palette[paletteBase + pixel];
                                    PriorityBuffer[xs] = myPriority;
                                }
                            }
                        }
                    }
                    if (width == 32)
                    {
                        patternNo++;
                        x += 16;
                        for (int xs = x >= 0 ? x : 0; xs < x + 16 && xs >= 0 && xs < FrameWidth; xs++)
                        {
                            byte pixel = vdc.SpriteBuffer[(patternNo * 256) + (yofs * 16) + (xs - x)];
                            if (pixel != 0 && InterSpritePriorityBuffer[xs] == 0)
                            {
                                InterSpritePriorityBuffer[xs] = 1;
                                byte myPriority = priority ? highPriority : lowPriority;
                                if (PriorityBuffer[xs] < myPriority)
                                {
                                    FrameBuffer[(vdc.ScanLine * FrameWidth) + xs] = VCE.Palette[paletteBase + pixel];
                                    PriorityBuffer[xs] = myPriority;
                                }
                            }
                        }
                    }
                }
                else
                { // hflip = true
                    if (x + width > 0 && y + height > 0)
                    {
                        if (width == 32)
                            patternNo++;
                        for (int xs = x >= 0 ? x : 0; xs < x + 16 && xs >= 0 && xs < FrameWidth; xs++)
                        {
                            byte pixel = vdc.SpriteBuffer[(patternNo * 256) + (yofs * 16) + 15 - (xs - x)];
                            if (pixel != 0 && InterSpritePriorityBuffer[xs] == 0)
                            {
                                InterSpritePriorityBuffer[xs] = 1;
                                byte myPriority = priority ? highPriority : lowPriority;
                                if (PriorityBuffer[xs] < myPriority)
                                {
                                    FrameBuffer[(vdc.ScanLine * FrameWidth) + xs] = VCE.Palette[paletteBase + pixel];
                                    PriorityBuffer[xs] = myPriority;
                                }
                            }
                        }
                        if (width == 32)
                        {
                            patternNo--;
                            x += 16;
                            for (int xs = x >= 0 ? x : 0; xs < x + 16 && xs >= 0 && xs < FrameWidth; xs++)
                            {
                                byte pixel = vdc.SpriteBuffer[(patternNo * 256) + (yofs * 16) + 15 - (xs - x)];
                                if (pixel != 0 && InterSpritePriorityBuffer[xs] == 0)
                                {
                                    InterSpritePriorityBuffer[xs] = 1;
                                    byte myPriority = priority ? highPriority : lowPriority;
                                    if (PriorityBuffer[xs] < myPriority)
                                    {
                                        FrameBuffer[(vdc.ScanLine * FrameWidth) + xs] = VCE.Palette[paletteBase + pixel];
                                        PriorityBuffer[xs] = myPriority;
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }

        public int[] GetVideoBuffer()
        {
            return FrameBuffer;
        }

        public int BufferWidth
        {
            get { return FrameWidth; }
        }

        public int BufferHeight
        {
            get { return FrameHeight; }
        }

        public int BackgroundColor
        {
            get { return VCE.Palette[0]; }
        }
    }
}