92 lines
3.4 KiB
C#
92 lines
3.4 KiB
C#
using Milk;
|
|
using rjw;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Verse;
|
|
using RimWorld;
|
|
using Verse.AI;
|
|
|
|
namespace CRIALactation
|
|
{
|
|
public class JobDriver_MassageBreasts : JobDriver
|
|
{
|
|
private readonly float WorkTotal = 600f;
|
|
|
|
public override bool TryMakePreToilReservations(bool errorOnFailed)
|
|
{
|
|
LocalTargetInfo Target = job.GetTarget(TargetIndex.A);
|
|
return pawn.Reserve(Target, job, 1, -1, null, errorOnFailed);
|
|
}
|
|
|
|
protected override IEnumerable<Toil> MakeNewToils()
|
|
{
|
|
this.FailOnDespawnedNullOrForbidden(TargetIndex.A);
|
|
yield return Toils_Goto.GotoThing(TargetIndex.A, PathEndMode.Touch);
|
|
Toil massage = new Toil();
|
|
massage.FailOnDespawnedOrNull(TargetIndex.A);
|
|
massage.FailOnAggroMentalStateAndHostile(TargetIndex.A);
|
|
float massageProgress = 0;
|
|
massage.initAction = delegate
|
|
{
|
|
Pawn p = job.GetTarget(TargetIndex.A).Thing as Pawn;
|
|
pawn.pather.StopDead();
|
|
PawnUtility.ForceWait(p, 15000, null, true);
|
|
};
|
|
massage.tickAction = delegate ()
|
|
{
|
|
pawn.skills.Learn(SkillDefOf.Animals, 0.13f, false);
|
|
|
|
massageProgress += pawn.GetStatValue(StatDefOf.AnimalGatherSpeed, true);
|
|
|
|
};
|
|
massage.AddEndCondition(delegate
|
|
{
|
|
Pawn p = job.GetTarget(TargetIndex.A).Thing as Pawn;
|
|
if (massageProgress >= WorkTotal)
|
|
{
|
|
|
|
Hediff induce = p.health.hediffSet.GetFirstHediffOfDef(HediffDefOf_Milk.InducingLactation);
|
|
induce.Severity += (1f / LactationSettings.totalMassagesUntilLactation);
|
|
//0.05f ; //20 times, 3 times a day = 7 days, give or take
|
|
|
|
if (induce.Severity >= 1)
|
|
{
|
|
//p.health.AddHediff(cooldown,
|
|
LactationUtility.StartLactating(p, true);
|
|
//remove the other hediff
|
|
p.health.RemoveHediff(induce);
|
|
}
|
|
|
|
//Add cooldown hediff. This has the pain debuff so i'm going to add it even if you're now lactating. The last one would have still hurt.
|
|
Hediff cooldown = HediffMaker.MakeHediff(HediffDefOf_Milk.InducingLactationCooldown, p, null);
|
|
cooldown.Severity = LactationSettings.massageCooldown;
|
|
p.health.AddHediff(cooldown, null, null, null);
|
|
|
|
return JobCondition.Succeeded;
|
|
}
|
|
|
|
return JobCondition.Ongoing;
|
|
|
|
});
|
|
|
|
massage.AddFinishAction(delegate {
|
|
Pawn pawn = this.job.GetTarget(TargetIndex.A).Thing as Pawn;
|
|
if (pawn != null && pawn.CurJobDef == JobDefOf.Wait_MaintainPosture)
|
|
{
|
|
pawn.jobs.EndCurrentJob(JobCondition.InterruptForced, true, true);
|
|
}
|
|
});
|
|
massage.defaultCompleteMode = ToilCompleteMode.Never;
|
|
|
|
massage.WithProgressBar(TargetIndex.A, () => massageProgress / WorkTotal);
|
|
massage.activeSkill = (() => SkillDefOf.Animals);
|
|
yield return massage;
|
|
yield break;
|
|
|
|
}
|
|
}
|
|
}
|