暂存

2025-11-22 16:58:29 +08:00
parent 87076bd957
commit 0536a94b01
11 changed files with 560 additions and 13 deletions
--- a/Source/WulaFallenEmpire/GlobalWorkTable/WorkGiver_GlobalWorkTable.cs
+++ b/Source/WulaFallenEmpire/GlobalWorkTable/WorkGiver_GlobalWorkTable.cs
@@ -0,0 +1,157 @@
+using RimWorld;
+using System.Collections.Generic;
+using System.Linq;
+using Verse;
+using Verse.AI;
+
+namespace WulaFallenEmpire
+{
+    public class WorkGiver_GlobalWorkTable : WorkGiver_Scanner
+    {
+        public override ThingRequest PotentialWorkThingRequest => ThingRequest.ForDef(ThingDef.Named("WULA_WeaponArmor_Productor"));
+        public override PathEndMode PathEndMode => PathEndMode.Touch;
+
+        public override bool HasJobOnThing(Pawn pawn, Thing t, bool forced = false)
+        {
+            if (!(t is Building_GlobalWorkTable table) || !table.Spawned || table.IsForbidden(pawn))
+            {
+                // Log.Message($"[WULA_DEBUG] HasJobOnThing: Target invalid or forbidden. {t}");
+                return false;
+            }
+
+            if (!pawn.CanReserve(table, 1, -1, null, forced))
+            {
+                // Log.Message($"[WULA_DEBUG] HasJobOnThing: Cannot reserve table.");
+                return false;
+            }
+
+            // 检查是否有需要收集材料的订单
+            var order = table.globalOrderStack.orders.FirstOrDefault(o => o.state == GlobalProductionOrder.ProductionState.Gathering && !o.paused);
+            if (order == null)
+            {
+                // Log.Message($"[WULA_DEBUG] HasJobOnThing: No gathering order found.");
+                return false;
+            }
+
+            // 检查是否已经有足够的材料在容器中或云端
+            if (order.HasEnoughResources())
+            {
+                // Log.Message($"[WULA_DEBUG] HasJobOnThing: Order has enough resources.");
+                return false;
+            }
+
+            // 查找所需材料
+            var ingredients = FindBestIngredients(pawn, table, order);
+            if (ingredients == null)
+            {
+                if (forced) Log.Message($"[WULA_DEBUG] HasJobOnThing: Could not find ingredients for {order.Label}.");
+                return false;
+            }
+            
+            return true;
+        }
+
+        public override Job JobOnThing(Pawn pawn, Thing t, bool forced = false)
+        {
+            if (!(t is Building_GlobalWorkTable table))
+                return null;
+
+            var order = table.globalOrderStack.orders.FirstOrDefault(o => o.state == GlobalProductionOrder.ProductionState.Gathering && !o.paused);
+            if (order == null)
+                return null;
+
+            var ingredients = FindBestIngredients(pawn, table, order);
+            if (ingredients == null)
+                return null;
+
+            Job job = JobMaker.MakeJob(DefDatabase<JobDef>.GetNamed("WULA_HaulToGlobalWorkTable"), t);
+            job.targetQueueB = ingredients.Select(i => new LocalTargetInfo(i.Key)).ToList();
+            job.countQueue = ingredients.Select(i => i.Value).ToList();
+            return job;
+        }
+
+        private List<KeyValuePair<Thing, int>> FindBestIngredients(Pawn pawn, Building_GlobalWorkTable table, GlobalProductionOrder order)
+        {
+            var result = new List<KeyValuePair<Thing, int>>();
+            var globalStorage = Find.World.GetComponent<GlobalStorageWorldComponent>();
+            
+            // 获取所需材料清单
+            var neededMaterials = GetNeededMaterials(order, table, globalStorage);
+            
+            // Log.Message($"[WULA_DEBUG] Needed materials for {order.Label}: {string.Join(", ", neededMaterials.Select(k => $"{k.Key.defName} x{k.Value}"))}");
+
+            foreach (var kvp in neededMaterials)
+            {
+                ThingDef def = kvp.Key;
+                int countNeeded = kvp.Value;
+
+                // 在地图上查找材料
+                // 注意：t.IsInAnyStorage() 可能会过滤掉放在地上的材料，如果玩家没有设置储存区
+                // 为了测试，先移除 IsInAnyStorage 限制，或者确保测试时材料在储存区
+                var things = pawn.Map.listerThings.ThingsOfDef(def)
+                    .Where(t => !t.IsForbidden(pawn) && pawn.CanReserve(t)) // 移除了 IsInAnyStorage() 以放宽条件
+                    .OrderBy(t => t.Position.DistanceTo(pawn.Position))
+                    .ToList();
+
+                int currentCount = 0;
+                foreach (var thing in things)
+                {
+                    int take = UnityEngine.Mathf.Min(thing.stackCount, countNeeded - currentCount);
+                    if (take > 0)
+                    {
+                        result.Add(new KeyValuePair<Thing, int>(thing, take));
+                        currentCount += take;
+                        if (currentCount >= countNeeded) break;
+                    }
+                }
+                
+                // Log.Message($"[WULA_DEBUG] Found {currentCount}/{countNeeded} of {def.defName}");
+            }
+
+            return result.Count > 0 ? result : null;
+        }
+
+        private Dictionary<ThingDef, int> GetNeededMaterials(GlobalProductionOrder order, Building_GlobalWorkTable table, GlobalStorageWorldComponent storage)
+        {
+            var needed = new Dictionary<ThingDef, int>();
+            
+            // 1. 计算总需求
+            var totalRequired = order.GetProductCostList();
+            if (totalRequired.Count == 0)
+            {
+                // 处理配方原料 (Ingredients) - 简化处理，假设配方只使用固定材料
+                // 实际情况可能更复杂，需要处理过滤器
+                foreach (var ingredient in order.recipe.ingredients)
+                {
+                    // 这里简化：只取第一个允许的物品作为需求
+                    // 更好的做法是动态匹配，但这需要更复杂的逻辑
+                    var def = ingredient.filter.AllowedThingDefs.FirstOrDefault();
+                    if (def != null)
+                    {
+                        int count = (int)ingredient.GetBaseCount();
+                        if (needed.ContainsKey(def)) needed[def] += count;
+                        else needed[def] = count;
+                    }
+                }
+            }
+
+            // 2. 减去云端已有的
+            foreach (var kvp in totalRequired)
+            {
+                int cloudCount = storage.GetInputStorageCount(kvp.Key);
+                int remaining = kvp.Value - cloudCount;
+                
+                // 3. 减去工作台容器中已有的
+                int containerCount = table.innerContainer.TotalStackCountOfDef(kvp.Key);
+                remaining -= containerCount;
+
+                if (remaining > 0)
+                {
+                    needed[kvp.Key] = remaining;
+                }
+            }
+
+            return needed;
+        }
+    }
+}