73 lines
3.5 KiB
C#
73 lines
3.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Verse;
|
|
using rjw;
|
|
using Autopsy;
|
|
using HarmonyLib;
|
|
|
|
namespace RJW_patch_Autopsy
|
|
{
|
|
[HarmonyPatch(typeof(NewMedicalRecipesUtility), "TraverseBody")]
|
|
public static class NewMedicalRecipesUtilityPatch
|
|
{
|
|
[HarmonyPrefix]
|
|
public static bool AddRjwParts(RecipeInfo recipeInfo, Corpse corpse, float skillChance, ref IEnumerable<Thing> __result)
|
|
{
|
|
//Collect vanilla parts
|
|
BodyPartRecord core = corpse.InnerPawn.RaceProps.body.corePart;
|
|
List<BodyPartRecord> queue = new List<BodyPartRecord> { core };
|
|
HediffSet hediffSet = corpse.InnerPawn.health.hediffSet;
|
|
List<Thing> results = new List<Thing>();
|
|
List<BodyPartRecord> damagedParts = new List<BodyPartRecord>();
|
|
while (queue.Count > 0)
|
|
{
|
|
BodyPartRecord part = queue.First();
|
|
queue.Remove(part);
|
|
//Drop parts and bionics that are higher on the body tree.
|
|
if (NewMedicalRecipesUtility.TryGetParts(corpse, recipeInfo, part, skillChance, ref results, ref damagedParts) && core != part)
|
|
continue;
|
|
queue.AddRange(part.parts.Where(x => !hediffSet.PartIsMissing(x)));
|
|
}
|
|
|
|
foreach (BodyPartRecord part in damagedParts) NewMedicalRecipesUtility.DamageHarvested(corpse.InnerPawn, part);
|
|
|
|
//Collect rjw rediffs
|
|
List<Hediff> rjwNaturalDiffs = (from x in corpse.InnerPawn.health.hediffSet.hediffs
|
|
where x is Hediff_PartBaseNatural
|
|
select x).ToList();
|
|
List<Hediff> rjwArtificialDiffs = (from x in corpse.InnerPawn.health.hediffSet.hediffs
|
|
where x is Hediff_PartBaseArtifical
|
|
select x).ToList();
|
|
|
|
//Log.Message(String.Format("Collected {0} natural and {1} artifical hediffs", rjwNaturalDiffs.Count(), rjwArtificialDiffs.Count()));
|
|
|
|
//Collect parts from hediffs rjw's surgery methods
|
|
List<Thing> rjwNaturalThings = rjwNaturalDiffs.Select(d => SexPartAdder.recipePartRemover(d)).ToList();
|
|
List<Thing> rjwArtificialThings = rjwArtificialDiffs.Select(d => SexPartAdder.recipePartRemover(d)).ToList();
|
|
|
|
//Log.Message(String.Format("Collected {0} things from {1} natural and {2} things from {3} artifical hediffs", rjwArtificialThings.Count(), rjwNaturalDiffs.Count(), rjwArtificialThings.Count(), rjwArtificialDiffs.Count()));
|
|
|
|
//Simulate success chance scaled with skill etc.
|
|
rjwNaturalThings.ToList().ForEach(t => { if (Rand.Chance(Math.Min(skillChance, recipeInfo.NaturalChance))) results.Add(t); });
|
|
rjwArtificialThings.ToList().ForEach(t => { if (Rand.Chance(Math.Min(skillChance, recipeInfo.BionicChance))) results.Add(t); });
|
|
|
|
//Remove all parts that were tried to harves from the corpse
|
|
rjwNaturalDiffs.ToList().ForEach(d => corpse.InnerPawn.health.RemoveHediff(d));
|
|
rjwArtificialDiffs.ToList().ForEach(d => corpse.InnerPawn.health.RemoveHediff(d));
|
|
|
|
if (results.Count() > recipeInfo.PartNumber)
|
|
{
|
|
Random random = new Random();
|
|
__result = results.OrderBy(i => random.Next()).Take(recipeInfo.PartNumber);
|
|
}
|
|
else
|
|
{
|
|
__result = results;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|