Add Fast FindModulesImplementing (and a const version), add FastRemov…

…eAt, add Swap, move Principia bit in KSPUtils to ModUtils
KSP-RO · Dec 4, 2023 · 6abbd4f · 6abbd4f
1 parent 71348d0
commit 6abbd4f
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 16 deletions.
diff --git a/Source/ROUtils/Utils/KSPUtils.cs b/Source/ROUtils/Utils/KSPUtils.cs
@@ -47,27 +47,34 @@ public static List<Type> GetAllLoadedTypes<T>(bool instantiableOnly = true)
             return list;
         }
 
-        public static double PrincipiaCorrectInclination(Orbit o)
+        public static List<T> FixedFindModulesImplementing<T>(this Part part) where T : class
         {
-            if (ModUtils.IsPrincipiaInstalled && o.referenceBody != (FlightGlobals.currentMainBody ?? Planetarium.fetch.Home))
-            {
-                Vector3d polarAxis = o.referenceBody.BodyFrame.Z;
+            var listPM = FixedFindModulesImplementingConst<T>(part);
+            int c = listPM.Count;
+            var newList = new List<T>(c);
+            for (int i = 0; i < c; ++i)
+                newList.Add(listPM[i] as T);
 
-                double hSqrMag = o.h.sqrMagnitude;
-                if (hSqrMag == 0d)
-                {
-                    return Math.Acos(Vector3d.Dot(polarAxis, o.pos) / o.pos.magnitude) * (180.0 / Math.PI);
-                }
-                else
+            return newList;
+        }
+
+        public static List<PartModule> FixedFindModulesImplementingConst<T>(this Part part) where T : class
+        {
+            Type typeFromHandle = typeof(T);
+            if (part.cachedModuleLists.TryGetValue(typeFromHandle, out var list))
+                return list;
+
+            list = new List<PartModule>();
+            for (int i = 0, ic = part.modules.Count; i < ic; ++i)
+            {
+                PartModule partModule = part.modules[i];
+                if (partModule is T)
                 {
-                    Vector3d orbitZ = o.h / Math.Sqrt(hSqrMag);
-                    return Math.Atan2((orbitZ - polarAxis).magnitude, (orbitZ + polarAxis).magnitude) * (2d * (180.0 / Math.PI));
+                    list.Add(partModule);
                 }
             }
-            else
-            {
-                return o.inclination;
-            }
+            part.cachedModuleLists[typeFromHandle] = list;
+            return list;
         }
 
         /// <summary>

diff --git a/Source/ROUtils/Utils/MiscUtils.cs b/Source/ROUtils/Utils/MiscUtils.cs
@@ -19,6 +19,20 @@ public static T Pop<T>(this List<T> list)
             return val;
         }
 
+        public static void Swap<T>(ref T a, ref T b)
+        {
+            T c = a;
+            a = b;
+            b = c;
+        }
+
+        public static void FastRemoveAt<T>(this List<T> list, int i)
+        {
+            int c = list.Count - 1;
+            list[i] = list[c];
+            list.RemoveAt(c);
+        }
+
         public static Texture2D GetReadableCopy(this Texture2D texture)
         {
             Texture2D readable = new Texture2D(texture.width, texture.height);

diff --git a/Source/ROUtils/Utils/ModUtils.cs b/Source/ROUtils/Utils/ModUtils.cs
@@ -98,5 +98,28 @@ private static bool FindPrincipia()
             _isPrincipiaInstalled = AssemblyLoader.loadedAssemblies.Any(a => a.name.Equals("ksp_plugin_adapter", StringComparison.OrdinalIgnoreCase));
             return _isPrincipiaInstalled;
         }
+
+        public static double PrincipiaCorrectInclination(Orbit o)
+        {
+            if (ModUtils.IsPrincipiaInstalled && o.referenceBody != (FlightGlobals.currentMainBody ?? Planetarium.fetch.Home))
+            {
+                Vector3d polarAxis = o.referenceBody.BodyFrame.Z;
+
+                double hSqrMag = o.h.sqrMagnitude;
+                if (hSqrMag == 0d)
+                {
+                    return Math.Acos(Vector3d.Dot(polarAxis, o.pos) / o.pos.magnitude) * (180.0 / Math.PI);
+                }
+                else
+                {
+                    Vector3d orbitZ = o.h / Math.Sqrt(hSqrMag);
+                    return Math.Atan2((orbitZ - polarAxis).magnitude, (orbitZ + polarAxis).magnitude) * (2d * (180.0 / Math.PI));
+                }
+            }
+            else
+            {
+                return o.inclination;
+            }
+        }
     }
 }