#pragma once

#include <map>
#include <memory>
#include <string>

#include "Runtime/CToken.hpp"
#include "Runtime/Character/CCharacterInfo.hpp"
#include "Runtime/Character/CParticleGenInfo.hpp"
#include "Runtime/Particle/CGenDescription.hpp"
#include "Runtime/Particle/CSwooshDescription.hpp"
#include "Runtime/Particle/CElectricDescription.hpp"

#include <zeus/CColor.hpp>
#include <zeus/CFrustum.hpp>

namespace urde {
class CCharLayoutInfo;
class CPoseAsTransforms;

class CParticleDatabase {
  std::map<CAssetId, std::shared_ptr<TLockedToken<CGenDescription>>> x0_particleDescs;
  std::map<CAssetId, std::shared_ptr<TLockedToken<CSwooshDescription>>> x14_swooshDescs;
  std::map<CAssetId, std::shared_ptr<TLockedToken<CElectricDescription>>> x28_electricDescs;
  std::map<std::string, std::unique_ptr<CParticleGenInfo>> x3c_rendererDrawLoop;
  std::map<std::string, std::unique_ptr<CParticleGenInfo>> x50_firstDrawLoop;
  std::map<std::string, std::unique_ptr<CParticleGenInfo>> x64_lastDrawLoop;
  std::map<std::string, std::unique_ptr<CParticleGenInfo>> x78_rendererDraw;
  std::map<std::string, std::unique_ptr<CParticleGenInfo>> x8c_firstDraw;
  std::map<std::string, std::unique_ptr<CParticleGenInfo>> xa0_lastDraw;
  bool xb4_24_updatesEnabled : 1 = true;
  bool xb4_25_anySystemsDrawnWithModel : 1 = false;

  static void SetModulationColorAllActiveEffectsForParticleDB(
      const zeus::CColor& color, std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map);
  static void SuspendAllActiveEffectsForParticleDB(CStateManager& mgr,
                                                   std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map);
  static void DeleteAllLightsForParticleDB(CStateManager& mgr,
                                           std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map);
  static void RenderParticleGenMap(const std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map);
  static void RenderParticleGenMapMasked(const std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map, int mask,
                                         int target);
  static void AddToRendererClippedParticleGenMap(const std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map,
                                                 const zeus::CFrustum& frustum);
  static void
  AddToRendererClippedParticleGenMapMasked(const std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map,
                                           const zeus::CFrustum& frustum, int mask, int target);
  static void UpdateParticleGenDB(float dt, const CPoseAsTransforms& pose, const CCharLayoutInfo& charInfo,
                                  const zeus::CTransform& xf, const zeus::CVector3f& vec, CStateManager& stateMgr,
                                  std::map<std::string, std::unique_ptr<CParticleGenInfo>>& map, bool deleteIfDone);

public:
  CParticleDatabase();
  void CacheParticleDesc(const SObjectTag& tag);
  void CacheParticleDesc(const CCharacterInfo::CParticleResData& desc);
  void SetModulationColorAllActiveEffects(const zeus::CColor& color);
  void SuspendAllActiveEffects(CStateManager& stateMgr);
  void DeleteAllLights(CStateManager& stateMgr);
  void Update(float dt, const CPoseAsTransforms& pose, const CCharLayoutInfo& charInfo, const zeus::CTransform& xf,
              const zeus::CVector3f& scale, CStateManager& stateMgr);
  void RenderSystemsToBeDrawnLastMasked(int mask, int target) const;
  void RenderSystemsToBeDrawnLast() const;
  void RenderSystemsToBeDrawnFirstMasked(int mask, int target) const;
  void RenderSystemsToBeDrawnFirst() const;
  void AddToRendererClippedMasked(const zeus::CFrustum& frustum, int mask, int target) const;
  void AddToRendererClipped(const zeus::CFrustum& frustum) const;
  CParticleGenInfo* GetParticleEffect(std::string_view name) const;
  void SetParticleEffectState(std::string_view name, bool active, CStateManager& mgr);
  void SetCEXTValue(std::string_view name, int idx, float value);
  void AddAuxiliaryParticleEffect(std::string_view name, int flags, const CAuxiliaryParticleData& data,
                                  const zeus::CVector3f& scale, CStateManager& mgr, TAreaId aid, int lightId);
  void AddParticleEffect(std::string_view name, int flags, const CParticleData& data, const zeus::CVector3f& scale,
                         CStateManager& mgr, TAreaId aid, bool oneShot, int lightId);
  void InsertParticleGen(bool oneShot, int flags, std::string_view name, std::unique_ptr<CParticleGenInfo>&& gen);
  bool AreAnySystemsDrawnWithModel() const { return xb4_25_anySystemsDrawnWithModel; }
  void SetUpdatesEnabled(bool active) { xb4_24_updatesEnabled = active; }
};
} // namespace urde