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

Bug fix

Browse Source
This commit is contained in:
Soar Qin
2023-10-08 22:55:12 +08:00
parent 74c3c57f13
commit 9d601e1cbc
2 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
CheatEnabler/FactoryPatch.cs
View File

@@ -584,7 +584,7 @@ public static class FactoryPatch
}
}
private static class BeltSignalGenerator
public static class BeltSignalGenerator
{
private static Harmony _beltSignalPatch;
private static Dictionary<int, BeltSignal>[] _signalBelts;
@@ -826,6 +826,11 @@ public static class FactoryPatch
GetSignalBelts(factory)?.Remove(beltId);
}
public static void RemovePlanetSignalBelts(int factory)
{
GetSignalBelts(factory)?.Clear();
}
private static void RemoveSignalBeltPortalEnd(int factory, int beltId)
{
var v = ((long)factory << 32) | (uint)beltId;
@@ -843,6 +848,16 @@ public static class FactoryPatch
InitItemSources();
}
[HarmonyPostfix]
[HarmonyPatch(typeof(PlanetFactory), nameof(PlanetFactory.Init))]
private static void PlanetFactory_Init_Postfix(PlanetFactory __instance)
{
if (BeltSignalGeneratorEnabled.Value)
{
RemovePlanetSignalBelts(__instance.index);
}
}
[HarmonyPrefix]
[HarmonyPatch(typeof(CargoTraffic), nameof(CargoTraffic.RemoveBeltComponent))]
public static void CargoTraffic_RemoveBeltComponent_Prefix(int id)
@@ -963,6 +978,7 @@ public static class FactoryPatch
var productRegister = factoryProductionStat.productRegister;
var consumeRegister = factoryProductionStat.consumeRegister;
var countRecipe = BeltSignalCountRecipeEnabled.Value;
var cargoTraffic = factory.cargoTraffic;
foreach (var pair in belts)
{
var beltSignal = pair.Value;
@@ -972,7 +988,6 @@ public static class FactoryPatch
case 404:
{
var beltId = pair.Key;
var cargoTraffic = factory.cargoTraffic;
ref var belt = ref cargoTraffic.beltPool[beltId];
var cargoPath = cargoTraffic.GetCargoPath(belt.segPathId);
int itemId;
@@ -986,7 +1001,6 @@ public static class FactoryPatch
case 600:
{
if (!_portalTo.TryGetValue(beltSignal.SpeedLimit, out var set)) continue;
var cargoTraffic = factory.cargoTraffic;
var beltId = pair.Key;
ref var belt = ref cargoTraffic.beltPool[beltId];
var cargoPath = cargoTraffic.GetCargoPath(belt.segPathId);
@@ -1019,7 +1033,6 @@ public static class FactoryPatch
}
var beltId = pair.Key;
var cargoTraffic = factory.cargoTraffic;
ref var belt = ref cargoTraffic.beltPool[beltId];
var stack = beltSignal.Stack;
var inc = beltSignal.Inc;