87 lines
3.2 KiB
C#
87 lines
3.2 KiB
C#
using System.Collections.Generic;
|
||
using UnityEngine;
|
||
using Verse;
|
||
using Verse.AI;
|
||
using RimWorld; // For JobDefOf, ThingDefOf, StatDefOf
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class JobGiver_WulaPackEnergy : ThinkNode_JobGiver
|
||
{
|
||
public float packEnergyThreshold = 0.5f; // 默认打包能量阈值
|
||
public int packEnergyCount = 2; // 默认打包数量
|
||
|
||
// 定义乌拉能量核心的ThingDef
|
||
private static ThingDef WULA_Charge_Cube_Def => ThingDef.Named("WULA_Charge_Cube");
|
||
|
||
public override ThinkNode DeepCopy(bool resolve = true)
|
||
{
|
||
JobGiver_WulaPackEnergy obj = (JobGiver_WulaPackEnergy)base.DeepCopy(resolve);
|
||
obj.packEnergyThreshold = packEnergyThreshold;
|
||
obj.packEnergyCount = packEnergyCount;
|
||
return obj;
|
||
}
|
||
|
||
protected override Job TryGiveJob(Pawn pawn)
|
||
{
|
||
if (pawn.inventory == null)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
// 检查背包中是否有足够的能量核心，这里可以根据Need_WulaEnergy的当前值来判断是否需要打包
|
||
// 简化逻辑：如果能量低于某个阈值，并且背包中没有能量核心，则尝试打包
|
||
Need_WulaEnergy energyNeed = pawn.needs.TryGetNeed<Need_WulaEnergy>();
|
||
if (energyNeed == null)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
// 只有当能量低于阈值，并且背包中能量核心数量少于2个时，才尝试打包
|
||
if (energyNeed.CurLevelPercentage > packEnergyThreshold || pawn.inventory.innerContainer.TotalStackCountOfDef(WULA_Charge_Cube_Def) >= 2)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
// 检查是否超重
|
||
if (MassUtility.IsOverEncumbered(pawn))
|
||
{
|
||
return null;
|
||
}
|
||
|
||
// 寻找地图上可触及的WULA_Charge_Cube
|
||
Thing thing = GenClosest.ClosestThing_Regionwise_ReachablePrioritized(
|
||
pawn.Position,
|
||
pawn.Map,
|
||
ThingRequest.ForDef(WULA_Charge_Cube_Def), // 只寻找WULA_Charge_Cube
|
||
PathEndMode.ClosestTouch,
|
||
TraverseParms.For(pawn),
|
||
20f, // 搜索距离
|
||
delegate(Thing t)
|
||
{
|
||
// 检查物品是否被禁止，是否可预留，是否社交得体
|
||
return !t.IsForbidden(pawn) && pawn.CanReserve(t) && t.IsSociallyProper(pawn);
|
||
},
|
||
(Thing x) => 0f // 优先级，这里可以根据距离或其他因素调整
|
||
);
|
||
|
||
if (thing == null)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
// 计算需要打包的数量，限制在1到2个
|
||
int countToTake = Mathf.Min(thing.stackCount, 2); // 限制为最多2个
|
||
if (WULA_Charge_Cube_Def != null)
|
||
{
|
||
countToTake = Mathf.Min(countToTake, WULA_Charge_Cube_Def.stackLimit);
|
||
}
|
||
|
||
// 创建TakeInventory Job
|
||
Job job = JobMaker.MakeJob(JobDefOf.TakeInventory, thing);
|
||
job.count = countToTake;
|
||
return job;
|
||
}
|
||
}
|
||
}
|