Salome HOME
Merge branch 'master' of newgeom:newgeom
[modules/shaper.git] / src / PartSetPlugin / PartSetPlugin_Part.cpp
index e2b47f4ce9b3f35c8982dbf7788a8549f75120b4..937820becdad0902bb4be7f36908548dc20d68f7 100644 (file)
@@ -3,7 +3,7 @@
 // Author:      Mikhail PONIKAROV
 
 #include "PartSetPlugin_Part.h"
-#include "ModelAPI_PluginManager.h"
+#include "ModelAPI_Session.h"
 #include "ModelAPI_Document.h"
 #include "ModelAPI_Data.h"
 #include "ModelAPI_AttributeDocRef.h"
@@ -16,26 +16,29 @@ PartSetPlugin_Part::PartSetPlugin_Part()
 }
 
 void PartSetPlugin_Part::initAttributes()
-{// all is in part result
+{  // all is in part result
 }
 
-void PartSetPlugin_Part::execute() 
+void PartSetPlugin_Part::execute()
 {
   ResultPartPtr aResult = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(firstResult());
   if (!aResult) {
     aResult = document()->createPart(data());
     setResult(aResult);
   }
-  boost::shared_ptr<ModelAPI_AttributeDocRef> aDocRef =
-    aResult->data()->docRef(ModelAPI_ResultPart::DOC_REF());
-
-  if (!aDocRef->value()) { // create a document if not yet created
-    boost::shared_ptr<ModelAPI_Document> aPartSetDoc = 
-      ModelAPI_PluginManager::get()->rootDocument();
+  /*
+  boost::shared_ptr<ModelAPI_AttributeDocRef> aDocRef = aResult->data()->docRef(
+      ModelAPI_ResultPart::DOC_REF());
+  
+  if (!aDocRef->value()) {  // create a document if not yet created
+    boost::shared_ptr<ModelAPI_Document> aPartSetDoc =
+        ModelAPI_Session::get()->moduleDocument();
     aDocRef->setValue(aPartSetDoc->subDocument(data()->name()));
   }
+  */
 }
 
-boost::shared_ptr<ModelAPI_Document> PartSetPlugin_Part::documentToAdd() {
-  return ModelAPI_PluginManager::get()->rootDocument();
+boost::shared_ptr<ModelAPI_Document> PartSetPlugin_Part::documentToAdd()
+{
+  return ModelAPI_Session::get()->moduleDocument();
 }