rjw-genes/Source/Genes/Gender/GenderUtility.cs

using Verse;
using rjw;
using RimWorld;
using System.Linq;
using System;
using System.Collections.Generic;
using HarmonyLib;

namespace RJW_Genes
{
    public class GenderUtility
    {
        /// <summary>
        /// Returns if a Pawn is female (Gender==Female) or if it should be (Gene==FemaleOnly)
        /// This is used as a small helper, as the genes might fire in different orders.
        /// </summary>
        public static bool IsFemale(Pawn pawn)
        {
            return
                pawn.gender == Gender.Female || pawn.genes.GenesListForReading.Any(gene => gene.def.defName.EqualsIgnoreCase(GeneDefOf.rjw_genes_female_only.defName));
        }

        /// <summary>
        /// Returns if a Pawn is male (Gender==Male) or if it should be (Gene==MaleOnly)
        /// This is used as a small helper, as the genes might fire in different orders.
        /// </summary>
        public static bool IsMale(Pawn pawn)
        {
            return
                pawn.gender == Gender.Male || pawn.genes.GenesListForReading.Any(gene => gene.def.defName.EqualsIgnoreCase(GeneDefOf.rjw_genes_male_only.defName));
        }

        /// <summary>
        /// Adjusts the Body Type to match the given target gender.
        /// This is only for "drawing" attributes of the pawn, the genitalia are untouched at this point. 
        /// (for male and female only, baby,child and hulks don't change)
        /// </summary>
        /// <param name="pawn"></param>
        /// <param name="targetGender"></param>
        public static void AdjustBodyToTargetGender(Pawn pawn, Gender targetGender)
        {
            if (pawn == null)
                return;
            if (pawn.story.bodyType == BodyTypeDefOf.Baby || pawn.story.bodyType == BodyTypeDefOf.Hulk || pawn.story.bodyType == BodyTypeDefOf.Child)
                return;

            if (targetGender == Gender.Male)
            {
                pawn.story.bodyType = BodyTypeDefOf.Male;
            }
            else if (targetGender == Gender.Female)
            {
                pawn.story.bodyType = BodyTypeDefOf.Female; 
                pawn.style.beardDef = BeardDefOf.NoBeard;
            }

            // Re-Choose heads if it is wrong gender
            if (pawn.story.headType.gender == Gender.None || pawn.story.headType.gender == targetGender)
            {
                // Do nothing, Gender of Heat is Neutral or matches
            }
            else
            {
                // Below line tries to get (and set) an available head from the backstory, if it returns true everything worked if it returns false we log it
                if(! pawn.story.TryGetRandomHeadFromSet(DefDatabase<HeadTypeDef>.AllDefs.Where((Func<HeadTypeDef, bool>)(x => x.randomChosen))))
                {
                    Log.Message("Failed to retrieve a correct-gender head for the pawn " + pawn.Name);
                };
            } 

            // Force Redraw at the spot
            pawn.Drawer.renderer.graphics.SetAllGraphicsDirty();
        }

        // Fetch these once at load time because they don't change inside RJW
        private static readonly List<HediffDef> wasSexThoughts = Traverse.Create(typeof(GenderHelper)).Field("old_sex_list").GetValue<List<HediffDef>>();
        private static readonly List<HediffDef> sexChangeThoughts = Traverse.Create(typeof(GenderHelper)).Field("SexChangeThoughts").GetValue<List<HediffDef>>();

        /// <summary>
        /// This method removes all RJW-Sexchange-Hediffs from the pawn. 
        /// It used with the RJW_Gene.Notify_OnPawnGeneration() to check for pawns on spawn.
        /// 
        /// Fixes Issue #32, where pawns that spawn fresh with a "all female" gene may have m2f thoughts. 
        /// </summary>
        /// <param name="pawn">The pawn that needs to have SexChange-Thoughts removed.</param>
        public static void RemoveAllSexChangeThoughts(Pawn pawn)
        {
            // Shouldn't ever be true in the normal case, but this stops someone from calling this with an incorrect setup
            if (pawn?.health == null)
                return;

            if(wasSexThoughts == null || sexChangeThoughts == null || !wasSexThoughts.Any() || !sexChangeThoughts.Any())
            {
                Log.Warning($"Couldn't get values from RJW.\nold_sex_list: {wasSexThoughts.ToStringSafeEnumerable()}\nSexChangeThoughts: {sexChangeThoughts.ToStringSafeEnumerable()}");
                return;
            }

            foreach(var def in wasSexThoughts.Concat(sexChangeThoughts))
            {
                var hediff = pawn.health.hediffSet.GetFirstHediffOfDef(def);
                if (hediff != null)
                    pawn.health.RemoveHediff(hediff);
            }
        }
    }
}