metaforce/Runtime/CObjectList.hpp

95 lines
2.9 KiB
C++

#pragma once
#include <array>
#include "Runtime/RetroTypes.hpp"
#include "Runtime/World/CEntity.hpp"
namespace metaforce {
enum class EGameObjectList {
Invalid = -1,
All,
Actor,
PhysicsActor,
GameCamera,
GameLight,
ListeningAi,
AiWaypoint,
PlatformAndDoor,
};
class CObjectList {
friend class CGameArea;
struct SObjectListEntry {
CEntity* entity = nullptr;
s16 next = -1;
s16 prev = -1;
};
std::array<SObjectListEntry, 1024> x0_list; // was an rstl::prereserved_vector
EGameObjectList x2004_listEnum;
s16 x2008_firstId = -1;
u16 x200a_count = 0;
public:
class iterator {
friend class CObjectList;
CObjectList& m_list;
s16 m_id;
iterator(CObjectList& list, s16 id) : m_list(list), m_id(id) {}
public:
iterator& operator++() {
m_id = m_list.GetNextObjectIndex(m_id);
return *this;
}
bool operator==(const iterator& other) const { return m_id == other.m_id; }
bool operator!=(const iterator& other) const { return !operator==(other); }
CEntity* operator*() const { return m_list.GetObjectByIndex(m_id); }
};
class const_iterator {
friend class CObjectList;
const CObjectList& m_list;
s16 m_id;
const_iterator(const CObjectList& list, s16 id) : m_list(list), m_id(id) {}
public:
const_iterator& operator++() {
m_id = m_list.GetNextObjectIndex(m_id);
return *this;
}
bool operator==(const iterator& other) const { return m_id == other.m_id; }
bool operator!=(const iterator& other) const { return !operator==(other); }
const CEntity* operator*() const { return m_list.GetObjectByIndex(m_id); }
};
[[nodiscard]] iterator begin() { return iterator(*this, x2008_firstId); }
[[nodiscard]] iterator end() { return iterator(*this, -1); }
[[nodiscard]] const_iterator begin() const { return const_iterator(*this, x2008_firstId); }
[[nodiscard]] const_iterator end() const { return const_iterator(*this, -1); }
[[nodiscard]] const_iterator cbegin() const { return begin(); }
[[nodiscard]] const_iterator cend() const { return end(); }
explicit CObjectList(EGameObjectList listEnum);
virtual ~CObjectList() = default;
void AddObject(CEntity& entity);
void RemoveObject(TUniqueId uid);
const CEntity* operator[](size_t i) const;
CEntity* operator[](size_t i);
const CEntity* GetObjectById(TUniqueId uid) const;
const CEntity* GetObjectByIndex(s16 index) const { return x0_list[index].entity; }
CEntity* GetObjectByIndex(s16 index) { return x0_list[index].entity; }
CEntity* GetObjectById(TUniqueId uid);
const CEntity* GetValidObjectById(TUniqueId uid) const;
CEntity* GetValidObjectById(TUniqueId uid);
s16 GetFirstObjectIndex() const { return x2008_firstId; }
s16 GetNextObjectIndex(s16 prev) const { return x0_list[prev].next; }
virtual bool IsQualified(const CEntity&) const;
u16 size() const { return x200a_count; }
};
} // namespace metaforce