RJW_Animal_Gene_Inheritance/Source/RJW_BGS/InheritanceUtility.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Verse;
using RimWorld;

namespace RJW_BGS
{
    public class InheritanceUtility
    {
        public static List<GeneDef> AnimalInheritedGenes(Pawn father, Pawn mother)
        {
            List<GeneDef> genelist = new List<GeneDef>();
            if (father != null && !father.RaceProps.Humanlike)
            {
                PawnKindDef pawnKindDef = father.kindDef;
                RaceGeneDef raceGeneDef = RJWcopy.GetRaceGenDefInternal(pawnKindDef);
                if (raceGeneDef != null)
                {
                    GeneDef gene = null;
                    //In case you hit a modded gene not currently active try again.
                    for (int i = 0; i < 50 || gene == null; i++)
                    {
                        if (raceGeneDef.genes.Any())
                        {
                          gene = DefDatabase<GeneDef>.GetNamed(raceGeneDef.genes.RandomElement());
                        }
                    }
                    if (gene != null)
                    {
                        genelist.Add(gene);
                    }
                }
            }

            if (mother != null && !mother.RaceProps.Humanlike)
            {
                PawnKindDef pawnKindDef = mother.kindDef;
                RaceGeneDef raceGeneDef = RJWcopy.GetRaceGenDefInternal(pawnKindDef);
                if (raceGeneDef != null)
                {
                    GeneDef gene = null;
                    //In case you hit a modded gene not currently active try again.
                    for (int i = 0; i < 50 || gene == null; i++)
                    {
                        if (raceGeneDef.genes.Any())
                        {
                            gene = DefDatabase<GeneDef>.GetNamed(raceGeneDef.genes.RandomElement());
                        }
                    }
                    if (gene != null)
                    {
                        genelist.Add(gene);

                    }
                    
                }
            }
            return genelist;
        }

        public static List<GeneDef> SelectGenes(Pawn pawn)
        {
            List<GeneDef> genelist = new List<GeneDef>();
            PawnKindDef pawnKindDef = pawn.kindDef;
            RaceGeneDef raceGeneDef = RJWcopy.GetRaceGenDefInternal(pawnKindDef);
            if (raceGeneDef != null)
            {
                int num1 = raceGeneDef.genes.Count;
                int num2 = raceGeneDef.genechances.Count;
                if (num1 != num2)
                {
                    Log.Error("The amount of genes and genechanches are different in " + raceGeneDef.defName + ". Can't select genes to inherit");
                    return genelist;
                }
                for (int i = 0; i<num1; i++)
                {
                    if (raceGeneDef.genechances[i] > Rand.Range(0.01f, 1f))
                    {
                        genelist.Add(DefDatabase<GeneDef>.GetNamed(raceGeneDef.genes[i]));
                    }
                }
            }
            return genelist;
        }

        public static void AddGenes(Pawn pawn, List<GeneDef> genes)
        {
            foreach (GeneDef gene in genes)
            {
                pawn.genes.AddGene(gene, false);
            }
        }

        public static void NewGenes(Pawn mother, Pawn dad, Pawn baby)
        {
            if (baby.RaceProps.Humanlike)
            {
                if (baby.genes == null)
                {
                    baby.genes = new Pawn_GeneTracker(baby);
                }

                //Remove the hair and skin genes pawns always start with, should get correct ones from human parent anyway.
                for (int i = baby.genes.Endogenes.Count - 1; i >= 0; i--)
                {
                    baby.genes.RemoveGene(baby.genes.Endogenes[i]);
                }

                List<GeneDef> genes = PregnancyUtility.GetInheritedGenes(dad, mother);
                List<GeneDef> beastgenes = InheritanceUtility.AnimalInheritedGenes(dad, mother);
                InheritanceUtility.AddGenes(baby, beastgenes);
                InheritanceUtility.AddGenes(baby, genes);
                if(baby.genes.GetFirstEndogeneByCategory(EndogeneCategory.Melanin) == null)
                {
                    AddSkinColor(mother, dad, baby);
                }
            }
        }

        public static void AddSkinColor(Pawn mother, Pawn father, Pawn baby)
        {
            if (mother != null && mother.genes != null)
            {
                GeneDef gene = mother.genes.GetFirstEndogeneByCategory(EndogeneCategory.Melanin);
                if (gene != null)
                {
                    baby.genes.AddGene(gene, false);
                }
            }
            else if (father != null && father.genes != null)
            {
                GeneDef gene = father.genes.GetFirstEndogeneByCategory(EndogeneCategory.Melanin);
                if (gene != null)
                {
                    baby.genes.AddGene(gene, false);
                }
            }
            else
            {
                Log.Message("Could not find skincolor of " + baby.Name + "'s parents, giving random skincolor.");
                baby.genes.AddGene(PawnSkinColors.RandomSkinColorGene(baby), false);
            }
        }

    }
}