Draft for Thoughts and Penalties of Missing or Multiple Queens

Source/Genes/Hive/Genes/ConditionalStatAffecters/ConditionalStatAffecter_MultipleQueens.cs

@@ -19,8 +19,15 @@ namespace RJW_Genes
 
         public override bool Applies(StatRequest req)
         {
-            // => ModsConfig.BiotechActive && req.HasThing && req.Thing.Spawned && req.Thing.Position.InSunlight(req.Thing.Map);
-            throw new NotImplementedException();
+            if (req.Pawn == null || !req.Pawn.Spawned)
+                return false;
+
+            if (GeneUtility.HasGeneNullCheck(req.Pawn, GeneDefOf.rjw_genes_zealous_loyalty))
+            {
+                return HiveUtility.QueensOnMap() >= 2;
+            }
+
+            return false;
         } 
     }
 }

Source/Genes/Hive/Genes/ConditionalStatAffecters/ConditionalStatAffecter_QueenAbsent.cs

@@ -19,8 +19,15 @@ namespace RJW_Genes
 
         public override bool Applies(StatRequest req)
         {
-            //ModsConfig.BiotechActive && req.HasThing && req.Thing.Spawned && req.Thing.Position.InSunlight(req.Thing.Map);
-            throw new NotImplementedException();
+            if (req.Pawn == null || !req.Pawn.Spawned)
+                return false;
+
+            if (GeneUtility.HasGeneNullCheck(req.Pawn, GeneDefOf.rjw_genes_zealous_loyalty))
+            {
+                return HiveUtility.QueensOnMap() == 0;
+            }
+
+            return false;
         }
     }
 }

Source/Genes/Hive/Genes/ConditionalStatAffecters/ConditionalStatAffecter_QueenCloseBy.cs

@@ -17,12 +17,36 @@ namespace RJW_Genes
     /// </summary>
     public class ConditionalStatAffecter_QueenCloseBy : ConditionalStatAffecter
     {
+
+        const float EFFECT_DISTANCE = 10.0f;
+
         public override string Label => (string)"StatsReport_QueenCloseBy".Translate();
 
         public override bool Applies(StatRequest req)
         {
-            // => ModsConfig.BiotechActive && req.HasThing && req.Thing.Spawned && req.Thing.Position.InSunlight(req.Thing.Map);
-            throw new NotImplementedException();
-        } 
+            if (req.Pawn == null || !req.Pawn.Spawned)
+                return false;
+
+            // Case A: Check for Loyal Pawns if their One Queen is nearby
+            if (GeneUtility.HasGeneNullCheck(req.Pawn, GeneDefOf.rjw_genes_zealous_loyalty) && HiveUtility.QueensOnMap() == 1)
+            {
+                Pawn queen = HiveUtility.GetQueensOnMap()[0];
+
+                return req.Pawn.Position.DistanceTo(queen.Position) <= EFFECT_DISTANCE;
+            }
+
+            // Case A: Check for Queen if another Queen is nearby
+            if (GeneUtility.HasGeneNullCheck(req.Pawn, GeneDefOf.rjw_genes_zealous_loyalty) && HiveUtility.QueensOnMap() >= 2)
+            {
+                foreach (Pawn queen in HiveUtility.GetQueensOnMap())
+                {
+                    if (queen != req.Pawn && req.Pawn.Position.DistanceTo(queen.Position) <= EFFECT_DISTANCE)
+                        return true;
+                }
+            }
+
+
+            return false;
+        }
     }
 }

Source/Genes/Hive/Helpers/HiveUtility.cs

@@ -0,0 +1,39 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Verse;
+
+namespace RJW_Genes
+{
+    internal class HiveUtility
+    {
+        public static bool IsAdultQueen(Pawn pawn)
+        {
+
+            if (pawn == null || !pawn.Spawned)
+                return false;
+
+            if (GeneUtility.HasGeneNullCheck(pawn, GeneDefOf.rjw_genes_queen)) 
+            { 
+                return pawn.ageTracker.Adult;
+            }
+
+            return false;
+        }
+
+        public static int QueensOnMap()
+        {
+            List<Pawn> playersPawns = Find.CurrentMap.mapPawns.SpawnedPawnsInFaction(Faction.OfPlayer);
+            return playersPawns.Count(pawn => pawn.Spawned && IsAdultQueen(pawn));
+        }
+
+        public static List<Pawn> GetQueensOnMap()
+        {
+            List<Pawn> playersPawns = Find.CurrentMap.mapPawns.SpawnedPawnsInFaction(Faction.OfPlayer);
+            return playersPawns.FindAll(pawn => pawn.Spawned && IsAdultQueen(pawn));
+        }
+    }
+}

Source/Genes/Hive/Thoughts/ThoughtWorker_QueenPresent_Social.cs

