rjw-genes/Source/Genes/Genitalia/GenitaliaChanger.cs

93 lines
2.8 KiB
C#

using rjw;
using RimWorld;
using Verse;
using System.Collections.Generic;
namespace RJW_Genes
{
internal class GenitaliaChanger
{
/// <summary>
/// This method changes the pawns genitalia to the given types of genitalia.
/// All genitals will be changed, e.g. pawns with 2 penises (for whatever reason) or is a futa, all genitals will be changed.
/// </summary>
/// <param name="pawn">the pawn who's genitals will be changed</param>
/// <param name="penisReplacement">the new type of penis</param>
/// <param name="vaginaReplacement">the new type of vagina</param>
/// <param name="anusReplacement">the new type of anus</param>
public static void ChangeGenitalia(Pawn pawn, HediffDef penisReplacement, HediffDef vaginaReplacement, HediffDef anusReplacement)
{
var oldParts = Genital_Helper.get_AllPartsHediffList(pawn);
BodyPartRecord correctBPR;
if (!oldParts.NullOrEmpty())
{
Hediff replacementGenital;
CompHediffBodyPart CompHediff;
foreach (Hediff existingGenital in oldParts)
{
if (IsArtificial(existingGenital))
continue;
correctBPR = Genital_Helper.get_genitalsBPR(pawn);
replacementGenital = null;
CompHediff = null;
if (Genital_Helper.is_penis(existingGenital))
replacementGenital = HediffMaker.MakeHediff(penisReplacement, pawn, correctBPR);
if (Genital_Helper.is_vagina(existingGenital))
replacementGenital = HediffMaker.MakeHediff(vaginaReplacement, pawn, correctBPR);
if (IsAnus(existingGenital))
{
correctBPR = Genital_Helper.get_anusBPR(pawn);
replacementGenital = HediffMaker.MakeHediff(anusReplacement, pawn, correctBPR);
}
if (replacementGenital != null)
{
CompHediff = replacementGenital.TryGetComp<rjw.CompHediffBodyPart>();
if (CompHediff != null)
{
CompHediff.initComp(pawn);
CompHediff.updatesize();
}
GenderHelper.ChangeSex(pawn, () =>
{
pawn.health.RemoveHediff(existingGenital);
pawn.health.AddHediff(replacementGenital, correctBPR);
});
}
}
}
else
{
Messages.Message("RJW_Genes_GenitalsNotAlterable".Translate(pawn), pawn, MessageTypeDefOf.SilentInput);
return;
}
}
public static bool IsAnus(Hediff candidate)
{
return candidate.def.defName.ToLower().Contains("anus"); }
public static bool IsArtificial(Hediff candidate)
{
return candidate.def.defName.ToLower().Contains("bionic") || candidate.def.defName.ToLower().Contains("archo");
}
public static void RemoveAllGenitalia(Pawn pawn)
{
var parts = Genital_Helper.get_AllPartsHediffList(pawn);
foreach (var part in parts)
{
pawn.health.RemoveHediff(part);
}
}
}
}