#include "ssound.h"

#include <cstring>

#include "game/runtime.h"
#include "game/sound/sndshim.h"

s32 gMusicFade = 0;
s32 gSema;
Sound gSounds[64];
Vec3w gEarTrans[2];
Vec3w gCamTrans;
s32 gMusicFadeDir = 0;
Curve gCurves[16];
s32 gCamAngle;
u8 gMirrorMode = 0;
u32 sLastTick = 0;

static s32 sqrt_table[256] = {
    0,     4096,  5793,  7094,  8192,  9159,  10033, 10837, 11585, 12288, 12953, 13585, 14189,
    14768, 15326, 15864, 16384, 16888, 17378, 17854, 18318, 18770, 19212, 19644, 20066, 20480,
    20886, 21283, 21674, 22058, 22435, 22806, 23170, 23530, 23884, 24232, 24576, 24915, 25249,
    25580, 25905, 26227, 26545, 26859, 27170, 27477, 27780, 28081, 28378, 28672, 28963, 29251,
    29537, 29819, 30099, 30377, 30652, 30924, 31194, 31462, 31727, 31991, 32252, 32511, 32768,
    33023, 33276, 33527, 33776, 34024, 34270, 34514, 34756, 34996, 35235, 35472, 35708, 35942,
    36175, 36406, 36636, 36864, 37091, 37316, 37540, 37763, 37985, 38205, 38424, 38642, 38858,
    39073, 39287, 39500, 39712, 39923, 40132, 40341, 40548, 40755, 40960, 41164, 41368, 41570,
    41771, 41972, 42171, 42369, 42567, 42763, 42959, 43154, 43348, 43541, 43733, 43925, 44115,
    44305, 44494, 44682, 44869, 45056, 45242, 45427, 45611, 45795, 45977, 46160, 46341, 46522,
    46702, 46881, 47059, 47237, 47415, 47591, 47767, 47942, 48117, 48291, 48465, 48637, 48809,
    48981, 49152, 49322, 49492, 49661, 49830, 49998, 50166, 50332, 50499, 50665, 50830, 50995,
    51159, 51323, 51486, 51649, 51811, 51972, 52134, 52294, 52454, 52614, 52773, 52932, 53090,
    53248, 53405, 53562, 53719, 53874, 54030, 54185, 54340, 54494, 54647, 54801, 54954, 55106,
    55258, 55410, 55561, 55712, 55862, 56012, 56162, 56311, 56459, 56608, 56756, 56903, 57051,
    57198, 57344, 57490, 57636, 57781, 57926, 58071, 58215, 58359, 58503, 58646, 58789, 58931,
    59073, 59215, 59357, 59498, 59639, 59779, 59919, 60059, 60199, 60338, 60477, 60615, 60753,
    60891, 61029, 61166, 61303, 61440, 61576, 61712, 61848, 61984, 62119, 62254, 62388, 62523,
    62657, 62790, 62924, 63057, 63190, 63323, 63455, 63587, 63719, 63850, 63982, 64113, 64243,
    64374, 64504, 64634, 64763, 64893, 65022, 65151, 65279, 65408,
};

static s32 atan_table[257] = {
    0,  0,  0,  0,  0,  1,  1,  1,  1,  2,  2,  2,  2,  2,  3,  3,  3,  3,  4,  4,  4,  4,  4,  5,
    5,  5,  5,  6,  6,  6,  6,  6,  7,  7,  7,  7,  8,  8,  8,  8,  8,  9,  9,  9,  9,  9,  10, 10,
    10, 10, 11, 11, 11, 11, 11, 12, 12, 12, 12, 12, 13, 13, 13, 13, 14, 14, 14, 14, 14, 15, 15, 15,
    15, 15, 16, 16, 16, 16, 16, 17, 17, 17, 17, 17, 18, 18, 18, 18, 18, 19, 19, 19, 19, 19, 20, 20,
    20, 20, 20, 21, 21, 21, 21, 21, 22, 22, 22, 22, 22, 23, 23, 23, 23, 23, 23, 24, 24, 24, 24, 24,
    25, 25, 25, 25, 25, 26, 26, 26, 26, 26, 26, 27, 27, 27, 27, 27, 27, 28, 28, 28, 28, 28, 29, 29,
    29, 29, 29, 29, 30, 30, 30, 30, 30, 30, 31, 31, 31, 31, 31, 31, 32, 32, 32, 32, 32, 32, 32, 33,
    33, 33, 33, 33, 33, 34, 34, 34, 34, 34, 34, 34, 35, 35, 35, 35, 35, 35, 35, 36, 36, 36, 36, 36,
    36, 37, 37, 37, 37, 37, 37, 37, 37, 38, 38, 38, 38, 38, 38, 38, 39, 39, 39, 39, 39, 39, 39, 40,
    40, 40, 40, 40, 40, 40, 40, 41, 41, 41, 41, 41, 41, 41, 41, 42, 42, 42, 42, 42, 42, 42, 42, 43,
    43, 43, 43, 43, 43, 43, 43, 43, 44, 44, 44, 44, 44, 44, 44, 44, 45,
};

