encompass-cs/encompass-cs/EntityManager.cs

using System.Linq;
using System;
using System.Collections.Generic;
using Encompass.Exceptions;

namespace Encompass
{
    internal class EntityManager
    {
        private readonly Dictionary<Guid, Entity> IDToEntity = new Dictionary<Guid, Entity>();

        private readonly HashSet<Guid> entitiesMarkedForDestroy = new HashSet<Guid>();

        private readonly ComponentManager componentManager;
        private readonly ComponentMessageManager componentMessageManager;

        public EntityManager(ComponentManager componentManager, ComponentMessageManager componentMessageManager)
        {
            this.componentManager = componentManager;
            this.componentMessageManager = componentMessageManager;
        }

        public Entity CreateEntity()
        {
            var id = NextID();
            var entity = new Entity(id);
            IDToEntity[id] = entity;
            componentManager.RegisterEntity(id);
            componentMessageManager.RegisterEntity(entity);
            return entity;
        }

        public bool EntityExists(Guid id)
        {
            return IDToEntity.ContainsKey(id);
        }

        public Entity GetEntity(Guid id)
        {
            if (IDToEntity.ContainsKey(id))
            {
                return IDToEntity[id];
            }
            else
            {
                throw new EntityNotFoundException("Entity with ID {0} does not exist.", id);
            }
        }

        public void MarkForDestroy(Guid entityID)
        {
            entitiesMarkedForDestroy.Add(entityID);
        }

        public void DestroyMarkedEntities()
        {
            foreach (var entityID in entitiesMarkedForDestroy)
            {
                componentMessageManager.RegisterDestroyedEntity(GetEntity(entityID));
                componentManager.MarkAllComponentsOnEntityForRemoval(entityID);
                IDToEntity.Remove(entityID);
                componentManager.RegisterDestroyedEntity(entityID);
            }

            entitiesMarkedForDestroy.Clear();
        }

        private Guid NextID()
        {
            return Guid.NewGuid();
        }
    }
}