Svelto.ECS/com.sebaslab.svelto.ecs/Core/Streams/EntitiesStreams.cs

using System;
using Svelto.DataStructures;

namespace Svelto.ECS
{
    /// <summary>
    ///     I eventually realised that, with the ECS design, no form of engines (systems) communication other
    ///     than polling entity components is effective.
    ///     The only purpose of this publisher/consumer model is to let two enginesroots communicate with each other
    ///     through a thread safe ring buffer.
    ///     The engines root A publishes entities.
    ///     The engines root B can consume those entities at any time, as they will be a copy of the original
    ///     entities and won't point directly to the database of the engines root A
    /// </summary>
    struct EntitiesStreams : IDisposable
    {
        internal Consumer<T> GenerateConsumer<T>(string name, uint capacity)
            where T : unmanaged, IEntityComponent
        {
            if (_streams.ContainsKey(TypeRefWrapper<T>.wrapper) == false)
                _streams[TypeRefWrapper<T>.wrapper] = new EntityStream<T>();

            return (_streams[TypeRefWrapper<T>.wrapper] as EntityStream<T>).GenerateConsumer(name, capacity);
        }

        public Consumer<T> GenerateConsumer<T>(ExclusiveGroupStruct group, string name, uint capacity)
            where T : unmanaged, IEntityComponent
        {
            if (_streams.ContainsKey(TypeRefWrapper<T>.wrapper) == false)
                _streams[TypeRefWrapper<T>.wrapper] = new EntityStream<T>();

            var typeSafeStream = (EntityStream<T>) _streams[TypeRefWrapper<T>.wrapper];
            return typeSafeStream.GenerateConsumer(group, name, capacity);
        }

        internal void PublishEntity<T>(ref T entity, EGID egid) where T : unmanaged, IEntityComponent
        {
            if (_streams.TryGetValue(TypeRefWrapper<T>.wrapper, out var typeSafeStream))
                (typeSafeStream as EntityStream<T>).PublishEntity(ref entity, egid);
            else
                Console.LogDebug("No Consumers are waiting for this entity to change ", typeof(T));
        }

        public void Dispose()
        {
            foreach (var stream in _streams)
                stream.value.Dispose();
        }

        public static EntitiesStreams Create()
        {
            var stream = new EntitiesStreams();
            stream._streams = FasterDictionary<RefWrapperType, ITypeSafeStream>.Construct();

            return stream;
        }

        FasterDictionary<RefWrapperType, ITypeSafeStream> _streams;
    }
}