Salome HOME
bos #29482 Export of colors and names to STEP.
[modules/shaper.git] / src / ExchangePlugin / ExchangePlugin_ExportFeature.cpp
index 93500cc5e7660a345a7aa6e851878269123e23fe..54400c68680dcd2c1f5242bf3d192cabc3b04435 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2014-2019  CEA/DEN, EDF R&D
+// Copyright (C) 2014-2022  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
@@ -34,6 +34,7 @@
 #include <GeomAlgoAPI_CompoundBuilder.h>
 #include <GeomAlgoAPI_IGESExport.h>
 #include <GeomAlgoAPI_STEPExport.h>
+#include <GeomAlgoAPI_STLExport.h>
 #include <GeomAlgoAPI_Tools.h>
 #include <GeomAlgoAPI_XAOExport.h>
 
 #include <GeomAPI_ShapeExplorer.h>
 #include <GeomAPI_Trsf.h>
 
+#include <Locale_Convert.h>
+
 #include <ModelAPI_AttributeSelectionList.h>
 #include <ModelAPI_AttributeString.h>
 #include <ModelAPI_AttributeStringArray.h>
 #include <ModelAPI_AttributeIntArray.h>
 #include <ModelAPI_AttributeTables.h>
+#include <ModelAPI_AttributeDouble.h>
 #include <ModelAPI_Data.h>
 #include <ModelAPI_Document.h>
 #include <ModelAPI_Object.h>
@@ -96,9 +100,27 @@ void ExchangePlugin_ExportFeature::initAttributes()
     ModelAPI_AttributeString::typeId());
   data()->addAttribute(ExchangePlugin_ExportFeature::XAO_SELECTION_LIST_ID(),
     ModelAPI_AttributeSelectionList::typeId());
+  data()->addAttribute(ExchangePlugin_ExportFeature::STL_FILE_PATH_ID(),
+    ModelAPI_AttributeString::typeId());
+  data()->addAttribute(ExchangePlugin_ExportFeature::STL_OBJECT_SELECTED(),
+    ModelAPI_AttributeSelection::typeId());
+  data()->addAttribute(ExchangePlugin_ExportFeature::STL_DEFLECTION_TYPE(),
+   ModelAPI_AttributeString::typeId());
+  data()->addAttribute(ExchangePlugin_ExportFeature::STL_RELATIVE(),
+    ModelAPI_AttributeDouble::typeId());
+
+  double defelection = Config_PropManager::real("Visualization", "body_deflection");
+  real(ExchangePlugin_ExportFeature::STL_RELATIVE())->setValue(defelection);
+
+  data()->addAttribute(ExchangePlugin_ExportFeature::STL_ABSOLUTE(),
+    ModelAPI_AttributeDouble::typeId());
+  data()->addAttribute(ExchangePlugin_ExportFeature::STL_FILE_TYPE(),
+   ModelAPI_AttributeString::typeId());
 
   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(),
     ExchangePlugin_ExportFeature::XAO_FILE_PATH_ID());
+  ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(),
+    ExchangePlugin_ExportFeature::STL_FILE_PATH_ID());
   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(),
     ExchangePlugin_ExportFeature::XAO_AUTHOR_ID());
   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(),
@@ -126,6 +148,10 @@ void ExchangePlugin_ExportFeature::attributeChanged(const std::string& theID)
     string(ExchangePlugin_ExportFeature::FILE_PATH_ID())->setValue(
       string(ExchangePlugin_ExportFeature::XAO_FILE_PATH_ID())->value());
   }
