127 lines
4.9 KiB
C#
127 lines
4.9 KiB
C#
using System.Collections.Generic;
|
||
using System.Linq; // Added for FirstOrDefault
|
||
using UnityEngine;
|
||
using Verse;
|
||
using Verse.AI;
|
||
using RimWorld; // For ThingDefOf, StatDefOf, etc.
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class JobDriver_IngestWulaEnergy : JobDriver
|
||
{
|
||
private const TargetIndex IngestibleSourceInd = TargetIndex.A;
|
||
private bool eatingFromInventory; // 新增字段
|
||
|
||
private Toil chewing; // 新增咀嚼Toil字段
|
||
|
||
private Thing IngestibleSource => job.GetTarget(IngestibleSourceInd).Thing;
|
||
|
||
// 新增咀嚼时间乘数属性
|
||
private float ChewDurationMultiplier
|
||
{
|
||
get
|
||
{
|
||
Thing ingestibleSource = IngestibleSource;
|
||
// 假设乌拉能量核心也有EatingSpeed属性影响咀嚼速度，或者固定为1f
|
||
return 1f / pawn.GetStatValue(StatDefOf.EatingSpeed);
|
||
}
|
||
}
|
||
|
||
public override void ExposeData()
|
||
{
|
||
base.ExposeData();
|
||
Scribe_Values.Look(ref eatingFromInventory, "eatingFromInventory", defaultValue: false);
|
||
}
|
||
|
||
public override bool TryMakePreToilReservations(bool errorOnFailed)
|
||
{
|
||
if (pawn.Faction != null)
|
||
{
|
||
Thing ingestibleSource = IngestibleSource;
|
||
// 使用FoodUtility.GetMaxAmountToPickup
|
||
int maxAmountToPickup = FoodUtility.GetMaxAmountToPickup(ingestibleSource, pawn, job.count);
|
||
if (!pawn.Reserve(ingestibleSource, job, 10, maxAmountToPickup, null, errorOnFailed))
|
||
{
|
||
return false;
|
||
}
|
||
job.count = maxAmountToPickup;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
protected override IEnumerable<Toil> MakeNewToils()
|
||
{
|
||
// 失败条件：如果能量核心被摧毁、为空或被禁止
|
||
this.FailOn(() => IngestibleSource.DestroyedOrNull() || !IngestibleSource.IngestibleNow);
|
||
|
||
// 初始化 eatingFromInventory
|
||
eatingFromInventory = pawn.inventory != null && pawn.inventory.Contains(IngestibleSource);
|
||
|
||
// 定义咀嚼Toil
|
||
chewing = Toils_Ingest.ChewIngestible(pawn, ChewDurationMultiplier, IngestibleSourceInd, TargetIndex.None)
|
||
.FailOn((Toil x) => !IngestibleSource.Spawned && (pawn.carryTracker == null || pawn.carryTracker.CarriedThing != IngestibleSource))
|
||
.FailOnCannotTouch(IngestibleSourceInd, PathEndMode.Touch);
|
||
|
||
// 根据是否从背包摄入，选择不同的Toil序列
|
||
foreach (Toil item in PrepareToIngestToils(chewing))
|
||
{
|
||
yield return item;
|
||
}
|
||
|
||
yield return chewing;
|
||
|
||
// 最终处理能量摄取
|
||
Toil finalizeToil = ToilMaker.MakeToil("FinalizeWulaEnergyIngest");
|
||
finalizeToil.initAction = delegate
|
||
{
|
||
Pawn actor = finalizeToil.actor;
|
||
Thing thing = actor.carryTracker.CarriedThing; // 从carryTracker获取，因为Toils_Ingest.ChewIngestible会处理携带
|
||
|
||
if (thing == null)
|
||
{
|
||
actor.jobs.EndCurrentJob(JobCondition.Incompletable);
|
||
return;
|
||
}
|
||
|
||
Need_WulaEnergy energyNeed = actor.needs.TryGetNeed<Need_WulaEnergy>();
|
||
if (energyNeed == null)
|
||
{
|
||
actor.jobs.EndCurrentJob(JobCondition.Errored);
|
||
return;
|
||
}
|
||
|
||
ThingDefExtension_EnergySource ext = thing.def.GetModExtension<ThingDefExtension_EnergySource>();
|
||
if (ext == null)
|
||
{
|
||
actor.jobs.EndCurrentJob(JobCondition.Errored);
|
||
return;
|
||
}
|
||
|
||
energyNeed.CurLevel += ext.energyAmount;
|
||
thing.Destroy(DestroyMode.Vanish);
|
||
};
|
||
finalizeToil.defaultCompleteMode = ToilCompleteMode.Instant;
|
||
yield return finalizeToil;
|
||
}
|
||
|
||
// 辅助方法，根据情况返回不同的Toil序列
|
||
private IEnumerable<Toil> PrepareToIngestToils(Toil chewToil)
|
||
{
|
||
if (eatingFromInventory)
|
||
{
|
||
yield return Toils_Misc.TakeItemFromInventoryToCarrier(pawn, IngestibleSourceInd);
|
||
}
|
||
else
|
||
{
|
||
// 类似原版JobDriver_Ingest的ToolUser逻辑
|
||
yield return Toils_Goto.GotoThing(IngestibleSourceInd, PathEndMode.ClosestTouch)
|
||
.FailOnDespawnedNullOrForbidden(IngestibleSourceInd);
|
||
yield return Toils_Ingest.PickupIngestible(IngestibleSourceInd, pawn);
|
||
}
|
||
// 不处理FindAdjacentEatSurface，因为乌拉能量核心可能不需要“吃表面”
|
||
// 也不处理takeExtraIngestibles，因为乌拉能量核心通常是单次消耗
|
||
yield break; // 确保迭代器结束
|
||
}
|
||
}
|
||
}
|