Salome HOME
Merge branch 'master' of newgeom:newgeom.git
[modules/shaper.git] / src / PartSetPlugin / PartSetPlugin_Part.cpp
1 // File:        PartSetPlugin_Part.cxx
2 // Created:     27 Mar 2014
3 // Author:      Mikhail PONIKAROV
4
5 #include "PartSetPlugin_Part.h"
6 #include "ModelAPI_PluginManager.h"
7 #include "ModelAPI_Document.h"
8 #include "ModelAPI_Data.h"
9 #include "ModelAPI_Iterator.h"
10 #include "ModelAPI_AttributeDocRef.h"
11
12 using namespace std;
13
14 PartSetPlugin_Part::PartSetPlugin_Part()
15 {
16 }
17
18 void PartSetPlugin_Part::initAttributes()
19 {
20   data()->addAttribute(PART_ATTR_DOC_REF, ModelAPI_AttributeDocRef::type());
21 }
22
23 void PartSetPlugin_Part::execute() 
24 {
25   boost::shared_ptr<ModelAPI_AttributeDocRef> aDocRef = data()->docRef(PART_ATTR_DOC_REF);
26   if (!aDocRef->value()) { // create a document if not yet created
27     boost::shared_ptr<ModelAPI_Document> aPartSetDoc = ModelAPI_PluginManager::get()->rootDocument();
28     aDocRef->setValue(aPartSetDoc->subDocument(data()->getName()));
29   }
30 }
31
32 boost::shared_ptr<ModelAPI_Document> PartSetPlugin_Part::documentToAdd() {
33   return ModelAPI_PluginManager::get()->rootDocument();
34 }