#pragma once

#include "CGunWeapon.hpp"

namespace urde {

class CPowerBeam final : public CGunWeapon {
  enum class ESmokeState { Inactive, Active, Done };
  TCachedToken<CGenDescription> x21c_shotSmoke;
  TCachedToken<CGenDescription> x228_power2nd1;
  std::unique_ptr<CElementGen> x234_shotSmokeGen;
  std::unique_ptr<CElementGen> x238_power2ndGen;
  float x23c_smokeTimer = 0.f;
  ESmokeState x240_smokeState = ESmokeState::Inactive;
  bool x244_24 : 1;
  bool x244_25_loaded : 1;
  void ReInitVariables();

public:
  CPowerBeam(CAssetId characterId, EWeaponType type, TUniqueId playerId, EMaterialTypes playerMaterial,
             const zeus::CVector3f& scale);

  void PreRenderGunFx(const CStateManager& mgr, const zeus::CTransform& xf);
  void PostRenderGunFx(const CStateManager& mgr, const zeus::CTransform& xf);
  void UpdateGunFx(bool shotSmoke, float dt, const CStateManager& mgr, const zeus::CTransform& xf);
  void Fire(bool underwater, float dt, EChargeState chargeState, const zeus::CTransform& xf, CStateManager& mgr,
            TUniqueId homingTarget, float chargeFactor1, float chargeFactor2);
  void EnableSecondaryFx(ESecondaryFxType type);
  void Update(float dt, CStateManager& mgr);
  void Load(CStateManager& mgr, bool subtypeBasePose);
  void Unload(CStateManager& mgr);
  bool IsLoaded() const;
};

} // namespace urde