#include "../copyright"

#include "snes9x.h"
#include "memmap.h"
#include "cpuops.h"
#include "ppu.h"
#include "cpuexec.h"
#include "gfx.h"
#include "apu.h"
#include "dma.h"
#include "fxemu.h"
#include "sa1.h"
#include "spc7110.h"
#include "arm_dynarec/dynaexec.h"

void S9xMainLoop_SA1_SFX(void);
void S9xMainLoop_SA1_NoSFX(void);
void S9xMainLoop_NoSA1_SFX(void);
void S9xMainLoop_NoSA1_NoSFX(void);

/*
 * This is a CATSFC modification inspired by a Snes9x-Euphoria modification.
 * The emulator selects a main loop based on the chips used in an entire
 * frame. This avoids the constant SA1.Executing and Settings.SuperFX checks.
 *
 * The original version of S9xMainLoop is S9xMainLoop_SA1_SFX below. Remember
 * to propagate modifications to the SA1_NoSFX, NoSA1_SFX and NoSA1_NoSFX
 * versions.
 */
void S9xMainLoop()
{
#ifdef USE_DYNAREC
   if (Settings.EnableDynarec)
   {
      S9xMainLoop_Dyna();
   }
   else
#endif
   if (Settings.SA1)
   {
      if (Settings.SuperFX)
         S9xMainLoop_SA1_SFX();
      else
         S9xMainLoop_SA1_NoSFX();
   }
   else
   {
      if (Settings.SuperFX)
         S9xMainLoop_NoSA1_SFX();
      else
         S9xMainLoop_NoSA1_NoSFX();
   }
}

void S9xMainLoop_SA1_SFX()
{
#ifdef LAGFIX
   do
   {
#endif
      do
      {
         APU_EXECUTE();
         if (CPU.Flags)
         {
            if (CPU.Flags & NMI_FLAG)
            {
               if (--CPU.NMICycleCount == 0)
               {
                  CPU.Flags &= ~NMI_FLAG;
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  S9xOpcode_NMI();
               }
            }

            if (CPU.Flags & IRQ_PENDING_FLAG)
            {
               if (CPU.IRQCycleCount == 0)
               {
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  if (CPU.IRQActive && !Settings.DisableIRQ)
                  {
                     if (!CheckFlag(IRQ))
                        S9xOpcode_IRQ();
                  }
                  else
                     CPU.Flags &= ~IRQ_PENDING_FLAG;
               }
               else if (--CPU.IRQCycleCount == 0 && CheckFlag(IRQ))
                  CPU.IRQCycleCount = 1;
            }
            if (CPU.Flags & SCAN_KEYS_FLAG)
               break;
         }

         CPU.PCAtOpcodeStart = CPU.PC;
         CPU.Cycles += CPU.MemSpeed;
         (*ICPU.S9xOpcodes [*CPU.PC++].S9xOpcode)();

         if (SA1.Executing)
            S9xSA1MainLoop();
         DO_HBLANK_CHECK_SFX();

#ifdef LAGFIX
         if(finishedFrame)
            break;
#endif
      } while(true);

      ICPU.Registers.PC = CPU.PC - CPU.PCBase;
#ifndef USE_BLARGG_APU
      IAPU.Registers.PC = IAPU.PC - IAPU.RAM;
#endif

#ifdef LAGFIX
      if(!finishedFrame)
      {
#endif
         S9xPackStatus();
#ifndef USE_BLARGG_APU
         S9xAPUPackStatus();
#endif
         CPU.Flags &= ~SCAN_KEYS_FLAG;
#ifdef LAGFIX
      }
      else
      {
         finishedFrame = false;
         break;
      }
   } while(!finishedFrame);
#endif
}

