rjw-plasticsurgeries/Source/RJW_PlasticSurgeries/Recipe_Surgery_Penoplasty.cs

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

namespace RJW_PlasticSurgeries
{
    public abstract class Recipe_Surgery_Penoplasty : Recipe_Plastic_Surgery
    {
        protected override BodyPartRecord GetPartCandidate(Pawn pawn)
        {
            return Genital_Helper.get_genitalsBPR(pawn);
        }

        protected override bool HasPart(Pawn pawn, List<Hediff> hediffs)
        {
            return Genital_Helper.has_penis_fertile(pawn, hediffs) ||
                   Genital_Helper.has_penis_infertile(pawn, hediffs);
        }

        protected override List<Hediff> GetHediffs(Pawn pawn)
        {
            return pawn.GetGenitalsList().FindAll(Genital_Helper.is_penis);
        }

        protected override (HediffDef, float) GetLicentiaDamage()
        {
            if (HasLicentia)
            {
                return (Licentia.HediffDefs.StretchTear, 0.2f);
            }

            throw new NotSupportedException();
        }
    }

    public class Recipe_Surgery_Penoplasty_Micro : Recipe_Surgery_Penoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.1f, true);
        }
    }

    public class Recipe_Surgery_Penoplasty_Small : Recipe_Surgery_Penoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.3f, true);
        }
    }

    public class Recipe_Surgery_Penoplasty_Average : Recipe_Surgery_Penoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.5f, true);
        }
    }

    public class Recipe_Surgery_Penoplasty_Large : Recipe_Surgery_Penoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.7f, true);
        }
    }

    public class Recipe_Surgery_Penoplasty_Huge : Recipe_Surgery_Penoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.9f, true);
        }
    }
}