144 lines
5.2 KiB
C#
144 lines
5.2 KiB
C#
using System.Collections.Generic;
|
||
using Verse;
|
||
using Verse.AI;
|
||
using RimWorld;
|
||
using UnityEngine;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class JobDriver_TransformPawn : JobDriver
|
||
{
|
||
private const TargetIndex GatherBuildingIndex = TargetIndex.A;
|
||
|
||
private Comp_Gather GatherComp => job.targetA.Thing?.TryGetComp<Comp_Gather>();
|
||
private Comp_AutonomousCat PawnComp => pawn.TryGetComp<Comp_AutonomousCat>();
|
||
|
||
public override bool TryMakePreToilReservations(bool errorOnFailed)
|
||
{
|
||
// 预留目标建筑
|
||
if (!pawn.Reserve(job.targetA, job, 1, -1, null, errorOnFailed))
|
||
{
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
protected override IEnumerable<Toil> MakeNewToils()
|
||
{
|
||
// 第1步：移动到目标建筑
|
||
yield return Toils_Goto.GotoCell(GatherBuildingIndex, PathEndMode.InteractionCell);
|
||
|
||
// 第2步：进行转化工作
|
||
Toil transformToil = new Toil();
|
||
transformToil.initAction = () =>
|
||
{
|
||
// 获取目标建筑
|
||
Thing gatherBuilding = job.targetA.Thing;
|
||
if (gatherBuilding == null || gatherBuilding.Destroyed)
|
||
{
|
||
ReadyForNextToil();
|
||
return;
|
||
}
|
||
|
||
// 获取Comp_Gather
|
||
var gatherComp = gatherBuilding.TryGetComp<Comp_Gather>();
|
||
if (gatherComp == null)
|
||
{
|
||
ReadyForNextToil();
|
||
return;
|
||
}
|
||
|
||
// 确保可以转化
|
||
if (!gatherComp.CanTransformPawn(pawn))
|
||
{
|
||
Messages.Message("Wula_CannotTransformHere".Translate(), MessageTypeDefOf.RejectInput);
|
||
ReadyForNextToil();
|
||
return;
|
||
}
|
||
};
|
||
|
||
transformToil.tickAction = () =>
|
||
{
|
||
// 确保目标建筑仍然有效
|
||
Thing gatherBuilding = job.targetA.Thing;
|
||
if (gatherBuilding == null || gatherBuilding.Destroyed ||
|
||
gatherBuilding.Map != pawn.Map ||
|
||
pawn.Position.DistanceTo(gatherBuilding.Position) > 3f)
|
||
{
|
||
// 中断转化，清除转化目标
|
||
PawnComp?.ClearTransformTarget();
|
||
EndJobWith(JobCondition.Incompletable);
|
||
return;
|
||
}
|
||
|
||
// 面向建筑
|
||
pawn.rotationTracker.FaceCell(gatherBuilding.Position);
|
||
|
||
// 播放转化效果
|
||
if (Find.TickManager.TicksGame % 20 == 0)
|
||
{
|
||
PlayTransformEffects();
|
||
}
|
||
};
|
||
|
||
// 从Comp_AutonomousCat获取转化时间
|
||
var compProps = PawnComp?.Props as CompProperties_AutonomousCat;
|
||
int transformDuration = compProps != null ?
|
||
Mathf.RoundToInt(compProps.transformTime * 60f) : 180; // 默认3秒
|
||
|
||
transformToil.defaultCompleteMode = ToilCompleteMode.Delay;
|
||
transformToil.defaultDuration = transformDuration;
|
||
yield return transformToil;
|
||
|
||
// 第3步：完成转化
|
||
yield return new Toil
|
||
{
|
||
initAction = () =>
|
||
{
|
||
// 获取目标建筑
|
||
Thing gatherBuilding = job.targetA.Thing;
|
||
if (gatherBuilding == null || gatherBuilding.Destroyed)
|
||
{
|
||
return;
|
||
}
|
||
|
||
// 获取Comp_Gather
|
||
var gatherComp = gatherBuilding.TryGetComp<Comp_Gather>();
|
||
if (gatherComp == null)
|
||
{
|
||
return;
|
||
}
|
||
|
||
// 获取要转化的PawnKindDef（从Comp_AutonomousCat中）
|
||
var targetPawnKind = PawnComp?.PendingTransformTarget;
|
||
if (targetPawnKind == null)
|
||
{
|
||
Messages.Message("Wula_NoTransformTarget".Translate(), MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 调用Comp_Gather的转化方法
|
||
gatherComp.TransformPawn(pawn, targetPawnKind);
|
||
|
||
// 清除转化目标
|
||
PawnComp?.ClearTransformTarget();
|
||
},
|
||
defaultCompleteMode = ToilCompleteMode.Instant
|
||
};
|
||
}
|
||
|
||
// 播放转化效果
|
||
private void PlayTransformEffects()
|
||
{
|
||
// 播放音效
|
||
var compProps = PawnComp?.Props as CompProperties_AutonomousCat;
|
||
|
||
// 播放特效
|
||
if (compProps?.transformEffect != null)
|
||
{
|
||
compProps.transformEffect.Spawn(pawn.Position, pawn.Map).Cleanup();
|
||
}
|
||
}
|
||
}
|
||
}
|