rjw-genes/Source/Genes/ExtraGenitalia/Gene_ExtraVagina.cs

89 lines
2.6 KiB
C#

using Verse;
using rjw;
using RimWorld;
using System.Linq;
namespace RJW_Genes
{
public class Gene_ExtraVagina : RJW_Gene
{
internal Hediff additional_vagina;
//TODO: This works ingame when genes are added, but if there is
//a gene (e.g. ovipositor) in creation it does not work as expected (only has one genital)
//Penis works as expected
public override void PostMake()
{
base.PostMake();
// Some sources add Genes before they fire, e.g. Character Editor
// This should harden the gene, to solve #19
if (HasAlreadyTwoVaginas())
{
return;
}
// Vaginas are only added for female pawns!
if (GenderUtility.IsFemale(pawn) && additional_vagina == null)
{
CreateAndAddVagina();
}
}
public override void PostAdd()
{
base.PostAdd();
// Some sources add Genes before they fire, e.g. Character Editor
// This should harden the gene, to solve #19
if (HasAlreadyTwoVaginas())
{
return;
}
// Vaginas are only added for female pawns!
if (GenderUtility.IsFemale(pawn) && additional_vagina == null)
{
CreateAndAddVagina();
}
}
public override void PostRemove()
{
base.PostRemove();
if(additional_vagina != null)
pawn.health.RemoveHediff(additional_vagina);
}
internal void CreateAndAddVagina()
{
var correctGene = GenitaliaUtility.GetGenitaliaTypeGeneForPawn(pawn);
HediffDef vaginaDef = GenitaliaUtility.GetVaginaForGene(correctGene);
var partBPR = Genital_Helper.get_genitalsBPR(pawn);
additional_vagina = HediffMaker.MakeHediff(vaginaDef, pawn);
var CompHediff = additional_vagina.TryGetComp<rjw.CompHediffBodyPart>();
if (CompHediff != null)
{
CompHediff.initComp(pawn);
CompHediff.updatesize();
}
pawn.health.AddHediff(additional_vagina, partBPR);
}
internal bool HasAlreadyTwoVaginas()
{
if (pawn == null)
return false;
var possible_breasts =
Genital_Helper.get_AllPartsHediffList(pawn).Where(t => Genital_Helper.is_vagina(t));
return possible_breasts.Count() >= 2;
}
}
}