86 lines
3.8 KiB
C#
86 lines
3.8 KiB
C#
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);
|
|
}
|
|
}
|
|
}
|