#include <Scheduler.hpp>
#include <Core.hpp>

void Scheduler::EnqueueRelative(const u64 t, const EventType type) { EnqueueAbsolute(t + ticks, type); }

void Scheduler::EnqueueAbsolute(const u64 t, const EventType type) { events.push({t, type}); }

u64 Scheduler::Remove(const EventType eventType) const {
  for (auto &[time, type] : events) {
    if (type == eventType) {
      const u64 ret = time - ticks;
      type = NONE;
      time = ticks;
      return ret;
    }
  }

  return 0;
}

void Scheduler::Tick(const u64 t) {
  n64::Mem& mem = n64::Core::GetMem();
  ticks += t;
  n64::MI &mi = mem.mmio.mi;
  n64::SI &si = mem.mmio.si;
  n64::PI &pi = mem.mmio.pi;

  while (ticks >= events.top().time) {
    switch (const auto type = events.top().type) {
    case SI_DMA:
      si.DMA();
      break;
    case PI_DMA_COMPLETE:
      mi.InterruptRaise(n64::MI::Interrupt::PI);
      pi.dmaBusy = false;
      break;
    case PI_BUS_WRITE_COMPLETE:
      pi.ioBusy = false;
      break;
    case NONE:
      break;
    case IMPOSSIBLE:
      Util::Error::GetInstance().Throw({Util::Error::Severity::UNRECOVERABLE}, {Util::Error::Type::ROM_LOAD_ERROR}, {}, {}, "Unrecognized rom endianness");
      return;
    default:
      Util::Error::GetInstance().Throw({Util::Error::Severity::UNRECOVERABLE}, {Util::Error::Type::ROM_LOAD_ERROR}, {}, {}, "Unknown scheduler event type {}", static_cast<int>(type));
      return;
    }
    events.pop();
  }
}