#include "CObjectList.hpp"

namespace urde
{

CObjectList::CObjectList(EGameObjectList listEnum)
: x2004_listEnum(listEnum)
{}

void CObjectList::AddObject(CEntity& entity)
{
    if (IsQualified(entity))
    {
        if (x2008_firstId != -1)
            x0_list[x2008_firstId].prev = entity.GetUniqueId().Value();
        s16 prevFirst = x2008_firstId;
        x2008_firstId = entity.GetUniqueId().Value();
        SObjectListEntry& newEnt = x0_list[x2008_firstId];
        newEnt.entity = &entity;
        newEnt.next = prevFirst;
        newEnt.prev = -1;
        ++x200a_count;
    }
}

void CObjectList::RemoveObject(TUniqueId uid)
{
    SObjectListEntry& ent = x0_list[uid.Value()];
    if (!ent.entity || ent.entity->GetUniqueId() != uid)
        return;
    if (uid.Value() == x2008_firstId)
    {
        x2008_firstId = ent.next;
        if (ent.next != -1)
            x0_list[ent.next].prev = -1;
    }
    else
    {
        x0_list[ent.prev].next = ent.next;
        if (ent.next != -1)
            x0_list[ent.next].prev = ent.prev;
    }
    ent.entity = nullptr;
    ent.next = -1;
    ent.prev = -1;
    --x200a_count;
}

const CEntity* CObjectList::operator[](size_t i) const
{
    const SObjectListEntry& ent = x0_list[i];
    if (!ent.entity || ent.entity->x30_26_scriptingBlocked)
        return nullptr;
    return ent.entity;
}

CEntity* CObjectList::operator[](size_t i)
{
    SObjectListEntry& ent = x0_list[i];
    if (!ent.entity || ent.entity->x30_26_scriptingBlocked)
        return nullptr;
    return ent.entity;
}

const CEntity* CObjectList::GetObjectById(TUniqueId uid) const
{
    if (uid == kInvalidUniqueId)
        return nullptr;
    const SObjectListEntry& ent = x0_list[uid.Value()];
    if (!ent.entity || ent.entity->x30_26_scriptingBlocked)
        return nullptr;
    return ent.entity;
}

CEntity* CObjectList::GetObjectById(TUniqueId uid)
{
    if (uid == kInvalidUniqueId)
        return nullptr;
    SObjectListEntry& ent = x0_list[uid.Value()];
    if (!ent.entity || ent.entity->x30_26_scriptingBlocked)
        return nullptr;
    return ent.entity;
}

const CEntity* CObjectList::GetValidObjectById(TUniqueId uid) const
{
    if (uid == kInvalidUniqueId)
        return nullptr;
    const SObjectListEntry& ent = x0_list[uid.Value()];
    if (!ent.entity)
        return nullptr;
    if (ent.entity->GetUniqueId() != uid)
        return nullptr;
    return ent.entity;
}

CEntity* CObjectList::GetValidObjectById(TUniqueId uid)
{
    if (uid == kInvalidUniqueId)
        return nullptr;
    SObjectListEntry& ent = x0_list[uid.Value()];
    if (!ent.entity)
        return nullptr;
    if (ent.entity->GetUniqueId() != uid)
        return nullptr;
    return ent.entity;
}

bool CObjectList::IsQualified(const CEntity&) {return true;}

}