void ssound_init_globals() {
  gMusicFade = 0;
  gSema = 0;
}

Sound* LookupSound(s32 id) {
  if (id == 0) {
    return nullptr;
  }

  for (auto& s : gSounds) {
    if (s.id == id) {
      s32 sndid = snd_SoundIsStillPlaying(s.sound_handle);
      s.sound_handle = sndid;

      if (sndid == 0) {
        s.id = 0;
        return nullptr;
      }

      return &s;
    }
  }

  return nullptr;
}

void CleanSounds() {
  for (auto& s : gSounds) {
    if (s.id != 0) {
      s32 sndid = snd_SoundIsStillPlaying(s.sound_handle);
      s.sound_handle = sndid;

      if (sndid == 0) {
        s.id = 0;
      }
    }
  }
}

s32 CalculateAngle(Vec3w* trans) {
  s32 diffX = gCamTrans.x - trans->x;
  s32 diffZ = gCamTrans.z - trans->z;
  s32 angle;

  s32 lookupX = diffX;
  s32 lookupZ = diffZ;

  if (diffX < 0) {
    lookupX = trans->x - gCamTrans.x;
  }

  if (diffZ < 0) {
    lookupZ = trans->z - gCamTrans.z;
  }

  if (lookupX == 0 && lookupZ == 0) {
    return 0;
  }

  if (lookupZ >= lookupX) {
    angle = atan_table[(lookupX << 8) / lookupZ];

    if (diffZ >= 0) {
      if (diffX < 0) {
        angle = 360 - angle;
      }
    } else if (diffX >= 0) {
      angle = 180 - angle;
    } else {
      angle += 180;
    }
  } else {
    angle = atan_table[(lookupZ << 8) / lookupX];

    if (diffX >= 0) {
      if (diffZ >= 0) {
        angle = 90 - angle;
      } else {
        angle = angle + 90;
      }
    } else {
      if (diffZ >= 0) {
        angle = angle + 270;
      } else {
        angle = 270 - angle;
      }
    }
  }

  angle = (angle - gCamAngle + 720) % 360;

  if (gMirrorMode) {
    angle = ((180 - angle) + 180) % 360;
  }

  return angle;
}

s32 GetPan(Sound* sound) {
  return CalculateAngle(&sound->params.trans);
}

s32 CalculateFalloffVolume(Vec3w* pos, s32 volume, s32 fo_curve, s32 fo_min, s32 fo_max) {
  s32 xdiff = 0;
  s32 ydiff = 0;
  s32 zdiff = 0;

  if (g_game_version == GameVersion::Jak1) {
    if (fo_curve == 0) {
      return volume;
    }

    xdiff = gEarTrans[0].x - pos->x;
    ydiff = gEarTrans[0].y - pos->y;
    zdiff = gEarTrans[0].z - pos->z;
  } else {
    if (fo_curve == 1) {
      return volume;
    } else if (fo_curve == 9 || fo_curve == 11) {
      xdiff = gEarTrans[1].x - pos->x;
      ydiff = gEarTrans[1].y - pos->y;
      zdiff = gEarTrans[1].z - pos->z;
    } else if (fo_curve == 10) {
      xdiff = 0;
      ydiff = gEarTrans[0].y - pos->y;
      zdiff = 0;
    } else {
      xdiff = gEarTrans[0].x - pos->x;
      ydiff = gEarTrans[0].y - pos->y;
      zdiff = gEarTrans[0].z - pos->z;
    }
  }

  if (xdiff < 0) {
    xdiff = -xdiff;
  }
  if (ydiff < 0) {
    ydiff = -ydiff;
  }
  if (zdiff < 0) {
    zdiff = -zdiff;
  }

  s32 min = fo_min << 8;
  s32 max = fo_max << 8;

  s32 new_vol = 0;
  if (xdiff <= max && ydiff <= max && zdiff <= max) {
    while (max > 0x7fff) {
      max >>= 1;
      min >>= 1;
      xdiff >>= 1;
      ydiff >>= 1;
      zdiff >>= 1;
    }
    u32 dist_squared = xdiff * xdiff + ydiff * ydiff + zdiff * zdiff;
    s32 dist_steps = 0;
    if (dist_squared != 0) {
      while ((dist_squared & 0xc0000000) == 0) {
        ++dist_steps;
        dist_squared <<= 2;
      }
      dist_steps = sqrt_table[dist_squared >> 24] >> (dist_steps & 0x1f);
    }
    new_vol = volume;
    if (min < dist_steps) {
      u32 voldiff = dist_steps - min;
      if (dist_steps < max) {
        dist_steps = max - min;
        while (voldiff > 0xffff) {
          dist_steps >>= 1;
          voldiff >>= 1;
        }
        voldiff = (voldiff << 0x10) / dist_steps;
        if (voldiff != 0x10000) {
          new_vol = (voldiff * voldiff) >> 0x10;
          new_vol = (gCurves[fo_curve].unk4 * 0x10000 + gCurves[fo_curve].unk3 * voldiff +
                     gCurves[fo_curve].unk2 * new_vol +
                     gCurves[fo_curve].unk1 * ((new_vol * voldiff) >> 0x10)) >>
                    0xc;
          if (new_vol < 0) {
            new_vol = 0;
          } else if (0x10000 < new_vol) {
            new_vol = 0x10000;
          }
          new_vol = (new_vol * volume) >> 0x10;
        }
      } else {
        new_vol = 0;
      }
    }
  }
  if (fo_curve == 11 && new_vol < 0x180) {
    new_vol = 0x180;
  }
  return new_vol;
}

