src/PartSetPlugin/PartSetPlugin_Duplicate.cpp

   1 // File:        PartSetPlugin_Duplicate.cxx
   2 // Created:     20 May 2014
   3 // Author:      Mikhail PONIKAROV
   4
   5 #include "PartSetPlugin_Duplicate.h"
   6 #include <ModelAPI_AttributeRefAttr.h>
   7 #include <ModelAPI_AttributeDocRef.h>
   8 #include <ModelAPI_Data.h>
   9 #include <ModelAPI_Document.h>
  10 #include <ModelAPI_ResultPart.h>
  11
  12 using namespace std;
  13
  14 PartSetPlugin_Duplicate::PartSetPlugin_Duplicate()
  15 {
  16 }
  17
  18 void PartSetPlugin_Duplicate::initAttributes()
  19 {
  20   PartSetPlugin_Part::initAttributes();
  21   data()->addAttribute(ORIGIN_REF(), ModelAPI_AttributeRefAttr::type());
  22
  23   boost::shared_ptr<ModelAPI_PluginManager> aPManager = ModelAPI_PluginManager::get();
  24   boost::shared_ptr<ModelAPI_Document> aRoot = aPManager->rootDocument();
  25   boost::shared_ptr<ModelAPI_ResultPart> aSource;  // searching for source document attribute
  26   for (int a = aRoot->size(getGroup()) - 1; a >= 0; a--) {
  27     aSource = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aRoot->object(getGroup(), a));
  28     if (aSource && aSource->data()
  29         && aSource->data()->docRef(ModelAPI_ResultPart::DOC_REF())->value()
  30             == aPManager->currentDocument())
  31       break;
  32     aSource.reset();
  33   }
  34   if (aSource) {
  35     boost::shared_ptr<ModelAPI_Document> aCopy = aPManager->copy(
  36         aSource->data()->docRef(ModelAPI_ResultPart::DOC_REF())->value(), data()->name());
  37     data()->refattr(ORIGIN_REF())->setObject(aSource);
  38   }
  39 }
  40
  41 void PartSetPlugin_Duplicate::execute()
  42 {
  43   if (data()->refattr(ORIGIN_REF())->object())
  44     PartSetPlugin_Part::execute();
  45 }