74 lines
2.3 KiB
C#
74 lines
2.3 KiB
C#
// WorkGiver_RepairMech.cs
|
|
using RimWorld;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Verse;
|
|
using Verse.AI;
|
|
|
|
namespace WulaFallenEmpire
|
|
{
|
|
public class WorkGiver_RepairMech : WorkGiver_Scanner
|
|
{
|
|
public override PathEndMode PathEndMode => PathEndMode.Touch;
|
|
|
|
public override Danger MaxPathDanger(Pawn pawn)
|
|
{
|
|
return Danger.Deadly;
|
|
}
|
|
|
|
public override IEnumerable<Thing> PotentialWorkThingsGlobal(Pawn pawn)
|
|
{
|
|
if (pawn.Faction != Faction.OfPlayer || pawn.Map == null)
|
|
return Enumerable.Empty<Thing>();
|
|
|
|
// 获取所有需要维修的玩家机甲
|
|
return pawn.Map.mapPawns.AllPawnsSpawned
|
|
.Where(p =>
|
|
p.Faction == Faction.OfPlayer &&
|
|
p.health != null &&
|
|
!p.Dead &&
|
|
p.TryGetComp<CompMechRepairable>()?.CanAutoRepair == true
|
|
)
|
|
.Cast<Thing>();
|
|
}
|
|
|
|
public override bool ShouldSkip(Pawn pawn, bool forced = false)
|
|
{
|
|
if (pawn.story != null && pawn.WorkTagIsDisabled(WorkTags.Crafting))
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
public override bool HasJobOnThing(Pawn pawn, Thing t, bool forced = false)
|
|
{
|
|
if (!(t is Pawn mech) || mech.Dead)
|
|
return false;
|
|
|
|
var repairableComp = t.TryGetComp<CompMechRepairable>();
|
|
if (repairableComp == null || !repairableComp.CanAutoRepair)
|
|
return false;
|
|
|
|
if (!repairableComp.NeedsRepair)
|
|
return false;
|
|
|
|
if (pawn.Faction != Faction.OfPlayer)
|
|
return false;
|
|
|
|
if (!pawn.CanReserveAndReach(t, PathEndMode.Touch, Danger.Some, 1, -1, null, forced))
|
|
return false;
|
|
|
|
// 检查工作标签
|
|
if (pawn.story != null && pawn.WorkTagIsDisabled(WorkTags.Crafting))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
public override Job JobOnThing(Pawn pawn, Thing t, bool forced = false)
|
|
{
|
|
return JobMaker.MakeJob(Wula_JobDefOf.WULA_RepairMech, t);
|
|
}
|
|
}
|
|
}
|