RJW-Sexperience/Source/RJWSexperience/SexHistory/UI/SexStatusViewModel.cs
2023-07-09 12:02:20 +05:00

357 lines
13 KiB
C#

using RimWorld;
using rjw;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using UnityEngine;
using Verse;
namespace RJWSexperience.SexHistory.UI
{
public sealed class SexStatusViewModel
{
private static readonly int[] Sextype =
{
(int)xxx.rjwSextype.Vaginal,
(int)xxx.rjwSextype.Anal,
(int)xxx.rjwSextype.Oral,
(int)xxx.rjwSextype.Fellatio,
(int)xxx.rjwSextype.Cunnilingus,
(int)xxx.rjwSextype.DoublePenetration,
(int)xxx.rjwSextype.Boobjob,
(int)xxx.rjwSextype.Handjob,
(int)xxx.rjwSextype.Footjob,
(int)xxx.rjwSextype.Fingering,
(int)xxx.rjwSextype.Scissoring,
(int)xxx.rjwSextype.MutualMasturbation,
(int)xxx.rjwSextype.Fisting,
(int)xxx.rjwSextype.Rimming,
(int)xxx.rjwSextype.Sixtynine
};
private readonly SexHistoryComp _history;
private readonly CompRJW _rjwComp;
private int _validUntilTick;
public SexStatusViewModel(SexHistoryComp history, PartnerOrderMode mode)
{
_history = history;
_rjwComp = history.ParentPawn.TryGetComp<CompRJW>();
SetPartnerOrder(mode);
Name = Pawn.Name?.ToStringFull ?? Pawn.Label;
if (Pawn.story != null)
{
AgeAndTitle = Pawn.ageTracker.AgeBiologicalYears + ", " + Pawn.story.Title;
}
else
{
AgeAndTitle = Pawn.ageTracker.AgeBiologicalYears + ", " + Pawn.def.label;
}
}
public Pawn Pawn => _history.ParentPawn;
public string Name { get; }
public string AgeAndTitle { get; }
public List<InfoCard> InfoCards { get; } = new List<InfoCard>();
public InfoCard SelectedPartnerCard { get; private set; }
public PreferedRaceCard PreferedRaceCard { get; private set; }
public List<BarInfo> SexTypes { get; } = new List<BarInfo>();
public BarInfo TotalSex { get; } = new BarInfo(HistoryUtility.TotalSex);
public BarInfo Lust { get; } = new BarInfo(HistoryUtility.Slaanesh);
public BarInfo BestSextype { get; } = new BarInfo();
public BarInfo RecentSextype { get; } = new BarInfo();
public BarInfo Necro { get; } = new BarInfo(HistoryUtility.Nurgle);
public BarInfo Incest { get; } = new BarInfo(HistoryUtility.Nurgle);
public BarInfo ConsumedCum { get; } = new BarInfo(Texture2D.linearGrayTexture);
public BarInfo CumHediff { get; } = new BarInfo(Texture2D.linearGrayTexture);
public BarInfo BeenRaped { get; } = new BarInfo(Texture2D.grayTexture);
public BarInfo Raped { get; } = new BarInfo(HistoryUtility.Khorne);
public BarInfo SexSatisfaction { get; } = new BarInfo(HistoryUtility.Satisfaction);
public BarInfo SexSkill { get; } = new BarInfo(HistoryUtility.Tzeentch);
public string VirginLabel { get; private set; }
public string SexualityLabel { get; private set; }
public string QuirksLabel { get; private set; }
public TipSignal QuirksTooltip { get; private set; }
public SexPartnerHistoryRecord SelectedPartner { get; private set; }
public IEnumerable<PartnerPortraitInfo> Partners { get; private set; }
public void Update()
{
if (Find.TickManager.TicksGame <= _validUntilTick)
{
return;
}
UpdateInfoCards();
UpdateBars();
UpdateQuirks();
UpdateVirginAndSexuality();
PreferedRaceCard = new PreferedRaceCard(_history);
int tickRateMultiplier = (int)Find.TickManager.TickRateMultiplier;
if (tickRateMultiplier == 0) // Paused
{
_validUntilTick = Find.TickManager.TicksGame;
return;
}
_validUntilTick = Find.TickManager.TicksGame + (60 * tickRateMultiplier);
}
private void UpdateInfoCards()
{
InfoCards.Clear();
InfoCards.Add(new InfoCard(
pawn: Pawn,
partnerRecord: _history.RecentPartnerRecord,
label: Keyed.RS_Recent_Sex_Partner,
tooltipLabel: Keyed.RS_Recent_Sex_Partner_ToolTip,
lastSexTimeTicks: _history.RecentSexTickAbs));
InfoCards.Add(new InfoCard(
pawn: Pawn,
partnerRecord: _history.FirstPartnerRecord,
label: Keyed.RS_First_Sex_Partner,
tooltipLabel: Keyed.RS_First_Sex_Partner_ToolTip,
lastSexTimeTicks: _history.FirstSexTickAbs));
InfoCards.Add(new InfoCard(
pawn: Pawn,
partnerRecord: _history.MostPartnerRecord,
label: Keyed.RS_Most_Sex_Partner,
tooltipLabel: Keyed.RS_Most_Sex_Partner_ToolTip,
lastSexTimeTicks: _history.MostSexTickAbs));
InfoCards.Add(new InfoCard(
pawn: Pawn,
partnerRecord: _history.BestSexPartnerRecord,
label: Keyed.RS_Best_Sex_Partner,
tooltipLabel: Keyed.RS_Best_Sex_Partner_ToolTip,
lastSexTimeTicks: _history.BestSexTickAbs));
if (SelectedPartner != null)
{
UpdateSelectedPartnerCard();
}
}
private void UpdateBars()
{
float maxSatisfaction = _history.GetBestSextype(out _);
if (maxSatisfaction == 0f)
{
maxSatisfaction = UIUtility.BASESAT;
}
SexTypes.Clear();
for (int i = 0; i < Sextype.Length; i++)
{
int sexIndex = Sextype[i];
float AverageSatisfaction = _history.GetAVGSat(sexIndex);
float relativeSat = AverageSatisfaction / maxSatisfaction;
float satisfactionRelativeToBase = AverageSatisfaction / UIUtility.BASESAT;
SexTypes.Add(new BarInfo(
label: Keyed.RS_SexInfo(Keyed.Sextype[sexIndex], _history.GetSexCount(sexIndex)),
fillPercent: relativeSat,
fillTexture: HistoryUtility.SextypeColor[sexIndex],
tooltip: Keyed.RS_LastSex + ": " + UIUtility.GetSexDays(_history.GetSextypeRecentTickAbs(sexIndex), true),
labelRight: Keyed.RS_SatAVG(satisfactionRelativeToBase)));
}
SexTypes.Add(new BarInfo(
label: string.Format(Keyed.RS_Sex_Partners + ": {0} ({1})", _history.PartnerCount, Pawn.records.GetValue(RsDefOf.Record.SexPartnerCount)),
fillPercent: _history.PartnerCount / 50,
fillTexture: HistoryUtility.Partners));
SexTypes.Add(new BarInfo(
label: string.Format(Keyed.RS_VirginsTaken + ": {0:0}", _history.VirginsTaken),
fillPercent: _history.VirginsTaken / 100,
fillTexture: HistoryUtility.Partners));
TotalSex.Label = string.Format(Keyed.RS_TotalSexHad + ": {0:0} ({1:0})", _history.TotalSexHad, Pawn.records.GetValue(xxx.CountOfSex));
TotalSex.FillPercent = _history.TotalSexHad / 100;
TotalSex.LabelRight = Keyed.RS_SatAVG(_history.AVGSat);
float lust = Pawn.records.GetValue(RsDefOf.Record.Lust);
float sexDrive = GetStatValue(xxx.sex_drive_stat);
float lustLimit = SexperienceMod.Settings.LustLimit * 3f;
Lust.Label = string.Format(Keyed.Lust + ": {0:0.00}", lust);
Lust.FillPercent = lust.Normalization(-lustLimit, lustLimit);
Lust.Tooltip = GetStatTooltip(xxx.sex_drive_stat, sexDrive);
Lust.LabelRight = xxx.sex_drive_stat.LabelCap + ": " + sexDrive.ToStringPercent();
float bestSextypeRelativeSatisfaction = _history.GetBestSextype(out xxx.rjwSextype bestSextype) / UIUtility.BASESAT;
BestSextype.Label = string.Format(Keyed.RS_Best_Sextype + ": {0}", Keyed.Sextype[(int)bestSextype]);
BestSextype.FillPercent = bestSextypeRelativeSatisfaction / 2;
BestSextype.FillTexture = HistoryUtility.SextypeColor[(int)bestSextype];
BestSextype.LabelRight = Keyed.RS_SatAVG(bestSextypeRelativeSatisfaction);
float recentSextypeRelativeSatisfaction = _history.GetRecentSextype(out xxx.rjwSextype recentSextype) / UIUtility.BASESAT;
RecentSextype.Label = string.Format(Keyed.RS_Recent_Sextype + ": {0}", Keyed.Sextype[(int)recentSextype]);
RecentSextype.FillPercent = recentSextypeRelativeSatisfaction / 2;
RecentSextype.FillTexture = HistoryUtility.SextypeColor[(int)recentSextype];
RecentSextype.LabelRight = recentSextypeRelativeSatisfaction.ToStringPercent();
Necro.Label = string.Format(Keyed.RS_Necrophile + ": {0}", _history.CorpseFuckCount);
Necro.FillPercent = _history.CorpseFuckCount / 50f;
Incest.Label = string.Format(Keyed.Incest + ": {0}", _history.IncestuousCount);
Incest.FillPercent = _history.IncestuousCount / 50f;
float amountofEatenCum = Pawn.records.GetValue(RsDefOf.Record.AmountofEatenCum);
ConsumedCum.Label = string.Format(Keyed.RS_Cum_Swallowed + ": {0} mL, {1} " + Keyed.RS_NumofTimes, amountofEatenCum, Pawn.records.GetValue(RsDefOf.Record.NumofEatenCum));
ConsumedCum.FillPercent = amountofEatenCum / 1000;
Hediff cumHediff = Pawn.health.hediffSet.GetFirstHediffOfDef(RsDefOf.Hediff.CumAddiction)
?? Pawn.health.hediffSet.GetFirstHediffOfDef(RsDefOf.Hediff.CumTolerance);
if (cumHediff != null)
{
CumHediff.Label = $"{cumHediff.Label}: {cumHediff.Severity.ToStringPercent()}";
CumHediff.FillPercent = cumHediff.Severity;
CumHediff.Tooltip = new TipSignal(() => cumHediff.GetTooltip(Pawn, false), cumHediff.Label.GetHashCode());
}
else
{
CumHediff.Label = "";
CumHediff.FillPercent = 0f;
CumHediff.Tooltip = default;
}
float vulnerability = GetStatValue(xxx.vulnerability_stat);
string vulnerabilityLabel = xxx.vulnerability_stat.LabelCap + ": " + vulnerability.ToStringPercent();
TipSignal vulnerabilityTip = GetStatTooltip(xxx.vulnerability_stat, vulnerability);
Raped.Label = string.Format(Keyed.RS_RapedSomeone + ": {0}", _history.RapedCount);
Raped.FillPercent = _history.RapedCount / 50f;
Raped.Tooltip = vulnerabilityTip;
Raped.LabelRight = vulnerabilityLabel;
BeenRaped.Label = string.Format(Keyed.RS_BeenRaped + ": {0}", _history.BeenRapedCount);
BeenRaped.FillPercent = _history.BeenRapedCount / 50f;
BeenRaped.Tooltip = vulnerabilityTip;
BeenRaped.LabelRight = vulnerabilityLabel;
float sexSatisfaction = GetStatValue(xxx.sex_satisfaction);
SexSatisfaction.Label = xxx.sex_satisfaction.LabelCap + ": " + sexSatisfaction.ToStringPercent();
SexSatisfaction.FillPercent = sexSatisfaction / 2;
SexSatisfaction.Tooltip = GetStatTooltip(xxx.sex_satisfaction, sexSatisfaction);
SkillRecord skill = Pawn.skills?.GetSkill(RsDefOf.Skill.Sex);
float sexSkillLevel = skill?.Level ?? 0f;
float sexStat = Pawn.GetSexStat();
SexSkill.Label = $"{Keyed.RS_SexSkill}: {sexSkillLevel}, {skill?.xpSinceLastLevel / skill?.XpRequiredForLevelUp:P2}";
SexSkill.FillPercent = sexSkillLevel / 20;
SexSkill.Tooltip = GetStatTooltip(RsDefOf.Stat.SexAbility, sexStat);
SexSkill.LabelRight = RsDefOf.Stat.SexAbility.LabelCap + ": " + sexStat.ToStringPercent();
SexSkill.Border = HistoryUtility.GetPassionBG(skill?.passion);
}
private void UpdateQuirks()
{
List<Quirk> quirks = Quirk.All.FindAll(x => Pawn.Has(x));
string quirkstr = quirks.Select(x => x.Key).ToCommaList();
QuirksLabel = "Quirks".Translate() + quirkstr;
if (quirks.NullOrEmpty())
{
QuirksTooltip = "NoQuirks".Translate();
}
else
{
QuirksTooltip = new TipSignal(() =>
{
StringBuilder stringBuilder = new StringBuilder();
foreach (Quirk q in quirks)
{
stringBuilder.AppendLine(q.Key.Colorize(Color.yellow));
stringBuilder.AppendLine(q.LocaliztionKey.Translate(Pawn.Named("pawn")).AdjustedFor(Pawn).Resolve());
stringBuilder.AppendLine("");
}
return stringBuilder.ToString().TrimEndNewlines();
}, "Quirks".GetHashCode());
}
}
private void UpdateVirginAndSexuality()
{
Trait virginity = Pawn.story?.traits?.GetTrait(RsDefOf.Trait.Virgin);
if (virginity != null && virginity.Degree != Virginity.TraitDegree.FemaleAfterSurgery)
{
VirginLabel = virginity.Label;
}
else
{
VirginLabel = null;
}
if (_rjwComp != null)
{
SexualityLabel = Keyed.RS_Sexuality + ": " + Keyed.Sexuality[(int)_rjwComp.orientation];
}
}
public void SetPartnerOrder(PartnerOrderMode mode)
{
if (_history == null)
{
return;
}
var partners = _history.PartnerList.Select(x => new PartnerPortraitInfo(Pawn, x));
switch (mode)
{
default:
Partners = partners;
break;
case PartnerOrderMode.Recent:
Partners = partners.OrderBy(x => x.partnerRecord.RecentSexTickAbs);
break;
case PartnerOrderMode.Most:
Partners = partners.OrderBy(x => x.partnerRecord.TotalSexCount);
break;
case PartnerOrderMode.Name:
Partners = partners.OrderBy(x => x.partnerRecord.Label);
break;
}
}
public void SetSelectedPartner(SexPartnerHistoryRecord sexPartner)
{
SelectedPartner = sexPartner;
UpdateSelectedPartnerCard();
}
private void UpdateSelectedPartnerCard()
{
if (SelectedPartner == null)
{
SelectedPartnerCard = default;
return;
}
SelectedPartnerCard = new InfoCard(
pawn: Pawn,
partnerRecord: SelectedPartner,
label: Keyed.RS_Selected_Partner,
tooltipLabel: Keyed.RS_Selected_Partner,
lastSexTimeTicks: SelectedPartner.RecentSexTickAbs);
}
private float GetStatValue(StatDef statDef) => Pawn.Dead ? 0f : Pawn.GetStatValue(statDef);
private TipSignal GetStatTooltip(StatDef stat, float val)
{
if (!Pawn.Dead)
{
return new TipSignal(
textGetter: () => stat.description + "\n\n" + stat.Worker.GetExplanationFull(StatRequest.For(Pawn), ToStringNumberSense.Undefined, val),
uniqueId: stat.GetHashCode());
}
return "Dead".Translate();
}
}
}