LoadOrder.FilterToMods

Mutagen-Modding · Dec 15, 2024 · d7ec5af · d7ec5af
1 parent 8097da4
commit d7ec5af
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/Mutagen.Bethesda.Core/Extensions/LoadOrderExt.cs b/Mutagen.Bethesda.Core/Extensions/LoadOrderExt.cs
@@ -205,4 +205,18 @@ public static LoadOrder<TListing> Where<TListing>(this ILoadOrderGetter<TListing
             loadOrder.ListedOrder
                 .Where(filter));
     }
+
+    public static LoadOrder<TListing> FilterToMods<TListing>(this ILoadOrderGetter<TListing> loadOrder, IReadOnlyCollection<ModKey> modKeys)
+        where TListing : IModKeyed
+    {
+        return new LoadOrder<TListing>(
+            loadOrder.ListedOrder
+                .Where(x => modKeys.Contains(x.ModKey)));
+    }
+
+    public static LoadOrder<TListing> FilterToMods<TListing>(this ILoadOrderGetter<TListing> loadOrder, params ModKey[] modKeys)
+        where TListing : IModKeyed
+    {
+        return FilterToMods(loadOrder, modKeys.ToHashSet());
+    }
 }