Add ModuleBoosterManager for configurable booster thrust multipliers

2025-02-24 09:33:27 +01:00
parent 9718e254aa
commit 320c032e30
1 changed files with 66 additions and 0 deletions
--- a/Projects/TerraTech/TerraTech/ModuleBoosterManager.cs
+++ b/Projects/TerraTech/TerraTech/ModuleBoosterManager.cs
@@ -0,0 +1,66 @@
+using System;
+using BepInEx.Configuration;
+using HarmonyLib;
+
+namespace TerraTech {
+    [HarmonyPatch]
+    public class ModuleBoosterManager {
+        private static readonly MultipliedObjectManager<Thruster> fanManager =
+            new MultipliedObjectManager<Thruster>(ConfigureFanThruster);
+        private static readonly MultipliedObjectManager<Thruster> jetManager =
+            new MultipliedObjectManager<Thruster>(ConfigureJetThruster);
+
+        public static ConfigEntry<float> boosterFanThrustMultiplier;
+        public static ConfigEntry<float> boosterJetThrustMultiplier;
+
+        public static void Setup(ConfigFile config) {
+            float min = 0.01f;
+            float max = 32f;
+
+            boosterFanThrustMultiplier = config.Bind(
+                "Booster", "Booster Fan Thrust Multiplier", 1f,
+                new ConfigDescription("Booster Fan Thrust Multiplier", new AcceptableValueRange<float>(min, max)));
+            boosterFanThrustMultiplier.SettingChanged += (sender, args) => DoPatch();
+
+            boosterJetThrustMultiplier = config.Bind(
+                "Booster", "Booster Jet Thrust Multiplier", 1f,
+                new ConfigDescription("Booster Jet Thrust Multiplier", new AcceptableValueRange<float>(min, max)));
+            boosterJetThrustMultiplier.SettingChanged += (sender, args) => DoPatch();
+        }
+
+        private static void ConfigureFanThruster(MultipliedObject<Thruster> obj) {
+            obj.AddField(new FieldConfiguration<float>("m_Force", boosterFanThrustMultiplier));
+        }
+
+        private static void ConfigureJetThruster(MultipliedObject<Thruster> obj) {
+            obj.AddField(new FieldConfiguration<float>("m_Force", boosterJetThrustMultiplier));
+        }
+
+        [HarmonyPrefix]
+        [HarmonyPatch(typeof(ModuleBooster), "OnAttached")]
+        static void PostfixCreate(ModuleBooster __instance) {
+            var trav = Traverse.Create(__instance);
+            var fans = trav.Field("fans").GetValue<Thruster[]>();
+            var jets = trav.Field("jets").GetValue<Thruster[]>();
+
+            foreach (var fan in fans) fanManager.OnObjectAttached(fan);
+            foreach (var jet in jets) jetManager.OnObjectAttached(jet);
+        }
+
+        [HarmonyPrefix]
+        [HarmonyPatch(typeof(ModuleBooster), "OnDetaching")]
+        static void PostfixDestroy(ModuleBooster __instance) {
+            var trav = Traverse.Create(__instance);
+            var fans = trav.Field("fans").GetValue<Thruster[]>();
+            var jets = trav.Field("jets").GetValue<Thruster[]>();
+
+            foreach (var fan in fans) fanManager.OnObjectDetached(fan);
+            foreach (var jet in jets) jetManager.OnObjectDetached(jet);
+        }
+
+        public static void DoPatch() {
+            fanManager.ApplyAll();
+            jetManager.ApplyAll();
+        }
+    }
+}