Don't display cum overlay when stuffed with insect eggs

2023-06-29 15:59:02 -07:00 · 2023-06-29 15:59:02 -07:00 · 30a5d84476
parent f5a302c272
commit 30a5d84476
2 changed files with 6 additions and 0 deletions
--- a/1.4/Assemblies/RJW_Menstruation.dll
+++ b/1.4/Assemblies/RJW_Menstruation.dll
--- a/1.4/source/RJW_Menstruation/RJW_Menstruation/HediffComps/MenstruationUtility.cs
+++ b/1.4/source/RJW_Menstruation/RJW_Menstruation/HediffComps/MenstruationUtility.cs
@ -166,6 +166,12 @@ namespace RJW_Menstruation

        public static Texture2D GetCumIcon(this HediffComp_Menstruation comp)
        {
+            Pawn pawn = comp.Pawn;
+            
+            List<Hediff_InsectEgg> insectEggs = new List<Hediff_InsectEgg>();
+            comp.Pawn.health.hediffSet.GetHediffs(ref insectEggs);
+            if (!insectEggs.NullOrEmpty() && insectEggs.Sum(hediff => hediff.eggssize) > 1.0f) return null; // same logic as "Stuffed" in GetInsectEggedIcon
+
            string icon = comp.WombTex;
            float cumpercent = comp.TotalCumPercent;
            if (cumpercent < 0.001f) return ContentFinder<Texture2D>.Get("Womb/Empty", true);