Added object, attribute and data model organization in the document.

[modules/shaper.git] / src / PartSetPlugin / PartSetPlugin_NewPart.cxx

diff --git a/src/PartSetPlugin/PartSetPlugin_NewPart.cxx b/src/PartSetPlugin/PartSetPlugin_NewPart.cxx
index 7959ef0442a988cea52d86d2660f1df6ceaf86eb..2b2c59bc31133f6c560e0250e32e422c5a831997 100644 (file)
--- a/src/PartSetPlugin/PartSetPlugin_NewPart.cxx
+++ b/src/PartSetPlugin/PartSetPlugin_NewPart.cxx
@@ -2,8 +2,26 @@
 // Created:     27 Mar 2014
 // Author:      Mikhail PONIKAROV
 
-#include "PartSetPlugin_NewPart.hxx"
+#include "PartSetPlugin_NewPart.h"
+#include "ModelAPI_PluginManager.h"
+#include "ModelAPI_Document.h"
+#include "ModelAPI_Object.h"
+#include "ModelAPI_AttributeDocRef.h"
+
+using namespace std;
 
 PartSetPlugin_NewPart::PartSetPlugin_NewPart()
 {
 }
+
+void PartSetPlugin_NewPart::initAttributes()
+{
+  data()->addAttribute(ModelAPI_AttributeDocRef::type(), PART_DOC_REF);
+}
+
+void PartSetPlugin_NewPart::execute() 
+{
+  shared_ptr<ModelAPI_Document> aPartSetDoc = ModelAPI_PluginManager::get()->rootDocument();
+  data()->setName(string("Part_") + "1");
+  aPartSetDoc->subDocument(string("Part_") + "1");
+}