DragonECS/src/Utils/WorldMetaStorage.cs

using DCFApixels.DragonECS.Utils;
using System;
using System.Collections.Generic;
using System.Runtime.CompilerServices;

namespace DCFApixels.DragonECS
{
    //TODO этот класс требует переработки, изначально такая конструкция имела хорошую производительность, но сейчас он слишком раздулся
    internal static class WorldMetaStorage 
    {
        private static int _tokenCount = 0;
        private static List<ResizerBase> _resizers = new List<ResizerBase>();
        private static WorldTypeMeta[] _metas = new WorldTypeMeta[0];
        private static Dictionary<Type, int> _worldIds = new Dictionary<Type, int>();
        private static class WorldIndex<TWorldArchetype>
        {
            public static int id = GetWorldID(typeof(TWorldArchetype));
        }
        private static int GetToken()
        {
            WorldTypeMeta meta = new WorldTypeMeta();
            meta.id = _tokenCount;
            Array.Resize(ref _metas, ++_tokenCount);
            _metas[_tokenCount - 1] = meta;

            foreach (var item in _resizers)
                item.Resize(_tokenCount);
            return _tokenCount - 1;
        }
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetWorldID(Type archetype)
        {
            if (!_worldIds.TryGetValue(archetype, out int id))
            {
                id = GetToken();
                _worldIds.Add(archetype, id);
            }
            return id;
        }
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetWorldID<TWorldArchetype>() => WorldIndex<TWorldArchetype>.id;
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetComponentID<T>(int worldID) => Component<T>.Get(worldID);
        public static int GetComponentID(Type type, int worldID) => _metas[worldID].GetComponentID(type);
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetPoolID<T>(int worldID) => Pool<T>.Get(worldID);
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetSubjectID<T>(int worldID) => Subject<T>.Get(worldID);
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetExecutorID<T>(int worldID) => Executor<T>.Get(worldID);
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        public static int GetWorldComponentID<T>(int worldID) => WorldComponent<T>.Get(worldID);
        public static bool IsComponentTypeDeclared(int worldID, Type type) => _metas[worldID].IsDeclaredType(type);
        public static Type GetComponentType(int worldID, int componentID) => _metas[worldID].GetComponentType(componentID);


        private abstract class ResizerBase
        {
            public abstract Type Type { get; }
            public abstract int[] IDS { get; }
            public abstract void Resize(int size);
        }

        #region Containers
        public static class PoolComponentIdArrays
        {
            private static Dictionary<Type, int[]> _componentTypeArrayPairs = new Dictionary<Type, int[]>();

            public static int[] GetIdsArray(Type type)
            {
                int targetSize = _tokenCount;
                if (!_componentTypeArrayPairs.TryGetValue(type, out int[] result))
                {
                    result = new int[targetSize];
                    for (int i = 0; i < result.Length; i++)
                        result[i] = -1;
                    _componentTypeArrayPairs.Add(type, result);
                }
                else
                {
                    if(result.Length < targetSize)
                    {
                        int oldSize = result.Length;
                        Array.Resize(ref result, targetSize);
                        ArrayUtility.Fill(result, -1, oldSize, targetSize);
                        _componentTypeArrayPairs[type] = result;
                    }
                }

                return result;
            }

