+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
// File: PartSetPlugin_Duplicate.cxx
// Created: 20 May 2014
// Author: Mikhail PONIKAROV
#include <ModelAPI_AttributeDocRef.h>
#include <ModelAPI_Data.h>
#include <ModelAPI_Document.h>
+#include <ModelAPI_ResultPart.h>
+#include <ModelAPI_Session.h>
using namespace std;
void PartSetPlugin_Duplicate::initAttributes()
{
PartSetPlugin_Part::initAttributes();
- data()->addAttribute(PART_DUPLICATE_ATTR_REF, ModelAPI_AttributeRefAttr::type());
-
- boost::shared_ptr<ModelAPI_AttributeRefAttr> aRef = data()->refattr(PART_DUPLICATE_ATTR_REF);
- if (!aRef->attr()) { // create a copy: if not created yet attribute is not initialized
- boost::shared_ptr<ModelAPI_PluginManager> aPManager = ModelAPI_PluginManager::get();
- boost::shared_ptr<ModelAPI_Document> aRoot = aPManager->rootDocument();
- boost::shared_ptr<PartSetPlugin_Part> aSource; // searching for source document attribute
- for(int a = aRoot->size(getGroup()) - 1; a >= 0; a--) {
- aSource = boost::dynamic_pointer_cast<PartSetPlugin_Part>(
- aRoot->object(ModelAPI_Feature::group(), a));
- if (aSource->data()->docRef(PART_ATTR_DOC_REF)->value() == aPManager->currentDocument())
- break;
- aSource.reset();
- }
- if (aSource) {
- boost::shared_ptr<ModelAPI_Document> aCopy =
- aPManager->copy(aSource->data()->docRef(PART_ATTR_DOC_REF)->value(), data()->name());
- aRef->setObject(aSource);
- }
+
+ std::shared_ptr<ModelAPI_Session> aPManager = ModelAPI_Session::get();
+ std::shared_ptr<ModelAPI_Document> aRoot = aPManager->moduleDocument();
+ std::shared_ptr<ModelAPI_ResultPart> aSource; // searching for source document attribute
+ for (int a = aRoot->size(getGroup()) - 1; a >= 0; a--) {
+ aSource = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aRoot->object(getGroup(), a));
+ if (aSource && aSource->data()
+ && aSource->data()->document(ModelAPI_ResultPart::DOC_REF())->value()
+ == aPManager->activeDocument())
+ break;
+ aSource.reset();
+ }
+ if (aSource) {
+ std::shared_ptr<ModelAPI_Document> aCopy = aPManager->copy(
+ aSource->data()->document(ModelAPI_ResultPart::DOC_REF())->value(), data()->name());
+ } else { // invalid part copy: do nothing, keep this in empty name
+ data()->setName("");
}
}
+
+void PartSetPlugin_Duplicate::execute()
+{
+ if (!data()->name().empty())
+ PartSetPlugin_Part::execute();
+}
+
+const std::string& PartSetPlugin_Duplicate::documentToAdd()
+{
+ // part must be added only to the module document
+ return ModelAPI_Session::get()->moduleDocument()->kind();
+}