static void Postfix(Pawn_JobTracker __instance, ref ThinkResult __result)
{
Pawn pawn = Traverse.Create(__instance).Field("pawn").GetValue <Pawn>();
if (!pawn.IsColonistPlayerControlled)
{
return;
}
ExtendedDataStorage store = Base.Instance.GetExtendedDataStorage();
ExtendedPawnData pawnData = store.GetExtendedDataFor(pawn);
if (store == null || pawnData == null)
{
return;
}
if (pawnData.owning == null || !pawnData.owning.Spawned || pawnData.owning.Downed || pawnData.owning.Dead || pawnData.mount != null)
{
return;
}
LocalTargetInfo target = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
float pawnTargetDistance = DistanceUtility.QuickDistance(pawnData.owning.Position, target.Cell);
if (pawnTargetDistance > 10 || __result.Job.def == JobDefOf.LayDown || __result.Job.def == JobDefOf.Research || pawn.InMentalState || pawn.Dead || pawn.Downed)
{
if (pawnData.owning.jobs.curJob != null && pawnData.owning.jobs.curJob.def == JobDefOf.Wait)
{
pawnData.owning.jobs.EndCurrentJob(JobCondition.InterruptForced);
}
ExtendedPawnData animalData = store.GetExtendedDataFor(pawnData.owning);
pawnData.owning = null;
animalData.ownedBy = null;
}
}
static void Postfix(Pawn_JobTracker __instance, ref ThinkResult __result)
{
Pawn pawn = Traverse.Create(__instance).Field("pawn").GetValue <Pawn>();
if (pawn.IsColonistPlayerControlled || pawn.RaceProps.Animal)
{
return;
}
LocalTargetInfo target = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
if (!target.IsValid)
{
return;
}
ExtendedDataStorage store = Base.Instance.GetExtendedDataStorage();
if (store == null)
{
return;
}
//Log.Message("wrong duty");
ExtendedPawnData PawnData = store.GetExtendedDataFor(pawn);
Lord lord = pawn.GetLord();
if (lord == null)
{
return;
}
QueuedJob qJob = pawn.jobs.jobQueue.FirstOrFallback(null);
if (qJob != null && (qJob.job.def == GUC_JobDefOf.Dismount))
{
return;
}
Log.Message("curLordToil: " + pawn.GetLord().CurLordToil.ToString() + ", pawn name: " + pawn.Name);
Log.Message("lordJob: " + pawn.GetLord().LordJob + ", pawn name: " + pawn.Name);
Log.Message("lord.CurLordToil.GetType().Name" + lord.CurLordToil.GetType().Name);
if (lord.CurLordToil is LordToil_ExitMapAndEscortCarriers)
{
if (PawnData.owning != null && PawnData.mount == null && !PawnData.owning.Downed && PawnData.owning.Spawned)
{
mountAnimal(__instance, pawn, PawnData, ref __result);
}
}
else if (lord.CurLordToil.GetType().Name == "LordToil_DefendTraderCaravan" || lord.CurLordToil is LordToil_DefendPoint) //first option is internal class, hence this way of accessing.
{
if (PawnData.mount != null)
{
parkAnimal(__instance, pawn, PawnData);
}
}
}
static void Postfix(Pawn_JobTracker __instance, ref ThinkResult __result)
{
Pawn pawn = Traverse.Create(__instance).Field("pawn").GetValue <Pawn>();
if (!pawn.IsColonistPlayerControlled)
{
return;
}
if (__result.Job == null)
{
return;
}
if (__result.Job.def == GUC_JobDefOf.Mount)
{
return;
}
if (pawn.Drafted)
{
return;
}
if (pawn.InMentalState)
{
return;
}
LocalTargetInfo target = null;
LocalTargetInfo targetB = null;
//For some jobs the first target is B, and the second A.
if (__result.Job.def == JobDefOf.TendPatient || __result.Job.def == JobDefOf.Refuel)
{
target = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.B);
targetB = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
}
else if (__result.Job.def == JobDefOf.DoBill && !__result.Job.targetQueueB.NullOrEmpty())
{
target = __result.Job.targetQueueB[0];
targetB = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
}
else
{
target = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
targetB = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.B);
}
if (!target.IsValid)
{
return;
}
Area_GU area = (Area_GU)pawn.Map.areaManager.GetLabeled(Base.NOMOUNT_LABEL);
//TODO: make sure mounts are parked of when pawn wants to enter area with mount.
if (Base.Instance == null)
{
return;
}
ExtendedDataStorage store = Base.Instance.GetExtendedDataStorage();
ExtendedPawnData pawnData = store.GetExtendedDataFor(pawn);
if (store == null || pawnData == null)
{
return;
}
if (pawnData.wasRidingToJob)
{
pawnData.wasRidingToJob = false;
return;
}
if (pawn.mindState != null && pawn.mindState.duty != null && (pawn.mindState.duty.def == DutyDefOf.TravelOrWait || pawn.mindState.duty.def == DutyDefOf.TravelOrLeave))
{
return;
}
Pawn bestChoiceAnimal = pawnData.mount;
//Pawn bestChoiceAnimal = null;
float pawnTargetDistance = DistanceUtility.QuickDistance(pawn.Position, target.Cell);
//Log.Message("pawnTargetDistance: " + pawnTargetDistance);
float firstToSecondTargetDistance = 0;
if (__result.Job.def == JobDefOf.HaulToCell || __result.Job.def == JobDefOf.HaulToContainer)
{
if (targetB.IsValid)
{
firstToSecondTargetDistance = DistanceUtility.QuickDistance(target.Cell, targetB.Cell);
}
}
float totalDistance = pawnTargetDistance + firstToSecondTargetDistance;
if (totalDistance > Base.minAutoMountDistance)
{
if (pawnData.mount == null)
{
bestChoiceAnimal = GetBestChoiceAnimal(pawn, target, targetB, pawnTargetDistance, firstToSecondTargetDistance, store);
}
if (bestChoiceAnimal != null)
{
__result = InsertMountingJobs(pawn, bestChoiceAnimal, target, targetB, ref pawnData, store.GetExtendedDataFor(bestChoiceAnimal), __instance, __result);
}
//Log.Message("timeNeededOriginal: " + timeNeededOriginal);
//Log.Message("adjusted ticks per move: " + TicksPerMoveUtility.adjustedTicksPerMove(pawn, closestAnimal, true));
//Log.Message("original ticks per move: " + pawn.TicksPerMoveDiagonal);
}
}
static void Postfix(Pawn_JobTracker __instance, ref ThinkResult __result, ref Pawn ___pawn)
{
if (___pawn.RaceProps.Animal && ___pawn.Faction != Faction.OfPlayer && ___pawn.Faction != null)
{
if (___pawn.GetLord() != null && (___pawn.GetLord().CurLordToil is LordToil_DefendPoint || ___pawn.GetLord().CurLordToil.GetType().Name == "LordToil_DefendTraderCaravan"))
{
if (__result.SourceNode is JobGiver_Wander)
{
JobGiver_Wander jgWander = (JobGiver_Wander)__result.SourceNode;
Traverse.Create(__result.SourceNode).Field("wanderRadius").SetValue(5f);
}
}
}
if (___pawn.IsColonistPlayerControlled || ___pawn.RaceProps.Animal || ___pawn.Faction.HostileTo(Faction.OfPlayer))
{
return;
}
if (___pawn.IsPrisoner)
{
return;
}
LocalTargetInfo target = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
if (!target.IsValid)
{
return;
}
ExtendedDataStorage store = Base.GetExtendedDataStorage();
if (store == null)
{
return;
}
//Log.Message("wrong duty");
ExtendedPawnData PawnData = store.GetExtendedDataFor(___pawn);
Lord lord = ___pawn.GetLord();
if (lord == null)
{
return;
}
if (__result.Job.def == GUC_JobDefOf.Dismount || __result.Job.def == GUC_JobDefOf.Mount)
{
return;
}
QueuedJob qJob = ___pawn.jobs.jobQueue.FirstOrFallback(null);
if (qJob != null && (qJob.job.def == GUC_JobDefOf.Dismount || qJob.job.def == GUC_JobDefOf.Mount))
{
return;
}
if (lord.CurLordToil is LordToil_ExitMapAndEscortCarriers || lord.CurLordToil is LordToil_Travel || lord.CurLordToil is LordToil_ExitMap || lord.CurLordToil is LordToil_ExitMapTraderFighting)
{
if (PawnData.owning != null && PawnData.owning.Faction == ___pawn.Faction && PawnData.mount == null && !PawnData.owning.Downed && PawnData.owning.Spawned && !___pawn.IsBurning() && !___pawn.Downed)
{
mountAnimal(__instance, ___pawn, PawnData, ref __result);
}
}
else if (lord.CurLordToil.GetType().Name == "LordToil_DefendTraderCaravan" || lord.CurLordToil is LordToil_DefendPoint) //first option is internal class, hence this way of accessing.
{
if (PawnData.mount != null)
{
parkAnimal(__instance, ___pawn, PawnData);
}
}
}
static void Postfix(Pawn_JobTracker __instance, ref ThinkResult __result, ref Pawn ___pawn)
{
if (!___pawn.IsColonistPlayerControlled || !___pawn.RaceProps.Humanlike)
{
return;
}
if (__result.Job == null)
{
return;
}
if (__result.Job.def == GUC_JobDefOf.Mount)
{
return;
}
if (___pawn.Drafted)
{
return;
}
if (___pawn.InMentalState)
{
return;
}
if (___pawn.IsBorrowedByAnyFaction())
{
return;
}
ExtendedDataStorage store = Base.Instance.GetExtendedDataStorage();
if (store == null)
{
return;
}
ExtendedPawnData pawnData = store.GetExtendedDataFor(___pawn);
if (pawnData.mount != null)
{
return;
}
LocalTargetInfo firstTarget = null;
LocalTargetInfo secondTarget = null;
//For some jobs the first target is B, and the second A.
if (__result.Job.def == JobDefOf.TendPatient || __result.Job.def == JobDefOf.Refuel || __result.Job.def == JobDefOf.FixBrokenDownBuilding)
{
firstTarget = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.B);
secondTarget = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
}
else if (__result.Job.def == JobDefOf.DoBill && !__result.Job.targetQueueB.NullOrEmpty())
{
firstTarget = __result.Job.targetQueueB[0];
secondTarget = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
}
else
{
firstTarget = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.A);
secondTarget = DistanceUtility.GetFirstTarget(__result.Job, TargetIndex.B);
}
if (!firstTarget.IsValid)
{
return;
}
if (Base.Instance == null)
{
return;
}
if (store == null || pawnData == null)
{
return;
}
if (pawnData.wasRidingToJob)
{
pawnData.wasRidingToJob = false;
return;
}
if (___pawn.mindState != null && ___pawn.mindState.duty != null && (___pawn.mindState.duty.def == DutyDefOf.TravelOrWait || ___pawn.mindState.duty.def == DutyDefOf.TravelOrLeave))
{
return;
}
Pawn bestChoiceAnimal = null;
float pawnTargetDistance = DistanceUtility.QuickDistance(___pawn.Position, firstTarget.Cell);
float firstToSecondTargetDistance = 0;
if (__result.Job.def == JobDefOf.HaulToCell || __result.Job.def == JobDefOf.HaulToContainer)
{
if (secondTarget.IsValid)
{
firstToSecondTargetDistance = DistanceUtility.QuickDistance(firstTarget.Cell, secondTarget.Cell);
}
}
float totalDistance = pawnTargetDistance + firstToSecondTargetDistance;
if (totalDistance > Base.minAutoMountDistance)
{
bestChoiceAnimal = GetBestChoiceAnimal(___pawn, firstTarget, secondTarget, pawnTargetDistance, firstToSecondTargetDistance, store);
if (bestChoiceAnimal != null)
{
__result = InsertMountingJobs(___pawn, bestChoiceAnimal, firstTarget, secondTarget, ref pawnData, store.GetExtendedDataFor(bestChoiceAnimal), __instance, __result);
}
//Log.Message("timeNeededOriginal: " + timeNeededOriginal);
//Log.Message("adjusted ticks per move: " + TicksPerMoveUtility.adjustedTicksPerMove(pawn, closestAnimal, true));
//Log.Message("original ticks per move: " + pawn.TicksPerMoveDiagonal);
}
}
