public static ThoughtState CurrentStateInternal(RimWorld.ThoughtWorker_SleepingRoomImpressiveness _this, Pawn p)
{
if (p == null)
{
return(ThoughtState.Inactive); // Added
}
// BASE
if (p.story.traits.HasTrait(TraitDefOf.Ascetic))
{
return(ThoughtState.Inactive);
}
#region Added
if (p.IsGuest())
{
return(CheckRoom(_this, p.GetGuestRoom()));
}
#endregion
// Sorta base
if (p.ownership == null || p.ownership.OwnedBed == null)
{
return(ThoughtState.Inactive);
}
// Refactored into function for double use
return(CheckRoom(_this, p.ownership.OwnedBed.GetRoom()));
}
private static ThoughtState CheckRoom(RimWorld.ThoughtWorker_SleepingRoomImpressiveness _this, Room room)
{
// BASE
if (room == null)
{
return(ThoughtState.Inactive);
}
int scoreStageIndex = RoomStatDefOf.Impressiveness.GetScoreStageIndex(room.GetStat(RoomStatDefOf.Impressiveness));
if (_this.def.stages[scoreStageIndex] != null)
{
return(ThoughtState.ActiveAtStage(scoreStageIndex));
}
return(ThoughtState.Inactive);
}