remove duplicate types

Signed-off-by: Ian Chen <[email protected]>
gazebosim · Oct 11, 2024 · ea75328 · ea75328
1 parent 0da35a8
commit ea75328
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/core/src/MessageFactory.cc b/core/src/MessageFactory.cc
@@ -15,6 +15,8 @@
  *
 */
 
+#include <unordered_set>
+
 #ifdef _MSC_VER
 #pragma warning(push)
 #pragma warning(disable: 4146 4251)
@@ -65,11 +67,6 @@ MessageFactory::MessagePtr MessageFactory::New(
  {
  type = kGzMsgsPrefix + _msgType.substr(9);
  }
- // Convert ".gz.msgs." prefix
- else if (_msgType.find(".gz.msgs.") == 0)
- {
- type = kGzMsgsPrefix + _msgType.substr(9);
- }
  else
  {
  type = _msgType;
@@ -130,15 +127,21 @@ void MessageFactory::Types(std::vector<std::string> &_types)
 {
  _types.clear();
 
+ // Add the types loaded from descriptor files
+ std::vector<std::string> dynTypes;
+ this->dynamicFactory->Types(dynTypes);
+
+ // Use set to remove duplicates
+ std::unordered_set<std::string> typesSet(dynTypes.begin(), dynTypes.end());
+
  // Return the list of all known message types.
  std::map<std::string, FactoryFn>::const_iterator iter;
  for (iter = msgMap.begin(); iter != msgMap.end(); ++iter)
  {
- _types.push_back(iter->first);
+ typesSet.insert(iter->first);
  }
 
- // Add the types loaded from descriptor files
- this->dynamicFactory->Types(_types);
+ std::copy(typesSet.begin(), typesSet.end(), std::back_inserter(_types));
 }
 
 /////////////////////////////////////////////////