+  else if (theID == STL_FILE_PATH_ID()) {
+    string(ExchangePlugin_ExportFeature::FILE_PATH_ID())->setValue(
+      string(ExchangePlugin_ExportFeature::STL_FILE_PATH_ID())->value());
+  }
 }
 
 /*
@@ -168,19 +194,33 @@ void ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName,
   if (aFormatName == "XAO") {
     exportXAO(theFileName);
     return;
+  }else if (aFormatName == "STL") {
+    exportSTL(theFileName);
+    return;
   }
 
   // make shape for export from selected shapes
   AttributeSelectionListPtr aSelectionListAttr =
       this->selectionList(ExchangePlugin_ExportFeature::SELECTION_LIST_ID());
   std::list<GeomShapePtr> aShapes;
+  std::list<ResultPtr> aContexts;
   for (int i = 0, aSize = aSelectionListAttr->size(); i < aSize; ++i) {
     AttributeSelectionPtr anAttrSelection = aSelectionListAttr->value(i);
+
+    /// do not export pictures
+    ResultPtr aBodyContext =
+      std::dynamic_pointer_cast<ModelAPI_Result>(anAttrSelection->context());
+    if (aBodyContext.get() && aBodyContext->hasTexture())
+      continue;
+
     std::shared_ptr<GeomAPI_Shape> aCurShape = anAttrSelection->value();
     if (aCurShape.get() == NULL)
       aCurShape = anAttrSelection->context()->shape();
     if (aCurShape.get() != NULL)
+    {
       aShapes.push_back(aCurShape);
+      aContexts.push_back(anAttrSelection->context());
+    }
   }
 
   // Store compound if we have more than one shape.
@@ -193,14 +233,14 @@ void ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName,
   if (aFormatName == "BREP") {
     aResult = BREPExport(theFileName, aFormatName, aShape, anError);
   } else if (aFormatName == "STEP") {
-    aResult = STEPExport(theFileName, aFormatName, aShape, anError);
+    aResult = STEPExport(theFileName, aShapes, aContexts, anError);
   } else if (aFormatName.substr(0, 4) == "IGES") {
     aResult = IGESExport(theFileName, aFormatName, aShape, anError);
   } else {
     anError = "Unsupported format: " + aFormatName;
   }
 
-  if (!anError.empty()) {
+  if (!aResult || !anError.empty()) {
     setError("An error occurred while exporting " + theFileName + ": " + anError);
     return;
   }
@@ -227,94 +267,48 @@ static std::string valToString(const ModelAPI_AttributeTables::Value& theVal,
   return aStr.str();
 }
 
-/// Returns true if something in selection is presented in the results list
-static bool isInResults(AttributeSelectionListPtr theSelection,
-                        const std::list<ResultPtr>& theResults,
-                        std::set<ResultPtr>& theCashedResults)
+void ExchangePlugin_ExportFeature::exportSTL(const std::string& theFileName)
 {
-  // collect all results into a cashed set
-  if (theCashedResults.empty()) {
-    std::list<ResultPtr> aResults;
-    std::list<ResultPtr>::const_iterator aRes = theResults.cbegin();
-    for(; aRes != theResults.cend(); aRes++) {
-      if (theCashedResults.count(*aRes))
-        continue;
-      else
-        theCashedResults.insert(*aRes);
-      if ((*aRes)->groupName() == ModelAPI_ResultBody::group()) {
-        ResultBodyPtr aResBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aRes);
-        std::list<ResultPtr> aResults;
-        ModelAPI_Tools::allSubs(aResBody, aResults, false);
-        for(std::list<ResultPtr>::iterator aR = aResults.begin(); aR != aResults.end(); aR++) {
-          theCashedResults.insert(std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aR));
-        }
-      } else if ((*aRes)->groupName() == ModelAPI_ResultPart::group()) { // all results of the part
-        ResultPartPtr aResPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(*aRes);
-        DocumentPtr aPartDoc = aResPart->partDoc();
-        if (!aPartDoc.get() || !aPartDoc->isOpened()) { // document is not accessible
-          return false;
-        }
-        int aBodyCount = aPartDoc->size(ModelAPI_ResultBody::group());
-        for (int aBodyIndex = 0; aBodyIndex < aBodyCount; ++aBodyIndex) {
-          ResultBodyPtr aResBody =
-            std::dynamic_pointer_cast<ModelAPI_ResultBody>(
-              aPartDoc->object(ModelAPI_ResultBody::group(), aBodyIndex));
-          if (aResBody.get()) {
-            theCashedResults.insert(aResBody);
-            std::list<ResultPtr> aResults;
-            ModelAPI_Tools::allSubs(aResBody, aResults, false);
-            for(std::list<ResultPtr>::iterator aR = aResults.begin(); aR != aResults.end(); aR++) {
-              theCashedResults.insert(std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aR));
-            }
-          }
-        }
-      }
-    }
+  // Get shape.
+  AttributeSelectionPtr aSelection = selection(STL_OBJECT_SELECTED());
+  GeomShapePtr aShape = aSelection->value();
+  if (!aShape.get()) {
+    aShape = aSelection->context()->shape();
   }
-  // if context is in results, return true
-  for(int a = 0; a < theSelection->size(); a++) {
-    AttributeSelectionPtr anAttr = theSelection->value(a);
-    ResultPtr aContext = anAttr->context();
-    // check is it group selected for groups BOP
-    if (aContext.get() && aContext->groupName() == ModelAPI_ResultGroup::group()) {
-      // it is impossible by used results check which result is used in this group result,
-      // so check the results shapes is it in results of this document or not
-      FeaturePtr aSelFeature =
-        std::dynamic_pointer_cast<ModelAPI_Feature>(theSelection->owner());
-      if (!aSelFeature.get() || aSelFeature->results().empty())
-        continue;
-      GeomShapePtr aGroupResShape = aSelFeature->firstResult()->shape();
-
-      std::set<ResultPtr>::iterator allResultsIter = theCashedResults.begin();
-      for(; allResultsIter != theCashedResults.end(); allResultsIter++) {
-        GeomShapePtr aResultShape = (*allResultsIter)->shape();
-
-        GeomAPI_Shape::ShapeType aType =
-          GeomAPI_Shape::shapeTypeByStr(theSelection->selectionType());
-        GeomAPI_ShapeExplorer aGroupResExp(aGroupResShape, aType);
-        for(; aGroupResExp.more(); aGroupResExp.next()) {
-          if (aResultShape->isSubShape(aGroupResExp.current(), false))
-            return true; // at least one shape of the group is in the used results
-        }
-      }
-    }
-    ResultBodyPtr aSelected = std::dynamic_pointer_cast<ModelAPI_ResultBody>(anAttr->context());
-    if (!aSelected.get()) { // try to get selected feature and all its results
-      FeaturePtr aContextFeature = anAttr->contextFeature();
-      if (aContextFeature.get() && !aContextFeature->results().empty()) {
-        const std::list<ResultPtr>& allResluts = aContextFeature->results();
-        std::list<ResultPtr>::const_iterator aResIter = allResluts.cbegin();
-        for(; aResIter != allResluts.cend(); aResIter++) {
-          if (aResIter->get() && theCashedResults.count(*aResIter))
-            return true;
-        }
-      }
-    } else if (aSelected.get() && theCashedResults.count(aSelected))
-      return true;
+
+  // Get relative value and percent flag.
+  double aValue;
+  bool anIsRelative = false;
+  bool anIsASCII = false;
+
+  if (string(STL_DEFLECTION_TYPE())->value() == STL_DEFLECTION_TYPE_RELATIVE()) {
+    aValue = real(STL_RELATIVE())->value();
+    anIsRelative = true;
+  } else {
+    aValue = real(STL_ABSOLUTE())->value();
+  }
+
+  if (string(STL_FILE_TYPE())->value() == STL_FILE_TYPE_ASCII()) {
+    anIsASCII = true;
+  }
+  // Perform the export
+  std::string anError;
+  bool aResult = false;
+
+  aResult = STLExport(theFileName,
+                      aShape,
+                      aValue,
+                      anIsRelative,
+                      anIsASCII,
+                      anError);
+
+  if (!aResult || !anError.empty()) {
+    setError("An error occurred while exporting " + theFileName + ": " + anError);
+    return;
   }
-  return false;
 }
 
+
 void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
 {
   try {
@@ -333,14 +327,20 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
   std::list<DocumentPtr> aDocuments; /// documents of Parts selected and used in export
   std::map<DocumentPtr, GeomTrsfPtr> aDocTrsf; /// translation of the part
 
+  bool anExCludedIsImage = false;
   AttributeSelectionListPtr aSelection = selectionList(XAO_SELECTION_LIST_ID());
   bool aIsSelection = aSelection->isInitialized() && aSelection->size() > 0;
   if (aIsSelection) { // a mode for export to geom result by result
-    for(int a = 0; a < aSelection->size(); a++) {
+    for (int a = 0; a < aSelection->size(); a++) {
       AttributeSelectionPtr anAttr = aSelection->value(a);
       ResultPtr aBodyContext =
         std::dynamic_pointer_cast<ModelAPI_Result>(anAttr->context());
       if (aBodyContext.get() && !aBodyContext->isDisabled() && aBodyContext->shape().get()) {
+          /// do not export pictures
+          if (aBodyContext->hasTexture()){
+            anExCludedIsImage = true;
+            continue;
+          }
         aResults.push_back(aBodyContext);
         GeomShapePtr aShape = anAttr->value();
         if (!aShape.get())
@@ -373,7 +373,8 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
     }
   }
   if (aShapes.empty()) {
-    setError("No shapes to export");
+    if(!anExCludedIsImage)
+      setError("No shapes to export");
     return;
   }
 
@@ -394,7 +395,7 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
   if (aGeometryName.empty() && aResults.size() == 1) {
     // get the name from the first result
     ResultPtr aResultBody = *aResults.begin();
-    aGeometryName = aResultBody->data()->name();
+    aGeometryName = Locale::Convert::toString(aResultBody->data()->name());
   }
 
   aXao.getGeometry()->setName(aGeometryName);
@@ -418,7 +419,9 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
 
       AttributeSelectionListPtr aSelectionList =
           aGroupFeature->selectionList("group_list");
-      if (!isInResults(aSelectionList, aResults, allResultsCashed))// skip group not used in result
+      if (!ModelAPI_Tools::isInResults(aSelectionList,
+                                       aResults,
+                                       allResultsCashed))// skip group not used in result
         continue;
 
       // conversion of dimension
@@ -429,7 +432,7 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
       XAO::Dimension aGroupDimension = XAO::XaoUtils::stringToDimension(aDimensionString);
 
       XAO::Group* aXaoGroup = aXao.addGroup(aGroupDimension,
-                                            aResultGroup->data()->name());
+        Locale::Convert::toString(aResultGroup->data()->name()));
 
       try {
         GeomAPI_ShapeExplorer aGroupResExplorer(aResultGroup->shape(), aSelType);
@@ -448,7 +451,7 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
       } catch (XAO::XAO_Exception& e) {
         // LCOV_EXCL_START
         std::string msg = "An error occurred while exporting group " +
-          aResultGroup->data()->name();
+          Locale::Convert::toString(aResultGroup->data()->name());
         msg += ".\n";
         msg += e.what();
         msg += "\n";
@@ -472,7 +475,8 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
       std::string aSelectionType = aSelectionList->selectionType();
       bool isWholePart = aSelectionType == "part";
       // skip field not used in results
-      if (!isWholePart && !isInResults(aSelectionList, aResults, allResultsCashed))
+      if (!isWholePart &&
+          !ModelAPI_Tools::isInResults(aSelectionList, aResults, allResultsCashed))
         continue;
 
       // conversion of dimension
@@ -485,7 +489,7 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
       XAO::Type aFieldType = XAO::XaoUtils::stringToFieldType(aTypeString);
 
       XAO::Field* aXaoField = aXao.addField(aFieldType, aFieldDimension, aTables->columns(),
-                                            aResultField->data()->name());
+        Locale::Convert::toString(aResultField->data()->name()));
 
 
       try {
@@ -511,8 +515,8 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
               int anElementID = 0;
               if (!isWholePart) {
                 // element index actually is the ID of the selection
-                AttributeSelectionPtr aSelection = aSelectionList->value(aRow - 1);
-                int aReferenceID = GeomAlgoAPI_CompoundBuilder::id(aShape, aSelection->value());
+                AttributeSelectionPtr aSel = aSelectionList->value(aRow - 1);
+                int aReferenceID = GeomAlgoAPI_CompoundBuilder::id(aShape, aSel->value());
                 if (aReferenceID == 0) // selected value does not found in the exported shape
                   continue;
 
@@ -546,7 +550,7 @@ void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
       } catch (XAO::XAO_Exception& e) {
         // LCOV_EXCL_START
         std::string msg = "An error occurred while exporting field " +
-          aResultField->data()->name();
+          Locale::Convert::toString(aResultField->data()->name());
         msg += ".\n";
         msg += e.what();
         msg += "\n";