soarqin/DSP_Mods
1
0
Fork 0
mirror of https://github.com/soarqin/DSP_Mods.git synced 2025-12-08 20:13:29 +08:00
Code Activity

work in progress

Browse Source
This commit is contained in:
Soar Qin
2025-11-13 22:16:28 +08:00
parent 6f5b779d58
commit 1467a6010d
1 changed files with 81 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
UXAssist/Patches/FactoryPatch.cs
View File

@@ -2323,23 +2323,55 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
}
itemBundle = factory._tmp_items;
full = false;
if (GameMain.instance.timei < nextTimei) return;
nextTimei = GameMain.instance.timei + 12;
if (entityId == 0 || factory.entityPool[entityId].id != entityId)
{
return;
}
if (GameMain.instance.timei < nextTimei) return;
nextTimei = GameMain.instance.timei + 12;
ref var entityData = ref factory.entityPool[entityId];
if (entityData.beltId <= 0) return;
var cargoTraffic = factory.cargoTraffic;
ref var belt = ref cargoTraffic.beltPool[entityData.beltId];
if (belt.id != entityData.beltId) return;
var cargoPath = cargoTraffic.GetCargoPath(belt.segPathId);
var end = cargoPath.bufferLength - 1;
var buffer = cargoPath.buffer;
Dictionary<int, long> takeOutItems = [];
HashSet<int> pathIds = [belt.segPathId];
if (PressShiftToTakeWholeBeltItemsIncludeBranches.Value)
{
List<int> pendingPathIds = [belt.segPathId];
while (pendingPathIds.Count > 0)
{
var lastIndex = pendingPathIds.Count - 1;
var thisPathId = pendingPathIds[lastIndex];
pendingPathIds.RemoveAt(lastIndex);
var path = cargoTraffic.GetCargoPath(thisPathId);
if (path == null) continue;
foreach (var inputPathId in path.inputPaths)
{
if (pathIds.Contains(inputPathId)) continue;
pathIds.Add(inputPathId);
pendingPathIds.Add(inputPathId);
}
if (path.outputPath == null) continue;
var outputPathId = path.outputPath.id;
if (pathIds.Contains(outputPathId)) continue;
pathIds.Add(outputPathId);
pendingPathIds.Add(outputPathId);
}
}
var mainPlayer = factory.gameData.mainPlayer;
var factorySystem = factory.factorySystem;
Dictionary<int, long> takeOutItems = [];
foreach (var pathId in pathIds)
{
var cargoPath = cargoTraffic.GetCargoPath(pathId);
if (cargoPath == null) continue;
var end = cargoPath.bufferLength - 5;
var buffer = cargoPath.buffer;
if (PressShiftToTakeWholeBeltItemsIncludeInserters.Value)
{
foreach (var beltId in cargoPath.belts)
{
ref var b = ref cargoTraffic.beltPool[beltId];
@@ -2364,8 +2396,8 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
}
}
}
int i = 0;
while (i <= end)
}
for (var i = 0; i <= end;)
{
if (buffer[i] >= 246)
{
@@ -2378,8 +2410,9 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
takeOutItems[item] = (takeOutItems.TryGetValue(item, out var value) ? value : 0)
+ ((long)stack | ((long)inc << 32));
Array.Clear(buffer, i - 4, 10);
i += 10;
i += 6;
if (cargoPath.updateLen < i) cargoPath.updateLen = i;
i += 4;
cargoPath.cargoContainer.RemoveCargo(index);
}
else
@@ -2391,6 +2424,7 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
}
}
}
}
foreach (var kvp in takeOutItems)
{
var added = mainPlayer.TryAddItemToPackage(kvp.Key, (int)(kvp.Value & 0xFFFFFFFF), (int)(kvp.Value >> 32), true, entityId);