MonoGame.Extended/Source/MonoGame.Extended.Entities/EntityWorld.cs

using Microsoft.Xna.Framework;
using MonoGame.Extended.Collections;
using MonoGame.Extended.Entities.Systems;

namespace MonoGame.Extended.Entities
{
    public class EntityWorld : SimpleDrawableGameComponent
    {
        private readonly Bag<IUpdateSystem> _updateSystems;
        private readonly Bag<IDrawSystem> _drawSystems;

        public EntityWorld()
        {
            _updateSystems = new Bag<IUpdateSystem>();
            _drawSystems = new Bag<IDrawSystem>();

            RegisterSystem(ComponentManager = new ComponentManager());
            RegisterSystem(EntityManager = new EntityManager(ComponentManager));
        }

        public override void Dispose()
        {
            foreach (var updateSystem in _updateSystems)
                updateSystem.Dispose();

            foreach (var drawSystem in _drawSystems)
                drawSystem.Dispose();

            base.Dispose();
        }

        internal EntityManager EntityManager { get; }
        internal ComponentManager ComponentManager { get; }
        
        public Bag<Entity> AllEntities => EntityManager.Entities;

        // TODO: Move this to world configuration
        public void RegisterSystem(ISystem system)
        {
            // ReSharper disable once ConvertIfStatementToSwitchStatement
            if (system is IUpdateSystem updateSystem)
                _updateSystems.Add(updateSystem);

            if (system is IDrawSystem drawSystem)
                _drawSystems.Add(drawSystem);

            system.Initialize(this);
        }

        public Entity GetEntity(int entityId)
        {
            return AllEntities[entityId];
        }

        public Entity CreateEntity()
        {
            return EntityManager.CreateEntity();
        }

        public void DestroyEntity(int entityId)
        {
            EntityManager.DestroyEntity(entityId);
        }

        public void DestroyEntity(Entity entity)
        {
            EntityManager.DestroyEntity(entity);
        }
        
        public override void Update(GameTime gameTime)
        {
            foreach (var system in _updateSystems)
                system.Update(gameTime);
        }

        public override void Draw(GameTime gameTime)
        {
            foreach (var system in _drawSystems)
                system.Draw(gameTime);
        }
    }
}