Bit of a refactor to AddWombGizmos, should be nicer to the Sexperience module

This commit is contained in:
lutepickle 2022-07-11 12:55:39 -07:00
parent 699c91106f
commit 3d34ed1e53

View file

@ -28,14 +28,16 @@ namespace RJW_Menstruation
foreach (Gizmo gizmo in gizmos) foreach (Gizmo gizmo in gizmos)
yield return gizmo; yield return gizmo;
foreach (HediffComp_Menstruation comp in __instance.GetMenstruationComps()) foreach (Gizmo gizmo in GetMenstruationGizmos(__instance))
foreach (Gizmo gizmo in GetMenstruationGizmos(__instance, comp)) yield return gizmo;
yield return gizmo;
} }
public static List<Gizmo> GetMenstruationGizmos(Pawn pawn, HediffComp_Menstruation comp) public static List<Gizmo> GetMenstruationGizmos(Pawn pawn)
{ {
return new List<Gizmo>() { CreateGizmo_WombStatus(pawn, comp) }; List<Gizmo> res = new List<Gizmo>();
foreach (HediffComp_Menstruation comp in pawn.GetMenstruationComps())
res.Add(CreateGizmo_WombStatus(pawn, comp));
return res;
} }