Keep models of a category in the order of registration

Author: Quincunx271

include/nodes/internal/DataModelRegistry.hpp

@@ -36,6 +36,7 @@ class NODE_EDITOR_PUBLIC DataModelRegistry
   using RegistryItemCreator = std::function<RegistryItemPtr()>;
   using RegisteredModelCreatorsMap = std::unordered_map<QString, RegistryItemCreator>;
   using RegisteredModelsCategoryMap = std::unordered_map<QString, QString>;
+  using RegisteredModelsOrder = std::vector<QString>;
   using CategoriesSet = std::set<QString>;
 
   using RegisteredTypeConvertersMap = std::map<TypeConverterId, TypeConverter>;
@@ -82,6 +83,8 @@ class NODE_EDITOR_PUBLIC DataModelRegistry
 
   RegisteredModelCreatorsMap const &registeredModelCreators() const;
 
+  RegisteredModelsOrder const& registeredModelsOrder() const;
+
   RegisteredModelsCategoryMap const &registeredModelsCategoryAssociation() const;
 
   CategoriesSet const &categories() const;
@@ -97,6 +100,8 @@ class NODE_EDITOR_PUBLIC DataModelRegistry
 
   RegisteredModelCreatorsMap _registeredItemCreators;
 
+  RegisteredModelsOrder _registeredModelsOrder;
+
   RegisteredTypeConvertersMap _registeredTypeConverters;
 
 private:
@@ -129,6 +134,7 @@ class NODE_EDITOR_PUBLIC DataModelRegistry
     {
       _registeredItemCreators[name] = std::move(creator);
       _categories.insert(category);
+      _registeredModelsOrder.push_back(name);
       _registeredModelsCategory[name] = category;
     }
   }
@@ -142,6 +148,7 @@ class NODE_EDITOR_PUBLIC DataModelRegistry
     {
       _registeredItemCreators[name] = std::move(creator);
       _categories.insert(category);
+      _registeredModelsOrder.push_back(name);
       _registeredModelsCategory[name] = category;
     }
   }

src/DataModelRegistry.cpp

@@ -30,6 +30,12 @@ registeredModelCreators() const
   return _registeredItemCreators;
 }
 
+DataModelRegistry::RegisteredModelsOrder const &
+DataModelRegistry::
+registeredModelsOrder() const
+{
+  return _registeredModelsOrder;
+}
 
 DataModelRegistry::RegisteredModelsCategoryMap const &
 DataModelRegistry::

src/FlowView.cpp

@@ -141,12 +141,13 @@ contextMenuEvent(QContextMenuEvent *event)
     topLevelItems[cat] = item;
   }
 
-  for (auto const &assoc : _scene->registry().registeredModelsCategoryAssociation())
+  auto const &assocCategory = _scene->registry().registeredModelsCategoryAssociation();
+  for (auto const &name : _scene->registry().registeredModelsOrder())
   {
-    auto parent = topLevelItems[assoc.second];
+    auto parent = topLevelItems[assocCategory.at(name)];
     auto item   = new QTreeWidgetItem(parent);
-    item->setText(0, assoc.first);
-    item->setData(0, Qt::UserRole, assoc.first);
+    item->setText(0, name);
+    item->setData(0, Qt::UserRole, name);
   }
 
   treeView->expandAll();