暂存充电床

Source/WulaFallenEmpire/JobGiver_WulaGetEnergy.cs

@@ -1,4 +1,5 @@
 using RimWorld;
+using System.Linq;
 using Verse;
 using Verse.AI;
 
@@ -12,8 +13,14 @@ namespace WulaFallenEmpire
 
         public override float GetPriority(Pawn pawn)
         {
+            if (pawn.health.hediffSet.HasHediff(DefDatabase<HediffDef>.GetNamed("WULA_ChargingHediff")))
+            {
+                Log.Message($"[JobGiver_WulaGetEnergy] {pawn.Name.ToStringShort} already has charging hediff. Priority 0.");
+                return 0f;
+            }
+            
             Need_WulaEnergy energyNeed = pawn.needs.TryGetNeed<Need_WulaEnergy>();
-            if (energyNeed == null || pawn.health.hediffSet.HasHediff(DefDatabase<HediffDef>.GetNamed("WULA_ChargingHediff")))
+            if (energyNeed == null)
             {
                 return 0f;
             }
@@ -27,22 +34,33 @@ namespace WulaFallenEmpire
 
         protected override Job TryGiveJob(Pawn pawn)
         {
+            Log.Message($"[JobGiver_WulaGetEnergy] Trying to give job to {pawn.Name.ToStringShort}.");
+
             if (pawn.health.hediffSet.HasHediff(DefDatabase<HediffDef>.GetNamed("WULA_ChargingHediff")))
             {
+                Log.Message($"[JobGiver_WulaGetEnergy] {pawn.Name.ToStringShort} already has charging hediff. Job cancelled.");
                 return null;
             }
 
             Need_WulaEnergy energyNeed = pawn.needs.TryGetNeed<Need_WulaEnergy>();
             if (energyNeed == null || energyNeed.CurLevelPercentage >= maxEnergyLevelPercentage)
             {
+                Log.Message($"[JobGiver_WulaGetEnergy] Energy level for {pawn.Name.ToStringShort} is sufficient. Job cancelled.");
                 return null;
             }
 
             if (!TryFindBestEnergySourceFor(pawn, out Thing energySource))
             {
+                Log.Message($"[JobGiver_WulaGetEnergy] No energy source found for {pawn.Name.ToStringShort}. Job cancelled.");
                 return null;
             }
 
+            if (energySource is Building_Bed)
+            {
+                Log.Message($"[JobGiver_WulaGetEnergy] Found bed for {pawn.Name.ToStringShort}. Creating WULA_LayDownToCharge job.");
+                return JobMaker.MakeJob(DefDatabase<JobDef>.GetNamed("WULA_LayDownToCharge"), energySource);
+            }
+
             Job job = JobMaker.MakeJob(DefDatabase<JobDef>.GetNamed("WULA_IngestWulaEnergy"), energySource);
             job.count = 1;
             return job;
@@ -50,6 +68,13 @@ namespace WulaFallenEmpire
 
         private bool TryFindBestEnergySourceFor(Pawn pawn, out Thing energySource)
         {
+            // 优先寻找可用的充电床
+            energySource = FindChargingBed(pawn);
+            if (energySource != null)
+            {
+                return true;
+            }
+
             // 优先从背包中寻找
             Thing thing = pawn.inventory.innerContainer.FirstOrFallback(t => t.def.GetModExtension<ThingDefExtension_EnergySource>() != null && t.IngestibleNow);
             if (thing != null)
@@ -71,5 +96,33 @@ namespace WulaFallenEmpire
 
             return energySource != null;
         }
+
+        private Building_Bed FindChargingBed(Pawn pawn)
+        {
+            // 寻找附近可用的 WULA_Charging_Station_Synth
+            Building_Bed bed = (Building_Bed)GenClosest.ClosestThingReachable(
+                pawn.Position,
+                pawn.Map,
+                ThingRequest.ForDef(ThingDefOf_WULA.WULA_Charging_Station_Synth),
+                PathEndMode.InteractionCell,
+                TraverseParms.For(pawn),
+                9999f,
+                b =>
+                {
+                    Building_Bed bed_internal = b as Building_Bed;
+                    if (bed_internal == null) return false;
+
+                    var powerComp = bed_internal.GetComp<CompPowerTrader>();
+                    return !bed_internal.IsForbidden(pawn) &&
+                           pawn.CanReserve(bed_internal) &&
+                           !bed_internal.Medical &&
+                           !bed_internal.IsBurning() &&
+                           powerComp != null &&
+                           powerComp.PowerOn &&
+                           !bed_internal.CurOccupants.Any();
+                }
+            );
+            return bed;
+        }
     }
 }