zc

Source/WulaFallenEmpire/BuildingComp/WULA_PathCostUpdater/CompPathCostUpdater.cs

@@ -0,0 +1,148 @@
+using System.Collections.Generic;
+using RimWorld;
+using Verse;
+using Verse.AI;
+
+namespace WulaFallenEmpire
+{
+    public class CompPathCostUpdater : ThingComp
+    {
+        private CompProperties_PathCostUpdater Props => (CompProperties_PathCostUpdater)props;
+        
+        // 记录是否需要更新路径成本
+        private bool needsPathUpdate = false;
+        
+        // 记录需要更新的区域
+        private CellRect updateRect;
+        
+        public override void PostSpawnSetup(bool respawningAfterLoad)
+        {
+            base.PostSpawnSetup(respawningAfterLoad);
+            
+            if (!respawningAfterLoad)
+            {
+                MarkForPathUpdate();
+            }
+        }
+        
+        public override void CompTick()
+        {
+            base.CompTick();
+            
+            // 每帧检查是否需要更新路径
+            if (needsPathUpdate && parent.Spawned)
+            {
+                UpdatePathCosts();
+                needsPathUpdate = false;
+            }
+        }
+        
+        /// <summary>
+        /// 标记需要更新路径成本
+        /// </summary>
+        public void MarkForPathUpdate()
+        {
+            if (!parent.Spawned)
+                return;
+                
+            needsPathUpdate = true;
+            
+            // 计算需要更新的区域
+            updateRect = parent.OccupiedRect();
+            
+            // 根据建筑大小决定扩展区域
+            int expandBy = CalculateExpandDistance();
+            updateRect = updateRect.ExpandedBy(expandBy);
+            updateRect = updateRect.ClipInsideMap(parent.Map);
+        }
+        
+        /// <summary>
+        /// 根据建筑大小计算需要扩展的距离
+        /// </summary>
+        private int CalculateExpandDistance()
+        {
+            if (Props == null || !Props.adaptiveExpansion)
+                return 1;
+                
+            // 根据建筑尺寸决定扩展距离
+            int size = parent.def.size.x * parent.def.size.z;
+            
+            if (size <= 1)      // 1x1 建筑
+                return 1;
+            else if (size <= 4) // 2x2 或 1x4 建筑
+                return 2;
+            else if (size <= 9) // 3x3 建筑
+                return 3;
+            else                // 大型建筑
+                return 4;
+        }
+        
+        /// <summary>
+        /// 更新路径成本
+        /// </summary>
+        public void UpdatePathCosts()
+        {
+            if (!parent.Spawned)
+                return;
+                
+            Map map = parent.Map;
+            
+            // 使用map.pathing来更新路径成本
+            // 根据Pathing.cs的RecalculatePerceivedPathCostUnderThing方法，我们可以直接调用它
+            map.pathing.RecalculatePerceivedPathCostUnderThing(parent);
+            
+            // 清理可达性缓存
+            ClearReachabilityCache(map);
+        }
+        
+        /// <summary>
+        /// 清理可达性缓存
+        /// </summary>
+        private void ClearReachabilityCache(Map map)
+        {
+            // 清理整个地图的可达性缓存
+            map.reachability.ClearCache();
+        }
+        
+        /// <summary>
+        /// 强制立即更新路径（用于特殊事件）
+        /// </summary>
+        public void ForceImmediateUpdate()
+        {
+            if (parent.Spawned)
+            {
+                UpdatePathCosts();
+                needsPathUpdate = false;
+            }
+        }
+        
+        public override IEnumerable<Gizmo> CompGetGizmosExtra()
+        {
+            foreach (var gizmo in base.CompGetGizmosExtra())
+            {
+                yield return gizmo;
+            }
+            
+            if (DebugSettings.godMode)
+            {
+                yield return new Command_Action
+                {
+                    defaultLabel = "DEBUG: Update Path Costs",
+                    defaultDesc = "Force update path costs for this building",
+                    action = ForceImmediateUpdate
+                };
+            }
+        }
+    }
+    
+    public class CompProperties_PathCostUpdater : CompProperties
+    {
+        // 是否启用自适应扩展（根据建筑大小决定更新区域）
+        public bool adaptiveExpansion = true;
+        
+        public CompProperties_PathCostUpdater()
+        {
+            compClass = typeof(CompPathCostUpdater);
+        }
+    }
+}