void S9xMainLoop_SA1_NoSFX()
{
#ifdef LAGFIX
   do
   {
#endif
      do
      {
         APU_EXECUTE();
         if (CPU.Flags)
         {
            if (CPU.Flags & NMI_FLAG)
            {
               if (--CPU.NMICycleCount == 0)
               {
                  CPU.Flags &= ~NMI_FLAG;
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  S9xOpcode_NMI();
               }
            }

            if (CPU.Flags & IRQ_PENDING_FLAG)
            {
               if (CPU.IRQCycleCount == 0)
               {
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  if (CPU.IRQActive && !Settings.DisableIRQ)
                  {
                     if (!CheckFlag(IRQ))
                        S9xOpcode_IRQ();
                  }
                  else
                     CPU.Flags &= ~IRQ_PENDING_FLAG;
               }
               else if (--CPU.IRQCycleCount == 0 && CheckFlag(IRQ))
                  CPU.IRQCycleCount = 1;
            }
            if (CPU.Flags & SCAN_KEYS_FLAG)
               break;
         }

         CPU.PCAtOpcodeStart = CPU.PC;
         CPU.Cycles += CPU.MemSpeed;
         (*ICPU.S9xOpcodes [*CPU.PC++].S9xOpcode)();

         if (SA1.Executing)
            S9xSA1MainLoop();
         DO_HBLANK_CHECK_NoSFX();

#ifdef LAGFIX
         if(finishedFrame)
            break;
#endif
      } while(true);

      ICPU.Registers.PC = CPU.PC - CPU.PCBase;
#ifndef USE_BLARGG_APU
      IAPU.Registers.PC = IAPU.PC - IAPU.RAM;
#endif

#ifdef LAGFIX
      if(!finishedFrame)
      {
#endif
         S9xPackStatus();
#ifndef USE_BLARGG_APU
         S9xAPUPackStatus();
#endif
         CPU.Flags &= ~SCAN_KEYS_FLAG;
#ifdef LAGFIX
      }
      else
      {
         finishedFrame = false;
         break;
      }
   } while(!finishedFrame);
#endif
}

void S9xMainLoop_NoSA1_SFX()
{
#ifdef LAGFIX
   do
   {
#endif
      do
      {
         APU_EXECUTE();
         if (CPU.Flags)
         {
            if (CPU.Flags & NMI_FLAG)
            {
               if (--CPU.NMICycleCount == 0)
               {
                  CPU.Flags &= ~NMI_FLAG;
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  S9xOpcode_NMI();
               }
            }

            if (CPU.Flags & IRQ_PENDING_FLAG)
            {
               if (CPU.IRQCycleCount == 0)
               {
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  if (CPU.IRQActive && !Settings.DisableIRQ)
                  {
                     if (!CheckFlag(IRQ))
                        S9xOpcode_IRQ();
                  }
                  else
                     CPU.Flags &= ~IRQ_PENDING_FLAG;
               }
               else if (--CPU.IRQCycleCount == 0 && CheckFlag(IRQ))
                  CPU.IRQCycleCount = 1;
            }
            if (CPU.Flags & SCAN_KEYS_FLAG)
               break;
         }

         CPU.PCAtOpcodeStart = CPU.PC;
         CPU.Cycles += CPU.MemSpeed;
         (*ICPU.S9xOpcodes [*CPU.PC++].S9xOpcode)();
         DO_HBLANK_CHECK_SFX();

#ifdef LAGFIX
         if(finishedFrame)
            break;
#endif
      } while(true);

      ICPU.Registers.PC = CPU.PC - CPU.PCBase;
#ifndef USE_BLARGG_APU
      IAPU.Registers.PC = IAPU.PC - IAPU.RAM;
#endif

#ifdef LAGFIX
      if(!finishedFrame)
      {
#endif
         S9xPackStatus();
#ifndef USE_BLARGG_APU
         S9xAPUPackStatus();
#endif
         CPU.Flags &= ~SCAN_KEYS_FLAG;
#ifdef LAGFIX
      }
      else
      {
         finishedFrame = false;
         break;
      }
   } while(!finishedFrame);
#endif
}

