Issue #2811: Update content of Object node on creation moment

[modules/shaper.git] / src / ExchangePlugin / ExchangePlugin_Dump.cpp

diff --git a/src/ExchangePlugin/ExchangePlugin_Dump.cpp b/src/ExchangePlugin/ExchangePlugin_Dump.cpp
index d89f6d98a7dfd8cb1eefcb0e1bf38fc3edfaf259..dbec8bef6c28bc8dc91876fdf2bda9c0af0fe3d4 100644 (file)
--- a/src/ExchangePlugin/ExchangePlugin_Dump.cpp
+++ b/src/ExchangePlugin/ExchangePlugin_Dump.cpp
@@ -1,14 +1,30 @@
-// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
-
-// File:    ExchangePlugin_ExportFeature.cpp
-// Created: May 14, 2015
-// Author:  Sergey POKHODENKO
+// Copyright (C) 2014-2017  CEA/DEN, EDF R&D
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+//
+// See http://www.salome-platform.org/ or
+// email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
+//
 
 #include <ExchangePlugin_Dump.h>
 
+#include <ModelAPI_AttributeBoolean.h>
 #include <ModelAPI_AttributeString.h>
 #include <ModelAPI_Document.h>
 #include <ModelAPI_Session.h>
+#include <ModelAPI_ResultPart.h>
 
 #include <ModelHighAPI_Dumper.h>
 
@@ -25,8 +41,11 @@ ExchangePlugin_Dump::~ExchangePlugin_Dump()
 
 void ExchangePlugin_Dump::initAttributes()
 {
-  data()->addAttribute(ExchangePlugin_Dump::FILE_PATH_ID(), ModelAPI_AttributeString::typeId());
-  data()->addAttribute(ExchangePlugin_Dump::FILE_FORMAT_ID(), ModelAPI_AttributeString::typeId());
+  data()->addAttribute(FILE_PATH_ID(), ModelAPI_AttributeString::typeId());
+  data()->addAttribute(FILE_FORMAT_ID(), ModelAPI_AttributeString::typeId());
+
+  data()->addAttribute(SELECTION_TYPE_ID(), ModelAPI_AttributeString::typeId());
+  //string(SELECTION_TYPE_ID())->setValue(TOPOLOGICAL_NAMING_DUMP_ID()); // default value
 }
 
 void ExchangePlugin_Dump::execute()
@@ -43,10 +62,50 @@ void ExchangePlugin_Dump::execute()
 void ExchangePlugin_Dump::dump(const std::string& theFileName)
 {
   // load DumpAssistant from Python side
-  Config_ModuleReader::loadScript("model.dump");
+  Config_ModuleReader::loadScript("salome.shaper.model.dump");
 
   ModelHighAPI_Dumper* aDumper = ModelHighAPI_Dumper::getInstance();
+  aDumper->clear();
+  aDumper->setSelectionByGeometry(string(SELECTION_TYPE_ID())->value() == GEOMETRIC_DUMP_ID());
+  aDumper->setSelectionWeakNaming(string(SELECTION_TYPE_ID())->value() == WEAK_NAMING_DUMP_ID());
+
   DocumentPtr aDoc = ModelAPI_Session::get()->moduleDocument();
+
+  int aFeaturesNb = aDoc->size(ModelAPI_Feature::group());
+  if(aFeaturesNb > 1) {
+    FeaturePtr aLastFeature =
+        ModelAPI_Feature::feature(aDoc->object(ModelAPI_Feature::group(), aFeaturesNb - 1));
+    if(aDoc->currentFeature(true) != aLastFeature) {
+        setError("Dump cannot be done. Please move the history line to the end before dumping.");
+        return;
+    }
+  }
+
+  DocumentPtr anActiveDoc = ModelAPI_Session::get()->activeDocument();
+  aFeaturesNb = anActiveDoc->size(ModelAPI_Feature::group());
+  if(aFeaturesNb > 1) {
+    FeaturePtr aLastFeature =
+        ModelAPI_Feature::feature(anActiveDoc->object(ModelAPI_Feature::group(), aFeaturesNb - 1));
+    if(anActiveDoc->currentFeature(true) != aLastFeature) {
+        setError("Dump cannot be done. Please move the history line to the end before dumping.");
+        return;
+    }
+  }
+
+  std::list<FeaturePtr> aFeatures = aDoc->allFeatures();
+  for(std::list<FeaturePtr>::const_iterator aFeatIt = aFeatures.begin();
+      aFeatIt != aFeatures.end();
+      ++aFeatIt) {
+    ResultPartPtr aResultPart =
+      std::dynamic_pointer_cast<ModelAPI_ResultPart>((*aFeatIt)->firstResult());
+    if(aResultPart.get()) {
+      if(!aResultPart->isActivated()) {
+        setError("Error: Not all parts are loaded. Can not dump.");
+        return;
+      }
+    }
+  }
+
   if (!aDumper || !aDumper->process(aDoc, theFileName))
     setError("An error occured while dumping to " + theFileName);
 }