167 lines
5.5 KiB
C#
167 lines
5.5 KiB
C#
using HarmonyLib;
|
|
using RimWorld;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using Verse;
|
|
|
|
namespace RJW_Menstruation
|
|
{
|
|
|
|
[HarmonyPatch(typeof(Pawn), nameof(Pawn.SpawnSetup))]
|
|
public class Pawn_Patch
|
|
{
|
|
public static void Postfix(Pawn __instance)
|
|
{
|
|
//Log.Message("Initialize on spawnsetup");
|
|
foreach (HediffComp_Menstruation comp in __instance.GetMenstruationComps())
|
|
{
|
|
comp.Initialize();
|
|
}
|
|
|
|
HediffComp_Breast bcomp = __instance.GetBreastComp();
|
|
if (bcomp != null)
|
|
{
|
|
bcomp.Initialize();
|
|
}
|
|
}
|
|
}
|
|
|
|
[HarmonyPatch(typeof(FloatMenuMakerMap), "AddHumanlikeOrders")]
|
|
public class HumanlikeOrder_Patch
|
|
{
|
|
public static void Postfix(Vector3 clickPos, Pawn pawn, List<FloatMenuOption> opts)
|
|
{
|
|
IEnumerable<LocalTargetInfo> selftargets = GenUI.TargetsAt(clickPos, TargetingParameters.ForSelf(pawn));
|
|
|
|
foreach (LocalTargetInfo t in selftargets)
|
|
{
|
|
if (t.Pawn == pawn && pawn.HasMenstruationComp()) opts.AddDistinct(MakeSelfMenu(pawn, t));
|
|
break;
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
public static FloatMenuOption MakeSelfMenu(Pawn pawn, LocalTargetInfo target)
|
|
{
|
|
FloatMenuOption option = FloatMenuUtility.DecoratePrioritizedTask(new FloatMenuOption(Translations.FloatMenu_CleanSelf, delegate ()
|
|
{
|
|
pawn.jobs.TryTakeOrderedJob(new Verse.AI.Job(VariousDefOf.VaginaWashing, null, null, target.Cell));
|
|
}, MenuOptionPriority.Low), pawn, target);
|
|
|
|
return option;
|
|
}
|
|
}
|
|
|
|
//[HarmonyPatch(typeof(HealthCardUtility), "DrawHediffListing")]
|
|
//public class DrawHediffListing_Patch
|
|
//{
|
|
// public const float buttonWidth = 80f;
|
|
// public const float buttonHeight = 20f;
|
|
//
|
|
// public static void Postfix(Rect rect, Pawn pawn, bool showBloodLoss)
|
|
// {
|
|
// if (Configurations.EnableButtonInHT && pawn.HasMenstruationComp())
|
|
// {
|
|
// Rect buttonrect = new Rect(rect.xMax - buttonWidth, rect.yMax - buttonHeight, buttonWidth, buttonHeight);
|
|
// if (Widgets.ButtonText(buttonrect, "Status"))
|
|
// {
|
|
// Dialog_WombStatus.ToggleWindow(pawn,pawn.GetMenstruationComp());
|
|
// }
|
|
// }
|
|
//
|
|
//
|
|
// }
|
|
//}
|
|
|
|
[HarmonyPatch(typeof(HealthCardUtility), "DrawHediffRow")]
|
|
public class DrawHediffRow_Patch
|
|
{
|
|
public const float buttonWidth = 50f;
|
|
public const float buttonHeight = 20f;
|
|
|
|
private static HediffComp_Menstruation GetFirstMenstruation(IEnumerable<Hediff> diffs)
|
|
{
|
|
foreach (Hediff diff in diffs)
|
|
{
|
|
HediffComp_Menstruation comp = diff.GetMenstruationCompFromVagina();
|
|
if (comp != null) return comp;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public static void Prefix(Rect rect, Pawn pawn, IEnumerable<Hediff> diffs, ref float curY)
|
|
{
|
|
if (Configurations.EnableButtonInHT && pawn.ShowStatus())
|
|
{
|
|
HediffComp_Menstruation comp = GetFirstMenstruation(diffs);
|
|
if (comp != null)
|
|
{
|
|
Rect buttonrect = new Rect((rect.xMax) / 2 - 5f, curY + 2f, buttonWidth, buttonHeight);
|
|
if (Widgets.ButtonText(buttonrect, Translations.Button_HealthTab))
|
|
{
|
|
Dialog_WombStatus.ToggleWindow(pawn, comp);
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
//Merged to RJW
|
|
//[HarmonyPatch(typeof(PawnColumnWorker_Pregnant), "GetIconFor")]
|
|
//public class PawnColumnWorker_Patch_Icon
|
|
//{
|
|
// public static void Postfix(Pawn pawn, ref Texture2D __result)
|
|
// {
|
|
// if (pawn.IsVisiblyPregnant()) __result = ContentFinder<Texture2D>.Get("UI/Icons/Animal/Pregnant", true);
|
|
// }
|
|
//
|
|
//}
|
|
//
|
|
//[HarmonyPatch(typeof(PawnColumnWorker_Pregnant), "GetTooltipText")]
|
|
//public class PawnColumnWorker_Patch_Tooltip
|
|
//{
|
|
// public static bool Prefix(Pawn pawn, ref string __result)
|
|
// {
|
|
// float gestationProgress = PregnancyHelper.GetPregnancy(pawn).Severity;
|
|
// int num = (int)(pawn.RaceProps.gestationPeriodDays * 60000f);
|
|
// int numTicks = (int)(gestationProgress * (float)num);
|
|
// __result = "PregnantIconDesc".Translate(numTicks.ToStringTicksToDays("F0"), num.ToStringTicksToDays("F0"));
|
|
// return false;
|
|
// }
|
|
//
|
|
//}
|
|
//
|
|
//[HarmonyPatch(typeof(TransferableUIUtility), "DoExtraAnimalIcons")]
|
|
//public class TransferableUIUtility_Patch_Icon
|
|
//{
|
|
// //private static readonly Texture2D PregnantIcon = ContentFinder<Texture2D>.Get("UI/Icons/Animal/Pregnant", true);
|
|
//
|
|
//
|
|
//
|
|
// public static void Postfix(Transferable trad, Rect rect, ref float curX, Texture2D ___PregnantIcon)
|
|
// {
|
|
// Pawn pawn = trad.AnyThing as Pawn;
|
|
// if (pawn?.health?.hediffSet != null && pawn.IsVisiblyPregnant())
|
|
// {
|
|
// Rect rect3 = new Rect(curX - 24f, (rect.height - 24f) / 2f, 24f, 24f);
|
|
// curX -= 24f;
|
|
// if (Mouse.IsOver(rect3))
|
|
// {
|
|
// TooltipHandler.TipRegion(rect3, PawnColumnWorker_Pregnant.GetTooltipText(pawn));
|
|
// }
|
|
// GUI.DrawTexture(rect3, ___PregnantIcon);
|
|
// }
|
|
// }
|
|
//}
|
|
|
|
|
|
|
|
|
|
}
|