135 lines
4.4 KiB
C#
135 lines
4.4 KiB
C#
using RimWorld;
|
|
using Verse;
|
|
using Verse.AI;
|
|
|
|
namespace WulaFallenEmpire
|
|
{
|
|
public class WorkGiver_Warden_DeliverEnergy : WorkGiver_Scanner
|
|
{
|
|
private WorkGiverDefExtension_FeedWula ext;
|
|
|
|
private WorkGiverDefExtension_FeedWula Ext
|
|
{
|
|
get
|
|
{
|
|
if (ext == null)
|
|
{
|
|
ext = def.GetModExtension<WorkGiverDefExtension_FeedWula>();
|
|
}
|
|
return ext;
|
|
}
|
|
}
|
|
|
|
public override ThingRequest PotentialWorkThingRequest => ThingRequest.ForDef(ThingDef.Named("WulaSpecies"));
|
|
|
|
public override PathEndMode PathEndMode => PathEndMode.ClosestTouch;
|
|
|
|
public override Danger MaxPathDanger(Pawn pawn) => Danger.Deadly;
|
|
|
|
public override bool HasJobOnThing(Pawn pawn, Thing t, bool forced = false)
|
|
{
|
|
Pawn prisoner = t as Pawn;
|
|
|
|
if (prisoner == null || prisoner == pawn || !prisoner.IsPrisonerOfColony || !prisoner.guest.CanBeBroughtFood)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
Need_WulaEnergy energyNeed = prisoner.needs.TryGetNeed<Need_WulaEnergy>();
|
|
if (energyNeed == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (energyNeed.CurLevelPercentage > Ext.feedThreshold)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (WardenFeedUtility.ShouldBeFed(prisoner))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!pawn.CanReserveAndReach(prisoner, PathEndMode.Touch, Danger.Deadly, 1, -1, null, forced))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (Ext == null || Ext.energySourceDef == null)
|
|
{
|
|
Log.ErrorOnce("WorkGiver_Warden_DeliverEnergy is missing the DefModExtension with a valid energySourceDef.", def.GetHashCode());
|
|
return false;
|
|
}
|
|
|
|
if (!FindBestEnergySourceFor(pawn, prisoner, out _, out _))
|
|
{
|
|
JobFailReason.Is("NoFood".Translate());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public override Job JobOnThing(Pawn pawn, Thing t, bool forced = false)
|
|
{
|
|
Pawn prisoner = (Pawn)t;
|
|
if (FindBestEnergySourceFor(pawn, prisoner, out Thing energySource, out _))
|
|
{
|
|
Job job = JobMaker.MakeJob(JobDefOf.DeliverFood, energySource, prisoner);
|
|
job.count = 1;
|
|
job.targetC = RCellFinder.SpotToChewStandingNear(prisoner, energySource);
|
|
return job;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private bool FindBestEnergySourceFor(Pawn getter, Pawn eater, out Thing foodSource, out ThingDef foodDef)
|
|
{
|
|
foodSource = null;
|
|
foodDef = null;
|
|
|
|
if (Ext == null || Ext.energySourceDef == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// Check if there's already an energy source in the eater's room that the eater can reach and use.
|
|
Thing existingEnergyInRoom = GenClosest.ClosestThingReachable(
|
|
eater.Position, // Start search from eater's position
|
|
eater.Map,
|
|
ThingRequest.ForDef(Ext.energySourceDef),
|
|
PathEndMode.OnCell,
|
|
TraverseParms.For(eater, Danger.Deadly, TraverseMode.ByPawn, false), // Use eater's traverse parms
|
|
9999f,
|
|
(Thing x) => !x.IsForbidden(eater) && eater.CanReserve(x) && x.GetRoom() == eater.GetRoom()
|
|
);
|
|
|
|
if (existingEnergyInRoom != null)
|
|
{
|
|
// If there's already an energy source in the room, no need for the warden to bring another.
|
|
return false;
|
|
}
|
|
|
|
// Search for an energy source anywhere, now that we've confirmed none are in the room.
|
|
foodSource = GenClosest.ClosestThingReachable(
|
|
getter.Position,
|
|
getter.Map,
|
|
ThingRequest.ForDef(Ext.energySourceDef),
|
|
PathEndMode.OnCell,
|
|
TraverseParms.For(getter),
|
|
9999f,
|
|
(Thing x) => !x.IsForbidden(getter) && getter.CanReserve(x)
|
|
);
|
|
|
|
if (foodSource != null)
|
|
{
|
|
foodDef = foodSource.def;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|