暂存能量改动

2025-07-21 02:05:37 +08:00
parent eed1c1b46a
commit ef074ec366
31 changed files with 4487 additions and 38 deletions
--- a/Source/WulaFallenEmpire/JobDriver_IngestWulaEnergy.cs
+++ b/Source/WulaFallenEmpire/JobDriver_IngestWulaEnergy.cs
@@ -0,0 +1,97 @@
+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;
+        }
+    }
+}