You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
3.0 KiB
C++
101 lines
3.0 KiB
C++
/*
|
|
* Copyright (c) Contributors to the Open 3D Engine Project.
|
|
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 OR MIT
|
|
*
|
|
*/
|
|
|
|
#include <Source/NetworkEntity/NetworkEntityTracker.h>
|
|
#include <Multiplayer/NetworkEntity/NetworkEntityHandle.h>
|
|
#include <Multiplayer/Components/NetBindComponent.h>
|
|
#include <AzCore/Console/IConsole.h>
|
|
#include <AzCore/Console/ILogger.h>
|
|
|
|
namespace Multiplayer
|
|
{
|
|
void NetworkEntityTracker::Add(NetEntityId netEntityId, AZ::Entity* entity)
|
|
{
|
|
++m_addChangeDirty;
|
|
AZ_Assert(m_entityMap.end() == m_entityMap.find(netEntityId), "Attempting to add the same entity to the entity map multiple times");
|
|
m_entityMap[netEntityId] = entity;
|
|
m_netEntityIdMap[entity->GetId()] = netEntityId;
|
|
}
|
|
|
|
void NetworkEntityTracker::RegisterNetBindComponent(AZ::Entity* entity, NetBindComponent* component)
|
|
{
|
|
m_netBindingMap[entity] = component;
|
|
}
|
|
|
|
void NetworkEntityTracker::UnregisterNetBindComponent(NetBindComponent* component)
|
|
{
|
|
m_netBindingMap.erase(component->GetEntity());
|
|
}
|
|
|
|
NetworkEntityHandle NetworkEntityTracker::Get(NetEntityId netEntityId)
|
|
{
|
|
AZ::Entity* entity = GetRaw(netEntityId);
|
|
return NetworkEntityHandle(entity, this);
|
|
}
|
|
|
|
ConstNetworkEntityHandle NetworkEntityTracker::Get(NetEntityId netEntityId) const
|
|
{
|
|
AZ::Entity* entity = GetRaw(netEntityId);
|
|
return ConstNetworkEntityHandle(entity, this);
|
|
}
|
|
|
|
NetEntityId NetworkEntityTracker::Get(const AZ::EntityId& entityId) const
|
|
{
|
|
auto found = m_netEntityIdMap.find(entityId);
|
|
if (found != m_netEntityIdMap.end())
|
|
{
|
|
return found->second;
|
|
}
|
|
return Multiplayer::InvalidNetEntityId;
|
|
}
|
|
|
|
bool NetworkEntityTracker::Exists(NetEntityId netEntityId) const
|
|
{
|
|
return (m_entityMap.find(netEntityId) != m_entityMap.end());
|
|
}
|
|
|
|
AZ::Entity* NetworkEntityTracker::GetRaw(NetEntityId netEntityId) const
|
|
{
|
|
auto found = m_entityMap.find(netEntityId);
|
|
if (found != m_entityMap.end())
|
|
{
|
|
return found->second;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
void NetworkEntityTracker::erase(NetEntityId netEntityId)
|
|
{
|
|
++m_deleteChangeDirty;
|
|
|
|
auto found = m_entityMap.find(netEntityId);
|
|
if (found != m_entityMap.end())
|
|
{
|
|
m_netEntityIdMap.erase(found->second->GetId());
|
|
m_entityMap.erase(found);
|
|
}
|
|
}
|
|
|
|
NetworkEntityTracker::EntityMap::iterator NetworkEntityTracker::erase(EntityMap::iterator iter)
|
|
{
|
|
++m_deleteChangeDirty;
|
|
if (iter != m_entityMap.end())
|
|
{
|
|
m_netEntityIdMap.erase(iter->second->GetId());
|
|
}
|
|
return m_entityMap.erase(iter);
|
|
}
|
|
|
|
AZ::Entity *NetworkEntityTracker::Move(EntityMap::iterator iter)
|
|
{
|
|
AZ::Entity *ptr = iter->second;
|
|
erase(iter);
|
|
return ptr;
|
|
}
|
|
}
|