prewarm preafb pools

Aeroluna · Jan 18, 2025 · 961bfec · 961bfec
1 parent 07f8ea9
commit 961bfec
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 2 deletions.
diff --git a/Vivify/Events/AssignObjectPrefab.cs b/Vivify/Events/AssignObjectPrefab.cs
@@ -13,7 +13,7 @@
 namespace Vivify.Events;
 
 [CustomEvent(ASSIGN_OBJECT_PREFAB)]
-internal class AssignObjectPrefab : ICustomEvent
+internal class AssignObjectPrefab : IInitializable, ICustomEvent
 {
     private readonly BeatmapObjectPrefabManager _beatmapObjectPrefabManager;
     private readonly DebrisPrefabManager _debrisPrefabManager;
@@ -38,6 +38,51 @@ private AssignObjectPrefab(
         _saberPrefabManager = saberPrefabManager;
     }
 
+    public void Initialize()
+    {
+        foreach (ICustomEventCustomData customEventCustomData in _deserializedData.CustomEventCustomDatas.Values)
+        {
+            if (customEventCustomData is not AssignObjectPrefabData data)
+            {
+                continue;
+            }
+
+            foreach ((string? key, AssignObjectPrefabData.IPrefabInfo value) in data.Assets)
+            {
+                if (key == null)
+                {
+                    continue;
+                }
+
+                if (value is not AssignObjectPrefabData.ObjectPrefabInfo objectPrefabInfo)
+                {
+                    continue;
+                }
+
+                string? asset = objectPrefabInfo.Asset;
+                if (!string.IsNullOrEmpty(asset))
+                {
+                    _beatmapObjectPrefabManager.PrewarmGameObjectPrefabPool(asset!, 10);
+                }
+
+                string? debrisAsset = objectPrefabInfo.DebrisAsset;
+                if (!string.IsNullOrEmpty(debrisAsset))
+                {
+                    _beatmapObjectPrefabManager.PrewarmGameObjectPrefabPool(asset!, 10);
+                }
+
+                string? anyDirectionAsset = objectPrefabInfo.AnyDirectionAsset;
+                if (!string.IsNullOrEmpty(anyDirectionAsset) &&
+                    key == NOTE_PREFAB)
+                {
+                    _beatmapObjectPrefabManager.PrewarmGameObjectPrefabPool(asset!, 10);
+                }
+
+                break;
+            }
+        }
+    }
+
     public void Callback(CustomEventData customEventData)
     {
         if (!_deserializedData.Resolve(customEventData, out AssignObjectPrefabData? data))

diff --git a/Vivify/ObjectPrefab/Managers/BeatmapObjectPrefabManager.cs b/Vivify/ObjectPrefab/Managers/BeatmapObjectPrefabManager.cs
@@ -63,6 +63,12 @@ public void Dispose()
         _prefabPools.Values.Do(n => n.Dispose());
     }
 
+    internal void PrewarmGameObjectPrefabPool(string assetName, int count)
+    {
+        PrefabPool? prefabPool = GetGameObjectPrefabPool(assetName);
+        prefabPool?.Prewarm(count);
+    }
+
     internal void AssignGameObjectPrefab(
         PrefabList prefabList,
         string? assetName,

diff --git a/Vivify/ObjectPrefab/Pools/PrefabPool.cs b/Vivify/ObjectPrefab/Pools/PrefabPool.cs
@@ -45,7 +45,7 @@ public GameObject Spawn(Component component, float startTime)
         GameObject spawned;
         if (_inactive.Count == 0)
         {
-            spawned = Object.Instantiate(_original);
+            spawned = _instantiator.InstantiatePrefab(_original);
         }
         else
         {
@@ -65,4 +65,15 @@ public GameObject Spawn(Component component, float startTime)
         _instantiator.SongSynchronize(spawned, startTime);
         return spawned;
     }
+
+    internal void Prewarm(int count)
+    {
+        for (int i = 0; i < count; i++)
+        {
+            GameObject result = _instantiator.InstantiatePrefab(_original);
+            result.SetActive(false);
+            _inactive.Push(result);
+            _instantiator.SongSynchronize(result, 0);
+        }
+    }
 }