Extract generic object field multiplier logic to separate file
This commit is contained in:
@@ -1,4 +1,6 @@
|
|||||||
using HarmonyLib;
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using HarmonyLib;
|
||||||
|
|
||||||
namespace TerraTech {
|
namespace TerraTech {
|
||||||
[HarmonyPatch]
|
[HarmonyPatch]
|
||||||
|
|||||||
@@ -4,142 +4,6 @@ using BepInEx.Configuration;
|
|||||||
using HarmonyLib;
|
using HarmonyLib;
|
||||||
|
|
||||||
namespace TerraTech {
|
namespace TerraTech {
|
||||||
public class MultipliedField<TObject, TValue> {
|
|
||||||
private readonly string fieldName;
|
|
||||||
private readonly ConfigEntry<TValue> multiplier;
|
|
||||||
private readonly Traverse parentTraverse;
|
|
||||||
private TValue originalValue;
|
|
||||||
|
|
||||||
public MultipliedField(string fieldName, ConfigEntry<TValue> multiplier, Traverse parentTraverse) {
|
|
||||||
this.fieldName = fieldName;
|
|
||||||
this.multiplier = multiplier;
|
|
||||||
this.parentTraverse = parentTraverse;
|
|
||||||
|
|
||||||
if (!parentTraverse.Field(fieldName).FieldExists()) {
|
|
||||||
throw new ArgumentException($"Field {fieldName} does not exist on {parentTraverse.ToString()}");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public TValue GetValue() {
|
|
||||||
var value = parentTraverse.Field(fieldName).GetValue();
|
|
||||||
if (value == null)
|
|
||||||
throw new InvalidOperationException($"Field {fieldName} returned null");
|
|
||||||
return (TValue)value;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void SetValue(TValue value) {
|
|
||||||
if (value == null)
|
|
||||||
throw new ArgumentNullException(nameof(value));
|
|
||||||
parentTraverse.Field(fieldName).SetValue(value);
|
|
||||||
var verifyValue = GetValue();
|
|
||||||
if (!verifyValue.Equals(value))
|
|
||||||
throw new InvalidOperationException($"Field {fieldName} set to {value} but read back as {verifyValue}");
|
|
||||||
}
|
|
||||||
|
|
||||||
public void CaptureOriginal() {
|
|
||||||
originalValue = GetValue();
|
|
||||||
if (Main.debug.Value)
|
|
||||||
Console.WriteLine($"Captured original value for {fieldName}: {originalValue}");
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Apply() {
|
|
||||||
try {
|
|
||||||
dynamic originalDynamic = originalValue;
|
|
||||||
dynamic multiplierDynamic = multiplier.Value;
|
|
||||||
var newValue = (TValue)(originalDynamic * multiplierDynamic);
|
|
||||||
|
|
||||||
if (Main.debug.Value)
|
|
||||||
Console.WriteLine($"Applying to {fieldName}: {originalValue} * {multiplier.Value} = {newValue}");
|
|
||||||
|
|
||||||
SetValue(newValue);
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new InvalidOperationException($"Failed to apply multiplication to {fieldName}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Restore() {
|
|
||||||
if (Main.debug.Value)
|
|
||||||
Console.WriteLine($"Restoring {fieldName} to original value: {originalValue}");
|
|
||||||
SetValue(originalValue);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void LogValue(string prefix) {
|
|
||||||
if (!Main.debug.Value)
|
|
||||||
return;
|
|
||||||
var currentValue = GetValue();
|
|
||||||
Console.WriteLine(
|
|
||||||
$"{prefix} {parentTraverse}; {fieldName}: {currentValue} (original: {originalValue}, multiplier: {multiplier.Value})");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public class MultipliedObject<T> {
|
|
||||||
private readonly Traverse objectTraverse;
|
|
||||||
private readonly List<MultipliedField<T, float>> fields = new List<MultipliedField<T, float>>();
|
|
||||||
|
|
||||||
public MultipliedObject(T instance) => objectTraverse = Traverse.Create(instance);
|
|
||||||
public void AddField(string fieldName, ConfigEntry<float> multiplier) =>
|
|
||||||
fields.Add(new MultipliedField<T, float>(fieldName, multiplier, objectTraverse));
|
|
||||||
public void CaptureFrom() => fields.ForEach(f => f.CaptureOriginal());
|
|
||||||
public void ApplyTo() => fields.ForEach(f => f.Apply());
|
|
||||||
public void RestoreTo() => fields.ForEach(f => f.Restore());
|
|
||||||
public void LogValues(string prefix) {
|
|
||||||
if (!Main.debug.Value)
|
|
||||||
return;
|
|
||||||
fields.ForEach(f => f.LogValue(prefix));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public class MultipliedObjectManager<T>
|
|
||||||
where T : class {
|
|
||||||
private readonly Dictionary<T, MultipliedObject<T>> managedObjects = new Dictionary<T, MultipliedObject<T>>();
|
|
||||||
private readonly Action<MultipliedObject<T>> configureObject;
|
|
||||||
|
|
||||||
public MultipliedObjectManager(Action<MultipliedObject<T>> configureObject) {
|
|
||||||
this.configureObject = configureObject;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void OnObjectAttached(T instance) {
|
|
||||||
if (!managedObjects.ContainsKey(instance)) {
|
|
||||||
if (Main.debug.Value)
|
|
||||||
Console.WriteLine($"{typeof(T).Name}.OnAttached");
|
|
||||||
|
|
||||||
var multipliedObject = new MultipliedObject<T>(instance);
|
|
||||||
configureObject(multipliedObject);
|
|
||||||
multipliedObject.CaptureFrom();
|
|
||||||
managedObjects.Add(instance, multipliedObject);
|
|
||||||
multipliedObject.LogValues("Patching");
|
|
||||||
|
|
||||||
ApplyTo(instance);
|
|
||||||
multipliedObject.LogValues("Patched");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public void OnObjectDetached(T instance) {
|
|
||||||
if (managedObjects.TryGetValue(instance, out var multipliedObject)) {
|
|
||||||
if (Main.debug.Value) {
|
|
||||||
Console.WriteLine($"{typeof(T).Name}.OnDetaching");
|
|
||||||
multipliedObject.LogValues("Restoring");
|
|
||||||
}
|
|
||||||
|
|
||||||
RestoreTo(instance);
|
|
||||||
multipliedObject.LogValues("Restored");
|
|
||||||
managedObjects.Remove(instance);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public void ApplyAll() {
|
|
||||||
if (Main.debug.Value)
|
|
||||||
Console.WriteLine($"Modifying {managedObjects.Count} {typeof(T).Name}");
|
|
||||||
foreach (var instance in managedObjects.Keys) {
|
|
||||||
RestoreTo(instance);
|
|
||||||
ApplyTo(instance);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void ApplyTo(T instance) if (managedObjects.TryGetValue(instance, out var obj)) obj.ApplyTo();
|
|
||||||
private void RestoreTo(T instance) if (managedObjects.TryGetValue(instance, out var obj)) obj.RestoreTo();
|
|
||||||
}
|
|
||||||
|
|
||||||
[HarmonyPatch]
|
[HarmonyPatch]
|
||||||
public class ModuleShieldGeneratorManager {
|
public class ModuleShieldGeneratorManager {
|
||||||
private static readonly MultipliedObjectManager<ModuleShieldGenerator> manager =
|
private static readonly MultipliedObjectManager<ModuleShieldGenerator> manager =
|
||||||
|
|||||||
165
Projects/TerraTech/TerraTech/ObjectFieldMultiplier.cs
Normal file
165
Projects/TerraTech/TerraTech/ObjectFieldMultiplier.cs
Normal file
@@ -0,0 +1,165 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using BepInEx.Configuration;
|
||||||
|
using HarmonyLib;
|
||||||
|
|
||||||
|
namespace TerraTech {
|
||||||
|
public class MultipliedField<TObject, TValue> {
|
||||||
|
private readonly string fieldName;
|
||||||
|
private readonly ConfigEntry<TValue> multiplier;
|
||||||
|
private readonly Traverse parentTraverse;
|
||||||
|
private TValue originalValue;
|
||||||
|
|
||||||
|
public MultipliedField(string fieldName, ConfigEntry<TValue> multiplier, Traverse parentTraverse) {
|
||||||
|
this.fieldName = fieldName;
|
||||||
|
this.multiplier = multiplier;
|
||||||
|
this.parentTraverse = parentTraverse;
|
||||||
|
|
||||||
|
if (!parentTraverse.Field(fieldName).FieldExists()) {
|
||||||
|
throw new ArgumentException(
|
||||||
|
string.Format("Field {0} does not exist on {1}", fieldName, parentTraverse.ToString()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public TValue GetValue() {
|
||||||
|
var value = parentTraverse.Field(fieldName).GetValue();
|
||||||
|
if (value == null)
|
||||||
|
throw new InvalidOperationException(string.Format("Field {0} returned null", fieldName));
|
||||||
|
return (TValue)value;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void SetValue(TValue value) {
|
||||||
|
parentTraverse.Field(fieldName).SetValue(value);
|
||||||
|
var verifyValue = GetValue();
|
||||||
|
if (!verifyValue.Equals(value))
|
||||||
|
throw new InvalidOperationException(
|
||||||
|
string.Format("Field {0} set to {1} but read back as {2}", fieldName, value, verifyValue));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void CaptureOriginal() {
|
||||||
|
originalValue = GetValue();
|
||||||
|
if (Main.debug.Value)
|
||||||
|
Console.WriteLine(string.Format("Captured original value for {0}: {1}", fieldName, originalValue));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Apply() {
|
||||||
|
try {
|
||||||
|
dynamic originalDynamic = originalValue;
|
||||||
|
dynamic multiplierDynamic = multiplier.Value;
|
||||||
|
var newValue = (TValue)(originalDynamic * multiplierDynamic);
|
||||||
|
|
||||||
|
if (Main.debug.Value)
|
||||||
|
Console.WriteLine(string.Format("Applying to {0}: {1} * {2} = {3}", fieldName, originalValue,
|
||||||
|
multiplier.Value, newValue));
|
||||||
|
|
||||||
|
SetValue(newValue);
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new InvalidOperationException(string.Format("Failed to apply multiplication to {0}", fieldName),
|
||||||
|
e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Restore() {
|
||||||
|
if (Main.debug.Value)
|
||||||
|
Console.WriteLine(string.Format("Restoring {0} to original value: {1}", fieldName, originalValue));
|
||||||
|
SetValue(originalValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void LogValue(string prefix) {
|
||||||
|
if (!Main.debug.Value)
|
||||||
|
return;
|
||||||
|
var currentValue = GetValue();
|
||||||
|
Console.WriteLine(
|
||||||
|
$"{prefix} {parentTraverse}; {fieldName}: {currentValue} (original: {originalValue}, multiplier: {multiplier.Value})");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class MultipliedObject<T> {
|
||||||
|
private readonly Traverse objectTraverse;
|
||||||
|
private readonly List<MultipliedField<T, float>> fields = new List<MultipliedField<T, float>>();
|
||||||
|
|
||||||
|
public MultipliedObject(T instance) {
|
||||||
|
objectTraverse = Traverse.Create(instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void AddField(string fieldName, ConfigEntry<float> multiplier) {
|
||||||
|
fields.Add(new MultipliedField<T, float>(fieldName, multiplier, objectTraverse));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void CaptureFrom() {
|
||||||
|
fields.ForEach(f => f.CaptureOriginal());
|
||||||
|
}
|
||||||
|
|
||||||
|
public void ApplyTo() {
|
||||||
|
fields.ForEach(f => f.Apply());
|
||||||
|
}
|
||||||
|
|
||||||
|
public void RestoreTo() {
|
||||||
|
fields.ForEach(f => f.Restore());
|
||||||
|
}
|
||||||
|
|
||||||
|
public void LogValues(string prefix) {
|
||||||
|
if (!Main.debug.Value)
|
||||||
|
return;
|
||||||
|
fields.ForEach(f => f.LogValue(prefix));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class MultipliedObjectManager<T>
|
||||||
|
where T : class {
|
||||||
|
private readonly Dictionary<T, MultipliedObject<T>> managedObjects = new Dictionary<T, MultipliedObject<T>>();
|
||||||
|
private readonly Action<MultipliedObject<T>> configureObject;
|
||||||
|
|
||||||
|
public MultipliedObjectManager(Action<MultipliedObject<T>> configureObject) {
|
||||||
|
this.configureObject = configureObject;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void OnObjectAttached(T instance) {
|
||||||
|
if (!managedObjects.ContainsKey(instance)) {
|
||||||
|
if (Main.debug.Value)
|
||||||
|
Console.WriteLine(string.Format("{0}.OnAttached", typeof(T).Name));
|
||||||
|
|
||||||
|
var multipliedObject = new MultipliedObject<T>(instance);
|
||||||
|
configureObject(multipliedObject);
|
||||||
|
multipliedObject.CaptureFrom();
|
||||||
|
managedObjects.Add(instance, multipliedObject);
|
||||||
|
multipliedObject.LogValues("Patching");
|
||||||
|
|
||||||
|
ApplyTo(instance);
|
||||||
|
multipliedObject.LogValues("Patched");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void OnObjectDetached(T instance) {
|
||||||
|
if (managedObjects.TryGetValue(instance, out var multipliedObject)) {
|
||||||
|
if (Main.debug.Value) {
|
||||||
|
Console.WriteLine($"{typeof(T).Name}.OnDetaching");
|
||||||
|
multipliedObject.LogValues("Restoring");
|
||||||
|
}
|
||||||
|
|
||||||
|
RestoreTo(instance);
|
||||||
|
multipliedObject.LogValues("Restored");
|
||||||
|
managedObjects.Remove(instance);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void ApplyAll() {
|
||||||
|
if (Main.debug.Value)
|
||||||
|
Console.WriteLine($"Modifying {managedObjects.Count} {typeof(T).Name}");
|
||||||
|
foreach (var instance in managedObjects.Keys) {
|
||||||
|
RestoreTo(instance);
|
||||||
|
ApplyTo(instance);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void ApplyTo(T instance) {
|
||||||
|
if (managedObjects.TryGetValue(instance, out var obj))
|
||||||
|
obj.ApplyTo()
|
||||||
|
}
|
||||||
|
|
||||||
|
private void RestoreTo(T instance) {
|
||||||
|
if (managedObjects.TryGetValue(instance, out var obj))
|
||||||
|
obj.RestoreTo()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -48,6 +48,7 @@
|
|||||||
<Compile Include="ModuleEnergy.cs" />
|
<Compile Include="ModuleEnergy.cs" />
|
||||||
<Compile Include="ModuleEnergyStoreManager.cs" />
|
<Compile Include="ModuleEnergyStoreManager.cs" />
|
||||||
<Compile Include="ModuleShieldGeneratorManager.cs" />
|
<Compile Include="ModuleShieldGeneratorManager.cs" />
|
||||||
|
<Compile Include="ObjectFieldMultiplier.cs" />
|
||||||
<Compile Include="Patches.cs" />
|
<Compile Include="Patches.cs" />
|
||||||
<Compile Include="Properties\AssemblyInfo.cs" />
|
<Compile Include="Properties\AssemblyInfo.cs" />
|
||||||
<Compile Include="SeekingProjectileManager.cs" />
|
<Compile Include="SeekingProjectileManager.cs" />
|
||||||
|
|||||||
Reference in New Issue
Block a user