add LogisticMiner (idea from PlanetMiner)

2025-12-08 21:33:28 +08:00 · 2022-08-22 21:53:53 +08:00
parent 16dcf732b3
commit 4a88a6a8bf
5 changed files with 292 additions and 32 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -0,0 +1,6 @@
+# Rider project files
+/.idea/
+
+# C# generated folders
+bin/
+obj/
--- a/DSP_Mods.sln
+++ b/DSP_Mods.sln
@@ -2,6 +2,8 @@
 Microsoft Visual Studio Solution File, Format Version 12.00
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "DevShortcuts", "DevShortcuts\DevShortcuts.csproj", "{F9F16B62-D1D3-466B-BE22-E64B9EA957C2}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "LogisticMiner", "LogisticMiner\LogisticMiner.csproj", "{7149D717-C913-4153-9425-38CB9D087F83}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -12,5 +14,9 @@ Global
 		{F9F16B62-D1D3-466B-BE22-E64B9EA957C2}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{F9F16B62-D1D3-466B-BE22-E64B9EA957C2}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{F9F16B62-D1D3-466B-BE22-E64B9EA957C2}.Release|Any CPU.Build.0 = Release|Any CPU
+		{7149D717-C913-4153-9425-38CB9D087F83}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{7149D717-C913-4153-9425-38CB9D087F83}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{7149D717-C913-4153-9425-38CB9D087F83}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{7149D717-C913-4153-9425-38CB9D087F83}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 EndGlobal
--- a/DevShortcuts/DevShortcuts.cs
+++ b/DevShortcuts/DevShortcuts.cs
@@ -1,44 +1,41 @@
 using BepInEx;
 using HarmonyLib;

-namespace DevShortcuts
+namespace DevShortcuts;
+
+[BepInPlugin(PluginInfo.PLUGIN_GUID, PluginInfo.PLUGIN_NAME, PluginInfo.PLUGIN_VERSION)]
+public class DevShortcuts : BaseUnityPlugin
 {
-    [BepInPlugin(PluginInfo.PLUGIN_GUID, PluginInfo.PLUGIN_NAME, PluginInfo.PLUGIN_VERSION)]
-    public class DevShortcuts : BaseUnityPlugin
+    private void Awake()
    {
-        private void Awake()
-        {
-            // Plugin startup logic
-            Logger.LogInfo($"Plugin {PluginInfo.PLUGIN_GUID} is loaded!");
-        }
+        Logger.LogInfo($"Plugin {PluginInfo.PLUGIN_GUID} is loaded!");
+    }

-        private void Start()
-        {
-            Harmony.CreateAndPatchAll(typeof(DevShortcuts));
-        }
+    private void Start()
+    {
+        Harmony.CreateAndPatchAll(typeof(DevShortcuts));
+    }

-        [HarmonyPostfix]
-        [HarmonyPatch(typeof(PlayerController), "Init")]
-        static void PlayerControllerInit(ref PlayerAction[] ___actions, Player ___player)
+    [HarmonyPostfix]
+    [HarmonyPatch(typeof(PlayerController), "Init")]
+    static void PlayerControllerInit(ref PlayerAction[] ___actions, Player ___player)
+    {
+        var cnt = ___actions.Length;
+        var newActions = new PlayerAction[cnt + 1];
+        for (int i = 0; i < cnt; i++)
        {
-            var cnt = ___actions.Length;
-            var newActions = new PlayerAction[cnt + 1];
-            for (int i = 0; i < cnt; i++)
-            {
-                newActions[i] = ___actions[i];
-            }
-            var test = new PlayerAction_Test();
-            test.Init(___player);
-            newActions[cnt] = test;
-            ___actions = null;
-            ___actions = newActions;
+            newActions[i] = ___actions[i];
        }
+        var test = new PlayerAction_Test();
+        test.Init(___player);
+        newActions[cnt] = test;
+        ___actions = newActions;
+    }

-        [HarmonyPostfix]
-        [HarmonyPatch(typeof(PlayerAction_Test), "GameTick")]
-        static void PlayerAction_TestGameTick(PlayerAction_Test __instance, long timei)
-        {
-            __instance.Update();
-        }
+    [HarmonyPostfix]
+    [HarmonyPatch(typeof(PlayerAction_Test), "GameTick")]
+    static void PlayerAction_TestGameTick(PlayerAction_Test __instance, long timei)
+    {
+        __instance.Update();
    }
 }
