#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; }
};

}