[Entities] Optimized ComponentVersionTracker;

Author: LittleCodingFox

Engine/Staple.Core/Entities/ComponentVersionTracker.cs

@@ -1,33 +1,27 @@
 using System;
-using System.Collections.Generic;
 
 namespace Staple;
 
-public class ComponentVersionTracker
+public class ComponentVersionTracker<T> where T: IComponent, IComponentVersion
 {
-    private readonly Dictionary<int, ulong> versions = [];
+    private ulong[] versions = [1024];
 
-    public void Clear()
-    {
-        versions.Clear();
-    }
-
-    public bool ShouldUpdateComponent<T>(Entity entity, in T component) where T : IComponent, IComponentVersion
+    public bool ShouldUpdateComponent(Entity entity, in T component)
     {
         if (component == null)
         {
             throw new ArgumentNullException(nameof(component), "Component can't be null");
         }
 
-        var hashCode = HashCode.Combine(entity, component.GetType().FullName.GetHashCode());
-
-        if (versions.TryGetValue(hashCode, out var version) == false)
+        if(versions.Length < entity.Identifier.ID)
         {
-            versions.Add(hashCode, component.Version);
-
-            return true;
+            Array.Resize(ref versions, entity.Identifier.ID);
         }
 
+        var version = versions[entity.Identifier.ID - 1];
+
+        versions[entity.Identifier.ID - 1] = component.Version;
+
         return version != component.Version;
     }
 }

Engine/Staple.Core/Rendering/Animation/SkinnedMeshRenderSystem.cs

@@ -30,7 +30,7 @@ private struct RenderInfo
 
     private readonly SceneQuery<SkinnedMeshInstance, Transform> instances = new();
 
-    private readonly ComponentVersionTracker transformVersions = new();
+    private readonly ComponentVersionTracker<Transform> transformVersions = new();
 
     public bool UsesOwnRenderProcess => false;
 

Engine/Staple.Core/Rendering/Mesh/MeshCombineSystem.cs

@@ -31,7 +31,7 @@ private struct RenderInfo
 
     private readonly ExpandableContainer<RenderInfo> renderers = new();
 
-    private readonly ComponentVersionTracker transformVersions = new();
+    private readonly ComponentVersionTracker<Transform> transformVersions = new();
 
     #region Lifecycle
     public void Prepare()

Engine/Staple.Core/Rendering/Mesh/MeshRenderSystem.cs

@@ -43,7 +43,7 @@ private class InstanceData
             .Build();
     });
 
-    private readonly ComponentVersionTracker transformVersions = new();
+    private readonly ComponentVersionTracker<Transform> transformVersions = new();
 
     public bool UsesOwnRenderProcess => false;