void S9xMainLoop_NoSA1_NoSFX()
{
#ifdef LAGFIX
   do
   {
#endif
      do
      {
         APU_EXECUTE();
         if (CPU.Flags)
         {
            if (CPU.Flags & NMI_FLAG)
            {
               if (--CPU.NMICycleCount == 0)
               {
                  CPU.Flags &= ~NMI_FLAG;
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  S9xOpcode_NMI();
               }
            }

            if (CPU.Flags & IRQ_PENDING_FLAG)
            {
               if (CPU.IRQCycleCount == 0)
               {
                  if (CPU.WaitingForInterrupt)
                  {
                     CPU.WaitingForInterrupt = false;
                     CPU.PC++;
                  }
                  if (CPU.IRQActive && !Settings.DisableIRQ)
                  {
                     if (!CheckFlag(IRQ))
                        S9xOpcode_IRQ();
                  }
                  else
                     CPU.Flags &= ~IRQ_PENDING_FLAG;
               }
               else if (--CPU.IRQCycleCount == 0 && CheckFlag(IRQ))
                  CPU.IRQCycleCount = 1;
            }
            if (CPU.Flags & SCAN_KEYS_FLAG)
               break;
         }

         CPU.PCAtOpcodeStart = CPU.PC;
         CPU.Cycles += CPU.MemSpeed;
         (*ICPU.S9xOpcodes [*CPU.PC++].S9xOpcode)();
         DO_HBLANK_CHECK_NoSFX();

#ifdef LAGFIX
         if(finishedFrame)
            break;
#endif
      } while(true);

      ICPU.Registers.PC = CPU.PC - CPU.PCBase;
#ifndef USE_BLARGG_APU
      IAPU.Registers.PC = IAPU.PC - IAPU.RAM;
#endif

#ifdef LAGFIX
      if(!finishedFrame)
      {
#endif
         S9xPackStatus();
#ifndef USE_BLARGG_APU
         S9xAPUPackStatus();
#endif
         CPU.Flags &= ~SCAN_KEYS_FLAG;
#ifdef LAGFIX
      }
      else
      {
         finishedFrame = false;
         break;
      }
   } while(!finishedFrame);
#endif
}

void S9xSetIRQ(uint32_t source)
{
   CPU.IRQActive |= source;
   CPU.Flags |= IRQ_PENDING_FLAG;
   CPU.IRQCycleCount = 3;
   if (CPU.WaitingForInterrupt)
   {
      /* Force IRQ to trigger immediately after WAI -
       * Final Fantasy Mystic Quest crashes without this. */
      CPU.IRQCycleCount = 0;
      CPU.WaitingForInterrupt = false;
      CPU.PC++;
   }
}

void S9xClearIRQ(uint32_t source)
{
   CLEAR_IRQ_SOURCE(source);
}

/*
 * This is a CATSFC modification inspired by a Snes9x-Euphoria modification.
 * The emulator selects an HBlank processor based on the chips used in an
 * entire frame. This avoids the constant Settings.SuperFX checks.
 *
 * The original version of S9xDoHBlankProcessing is S9xDoHBlankProcessing_SFX
 * below. Remember to propagate modifications to the NoSFX version.
 */
