rjw_menstruation/1.4/source/RJW_Menstruation/RJW_Menstruation/CompBiosculpterPod_EggResto...

31 lines
1.0 KiB
C#

using RimWorld;
using System.Linq;
using Verse;
namespace RJW_Menstruation
{
public class CompProperties_BiosculpterPod_EggRestorationCycle : CompProperties_BiosculpterPod_BaseCycle
{
public CompProperties_BiosculpterPod_EggRestorationCycle()
{
compClass = typeof(CompBiosculpterPod_EggRestorationCycle);
}
public float yearsToRestore;
}
public class CompBiosculpterPod_EggRestorationCycle : CompBiosculpterPod_Cycle
{
public override void CycleCompleted(Pawn occupant)
{
foreach (HediffComp_Menstruation comp in occupant.GetMenstruationComps())
comp.RestoreEggs(((CompProperties_BiosculpterPod_EggRestorationCycle)Props).yearsToRestore);
Messages.Message(Translations.EggRestorationCompleted(occupant.Named("PAWN")), occupant, MessageTypeDefOf.PositiveEvent);
if (occupant.GetMenstruationComps().Any())
occupant.needs.mood?.thoughts?.memories?.TryGainMemoryFast(VariousDefOf.EggRestorationReceived);
}
}
}