Merge branch 'master' of newgeom:newgeom

[modules/shaper.git] / src / PartSetPlugin / PartSetPlugin_Plugin.cpp

diff --git a/src/PartSetPlugin/PartSetPlugin_Plugin.cpp b/src/PartSetPlugin/PartSetPlugin_Plugin.cpp
index 85ea1c37eacf527337a523ae115b3684134ef227..dc384608e6552ca7701115cbfd68d0c1889cd376 100644 (file)
--- a/src/PartSetPlugin/PartSetPlugin_Plugin.cpp
+++ b/src/PartSetPlugin/PartSetPlugin_Plugin.cpp
@@ -1,5 +1,7 @@
 #include "PartSetPlugin_Plugin.h"
 #include "PartSetPlugin_Part.h"
+#include "PartSetPlugin_Duplicate.h"
+#include "PartSetPlugin_Remove.h"
 #include <ModelAPI_PluginManager.h>
 #include <ModelAPI_Document.h>
 
@@ -14,11 +16,17 @@ PartSetPlugin_Plugin::PartSetPlugin_Plugin()
   ModelAPI_PluginManager::get()->registerPlugin(this);
 }
 
-boost::shared_ptr<ModelAPI_Feature> PartSetPlugin_Plugin::createFeature(string theFeatureID)
+FeaturePtr PartSetPlugin_Plugin::createFeature(string theFeatureID)
 {
-  if (theFeatureID == "Part") {
-    return boost::shared_ptr<ModelAPI_Feature>(new PartSetPlugin_Part);
+  if (theFeatureID == PartSetPlugin_Part::ID()) {
+    return FeaturePtr(new PartSetPlugin_Part);
   }
-  // feature of such kind is not found
-  return boost::shared_ptr<ModelAPI_Feature>();
+  if (theFeatureID == "duplicate") {
+    return FeaturePtr(new PartSetPlugin_Duplicate);
+  }
+  if (theFeatureID == "remove") {
+    return FeaturePtr(new PartSetPlugin_Remove);
+  }
+    // feature of such kind is not found
+  return FeaturePtr();
 }