Refactor Thought_IncreaseRecord, also fix null error in it if no sexperience

This commit is contained in:
amevarashi 2022-08-01 17:57:29 +05:00
parent 75023a9d05
commit 6767feb57c
2 changed files with 31 additions and 57 deletions

View File

@ -1,4 +1,5 @@
using RimWorld;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using Verse;
@ -10,5 +11,23 @@ namespace RJWSexperience
public RecordDef recordDef;
[SuppressMessage("Minor Code Smell", "S1104:Fields should not have public accessibility", Justification = "Field value loaded from XML")]
public float increment;
public override IEnumerable<string> ConfigErrors()
{
foreach (string error in base.ConfigErrors())
{
yield return error;
}
if (recordDef == null)
{
yield return "<recordDef> is empty";
}
if (increment == 0f)
{
yield return "<increment> is empty or 0";
}
}
}
}

View File

@ -1,70 +1,25 @@
using RimWorld;
using Verse;
namespace RJWSexperience
{
/// <summary>
/// Increments record every time the thought is added
/// The increment is permanent
/// </summary>
public class Thought_IncreaseRecord : Thought_Memory
{
private ThoughtDefExtension_IncreaseRecord extension;
protected ThoughtDefExtension_IncreaseRecord Extension
{
get
{
if (extension == null)
extension = def.GetModExtension<ThoughtDefExtension_IncreaseRecord>();
return extension;
}
}
protected RecordDef RecordDef => Extension.recordDef;
protected float Increment => Extension.increment;
protected float recordIncrement;
public override void ExposeData()
{
base.ExposeData();
Scribe_Values.Look(ref recordIncrement, "recordIncrement", recordIncrement, true);
}
public override void ThoughtInterval()
{
base.ThoughtInterval();
if (recordIncrement != 0)
{
pawn.records.AddTo(RecordDef, recordIncrement);
recordIncrement = 0;
}
}
/// <summary>
/// This method is called for every thought right after the pawn is assigned
/// </summary>
public override bool TryMergeWithExistingMemory(out bool showBubble)
{
ThoughtHandler thoughts = pawn.needs.mood.thoughts;
if (thoughts.memories.NumMemoriesInGroup(this) >= def.stackLimit)
var defExtension = def.GetModExtension<ThoughtDefExtension_IncreaseRecord>();
if (defExtension != null)
{
Thought_IncreaseRecord thought_Memory = (Thought_IncreaseRecord)thoughts.memories.OldestMemoryInGroup(this);
if (thought_Memory != null)
{
showBubble = (thought_Memory.age > thought_Memory.def.DurationTicks / 2);
thought_Memory.Merged();
return true;
}
pawn.records.AddTo(defExtension.recordDef, defExtension.increment);
}
showBubble = true;
return false;
}
public override void Init()
{
base.Init();
recordIncrement = Increment;
}
protected virtual void Merged()
{
age = 0;
recordIncrement += Increment;
return base.TryMergeWithExistingMemory(out showBubble);
}
}
}