68 lines
2.1 KiB
C#
68 lines
2.1 KiB
C#
using System.Collections.Generic;
|
||
using Verse;
|
||
using Verse.AI;
|
||
using RimWorld;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class JobDriver_BoardMech : JobDriver
|
||
{
|
||
private const TargetIndex MechIndex = TargetIndex.A;
|
||
|
||
private CompMechCrewHolder CrewComp => job.targetA.Thing?.TryGetComp<CompMechCrewHolder>();
|
||
|
||
public override bool TryMakePreToilReservations(bool errorOnFailed)
|
||
{
|
||
// 预留目标机甲
|
||
if (!pawn.Reserve(job.targetA, job, 1, -1, null, errorOnFailed))
|
||
{
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
protected override IEnumerable<Toil> MakeNewToils()
|
||
{
|
||
// 第0步：添加失败条件
|
||
AddFailCondition(() =>
|
||
{
|
||
var mech = TargetThingA as Pawn;
|
||
if (mech == null || mech.Destroyed || mech.Dead)
|
||
return true;
|
||
|
||
var comp = CrewComp;
|
||
if (comp == null || comp.IsFull || !comp.CanAddCrew(pawn))
|
||
return true;
|
||
|
||
if (pawn.Downed || pawn.Dead)
|
||
return true;
|
||
|
||
return false;
|
||
});
|
||
|
||
// 第1步：走到机甲旁边
|
||
yield return Toils_Goto.GotoThing(MechIndex, PathEndMode.Touch);
|
||
|
||
// 第2步：等待短暂时间（可选）
|
||
yield return Toils_General.Wait(10).WithProgressBarToilDelay(MechIndex);
|
||
|
||
// 第3步：登上机甲
|
||
Toil boardToil = new Toil();
|
||
boardToil.initAction = () =>
|
||
{
|
||
var mech = TargetThingA as Pawn;
|
||
if (mech == null)
|
||
return;
|
||
|
||
var comp = CrewComp;
|
||
if (comp != null && comp.CanAddCrew(pawn))
|
||
{
|
||
comp.AddCrew(pawn);
|
||
}
|
||
};
|
||
boardToil.defaultCompleteMode = ToilCompleteMode.Instant;
|
||
yield return boardToil;
|
||
}
|
||
}
|
||
}
|