rjw-genes/Source/Genes/Life_Force/Interactions/SuccubusTailjob/GenesPartKindUsageRule.cs

using System;
using System.Collections.Generic;
using rjw;
using rjw.Modules.Interactions.Contexts;
using rjw.Modules.Interactions.Enums;
using rjw.Modules.Interactions.Rules.PartKindUsageRules;
using rjw.Modules.Shared;
using Verse;

namespace RJW_Genes.Interactions
{
    //Summary//
    //Set custom preferences for pawn. Gets integrated into rjw by AddtoIPartPreferenceRule in First
    //Depending on the level of lifeforce increase the chance for using the mouth.
    public class GenesPartKindUsageRule : IPartPreferenceRule
    {
        public IEnumerable<Weighted<LewdablePartKind>> ModifiersForDominant(InteractionContext context)
        {
            Pawn pawn = context.Internals.Dominant.Pawn;
            Gene_LifeForce gene = pawn.genes.GetFirstGeneOfType<Gene_LifeForce>();
            if (gene != null)
            {
                float weight = 2f;
                if (gene.Value < gene.MinLevelForAlert)
                {
                    weight *= 10;
                }
                else if (gene.Value < gene.targetValue)
                {
                    weight *= 2.5f;
                }
                if (pawn.genes.HasGene(GeneDefOf.rjw_genes_cum_eater))
                {
                    yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Mouth);
                    yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Beak);
                }

                if (pawn.genes.HasGene(GeneDefOf.rjw_genes_fertilin_absorber))
                {
                    yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Vagina);
                    yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Anus);
                }
            }
            yield break;
        }

        public IEnumerable<Weighted<LewdablePartKind>> ModifiersForSubmissive(InteractionContext context)
        {
            Pawn pawn = context.Internals.Dominant.Pawn;
            Gene_LifeForce gene = pawn.genes.GetFirstGeneOfType<Gene_LifeForce>();
            if (gene != null)
            {
                float weight = 2f;
                if (gene.Value < gene.MinLevelForAlert)
                {
                    weight *= 10;
                }
                else if (gene.Value < gene.targetValue)
                {
                    weight *= 2.5f;
                }
                yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Mouth);
                yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Beak);

                if (pawn.genes.HasGene(GeneDefOf.rjw_genes_fertilin_absorber))
                {
                    yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Vagina);
                    yield return new Weighted<LewdablePartKind>(weight, LewdablePartKind.Anus);
                }
            }
            yield break;
        }
    }
}