Add TechBoosterManager with configurable fuel multipliers
This commit is contained in:
@@ -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,
|
||||
|
@@ -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));
|
||||
|
85
Projects/TerraTech/TerraTech/TechBoosterManager.cs
Normal file
85
Projects/TerraTech/TerraTech/TechBoosterManager.cs
Normal 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);
|
||||
}
|
||||
}
|
||||
}
|
@@ -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">
|
||||
|
Reference in New Issue
Block a user