Salome HOME
bos #29482 Export of colors and names to STEP.
[modules/shaper.git] / src / ExchangePlugin / ExchangePlugin_ExportFeature.cpp
index 1da829a6f1e2ea5f6a5524b55268cf7c357ff7d8..54400c68680dcd2c1f5242bf3d192cabc3b04435 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2014-2021  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
@@ -27,7 +27,6 @@
 #include <ostream>
 #endif
 
-
 #include <Config_Common.h>
 #include <Config_PropManager.h>
 
@@ -153,7 +152,6 @@ void ExchangePlugin_ExportFeature::attributeChanged(const std::string& theID)
     string(ExchangePlugin_ExportFeature::FILE_PATH_ID())->setValue(
       string(ExchangePlugin_ExportFeature::STL_FILE_PATH_ID())->value());
   }
-
 }
 
 /*
@@ -205,20 +203,24 @@ void ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName,
   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->hasTexture())
+    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.
@@ -231,7 +233,7 @@ 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 {
@@ -265,93 +267,6 @@ 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)
-{
-  // collect all results into a cashed set
-  if (theCashedResults.empty()) {
-    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));
-            }
-          }
-        }
-      }
-    }
-  }
-  // 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;
-  }
-  return false;
-}
-
 void ExchangePlugin_ExportFeature::exportSTL(const std::string& theFileName)
 {
   // Get shape.
@@ -504,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
@@ -558,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