#include "Base/Types.h"


#include <string.h>



#include "Utility/MathUtil.h"
#include "Core/Memory.h"
#include "Debug/DBGConsole.h"
#include "HLEAudio/HLEAudioInternal.h"
#include "HLEAudio/HLEAudioState.h"
#include <algorithm>
extern bool isMKABI;
extern bool isZeldaABI;

inline u16 Sample_Mask(u32 x) { return (u16)(x & 0xffff); }

void CLEARBUFF(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "CLEARBUFF");
#endif
  u16 addr = command.Abi1ClearBuffer.Address;
  u16 count = command.Abi1ClearBuffer.Count;

  gAudioHLEState.ClearBuffer(addr, count);
}

void CLEARBUFF2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "CLEARBUFF2");
#endif
  u16 addr = command.Abi2ClearBuffer.Address;
  u16 count = command.Abi2ClearBuffer.Count;

  gAudioHLEState.ClearBuffer(addr, count);
}
void CLEARBUFF3(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "CLEARBUFF3");
#endif
  u16 addr = (u16)(command.cmd0 & 0xffff);
  u16 count = (u16)(command.cmd1 & 0xffff);
  memset(gAudioHLEState.Buffer + addr + 0x4f0, 0, count);
}

void DMEMMOVE(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "DMEMMOVE");
#endif
  u16 src = command.Abi1DmemMove.Src;
  u16 dst = command.Abi1DmemMove.Dst;
  u16 count = command.Abi1DmemMove.Count;

  gAudioHLEState.DmemMove(dst, src, count);
}

void DMEMMOVE2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "DMEMMOVE2");
#endif
  u16 src = command.Abi2DmemMove.Src;
  u16 dst = command.Abi2DmemMove.Dst;
  u16 count = command.Abi2DmemMove.Count;

  gAudioHLEState.DmemMove(dst, src, count);
}

void DMEMMOVE3(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "DMEMMOVE3");
#endif
  u16 src = command.Abi3DmemMove.Src;
  u16 dst = command.Abi3DmemMove.Dst;
  u16 count = command.Abi3DmemMove.Count;

  gAudioHLEState.DmemMove(dst + 0x4f0, src + 0x4f0, count);
}

void DUPLICATE2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "DUPLICATE2");
#endif
  u32 Count = (command.cmd0 >> 16) & 0xff;
  u32 In = command.cmd0 & 0xffff;
  u32 Out = command.cmd1 >> 16;

  u16 buff[64];

  memmove(buff, gAudioHLEState.Buffer + In, 128);

  while (Count) {
    memmove(gAudioHLEState.Buffer + Out, buff, 128);
    Out += 128;
    Count--;
  }
}

void LOADBUFF(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "LOADBUFF");
#endif
  u32 addr = command.Abi1LoadBuffer.Address+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];

  gAudioHLEState.LoadBuffer(addr);
}

void LOADBUFF2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "LOADBUFF2");
#endif
  // Needs accuracy verification...
  u32 src = command.Abi2LoadBuffer.SrcAddr+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];
  u16 dst = command.Abi2LoadBuffer.DstAddr;
  u16 count = command.Abi2LoadBuffer.Count;

  gAudioHLEState.LoadBuffer(dst, src, count);
}

void LOADBUFF3(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "LOADBUFF3");
#endif
  u32 v0;
  u32 cnt = (((command.cmd0 >> 0xC) + 3) & 0xFFC);
  v0 = (command.cmd1 & 0xfffffc);
  u32 src = (command.cmd0 & 0xffc) + 0x4f0;
  memmove(gAudioHLEState.Buffer + src, rdram + v0, cnt);
}

void SAVEBUFF(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SAVEBUFF");
#endif
  u32 addr = command.Abi1SaveBuffer.Address+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];

  gAudioHLEState.SaveBuffer(addr);
}

void SAVEBUFF2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SAVEBUFF2");
#endif
  // Needs accuracy verification...
  u32 dst = command.Abi2SaveBuffer.DstAddr+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];
  u16 src = command.Abi2SaveBuffer.SrcAddr;
  u16 count = command.Abi2SaveBuffer.Count;

  gAudioHLEState.SaveBuffer(dst, src, count);
}

void SAVEBUFF3(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SAVEBUFF3");
#endif
  u32 cnt = ((command.cmd0 >> 0xC) + 3) & 0xFFC;
  u32 v0 = (command.cmd1 & 0xfffffc);
  u32 src = (command.cmd0 & 0xffc) + 0x4f0;
  memmove(rdram + v0, gAudioHLEState.Buffer + src, cnt);
}

void SEGMENT(AudioHLECommand command [[maybe_unused]]) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SEGMENT - Implementing SEGMENT command");
#endif

  // Extract segment and offset
  uint8_t segment = (command.Abi1SetSegment.Segment >> 24) & 0x3f;   // Extract the 6-bit segment ID
  uint32_t offset = command.Abi1SetSegment.Segment & 0xffffff;        // Extract the 24-bit offset

  // Check if the segment is valid (assuming 'n' is the maximum number of segments, e.g., 16 or 32)
  if (segment >= 16) {
#ifdef DEBUG_AUDIO
    DBGConsole_Msg(0, "SEGMENT - Invalid segment %u", segment);
#endif
    return;
  }

  // Store the segment offset in the audio HLE state
  gAudioHLEState.SetSegment(segment, offset);

#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SEGMENT - Segment %u set to offset 0x%06x", segment, offset);
#endif
}
void SEGMENT2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SEGMENT2");
#endif
  if (isZeldaABI) {
    FILTER2(command);
    return;
  }
  if ((command.cmd0 & 0xffffff) == 0) {
    isMKABI = true;
    // gAudioHLEState.Segments[(command.cmd1>>24)&0xf] = (command.cmd1 &
    // 0xffffff);
  } else {
    isMKABI = false;
    isZeldaABI = true;
    FILTER2(command);
  }
}

void SETBUFF(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SETBUFF");
#endif
  u8 flags = command.Abi1SetBuffer.Flags;
  u16 in = command.Abi1SetBuffer.In;
  u16 out = command.Abi1SetBuffer.Out;
  u16 count = command.Abi1SetBuffer.Count;

  gAudioHLEState.SetBuffer(flags, in, out, count);
}

void SETBUFF2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SETBUFF2");
#endif
  u16 in = command.Abi2SetBuffer.In;
  u16 out = command.Abi2SetBuffer.Out;
  u16 count = command.Abi2SetBuffer.Count;

#ifdef DAEDALUS_ENABLE_ASSERTS
  u8 flags(command.Abi2SetBuffer.Flags);
  DAEDALUS_ASSERT(flags == 0, "SETBUFF flags set: %02x", flags);
#endif

  gAudioHLEState.SetBuffer(0, in, out, count);
}

void SETLOOP(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SETLOOP");
#endif
  u32 loopval = command.Abi1SetLoop.LoopVal+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];

  gAudioHLEState.SetLoop(loopval);
}

void SETLOOP2(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SETLOOP2");
#endif
  u32 loopval = command.Abi2SetLoop.LoopVal+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];

  gAudioHLEState.SetLoop(loopval);
}

void SETLOOP3(AudioHLECommand command) {
#ifdef DEBUG_AUDIO
  DBGConsole_Msg(0, "SETLOOP3");
#endif
  u32 loopval = command.Abi3SetLoop.LoopVal+ gAudioHLEState.Segments[(command.cmd1>>24)&0xf];

  gAudioHLEState.SetLoop(loopval);
}