64 lines
2.4 KiB
C#
64 lines
2.4 KiB
C#
// WorkGiver_DoMaintenance.cs (修复版)
|
||
using RimWorld;
|
||
using Verse;
|
||
using Verse.AI;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class WorkGiver_DoMaintenance : WorkGiver_Scanner
|
||
{
|
||
public override ThingRequest PotentialWorkThingRequest =>
|
||
ThingRequest.ForDef(ThingDef.Named("WULA_MaintenancePod"));
|
||
|
||
public override PathEndMode PathEndMode => PathEndMode.Touch;
|
||
|
||
public override bool HasJobOnThing(Pawn pawn, Thing t, bool forced = false)
|
||
{
|
||
// 检查维护舱是否可用
|
||
if (!pawn.CanReserve(t, 1, -1, null, forced))
|
||
return false;
|
||
|
||
var podComp = t.TryGetComp<CompMaintenancePod>();
|
||
if (podComp == null || podComp.State != MaintenancePodState.Idle || !podComp.PowerOn)
|
||
return false;
|
||
|
||
// 检查是否有足够的燃料（零部件）
|
||
// 如果是强制工作（玩家右键），我们允许通过检查，让 JobDriver 去处理（可能会提示燃料不足）
|
||
// 这样玩家能知道为什么不能工作，而不是默默失败
|
||
if (!forced)
|
||
{
|
||
float requiredFuel = podComp.GetRequiredComponentsFor(pawn);
|
||
var refuelable = t.TryGetComp<CompRefuelable>();
|
||
if (refuelable != null && refuelable.Fuel < requiredFuel)
|
||
{
|
||
JobFailReason.Is("WULA_MaintenancePod_NotEnoughComponents".Translate(requiredFuel.ToString("F0")));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// 检查当前pawn是否有维护需求且需要维护
|
||
return PawnNeedsMaintenance(pawn);
|
||
}
|
||
|
||
public override Job JobOnThing(Pawn pawn, Thing t, bool forced = false)
|
||
{
|
||
return JobMaker.MakeJob(Wula_JobDefOf.WULA_EnterMaintenancePod, t);
|
||
}
|
||
|
||
// 检查单个Pawn是否需要维护
|
||
private bool PawnNeedsMaintenance(Pawn pawn)
|
||
{
|
||
// 检查是否有维护需求组件
|
||
var maintenanceNeed = pawn.needs?.TryGetNeed<Need_Maintenance>();
|
||
if (maintenanceNeed == null)
|
||
{
|
||
// 这个Pawn没有维护需求，不应该使用维护舱
|
||
return false;
|
||
}
|
||
|
||
// 检查维护水平是否低于阈值
|
||
return maintenanceNeed.CurLevel <= 0.3f; // 需要维护的阈值
|
||
}
|
||
}
|
||
}
|