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 #include <ModelAPI_Session.h>
  12
  13 using namespace std;
  14
  15 PartSetPlugin_Duplicate::PartSetPlugin_Duplicate()
  16 {
  17 }
  18
  19 void PartSetPlugin_Duplicate::initAttributes()
  20 {
  21   PartSetPlugin_Part::initAttributes();
  22   data()->addAttribute(ORIGIN_REF(), ModelAPI_AttributeRefAttr::type());
  23
  24   std::shared_ptr<ModelAPI_Session> aPManager = ModelAPI_Session::get();
  25   std::shared_ptr<ModelAPI_Document> aRoot = aPManager->moduleDocument();
  26   std::shared_ptr<ModelAPI_ResultPart> aSource;  // searching for source document attribute
  27   for (int a = aRoot->size(getGroup()) - 1; a >= 0; a--) {
  28     aSource = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aRoot->object(getGroup(), a));
  29     if (aSource && aSource->data()
  30         && aSource->data()->document(ModelAPI_ResultPart::DOC_REF())->value()
  31             == aPManager->activeDocument())
  32       break;
  33     aSource.reset();
  34   }
  35   if (aSource) {
  36     std::shared_ptr<ModelAPI_Document> aCopy = aPManager->copy(
  37         aSource->data()->document(ModelAPI_ResultPart::DOC_REF())->value(), data()->name());
  38     data()->refattr(ORIGIN_REF())->setObject(aSource);
  39   }
  40 }
  41
  42 void PartSetPlugin_Duplicate::execute()
  43 {
  44   if (data()->refattr(ORIGIN_REF())->object())
  45     PartSetPlugin_Part::execute();
  46 }