void S9xDoHBlankProcessing_SFX()
{
   CPU.WaitCounter++;
   switch (CPU.WhichEvent)
   {
   case HBLANK_START_EVENT:
      if (IPPU.HDMA && CPU.V_Counter <= PPU.ScreenHeight)
         IPPU.HDMA = S9xDoHDMA(IPPU.HDMA);
      break;
   case HBLANK_END_EVENT:
      S9xSuperFXExec();
#ifndef USE_BLARGG_APU
      CPU.Cycles -= Settings.H_Max;
      if (IAPU.APUExecuting)
         APU.Cycles -= Settings.H_Max;
      else
         APU.Cycles = 0;
#else
      S9xAPUExecute();
      CPU.Cycles -= Settings.H_Max;
      S9xAPUSetReferenceTime(CPU.Cycles);
#endif
      CPU.NextEvent = -1;

      if (++CPU.V_Counter >= (Settings.PAL ? SNES_MAX_PAL_VCOUNTER : SNES_MAX_NTSC_VCOUNTER))
      {
         CPU.V_Counter = 0;
         Memory.FillRAM[0x213F] ^= 0x80;
         PPU.RangeTimeOver = 0;
         CPU.NMIActive = false;
         ICPU.Frame++;
         CPU.Flags |= SCAN_KEYS_FLAG;
         S9xStartHDMA();
      }

      if (PPU.VTimerEnabled && !PPU.HTimerEnabled && CPU.V_Counter == PPU.IRQVBeamPos)
         S9xSetIRQ(PPU_V_BEAM_IRQ_SOURCE);

      if (CPU.V_Counter == PPU.ScreenHeight + FIRST_VISIBLE_LINE)
      {
         /* Start of V-blank */
         S9xEndScreenRefresh();
         IPPU.HDMA = 0;
         /* Bits 7 and 6 of $4212 are computed when read in S9xGetPPU. */
         PPU.ForcedBlanking = (Memory.FillRAM [0x2100] >> 7) & 1;

         if (!PPU.ForcedBlanking)
         {
            uint8_t tmp = 0;
            PPU.OAMAddr = PPU.SavedOAMAddr;

            if (PPU.OAMPriorityRotation)
               tmp = (PPU.OAMAddr & 0xFE) >> 1;
            if ((PPU.OAMFlip & 1) || PPU.FirstSprite != tmp)
            {
               PPU.FirstSprite = tmp;
               IPPU.OBJChanged = true;
            }

            PPU.OAMFlip = 0;
         }

         Memory.FillRAM[0x4210] = 0x80 | Model->_5A22;
         if (Memory.FillRAM[0x4200] & 0x80)
         {
            CPU.NMIActive = true;
            CPU.Flags |= NMI_FLAG;
            CPU.NMICycleCount = CPU.NMITriggerPoint;
         }
      }

      if (CPU.V_Counter == PPU.ScreenHeight + 3)
         S9xUpdateJoypads();

      if (CPU.V_Counter == FIRST_VISIBLE_LINE)
      {
         Memory.FillRAM[0x4210] = Model->_5A22;
         CPU.Flags &= ~NMI_FLAG;
         S9xStartScreenRefresh();
      }
      if (CPU.V_Counter >= FIRST_VISIBLE_LINE && CPU.V_Counter < PPU.ScreenHeight + FIRST_VISIBLE_LINE)
         RenderLine(CPU.V_Counter - FIRST_VISIBLE_LINE);
#ifndef USE_BLARGG_APU
      if (APU.TimerEnabled [2])
      {
         APU.Timer [2] += 4;
         while (APU.Timer [2] >= APU.TimerTarget [2])
         {
            IAPU.RAM [0xff] = (IAPU.RAM [0xff] + 1) & 0xf;
            APU.Timer [2] -= APU.TimerTarget [2];
            IAPU.WaitCounter++;
            IAPU.APUExecuting = true;
         }
      }
      if (CPU.V_Counter & 1)
      {
         if (APU.TimerEnabled [0])
         {
            APU.Timer [0]++;
            if (APU.Timer [0] >= APU.TimerTarget [0])
            {
               IAPU.RAM [0xfd] = (IAPU.RAM [0xfd] + 1) & 0xf;
               APU.Timer [0] = 0;
               IAPU.WaitCounter++;
               IAPU.APUExecuting = true;
            }
         }
         if (APU.TimerEnabled [1])
         {
            APU.Timer [1]++;
            if (APU.Timer [1] >= APU.TimerTarget [1])
            {
               IAPU.RAM [0xfe] = (IAPU.RAM [0xfe] + 1) & 0xf;
               APU.Timer [1] = 0;
               IAPU.WaitCounter++;
               IAPU.APUExecuting = true;
            }
         }
      }
#endif
      break;
   case HTIMER_BEFORE_EVENT:
   case HTIMER_AFTER_EVENT:
      if (PPU.HTimerEnabled && (!PPU.VTimerEnabled || CPU.V_Counter == PPU.IRQVBeamPos))
         S9xSetIRQ(PPU_H_BEAM_IRQ_SOURCE);
      break;
   }

   S9xReschedule();
}

