1

Source/WulaFallenEmpire/WorkGiver/WorkGiver_DeepDrill_WulaConstructor.cs

@@ -0,0 +1,98 @@
+using System.Collections.Generic;
+using RimWorld;
+using Verse;
+using Verse.AI;
+
+namespace WulaFallenEmpire
+{
+    // 扩展方法，用于检查特定种族
+    public static class WorkGiverExtensions
+    {
+        // 检查是否为指定种族
+        public static bool IsRace(this Pawn pawn, string raceDefName)
+        {
+            return pawn.def.defName == raceDefName;
+        }
+        
+        // 检查是否为指定种族之一
+        public static bool IsAnyRace(this Pawn pawn, params string[] raceDefNames)
+        {
+            string pawnRace = pawn.def.defName;
+            foreach (string race in raceDefNames)
+            {
+                if (pawnRace == race)
+                {
+                    return true;
+                }
+            }
+            return false;
+        }
+    }
+    
+    // 使用扩展方法的子类
+    public class WorkGiver_DeepDrill_WulaCatConstructor : WorkGiver_DeepDrill
+    {
+        private const string AllowedRace = "Mech_WULA_Cat_Constructor";
+        
+        public override bool ShouldSkip(Pawn pawn, bool forced = false)
+        {
+            
+            // 检查种族
+            if (!pawn.IsRace(AllowedRace))
+            {
+                return true;
+            }
+            
+            // 调用基类方法
+            bool shouldSkip = base.ShouldSkip(pawn, forced);
+            
+            return shouldSkip;
+        }
+        
+        public override bool HasJobOnThing(Pawn pawn, Thing t, bool forced = false)
+        {
+            // 检查种族
+            if (!pawn.IsRace(AllowedRace))
+            {
+                return false;
+            }
+            
+            // 调用基类方法
+            bool hasJob = base.HasJobOnThing(pawn, t, forced);
+            
+            return hasJob;
+        }
+        
+        // 可选：自定义工作优先级
+        public override float GetPriority(Pawn pawn, TargetInfo t)
+        {
+            if (!pawn.IsRace(AllowedRace))
+            {
+                return 0f; // 不是允许的种族，优先级为0
+            }
+            
+            // 如果是允许的种族，提高优先级
+            float basePriority = base.GetPriority(pawn, t);
+            return basePriority * 1.5f; // 提高50%优先级
+        }
+        
+        // 可选：提供自定义的工作描述
+        public override Job JobOnThing(Pawn pawn, Thing t, bool forced = false)
+        {
+            if (!pawn.IsRace(AllowedRace))
+            {
+                return null;
+            }
+            
+            
+            Job job = base.JobOnThing(pawn, t, forced);
+            if (job != null)
+            {
+                // 可以在这里添加自定义标签或其他修改
+                job.playerForced = forced;
+            }
+            
+            return job;
+        }
+    }
+}