Add TechBoosterManager with configurable fuel multipliers

This commit is contained in:
2025-02-24 17:15:38 +01:00
parent 5723e9592e
commit 4f51f8599e
4 changed files with 89 additions and 2 deletions

View File

@@ -41,6 +41,7 @@ namespace TerraTech {
TankBeamManager.Setup(Config);
ModuleWeaponManager.Setup(Config);
ModuleHeartManager.Setup(Config);
TechBoosterManager.Setup(Config);
xpMultiplier =
Config.Bind("General", "XP Multiplier", 1f,

View File

@@ -38,8 +38,8 @@ namespace TerraTech {
}
private static void ConfigureHeart(MultipliedObject<ModuleHeart> obj) {
obj.AddField(
new FieldConfiguration<float, float>("m_EventHorizonRadius", eventHorizonRadiusMultiplier, ShouldApply));
obj.AddField(new FieldConfiguration<float, float>("m_EventHorizonRadius", eventHorizonRadiusMultiplier,
ShouldApply));
obj.AddField(new FieldConfiguration<float, float>("m_SetupTime", setupTimeMultiplier, ShouldApply));
obj.AddField(new FieldConfiguration<float, float>("m_StartShrinkingRadius", startShrinkingRadiusMultiplier,
ShouldApply));

View File

@@ -0,0 +1,85 @@
using System;
using System.Collections.Generic;
using BepInEx.Configuration;
using HarmonyLib;
namespace TerraTech {
[HarmonyPatch]
public class TechBoosterManager {
private struct TechBoosterData {
public float m_FuelCapacity;
public float m_FuelRefill;
}
private static Dictionary<TechBooster, TechBoosterData> techBoosterData =
new Dictionary<TechBooster, TechBoosterData>();
public static ConfigEntry<bool> playerOnly;
public static ConfigEntry<float> fuelCapacityMultiplier;
public static ConfigEntry<float> fuelRefillMultiplier;
public static void Setup(ConfigFile config) {
float min = 0.01f;
float max = 32f;
playerOnly = config.Bind("TechBooster", "Player Only", false, new ConfigDescription("Player Only"));
fuelCapacityMultiplier = config.Bind(
"TechBooster", "Fuel Capacity Multiplier", 1f,
new ConfigDescription("Fuel Capacity Multiplier", new AcceptableValueRange<float>(min, max)));
fuelRefillMultiplier =
config.Bind("TechBooster", "Fuel Refill Multiplier", 1f,
new ConfigDescription("Fuel Refill Multiplier", new AcceptableValueRange<float>(min, max)));
}
private static readonly Func<object, bool> ShouldApply = obj => {
if (!playerOnly.Value)
return true;
return CykUtil.IsObjectPlayerTank(obj);
};
[HarmonyPrefix]
[HarmonyPatch(typeof(TechBooster), "AddFuelTank")]
static void PostfixCreate(TechBooster __instance, ref ModuleFuelTank tank) {
var trav = Traverse.Create(__instance);
var fuelCapacityField = trav.Field("m_FuelCapacity");
var fuelRefillField = trav.Field("m_FuelRefill");
TechBoosterData boosterData;
bool ok = techBoosterData.TryGetValue(__instance, out boosterData);
if (ok) {
boosterData.m_FuelCapacity = fuelCapacityField.GetValue<float>() + tank.Capacity;
boosterData.m_FuelRefill = fuelRefillField.GetValue<float>() + tank.RefillRate;
} else {
boosterData = new TechBoosterData { m_FuelCapacity = fuelCapacityField.GetValue<float>(),
m_FuelRefill = fuelRefillField.GetValue<float>() };
techBoosterData[__instance] = boosterData;
}
fuelCapacityField.SetValue(fuelCapacityField.GetValue<float>() * fuelCapacityMultiplier.Value);
fuelRefillField.SetValue(fuelRefillField.GetValue<float>() * fuelRefillMultiplier.Value);
}
[HarmonyPrefix]
[HarmonyPatch(typeof(TechBooster), "RemoveFuelTank")]
static void PostfixDestroy(TechBooster __instance, ref ModuleFuelTank tank) {
var trav = Traverse.Create(__instance);
var fuelCapacityField = trav.Field("m_FuelCapacity");
var fuelRefillField = trav.Field("m_FuelRefill");
TechBoosterData boosterData;
bool ok = techBoosterData.TryGetValue(__instance, out boosterData);
if (ok) {
boosterData.m_FuelCapacity -= tank.Capacity;
boosterData.m_FuelRefill -= tank.RefillRate;
} else {
boosterData = new TechBoosterData { m_FuelCapacity = fuelCapacityField.GetValue<float>(),
m_FuelRefill = fuelRefillField.GetValue<float>() };
techBoosterData[__instance] = boosterData;
}
fuelCapacityField.SetValue(fuelCapacityField.GetValue<float>() * fuelCapacityMultiplier.Value);
fuelRefillField.SetValue(fuelRefillField.GetValue<float>() * fuelRefillMultiplier.Value);
}
}
}

View File

@@ -57,6 +57,7 @@
<Compile Include="ModuleWeaponManager.cs" />
<Compile Include="CykUtil.cs" />
<Compile Include="ModuleHeartManager.cs" />
<Compile Include="TechBoosterManager.cs" />
</ItemGroup>
<ItemGroup>
<Reference Include="0Harmony">