Copyright update 2020

[modules/shaper.git] / src / ExchangeAPI / ExchangeAPI_Import.cpp

diff --git a/src/ExchangeAPI/ExchangeAPI_Import.cpp b/src/ExchangeAPI/ExchangeAPI_Import.cpp
index 81b34ac03e66d53ddfbf1a6f921d0c6833c51327..b1a6c832d0cee6027f99b2ae059de409be245b16 100644 (file)
--- a/src/ExchangeAPI/ExchangeAPI_Import.cpp
+++ b/src/ExchangeAPI/ExchangeAPI_Import.cpp
@@ -1,4 +1,4 @@
-// Copyright (C) 2014-2019  CEA/DEN, EDF R&D
+// Copyright (C) 2014-2020  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
@@ -25,6 +25,10 @@
 #include <ModelHighAPI_Services.h>
 #include <ModelHighAPI_Tools.h>
 //--------------------------------------------------------------------------------------
+#include <ModelAPI_AttributeStringArray.h>
+#include <ModelAPI_Session.h>
+#include <ModelAPI_Tools.h>
+//--------------------------------------------------------------------------------------
 #include <algorithm>
 
 ExchangeAPI_Import::ExchangeAPI_Import(
@@ -111,7 +115,25 @@ void importPart(const std::shared_ptr<ModelAPI_Document> & thePart,
 
   FeaturePtr aFeature = thePart->addFeature(ExchangePlugin_ImportPart::ID());
   aFeature->string(ExchangePlugin_ImportPart::FILE_PATH_ID())->setValue(theFilePath);
-  // restart transaction to execute and delete the marcro-feature
+
+  // specify the ID of selected document
+  int aTargetPartIndex = 0;
+  SessionPtr aSession = ModelAPI_Session::get();
+  if (aSession->moduleDocument() == thePart) {
+    // Importing to PartSet has 2 choices: import directly to PartSet (if possible)
+    // or create a new part. Because then importing to existing part the document
+    // has to be specified explicitly.
+    // As a result, parse the list of possible target documents and generate new part
+    // if the import document is not applicable on PartSet level
+    // (there is no 'PartSet' in the list of applicable documents).
+    AttributeStringArrayPtr aDocsList =
+        aFeature->stringArray(ExchangePlugin_ImportPart::TARGET_PARTS_LIST_ID());
+    if (aDocsList->size() > 1 && aDocsList->value(1) == "PartSet")
+      aTargetPartIndex = 1;
+  }
+  aFeature->integer(ExchangePlugin_ImportPart::TARGET_PART_ID())->setValue(aTargetPartIndex);
+
+  // restart transaction to execute and delete the macro-feature
   apply();
 
   // restore current feature