#include <GeomAlgoAPI_XAOExport.h>
#include <GeomAPI_Shape.h>
+#include <GeomAPI_ShapeExplorer.h>
#include <ModelAPI_AttributeSelectionList.h>
#include <ModelAPI_AttributeString.h>
#include <ModelAPI_Document.h>
#include <ModelAPI_Object.h>
#include <ModelAPI_ResultBody.h>
+#include <ModelAPI_ResultPart.h>
#include <ModelAPI_ResultGroup.h>
#include <ModelAPI_ResultField.h>
#include <ModelAPI_Session.h>
aFormatName = "STEP";
} else if (anExtension == "IGES" || anExtension == "IGS") {
aFormatName = "IGES-5.1";
- } else if (anExtension == "XAO") {
- aFormatName = "XAO";
} else {
aFormatName = anExtension;
}
}
// Store compound if we have more than one shape.
- std::shared_ptr<GeomAPI_Shape> aShape;
- if(aShapes.size() == 1) {
- aShape = aShapes.front();
- } else {
- aShape = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
- }
+ std::shared_ptr<GeomAPI_Shape> aShape =
+ aShapes.size() == 1 ? aShapes.front() : GeomAlgoAPI_CompoundBuilder::compound(aShapes);
// Perform the export
std::string anError;
/// Returns true if something in selection is presented in the results list
static bool isInResults(AttributeSelectionListPtr theSelection,
- const std::list<ResultBodyPtr>& theResults,
- std::set<ResultBodyPtr>& theCashedResults)
+ const std::list<ResultPtr>& theResults,
+ std::set<ResultPtr>& theCashedResults)
{
// collect all results into a cashed set
if (theCashedResults.empty()) {
std::list<ResultPtr> aResults;
- std::list<ResultBodyPtr>::const_iterator aRes = theResults.cbegin();
+ std::list<ResultPtr>::const_iterator aRes = theResults.cbegin();
for(; aRes != theResults.cend(); aRes++) {
if (theCashedResults.count(*aRes))
continue;
else
theCashedResults.insert(*aRes);
- std::list<ResultPtr> aResults;
- ModelAPI_Tools::allSubs(*aRes, aResults, false);
- for(std::list<ResultPtr>::iterator aR = aResults.begin(); aR != aResults.end(); aR++) {
- theCashedResults.insert(std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aR));
+ 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);
- ResultBodyPtr aSelected= std::dynamic_pointer_cast<ModelAPI_ResultBody>(anAttr->context());
- if (aSelected.get() && theCashedResults.count(aSelected))
+ 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;
// make shape for export from all results
std::list<GeomShapePtr> aShapes;
- std::list<ResultBodyPtr> aResults;
+ std::list<ResultPtr> aResults;
+ std::list<DocumentPtr> aDocuments; /// documents of Parts selected and used in export
AttributeSelectionListPtr aSelection = selectionList(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++) {
AttributeSelectionPtr anAttr = aSelection->value(a);
- ResultBodyPtr aBodyContext =
- std::dynamic_pointer_cast<ModelAPI_ResultBody>(anAttr->context());
+ ResultPtr aBodyContext =
+ std::dynamic_pointer_cast<ModelAPI_Result>(anAttr->context());
if (aBodyContext.get() && !aBodyContext->isDisabled() && aBodyContext->shape().get()) {
aResults.push_back(aBodyContext);
GeomShapePtr aShape = anAttr->value();
if (!aShape.get())
aShape = aBodyContext->shape();
aShapes.push_back(aShape);
+ if (aBodyContext->groupName() == ModelAPI_ResultPart::group()) {
+ ResultPartPtr aResPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aBodyContext);
+ DocumentPtr aPartDoc = aResPart->partDoc();
+ if (!aPartDoc.get() || !aPartDoc->isOpened()) { // document is not accessible
+ std::string msg = "Can not export XAO for not loaded part";
+ Events_InfoMessage("ExportFeature", msg, this).send();
+ return;
+ } else {
+ aDocuments.push_back(aPartDoc);
+ }
+ }
}
}
} else {
std::string aGeometryName = string(ExchangePlugin_ExportFeature::XAO_GEOMETRY_NAME_ID())->value();
if (aGeometryName.empty() && aResults.size() == 1) {
// get the name from the first result
- ResultBodyPtr aResultBody = *aResults.begin();
+ ResultPtr aResultBody = *aResults.begin();
aGeometryName = aResultBody->data()->name();
}
aXao.getGeometry()->setName(aGeometryName);
- std::set<ResultBodyPtr> allResultsCashed; // cash to speed up searching in all results selected
-
- // groups
- int aGroupCount = document()->size(ModelAPI_ResultGroup::group());
- for (int aGroupIndex = 0; aGroupIndex < aGroupCount; ++aGroupIndex) {
- ResultGroupPtr aResultGroup =
- std::dynamic_pointer_cast<ModelAPI_ResultGroup>(
- document()->object(ModelAPI_ResultGroup::group(), aGroupIndex));
-
- FeaturePtr aGroupFeature = document()->feature(aResultGroup);
-
- AttributeSelectionListPtr aSelectionList =
- aGroupFeature->selectionList("group_list");
-
- if (!isInResults(aSelectionList, aResults, allResultsCashed))// skip group not used in results
- continue;
-
- // conversion of dimension
- std::string aSelectionType = aSelectionList->selectionType();
- std::string aDimensionString =
- ExchangePlugin_Tools::selectionType2xaoDimension(aSelectionType);
- XAO::Dimension aGroupDimension = XAO::XaoUtils::stringToDimension(aDimensionString);
-
- XAO::Group* aXaoGroup = aXao.addGroup(aGroupDimension,
- aResultGroup->data()->name());
-
- try {
- for (int aSelectionIndex = 0; aSelectionIndex < aSelectionList->size(); ++aSelectionIndex) {
- AttributeSelectionPtr aSelection = aSelectionList->value(aSelectionIndex);
-
- // complex conversion of reference id to element index
- // gives bad id in case the selection is done from python script
- // => using GeomAlgoAPI_CompoundBuilder::id instead
- // int aReferenceID_old = aSelection->Id();
-
- int aReferenceID = GeomAlgoAPI_CompoundBuilder::id(aShape, aSelection->value());
+ std::set<ResultPtr> allResultsCashed; // cash to speed up searching in all results selected
+
+ // iterate all documents used
+ if (aDocuments.empty())
+ aDocuments.push_back(document());
+ std::list<DocumentPtr>::iterator aDoc = aDocuments.begin();
+ for(; aDoc != aDocuments.end(); aDoc++) {
+ // groups
+ int aGroupCount = (*aDoc)->size(ModelAPI_ResultGroup::group());
+ for (int aGroupIndex = 0; aGroupIndex < aGroupCount; ++aGroupIndex) {
+ ResultGroupPtr aResultGroup = std::dynamic_pointer_cast<ModelAPI_ResultGroup>(
+ (*aDoc)->object(ModelAPI_ResultGroup::group(), aGroupIndex));
+ if (!aResultGroup.get() || !aResultGroup->shape().get())
+ continue;
- if (aReferenceID == 0) // selected value does not found in the exported shape
- continue;
+ FeaturePtr aGroupFeature = (*aDoc)->feature(aResultGroup);
- std::string aReferenceString = XAO::XaoUtils::intToString(aReferenceID);
- int anElementID =
- aXao.getGeometry()->getElementIndexByReference(aGroupDimension, aReferenceString);
+ AttributeSelectionListPtr aSelectionList =
+ aGroupFeature->selectionList("group_list");
+ if (!isInResults(aSelectionList, aResults, allResultsCashed))// skip group not used in result
+ continue;
- aXaoGroup->add(anElementID);
+ // conversion of dimension
+ std::string aSelectionType = aSelectionList->selectionType();
+ GeomAPI_Shape::ShapeType aSelType = GeomAPI_Shape::shapeTypeByStr(aSelectionType);
+ std::string aDimensionString =
+ ExchangePlugin_Tools::selectionType2xaoDimension(aSelectionType);
+ XAO::Dimension aGroupDimension = XAO::XaoUtils::stringToDimension(aDimensionString);
+
+ XAO::Group* aXaoGroup = aXao.addGroup(aGroupDimension,
+ aResultGroup->data()->name());
+
+ try {
+ GeomAPI_ShapeExplorer aGroupResExplorer(aResultGroup->shape(), aSelType);
+ for(; aGroupResExplorer.more(); aGroupResExplorer.next()) {
+ int aReferenceID = GeomAlgoAPI_CompoundBuilder::id(aShape, aGroupResExplorer.current());
+ if (aReferenceID == 0) // selected value does not found in the exported shape
+ continue;
+ std::string aReferenceString = XAO::XaoUtils::intToString(aReferenceID);
+ int anElementID =
+ aXao.getGeometry()->getElementIndexByReference(aGroupDimension, aReferenceString);
+ aXaoGroup->add(anElementID);
+ }
+ } catch (XAO::XAO_Exception& e) {
+ // LCOV_EXCL_START
+ std::string msg = "An error occurred while exporting group " +
+ aResultGroup->data()->name();
+ msg += ".\n";
+ msg += e.what();
+ msg += "\n";
+ msg += "=> skipping this group from XAO export.";
+ Events_InfoMessage("ExportFeature", msg, this).send();
+ aXao.removeGroup(aXaoGroup);
+ // LCOV_EXCL_STOP
}
- } catch (XAO::XAO_Exception& e) {
- // LCOV_EXCL_START
- std::string msg = "An error occurred while exporting group " + aResultGroup->data()->name();
- msg += ".\n";
- msg += e.what();
- msg += "\n";
- msg += "=> skipping this group from XAO export.";
- Events_InfoMessage("ExportFeature", msg, this).send();
- aXao.removeGroup(aXaoGroup);
- // LCOV_EXCL_STOP
}
- }
- // fields
- int aFieldCount = document()->size(ModelAPI_ResultField::group());
- for (int aFieldIndex = 0; aFieldIndex < aFieldCount; ++aFieldIndex) {
- ResultFieldPtr aResultField =
- std::dynamic_pointer_cast<ModelAPI_ResultField>(
- document()->object(ModelAPI_ResultField::group(), aFieldIndex));
-
- FeaturePtr aFieldFeature = document()->feature(aResultField);
-
- AttributeSelectionListPtr aSelectionList =
- aFieldFeature->selectionList("selected");
- std::string aSelectionType = aSelectionList->selectionType();
- bool isWholePart = aSelectionType == "part";
-
- if (!isWholePart &&
- !isInResults(aSelectionList, aResults, allResultsCashed))// skip field not used in results
- continue;
-
- // conversion of dimension
- std::string aDimensionString =
- ExchangePlugin_Tools::selectionType2xaoDimension(aSelectionType);
- XAO::Dimension aFieldDimension = XAO::XaoUtils::stringToDimension(aDimensionString);
- // get tables and their type
- std::shared_ptr<ModelAPI_AttributeTables> aTables = aFieldFeature->tables("values");
- std::string aTypeString = ExchangePlugin_Tools::valuesType2xaoType(aTables->type());
- XAO::Type aFieldType = XAO::XaoUtils::stringToFieldType(aTypeString);
-
- XAO::Field* aXaoField = aXao.addField(aFieldType, aFieldDimension, aTables->columns(),
- aResultField->data()->name());
-
-
- try {
- // set components names
- AttributeStringArrayPtr aComponents = aFieldFeature->stringArray("components_names");
- for(int aComp = 0; aComp < aComponents->size(); aComp++) {
- std::string aName = aComponents->value(aComp);
- aXaoField->setComponentName(aComp, aName);
- }
+ // fields
+ int aFieldCount = (*aDoc)->size(ModelAPI_ResultField::group());
+ for (int aFieldIndex = 0; aFieldIndex < aFieldCount; ++aFieldIndex) {
+ ResultFieldPtr aResultField = std::dynamic_pointer_cast<ModelAPI_ResultField>(
+ (*aDoc)->object(ModelAPI_ResultField::group(), aFieldIndex));
- AttributeIntArrayPtr aStamps = aFieldFeature->intArray("stamps");
- for (int aStepIndex = 0; aStepIndex < aTables->tables(); aStepIndex++) {
- XAO::Step* aStep = aXaoField->addNewStep(aStepIndex);
- aStep->setStep(aStepIndex);
- int aStampIndex = aStamps->value(aStepIndex);
- aStep->setStamp(aStampIndex);
- int aNumElements = isWholePart ? aXaoField->countElements() : aTables->rows();
- int aNumComps = aTables->columns();
- std::set<int> aFilledIDs; // to fill the rest by defaults
- // omit default values first row
- for(int aRow = isWholePart ? 0 : 1; aRow < aNumElements; aRow++) {
- for(int aCol = 0; aCol < aNumComps; aCol++) {
- 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());
- if (aReferenceID == 0) // selected value does not found in the exported shape
- continue;
+ FeaturePtr aFieldFeature = (*aDoc)->feature(aResultField);
- std::string aReferenceString = XAO::XaoUtils::intToString(aReferenceID);
- anElementID =
- aXao.getGeometry()->getElementIndexByReference(aFieldDimension, aReferenceString);
- }
+ AttributeSelectionListPtr aSelectionList =
+ aFieldFeature->selectionList("selected");
+ std::string aSelectionType = aSelectionList->selectionType();
+ bool isWholePart = aSelectionType == "part";
+ // skip field not used in results
+ if (!isWholePart && !isInResults(aSelectionList, aResults, allResultsCashed))
+ continue;
- ModelAPI_AttributeTables::Value aVal = aTables->value(
- isWholePart ? 0 : aRow, aCol, aStepIndex);
- std::string aStrVal = valToString(aVal, aTables->type());
- aStep->setStringValue(isWholePart ? aRow : anElementID, aCol, aStrVal);
- aFilledIDs.insert(anElementID);
- }
+ // conversion of dimension
+ std::string aDimensionString =
+ ExchangePlugin_Tools::selectionType2xaoDimension(aSelectionType);
+ XAO::Dimension aFieldDimension = XAO::XaoUtils::stringToDimension(aDimensionString);
+ // get tables and their type
+ std::shared_ptr<ModelAPI_AttributeTables> aTables = aFieldFeature->tables("values");
+ std::string aTypeString = ExchangePlugin_Tools::valuesType2xaoType(aTables->type());
+ XAO::Type aFieldType = XAO::XaoUtils::stringToFieldType(aTypeString);
+
+ XAO::Field* aXaoField = aXao.addField(aFieldType, aFieldDimension, aTables->columns(),
+ aResultField->data()->name());
+
+
+ try {
+ // set components names
+ AttributeStringArrayPtr aComponents = aFieldFeature->stringArray("components_names");
+ for(int aComp = 0; aComp < aComponents->size(); aComp++) {
+ std::string aName = aComponents->value(aComp);
+ aXaoField->setComponentName(aComp, aName);
}
- if (!isWholePart) { // fill the rest values by default ones
- XAO::GeometricElementList::iterator allElem = aXao.getGeometry()->begin(aFieldDimension);
- for(; allElem != aXao.getGeometry()->end(aFieldDimension); allElem++) {
- if (aFilledIDs.find(allElem->first) != aFilledIDs.end())
- continue;
+
+ AttributeIntArrayPtr aStamps = aFieldFeature->intArray("stamps");
+ for (int aStepIndex = 0; aStepIndex < aTables->tables(); aStepIndex++) {
+ XAO::Step* aStep = aXaoField->addNewStep(aStepIndex + 1);
+ aStep->setStep(aStepIndex + 1);
+ int aStampIndex = aStamps->value(aStepIndex);
+ aStep->setStamp(aStampIndex);
+ int aNumElements = isWholePart ? aXaoField->countElements() : aTables->rows();
+ int aNumComps = aTables->columns();
+ std::set<int> aFilledIDs; // to fill the rest by defaults
+ // omit default values first row
+ for(int aRow = isWholePart ? 0 : 1; aRow < aNumElements; aRow++) {
for(int aCol = 0; aCol < aNumComps; aCol++) {
- ModelAPI_AttributeTables::Value aVal = aTables->value(0, aCol, aStepIndex); // default
+ 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());
+ if (aReferenceID == 0) // selected value does not found in the exported shape
+ continue;
+
+ std::string aReferenceString = XAO::XaoUtils::intToString(aReferenceID);
+ anElementID = aXao.getGeometry()->
+ getElementIndexByReference(aFieldDimension, aReferenceString);
+ }
+
+ ModelAPI_AttributeTables::Value aVal = aTables->value(
+ isWholePart ? 0 : aRow, aCol, aStepIndex);
std::string aStrVal = valToString(aVal, aTables->type());
- aStep->setStringValue(allElem->first, aCol, aStrVal);
+ aStep->setStringValue(isWholePart ? aRow : anElementID, aCol, aStrVal);
+ aFilledIDs.insert(anElementID);
+ }
+ }
+ if (!isWholePart) { // fill the rest values by default ones
+ XAO::GeometricElementList::iterator allElem =
+ aXao.getGeometry()->begin(aFieldDimension);
+ for(; allElem != aXao.getGeometry()->end(aFieldDimension); allElem++) {
+ if (aFilledIDs.find(allElem->first) != aFilledIDs.end())
+ continue;
+ for(int aCol = 0; aCol < aNumComps; aCol++) {
+ // default value
+ ModelAPI_AttributeTables::Value aVal = aTables->value(0, aCol, aStepIndex);
+ std::string aStrVal = valToString(aVal, aTables->type());
+ aStep->setStringValue(allElem->first, aCol, aStrVal);
+ }
}
}
}
+ } catch (XAO::XAO_Exception& e) {
+ // LCOV_EXCL_START
+ std::string msg = "An error occurred while exporting field " +
+ aResultField->data()->name();
+ msg += ".\n";
+ msg += e.what();
+ msg += "\n";
+ msg += "=> skipping this field from XAO export.";
+ Events_InfoMessage("ExportFeature", msg, this).send();
+ aXao.removeField(aXaoField);
+ // LCOV_EXCL_STOP
}
- } catch (XAO::XAO_Exception& e) {
- // LCOV_EXCL_START
- std::string msg = "An error occurred while exporting field " + aResultField->data()->name();
- msg += ".\n";
- msg += e.what();
- msg += "\n";
- msg += "=> skipping this field from XAO export.";
- Events_InfoMessage("ExportFeature", msg, this).send();
- aXao.removeField(aXaoField);
- // LCOV_EXCL_STOP
}
}
}
// LCOV_EXCL_STOP
}
+
+bool ExchangePlugin_ExportFeature::isMacro() const
+{
+ if (!data().get() || !data()->isValid())
+ return false;
+ ExchangePlugin_ExportFeature* aThis = ((ExchangePlugin_ExportFeature*)(this));
+ AttributeStringPtr aFormatAttr = aThis->string(FILE_FORMAT_ID());
+ std::string aFormat(aFormatAttr.get() ? aFormatAttr->value() : "");
+
+ if (aFormat.empty()) { // get default format for the extension
+ AttributeStringPtr aFilePathAttr = aThis->string(FILE_PATH_ID());
+ std::string aFilePath = aFilePathAttr->value();
+ if (!aFilePath.empty()) {
+ std::string anExtension = GeomAlgoAPI_Tools::File_Tools::extension(aFilePath);
+ aFormat = anExtension;
+ }
+ }
+
+ if (aFormat == "XAO") { // on export to GEOm the selection attribute is filled - this is
+ // an exceptional case where export to XAO feature must be kept
+ AttributeSelectionListPtr aList = aThis->selectionList(SELECTION_LIST_ID());
+ return !aList->isInitialized() || aList->size() == 0;
+ }
+ return true;
+}