            public static int GetComponentID(Type type, int token)
            {
                GetIdsArray(type);
                ref int id = ref _componentTypeArrayPairs[type][token];
                if (id < 0)
                {
                    var meta = _metas[token];
                    id = meta.componentCount++;
                    meta.AddType(id, type);
                }
                return id;
            }
        }
        private static class Pool<T>
        {
            public static int[] ids;
            private static Type componentType = typeof(T).GetGenericArguments()[0];
            static Pool()
            {
                ids = PoolComponentIdArrays.GetIdsArray(componentType);
                _resizers.Add(new Resizer());
            }
            [MethodImpl(MethodImplOptions.AggressiveInlining)]
            public static int Get(int token)
            {
                ref int id = ref ids[token];
                if (id < 0)
                {
                    id = PoolComponentIdArrays.GetComponentID(componentType, token);
                }
                return id;
            }
            private sealed class Resizer : ResizerBase
            {
                public override Type Type => typeof(T);
                public override int[] IDS => ids;
                public override void Resize(int size)
                {
                    ids = PoolComponentIdArrays.GetIdsArray(componentType);
                }
            }
        }
        private static class Component<T>
        {
            public static int[] ids;
            static Component()
            {
                ids = PoolComponentIdArrays.GetIdsArray(typeof(T));
                _resizers.Add(new Resizer());
            }
            [MethodImpl(MethodImplOptions.AggressiveInlining)]
            public static int Get(int token)
            {
                ref int id = ref ids[token];
                if (id < 0)
                {
                    id = PoolComponentIdArrays.GetComponentID(typeof(T), token);
                }
                return id;
            }
            private sealed class Resizer : ResizerBase
            {
                public override Type Type => typeof(T);
                public override int[] IDS => ids;
                public override void Resize(int size)
                {
                    ids = PoolComponentIdArrays.GetIdsArray(typeof(T));
                }
            }
        }
        private static class Subject<T>
        {
            public static int[] ids;
            static Subject()
            {
                ids = new int[_tokenCount];
                for (int i = 0; i < ids.Length; i++)
                    ids[i] = -1;
                _resizers.Add(new Resizer());
            }
            [MethodImpl(MethodImplOptions.AggressiveInlining)]
            public static int Get(int token)
            {
                ref int id = ref ids[token];
                if (id < 0)
                    id = _metas[token].subjectsCount++;
                return id;
            }
            private sealed class Resizer : ResizerBase
            {
                public override Type Type => typeof(T);
                public override int[] IDS => ids;
                public override void Resize(int size)
                {
                    int oldSize = ids.Length;
                    Array.Resize(ref ids, size);
                    ArrayUtility.Fill(ids, -1, oldSize, size);
                }
            }
        }
        private static class Executor<T>
        {
            public static int[] ids;
            static Executor()
            {
                ids = new int[_tokenCount];
                for (int i = 0; i < ids.Length; i++)
                    ids[i] = -1;
                _resizers.Add(new Resizer());
            }
            [MethodImpl(MethodImplOptions.AggressiveInlining)]
            public static int Get(int token)
            {
                ref int id = ref ids[token];
                if (id < 0)
                    id = _metas[token].executorsCount++;
                return id;
            }
            private sealed class Resizer : ResizerBase
            {
                public override Type Type => typeof(T);
                public override int[] IDS => ids;
                public override void Resize(int size)
                {
                    int oldSize = ids.Length;
                    Array.Resize(ref ids, size);
                    ArrayUtility.Fill(ids, -1, oldSize, size);
                }
            }
        }
        private static class WorldComponent<T>
        {
            public static int[] ids;
            static WorldComponent()
            {
                ids = new int[_tokenCount];
                for (int i = 0; i < ids.Length; i++)
                    ids[i] = -1;
                _resizers.Add(new Resizer());
            }
            [MethodImpl(MethodImplOptions.AggressiveInlining)]
            public static int Get(int token)
            {
                ref int id = ref ids[token];
                if (id < 0)
                    id = _metas[token].worldComponentCount++;
                return id;
            }
            private sealed class Resizer : ResizerBase
            {
                public override Type Type => typeof(T);
                public override int[] IDS => ids;
                public override void Resize(int size)
                {
                    int oldSize = ids.Length;
                    Array.Resize(ref ids, size);
                    ArrayUtility.Fill(ids, -1, oldSize, size);
                }
            }
        }
        #endregion

        public struct XXX : IEcsComponent { }
        private class WorldTypeMeta
        {
            public int id;
            public int componentCount;
            public int subjectsCount;
            public int executorsCount;
            public int worldComponentCount;
            private Type[] _types;
            private Dictionary<Type, int> _declaredComponentTypes;
            public void AddType(int id, Type type)
            {
                if (_types.Length <= id)
                    Array.Resize(ref _types, id + 10);
                _types[id] = type;

                _declaredComponentTypes.Add(type, id);
            }
            public Type GetComponentType(int componentID) => _types[componentID];
            public bool IsDeclaredType(Type type) => _declaredComponentTypes.ContainsKey(type);
            public int GetComponentID(Type type)
            {
                return PoolComponentIdArrays.GetComponentID(type, id);
            }
            public WorldTypeMeta()
            {
                _types = new Type[10];
                _declaredComponentTypes = new Dictionary<Type, int>();
            }
        }
    }
}