Add ModuleBoosterManager for configurable booster thrust multipliers

This commit is contained in:
2025-02-24 09:33:27 +01:00
parent 9718e254aa
commit 320c032e30

View File

@@ -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();
}
}
}