425 lines
20 KiB
C#
425 lines
20 KiB
C#
using RimWorld;
|
|
using rjw;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using UnityEngine;
|
|
using Verse;
|
|
using Verse.AI;
|
|
|
|
namespace RJW_Menstruation
|
|
{
|
|
public static class MenstruationUtility
|
|
{
|
|
[Obsolete("This method is obsolete. Use GetMenstruationComps or a related function instead", false)]
|
|
public static HediffComp_Menstruation GetMenstruationComp(this Pawn pawn)
|
|
{
|
|
return pawn.GetFirstMenstruationComp();
|
|
}
|
|
|
|
public static IEnumerable<HediffComp_Menstruation> GetMenstruationComps(this Pawn pawn)
|
|
{
|
|
List<Hediff> hedifflist = Genital_Helper.get_PartsHediffList(pawn, Genital_Helper.get_genitalsBPR(pawn))?.FindAll(h => VariousDefOf.AllVaginas.Contains(h.def));
|
|
if (hedifflist == null) yield break;
|
|
foreach (Hediff hediff in hedifflist)
|
|
{
|
|
HediffComp_Menstruation result = hediff.TryGetComp<HediffComp_Menstruation>();
|
|
if (result != null) yield return result;
|
|
}
|
|
}
|
|
|
|
public static HediffComp_Menstruation GetFirstMenstruationComp(this Pawn pawn)
|
|
{
|
|
return pawn.GetMenstruationComps().FirstOrDefault();
|
|
}
|
|
|
|
public static HediffComp_Menstruation GetRandomMenstruationComp(this Pawn pawn)
|
|
{
|
|
return pawn.GetMenstruationComps().RandomElementWithFallback();
|
|
}
|
|
|
|
public static HediffComp_Menstruation GetFertileMenstruationComp(this Pawn pawn)
|
|
{
|
|
List<HediffComp_Menstruation> comps = pawn.GetMenstruationComps().ToList();
|
|
return comps.Where(c => c.IsDangerDay).RandomElementWithFallback() ?? comps.RandomElementWithFallback();
|
|
}
|
|
|
|
public static HediffComp_Menstruation GetMenstruationCompFromVagina(this Hediff vagina)
|
|
{
|
|
if (VariousDefOf.AllVaginas.Contains(vagina?.def))
|
|
{
|
|
return vagina.TryGetComp<HediffComp_Menstruation>();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public static HediffComp_Menstruation GetMenstruationCompFromPregnancy(this Hediff pregnancy)
|
|
{
|
|
return pregnancy?.pawn.GetMenstruationComps().FirstOrDefault(comp => comp.Pregnancy == pregnancy);
|
|
}
|
|
|
|
public static HediffComp_Anus GetAnusComp(this Pawn pawn)
|
|
{
|
|
List<Hediff> hedifflist = Genital_Helper.get_PartsHediffList(pawn, Genital_Helper.get_genitalsBPR(pawn))?.FindAll((Hediff h) => h.def.defName.ToLower().Contains("anus"));
|
|
HediffComp_Anus result;
|
|
if (!hedifflist.NullOrEmpty())
|
|
{
|
|
foreach (Hediff h in hedifflist)
|
|
{
|
|
result = h.TryGetComp<HediffComp_Anus>();
|
|
if (result != null) return result;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
[Obsolete("This method is obsolete and can cause ambiguity. Use GetMenstruationCompFromVagina or GetMenstruationCompFromPregnancy instead.", true)]
|
|
public static HediffComp_Menstruation GetMenstruationComp(Hediff hediff)
|
|
{
|
|
switch (hediff)
|
|
{
|
|
case Hediff_BasePregnancy rjwPreg:
|
|
return rjwPreg.GetMenstruationCompFromPregnancy();
|
|
case Hediff_Pregnant vanillaPreg:
|
|
return vanillaPreg.GetMenstruationCompFromPregnancy();
|
|
case Hediff_Labor vanillaLabor:
|
|
return vanillaLabor.GetMenstruationCompFromPregnancy();
|
|
case Hediff_LaborPushing vanillaLaborPushing:
|
|
return vanillaLaborPushing.GetMenstruationCompFromPregnancy();
|
|
case Hediff_PartBaseNatural rjwNatrual:
|
|
return rjwNatrual.GetMenstruationCompFromVagina();
|
|
case Hediff_PartBaseArtifical rjwArtificial:
|
|
return rjwArtificial.GetMenstruationCompFromVagina();
|
|
default:
|
|
Log.Warning("Obsolete GetMenstruationComp called with unknown hediff. Ensure your submods are up to date.");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static HediffComp_Anus GetAnusComp(this Hediff hediff)
|
|
{
|
|
if (hediff is Hediff_PartBaseNatural || hediff is Hediff_PartBaseArtifical)
|
|
{
|
|
return hediff.TryGetComp<HediffComp_Anus>();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public static float GetFertilityChance(this HediffComp_Menstruation comp)
|
|
{
|
|
return 1.0f - Mathf.Pow(1.0f - Configurations.FertilizeChance, comp.TotalFertCum * comp.Props.basefertilizationChanceFactor);
|
|
}
|
|
|
|
public static Texture2D GetPregnancyIcon(this HediffComp_Menstruation comp, Hediff hediff)
|
|
{
|
|
if (hediff is Hediff_MechanoidPregnancy)
|
|
return ContentFinder<Texture2D>.Get(("Womb/Mechanoid_Fluid"), true);
|
|
|
|
float gestationProgress = comp.StageProgress;
|
|
|
|
ThingDef babydef;
|
|
int babycount;
|
|
HediffComp_PregeneratedBabies babiescomp = hediff?.TryGetComp<HediffComp_PregeneratedBabies>();
|
|
if (hediff is Hediff_BasePregnancy preg)
|
|
{
|
|
babydef = preg.babies?.FirstOrDefault()?.def ?? ThingDefOf.Human;
|
|
babycount = preg.babies?.Count ?? 1;
|
|
}
|
|
else if (babiescomp?.HasBaby ?? false)
|
|
{
|
|
babydef = babiescomp.babies.First().def;
|
|
babycount = babiescomp.babies.Count;
|
|
}
|
|
else
|
|
{
|
|
babydef = comp.Pawn.def;
|
|
babycount = 1;
|
|
}
|
|
|
|
string fetustex = babydef.GetModExtension<PawnDNAModExtension>()?.fetusTexPath ?? "Fetus/Fetus_Default";
|
|
string icon;
|
|
if (gestationProgress < 0.2f) icon = comp.WombTex + "_Implanted";
|
|
else if (gestationProgress < 0.3f)
|
|
{
|
|
if (babydef.race?.FleshType == FleshTypeDefOf.Insectoid) icon = "Fetus/Insects/Insect_Early00";
|
|
else icon = "Fetus/Fetus_Early00";
|
|
}
|
|
else if (gestationProgress < 0.4f) icon = fetustex + "00";
|
|
else if (gestationProgress < 0.5f) icon = fetustex + "01";
|
|
else if (gestationProgress < 0.6f) icon = fetustex + "02";
|
|
else if (gestationProgress < 0.7f) icon = fetustex + "03";
|
|
else if (gestationProgress < 0.8f) icon = fetustex + "04";
|
|
else icon = fetustex + "05";
|
|
|
|
return TryGetTwinsIcon(icon, babycount) ?? ContentFinder<Texture2D>.Get((icon), true);
|
|
}
|
|
|
|
public static Texture2D TryGetTwinsIcon(string path, int babycount)
|
|
{
|
|
for (int i = babycount; i > 1; i--)
|
|
{
|
|
Texture2D result = ContentFinder<Texture2D>.Get(path + "_Multiplet_" + i, false);
|
|
if (result != null) return result;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public static Texture2D GetCumIcon(this HediffComp_Menstruation comp)
|
|
{
|
|
string icon = comp.WombTex;
|
|
float cumpercent = comp.TotalCumPercent;
|
|
if (cumpercent < 0.001f) return ContentFinder<Texture2D>.Get("Womb/Empty", true);
|
|
else if (cumpercent < 0.01f) icon += "_Cum_00";
|
|
else if (cumpercent < 0.05f) icon += "_Cum_01";
|
|
else if (cumpercent < 0.11f) icon += "_Cum_02";
|
|
else if (cumpercent < 0.17f) icon += "_Cum_03";
|
|
else if (cumpercent < 0.23f) icon += "_Cum_04";
|
|
else if (cumpercent < 0.29f) icon += "_Cum_05";
|
|
else if (cumpercent < 0.35f) icon += "_Cum_06";
|
|
else if (cumpercent < 0.41f) icon += "_Cum_07";
|
|
else if (cumpercent < 0.47f) icon += "_Cum_08";
|
|
else if (cumpercent < 0.53f) icon += "_Cum_09";
|
|
else if (cumpercent < 0.59f) icon += "_Cum_10";
|
|
else if (cumpercent < 0.65f) icon += "_Cum_11";
|
|
else if (cumpercent < 0.71f) icon += "_Cum_12";
|
|
else if (cumpercent < 0.77f) icon += "_Cum_13";
|
|
else if (cumpercent < 0.83f) icon += "_Cum_14";
|
|
else if (cumpercent < 0.89f) icon += "_Cum_15";
|
|
else if (cumpercent < 0.95f) icon += "_Cum_16";
|
|
else icon += "_Cum_17";
|
|
Texture2D cumtex = ContentFinder<Texture2D>.Get((icon), true);
|
|
return cumtex;
|
|
}
|
|
public static Texture2D GetInsectEggedIcon(this HediffComp_Menstruation comp)
|
|
{
|
|
List<Hediff_InsectEgg> hediffs = new List<Hediff_InsectEgg>();
|
|
comp.Pawn.health.hediffSet.GetHediffs(ref hediffs);
|
|
|
|
if (hediffs.NullOrEmpty()) return null;
|
|
string path = "Womb/Insect_Egged/";
|
|
if (hediffs.Max(hediff => hediff.eggssize) > 0.3) // The threshold for "large egg" in the debug
|
|
{
|
|
if (hediffs.Count == 1) return ContentFinder<Texture2D>.Get(path + "Womb_Egged_Large", true);
|
|
else return ContentFinder<Texture2D>.Get(path + "Womb_Egged_ManyMixed", true);
|
|
}
|
|
return ContentFinder<Texture2D>.Get(path + "Womb_Egged_" + hediffs.Count(), false) ?? ContentFinder<Texture2D>.Get(path + "Womb_Egged_Many", true);
|
|
}
|
|
public static Texture2D GetWombIcon(this HediffComp_Menstruation comp)
|
|
{
|
|
Texture2D wombtex = comp.GetInsectEggedIcon();
|
|
if (wombtex != null) return wombtex;
|
|
string icon = comp.WombTex;
|
|
HediffComp_Menstruation.Stage stage = comp.curStage;
|
|
if (stage == HediffComp_Menstruation.Stage.Bleeding) icon += "_Bleeding";
|
|
|
|
wombtex = ContentFinder<Texture2D>.Get(icon, true);
|
|
|
|
return wombtex;
|
|
}
|
|
public static Texture2D GetEggIcon(this HediffComp_Menstruation comp, bool includeOvary)
|
|
{
|
|
switch (comp.CurrentVisibleStage)
|
|
{
|
|
case HediffComp_Menstruation.Stage.Follicular:
|
|
if (!includeOvary) break;
|
|
if (comp is HediffComp_InducedOvulator)
|
|
{
|
|
if (comp.Pawn.jobs.curDriver is JobDriver_Sex job &&
|
|
job.Sexprops != null &&
|
|
!job.Sexprops.usedCondom &&
|
|
(job.Sexprops.sexType == xxx.rjwSextype.Vaginal || job.Sexprops.sexType == xxx.rjwSextype.DoublePenetration))
|
|
return ContentFinder<Texture2D>.Get("Ovaries/Ovary_01", true);
|
|
else break;
|
|
}
|
|
if (comp.curStageHrs > comp.CurStageIntervalHours - 30) // Approximate time for ovulation to occur
|
|
return ContentFinder<Texture2D>.Get("Ovaries/Ovary_01", true);
|
|
else break;
|
|
case HediffComp_Menstruation.Stage.Ovulatory:
|
|
if (!includeOvary) break;
|
|
return ContentFinder<Texture2D>.Get("Ovaries/Ovary_02", true);
|
|
case HediffComp_Menstruation.Stage.Luteal:
|
|
if (!comp.IsEggExist) break;
|
|
int fertstage = comp.IsFertilized;
|
|
if (fertstage >= 0)
|
|
{
|
|
if (fertstage <= Configurations.CycleAcceleration) return ContentFinder<Texture2D>.Get("Eggs/Egg_Fertilizing02", true);
|
|
else if (fertstage <= 18) return ContentFinder<Texture2D>.Get("Eggs/Egg_Fertilized00", true);
|
|
else if (fertstage <= 54) return ContentFinder<Texture2D>.Get("Eggs/Egg_Fertilized01", true);
|
|
else return ContentFinder<Texture2D>.Get("Eggs/Egg_Fertilized02", true);
|
|
}
|
|
else if (comp.IsEggFertilizing)
|
|
{
|
|
if (comp.GetFertilityChance() < 0.5f)
|
|
return ContentFinder<Texture2D>.Get("Eggs/Egg_Fertilizing00", true);
|
|
else
|
|
return ContentFinder<Texture2D>.Get("Eggs/Egg_Fertilizing01", true);
|
|
}
|
|
else return ContentFinder<Texture2D>.Get("Eggs/Egg", true);
|
|
case HediffComp_Menstruation.Stage.Pregnant:
|
|
if (comp.Pregnancy is Hediff_MechanoidPregnancy) break;
|
|
else if (comp.GetPregnancyProgress() < 0.2f) return ContentFinder<Texture2D>.Get("Eggs/Egg_Implanted00", true);
|
|
else break;
|
|
}
|
|
return ContentFinder<Texture2D>.Get("Womb/Empty", true);
|
|
}
|
|
|
|
public static void DrawEggOverlay(this HediffComp_Menstruation comp, Rect wombRect, bool includeOvary)
|
|
{
|
|
Rect rect = new Rect(wombRect.xMax - wombRect.width / 3, wombRect.y, wombRect.width / 3, wombRect.width / 3);
|
|
GUI.color = Color.white;
|
|
GUI.DrawTexture(rect, comp.GetEggIcon(includeOvary), ScaleMode.ScaleToFit);
|
|
}
|
|
|
|
|
|
public static Texture2D GetGenitalIcon(this Pawn pawn, HediffComp_Menstruation comp, bool drawOrigin = false)
|
|
{
|
|
Hediff hediff = comp?.parent;
|
|
if (hediff == null) return ContentFinder<Texture2D>.Get("Genitals/Vagina00", true);
|
|
//HediffComp_Menstruation comp = hediff.GetMenstruationComp();
|
|
string icon;
|
|
float severity;
|
|
if (drawOrigin) severity = comp.OriginVagSize;
|
|
else severity = hediff.Severity;
|
|
if (comp != null) icon = comp.VagTex;
|
|
else icon = "Genitals/Vagina";
|
|
|
|
if (severity < 0.20f) icon += "00"; //micro
|
|
else if (severity < 0.30f) icon += "01"; //tight
|
|
else if (severity < 0.40f) icon += "02"; //tight
|
|
else if (severity < 0.47f) icon += "03"; //average
|
|
else if (severity < 0.53f) icon += "04"; //average
|
|
else if (severity < 0.60f) icon += "05"; //average
|
|
else if (severity < 0.70f) icon += "06"; //accomodating
|
|
else if (severity < 0.80f) icon += "07"; //accomodating
|
|
else if (severity < 0.87f) icon += "08"; //cavernous
|
|
else if (severity < 0.94f) icon += "09"; //cavernous
|
|
else if (severity < 1.01f) icon += "10"; //cavernous
|
|
else icon += "11"; //abyssal
|
|
|
|
return ContentFinder<Texture2D>.Get((icon), true);
|
|
}
|
|
|
|
public static Texture2D GetAnalIcon(this Pawn pawn, bool drawOrigin = false)
|
|
{
|
|
Hediff hediff = Genital_Helper.get_PartsHediffList(pawn, Genital_Helper.get_anusBPR(pawn)).FirstOrDefault(h => VariousDefOf.AllAnuses.Contains(h.def)) ??
|
|
Genital_Helper.get_PartsHediffList(pawn, Genital_Helper.get_anusBPR(pawn)).FirstOrDefault(h => h.def.defName.ToLower().Contains("anus"));
|
|
if (hediff == null) return ContentFinder<Texture2D>.Get(("Genitals/Anal00"), true);
|
|
|
|
string icon;
|
|
float severity;
|
|
HediffComp_Anus comp = hediff.GetAnusComp();
|
|
if (comp != null)
|
|
{
|
|
CompProperties_Anus Props = (CompProperties_Anus)comp.props;
|
|
icon = Props.analTex ?? "Genitals/Anal";
|
|
if (drawOrigin) severity = comp.OriginAnusSize;
|
|
else severity = hediff.Severity;
|
|
}
|
|
else
|
|
{
|
|
icon = "Genitals/Anal";
|
|
severity = hediff.Severity;
|
|
}
|
|
if (severity < 0.20f) icon += "00"; //micro
|
|
else if (severity < 0.40f) icon += "01"; //tight
|
|
else if (severity < 0.60f) icon += "02"; //average
|
|
else if (severity < 0.80f) icon += "03"; //accomodating
|
|
else if (severity < 1.01f) icon += "04"; //cavernous
|
|
else icon += "05"; //abyssal
|
|
|
|
return ContentFinder<Texture2D>.Get((icon), true);
|
|
}
|
|
|
|
public static float GestationHours(this Hediff hediff)
|
|
{
|
|
if (hediff == null)
|
|
{
|
|
Log.Error("Tried to get gestation length without a pregnancy.");
|
|
return 1f;
|
|
}
|
|
else if (hediff is Hediff_BasePregnancy rjw_preg)
|
|
return (rjw_preg.p_end_tick - rjw_preg.p_start_tick) / GenDate.TicksPerHour;
|
|
// TODO: Biotech pregnancy
|
|
else return hediff.pawn.def.race.gestationPeriodDays * GenDate.HoursPerDay;
|
|
}
|
|
|
|
public static float RandomVariabilityPercent(int recursion = 0)
|
|
{
|
|
// Humans, in days
|
|
const float mean = 1.635f;
|
|
const float stddev = 0.9138f;
|
|
const float lambda = 0.234f;
|
|
if (recursion >= 10) return mean / (28 * 2);
|
|
|
|
float variability = Rand.Gaussian(mean, stddev) - Mathf.Log(Rand.Value) / lambda;
|
|
variability /= 28 * 2; // Convert to percentage
|
|
|
|
if (variability < 0 || variability > 0.35f) return RandomVariabilityPercent(recursion + 1); // ~2% chance, about the limit on how far variability can go before things start to break
|
|
else return variability;
|
|
}
|
|
|
|
public static bool IsInEstrus(this Pawn pawn, bool visible = true)
|
|
{
|
|
if (pawn.Dead) return false;
|
|
return pawn.health?.hediffSet?.HasHediff(visible ? VariousDefOf.Hediff_Estrus : VariousDefOf.Hediff_Estrus_Concealed) ?? false;
|
|
}
|
|
|
|
public static HediffComp_Menstruation.EstrusLevel HighestEstrus(this Pawn pawn)
|
|
{
|
|
HediffComp_Menstruation.EstrusLevel res = HediffComp_Menstruation.EstrusLevel.None;
|
|
|
|
foreach(HediffComp_Menstruation comp in pawn.GetMenstruationComps())
|
|
{
|
|
switch (comp.GetEstrusLevel())
|
|
{
|
|
case HediffComp_Menstruation.EstrusLevel.None:
|
|
break;
|
|
case HediffComp_Menstruation.EstrusLevel.Concealed:
|
|
res = HediffComp_Menstruation.EstrusLevel.Concealed;
|
|
break;
|
|
case HediffComp_Menstruation.EstrusLevel.Visible:
|
|
return HediffComp_Menstruation.EstrusLevel.Visible;
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
public static bool HasIUD(this Pawn pawn)
|
|
{
|
|
if (pawn.health.hediffSet.HasHediff(VariousDefOf.RJW_IUD)) return true;
|
|
if (ModsConfig.BiotechActive && pawn.health.hediffSet.HasHediff(HediffDefOf.ImplantedIUD)) return true;
|
|
return false;
|
|
}
|
|
|
|
public static float DamagePants(this Pawn pawn, float fluidAmount)
|
|
{
|
|
if (pawn.apparel == null) return 0;
|
|
Apparel pants = null;
|
|
foreach(Apparel apparel in pawn.apparel.WornApparel.Where(app => app.def.apparel.bodyPartGroups.Contains(BodyPartGroupDefOf.Legs)))
|
|
{
|
|
if (apparel.def.apparel.LastLayer == ApparelLayerDefOf.OnSkin)
|
|
{
|
|
pants = apparel;
|
|
break;
|
|
}
|
|
else if (pants == null || apparel.def.apparel.LastLayer == ApparelLayerDefOf.Middle)
|
|
// Either grab whatever's available or reassign the pants from shell to a middle
|
|
pants = apparel;
|
|
// Pants are middle and this is a shell
|
|
else continue;
|
|
}
|
|
if (pants == null) return 0;
|
|
|
|
const float HPPerMl = 0.5f;
|
|
|
|
DamageWorker.DamageResult damage = pants.TakeDamage(new DamageInfo(DamageDefOf.Deterioration, fluidAmount * HPPerMl, spawnFilth: false));
|
|
|
|
if (pants.Destroyed && PawnUtility.ShouldSendNotificationAbout(pawn) && !pawn.Dead)
|
|
Messages.Message("MessageWornApparelDeterioratedAway".Translate(GenLabel.ThingLabel(pants.def, pants.Stuff), pawn).CapitalizeFirst(), pawn, MessageTypeDefOf.NegativeEvent);
|
|
|
|
return damage.totalDamageDealt;
|
|
}
|
|
}
|
|
}
|