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

59 lines
1.6 KiB
C#

using Verse;
using rjw;
using RimWorld;
namespace RJW_Genes
{
public class Gene_ExtraBreasts : RJW_Gene
{
internal Hediff additional_breasts;
public override void PostMake()
{
base.PostMake();
// Tits are only added for female pawns!
if (GenderUtility.IsFemale(pawn) && additional_breasts == null)
{
CreateAndAddBreasts();
}
}
public override void PostAdd()
{
base.PostAdd();
// Tits are only added for female pawns!
if (GenderUtility.IsFemale(pawn) && additional_breasts == null)
{
CreateAndAddBreasts();
}
}
public override void PostRemove()
{
base.PostRemove();
if(additional_breasts != null)
pawn.health.RemoveHediff(additional_breasts);
}
internal void CreateAndAddBreasts()
{
var correctGene = GenitaliaUtility.GetGenitaliaTypeGeneForPawn(pawn);
var breastDef = GenitaliaUtility.GetBreastsForGene(correctGene);
var partBPR = Genital_Helper.get_breastsBPR(pawn);
additional_breasts = HediffMaker.MakeHediff(breastDef, pawn);
var CompHediff = additional_breasts.TryGetComp<rjw.CompHediffBodyPart>();
if (CompHediff != null)
{
CompHediff.initComp(pawn);
CompHediff.updatesize();
}
pawn.health.AddHediff(additional_breasts, partBPR);
}
}
}