--- a/LogisticMiner/LogisticMiner.cs
+++ b/LogisticMiner/LogisticMiner.cs
@@ -0,0 +1,227 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using BepInEx;
+using HarmonyLib;
+using Random = UnityEngine.Random;
+
+namespace LogisticMiner;
+
+[BepInPlugin(PluginInfo.PLUGIN_GUID, PluginInfo.PLUGIN_NAME, PluginInfo.PLUGIN_VERSION)]
+public class LogisticMiner : BaseUnityPlugin
+{
+    private static LogisticMiner _this;
+    private const int VeinEnergyRatio = 200000;
+    private const int WaterEnergyRatio = 50000;
+    private const int WaterSpeed = 100;
+
+    private static float _frame, _nextFrame;
+    private static float _miningCostRate;
+    private static uint _miningCostBarrier;
+
+    private static uint _seed = (uint)Random.Range(int.MinValue, int.MaxValue);
+    private static readonly Dictionary<int, List<int>> Veins = new();
+
+    private void Awake()
+    {
+        Logger.LogInfo($"Plugin {PluginInfo.PLUGIN_GUID} is loaded!");
+        _this = this;
+    }
+
+    private void Start()
+    {
+        Harmony.CreateAndPatchAll(typeof(LogisticMiner));
+    }
+
+    private void FixedUpdate()
+    {
+        _frame++;
+    }
+
+    [HarmonyPostfix]
+    [HarmonyPatch(typeof(PlanetFactory), "Init")]
+    [HarmonyPatch(typeof(PlanetFactory), "RecalculateVeinGroup")]
+    [HarmonyPatch(typeof(PlanetFactory), "RecalculateAllVeinGroups")]
+    private static void NeedRecalcVeins(PlanetFactory __instance)
+    {
+        VeinData[] veinPool = __instance.veinPool;
+        Veins.Clear();
+        for (var i = 0; i < veinPool.Length; i++)
+        {
+            var veinData = veinPool[i];
+            if (veinData.amount > 0 && veinData.type > EVeinType.None)
+            {
+                AddVeinData(__instance.planetId, veinData.productId, i);
+            }
+        }
+    }
+
+    [HarmonyPostfix]
+    [HarmonyPatch(typeof(FactorySystem), "GameTickLabResearchMode")]
+    private static void Miner(FactorySystem __instance)
+    {
+        if (_nextFrame > _frame)
+            return;
+        var history = GameMain.history;
+        var miningSpeedScale = history.miningSpeedScale;
+        if (miningSpeedScale <= 0f)
+            return;
+        var miningCostRate = history.miningCostRate;
+        if (!miningCostRate.Equals(_miningCostRate))
+        {
+            _miningCostRate = miningCostRate;
+            _miningCostBarrier = (uint)(int)Math.Ceiling(2147483646.0 * miningCostRate);
+        }
+
+        var miningFrames = 60f / miningSpeedScale;
+        var factory = __instance.factory;
+        var key0 = factory.planetId << 16;
+        var veinPool = factory.veinPool;
+        var planetTransport = __instance.planet.factory.transport;
+        var factoryProductionStat =
+            GameMain.statistics.production.factoryStatPool[__instance.factory.index];
+        var productRegister = factoryProductionStat?.productRegister;
+        do
+        {
+            for (var j = 1; j < planetTransport.stationCursor; j++)
+            {
+                var stationComponent = planetTransport.stationPool[j];
+                if (stationComponent == null) continue;
+                if (stationComponent.isCollector || stationComponent.isVeinCollector) continue;
+                var storage = stationComponent.storage;
+                if (storage == null) continue;
+                var isCollecting = false;
+                for (var k = 0; k < stationComponent.storage.Length; k++)
+                {
+                    ref var stationStore = ref storage[k];
+                    if (stationStore.localLogic != ELogisticStorage.Demand ||
+                        stationStore.max <= stationStore.count)
+                        continue;
+
+                    var isVein = Veins.TryGetValue(key0 | stationStore.itemId, out var val);
+                    var isVeinOrWater = isVein || stationStore.itemId == __instance.planet.waterItemId;
+                    if (!isVeinOrWater) continue;
+                    int amount;
+                    int energyRatio;
+                    if (isVein)
+                    {
+                        if (stationComponent.energy < VeinEnergyRatio)
+                        {
+                            isCollecting = true;
+                            k = int.MaxValue - 1;
+                            continue;
+                        }
+
+                        if (veinPool[val.First()].type == EVeinType.Oil)
+                            amount = (int)val
+                                .Where(item => veinPool.Length > item && veinPool[item].type > EVeinType.None)
+                                .Sum(item => veinPool[item].amount / 6000f);
+                        else
+                            amount = val.Sum(
+                                item2 => GetMine(veinPool, item2, __instance.planet.factory) ? 1 : 0);
+                        energyRatio = VeinEnergyRatio;
+                    }
+                    else
+                    {
+                        if (stationComponent.energy < WaterEnergyRatio)
+                        {
+                            isCollecting = true;
+                            k = int.MaxValue - 1;
+                            continue;
+                        }
+
+                        amount = WaterSpeed;
+                        energyRatio = WaterEnergyRatio;
+                    }
+
+                    if (amount == 0) continue;
+                    isCollecting = true;
+                    var energyConsume = energyRatio * amount;
+                    if (energyConsume > stationComponent.energy)
+                    {
+                        amount = (int)(stationComponent.energy / energyRatio);
+                        energyConsume = energyRatio * amount;
+                    }
+
+                    stationStore.count += amount;
+                    if (factoryProductionStat != null)
+                        productRegister[stationStore.itemId] += amount;
+                    stationComponent.energy -= energyConsume;
+                }
+
+                if (isCollecting && stationComponent.energyMax > stationComponent.energy * 2)
+                {
+                    var index = storage.Length - 2;
+                    var fuelCount = storage[index].count;
+                    if (fuelCount > 0)
+                    {
+                        var heatValue = LDB.items.Select(storage[index].itemId).HeatValue;
+                        if (heatValue > 0)
+                        {
+                            var count = (int)((stationComponent.energyMax - stationComponent.energy) /
+                                              heatValue);
+                            if (count > fuelCount)
+                                count = fuelCount;
+                            storage[index].count -= count;
+                            stationComponent.energy += count * heatValue;
+                        }
+                    }
+                }
+            }
+
+            _nextFrame += miningFrames;
+        } while (_nextFrame <= _frame);
+
+        if (_frame >= 1000000f)
+        {
+            _frame -= 1000000f;
+            _nextFrame -= 1000000f;
+        }
+    }
+
+    private static void AddVeinData(int planetId, int productId, int index)
+    {
+        var key = (planetId << 16) + productId;
+        if (Veins.TryGetValue(key, out var val))
+            val.Add(index);
+        else
+            Veins.Add(key, new List<int> { index });
+    }
+
+    private static bool GetMine(VeinData[] veinDatas, int index, PlanetFactory factory)
+    {
+        if (veinDatas.Length == 0 || veinDatas[index].type == EVeinType.None)
+            return false;
+
+        if (veinDatas[index].amount > 0)
+        {
+            bool flag = true;
+            if (_miningCostBarrier < 2147483646u)
+            {
+                _seed = (uint)((int)((ulong)((_seed % 2147483646u + 1) * 48271L) % 2147483647uL) - 1);
+                flag = _seed < _miningCostBarrier;
+            }
+
+            if (flag)
+            {
+                veinDatas[index].amount--;
+                factory.veinGroups[veinDatas[index].groupIndex].amount--;
+                if (veinDatas[index].amount <= 0)
+                {
+                    short groupIndex = veinDatas[index].groupIndex;
+                    factory.veinGroups[groupIndex].count--;
+                    factory.RemoveVeinWithComponents(index);
+                    factory.RecalculateVeinGroup(groupIndex);
+                }
+            }
+
+            return true;
+        }
+
+        short groupIndex2 = veinDatas[index].groupIndex;
+        factory.veinGroups[groupIndex2].count--;
+        factory.RemoveVeinWithComponents(index);
+        factory.RecalculateVeinGroup(groupIndex2);
+        return false;
+    }
+}
--- a/LogisticMiner/LogisticMiner.csproj
+++ b/LogisticMiner/LogisticMiner.csproj
@@ -0,0 +1,24 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <TargetFramework>netstandard2.0</TargetFramework>
+    <AssemblyName>LogisticMiner</AssemblyName>
+    <BepInExPluginGuid>org.soardev.logisticminer</BepInExPluginGuid>
+    <Description>DSP MOD - LogisticMiner</Description>
+    <Version>1.0.0</Version>
+    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
+    <LangVersion>latest</LangVersion>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <PackageReference Include="BepInEx.Analyzers" Version="1.*" PrivateAssets="all" />
+    <PackageReference Include="BepInEx.Core" Version="5.*" />
+    <PackageReference Include="BepInEx.PluginInfoProps" Version="1.*" />
+    <PackageReference Include="DysonSphereProgram.GameLibs" Version="0.9.26.13026-r.0" />
+    <PackageReference Include="UnityEngine.Modules" Version="2018.4.12" IncludeAssets="compile" />
+  </ItemGroup>
+  
+  <ItemGroup Condition="'$(TargetFramework.TrimEnd(`0123456789`))' == 'net'">
+    <PackageReference Include="Microsoft.NETFramework.ReferenceAssemblies" Version="1.0.2" PrivateAssets="all" />
+  </ItemGroup>
+</Project>