RJW-Sexperience/RJWSexperience/RJWSexperience/SexHistory/SexPartnerHistoryRecord.cs

using RimWorld;
using rjw;
using RJWSexperience.ExtensionMethods;
using System.Collections.Generic;
using System.Linq;
using Verse;

namespace RJWSexperience
{
	public class SexPartnerHistoryRecord : IExposable
	{
		public string PartnerID { get; set; }

		protected Pawn partner = null;
		protected string namecache;
		protected int totalsexhad = 0;
		protected int raped = 0;
		protected int rapedme = 0;
		protected int orgasms = 0;
		protected xxx.rjwSextype bestsextype = xxx.rjwSextype.None;
		protected float bestsatisfaction = 0;
		protected bool itookvirgin = false;
		protected bool incest = false;
		protected int recentsextickabs = 0;
		protected int bestsextickabs = 0;
		protected bool cannotLoadPawnData = false;
		protected ThingDef race;

		public string Label
		{
			get
			{
				if (partner != null)
				{
					namecache = partner.Label;
					return namecache;
				}
				else return namecache;
			}
		}
		public xxx.rjwSextype BestSextype
		{
			get
			{
				return bestsextype;
			}
		}
		public float BestSatisfaction
		{
			get
			{
				return bestsatisfaction;
			}
		}
		public int TotalSexCount
		{
			get
			{
				return totalsexhad;
			}
		}
		public Pawn Partner
		{
			get
			{
				if (!cannotLoadPawnData && partner == null)
				{
					LoadPartnerPawn(PartnerID);
					if (partner == null) cannotLoadPawnData = true;
				}
				return partner;
			}
		}
		public string RapeInfo
		{
			get
			{
				string res = "";
				if (raped > 0) res += Keyed.RS_Raped + raped + " ";
				if (rapedme > 0) res += Keyed.RS_RapedMe + rapedme + " ";
				return res;
			}
		}
		public int OrgasmCount => orgasms;
		public bool IamFirst => itookvirgin;
		public bool Incest => incest;
		public int Raped => raped;
		public int RapedMe => rapedme;
		public int RecentSexTickAbs => recentsextickabs;
		public int BestSexTickAbs => bestsextickabs;
		public int BestSexElapsedTicks => GenTicks.TicksAbs - bestsextickabs;
		public string BestSexDays
		{
			get
			{
				if (bestsextickabs != 0) return Keyed.RS_HadBestSexDaysAgo(GenDate.ToStringTicksToDays(BestSexElapsedTicks) + " " + Keyed.RS_Ago);
				return "";
			}
		}
		public ThingDef Race
		{
			get
			{
				if (Partner != null)
				{
					race = Partner.def;
					return race;
				}
				else return race;
			}
		}

		public SexPartnerHistoryRecord() { }

		public SexPartnerHistoryRecord(Pawn pawn, bool incest = false)
		{
			this.partner = pawn;
			this.namecache = pawn.Label;
			this.incest = incest;
			this.race = pawn.def;
		}

		public void ExposeData()
		{
			Scribe_Values.Look(ref namecache, "namecache");
			Scribe_Values.Look(ref totalsexhad, "totalsexhad", 0);
			Scribe_Values.Look(ref raped, "raped", 0);
			Scribe_Values.Look(ref rapedme, "rapedme", 0);
			Scribe_Values.Look(ref orgasms, "orgasms", 0);
			Scribe_Values.Look(ref bestsextype, "bestsextype", xxx.rjwSextype.None);
			Scribe_Values.Look(ref bestsatisfaction, "bestsatisfaction", 0f);
			Scribe_Values.Look(ref itookvirgin, "itookvirgin", false);
			Scribe_Values.Look(ref incest, "incest", false);
			Scribe_Values.Look(ref recentsextickabs, "recentsextickabs", 0);
			Scribe_Values.Look(ref bestsextickabs, "bestsextickabs", 0);
			Scribe_Defs.Look(ref race, "race");
		}

		public void RecordSex(SexProps props)
		{
			totalsexhad++;
			if (props.isRape)
			{
				if (partner == props.GetInteractionInitiator())
				{
					rapedme++;
				}
				else if (partner == props.GetInteractionRecipient())
				{
					raped++;
				}
			}
			recentsextickabs = GenTicks.TicksAbs;
		}

		public void RecordSatisfaction(SexProps props, float satisfaction)
		{
			if (satisfaction > bestsatisfaction)
			{
				orgasms++;
				bestsextype = props.sexType;
				bestsatisfaction = satisfaction;
				bestsextickabs = GenTicks.TicksAbs;
			}
		}

		public void TookVirgin()
		{
			itookvirgin = true;
		}

		protected void LoadPartnerPawn(string partnerID)
		{
			foreach (Map map in Find.Maps)
			{
				partner = map.mapPawns.AllPawns.FirstOrDefault(x => x.ThingID.Equals(partnerID));
				if (partner != null) return;
			}
			partner = Find.WorldPawns.AllPawnsAliveOrDead.FirstOrDefault(x => x.ThingID.Equals(partnerID));
		}

		public class RecentOrderComparer : IComparer<SexPartnerHistoryRecord>
		{
			public int Compare(SexPartnerHistoryRecord x, SexPartnerHistoryRecord y)
			{
				return y.RecentSexTickAbs.CompareTo(x.RecentSexTickAbs);
			}
		}

		public class MostOrderComparer : IComparer<SexPartnerHistoryRecord>
		{
			public int Compare(SexPartnerHistoryRecord x, SexPartnerHistoryRecord y)
			{
				return y.TotalSexCount.CompareTo(x.TotalSexCount);
			}
		}

		public class NameOrderComparer : IComparer<SexPartnerHistoryRecord>
		{
			public int Compare(SexPartnerHistoryRecord x, SexPartnerHistoryRecord y)
			{
				return x.Label.CompareTo(y.Label);
			}
		}
	}

}