refactor(CheatEnabler): split DysonSphereFunctions into focused helpers

2026-08-05 16:50:21 +08:00 · 2026-06-23 04:10:09 +08:00
parent eae4ba0376
commit 2dce3a7b93
6 changed files with 1664 additions and 1680 deletions
@@ -0,0 +1,133 @@
+using System;
+using System.Linq;
+using HarmonyLib;
+using UnityEngine;
+using UXAssist.Common;
+using UXAssist.Common.ModFeatures;
+
+namespace CheatEnabler.Functions.DysonSphere;
+
+[ModFeature("ShellCompletionFunctions")]
+public static class ShellCompletionFunctions
+{
+    public static void CompleteShellsInstantly()
+    {
+        var resolved = DysonSphereResolver.ResolveEditorOrLocalSphere();
+        if (resolved == null) return;
+        var (dysonSphere, star) = resolved.Value;
+
+        UIMessageBox.Show("CheatEnabler".Translate(), string.Format("This will complete all Dyson Sphere shells on \"{0}\". Are you sure?".Translate(), star.displayName), "取消".Translate(), "确定".Translate(), UIMessageBox.QUESTION, null, () =>
+        {
+            var totalNodeSpInfo = AccessTools.Field(typeof(DysonSphereLayer), "totalNodeSP");
+            var totalFrameSpInfo = AccessTools.Field(typeof(DysonSphereLayer), "totalFrameSP");
+            var totalCpInfo = AccessTools.Field(typeof(DysonSphereLayer), "totalCP");
+
+            var rocketCount = 0L;
+            var solarSailCount = 0L;
+            foreach (var dysonSphereLayer in dysonSphere.layersIdBased)
+            {
+                if (dysonSphereLayer == null) continue;
+                long totalNodeSp = 0;
+                long totalFrameSp = 0;
+                long totalCp = 0;
+                for (var i = dysonSphereLayer.frameCursor - 1; i >= 0; i--)
+                {
+                    var dysonFrame = dysonSphereLayer.framePool[i];
+                    if (dysonFrame == null || dysonFrame.id != i) continue;
+                    totalFrameSp += dysonFrame.spMax;
+                    var spMax = dysonFrame.spMax / 2;
+                    if (dysonFrame.spA < spMax)
+                    {
+                        rocketCount += spMax - dysonFrame.spA;
+                        dysonFrame.spA = spMax;
+                        dysonSphere.UpdateProgress(dysonFrame);
+                    }
+                    if (dysonFrame.spB < spMax)
+                    {
+                        rocketCount += spMax - dysonFrame.spB;
+                        dysonFrame.spB = spMax;
+                        dysonSphere.UpdateProgress(dysonFrame);
+                    }
+                }
+                for (var i = dysonSphereLayer.nodeCursor - 1; i >= 0; i--)
+                {
+                    var dysonNode = dysonSphereLayer.nodePool[i];
+                    if (dysonNode == null || dysonNode.id != i) continue;
+                    dysonNode.spOrdered = 0;
+                    dysonNode._spReq = 0;
+                    totalNodeSp += dysonNode.spMax;
+                    var diff = dysonNode.spMax - dysonNode.sp;
+                    if (diff > 0)
+                    {
+                        rocketCount += diff;
+                        dysonNode.sp = dysonNode.spMax;
+                        dysonSphere.UpdateProgress(dysonNode);
+                    }
+                    dysonNode._cpReq = 0;
+                    dysonNode.cpOrdered = 0;
+                    foreach (var shell in dysonNode.shells)
+                    {
+                        var nodeIndex = shell.nodeIndexMap[dysonNode.id];
+                        var cpMax = (shell.vertsqOffset[nodeIndex + 1] - shell.vertsqOffset[nodeIndex]) * shell.cpPerVertex;
+                        totalCp += cpMax;
+                        diff = cpMax - shell.nodecps[nodeIndex];
+                        shell.nodecps[nodeIndex] = cpMax;
+                        shell.nodecps[shell.nodecps.Length - 1] += diff;
+                        solarSailCount += diff;
+                        if (totalCpInfo != null)
+                        {
+                            shell.SetMaterialDynamicVars();
+                        }
+                    }
+                }
+
+                totalNodeSpInfo?.SetValue(dysonSphereLayer, totalNodeSp);
+                totalFrameSpInfo?.SetValue(dysonSphereLayer, totalFrameSp);
+                totalCpInfo?.SetValue(dysonSphereLayer, totalCp);
+            }
+            dysonSphere.CheckAutoNodes();
+
+            var productRegister = dysonSphere.productRegister;
+            if (productRegister != null)
+            {
+                lock (productRegister)
+                {
+                    var count = rocketCount;
+                    while (count > 0x40000000L)
+                    {
+                        productRegister[ProductionStatistics.DYSON_STRUCTURE_ID] += 0x40000000;
+                        count -= 0x40000000;
+                    }
+                    if (count > 0L) productRegister[ProductionStatistics.DYSON_STRUCTURE_ID] += (int)count;
+                    count = solarSailCount;
+                    while (count > 0x40000000L)
+                    {
+                        productRegister[ProductionStatistics.SOLAR_SAIL_ID] += 0x40000000;
+                        productRegister[ProductionStatistics.DYSON_CELL_ID] += 0x40000000;
+                        count -= 0x40000000;
+                    }
+                    if (count > 0L)
+                    {
+                        productRegister[ProductionStatistics.SOLAR_SAIL_ID] += (int)count;
+                        productRegister[ProductionStatistics.DYSON_CELL_ID] += (int)count;
+                    }
+                }
+            }
+            var consumeRegister = dysonSphere.consumeRegister;
+            if (consumeRegister != null)
+            {
+                lock (consumeRegister)
+                {
+                    var count = solarSailCount;
+                    while (count > 0x40000000L)
+                    {
+                        consumeRegister[ProductionStatistics.SOLAR_SAIL_ID] += 0x40000000;
+                        count -= 0x40000000;
+                    }
+                    if (count > 0L) consumeRegister[ProductionStatistics.SOLAR_SAIL_ID] += (int)count;
+                }
+            }
+            dysonSphere.needRecalculatePower = true;
+        });
+    }
+}