From 6eaaa539e91e249594854cd47a936b5216e3874c Mon Sep 17 00:00:00 2001
From: c0ffee <platinumspoons@gmail.com>
Date: Wed, 8 May 2024 09:25:17 -0700
Subject: [PATCH] Added trait-based pawn test

---
 .../Animations/PawnTests/PawnTest_Trait.cs    | 27 +++++++++++++++++++
 Rimworld-Animations.csproj                    |  1 +
 2 files changed, 28 insertions(+)
 create mode 100644 1.5/Source/Animations/PawnTests/PawnTest_Trait.cs

diff --git a/1.5/Source/Animations/PawnTests/PawnTest_Trait.cs b/1.5/Source/Animations/PawnTests/PawnTest_Trait.cs
new file mode 100644
index 0000000..dd67038
--- /dev/null
+++ b/1.5/Source/Animations/PawnTests/PawnTest_Trait.cs
@@ -0,0 +1,27 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Verse;
+
+namespace Rimworld_Animations
+{
+    public class PawnTest_Trait : BasePawnTest
+    {
+        TraitDef traitDef;
+        int? degree;
+
+        public override bool PawnTest(Pawn pawn)
+        {
+
+            if (degree != null)
+            {
+                return pawn.story.traits.HasTrait(traitDef, (int)degree);
+            }
+
+            return pawn.story.traits.HasTrait(traitDef);
+        }
+    }
+}
diff --git a/Rimworld-Animations.csproj b/Rimworld-Animations.csproj
index b8ae4cf..9e14a16 100644
--- a/Rimworld-Animations.csproj
+++ b/Rimworld-Animations.csproj
@@ -93,6 +93,7 @@
     <Compile Include="1.5\Source\Animations\PawnTests\PawnTest_Race.cs" />
     <Compile Include="1.5\Source\Animations\PawnTests\PawnTest_RJWCanBeFucked.cs" />
     <Compile Include="1.5\Source\Animations\PawnTests\PawnTest_RJWCanFuck.cs" />
+    <Compile Include="1.5\Source\Animations\PawnTests\PawnTest_Trait.cs" />
     <Compile Include="1.5\Source\Comps\CompExtendedAnimator.cs" />
     <Compile Include="1.5\Source\Comps\CompProperties_ExtendedAnimator.cs" />
     <Compile Include="1.5\Source\Comps\CompProperties_ThingAnimator.cs" />