private static IEnumerable <BodyPartRecord> WhichPartsMutationIsAllowedOn(
Pawn pawn,
HediffMutationDef def,
BodyPartRecord part,
Dictionary <string, HashSet <BodyPartDef> > excludedPartDefsForTag,
Dictionary <string, HashSet <BodyPartRecord> > excludedPartsForTag
)
{
IEnumerable <BodyPartRecord> allParts;
if (def.affectedParts == null)
{
List <BodyPartRecord> list = new List <BodyPartRecord>();
list.AddRange(GetChildBodyParts(part));
allParts = list;
}
else
{
allParts = pawn.health.hediffSet.GetNotMissingParts().Where(x => def.affectedParts.Contains(x.def));
}
foreach (string tag in def.exclusives)
{
if (def.affectsAllParts)
{
HashSet <BodyPartDef> excluded;
if (excludedPartDefsForTag.TryGetValue(tag, out excluded))
{
allParts = allParts.Where(x => !excluded.Contains(x.def));
}
}
else
{
HashSet <BodyPartRecord> excluded;
if (excludedPartsForTag.TryGetValue(tag, out excluded))
{
allParts = allParts.Except(excluded);
}
}
}
return(allParts);
}
public static IEnumerable <BodyPartRecord> MutatePawn(Pawn pawn, HediffRadiation radiation, float mutateAmount, float rareRatio, out Mutation mutationResult)
{
mutationResult = null;
BodyPartRecord part = radiation.Part;
Debug.Log("Finding mutation for part: " + part);
Debug.Log("Rare ratio: " + rareRatio);
Dictionary <string, HashSet <BodyPartDef> > excludedPartDefsForTag = new Dictionary <string, HashSet <BodyPartDef> >();
Dictionary <string, HashSet <BodyPartRecord> > excludedPartsForTag = new Dictionary <string, HashSet <BodyPartRecord> >();
foreach (Mutation existingMutation in pawn.health.hediffSet.GetHediffs <Mutation>())
{
foreach (string tag in existingMutation.def.exclusives)
{
HashSet <BodyPartDef> setDef = excludedPartDefsForTag.TryGetValue(tag);
if (setDef == null)
{
excludedPartDefsForTag[tag] = setDef = new HashSet <BodyPartDef>();
}
setDef.Add(existingMutation.Part.def);
HashSet <BodyPartRecord> set = excludedPartsForTag.TryGetValue(tag);
if (set == null)
{
excludedPartsForTag[tag] = set = new HashSet <BodyPartRecord>();
}
set.Add(existingMutation.Part);
}
}
Debug.Log("Excluded parts for tags: " + Debug.AsText(excludedPartsForTag));
Debug.Log("Excluded defs for tags: " + Debug.AsText(excludedPartDefsForTag));
var mutations =
MutationsHelper.Mutations.Where(x => x.relatedParts == null || x.relatedParts.Contains(part.def));
Debug.Log("All applicable mutations: " + Debug.AsText(mutations));
Dictionary <HediffMutationDef, IEnumerable <BodyPartRecord> > allowedMutations = new Dictionary <HediffMutationDef, IEnumerable <BodyPartRecord> >();
foreach (var mutation in mutations)
{
var parts = WhichPartsMutationIsAllowedOn(pawn, mutation, part, excludedPartDefsForTag, excludedPartsForTag);
Debug.Log(" " + mutation + ": to " + Debug.AsText(parts));
if (parts.Count() == 0)
{
continue;
}
allowedMutations[mutation] = parts;
}
Debug.Log("All allowed mutations: " + Debug.AsText(allowedMutations));
if (allowedMutations.Count() == 0)
{
Debug.Log("No mutation possible!");
return(null);
}
HediffMutationDef mutationDef = allowedMutations.Keys.RandomElementByWeight(x => (float)Math.Pow(x.likelihood, 1f - rareRatio));
Debug.Log("Chose mutation: " + mutationDef);
var applicableParts = allowedMutations[mutationDef];
Debug.Log("Can be applied to body parts: " + Debug.AsText(applicableParts));
if (allowedMutations.Count() == 0)
{
Debug.Log("No matching body parts!");
return(null);
}
var chosenPart = applicableParts.RandomElement();
Debug.Log("Chose part: " + chosenPart);
var chosenParts = pawn.health.hediffSet.GetNotMissingParts()
.Where(x => mutationDef.affectsAllParts ? x.def == chosenPart.def : x == chosenPart);
foreach (var partToMutate in chosenParts)
{
Debug.Log("Adding to part: " + partToMutate);
if (pawn.health.hediffSet.GetHediffs <Mutation>().Where(x => x.def == mutationDef && x.Part == partToMutate).Any())
{
Debug.Log(" But it already has this mutation!");
continue;
}
Mutation mutation = HediffMaker.MakeHediff(mutationDef, pawn, partToMutate) as Mutation;
if (mutation == null)
{
continue;
}
mutationResult = mutation;
pawn.health.AddHediff(mutation, partToMutate, null, null);
}
return(chosenParts);
}
