暂存

Source/WulaFallenEmpire/CompChargingBed.cs

@@ -1,6 +1,6 @@
 using RimWorld;
+using System.Collections.Generic;
 using Verse;
-using System.Linq;
 
 namespace WulaFallenEmpire
 {
@@ -17,60 +17,53 @@ namespace WulaFallenEmpire
 
     public class CompChargingBed : ThingComp
     {
-        private Pawn lastOccupant;
-        private CompProperties_ChargingBed Props => (CompProperties_ChargingBed)props;
+        public CompProperties_ChargingBed Props => (CompProperties_ChargingBed)props;
+
+        private List<Pawn> chargingPawns = new List<Pawn>();
 
         public override void CompTick()
         {
             base.CompTick();
+            var bed = (Building_Bed)parent;
+            var occupants = new HashSet<Pawn>(bed.CurOccupants);
 
-            if (parent is Building_Bed bed)
+            for (int i = chargingPawns.Count - 1; i >= 0; i--)
             {
-                Pawn currentOccupant = bed.CurOccupants.FirstOrDefault();
-
-                if (currentOccupant != lastOccupant)
+                var pawn = chargingPawns[i];
+                if (!occupants.Contains(pawn))
                 {
-                    Log.Message($"[CompChargingBed] Occupant changed. Old: {lastOccupant?.Name.ToStringShort ?? "null"}, New: {currentOccupant?.Name.ToStringShort ?? "null"} on {parent.Label}");
-                }
-
-                // Pawn starts resting
-                if (currentOccupant != null && lastOccupant == null)
-                {
-                    if (IsWula(currentOccupant))
+                    var hediff = pawn.health.hediffSet.GetFirstHediffOfDef(Props.hediffDef);
+                    if (hediff != null)
                     {
-                        Log.Message($"[CompChargingBed] {currentOccupant.Name.ToStringShort} started resting. Applying hediff.");
-                        ApplyChargingHediff(currentOccupant);
+                        pawn.health.RemoveHediff(hediff);
+                    }
+                    chargingPawns.RemoveAt(i);
+                }
+            }
+
+            if (bed.AnyOccupants)
+            {
+                foreach (var pawn in bed.CurOccupants)
+                {
+                    if (pawn.def.defName == Props.raceDefName)
+                    {
+                        if (!pawn.health.hediffSet.HasHediff(Props.hediffDef))
+                        {
+                            pawn.health.AddHediff(Props.hediffDef);
+                            if (!chargingPawns.Contains(pawn))
+                            {
+                                chargingPawns.Add(pawn);
+                            }
+                        }
                     }
                 }
-                // Pawn stops resting
-                else if (currentOccupant == null && lastOccupant != null)
-                {
-                    // Logic to remove hediff is now in the JobDriver, but we can log the event.
-                    if (IsWula(lastOccupant))
-                    {
-                         Log.Message($"[CompChargingBed] {lastOccupant.Name.ToStringShort} stopped resting.");
-                    }
-                }
-                
-                lastOccupant = currentOccupant;
             }
         }
 
-        private bool IsWula(Pawn pawn)
+        public override void PostExposeData()
         {
-            return pawn.def.defName == Props.raceDefName || pawn.def.defName == (Props.raceDefName + "Real");
+            base.PostExposeData();
+            Scribe_Collections.Look(ref chargingPawns, "chargingPawns", LookMode.Reference);
         }
-
-        private void ApplyChargingHediff(Pawn pawn)
-        {
-            var powerComp = parent.GetComp<CompPowerTrader>();
-            Log.Message($"[CompChargingBed] Trying to apply hediff to {pawn.Name.ToStringShort}. PowerOn: {powerComp?.PowerOn}. HasHediff: {pawn.health.hediffSet.HasHediff(Props.hediffDef)}");
-            if (powerComp != null && powerComp.PowerOn && !pawn.health.hediffSet.HasHediff(Props.hediffDef))
-            {
-                Log.Message($"[CompChargingBed] Adding hediff to {pawn.Name.ToStringShort}.");
-                pawn.health.AddHediff(Props.hediffDef);
-            }
-        }
-
     }
 }

Source/WulaFallenEmpire/JobDriver_WulaLayDownToCharge.cs

