GhostEngine/Ghost.Entities/Systems/SystemStorage.cs

using System.Runtime.CompilerServices;

namespace Ghost.Entities.Systems;

[SkipLocalsInit]
public struct SystemStorage
{
    private readonly List<Type> _systems = new();
    private readonly List<ISystem> _executionList = new();

    private readonly World _world;

    public event Action<Type>? SystemAdded;
    public event Action<Type>? SystemRemoved;

    internal SystemStorage(World world)
    {
        _world = world;
    }

    public readonly void AddSystem(Type systemType)
    {
        _systems.Add(systemType);
        SystemAdded?.Invoke(systemType);
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public readonly void AddSystem<T>()
        where T : ISystem, new()
    {
        AddSystem(typeof(T));
    }

    public readonly void RemoveSystem(Type systemType)
    {
        _systems.Remove(systemType);
        SystemRemoved?.Invoke(systemType);
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public readonly void RemoveSystem<T>()
        where T : ISystem, new()
    {
        RemoveSystem(typeof(T));
    }

    internal void CreateSystems()
    {
        var builder = new SystemDependencyBuilder(_systems);
        builder.BuildDependencyGraph();
        var executionOrder = builder.BuildExecutionOrder();

        var state = new SystemState()
        {
            World = _world,
        };

        foreach (var systemType in executionOrder)
        {
            var system = (ISystem?)Activator.CreateInstance(systemType) ?? throw new InvalidOperationException($"Failed to create instance of system type {systemType.Name}.");

            _executionList.Add(system);
            system.OnCreate(in state);
        }
    }

    internal void UpdateSystems()
    {
        var state = new SystemState()
        {
            World = _world,
        };

        foreach (var system in _executionList)
        {
            system.OnUpdate(in state);
        }
    }

    internal void Dispose()
    {
        var state = new SystemState()
        {
            World = _world,
        };

        foreach (var system in _executionList)
        {
            system.OnDestroy(in state);
        }

        _systems.Clear();
        _executionList.Clear();
    }
}