98 lines
3.8 KiB
C#
98 lines
3.8 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 Thing IngestibleSource => job.GetTarget(IngestibleSourceInd).Thing;
|
|
|
|
public override bool TryMakePreToilReservations(bool errorOnFailed)
|
|
{
|
|
// 尝试预留能量核心
|
|
if (pawn.Faction != null)
|
|
{
|
|
Thing ingestibleSource = IngestibleSource;
|
|
int maxAmountToPickup = FoodUtility.GetMaxAmountToPickup(ingestibleSource, pawn, job.count);
|
|
if (!pawn.Reserve(ingestibleSource, job, 10, maxAmountToPickup, null, errorOnFailed))
|
|
{
|
|
return false;
|
|
}
|
|
job.count = maxAmountToPickup; // 更新job.count以匹配实际预留数量
|
|
}
|
|
return true;
|
|
}
|
|
|
|
protected override IEnumerable<Toil> MakeNewToils()
|
|
{
|
|
// 失败条件：如果能量核心被摧毁、为空或被禁止
|
|
this.FailOn(() => IngestibleSource.DestroyedOrNull() || !IngestibleSource.IngestibleNow);
|
|
|
|
// Toil 1: 前往能量核心
|
|
yield return Toils_Goto.GotoThing(IngestibleSourceInd, PathEndMode.ClosestTouch)
|
|
.FailOnDespawnedNullOrForbidden(IngestibleSourceInd);
|
|
|
|
// Toil 2: 拾取能量核心并放入carryTracker
|
|
yield return Toils_Haul.StartCarryThing(IngestibleSourceInd);
|
|
|
|
// Toil 3: “摄取”能量核心 (模拟咀嚼过程，可以是一个简单的延迟)
|
|
Toil chewToil = ToilMaker.MakeToil("ChewWulaEnergy");
|
|
chewToil.initAction = delegate
|
|
{
|
|
// 设定一个短暂的“咀嚼”时间
|
|
pawn.jobs.curDriver.ticksLeftThisToil = 60; // 1秒
|
|
};
|
|
chewToil.defaultCompleteMode = ToilCompleteMode.Delay;
|
|
yield return chewToil;
|
|
|
|
// Toil 4: 最终处理能量摄取
|
|
Toil finalizeToil = ToilMaker.MakeToil("FinalizeWulaEnergyIngest");
|
|
finalizeToil.initAction = delegate
|
|
{
|
|
Pawn actor = finalizeToil.actor;
|
|
// 从Pawn的carryTracker中获取能量核心
|
|
Thing thing = actor.carryTracker.CarriedThing;
|
|
|
|
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);
|
|
|
|
// 记录能量摄入 (可选，如果需要类似 NutritionEaten 的记录)
|
|
// actor.records.AddTo(RecordDefOf.NutritionEaten, ext.energyAmount);
|
|
};
|
|
finalizeToil.defaultCompleteMode = ToilCompleteMode.Instant;
|
|
yield return finalizeToil;
|
|
}
|
|
}
|
|
}
|