Fix setting the primary category of multiple mods at once.

1 files changed, 7 insertions, 1 deletions

diff --git a/src/modlistviewactions.h b/src/modlistviewactions.h
index 65f323d9..2850d30a 100644
--- a/src/modlistviewactions.h
+++ b/src/modlistviewactions.h
@@ -87,7 +87,7 @@ public:
   void setColor(const QModelIndexList& indices, const QModelIndex& refIndex) const;
   void resetColor(const QModelIndexList& indices) const;
 
-  // set the category of the mod in the given list, using the given index as reference
+  // set the category of the mods in the given list, using the given index as reference
   // - the categories are set as-is on the refernce mod
   // - for the other mods, the category is only set if the current state of the category
   //   on the reference is different
@@ -95,6 +95,12 @@ public:
   void setCategories(const QModelIndexList& selected, const QModelIndex& ref,
     const std::vector<std::pair<int, bool>>& categories) const;
 
+  // set the primary category of the mods in the given list, adding the appropriate
+  // category to the mods unless force is false, in which case the primary category
+  // is set only on mods that have this category
+  //
+  void setPrimaryCategory(const QModelIndexList& selected, int category, bool force = true);
+
   // open the Windows explorer for the specified mods
   //
   void openExplorer(const QModelIndexList& index) const;