dave/BepInEx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Add battery capacity multiplier configuration

Browse Source
This commit is contained in:
PhatPhuckDave
2025-02-23 23:50:36 +01:00
parent 4ad206a198
commit 5df952a69a
4 changed files with 71 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
Projects/TerraTech/TerraTech/ModuleEnergyStoreManager.cs Normal file
View File

@@ -0,0 +1,65 @@
using HarmonyLib;
namespace TerraTech {
[HarmonyPatch]
public class ModuleEnergyStoreManager {
private static Dictionary<ModuleEnergyStore, float> batteryCapacity =
new Dictionary<ModuleEnergyStore, float>();
[HarmonyPrefix]
[HarmonyPatch(typeof(ModuleEnergyStore), "OnAttached")]
static void PostfixCreate(ModuleEnergyStore __instance) {
if (Main.debug.Value)
Console.WriteLine("ModuleEnergyStore.OnAttached");
if (!batteryCapacity.ContainsKey(__instance)) {
batteryCapacity.Add(__instance, __instance.m_BatteryCapacity);
if (Main.debug.Value)
Console.WriteLine("Patching {0}; m_BatteryCapacity: {1}", __instance.ToString(),
__instance.m_BatteryCapacity);
}
DoPatchSingle(__instance);
if (Main.debug.Value)
Console.WriteLine("Patched {0}; m_BatteryCapacity: {1}", __instance.ToString(),
__instance.m_BatteryCapacity);
}
[HarmonyPrefix]
[HarmonyPatch(typeof(ModuleEnergyStore), "OnDetaching")]
static void PostfixDestroy(ModuleEnergyStore __instance) {
if (Main.debug.Value)
Console.WriteLine("ModuleEnergyStore.OnDetaching");
if (Main.debug.Value)
Console.WriteLine("Restoring {0}; m_BatteryCapacity: {1}", __instance.ToString(),
__instance.m_BatteryCapacity);
DoRestoreSingle(__instance);
if (Main.debug.Value)
Console.WriteLine("Restored {0}; m_BatteryCapacity: {1}", __instance.ToString(),
__instance.m_BatteryCapacity);
batteryCapacity.Remove(__instance);
}
public static void DoPatch() {
if (Main.debug.Value)
Console.WriteLine("Modifying {0} ModuleEnergyStore", batteryCapacity.Count);
foreach (KeyValuePair<ModuleEnergyStore, float> keyValuePair in batteryCapacity) {
DoRestoreSingle(keyValuePair.Key);
DoPatchSingle(keyValuePair.Key);
}
}
static void DoPatchSingle(ModuleEnergyStore moduleEnergyStore) {
moduleEnergyStore.m_BatteryCapacity =
batteryCapacity[moduleEnergyStore] * Main.batteryCapacityMultiplier.Value;
}
static void DoRestoreSingle(ModuleEnergyStore moduleEnergyStore) {
if (batteryCapacity.ContainsKey(moduleEnergyStore))
moduleEnergyStore.m_BatteryCapacity = batteryCapacity[moduleEnergyStore];
}
}
}