void S9xDoHBlankProcessing_NoSFX()
{
   CPU.WaitCounter++;
   switch (CPU.WhichEvent)
   {
   case HBLANK_START_EVENT:
      if (IPPU.HDMA && CPU.V_Counter <= PPU.ScreenHeight)
         IPPU.HDMA = S9xDoHDMA(IPPU.HDMA);
      break;
   case HBLANK_END_EVENT:
#ifndef USE_BLARGG_APU
      CPU.Cycles -= Settings.H_Max;
      if (IAPU.APUExecuting)
         APU.Cycles -= Settings.H_Max;
      else
         APU.Cycles = 0;
#else
      S9xAPUExecute();
      CPU.Cycles -= Settings.H_Max;
      S9xAPUSetReferenceTime(CPU.Cycles);
#endif
      CPU.NextEvent = -1;

      if (++CPU.V_Counter >= (Settings.PAL ? SNES_MAX_PAL_VCOUNTER : SNES_MAX_NTSC_VCOUNTER))
      {
         CPU.V_Counter = 0;
         Memory.FillRAM[0x213F] ^= 0x80;
         PPU.RangeTimeOver = 0;
         CPU.NMIActive = false;
         ICPU.Frame++;
         CPU.Flags |= SCAN_KEYS_FLAG;
         S9xStartHDMA();
      }

      if (PPU.VTimerEnabled && !PPU.HTimerEnabled && CPU.V_Counter == PPU.IRQVBeamPos)
         S9xSetIRQ(PPU_V_BEAM_IRQ_SOURCE);

      if (CPU.V_Counter == PPU.ScreenHeight + FIRST_VISIBLE_LINE)
      {
         /* Start of V-blank */
         S9xEndScreenRefresh();
         IPPU.HDMA = 0;
         /* Bits 7 and 6 of $4212 are computed when read in S9xGetPPU. */
         PPU.ForcedBlanking = (Memory.FillRAM [0x2100] >> 7) & 1;

         if (!PPU.ForcedBlanking)
         {
            uint8_t tmp = 0;
            PPU.OAMAddr = PPU.SavedOAMAddr;

            if (PPU.OAMPriorityRotation)
               tmp = (PPU.OAMAddr & 0xFE) >> 1;
            if ((PPU.OAMFlip & 1) || PPU.FirstSprite != tmp)
            {
               PPU.FirstSprite = tmp;
               IPPU.OBJChanged = true;
            }

            PPU.OAMFlip = 0;
         }

         Memory.FillRAM[0x4210] = 0x80 | Model->_5A22;
         if (Memory.FillRAM[0x4200] & 0x80)
         {
            CPU.NMIActive = true;
            CPU.Flags |= NMI_FLAG;
            CPU.NMICycleCount = CPU.NMITriggerPoint;
         }
      }

      if (CPU.V_Counter == PPU.ScreenHeight + 3)
         S9xUpdateJoypads();

      if (CPU.V_Counter == FIRST_VISIBLE_LINE)
      {
         Memory.FillRAM[0x4210] = Model->_5A22;
         CPU.Flags &= ~NMI_FLAG;
         S9xStartScreenRefresh();
      }
      if (CPU.V_Counter >= FIRST_VISIBLE_LINE && CPU.V_Counter < PPU.ScreenHeight + FIRST_VISIBLE_LINE)
         RenderLine(CPU.V_Counter - FIRST_VISIBLE_LINE);
#ifndef USE_BLARGG_APU
      if (APU.TimerEnabled [2])
      {
         APU.Timer [2] += 4;
         while (APU.Timer [2] >= APU.TimerTarget [2])
         {
            IAPU.RAM [0xff] = (IAPU.RAM [0xff] + 1) & 0xf;
            APU.Timer [2] -= APU.TimerTarget [2];
            IAPU.WaitCounter++;
            IAPU.APUExecuting = true;
         }
      }
      if (CPU.V_Counter & 1)
      {
         if (APU.TimerEnabled [0])
         {
            APU.Timer [0]++;
            if (APU.Timer [0] >= APU.TimerTarget [0])
            {
               IAPU.RAM [0xfd] = (IAPU.RAM [0xfd] + 1) & 0xf;
               APU.Timer [0] = 0;
               IAPU.WaitCounter++;
               IAPU.APUExecuting = true;
            }
         }
         if (APU.TimerEnabled [1])
         {
            APU.Timer [1]++;
            if (APU.Timer [1] >= APU.TimerTarget [1])
            {
               IAPU.RAM [0xfe] = (IAPU.RAM [0xfe] + 1) & 0xf;
               APU.Timer [1] = 0;
               IAPU.WaitCounter++;
               IAPU.APUExecuting = true;
            }
         }
      }
#endif
      break;
   case HTIMER_BEFORE_EVENT:
   case HTIMER_AFTER_EVENT:
      if (PPU.HTimerEnabled && (!PPU.VTimerEnabled || CPU.V_Counter == PPU.IRQVBeamPos))
         S9xSetIRQ(PPU_H_BEAM_IRQ_SOURCE);
      break;
   }
   S9xReschedule();
}