#include "sdshim.h"
#include <cstring>
#include "common/common_types.h"
#include "game/sound/common/voice.h"
#include "third-party/fmt/core.h"

std::shared_ptr<snd::voice> voice;
u8 spu_memory[0xc060];

static sceSdTransIntrHandler trans_handler[2] = {nullptr, nullptr};
static void* userdata[2] = {nullptr, nullptr};

u32 sceSdGetSwitch(u32 entry) {
  // we can ignore this, only used for getting vmix
  return 0;
}

u32 sceSdGetAddr(u32 entry) {
  // u32 core = entry & 1;
  // u32 voice->id = (entry >> 1) & 0x1f;
  // u32 reg = entry & ~0x3f;

  // Only ever used for getting NAX

  return voice->get_nax() << 1;
}

void sceSdSetSwitch(u32 entry, u32 value) {
  // we can ignore this, only used for vmix
}

void sceSdSetAddr(u32 entry, u32 value) {
  [[maybe_unused]] u32 core = entry & 1;
  [[maybe_unused]] u32 voice_id = (entry >> 1) & 0x1f;
  u32 reg = entry & ~0x3f;

  switch (reg) {
    case SD_VA_SSA: {
      voice->set_ssa(value >> 1);
    } break;
    case SD_VA_LSAX: {
      voice->set_lsa(value >> 1);
    } break;
  }
}

void sceSdSetParam(u32 entry, u32 value) {
  [[maybe_unused]] u32 core = entry & 1;
  [[maybe_unused]] u32 voice_id = (entry >> 1) & 0x1f;
  u32 reg = entry & ~0x3f;

  switch (reg) {
    case SD_VP_VOLL: {
      voice->set_volume_l(value);
    } break;
    case SD_VP_VOLR: {
      voice->set_volume_r(value);
    } break;
    case SD_VP_PITCH: {
      voice->set_pitch(value);
    } break;
    case SD_VP_ADSR1: {
      voice->set_asdr1(value);
    } break;
    case SD_VP_ADSR2: {
      voice->set_asdr2(value);
    } break;
    default: {
    } break;
  }
}

void sceSdSetTransIntrHandler(s32 channel, sceSdTransIntrHandler handler, void* data) {
  trans_handler[channel] = handler;
  userdata[channel] = data;
}

u32 sceSdVoiceTrans(s32 channel, s32 mode, void* iop_addr, u32 spu_addr, u32 size) {
  memcpy(&spu_memory[spu_addr], iop_addr, size);
  if (trans_handler[channel] != nullptr) {
    trans_handler[channel](channel, userdata);
  }
  return size;
}