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

using RimWorld;
using Verse;
using rjw;
using System;

namespace RJW_Genes
{
    public class GenitaliaUtility
    {

        /// <summary>
        /// Returns the first (non) overwritten gene from the rjw_genes genitalia genes.
        /// In case the pawn has none, as default the human one is returned.
        /// </summary>
        /// <param name="pawn">the pawn whom to find genitaliagenes for</param>
        /// <returns>The first GeneDef of the pawn related to GenitaliaTypes</returns>
        public static GeneDef GetGenitaliaTypeGeneForPawn(Pawn pawn)
        {
           
            foreach (var gene in pawn.genes.GenesListForReading)
            {
                if (gene.def.defName.Contains("rjw_genes") && gene.def.defName.EndsWith("_genitalia"))
                    if (!gene.Overridden)
                        return gene.def;
            }
            return null;
        }


        public static HediffDef GetPenisForGene(GeneDef gene)
        {
            if (gene == null)
                return Genital_Helper.average_penis;

            switch (gene.defName)
            {
                case "rjw_genes_equine_genitalia":      return Genital_Helper.equine_penis;
                case "rjw_genes_canine_genitalia":      return Genital_Helper.canine_penis;
                case "rjw_genes_feline_genitalia":      return Genital_Helper.feline_penis;
                case "rjw_genes_demonic_genitalia":     return Genital_Helper.demon_penis;
                case "rjw_genes_dragon_genitalia":      return Genital_Helper.dragon_penis;
                case "rjw_genes_slime_genitalia":       return Genital_Helper.slime_penis;
                case "rjw_genes_ovipositor_genitalia":  return Genital_Helper.ovipositorM;

                default: return Genital_Helper.average_penis;
            }
        }


        public static HediffDef GetVaginaForGene(GeneDef gene)
        {
            if (gene == null)
                return Genital_Helper.average_vagina;

            switch (gene.defName)
            {
                case "rjw_genes_equine_genitalia": return Genital_Helper.equine_vagina;
                case "rjw_genes_canine_genitalia": return Genital_Helper.canine_vagina;
                case "rjw_genes_feline_genitalia": return Genital_Helper.feline_vagina;
                case "rjw_genes_demonic_genitalia": return Genital_Helper.demon_vagina;
                case "rjw_genes_dragon_genitalia": return Genital_Helper.dragon_vagina;
                case "rjw_genes_slime_genitalia": return Genital_Helper.slime_vagina;
                case "rjw_genes_ovipositor_genitalia": return Genital_Helper.ovipositorF;

                default: return Genital_Helper.average_vagina;
            }
        }

        public static HediffDef GetAnusForGene(GeneDef gene)
        {
            if (gene == null)
                return Genital_Helper.average_anus;
            switch (gene.defName)
            {
                //TODO: Do I want the default to be generic or average for feline,equine and canine?
                case "rjw_genes_equine_genitalia": return Genital_Helper.average_anus;
                case "rjw_genes_canine_genitalia": return Genital_Helper.average_anus;
                case "rjw_genes_feline_genitalia": return Genital_Helper.average_anus;
                case "rjw_genes_demonic_genitalia": return Genital_Helper.demon_anus;
                case "rjw_genes_dragon_genitalia": return Genital_Helper.average_anus;
                case "rjw_genes_slime_genitalia": return Genital_Helper.slime_anus;
                case "rjw_genes_ovipositor_genitalia": return Genital_Helper.insect_anus;

                default: return Genital_Helper.generic_anus;
            }
        }

        public static HediffDef GetBreastsForGene(GeneDef gene)
        {
            if (gene == null)
                return Genital_Helper.average_breasts;

            switch (gene.defName)
            {
                //TODO: Do I want the default to be generic or average?
                case "rjw_genes_equine_genitalia": return Genital_Helper.average_breasts;
                case "rjw_genes_canine_genitalia": return Genital_Helper.average_breasts;
                case "rjw_genes_feline_genitalia": return Genital_Helper.average_breasts;
                case "rjw_genes_demonic_genitalia": return Genital_Helper.average_breasts;
                case "rjw_genes_dragon_genitalia": return Genital_Helper.average_breasts;
                case "rjw_genes_slime_genitalia": return Genital_Helper.slime_breasts;
                case "rjw_genes_udder_breasts": return Genital_Helper.udder_breasts;
                case "rjw_genes_ovipositor_genitalia": return Genital_Helper.average_breasts;

                default: return Genital_Helper.generic_breasts;
            }
        }

        public static bool PawnStillNeedsGenitalia(Pawn pawn)
        {
            // There is the issue that the genes fire in a pseudo-random order 
            // Hence it can happen that the pawn still needs genitalia 
            // I wanted to make a simple lookup, but I think the genes are applied for all humans encountered so it could be huge
            // So the heuristic is to check if the pawn has any of the 3 standard genitalia OR has all genes ticked that says "I don't want genitalia".
            if (pawn == null) return false;

            bool pawn_has_any_genitalia = 
                Genital_Helper.has_genitals(pawn) || Genital_Helper.has_anus(pawn) || Genital_Helper.has_breasts(pawn);

            bool pawn_is_not_supposed_to_have_genitalia =
                pawn.genes.GenesListForReading.Any(x => x.def.defName == "rjw_genes_no_penis");

            if (pawn_is_not_supposed_to_have_genitalia)
                return false;
            else
                return !pawn_has_any_genitalia;

        }

        public static bool IsBreasts(Hediff candidate)
        {
            return candidate.def.defName.ToLower().Contains("breast");
        }
    }
}