AblativeAbsolute/rjw_menstruation
1
0
Fork 0
mirror of https://gitgud.io/lutepickle/rjw_menstruation.git synced 2024-08-14 22:46:52 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add PeekBaby and BabyCount to PregeneratedBabies

Browse source
This commit is contained in:
lutepickle 2024-05-17 08:36:12 -07:00
parent afcb7bf7c5
commit 64923264b9
4 changed files with 21 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
1.5/Assemblies/RJW_Menstruation.dll
View file

Binary file not shown.

21
1.5/source/RJW_Menstruation/RJW_Menstruation/HediffComps/HediffComp_PregeneratedBabies.cs
View file

@ -21,7 +21,22 @@ namespace RJW_Menstruation
public bool HasBaby public bool HasBaby
{ {
get => !babies.NullOrEmpty(); get => !babies.NullOrEmpty();
}
public int BabyCount
{
get
{
if (babies == null) return 0;
else return babies.Count;
}
}
public Pawn PeekBaby()
{
if (babies.NullOrEmpty()) return null;
else return babies.First();
} }
public Pawn PopBaby() public Pawn PopBaby()
@ -210,7 +225,7 @@ namespace RJW_Menstruation
float birthQuality = PregnancyUtility.GetBirthQualityFor(mother); float birthQuality = PregnancyUtility.GetBirthQualityFor(mother);
do do
{ {
Pawn baby = comp.babies[0]; Pawn baby = comp.PeekBaby();
Pawn thisFather = baby.GetFather() ?? father; Pawn thisFather = baby.GetFather() ?? father;
baby.relations.ClearAllRelations(); // To keep ApplyBirthOutcome from erroring when it tries to set up relations baby.relations.ClearAllRelations(); // To keep ApplyBirthOutcome from erroring when it tries to set up relations
@ -269,7 +284,7 @@ namespace RJW_Menstruation
// Though even if it does, that's probably okay. More babies more memories after all // Though even if it does, that's probably okay. More babies more memories after all
do do
{ {
Pawn baby = comp.babies[0]; Pawn baby = comp.PeekBaby();
Pawn thisFather = baby.GetFather() ?? father; Pawn thisFather = baby.GetFather() ?? father;
baby.relations.ClearAllRelations(); baby.relations.ClearAllRelations();

4
1.5/source/RJW_Menstruation/RJW_Menstruation/HediffComps/MenstruationUtility.cs
View file

@ -118,8 +118,8 @@ namespace RJW_Menstruation
} }
else if (babiescomp?.HasBaby ?? false) else if (babiescomp?.HasBaby ?? false)
{ {
babydef = babiescomp.babies.First().def; babydef = babiescomp.PeekBaby().def;
babycount = babiescomp.babies.Count; babycount = babiescomp.BabyCount;
} }
else else
{ {

2
1.5/source/RJW_Menstruation/RJW_Menstruation/Utility.cs
View file

@ -178,7 +178,7 @@ namespace RJW_Menstruation
} }
} }
HediffComp_PregeneratedBabies babiescomp = comp.Pregnancy.TryGetComp<HediffComp_PregeneratedBabies>(); HediffComp_PregeneratedBabies babiescomp = comp.Pregnancy.TryGetComp<HediffComp_PregeneratedBabies>();
return babiescomp?.babies?.FirstOrDefault(); return babiescomp?.PeekBaby();
} }
public static void DrawBreastIcon(this Pawn pawn, Rect rect) public static void DrawBreastIcon(this Pawn pawn, Rect rect)