work in progress

LogisticHub/Module/Miner.cs

@@ -1,7 +1,6 @@
 using System;
 using BepInEx.Configuration;
 using HarmonyLib;
-using UnityEngine;
 using UXAssist.Common;
 using Random = UnityEngine.Random;
 using GameLogicProc = UXAssist.Common.GameLogic;
@@ -72,7 +71,6 @@ public class Miner : PatchImpl<Miner>
 
     private static void OnGameBegin()
     {
-        VeinManager.Clear();
         _frame = 0L;
         UpdateMiningCostRate();
         UpdateSpeedScale();
@@ -135,7 +133,6 @@ public class Miner : PatchImpl<Miner>
         var frameCounter = _frame / 1200000L;
         if (frameCounter <= 0) return;
         _frame -= frameCounter * 1200000L;
-        // LogisticHub.Logger.LogDebug($"FrameCounter: {frameCounter}");
 
         DeepProfiler.BeginSample(DPEntry.Miner);
         var data = GameMain.data;
@@ -155,6 +152,7 @@ public class Miner : PatchImpl<Miner>
             {
                 foreach (var storageIndex in demands[itemIndex])
                 {
+                    LogisticHub.Logger.LogDebug($"StorageIndex: {storageIndex}");
                     var station = planetTransport.stationPool[storageIndex / 100];
                     if (station == null)
                         continue;
@@ -195,7 +193,7 @@ public class Miner : PatchImpl<Miner>
             for (var i = planetTransport.stationCursor - 1; i > 0; i--)
             {
                 var stationComponent = planetTransport.stationPool[i];
-                if (stationComponent.isCollector || stationComponent.isVeinCollector || stationComponent.energy * 2 >= stationComponent.energyMax) continue;
+                if (stationComponent == null || stationComponent.isCollector || stationComponent.isVeinCollector || stationComponent.energy * 2 >= stationComponent.energyMax) continue;
                 var index = (stationComponent.isStellar ? FuelIlsSlot.Value : FuelPlsSlot.Value) - 1;
                 var storage = stationComponent.storage;
                 if (index < 0 || index >= storage.Length)

LogisticHub/Module/StationManager.cs

@@ -54,7 +54,17 @@ public class StationManager : PatchImpl<StationManager>
 
     public static void Init()
     {
-        GameLogicProc.OnGameBegin += () =>
+        GameLogicProc.OnGameBegin += OnGameBegin;
+        Enable(true);
+    }
+
+    public static void Uninit()
+    {
+        GameLogicProc.OnGameBegin -= OnGameBegin;
+        Enable(false);
+    }
+
+    private static void OnGameBegin()
     {
         _stations = null;
         var data = GameMain.data;
@@ -73,13 +83,6 @@ public class StationManager : PatchImpl<StationManager>
                 UpdateStationInfo(stations, station);
             }
         }
-        };
-        Enable(true);
-    }
-
-    public static void Uninit()
-    {
-        Enable(false);
     }
 
     private static int ItemIdToIndex(int itemId)

LogisticHub/Module/VeinManager.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using HarmonyLib;
 using UXAssist.Common;
+using GameLogicProc = UXAssist.Common.GameLogic;
 
 namespace LogisticHub.Module;
 
@@ -19,16 +20,25 @@ public class VeinManager : PatchImpl<VeinManager>
     public static void Init()
     {
         Enable(true);
+        GameLogicProc.OnGameBegin += OnGameBegin;
     }
 
     public static void Uninit()
     {
+        GameLogicProc.OnGameBegin -= OnGameBegin;
         Enable(false);
     }
 
-    public static void Clear()
+    private static void OnGameBegin()
     {
         _veins = null;
+        var data = GameMain.data;
+        for (var index = data.factoryCount - 1; index >= 0; index--)
+        {
+            var factory = data.factories[index];
+            if (factory == null || factory.index != index) continue;
+            RecalcVeins(factory);
+        }
     }
 
     public static ProductVeinData[] GetVeins(int planetIndex)