// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
-/*
- * ExchangePlugin_ExportFeature.cpp
- *
- * Created on: May 14, 2015
- * Author: spo
- */
+// File: ExchangePlugin_ExportFeature.cpp
+// Created: May 14, 2015
+// Author: Sergey POKHODENKO
#include <ExchangePlugin_ExportFeature.h>
-#include <GeomAlgoAPI_BREPExport.h>
-#include <GeomAlgoAPI_STEPExport.h>
-#include <GeomAlgoAPI_IGESExport.h>
+
+#include <algorithm>
+#include <iterator>
+#include <string>
+#ifdef _DEBUG
+#include <iostream>
+#include <ostream>
+#endif
#include <Config_Common.h>
#include <Config_PropManager.h>
+#include <GeomAlgoAPI_BREPExport.h>
#include <GeomAlgoAPI_CompoundBuilder.h>
+#include <GeomAlgoAPI_IGESExport.h>
+#include <GeomAlgoAPI_STEPExport.h>
+#include <GeomAlgoAPI_Tools.h>
+#include <GeomAlgoAPI_XAOExport.h>
#include <GeomAPI_Shape.h>
-#include <ModelAPI_AttributeString.h>
#include <ModelAPI_AttributeSelectionList.h>
+#include <ModelAPI_AttributeString.h>
#include <ModelAPI_Data.h>
#include <ModelAPI_Document.h>
#include <ModelAPI_Object.h>
#include <ModelAPI_ResultBody.h>
-#include <TCollection_AsciiString.hxx>
-#include <TDF_Label.hxx>
-#include <TopoDS_Shape.hxx>
-#include <OSD_Path.hxx>
+#include <ModelAPI_ResultGroup.h>
+#include <ModelAPI_Session.h>
+#include <ModelAPI_Validator.h>
-#include <algorithm>
-#include <string>
-#ifdef _DEBUG
-#include <iostream>
-#include <ostream>
-#endif
+#include <XAO_Group.hxx>
+#include <XAO_Xao.hxx>
+
+#include <ExchangePlugin_Tools.h>
ExchangePlugin_ExportFeature::ExchangePlugin_ExportFeature()
{
// TODO Auto-generated destructor stub
}
-/*
- * Returns the unique kind of a feature
- */
-const std::string& ExchangePlugin_ExportFeature::getKind()
-{
- return ExchangePlugin_ExportFeature::ID();
-}
-
/*
* Request for initialization of data model of the feature: adding all attributes
*/
void ExchangePlugin_ExportFeature::initAttributes()
{
+ data()->addAttribute(ExchangePlugin_ExportFeature::EXPORT_TYPE_ID(), ModelAPI_AttributeString::typeId());
data()->addAttribute(ExchangePlugin_ExportFeature::FILE_PATH_ID(), ModelAPI_AttributeString::typeId());
+ data()->addAttribute(ExchangePlugin_ExportFeature::XAO_FILE_PATH_ID(), ModelAPI_AttributeString::typeId());
+ data()->addAttribute(ExchangePlugin_ExportFeature::FILE_FORMAT_ID(), ModelAPI_AttributeString::typeId());
data()->addAttribute(ExchangePlugin_ExportFeature::SELECTION_LIST_ID(), ModelAPI_AttributeSelectionList::typeId());
+ data()->addAttribute(ExchangePlugin_ExportFeature::XAO_AUTHOR_ID(), ModelAPI_AttributeString::typeId());
+ data()->addAttribute(ExchangePlugin_ExportFeature::XAO_GEOMETRY_NAME_ID(), ModelAPI_AttributeString::typeId());
+
+ //ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), ExchangePlugin_ExportFeature::SELECTION_LIST_ID());
+ ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), ExchangePlugin_ExportFeature::XAO_FILE_PATH_ID());
+ ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), ExchangePlugin_ExportFeature::XAO_AUTHOR_ID());
+ ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), ExchangePlugin_ExportFeature::XAO_GEOMETRY_NAME_ID());
+}
+
+void ExchangePlugin_ExportFeature::attributeChanged(const std::string& theID)
+{
+ if (theID == XAO_FILE_PATH_ID()) {
+ string(ExchangePlugin_ExportFeature::FILE_PATH_ID())->setValue(
+ string(ExchangePlugin_ExportFeature::XAO_FILE_PATH_ID())->value());
+ }
}
/*
*/
void ExchangePlugin_ExportFeature::execute()
{
- AttributeStringPtr aFilePathAttr = std::dynamic_pointer_cast<ModelAPI_AttributeString>(
- data()->attribute(ExchangePlugin_ExportFeature::FILE_PATH_ID()));
+ AttributeStringPtr aFormatAttr =
+ this->string(ExchangePlugin_ExportFeature::FILE_FORMAT_ID());
+ std::string aFormat = aFormatAttr->value();
+
+ AttributeStringPtr aFilePathAttr =
+ this->string(ExchangePlugin_ExportFeature::FILE_PATH_ID());
std::string aFilePath = aFilePathAttr->value();
if (aFilePath.empty())
return;
+ exportFile(aFilePath, aFormat);
+}
+
+void ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName,
+ const std::string& theFormat)
+{
+ std::string aFormatName = theFormat;
+
+ if (aFormatName.empty()) { // get default format for the extension
+ // ".brep" -> "BREP"
+ std::string anExtension = GeomAlgoAPI_Tools::File_Tools::extension(theFileName);
+ if (anExtension == "BREP" || anExtension == "BRP") {
+ aFormatName = "BREP";
+ } else if (anExtension == "STEP" || anExtension == "STP") {
+ aFormatName = "STEP";
+ } else if (anExtension == "IGES" || anExtension == "IGS") {
+ aFormatName = "IGES-5.1";
+ } else if (anExtension == "XAO") {
+ aFormatName = "XAO";
+ } else {
+ aFormatName = anExtension;
+ }
+ }
+
+ if (aFormatName == "XAO") {
+ exportXAO(theFileName);
+ return;
+ }
+
+ // make shape for export from selected shapes
AttributeSelectionListPtr aSelectionListAttr =
- std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(
- data()->attribute(ExchangePlugin_ExportFeature::SELECTION_LIST_ID()));
+ this->selectionList(ExchangePlugin_ExportFeature::SELECTION_LIST_ID());
+ std::list<GeomShapePtr> aShapes;
+ for (int i = 0, aSize = aSelectionListAttr->size(); i < aSize; ++i) {
+ AttributeSelectionPtr anAttrSelection = aSelectionListAttr->value(i);
+ std::shared_ptr<GeomAPI_Shape> aCurShape = anAttrSelection->value();
+ if (aCurShape.get() == NULL)
+ aCurShape = anAttrSelection->context()->shape();
+ if (aCurShape.get() != NULL)
+ aShapes.push_back(aCurShape);
+ }
+
+ // 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::list<std::shared_ptr<GeomAPI_Shape> > aShapes;
- for ( int i = 0, aSize = aSelectionListAttr->size(); i < aSize; ++i ) {
- std::shared_ptr<ModelAPI_AttributeSelection> anSelectionAttr = aSelectionListAttr->value(i);
- aShapes.push_back(anSelectionAttr->value());
+ // Perform the export
+ std::string anError;
+ bool aResult = false;
+ if (aFormatName == "BREP") {
+ aResult = BREPExport(theFileName, aFormatName, aShape, anError);
+ } else if (aFormatName == "STEP") {
+ aResult = STEPExport(theFileName, aFormatName, aShape, anError);
+ } else if (aFormatName.substr(0, 4) == "IGES") {
+ aResult = IGESExport(theFileName, aFormatName, aShape, anError);
+ } else {
+ anError = "Unsupported format: " + aFormatName;
}
- std::shared_ptr<GeomAPI_Shape> aShape =
- GeomAlgoAPI_CompoundBuilder::compound(aShapes);
- exportFile(aFilePath, aShape);
+ if (!anError.empty()) {
+ setError("An error occurred while exporting " + theFileName + ": " + anError);
+ return;
+ }
}
-bool ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName,
- std::shared_ptr<GeomAPI_Shape> theShape)
+void ExchangePlugin_ExportFeature::exportXAO(const std::string& theFileName)
{
- // retrieve the file and plugin library names
- TCollection_AsciiString aFileName(theFileName.c_str());
- OSD_Path aPath(aFileName);
- TCollection_AsciiString aFormatName = aPath.Extension();
- // ".brep" -> "BREP", TCollection_AsciiString are numbered from 1
- aFormatName = aFormatName.SubString(2, aFormatName.Length());
- aFormatName.UpperCase();
+ try {
+
+ std::string anError;
+ XAO::Xao aXao;
+
+ // author
+
+ std::string anAuthor = string(ExchangePlugin_ExportFeature::XAO_AUTHOR_ID())->value();
+ aXao.setAuthor(anAuthor);
+
+ // make shape for export from all results
+ std::list<GeomShapePtr> aShapes;
+ int aBodyCount = document()->size(ModelAPI_ResultBody::group());
+ for (int aBodyIndex = 0; aBodyIndex < aBodyCount; ++aBodyIndex) {
+ ResultBodyPtr aResultBody =
+ std::dynamic_pointer_cast<ModelAPI_ResultBody>(
+ document()->object(ModelAPI_ResultBody::group(), aBodyIndex));
+ if (!aResultBody.get())
+ continue;
+ aShapes.push_back(aResultBody->shape());
+ }
+ GeomShapePtr aShape = (aShapes.size() == 1)
+ ? *aShapes.begin()
+ : GeomAlgoAPI_CompoundBuilder::compound(aShapes);
- // Perform the export
- TCollection_AsciiString anError;
- TDF_Label anUnknownLabel = TDF_Label();
+ SetShapeToXAO(aShape, &aXao, anError);
- TopoDS_Shape aShape(theShape->impl<TopoDS_Shape>());
- bool aResult = true;
- if (aFormatName == "BREP") {
- aResult = BREPExport::Export(aFileName, aFormatName, aShape, anError, anUnknownLabel);
- } else if (aFormatName == "STEP" || aFormatName == "STP") {
- aResult = STEPExport::Export(aFileName, aFormatName, aShape, anError, anUnknownLabel);
- } else if (aFormatName == "IGES") {
- aResult = IGESExport::Export(aFileName, aFormatName, aShape, anError, anUnknownLabel);
+ if (!anError.empty()) {
+ setError("An error occurred while exporting " + theFileName + ": " + anError);
+ return;
}
- if ( !aResult ) {
- const static std::string aShapeError =
- "An error occurred while exporting " + theFileName + ": " + anError.ToCString();
- setError(aShapeError);
- return false;
+ // geometry name
+
+ std::string aGeometryName = string(ExchangePlugin_ExportFeature::XAO_GEOMETRY_NAME_ID())->value();
+ aXao.getGeometry()->setName(aGeometryName);
+
+ // 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");
+
+ // 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());
+
+ for (int aSelectionIndex = 0; aSelectionIndex < aSelectionList->size(); ++aSelectionIndex) {
+ AttributeSelectionPtr aSelection = aSelectionList->value(aSelectionIndex);
+
+ // complex conversion of reference id to element index
+ int aReferenceID = aSelection->Id();
+ std::string aReferenceString = XAO::XaoUtils::intToString(aReferenceID);
+ int anElementID = aXao.getGeometry()->getElementIndexByReference(aGroupDimension, aReferenceString);
+
+ aXaoGroup->add(anElementID);
+ }
+ }
+
+ // exporting
+
+ XAOExport(theFileName, &aXao, anError);
+
+ if (!anError.empty()) {
+ setError("An error occurred while exporting " + theFileName + ": " + anError);
+ return;
}
- return true;
+ } catch (XAO::XAO_Exception& e) {
+ std::string anError = e.what();
+ setError("An error occurred while importing " + theFileName + ": " + anError);
+ return;
+ }
}
