rjw-genes/Source/Animal_Inheritance/Defs/RaceGeneDef_Helper.cs

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

namespace RJW_BGS
{
    public class RaceGeneDef_Helper
    {
      	//code based on racegroupdefinternal which has a similar function
		public static RaceGeneDef GetRaceGeneDefInternal(Pawn pawn)
        {
			List<RaceGeneDef> Valids = ValidRaceGeneDefs(pawn);
			if (Valids.Count > 0)
            {
				RaceGeneDef result = Valids.MaxBy(r => r.priority);
				return result;
			}			
			return null;
		}

		public static List<RaceGeneDef> ValidRaceGeneDefs(Pawn pawn)
		{
			PawnKindDef kindDef = pawn.kindDef;
			if (kindDef == null)
			{
				ModLog.Warning($"Error looking up PawnKindDef for {pawn.Name} - Could not lookup Animal Inheritance Genes");
				return null;
			}
			
			string raceName = kindDef.race.defName;
			string pawnKindName = kindDef.defName;
			//Wild animals have no name, so we will use pawnkindname instead
			string pawnName = pawn.Name != null ? pawn.Name.ToStringFull : pawnKindName; 
			PawnData pawnData = SaveStorage.DataStore.GetPawnData(pawn);
			RaceGroupDef raceGroupDef = pawnData.RaceSupportDef;

			if (RJW_BGSSettings.rjw_bgs_detailed_debug)
				ModLog.Message($"Looking up Animal-Inheritable Genes for {pawnName} with KindDef {kindDef.defName},RaceName {raceName}, PawnKind {pawnKindName} and RaceGroup {raceGroupDef.defName}");

			IEnumerable<RaceGeneDef> allDefs = DefDatabase<RaceGeneDef>.AllDefs;
			List<RaceGeneDef> pawnKindDefs = allDefs.Where(delegate (RaceGeneDef group)
			{
				List<string> pawnKindNames = group.pawnKindNames;
				return pawnKindNames != null && pawnKindNames.Contains(pawnKindName);
			}).ToList<RaceGeneDef>();
			if (pawnKindDefs.Count() > 0)
            {
				DebugPrintRaceGeneDefs("PawnKindDefs", pawnName,pawnKindDefs);
				return pawnKindDefs;
			}
			else if (RJW_BGSSettings.rjw_bgs_detailed_debug)
				ModLog.Message($"Did not find PawnKindDefs for {pawnName}");

			List<RaceGeneDef> raceKindDefs = allDefs.Where(delegate (RaceGeneDef group)
			{
				List<string> raceNames = group.raceNames;
				return raceNames != null && raceNames.Contains(raceName);
			}).ToList<RaceGeneDef>();
			if (raceKindDefs.Count() > 0)
			{
				DebugPrintRaceGeneDefs("PawnKindDefs", pawnName, raceKindDefs);
				return raceKindDefs;
			}
			else if (RJW_BGSSettings.rjw_bgs_detailed_debug)
				ModLog.Message($"Did not find RaceKindDefs for {pawnName}");

			List<RaceGeneDef> raceGroupDefs = new List<RaceGeneDef>();
			if (raceGroupDef != null)
			{
				raceGroupDefs = allDefs.Where(delegate (RaceGeneDef group)
				{
					string raceGroupDefName = group.raceGroup;
					List<string> list_raceGroupDefName = group.raceGroups;
					return (raceGroupDefName != null && raceGroupDefName == raceGroupDef.defName) 
							|| (list_raceGroupDefName != null && list_raceGroupDefName.Contains(raceGroupDef.defName));
				}).ToList<RaceGeneDef>();
			}
			
			if (raceGroupDefs.Count() > 0)
            {
				DebugPrintRaceGeneDefs("RaceKindDefs", pawnName, raceGroupDefs);
				return raceGroupDefs;
			}
			else if (RJW_BGSSettings.rjw_bgs_detailed_debug)
				ModLog.Message($"Did not find RaceGroupDefs for {pawnName}");

			ModLog.Message($"Did not find any Genes inheritable for {pawnName}");
			return new List<RaceGeneDef>();
		}

		private static void DebugPrintRaceGeneDefs(String header,String identifier,List<RaceGeneDef> defs)
        {
			if (RJW_BGSSettings.rjw_bgs_detailed_debug)
            {
				var defString = "[";
				foreach (RaceGeneDef raceGeneDef in defs)
					defString += $"({raceGeneDef.priority}:{raceGeneDef.defName} - {raceGeneDef.genes.Count} Genes)";
				defString += "]";
				ModLog.Message($"Found the following {header}-Genes for {identifier}: {defString}");
			}
		}
    } 
}