X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FExchangePlugin%2FExchangePlugin_ImportFeature.cpp;h=70bab6e93944316af75e56853536b65dbce28718;hb=6ffc58e695559e8997bbc8b977b537f1c63d45a8;hp=6493870df6ea6a6260f552f10576471c83837246;hpb=4783f146b71a48c651523fcf0e12367bcf3d1fa8;p=modules%2Fshaper.git diff --git a/src/ExchangePlugin/ExchangePlugin_ImportFeature.cpp b/src/ExchangePlugin/ExchangePlugin_ImportFeature.cpp index 6493870df..70bab6e93 100644 --- a/src/ExchangePlugin/ExchangePlugin_ImportFeature.cpp +++ b/src/ExchangePlugin/ExchangePlugin_ImportFeature.cpp @@ -1,42 +1,42 @@ -/* - * ExchangePlugin_ImportFeature.cpp - * - * Created on: Aug 28, 2014 - * Author: sbh - */ +// Copyright (C) 2014-20xx CEA/DEN, EDF R&D + +// File: ExchangePlugin_ImportFeature.cpp +// Created: Aug 28, 2014 +// Authors: Sergey BELASH, Sergey POKHODENKO #include -#include +#include +#include + #include -#include +#include + +#include +#include +#include +#include +#include + +#include + +#include +#include #include +#include #include #include +#include #include #include -#include -#include -#include -#include +#include +#include +#include -#include -#include -#ifdef _DEBUG -#include -#include -#endif - -#ifdef WIN32 -# define _separator_ '\\' -#else -# define _separator_ '/' -#endif - -typedef TopoDS_Shape (*importFunctionPointer)(const TCollection_AsciiString&, - const TCollection_AsciiString&, - TCollection_AsciiString&, - const TDF_Label&); +#include +#include + +#include ExchangePlugin_ImportFeature::ExchangePlugin_ImportFeature() { @@ -47,20 +47,20 @@ ExchangePlugin_ImportFeature::~ExchangePlugin_ImportFeature() // TODO Auto-generated destructor stub } -/* - * Returns the unique kind of a feature - */ -const std::string& ExchangePlugin_ImportFeature::getKind() -{ - return ExchangePlugin_ImportFeature::ID(); -} - /* * Request for initialization of data model of the feature: adding all attributes */ void ExchangePlugin_ImportFeature::initAttributes() { - data()->addAttribute(ExchangePlugin_ImportFeature::FILE_PATH_ID(), ModelAPI_AttributeString::type()); + data()->addAttribute(ExchangePlugin_ImportFeature::FILE_PATH_ID(), + ModelAPI_AttributeString::typeId()); + AttributePtr aFeaturesAttribute = + data()->addAttribute(ExchangePlugin_ImportFeature::FEATURES_ID(), + ModelAPI_AttributeRefList::typeId()); + aFeaturesAttribute->setIsArgument(false); + + ModelAPI_Session::get()->validators()->registerNotObligatory( + getKind(), ExchangePlugin_ImportFeature::FEATURES_ID()); } /* @@ -68,108 +68,209 @@ void ExchangePlugin_ImportFeature::initAttributes() */ void ExchangePlugin_ImportFeature::execute() { - AttributeStringPtr aFilePathAttr = std::dynamic_pointer_cast( - data()->attribute(ExchangePlugin_ImportFeature::FILE_PATH_ID())); + AttributeStringPtr aFilePathAttr = string(ExchangePlugin_ImportFeature::FILE_PATH_ID()); std::string aFilePath = aFilePathAttr->value(); - if(aFilePath.empty()) + if (aFilePath.empty()) { + setError("File path is empty."); return; + } + importFile(aFilePath); } -bool ExchangePlugin_ImportFeature::importFile(const std::string& theFileName) +std::shared_ptr ExchangePlugin_ImportFeature::createResultBody( + std::shared_ptr aGeomShape) { - // 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(); - - // Load plugin library and get the "Import" method - LibHandle anImportLib = loadImportPlugin(std::string(aFormatName.ToCString())); - if(!anImportLib) - return false; - importFunctionPointer fp = (importFunctionPointer) GetProc(anImportLib, "Import"); - // Perform the import - TCollection_AsciiString anError; - TDF_Label anUnknownLabel = TDF_Label(); - TopoDS_Shape aShape = fp(aFileName, - aFormatName, - anError, - anUnknownLabel); - // Check if shape is valid - if ( aShape.IsNull() ) { - std::string aShapeError = "An error occurred while importing " + theFileName + ": "; - aShapeError = aShapeError + std::string(anError.ToCString()); - Events_Error::send(aShapeError, this); - #ifdef _DEBUG - std::cerr << aShapeError << std::endl; - #endif - return false; - } - // - // Pass the results into the model - std::string anObjectName = aPath.Name().ToCString(); - data()->setName(anObjectName); - std::shared_ptr aResultBody = document()->createBody(data()); - std::shared_ptr aGeomShape(new GeomAPI_Shape); - aGeomShape->setImpl(new TopoDS_Shape(aShape)); - - //LoadNamingDS of the imported shape - loadNamingDS(aGeomShape, aResultBody); - - setResult(aResultBody); - - return true; + std::shared_ptr aResultBody = document()->createBody(data()); + //LoadNamingDS of the imported shape + loadNamingDS(aGeomShape, aResultBody); + return aResultBody; } -//============================================================================ -void ExchangePlugin_ImportFeature::loadNamingDS( - std::shared_ptr theGeomShape, - std::shared_ptr theResultBody) -{ - //load result - theResultBody->store(theGeomShape); - int aTag(1); - theResultBody->loadFirstLevel(theGeomShape, aTag); - theResultBody->loadDisconnectedEdges(theGeomShape, aTag); - theResultBody->loadDisconnectedVertexes(theGeomShape, aTag); +void ExchangePlugin_ImportFeature::importFile(const std::string& theFileName) +{ + // "*.brep" -> "BREP" + std::string anExtension = GeomAlgoAPI_Tools::File_Tools::extension(theFileName); + + if (anExtension == "XAO") { + importXAO(theFileName); + return; + } + + // Perform the import + std::string anError; + std::shared_ptr aGeomShape; + if (anExtension == "BREP" || anExtension == "BRP") { + aGeomShape = BREPImport(theFileName, anExtension, anError); + } else if (anExtension == "STEP" || anExtension == "STP") { + aGeomShape = STEPImport(theFileName, anExtension, anError); + } else if (anExtension == "IGES" || anExtension == "IGS") { + aGeomShape = IGESImport(theFileName, anExtension, anError); + } else { + anError = "Unsupported format: " + anExtension; + } + + // Check if shape is valid + if (!anError.empty()) { + setError("An error occurred while importing " + theFileName + ": " + anError); + return; + } + + // Pass the results into the model + std::string anObjectName = GeomAlgoAPI_Tools::File_Tools::name(theFileName); + data()->setName(anObjectName); + + setResult(createResultBody(aGeomShape)); } -LibHandle ExchangePlugin_ImportFeature::loadImportPlugin(const std::string& theFormatName) +void ExchangePlugin_ImportFeature::importXAO(const std::string& theFileName) { - std::string aLibName = library(theFormatName + ID()); - LibHandle anImportLib = LoadLib(aLibName.c_str()); - std::string anImportError = "Failed to load " + aLibName + ": "; - if(!anImportLib) { -#ifdef WIN32 - LPVOID lpMsgBuf; - ::FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | - FORMAT_MESSAGE_FROM_SYSTEM | - FORMAT_MESSAGE_IGNORE_INSERTS, - 0, ::GetLastError(), 0, (LPTSTR) &lpMsgBuf, 0, 0); - anImportError = anImportError + std::string((char*) lpMsgBuf); - ::LocalFree(lpMsgBuf); -#else - anImportError = anImportError + std::string(dlerror()); -#endif - Events_Error::send(anImportError, this); -#ifdef _DEBUG - std::cerr << anImportError << std::endl; -#endif - return false; + try { + std::string anError; + + XAO::Xao aXao; + std::shared_ptr aGeomShape = XAOImport(theFileName, anError, &aXao); + + if (!anError.empty()) { + setError("An error occurred while importing " + theFileName + ": " + anError); + return; } - // Test loaded plugin for existence of valid "Import" function: - importFunctionPointer fp = (importFunctionPointer) GetProc(anImportLib, "Import"); - if (!fp) { - std::string aFunctionError = "No valid \"Import\" function was found in the " + aLibName; - Events_Error::send(aFunctionError, this); -#ifdef _DEBUG - std::cerr << aFunctionError << std::endl; -#endif - UnLoadLib(anImportLib) - return NULL; + + XAO::Geometry* aXaoGeometry = aXao.getGeometry(); + + // use the geometry name or the file name for the feature + std::string aBodyName = aXaoGeometry->getName(); + if (aBodyName.empty()) + aBodyName = GeomAlgoAPI_Tools::File_Tools::name(theFileName); + data()->setName(aBodyName); + + ResultBodyPtr aResultBody = createResultBody(aGeomShape); + setResult(aResultBody); + + // Process groups + std::shared_ptr aRefListOfGroups = + std::dynamic_pointer_cast(data()->attribute(FEATURES_ID())); + + // Remove previous groups stored in RefList + std::list anGroupList = aRefListOfGroups->list(); + std::list::iterator anGroupIt = anGroupList.begin(); + for (; anGroupIt != anGroupList.end(); ++anGroupIt) { + std::shared_ptr aFeature = ModelAPI_Feature::feature(*anGroupIt); + if (aFeature) + document()->removeFeature(aFeature); + } + + // Create new groups + for (int aGroupIndex = 0; aGroupIndex < aXao.countGroups(); ++aGroupIndex) { + XAO::Group* aXaoGroup = aXao.getGroup(aGroupIndex); + + std::shared_ptr aGroupFeature = addFeature("Group"); + + // group name + if (!aXaoGroup->getName().empty()) + aGroupFeature->data()->setName(aXaoGroup->getName()); + + // fill selection + AttributeSelectionListPtr aSelectionList = aGroupFeature->selectionList("group_list"); + + // conversion of dimension + XAO::Dimension aGroupDimension = aXaoGroup->getDimension(); + std::string aDimensionString = XAO::XaoUtils::dimensionToString(aXaoGroup->getDimension()); + std::string aSelectionType = ExchangePlugin_Tools::xaoDimension2selectionType(aDimensionString); + + aSelectionList->setSelectionType(aSelectionType); + for (int anElementIndex = 0; anElementIndex < aXaoGroup->count(); ++anElementIndex) { + aSelectionList->append(aResultBody, GeomShapePtr()); + // complex conversion of element index to reference id + int anElementID = aXaoGroup->get(anElementIndex); + std::string aReferenceString = + aXaoGeometry->getElementReference(aXaoGroup->getDimension(), anElementID); + int aReferenceID = XAO::XaoUtils::stringToInt(aReferenceString); + + aSelectionList->value(anElementIndex)->setId(aReferenceID); + } + } + // Top avoid problems in Object Browser update: issue #1647. + ModelAPI_EventCreator::get()->sendReordered( + std::dynamic_pointer_cast(aRefListOfGroups->owner())); + + } catch (XAO::XAO_Exception& e) { + std::string anError = e.what(); + setError("An error occurred while importing " + theFileName + ": " + anError); + return; } - return anImportLib; +} + +//============================================================================ +std::shared_ptr ExchangePlugin_ImportFeature::addFeature( + std::string theID) +{ + std::shared_ptr aNew = document()->addFeature(theID, false); + if (aNew) + data()->reflist(FEATURES_ID())->append(aNew); + // set as current also after it becomes sub to set correctly enabled for other subs + //document()->setCurrentFeature(aNew, false); + return aNew; +} + +void ExchangePlugin_ImportFeature::removeFeature( + std::shared_ptr theFeature) +{ + if (!data()->isValid()) + return; + AttributeRefListPtr aList = reflist(FEATURES_ID()); + aList->remove(theFeature); +} + +int ExchangePlugin_ImportFeature::numberOfSubs(bool forTree) const +{ + return data()->reflist(FEATURES_ID())->size(true); +} + +std::shared_ptr ExchangePlugin_ImportFeature::subFeature( + const int theIndex, bool forTree) +{ + ObjectPtr anObj = data()->reflist(FEATURES_ID())->object(theIndex, false); + FeaturePtr aRes = std::dynamic_pointer_cast(anObj); + return aRes; +} + +int ExchangePlugin_ImportFeature::subFeatureId(const int theIndex) const +{ + std::shared_ptr aRefList = std::dynamic_pointer_cast< + ModelAPI_AttributeRefList>(data()->attribute(FEATURES_ID())); + std::list aFeatures = aRefList->list(); + std::list::const_iterator anIt = aFeatures.begin(); + int aResultIndex = 1; // number of the counted (created) features, started from 1 + int aFeatureIndex = -1; // number of the not-empty features in the list + for (; anIt != aFeatures.end(); anIt++) { + if (anIt->get()) + aFeatureIndex++; + if (aFeatureIndex == theIndex) + break; + aResultIndex++; + } + return aResultIndex; +} + +bool ExchangePlugin_ImportFeature::isSub(ObjectPtr theObject) const +{ + // check is this feature of result + FeaturePtr aFeature = ModelAPI_Feature::feature(theObject); + if (aFeature) + return data()->reflist(FEATURES_ID())->isInList(aFeature); + return false; +} + +//============================================================================ +void ExchangePlugin_ImportFeature::loadNamingDS( + std::shared_ptr theGeomShape, + std::shared_ptr theResultBody) +{ + //load result + theResultBody->store(theGeomShape); + + int aTag(1); + std::string aNameMS = "Shape"; + theResultBody->loadFirstLevel(theGeomShape, aNameMS, aTag); }