Create ThingSetMaker_CountEnabledAmmoOnly.cs

CombatExtended-Continued · Aug 25, 2024 · 3c47a1e · 3c47a1e
1 parent 88d45a3
commit 3c47a1e
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/Source/CombatExtended/ThingSetMaker_CountEnabledAmmoOnly.cs b/Source/CombatExtended/ThingSetMaker_CountEnabledAmmoOnly.cs
@@ -0,0 +1,22 @@
+using System.Collections.Generic;
+using RimWorld;
+using Verse;
+
+namespace CombatExtended
+{
+    public class ThingSetMaker_CountEnabledAmmoOnly : ThingSetMaker_Count
+    {
+        public override void Generate(ThingSetMakerParams parms, List<Thing> outThings)
+        {
+            parms.validator = delegate (ThingDef d)
+            {
+                if (d is AmmoDef ammodef && (ammodef.tradeTags?.Contains(AmmoInjector.destroyWithAmmoDisabledTag) ?? false))
+                {
+                    return AmmoUtility.IsAmmoSystemActive(ammodef);
+                }
+                return true;
+            };
+            base.Generate(parms, outThings);
+        }
+    }
+}