124 lines
4.1 KiB
C#
124 lines
4.1 KiB
C#
// WorkGiver_RefuelMech.cs
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using RimWorld;
|
|
using Verse;
|
|
using Verse.AI;
|
|
|
|
namespace WulaFallenEmpire
|
|
{
|
|
public class WorkGiver_RefuelMech : WorkGiver_Scanner
|
|
{
|
|
public override PathEndMode PathEndMode => PathEndMode.Touch;
|
|
|
|
public override IEnumerable<Thing> PotentialWorkThingsGlobal(Pawn pawn)
|
|
{
|
|
// 返回所有需要燃料的机甲
|
|
// 修复：使用 LINQ 的 Where 方法而不是 FindAll
|
|
var mechs = pawn.Map.mapPawns.AllPawnsSpawned.Where(p =>
|
|
p.TryGetComp<CompMechFuel>() != null);
|
|
|
|
foreach (Pawn mech in mechs)
|
|
{
|
|
yield return mech;
|
|
}
|
|
}
|
|
|
|
public override bool ShouldSkip(Pawn pawn, bool forced = false)
|
|
{
|
|
// 如果没有需要燃料的机甲，跳过
|
|
return !PotentialWorkThingsGlobal(pawn).Any();
|
|
}
|
|
|
|
public override bool HasJobOnThing(Pawn pawn, Thing t, bool forced = false)
|
|
{
|
|
if (!(t is Pawn mech))
|
|
return false;
|
|
|
|
var fuelComp = mech.TryGetComp<CompMechFuel>();
|
|
if (fuelComp == null)
|
|
return false;
|
|
|
|
// 检查机甲是否已加满燃料
|
|
if (fuelComp.IsFull)
|
|
return false;
|
|
|
|
// 检查是否有可用的燃料
|
|
if (FindFuel(pawn, fuelComp) == null)
|
|
return false;
|
|
|
|
// 检查是否能接触到机甲
|
|
if (!pawn.CanReserveAndReach(t, PathEndMode.Touch, Danger.Some))
|
|
return false;
|
|
|
|
// 检查机甲状态
|
|
var pilotComp = mech.TryGetComp<CompMechPilotHolder>();
|
|
bool hasPilot = pilotComp != null && pilotComp.HasPilots;
|
|
|
|
// 如果有驾驶员且不是强制命令，不自动加注
|
|
if (hasPilot && !forced)
|
|
return false;
|
|
|
|
// 检查燃料组件是否允许自动加注
|
|
if (!fuelComp.Props.allowAutoRefuel && !forced)
|
|
return false;
|
|
|
|
// 检查是否达到自动加注阈值
|
|
if (!forced && !fuelComp.NeedsRefueling)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
public override Job JobOnThing(Pawn pawn, Thing t, bool forced = false)
|
|
{
|
|
var fuelComp = t.TryGetComp<CompMechFuel>();
|
|
if (fuelComp == null)
|
|
return null;
|
|
|
|
// 寻找燃料
|
|
Thing fuel = FindFuel(pawn, fuelComp);
|
|
if (fuel == null)
|
|
return null;
|
|
|
|
// 创建加注工作
|
|
Job job = JobMaker.MakeJob(Wula_JobDefOf.WULA_RefuelMech, t, fuel);
|
|
job.count = fuelComp.GetFuelCountToFullyRefuel();
|
|
return job;
|
|
}
|
|
|
|
// 修改方法：返回 Thing 而不是 bool
|
|
private Thing FindFuel(Pawn pawn, CompMechFuel fuelComp)
|
|
{
|
|
if (fuelComp.FuelType == null)
|
|
return null;
|
|
|
|
// 在库存中寻找燃料
|
|
Thing fuel = FindFuelInInventory(pawn, fuelComp.FuelType);
|
|
if (fuel != null)
|
|
return fuel;
|
|
|
|
// 在地图上寻找燃料
|
|
fuel = GenClosest.ClosestThingReachable(
|
|
pawn.Position,
|
|
pawn.Map,
|
|
ThingRequest.ForDef(fuelComp.FuelType),
|
|
PathEndMode.ClosestTouch,
|
|
TraverseParms.For(pawn),
|
|
9999f,
|
|
validator: thing => !thing.IsForbidden(pawn) && pawn.CanReserve(thing)
|
|
);
|
|
|
|
return fuel;
|
|
}
|
|
|
|
private Thing FindFuelInInventory(Pawn pawn, ThingDef fuelType)
|
|
{
|
|
if (pawn.inventory == null)
|
|
return null;
|
|
|
|
return pawn.inventory.innerContainer.FirstOrDefault(t => t.def == fuelType);
|
|
}
|
|
}
|
|
}
|