soarqin/DSP_Mods
1
0
Fork 0
mirror of https://github.com/soarqin/DSP_Mods.git synced 2025-12-09 19:33:28 +08:00
Code Activity

Work in progress

Browse Source
This commit is contained in:
Soar Qin
2022-11-30 01:34:01 +08:00
parent 45485c35f3
commit 1968631cbd
9 changed files with 437 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

98
OverclockEverything/BeltFix.cs Normal file
View File

@@ -0,0 +1,98 @@
using System.Collections.Generic;
using System.Reflection.Emit;
using HarmonyLib;
namespace OverclockEverything;
[HarmonyPatch]
public static class BeltFix
{
private static CodeInstruction[] LdcInstrs = new[]
{
new CodeInstruction(OpCodes.Ldc_I4_0), new CodeInstruction(OpCodes.Ldc_I4_1), new CodeInstruction(OpCodes.Ldc_I4_2),
new CodeInstruction(OpCodes.Ldc_I4_3), new CodeInstruction(OpCodes.Ldc_I4_4), new CodeInstruction(OpCodes.Ldc_I4_5),
new CodeInstruction(OpCodes.Ldc_I4_6), new CodeInstruction(OpCodes.Ldc_I4_7), new CodeInstruction(OpCodes.Ldc_I4_8),
new CodeInstruction(OpCodes.Ldc_I4_S, 9), new CodeInstruction(OpCodes.Ldc_I4_S, 10)
};
[HarmonyTranspiler, HarmonyPatch(typeof(CargoTraffic), "AlterBeltRenderer")]
public static IEnumerable<CodeInstruction> CargoTraffic_AlterBeltRenderer_Transpiler(IEnumerable<CodeInstruction> instructions)
{
bool lastIsSpeed = false;
foreach (var instr in instructions)
{
if (lastIsSpeed)
{
lastIsSpeed = false;
if (instr.opcode == OpCodes.Ldc_I4_1)
{
yield return LdcInstrs[Patch.beltSpeed[0]];
}
else if (instr.opcode == OpCodes.Ldc_I4_2)
{
yield return LdcInstrs[Patch.beltSpeed[1]];
}
else
{
yield return instr;
}
}
else
{
lastIsSpeed = instr.opcode == OpCodes.Ldfld &&
instr.OperandIs(AccessTools.Field(typeof(BeltComponent), nameof(BeltComponent.speed)));
yield return instr;
}
}
}
[HarmonyPrefix]
[HarmonyPatch(typeof(ConnGizmoRenderer), "AddBlueprintBeltMajorPoint")]
[HarmonyPatch(typeof(ConnGizmoRenderer), "AddBlueprintBeltPoint")]
[HarmonyPatch(typeof(ConnGizmoRenderer), "AddBlueprintBeltConn")]
public static void ConnGizmoRenderer_AddBlueprintBelt_Prefix(ref ConnGizmoRenderer __instance, ref uint color)
{
var bspeed = Patch.beltSpeed;
color = color >= bspeed[2] ? 3u : color >= bspeed[1] ? 2u : 1u;
}
[HarmonyTranspiler]
[HarmonyPatch(typeof(ConnGizmoRenderer), "Update")]
public static IEnumerable<CodeInstruction> ConnGizmoRenderer_Update_Transpiler(IEnumerable<CodeInstruction> instructions, ILGenerator generator)
{
bool lastIsLdcI4_3 = false;
foreach (var instr in instructions)
{
if (lastIsLdcI4_3)
{
lastIsLdcI4_3 = false;
yield return instr;
if (instr.opcode == OpCodes.Stfld && instr.OperandIs(AccessTools.Field(typeof(ConnGizmoObj), nameof(ConnGizmoObj.color))))
{
var label1 = generator.DefineLabel();
var label2 = generator.DefineLabel();
yield return new CodeInstruction(OpCodes.Ldloc_S, 6);
yield return LdcInstrs[Patch.beltSpeed[1]];
yield return new CodeInstruction(OpCodes.Bne_Un_S, label1);
yield return new CodeInstruction(OpCodes.Ldloca_S, 0);
yield return new CodeInstruction(OpCodes.Ldc_I4_2);
yield return new CodeInstruction(OpCodes.Stfld,
AccessTools.Field(typeof(ConnGizmoObj), nameof(ConnGizmoObj.color)));
yield return new CodeInstruction(OpCodes.Br, label2);
yield return new CodeInstruction(OpCodes.Ldloc_S, 6).WithLabels(label1);
yield return LdcInstrs[Patch.beltSpeed[0]];
yield return new CodeInstruction(OpCodes.Bne_Un_S, label2);
yield return new CodeInstruction(OpCodes.Ldloca_S, 0);
yield return new CodeInstruction(OpCodes.Ldc_I4_1);
yield return new CodeInstruction(OpCodes.Stfld,
AccessTools.Field(typeof(ConnGizmoObj), nameof(ConnGizmoObj.color)));
yield return new CodeInstruction(OpCodes.Nop).WithLabels(label2);
}
}
else
{
lastIsLdcI4_3 = instr.opcode == OpCodes.Ldc_I4_3;
yield return instr;
}
}
}
}