mirror of https://github.com/AxioDL/metaforce.git
61 lines
1.7 KiB
C++
61 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <map>
|
|
#include "CSegId.hpp"
|
|
|
|
namespace urde {
|
|
|
|
template <class T>
|
|
class TSegIdMap {
|
|
CSegId x0_boneCount = 0;
|
|
CSegId x1_capacity = 0;
|
|
u32 x4_maxCapacity = 100;
|
|
std::pair<CSegId, CSegId> x8_indirectionMap[100];
|
|
std::unique_ptr<T[]> xd0_bones;
|
|
CSegId xd4_curPrevBone = 0;
|
|
|
|
public:
|
|
TSegIdMap(const CSegId& capacity) : x1_capacity(capacity), xd0_bones(new T[capacity]) {}
|
|
T& operator[](const CSegId& id) { return SetElement(id); }
|
|
const T& operator[](const CSegId& id) const { return xd0_bones[x8_indirectionMap[id].second]; }
|
|
T& SetElement(const CSegId& id, T&& obj) {
|
|
size_t idx;
|
|
if (x8_indirectionMap[id].first == 0xff) {
|
|
x8_indirectionMap[id].first = xd4_curPrevBone;
|
|
x8_indirectionMap[id].second = x0_boneCount;
|
|
xd4_curPrevBone = id;
|
|
idx = x0_boneCount;
|
|
++x0_boneCount;
|
|
} else
|
|
idx = x8_indirectionMap[id].second;
|
|
xd0_bones[idx] = std::move(obj);
|
|
return xd0_bones[idx];
|
|
}
|
|
T& SetElement(const CSegId& id) {
|
|
size_t idx;
|
|
if (x8_indirectionMap[id].first == 0xff) {
|
|
x8_indirectionMap[id].first = xd4_curPrevBone;
|
|
x8_indirectionMap[id].second = x0_boneCount;
|
|
xd4_curPrevBone = id;
|
|
idx = x0_boneCount;
|
|
++x0_boneCount;
|
|
} else
|
|
idx = x8_indirectionMap[id].second;
|
|
return xd0_bones[idx];
|
|
}
|
|
void DelElement(const CSegId& id) {
|
|
if (x8_indirectionMap[id].first != 0xff) {
|
|
if (id == xd4_curPrevBone)
|
|
xd4_curPrevBone = x8_indirectionMap[id].first;
|
|
x8_indirectionMap[id].first = 0xff;
|
|
x8_indirectionMap[id].second = 0xff;
|
|
--x0_boneCount;
|
|
}
|
|
}
|
|
bool HasElement(const CSegId& id) const { return x8_indirectionMap[id].first != 0xff; }
|
|
|
|
u32 GetCapacity() const { return x1_capacity; }
|
|
};
|
|
|
|
} // namespace urde
|