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; } } }