67 lines
2.1 KiB
C#
67 lines
2.1 KiB
C#
using HarmonyLib;
|
|
using RimWorld;
|
|
using Verse;
|
|
using Verse.AI;
|
|
|
|
namespace WulaFallenEmpire
|
|
{
|
|
[HarmonyPatch(typeof(Pawn_JobTracker), "StartJob")]
|
|
public static class Patch_Pawn_JobTracker_StartJob
|
|
{
|
|
[HarmonyPostfix]
|
|
public static void Postfix(Pawn_JobTracker __instance, Pawn ___pawn)
|
|
{
|
|
// 检查是否是移动相关的任务
|
|
if (___pawn?.Map == null || ___pawn.Dead || ___pawn.Downed)
|
|
return;
|
|
|
|
if (__instance.curJob == null)
|
|
return;
|
|
|
|
// 只处理移动任务
|
|
if (__instance.curJob.def == JobDefOf.Goto ||
|
|
__instance.curJob.def == JobDefOf.GotoWander ||
|
|
__instance.curJob.def == JobDefOf.Follow)
|
|
{
|
|
// 通知区域传送器检查这个Pawn
|
|
NotifyAreaTeleporters(___pawn);
|
|
}
|
|
}
|
|
|
|
private static void NotifyAreaTeleporters(Pawn pawn)
|
|
{
|
|
if (pawn.Map == null)
|
|
return;
|
|
|
|
// 查找范围内的所有区域传送器
|
|
foreach (var thing in pawn.Map.listerThings.ThingsOfDef(ThingDef.Named("WULA_Support_AreaTeleporter")))
|
|
{
|
|
var comp = thing.TryGetComp<ThingComp_AreaTeleporter>();
|
|
if (comp != null && pawn.Position.DistanceTo(thing.Position) <= comp.Props.teleportRadius)
|
|
{
|
|
// 强制立即检查这个Pawn
|
|
comp.ForceCheckPawn(pawn);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 为ThingComp_AreaTeleporter添加强制检查方法
|
|
public partial class ThingComp_AreaTeleporter
|
|
{
|
|
public void ForceCheckPawn(Pawn pawn)
|
|
{
|
|
if (parent == null || !parent.Spawned || parent.Map == null)
|
|
return;
|
|
|
|
if (pawn.Position.DistanceTo(parent.Position) > Props.teleportRadius)
|
|
return;
|
|
|
|
if (!ShouldAffectPawn(pawn))
|
|
return;
|
|
|
|
TryTeleportPawn(pawn);
|
|
}
|
|
}
|
|
}
|