1

Source/WulaFallenEmpire/Pawn_Comps/AutonomousMech/ThinkNode_ConditionalAutonomousWorkMode.cs

@@ -0,0 +1,36 @@
+using RimWorld;
+using Verse;
+using Verse.AI;
+
+namespace WulaFallenEmpire
+{
+    public class ThinkNode_ConditionalAutonomousWorkMode : ThinkNode_Conditional
+    {
+        public MechWorkModeDef requiredMode;
+        
+        protected override bool Satisfied(Pawn pawn)
+        {
+            try
+            {
+                if (pawn == null || pawn.Dead || pawn.Downed)
+                    return false;
+                
+                // 检查是否有自主机械组件
+                var comp = pawn.GetComp<CompAutonomousMech>();
+                if (comp == null)
+                    return false;
+                
+                // 检查当前工作模式是否匹配要求
+                if (comp.CurrentWorkMode != requiredMode)
+                    return false;
+                
+                return true;
+            }
+            catch (System.Exception ex)
+            {
+                WulaLog.Debug($"[WULA] Exception in ThinkNode_ConditionalAutonomousWorkMode.Satisfied for pawn {pawn?.LabelShort}: {ex}");
+                return false;
+            }
+        }
+    }
+}