Added overload for "SetCapacity()", added it where necessary.

UnderminersTeam · Jan 23, 2023 · 98558f6 · 98558f6
1 parent fc87007
commit 98558f6
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/UndertaleModLib/Models/UndertaleGameObject.cs b/UndertaleModLib/Models/UndertaleGameObject.cs
@@ -160,6 +160,7 @@ protected void OnPropertyChanged([CallerMemberName] string name = null)
     /// </summary>
     public UndertaleGameObject()
     {
+        Events.SetCapacity(eventTypesLength);
         for (int i = 0; i < eventTypesLength; i++)
             Events.internalList.Add(new UndertalePointerList<Event>());
     }

diff --git a/UndertaleModLib/Models/UndertaleRoom.cs b/UndertaleModLib/Models/UndertaleRoom.cs
@@ -255,6 +255,8 @@ protected void OnPropertyChanged([CallerMemberName] string name = null)
     /// </summary>
     public UndertaleRoom()
     {
+        Backgrounds.SetCapacity(8);
+        Views.SetCapacity(8);
         for (int i = 0; i < 8; i++)
             Backgrounds.internalList.Add(new Background());
         for (int i = 0; i < 8; i++)

diff --git a/UndertaleModLib/UndertaleChunkTypes.cs b/UndertaleModLib/UndertaleChunkTypes.cs
@@ -225,6 +225,8 @@ internal override void SerializeChunk(UndertaleWriter writer)
         internal override void UnserializeChunk(UndertaleReader reader)
         {
             uint count = reader.ReadUInt32();
+            List.SetCapacity(count);
+
             for (int i = 0; i < count; i++)
                 Align &= (reader.ReadUInt32() % Alignment == 0);
             for (int i = 0; i < count; i++)

diff --git a/UndertaleModLib/UndertaleLists.cs b/UndertaleModLib/UndertaleLists.cs
@@ -33,17 +33,18 @@ public UndertaleListBase()
         /// <inheritdoc />
         public abstract void Unserialize(UndertaleReader reader);
 
-        public void SetCapacity(uint capacity)
+        public void SetCapacity(int capacity)
         {
             try
             {
-                internalList.Capacity = (int)capacity;
+                internalList.Capacity = capacity;
             }
             catch (Exception e)
             {
                 throw new UndertaleSerializationException($"{e.Message}\nwhile trying to \"SetCapacity()\" of \"UndertalePointerList<{typeof(T).FullName}>\".");
             }
         }
+        public void SetCapacity(uint capacity) => SetCapacity((int)capacity);
     }
 
     public class UndertaleSimpleList<T> : UndertaleListBase<T>, UndertaleObject where T : UndertaleObject, new()