@@ -5,40 +5,52 @@ using Verse.AI;
 
 namespace WulaFallenEmpire
 {
-    public class JobDriver_WulaLayDownToCharge : JobDriver_LayDown
+    public class JobDriver_WulaLayDownToCharge : JobDriver
     {
+        public override bool TryMakePreToilReservations(bool errorOnFailed)
+        {
+            return pawn.Reserve(job.targetA, job, 1, -1, null, errorOnFailed);
+        }
+
         protected override IEnumerable<Toil> MakeNewToils()
         {
             this.AddFinishAction(jobCondition =>
             {
-                Log.Message($"[JobDriver_WulaLayDownToCharge] Job finishing for {pawn.Name.ToStringShort} with condition {jobCondition}. Removing hediff.");
-                var hediff = pawn.health.hediffSet.GetFirstHediffOfDef(HediffDef.Named("WULA_ChargingHediff"));
+                var bed = (Building_Bed)job.targetA.Thing;
+                var comp = bed.GetComp<CompChargingBed>();
+                if (comp == null) return;
+                var hediff = pawn.health.hediffSet.GetFirstHediffOfDef(comp.Props.hediffDef);
                 if (hediff != null)
                 {
                     pawn.health.RemoveHediff(hediff);
-                    Log.Message($"[JobDriver_WulaLayDownToCharge] Hediff removed from {pawn.Name.ToStringShort}.");
-                }
-                else
-                {
-                    Log.Message($"[JobDriver_WulaLayDownToCharge] No hediff found on {pawn.Name.ToStringShort} to remove.");
                 }
             });
 
-            foreach (Toil toil in base.MakeNewToils())
-            {
-                yield return toil;
-            }
+            yield return Toils_Bed.GotoBed(TargetIndex.A);
 
-            var bed = (Building_Bed)job.targetA.Thing;
-            var powerComp = bed.GetComp<CompPowerTrader>();
-
-            var checkToil = new Toil
+            Toil layDownAndCharge = Toils_LayDown.LayDown(TargetIndex.A, true, false, false, false);
+            layDownAndCharge.AddPreInitAction(delegate
             {
-                tickAction = delegate
+                if (!pawn.health.hediffSet.HasHediff(HediffDef.Named("WULA_ChargingHediff")))
                 {
+                    var bed = (Building_Bed)job.targetA.Thing;
+                    var comp = bed.GetComp<CompChargingBed>();
+                    if (comp != null && !pawn.health.hediffSet.HasHediff(comp.Props.hediffDef))
+                    {
+                        pawn.health.AddHediff(comp.Props.hediffDef);
+                    }
+                }
+            });
+            
+            layDownAndCharge.tickAction = delegate
+            {
+                if (pawn.IsHashIntervalTick(60))
+                {
+                    var bed = (Building_Bed)job.targetA.Thing;
+                    var powerComp = bed.GetComp<CompPowerTrader>();
+
                     if (powerComp != null && !powerComp.PowerOn)
                     {
-                        Log.Message($"[JobDriver_WulaLayDownToCharge] Power lost for {pawn.Name.ToStringShort}. Ending job.");
                         EndJobWith(JobCondition.Incompletable);
                         return;
                     }
@@ -46,13 +58,12 @@ namespace WulaFallenEmpire
                     Need_WulaEnergy energyNeed = pawn.needs.TryGetNeed<Need_WulaEnergy>();
                     if (energyNeed != null && energyNeed.CurLevelPercentage >= 0.99f)
                     {
-                        Log.Message($"[JobDriver_WulaLayDownToCharge] {pawn.Name.ToStringShort} is fully charged. Ending job.");
                         EndJobWith(JobCondition.Succeeded);
                     }
-                },
-                defaultCompleteMode = ToilCompleteMode.Never
+                }
             };
-            yield return checkToil;
+            
+            yield return layDownAndCharge;
         }
     }
 }

Source/WulaFallenEmpire/JobGiver_WulaGetEnergy.cs

@@ -103,7 +103,7 @@ namespace WulaFallenEmpire
             Building_Bed bed = (Building_Bed)GenClosest.ClosestThingReachable(
                 pawn.Position,
                 pawn.Map,
-                ThingRequest.ForDef(ThingDefOf_WULA.WULA_Charging_Station_Synth),
+                ThingRequest.ForGroup(ThingRequestGroup.BuildingArtificial),
                 PathEndMode.InteractionCell,
                 TraverseParms.For(pawn),
                 9999f,
@@ -112,6 +112,9 @@ namespace WulaFallenEmpire
                     Building_Bed bed_internal = b as Building_Bed;
                     if (bed_internal == null) return false;
 
+                    var chargingComp = bed_internal.GetComp<CompChargingBed>();
+                    if (chargingComp == null) return false;
+
                     var powerComp = bed_internal.GetComp<CompPowerTrader>();
                     return !bed_internal.IsForbidden(pawn) &&
                            pawn.CanReserve(bed_internal) &&