99 lines
3.6 KiB
C#
99 lines
3.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using HarmonyLib;
|
|
using Verse;
|
|
using rjw;
|
|
|
|
namespace RJW_Genes
|
|
{
|
|
/// <summary>
|
|
/// This Class patches the AfterSexUtility to also fertilize eggs if Pawn A has "InsectBreeder" and Pawn B has Insect Eggs.
|
|
/// Patched Class is https://gitgud.io/Ed86/rjw/-/blob/master/1.4/Source/Common/Helpers/SexUtility.cs
|
|
///
|
|
/// Normal Egg-Pregnancy logic is in https://gitgud.io/Ed86/rjw/-/blob/master/1.4/Source/Modules/Pregnancy/Pregnancy_Helper.cs
|
|
/// </summary>
|
|
[HarmonyPatch(typeof(SexUtility), "Aftersex")]
|
|
static class Patch_EggFertilization
|
|
{
|
|
public static void Postfix(SexProps props)
|
|
{
|
|
// Only Fertilize on vaginal / anal sex
|
|
if (!(props.sexType == xxx.rjwSextype.Vaginal || props.sexType == xxx.rjwSextype.Anal))
|
|
{
|
|
return;
|
|
}
|
|
|
|
|
|
if (canDoEggFertilization(props.pawn, props.partner))
|
|
{
|
|
// Pawn has gene and Partner has eggs
|
|
if (props.pawn.genes.GenesListForReading.Any(x => x.def == GeneDefOf.rjw_genes_insectbreeder) && !getEggsforPawn(props.partner).NullOrEmpty())
|
|
{
|
|
Pawn eggHolder = props.partner;
|
|
Pawn impregnator = props.pawn;
|
|
|
|
foreach (Hediff_InsectEgg egg in getEggsforPawn(eggHolder))
|
|
{
|
|
if (!egg.fertilized)
|
|
egg.Fertilize(impregnator);
|
|
}
|
|
}
|
|
|
|
// Partner has gene and Pawn has eggs
|
|
if (props.partner.genes.GenesListForReading.Any(x => x.def == GeneDefOf.rjw_genes_insectbreeder) && !getEggsforPawn(props.pawn).NullOrEmpty())
|
|
{
|
|
Pawn eggHolder = props.pawn;
|
|
Pawn impregnator = props.partner;
|
|
|
|
foreach (Hediff_InsectEgg egg in getEggsforPawn(eggHolder))
|
|
{
|
|
if (!egg.fertilized)
|
|
egg.Fertilize(impregnator);
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
private static bool canDoEggFertilization(Pawn a, Pawn b)
|
|
{
|
|
|
|
// No Partner / Other Errors
|
|
if (a != null || b != null)
|
|
return false;
|
|
// None of the pawns has the relevant gene
|
|
if (!a.genes.GenesListForReading.Any(x => x.def == GeneDefOf.rjw_genes_insectbreeder) && !b.genes.GenesListForReading.Any(x => x.def == GeneDefOf.rjw_genes_insectbreeder))
|
|
return false;
|
|
// None of the pawns has eggs
|
|
if (getEggsforPawn(a).NullOrEmpty() && getEggsforPawn(b).NullOrEmpty())
|
|
return false;
|
|
|
|
// A has gene and B has eggs
|
|
if (a.genes.GenesListForReading.Any(x => x.def == GeneDefOf.rjw_genes_insectbreeder) && !getEggsforPawn(b).NullOrEmpty())
|
|
{
|
|
return true;
|
|
}
|
|
// B has gene and A has eggs
|
|
if (b.genes.GenesListForReading.Any(x => x.def == GeneDefOf.rjw_genes_insectbreeder) && !getEggsforPawn(a).NullOrEmpty())
|
|
{
|
|
return true;
|
|
}
|
|
// Any other case: Do nothing
|
|
return false;
|
|
}
|
|
|
|
private static List<Hediff_InsectEgg> getEggsforPawn(Pawn pawn)
|
|
{
|
|
List<Hediff_InsectEgg> eggs = new List<Hediff_InsectEgg>();
|
|
pawn.health.hediffSet.GetHediffs(ref eggs);
|
|
foreach (var egg in eggs)
|
|
egg.Fertilize(pawn);
|
|
|
|
return eggs;
|
|
}
|
|
}
|
|
}
|