diff --git a/Source/Genes/Gender/Gene_FemaleOnly.cs b/Source/Genes/Gender/Gene_FemaleOnly.cs index 0cfbe0d..110370f 100644 --- a/Source/Genes/Gender/Gene_FemaleOnly.cs +++ b/Source/Genes/Gender/Gene_FemaleOnly.cs @@ -4,7 +4,7 @@ using rjw; namespace RJW_Genes { - public class Gene_FemaleOnly : Gene + public class Gene_FemaleOnly : RJW_Gene { public override void PostMake() { @@ -24,6 +24,8 @@ namespace RJW_Genes private void AdjustPawnToFemale() { + Log.Message($"Gene_FemaleOnly AdjustPawnToFemale | {pawn} | {pawn.gender}"); + // Here we really use the Gender.Female and not our helper IsFemale(pawn) if (pawn.gender == Gender.Female) return; @@ -38,5 +40,11 @@ namespace RJW_Genes } } + public override void Notify_OnPawnGeneration() + { + base.Notify_OnPawnGeneration(); + // If this is Pawn generation, then we can assume that the pawn was never any gender other than female, so they shouldn't have sex change thoughts. + GenderUtility.RemoveAllSexChangeThoughts(pawn); + } } } diff --git a/Source/Genes/Gender/Gene_MaleOnly.cs b/Source/Genes/Gender/Gene_MaleOnly.cs index 51806cf..eba7ce5 100644 --- a/Source/Genes/Gender/Gene_MaleOnly.cs +++ b/Source/Genes/Gender/Gene_MaleOnly.cs @@ -4,7 +4,7 @@ using rjw; namespace RJW_Genes { - public class Gene_MaleOnly : Gene + public class Gene_MaleOnly : RJW_Gene { public override void PostMake() { @@ -38,5 +38,11 @@ namespace RJW_Genes } } + public override void Notify_OnPawnGeneration() + { + base.Notify_OnPawnGeneration(); + // If this is Pawn generation, then we can assume that the pawn was never any gender other than male, so they shouldn't have sex change thoughts. + GenderUtility.RemoveAllSexChangeThoughts(pawn); + } } } diff --git a/Source/Genes/Patch_AddNotifyOnGeneration.cs b/Source/Genes/Patch_AddNotifyOnGeneration.cs new file mode 100644 index 0000000..3395a81 --- /dev/null +++ b/Source/Genes/Patch_AddNotifyOnGeneration.cs @@ -0,0 +1,23 @@ +using HarmonyLib; +using System.Linq; +using Verse; + +namespace RJW_Genes.Genes +{ + [HarmonyPatch] + public static class Patch_AddNotifyOnGeneration + { + [HarmonyPatch(typeof(PawnGenerator), "GenerateGenes")] + [HarmonyPostfix] + public static void PawnGenerator_GenerateGenes_Postfix(Pawn pawn) + { + if (pawn.genes == null) return; + + foreach(var gene in pawn.genes.GenesListForReading) + { + if (gene is RJW_Gene rjwGene) + rjwGene.Notify_OnPawnGeneration(); + } + } + } +} diff --git a/Source/Genes/RJW_Gene.cs b/Source/Genes/RJW_Gene.cs index d03440c..dfa05bf 100644 --- a/Source/Genes/RJW_Gene.cs +++ b/Source/Genes/RJW_Gene.cs @@ -12,5 +12,13 @@ namespace RJW_Genes if (GenitaliaUtility.PawnStillNeedsGenitalia(pawn)) Sexualizer.sexualize_pawn(pawn); } + + /// + /// Executed via PawnGenerator.GenerateGenes at Pawn generation + /// Allows for execution of code that should only happen during PawnGeneration + /// + public virtual void Notify_OnPawnGeneration() + { + } } } diff --git a/Source/Rjw-Genes.csproj b/Source/Rjw-Genes.csproj index 68a3e34..d85935b 100644 --- a/Source/Rjw-Genes.csproj +++ b/Source/Rjw-Genes.csproj @@ -131,6 +131,7 @@ +