@@ -0,0 +1,38 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Verse;
+
+namespace RJW_Genes
+{
+    public class ThoughtWorker_QueenPresent_Social : ThoughtWorker
+    {
+        protected override ThoughtState CurrentSocialStateInternal(Pawn p, Pawn other)
+        {
+            if (!p.RaceProps.Humanlike)
+                return (ThoughtState) false;
+
+            if (!other.RaceProps.Humanlike)
+                return (ThoughtState) false;
+
+            if (!RelationsUtility.PawnsKnowEachOther(p, other))
+                return (ThoughtState) false;
+
+            // Only check if they are spawned 
+            if (!p.Spawned || !other.Spawned)
+            {
+                return (ThoughtState)false;
+            }
+
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_zealous_loyalty) && HiveUtility.QueensOnMap() == 1)
+            {
+                return (ThoughtState) HiveUtility.IsAdultQueen(other);
+            }
+
+            return (ThoughtState)false;
+        }
+    }
+}

Source/Genes/Hive/Thoughts/ThoughtWorker_RivalQueen_Social.cs

@@ -0,0 +1,36 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Verse;
+
+namespace RJW_Genes
+{
+    public class ThoughtWorker_RivalQueen_Social : ThoughtWorker
+    {
+        protected override ThoughtState CurrentSocialStateInternal(Pawn p, Pawn other)
+        {
+            if (!p.RaceProps.Humanlike)
+                return (ThoughtState)false;
+            if (!other.RaceProps.Humanlike)
+                return (ThoughtState)false;
+
+            if (!RelationsUtility.PawnsKnowEachOther(p, other))
+                return (ThoughtState)false;
+
+            // Only check if they are spawned 
+            if (!p.Spawned || !other.Spawned)
+            {
+                return (ThoughtState)false;
+            }
+
+            if(HiveUtility.IsAdultQueen(p) && HiveUtility.IsAdultQueen(other))
+            {
+                return (ThoughtState)true;
+            }
+            return (ThoughtState)false;
+        }
+    }
+}

Source/Genes/Hive/Thoughts/Thoughtworker_MultipleQueens_Mood.cs

@@ -0,0 +1,33 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+using Verse;
+
+namespace RJW_Genes
+{
+    public class Thoughtworker_MultipleQueens_Mood : ThoughtWorker
+    {
+
+        protected override ThoughtState CurrentStateInternal(Pawn p)
+        {
+            // Error Handling and Check for Pawn being on Map
+            if (p == null || !p.Spawned)
+                return (ThoughtState) false;
+            // Queens cannot have loyalty thoughts
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_queen))
+                return (ThoughtState)false;
+
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_zealous_loyalty) && HiveUtility.QueensOnMap() >= 2)
+            {
+                return (ThoughtState)true;
+            }
+
+            return (ThoughtState) false;
+        }
+
+    }
+}

Source/Genes/Hive/Thoughts/Thoughtworker_QueenAbsent_Mood.cs

@@ -0,0 +1,33 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+using Verse;
+
+namespace RJW_Genes
+{
+    public class Thoughtworker_QueenAbsent_Mood : ThoughtWorker
+    {
+
+        protected override ThoughtState CurrentStateInternal(Pawn p)
+        {
+            // Error Handling and Check for Pawn being on Map
+            if (p == null || !p.Spawned)
+                return (ThoughtState) false;
+            // Queens cannot have loyalty thoughts
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_queen))
+                return (ThoughtState)false;
+
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_zealous_loyalty) && HiveUtility.QueensOnMap() == 0)
+            {
+                return (ThoughtState)true;
+            }
+
+            return (ThoughtState) false;
+        }
+
+    }
+}

Source/Genes/Hive/Thoughts/Thoughtworker_QueenPresent_Mood.cs

@@ -0,0 +1,33 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+using Verse;
+
+namespace RJW_Genes
+{
+    public class Thoughtworker_QueenPresent_Mood : ThoughtWorker
+    {
+
+        protected override ThoughtState CurrentStateInternal(Pawn p)
+        {
+            // Error Handling and Check for Pawn being on Map
+            if (p == null || !p.Spawned)
+                return (ThoughtState) false;
+            // Queens cannot have loyalty thoughts
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_queen))
+                return (ThoughtState)false;
+
+            if (GeneUtility.HasGeneNullCheck(p, GeneDefOf.rjw_genes_zealous_loyalty) && HiveUtility.QueensOnMap() == 1)
+            {
+                return (ThoughtState)true;
+            }
+
+            return (ThoughtState) false;
+        }
+
+    }
+}

Source/Genes/Hive/Thoughts/Thoughtworker_RivalQueen_Mood.cs

@@ -0,0 +1,29 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+using Verse;
+
+namespace RJW_Genes
+{
+    public class Thoughtworker_RivalQueen_Mood : ThoughtWorker
+    {
+
+        protected override ThoughtState CurrentStateInternal(Pawn p)
+        {
+            if (p == null || !p.Spawned)
+                return (ThoughtState) false;
+
+            if (HiveUtility.IsAdultQueen(p) && HiveUtility.QueensOnMap() >= 2)
+            {
+                return (ThoughtState) true;
+            }
+
+            return (ThoughtState) false;
+        }
+
+    }
+}