mirror of
https://github.com/vegapnk/RJW-Genes.git
synced 2024-08-15 00:23:31 +00:00
90 lines
2.6 KiB
C#
90 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;
|
|
}
|
|
}
|
|
}
|