rimworld-animations/Source/Patches/HarmonyPatch_Pawn_DrawTrack...

50 lines
1.8 KiB
C#

using HarmonyLib;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using UnityEngine;
using Verse;
using Verse.AI;
namespace Rimworld_Animations {
[HarmonyPatch(typeof(Pawn_DrawTracker), "DrawPos", MethodType.Getter)]
public static class HarmonyPatch_Pawn_DrawTracker {
public static bool Prefix(ref Pawn ___pawn, ref Vector3 __result) {
if(___pawn.TryGetComp<CompBodyAnimator>() != null) {
if(___pawn.TryGetComp<CompBodyAnimator>().isAnimating) {
__result = ___pawn.TryGetComp<CompBodyAnimator>().anchor + ___pawn.TryGetComp<CompBodyAnimator>().deltaPos;
___pawn.Position = __result.ToIntVec3();
return false;
} else if(___pawn.TryGetComp<CompBodyAnimator>().resetPosition) {
//resetting position to anchor
__result = ___pawn.TryGetComp<CompBodyAnimator>().anchor;
___pawn.Position = ___pawn.TryGetComp<CompBodyAnimator>().positionReset;
return false;
}
}
return true;
}
}
[HarmonyPatch(typeof(Pawn_PathFollower), "StartPath")]
public static class HarmonyPatch_ResetPositionIfPathing {
public static void Prefix(ref Pawn ___pawn) {
if(___pawn.TryGetComp<CompBodyAnimator>() != null) {
if(___pawn.TryGetComp<CompBodyAnimator>().isAnimating || ___pawn.TryGetComp<CompBodyAnimator>().resetPosition) {
___pawn.TryGetComp<CompBodyAnimator>().isAnimating = false;
___pawn.Position = ___pawn.TryGetComp<CompBodyAnimator>().positionReset;
}
}
}
}
}