X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2FPartSetPlugin%2FPartSetPlugin_Plugin.cpp;h=dc384608e6552ca7701115cbfd68d0c1889cd376;hb=036ba4eb4c2df048fd651a54f68882a01769eb08;hp=85ea1c37eacf527337a523ae115b3684134ef227;hpb=6654a921211670c15a79a997ca666a174fe8a6f6;p=modules%2Fshaper.git

diff --git a/src/PartSetPlugin/PartSetPlugin_Plugin.cpp b/src/PartSetPlugin/PartSetPlugin_Plugin.cpp
index 85ea1c37e..dc384608e 100644
--- 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();
 }