mirror of https://github.com/AxioDL/metaforce.git
123 lines
4.7 KiB
C++
123 lines
4.7 KiB
C++
#pragma once
|
|
|
|
#include "Runtime/RetroTypes.hpp"
|
|
#include "Runtime/World/CEntity.hpp"
|
|
|
|
#include <zeus/CVector3f.hpp>
|
|
|
|
namespace metaforce {
|
|
class CStateManager;
|
|
class CAi;
|
|
|
|
class CTeamAiRole {
|
|
friend class CTeamAiMgr;
|
|
|
|
public:
|
|
enum class ETeamAiRole { Invalid = -1, Initial, Melee, Ranged, Unknown, Unassigned };
|
|
|
|
private:
|
|
TUniqueId x0_ownerId;
|
|
ETeamAiRole x4_roleA = ETeamAiRole::Invalid;
|
|
ETeamAiRole x8_roleB = ETeamAiRole::Invalid;
|
|
ETeamAiRole xc_roleC = ETeamAiRole::Invalid;
|
|
ETeamAiRole x10_curRole = ETeamAiRole::Invalid;
|
|
s32 x14_roleIndex = -1;
|
|
s32 x18_captainPriority = 0;
|
|
zeus::CVector3f x1c_position;
|
|
|
|
public:
|
|
CTeamAiRole(TUniqueId ownerId, ETeamAiRole a = ETeamAiRole::Invalid, ETeamAiRole b = ETeamAiRole::Invalid,
|
|
ETeamAiRole c = ETeamAiRole::Invalid)
|
|
: x0_ownerId(ownerId), x4_roleA(a), x8_roleB(b), xc_roleC(c) {}
|
|
TUniqueId GetOwnerId() const { return x0_ownerId; }
|
|
bool HasTeamAiRole() const { return false; }
|
|
ETeamAiRole GetTeamAiRole() const { return x10_curRole; }
|
|
void SetTeamAiRole(ETeamAiRole role) { x10_curRole = role; }
|
|
s32 GetRoleIndex() const { return x14_roleIndex; }
|
|
void SetRoleIndex(s32 idx) { x14_roleIndex = idx; }
|
|
const zeus::CVector3f& GetTeamPosition() const { return x1c_position; }
|
|
void SetTeamPosition(const zeus::CVector3f& pos) { x1c_position = pos; }
|
|
bool operator<(const CTeamAiRole& other) const { return x0_ownerId < other.x0_ownerId; }
|
|
};
|
|
|
|
class CTeamAiData {
|
|
friend class CTeamAiMgr;
|
|
u32 x0_aiCount;
|
|
u32 x4_meleeCount;
|
|
u32 x8_rangedCount;
|
|
u32 xc_unknownCount;
|
|
u32 x10_maxMeleeAttackerCount;
|
|
u32 x14_maxRangedAttackerCount;
|
|
u32 x18_positionMode;
|
|
float x1c_meleeTimeInterval;
|
|
float x20_rangedTimeInterval;
|
|
|
|
public:
|
|
CTeamAiData(CInputStream& in, s32 propCount);
|
|
};
|
|
|
|
class CTeamAiMgr : public CEntity {
|
|
public:
|
|
enum class EAttackType { Melee, Ranged };
|
|
|
|
private:
|
|
CTeamAiData x34_data;
|
|
std::vector<CTeamAiRole> x58_roles;
|
|
std::vector<TUniqueId> x68_meleeAttackers;
|
|
std::vector<TUniqueId> x78_rangedAttackers;
|
|
float x88_timeDirty = 0.0f;
|
|
TUniqueId x8c_teamCaptainId = kInvalidUniqueId;
|
|
float x90_timeSinceMelee = 0.0f;
|
|
float x94_timeSinceRanged = 0.0f;
|
|
|
|
void UpdateTeamCaptain();
|
|
bool ShouldUpdateRoles(float dt);
|
|
void ResetRoles(CStateManager& mgr);
|
|
void AssignRoles(CTeamAiRole::ETeamAiRole role, s32 count);
|
|
void UpdateRoles(CStateManager& mgr);
|
|
void SpacingSort(CStateManager& mgr, const zeus::CVector3f& pos);
|
|
void PositionTeam(CStateManager& mgr);
|
|
|
|
public:
|
|
CTeamAiMgr(TUniqueId uid, std::string_view name, const CEntityInfo& info, const CTeamAiData& data);
|
|
|
|
void Accept(IVisitor&) override;
|
|
void Think(float dt, CStateManager& mgr) override;
|
|
void AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId objId, CStateManager& mgr) override;
|
|
CTeamAiRole* GetTeamAiRole(TUniqueId aiId);
|
|
bool IsPartOfTeam(TUniqueId aiId) const;
|
|
bool HasTeamAiRole(TUniqueId aiId) const;
|
|
bool AssignTeamAiRole(const CAi& ai, CTeamAiRole::ETeamAiRole roleA, CTeamAiRole::ETeamAiRole roleB,
|
|
CTeamAiRole::ETeamAiRole roleC);
|
|
void RemoveTeamAiRole(TUniqueId aiId);
|
|
void ClearTeamAiRole(TUniqueId aiId);
|
|
s32 GetNumAssignedOfRole(CTeamAiRole::ETeamAiRole role) const;
|
|
s32 GetNumAssignedAiRoles() const;
|
|
|
|
bool IsMeleeAttacker(TUniqueId aiId) const;
|
|
bool CanAcceptMeleeAttacker(TUniqueId aiId) const;
|
|
bool AddMeleeAttacker(TUniqueId aiId);
|
|
void RemoveMeleeAttacker(TUniqueId aiId);
|
|
bool IsRangedAttacker(TUniqueId aiId) const;
|
|
bool CanAcceptRangedAttacker(TUniqueId aiId) const;
|
|
bool AddRangedAttacker(TUniqueId aiId);
|
|
void RemoveRangedAttacker(TUniqueId aiId);
|
|
|
|
bool HasMeleeAttackers() const { return !x68_meleeAttackers.empty(); }
|
|
const std::vector<TUniqueId>& GetMeleeAttackers() const { return x68_meleeAttackers; }
|
|
bool HasRangedAttackers() const { return !x78_rangedAttackers.empty(); }
|
|
const std::vector<TUniqueId>& GetRangedAttackers() const { return x78_rangedAttackers; }
|
|
s32 GetNumRoles() const { return x58_roles.size(); }
|
|
const std::vector<CTeamAiRole>& GetRoles() const { return x58_roles; }
|
|
s32 GetMaxMeleeAttackerCount() const { return x34_data.x10_maxMeleeAttackerCount; }
|
|
s32 GetMaxRangedAttackerCount() const { return x34_data.x14_maxRangedAttackerCount; }
|
|
|
|
static CTeamAiRole* GetTeamAiRole(CStateManager& mgr, TUniqueId mgrId, TUniqueId aiId);
|
|
static void ResetTeamAiRole(EAttackType type, CStateManager& mgr, TUniqueId mgrId, TUniqueId aiId, bool clearRole);
|
|
static bool CanAcceptAttacker(EAttackType type, CStateManager& mgr, TUniqueId mgrId, TUniqueId aiId);
|
|
static bool AddAttacker(EAttackType type, CStateManager& mgr, TUniqueId mgrId, TUniqueId aiId);
|
|
|
|
static TUniqueId GetTeamAiMgr(const CAi& ai, const CStateManager& mgr);
|
|
};
|
|
} // namespace metaforce
|