From 0dafd8a763fe50bf56db1adab7f5486574daea52 Mon Sep 17 00:00:00 2001 From: Soar Qin Date: Fri, 30 Aug 2024 19:20:32 +0800 Subject: [PATCH] fix Registers --- CheatEnabler/DysonSphereFunctions.cs | 14 +++++++++++++- UXAssist/DysonSpherePatch.cs | 12 +++++++++++- 2 files changed, 24 insertions(+), 2 deletions(-) diff --git a/CheatEnabler/DysonSphereFunctions.cs b/CheatEnabler/DysonSphereFunctions.cs index ec7b4f9..3c800d7 100644 --- a/CheatEnabler/DysonSphereFunctions.cs +++ b/CheatEnabler/DysonSphereFunctions.cs @@ -113,7 +113,19 @@ public static class DysonSphereFunctions lock (productRegister) { if (rocketCount > 0) productRegister[11902] += rocketCount; - if (solarSailCount > 0) productRegister[11903] += solarSailCount; + if (solarSailCount > 0) + { + productRegister[11901] += solarSailCount; + productRegister[11903] += solarSailCount; + } + } + } + var consumeRegister = dysonSphere.consumeRegister; + if (consumeRegister != null) + { + lock (consumeRegister) + { + if (solarSailCount > 0) consumeRegister[11901] += solarSailCount; } } }); diff --git a/UXAssist/DysonSpherePatch.cs b/UXAssist/DysonSpherePatch.cs index 898cd84..da9bb0e 100644 --- a/UXAssist/DysonSpherePatch.cs +++ b/UXAssist/DysonSpherePatch.cs @@ -233,12 +233,22 @@ public static class DysonSpherePatch dysonNode.shellTurn = shellIndex; + var solarSailCount = todoCount - count; productRegister = __instance.productRegister; if (productRegister != null) { lock (productRegister) { - productRegister[11903] += todoCount - count; + productRegister[11901] += solarSailCount; + productRegister[11903] += solarSailCount; + } + } + var consumeRegister = __instance.consumeRegister; + if (consumeRegister != null) + { + lock (consumeRegister) + { + consumeRegister[11901] += solarSailCount; } } }