// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
-/*
- * ExchangePlugin_ExportFeature.cpp
- *
- * Created on: Aug 28, 2014
- * Author: sbh
- */
+// 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 <GeomAPI_Shape.h>
+#include <ExchangePlugin_Tools.h>
+
+#include <GeomAlgoAPI_BREPExport.h>
+#include <GeomAlgoAPI_CompoundBuilder.h>
+#include <GeomAlgoAPI_IGESExport.h>
+#include <GeomAlgoAPI_STEPExport.h>
+#include <GeomAlgoAPI_Tools.h>
+
#include <Config_Common.h>
#include <Config_PropManager.h>
+#include <GeomAPI_Shape.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 <algorithm>
+#include <iterator>
#include <string>
#ifdef _DEBUG
#include <iostream>
void ExchangePlugin_ExportFeature::initAttributes()
{
data()->addAttribute(ExchangePlugin_ExportFeature::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());
}
/*
*/
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();
+ // Format may be empty. In this case look at extension.
+// if (aFormat.empty())
+// return;
+
+ AttributeStringPtr aFilePathAttr =
+ this->string(ExchangePlugin_ExportFeature::FILE_PATH_ID());
std::string aFilePath = aFilePathAttr->value();
- if(aFilePath.empty())
+ if (aFilePath.empty())
return;
- exportFile(aFilePath);
+
+ AttributeSelectionListPtr aSelectionListAttr =
+ this->selectionList(ExchangePlugin_ExportFeature::SELECTION_LIST_ID());
+ std::list<std::shared_ptr<GeomAPI_Shape> > 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);
+ }
+ std::shared_ptr<GeomAPI_Shape> aShape =
+ GeomAlgoAPI_CompoundBuilder::compound(aShapes);
+
+ exportFile(aFilePath, aFormat, aShape);
}
-bool ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName)
+bool ExchangePlugin_ExportFeature::exportFile(const std::string& theFileName,
+ const std::string& theFormat,
+ std::shared_ptr<GeomAPI_Shape> theShape)
{
// 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();
+ std::string aFormatName = theFormat;
+
+ if (theFormat.empty()) { // look at 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 {
+ aFormatName = anExtension;
+ }
+ }
// Perform the export
- TCollection_AsciiString anError;
- TDF_Label anUnknownLabel = TDF_Label();
-
- TopoDS_Shape aShape;
-// if (aFormatName == "BREP") {
-// aShape = BREPExport::Export(aFileName, aFormatName, anError, anUnknownLabel);
-// } else if (aFormatName == "STEP" || aFormatName == "STP") {
-// aShape = STEPExport::Export(aFileName, aFormatName, anError, anUnknownLabel);
-// } else if (aFormatName == "IGES") {
-// aShape = IGESExport::Export(aFileName, aFormatName, anError, anUnknownLabel);
-// }
-// // Check if shape is valid
-// if ( aShape.IsNull() ) {
-// const static std::string aShapeError =
-// "An error occurred while exporting " + theFileName + ": " + anError.ToCString();
-// setError(aShapeError);
-// return false;
-// }
-//
-// // Pass the results into the model
-// std::string anObjectName = aPath.Name().ToCString();
-// data()->setName(anObjectName);
-// std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data());
-// std::shared_ptr<GeomAPI_Shape> aGeomShape(new GeomAPI_Shape);
-// aGeomShape->setImpl(new TopoDS_Shape(aShape));
-//
-// //LoadNamingDS of the exported shape
-// loadNamingDS(aGeomShape, aResultBody);
-//
-// setResult(aResultBody);
+ std::string anError;
+ bool aResult = false;
+ if (aFormatName == "BREP") {
+ aResult = BREPExport(theFileName, aFormatName, theShape, anError);
+ } else if (aFormatName == "STEP") {
+ aResult = STEPExport(theFileName, aFormatName, theShape, anError);
+ } else if (aFormatName.substr(0, 4) == "IGES") {
+ aResult = IGESExport(theFileName, aFormatName, theShape, anError);
+ } else {
+ anError = "Unsupported format " + aFormatName;
+ }
+
+ if (!aResult) {
+ std::string aShapeError =
+ "An error occurred while exporting " + theFileName + ": " + anError;
+ setError(aShapeError);
+ return false;
+ }
return true;
}
-
-//============================================================================
-void ExchangePlugin_ExportFeature::loadNamingDS(
- std::shared_ptr<GeomAPI_Shape> theGeomShape,
- std::shared_ptr<ModelAPI_ResultBody> theResultBody)
-{
- //load result
- theResultBody->store(theGeomShape);
-
- int aTag(1);
- std::string aNameMS = "Shape";
- theResultBody->loadFirstLevel(theGeomShape, aNameMS, aTag);
- //std::string aNameDE = "DiscEdges";
- //theResultBody->loadDisconnectedEdges(theGeomShape, aNameDE, aTag);
- //std::string aNameDV = "DiscVertexes";
- //theResultBody->loadDisconnectedVertexes(theGeomShape, aNameDV, aTag);
-}