src/PartSetPlugin/PartSetPlugin_Duplicate.cpp

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