#ifndef _CGUNCONTROLLER
#define _CGUNCONTROLLER

#include "types.h"

#include "MetroidPrime/Weapons/GunController/CGSFreeLook.hpp"

class CModelData;

enum EGunState {
  kGS_Inactive,
  kGS_Default,
  kGS_FreeLook,
  kGS_ComboFire,
  kGS_Idle,
  kGS_Fidget,
  kGS_Strike,
  kGS_BigStrike
};

class CStateManager;

class CGunController {
  CModelData& x0_modelData;
  CGSFreeLook x4_freeLook;
  // CGSComboFire x1c_comboFire;
  // CGSFidget x30_fidget;
  u8 x1c_pad[0x34];
  EGunState x50_gunState;
  int x54_curAnimId;
  bool x58_24_animDone : 1;
  bool x58_25_enteredComboFire : 1;

public:
  explicit CGunController(CModelData& modelData);
  ~CGunController();

  void EnterFreeLook(CStateManager&, int, int);
  void EnterComboFire(CStateManager&, int);
  void EnterFidget(CStateManager&, int, int, int);
  void EnterStruck(CStateManager&, float);
  void LoadFidgetAnimAsync(CStateManager&, int, int, int);
  int Update(float, CStateManager&);
  void EnterIdle(CStateManager&);
  void ReturnToDefault(CStateManager&, float);
  void Reset();
  void ReturnToBasePosition(CStateManager&);

  int GetFreeLookSetId() const { return x4_freeLook.GetSetId(); }
};
CHECK_SIZEOF(CGunController, 0x5C)

#endif // _CGUNCONTROLLER