#include "CBodyStateInfo.hpp"
#include "World/CActor.hpp"
#include "CBodyController.hpp"

namespace urde {

CBodyStateInfo::CBodyStateInfo(CActor& actor, EBodyType type) {
  x34_24_changeLocoAtEndOfAnimOnly = false;
  const CPASDatabase& pasDatabase = actor.GetModelData()->GetAnimationData()->GetCharacterInfo().GetPASDatabase();
  for (int i = 0; i < pasDatabase.GetNumAnimStates(); ++i) {
    const CPASAnimState* state = pasDatabase.GetAnimStateByIndex(i);
    std::unique_ptr<CBodyState> bs;

    switch (type) {
    case EBodyType::BiPedal:
      bs = SetupBiPedalBodyStates(state->GetStateId(), actor);
      break;
    case EBodyType::Restricted:
    default:
      bs = SetupRestrictedBodyStates(state->GetStateId(), actor);
      break;
    case EBodyType::Flyer:
      bs = SetupFlyerBodyStates(state->GetStateId(), actor);
      break;
    case EBodyType::Pitchable:
      bs = SetupPitchableBodyStates(state->GetStateId(), actor);
      break;
    case EBodyType::WallWalker:
      bs = SetupWallWalkerBodyStates(state->GetStateId(), actor);
      break;
    case EBodyType::NewFlyer:
      bs = SetupNewFlyerBodyStates(state->GetStateId(), actor);
      break;
    case EBodyType::RestrictedFlyer:
      bs = SetupRestrictedFlyerBodyStates(state->GetStateId(), actor);
      break;
    }

    if (bs)
      x0_stateMap[pas::EAnimationState(state->GetStateId())] = std::move(bs);
  }

  x1c_additiveStates.reserve(4);
  x1c_additiveStates.push_back({pas::EAnimationState::AdditiveIdle, std::make_unique<CABSIdle>()});
  x1c_additiveStates.push_back({pas::EAnimationState::AdditiveAim, std::make_unique<CABSAim>()});
  x1c_additiveStates.push_back({pas::EAnimationState::AdditiveFlinch, std::make_unique<CABSFlinch>()});
  x1c_additiveStates.push_back({pas::EAnimationState::AdditiveReaction, std::make_unique<CABSReaction>()});
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupRestrictedFlyerBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSRestrictedFlyerLocomotion>(actor);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  default:
    return {};
  }
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupNewFlyerBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSNewFlyerLocomotion>(actor);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  default:
    return {};
  }
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupWallWalkerBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSWallWalkerLocomotion>(actor);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSFlyerTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  default:
    return {};
  }
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupPitchableBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSFlyerLocomotion>(actor, true);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSFlyerTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  default:
    return {};
  }
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupFlyerBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSFlyerLocomotion>(actor, false);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSFlyerTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  default:
    return {};
  }
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupRestrictedBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSRestrictedLocomotion>(actor);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  case 19:
    return std::make_unique<CBSCover>();
  default:
    return {};
  }
}

std::unique_ptr<CBodyState> CBodyStateInfo::SetupBiPedalBodyStates(int stateId, CActor& actor) {
  switch (stateId) {
  case 0:
    return std::make_unique<CBSFall>();
  case 1:
    return std::make_unique<CBSGetup>();
  case 2:
    return std::make_unique<CBSLieOnGround>(actor);
  case 3:
    return std::make_unique<CBSStep>();
  case 4:
    return std::make_unique<CBSDie>();
  case 5:
    return std::make_unique<CBSBiPedLocomotion>(actor);
  case 6:
    return std::make_unique<CBSKnockBack>();
  case 7:
    return std::make_unique<CBSAttack>();
  case 18:
    return std::make_unique<CBSProjectileAttack>();
  case 9:
    return std::make_unique<CBSLoopAttack>();
  case 8:
    return std::make_unique<CBSTurn>();
  case 10:
    return std::make_unique<CBSLoopReaction>();
  case 11:
    return std::make_unique<CBSGroundHit>();
  case 12:
    return std::make_unique<CBSGenerate>();
  case 13:
    return std::make_unique<CBSJump>();
  case 14:
    return std::make_unique<CBSHurled>();
  case 15:
    return std::make_unique<CBSSlide>();
  case 16:
    return std::make_unique<CBSTaunt>();
  case 17:
    return std::make_unique<CBSScripted>();
  case 19:
    return std::make_unique<CBSCover>();
  case 20:
    return std::make_unique<CBSWallHang>();
  default:
    return {};
  }
}

float CBodyStateInfo::GetLocomotionSpeed(pas::ELocomotionAnim anim) const {
  auto search = x0_stateMap.find(pas::EAnimationState::Locomotion);
  if (search != x0_stateMap.cend() && search->second && x18_bodyController) {
    const CBSLocomotion& bs = static_cast<const CBSLocomotion&>(*search->second);
    return bs.GetLocomotionSpeed(x18_bodyController->GetLocomotionType(), anim);
  }
  return 0.f;
}

float CBodyStateInfo::GetMaxSpeed() const {
  float ret = GetLocomotionSpeed(pas::ELocomotionAnim::Run);
  if (std::fabs(ret) < 0.00001f) {
    for (int i = 0; i < 8; ++i) {
      float tmp = GetLocomotionSpeed(pas::ELocomotionAnim(i));
      if (tmp > ret)
        ret = tmp;
    }
  }
  return ret;
}

CBodyState* CBodyStateInfo::GetCurrentState() {
  auto search = x0_stateMap.find(x14_state);
  if (search == x0_stateMap.end())
    return nullptr;
  return search->second.get();
}

const CBodyState* CBodyStateInfo::GetCurrentState() const {
  auto search = x0_stateMap.find(x14_state);
  if (search == x0_stateMap.end())
    return nullptr;
  return search->second.get();
}

void CBodyStateInfo::SetState(pas::EAnimationState s) {
  auto search = x0_stateMap.find(s);
  if (search == x0_stateMap.end())
    return;
  x14_state = s;
}

CAdditiveBodyState* CBodyStateInfo::GetCurrentAdditiveState() {
  for (auto& state : x1c_additiveStates) {
    if (x2c_additiveState == state.first)
      return state.second.get();
  }
  return nullptr;
}

void CBodyStateInfo::SetAdditiveState(pas::EAnimationState s) {
  for (auto& state : x1c_additiveStates) {
    if (s == state.first) {
      x2c_additiveState = s;
      return;
    }
  }
}

bool CBodyStateInfo::ApplyHeadTracking() const {
  if (x14_state == pas::EAnimationState::Invalid)
    return false;
  return GetCurrentState()->ApplyHeadTracking();
}

} // namespace urde