#include "CMorphBallShadow.hpp"
#include "CStateManager.hpp"
#include "World/CWorld.hpp"
#include "GameGlobalObjects.hpp"
#include "Graphics/CBooRenderer.hpp"
#include "World/CPlayer.hpp"
#include "Particle/CGenDescription.hpp"

namespace urde {

void CMorphBallShadow::GatherAreas(const CStateManager& mgr) {
  x18_areas.clear();
  for (const CGameArea& area : *mgr.GetWorld()) {
    CGameArea::EOcclusionState occState = CGameArea::EOcclusionState::Occluded;
    if (area.IsPostConstructed())
      occState = area.GetPostConstructed()->x10dc_occlusionState;
    if (occState == CGameArea::EOcclusionState::Visible)
      x18_areas.push_back(area.GetAreaId());
  }
}

void CMorphBallShadow::RenderIdBuffer(const zeus::CAABox& aabb, const CStateManager& mgr, CPlayer& player) {
  xb8_shadowVolume = aabb;
  x0_actors.clear();
  x18_areas.clear();
  x30_worldModelBits.clear();
  g_Renderer->x318_26_requestRGBA6 = true;

  if (!g_Renderer->x318_27_currentRGBA6) {
    xd0_hasIds = false;
    return;
  }

  GatherAreas(mgr);

  SViewport backupVp = g_Viewport;
  g_Renderer->BindBallShadowIdTarget();
  CGraphics::g_BooMainCommandQueue->clearTarget();

  zeus::CTransform backupViewMtx = CGraphics::g_ViewMatrix;
  CGraphics::CProjectionState backupProjection = CGraphics::g_Proj;
  zeus::CVector2f backupDepth = CGraphics::g_CachedDepthRange;
  zeus::CTransform viewMtx(
      zeus::CVector3f::skRight, zeus::CVector3f::skDown, zeus::CVector3f::skForward,
      zeus::CVector3f((aabb.min.x() + aabb.max.x()) * 0.5f, (aabb.min.y() + aabb.max.y()) * 0.5f, aabb.max.z()));

  CGraphics::SetDepthRange(DEPTH_NEAR, DEPTH_FAR);
  float vpX = (aabb.max.x() - aabb.min.x()) * 0.5f;
  float vpY = (aabb.max.y() - aabb.min.y()) * 0.5f;
  float vpZ = (aabb.max.z() - aabb.min.z()) + FLT_EPSILON;
  CGraphics::SetOrtho(-vpX, vpX, vpY, -vpY, 0.f, vpZ);

  rstl::reserved_vector<TUniqueId, 1024> nearItems;
  mgr.BuildNearList(nearItems, aabb, CMaterialFilter::skPassEverything, &player);

  CGraphics::SetViewPointMatrix(viewMtx);

  int alphaVal = 4;
  for (TUniqueId id : nearItems) {
    if (alphaVal > 255)
      break;

    const CActor* actor = static_cast<const CActor*>(mgr.GetObjectById(id));
    if (!actor || !actor->CanDrawStatic())
      continue;

    x0_actors.push_back(actor);

    const CModelData* modelData = actor->GetModelData();
    zeus::CTransform modelXf = actor->GetTransform() * zeus::CTransform::Scale(modelData->GetScale());
    CGraphics::SetModelMatrix(modelXf);

    CModelFlags flags(0, 0, 3, zeus::CColor{1.f, 1.f, 1.f, alphaVal / 255.f});
    flags.m_extendedShader = EExtendedShader::SolidColor; // Do solid color draw
    const CBooModel& model = *modelData->PickStaticModel(CModelData::EWhichModel::Normal);
    const_cast<CBooModel&>(model).VerifyCurrentShader(flags.x1_matSetIdx);
    model.DrawNormal(flags, nullptr, nullptr);
    alphaVal += 4;
  }

  CGraphics::SetModelMatrix(zeus::CTransform::Identity());

  g_Renderer->FindOverlappingWorldModels(x30_worldModelBits, aabb);
  alphaVal = g_Renderer->DrawOverlappingWorldModelIDs(alphaVal, x30_worldModelBits, aabb);

  g_Renderer->ResolveBallShadowIdTarget();

  g_Renderer->BindMainDrawTarget();
  CGraphics::SetViewPointMatrix(backupViewMtx);
  CGraphics::SetProjectionState(backupProjection);
  g_Renderer->SetViewport(backupVp.x0_left, backupVp.x4_top, backupVp.x8_width, backupVp.xc_height);
  CGraphics::SetDepthRange(backupDepth[0], backupDepth[1]);

  xd0_hasIds = alphaVal != 4;
}

bool CMorphBallShadow::AreasValid(const CStateManager& mgr) const {
  auto it = x18_areas.begin();
  for (const CGameArea& area : *mgr.GetWorld()) {
    CGameArea::EOcclusionState occState = CGameArea::EOcclusionState::Occluded;
    if (area.IsPostConstructed())
      occState = area.GetPostConstructed()->x10dc_occlusionState;
    if (occState != CGameArea::EOcclusionState::Visible)
      continue;
    if (it == x18_areas.end())
      return false;
    if (*it != area.GetAreaId())
      return false;
    ++it;
  }
  return true;
}

void CMorphBallShadow::Render(const CStateManager& mgr, float alpha) {
  if (!xd0_hasIds || !AreasValid(mgr))
    return;

  CModelFlags flags;
  flags.x4_color.a() = alpha;
  flags.m_extendedShader = EExtendedShader::MorphBallShadow;

  int alphaVal = 4;
  for (const CActor* actor : x0_actors) {
    const CModelData* modelData = actor->GetModelData();
    zeus::CTransform modelXf = actor->GetTransform() * zeus::CTransform::Scale(modelData->GetScale());
    CGraphics::SetModelMatrix(modelXf);

    flags.x4_color.r() = alphaVal / 255.f;
    const CBooModel& model = *modelData->PickStaticModel(CModelData::EWhichModel::Normal);
    const_cast<CBooModel&>(model).VerifyCurrentShader(flags.x1_matSetIdx);
    model.DrawNormal(flags, nullptr, nullptr);
    alphaVal += 4;
  }

  CGraphics::SetModelMatrix(zeus::CTransform::Identity());
  g_Renderer->DrawOverlappingWorldModelShadows(alphaVal, x30_worldModelBits, xb8_shadowVolume, alpha);
}

} // namespace urde