86 lines
2.4 KiB
C#
86 lines
2.4 KiB
C#
using RimWorld;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
using Verse;
|
|
|
|
namespace RJWSexperience.Ideology.Filters
|
|
{
|
|
/// <summary>
|
|
/// Filter to describe how one pawn sees another
|
|
/// </summary>
|
|
[SuppressMessage("Minor Code Smell", "S1104:Fields should not have public accessibility", Justification = "Def loader")]
|
|
public class RelationFilter
|
|
{
|
|
public bool? isVeneratedAnimal;
|
|
public List<PawnRelationDef> hasOneOfRelations;
|
|
public List<PawnRelationDef> hasNoneOfRelations;
|
|
public List<BloodRelationDegree> hasOneOfRelationDegrees;
|
|
|
|
private bool initialized = false;
|
|
private HashSet<PawnRelationDef> hasOneOfRelationsHashed;
|
|
private HashSet<PawnRelationDef> hasNoneOfRelationsHashed;
|
|
private HashSet<BloodRelationDegree> hasOneOfRelationDegreesHashed;
|
|
|
|
/// <summary>
|
|
/// Check if the pair of pawns fits filter conditions
|
|
/// </summary>
|
|
public bool Applies(Pawn pawn, Pawn partner)
|
|
{
|
|
// Fail if any single condition fails
|
|
if (isVeneratedAnimal != null && isVeneratedAnimal != pawn.Ideo.IsVeneratedAnimal(partner))
|
|
return false;
|
|
|
|
if (!CheckRelations(pawn, partner))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
private bool CheckRelations(Pawn pawn, Pawn partner)
|
|
{
|
|
if (!initialized)
|
|
Initialize();
|
|
|
|
if (hasNoneOfRelationsHashed == null && hasOneOfRelationsHashed == null && hasOneOfRelationDegreesHashed == null)
|
|
return true;
|
|
|
|
if (hasOneOfRelationDegreesHashed != null && !hasOneOfRelationDegreesHashed.Contains(RelationHelpers.GetBloodRelationDegree(pawn, partner)))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
IEnumerable<PawnRelationDef> relations = pawn.GetRelations(partner);
|
|
|
|
if (hasOneOfRelationsHashed != null)
|
|
{
|
|
if (relations.EnumerableNullOrEmpty())
|
|
return false;
|
|
|
|
if (!hasOneOfRelationsHashed.Overlaps(relations))
|
|
return false;
|
|
}
|
|
|
|
if (hasNoneOfRelationsHashed != null && !relations.EnumerableNullOrEmpty() && hasNoneOfRelationsHashed.Overlaps(relations))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
private void Initialize()
|
|
{
|
|
if (!hasNoneOfRelations.NullOrEmpty())
|
|
hasNoneOfRelationsHashed = new HashSet<PawnRelationDef>(hasNoneOfRelations);
|
|
|
|
if (!hasOneOfRelations.NullOrEmpty())
|
|
hasOneOfRelationsHashed = new HashSet<PawnRelationDef>(hasOneOfRelations);
|
|
|
|
if (!hasOneOfRelationDegrees.NullOrEmpty())
|
|
hasOneOfRelationDegreesHashed = new HashSet<BloodRelationDegree>(hasOneOfRelationDegrees);
|
|
|
|
initialized = true;
|
|
}
|
|
}
|
|
}
|