Code/CryEngine/CrySchematyc2/BaseEnv/Utils/BaseEnv_EntityMap.cpp

   1 // Copyright 2001-2019 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "BaseEnv/Utils/BaseEnv_EntityMap.h"
   5
   6 #include <CryEntitySystem/IEntitySystem.h>
   7
   8 namespace SchematycBaseEnv
   9 {
  10         void CEntityMap::MarkEntityForDestruction(EntityId entityId)
  11         {
  12                 m_entitiesMarkedForDestruction.push_back(entityId);
  13         }
  14
  15         void CEntityMap::DestroyMarkedEntites()
  16         {
  17                 EntityIds entitiesMarkedForDestruction = m_entitiesMarkedForDestruction;
  18                 m_entitiesMarkedForDestruction.clear();
  19                 for(EntityId entityId : entitiesMarkedForDestruction)
  20                 {
  21                         gEnv->pEntitySystem->RemoveEntity(entityId);
  22                 }
  23         }
  24
  25         void CEntityMap::AddObject(EntityId entityId, const Schematyc2::ObjectId& objectId)
  26         {
  27                 m_entityObjectIds.insert(EntityObjectIds::value_type(entityId, objectId));
  28         }
  29
  30         void CEntityMap::RemoveObject(EntityId entityId)
  31         {
  32                 m_entityObjectIds.erase(entityId);
  33         }
  34
  35         Schematyc2::ObjectId CEntityMap::FindObjectId(EntityId entityId) const
  36         {
  37                 EntityObjectIds::const_iterator itObjectId = m_entityObjectIds.find(entityId);
  38                 return itObjectId != m_entityObjectIds.end() ? itObjectId->second : Schematyc2::ObjectId();
  39         }
  40
  41         Schematyc2::IObject* CEntityMap::FindObject(EntityId entityId) const
  42         {
  43                 return gEnv->pSchematyc2->GetObjectManager().GetObjectById(FindObjectId(entityId));
  44         }
  45
  46         void CEntityMap::VisitEntities(const EntityMapVisitor& visitor)
  47         {
  48                 CRY_ASSERT(visitor);
  49                 if(visitor)
  50                 {
  51                         for(EntityObjectIds::value_type& entityObjectId : m_entityObjectIds)
  52                         {
  53                                 visitor(entityObjectId.first);
  54                         }
  55                 }
  56         }
  57 }