s32 GetVolume(Sound* sound) {
  return CalculateFalloffVolume(&sound->params.trans, sound->params.volume, sound->params.fo_curve,
                                sound->params.fo_min, sound->params.fo_max);
}

void UpdateVolume(Sound* sound) {
  s32 id = snd_SoundIsStillPlaying(sound->sound_handle);
  sound->sound_handle = id;
  if (sound->sound_handle == 0) {
    sound->id = 0;
  } else {
    s32 volume = GetVolume(sound);
    snd_SetSoundVolPan(id, volume, -2);
  }
}

Sound* AllocateSound() {
  for (auto& s : gSounds) {
    if (s.id == 0) {
      return &s;
    }
  }

  CleanSounds();
  for (auto& s : gSounds) {
    if (s.id == 0) {
      return &s;
    }
  }

  return nullptr;
}

void KillSoundsInGroup(u8 group) {
  for (auto& s : gSounds) {
    if (s.id != 0) {
      s32 sndid = snd_SoundIsStillPlaying(s.sound_handle);
      s.sound_handle = sndid;

      if (sndid == 0) {
        s.id = 0;
      } else if (s.params.group & group) {
        snd_StopSound(s.sound_handle);
        s.id = 0;
      }
    }
  }
}

void UpdateAutoVol(Sound* sound, s32 ticks) {
  if (ticks < sound->auto_time) {
    s32 nvol = sound->new_volume;
    if (nvol == -4) {
      nvol = 0;
    }

    s32 step = (nvol - sound->params.volume) * ticks;
    step /= sound->auto_time;

    if (step >= 0) {
      if (step == 0) {
        step = 1;
      }

      sound->params.volume += step;
      if (sound->new_volume < sound->params.volume) {
        sound->params.volume = sound->new_volume;
      }

    } else {
      sound->params.volume += step;
      if (sound->new_volume > sound->params.volume) {
        sound->params.volume = sound->new_volume;
      }
    }

    sound->auto_time -= ticks;
    return;
  }

  if (sound->new_volume == -4) {
    snd_StopSound(sound->sound_handle);
    sound->id = 0;
  } else {
    sound->params.volume = sound->new_volume;
  }

  sound->auto_time = 0;
}

void PrintActiveSounds() {
  char string[64];

  for (auto& s : gSounds) {
    if (s.id != 0 && s.is_music == 0) {
      if (s.bank_entry != nullptr) {
        u32 len = strlen(s.bank_entry->name.data());
        if (len > 16) {
          len = 16;
        }
        sprintf(string, "                 : Vol %d", GetVolume(&s));
        memcpy(string, s.bank_entry->name.data(), len);
        printf("%s\n", string);
      } else {  // added for printing jak2 sounds
        u32 len = strlen(s.name);
        if (len > 16) {
          len = 16;
        }
        sprintf(string, "                 : Vol %d, ID %d, Curve %d", GetVolume(&s), s.id,
                s.params.fo_curve);
        memcpy(string, s.name, len);
        printf("%s\n", string);
      }
    }
  }
}

void SetCurve(s32 curve, s32 falloff, s32 ease) {
  gCurves[curve].unk1 = ease * 2;
  gCurves[curve].unk2 = falloff - 3 * ease;
  gCurves[curve].unk3 = ease - falloff - 0x1000;
  gCurves[curve].unk4 = 0x1000;
}