Salome HOME
Merge branch 'master' into Dev_1.1.0
[modules/shaper.git] / src / ExchangePlugin / ExchangePlugin_ImportFeature.cpp
index 13a61e5469ce0c4e152e4415f9c09f38b525976d..f3f53a273e26fe136ecae73b051a9fdb9531c797 100644 (file)
@@ -1,3 +1,5 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
 /*
  * ExchangePlugin_ImportFeature.cpp
  *
@@ -6,10 +8,13 @@
  */
 
 #include <ExchangePlugin_ImportFeature.h>
+#include <GeomAlgoAPI_BREPImport.h>
+#include <GeomAlgoAPI_STEPImport.h>
 
 #include <GeomAPI_Shape.h>
 #include <Config_Common.h>
-#include <Events_Error.h>
+#include <Config_PropManager.h>
+
 #include <ModelAPI_AttributeString.h>
 #include <ModelAPI_Data.h>
 #include <ModelAPI_Document.h>
 #include <ostream>
 #endif
 
-#ifdef WIN32
-# define _separator_ '\\'
-#else
-# define _separator_ '/'
-#endif
-
-typedef TopoDS_Shape (*importFunctionPointer)(const TCollection_AsciiString&,
-                                              const TCollection_AsciiString&,
-                                              TCollection_AsciiString&,
-                                              const TDF_Label&);
-
 ExchangePlugin_ImportFeature::ExchangePlugin_ImportFeature()
 {
 }
@@ -60,7 +54,7 @@ const std::string& ExchangePlugin_ImportFeature::getKind()
  */
 void ExchangePlugin_ImportFeature::initAttributes()
 {
-  data()->addAttribute(ExchangePlugin_ImportFeature::FILE_PATH_ID(), ModelAPI_AttributeString::type());
+  data()->addAttribute(ExchangePlugin_ImportFeature::FILE_PATH_ID(), ModelAPI_AttributeString::typeId());
 }
 
 /*
@@ -68,7 +62,7 @@ void ExchangePlugin_ImportFeature::initAttributes()
  */
 void ExchangePlugin_ImportFeature::execute()
 {
-  AttributeStringPtr aFilePathAttr = boost::dynamic_pointer_cast<ModelAPI_AttributeString>(
+  AttributeStringPtr aFilePathAttr = std::dynamic_pointer_cast<ModelAPI_AttributeString>(
       data()->attribute(ExchangePlugin_ImportFeature::FILE_PATH_ID()));
   std::string aFilePath = aFilePathAttr->value();
   if(aFilePath.empty())
@@ -82,94 +76,56 @@ bool ExchangePlugin_ImportFeature::importFile(const std::string& theFileName)
   TCollection_AsciiString aFileName (theFileName.c_str());
   OSD_Path aPath(aFileName);
   TCollection_AsciiString aFormatName = aPath.Extension();
-  // ".brep" -> "BREP". TCollection_AsciiString are numbered from 1
+  // ".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);
+  // Perform the import
+  TCollection_AsciiString anError;
+  TDF_Label anUnknownLabel = TDF_Label();
+
+  TopoDS_Shape aShape;
+  if (aFormatName == "BREP") {
+    aShape = BREPImport::Import(aFileName, aFormatName, anError, anUnknownLabel);
+  } else if (aFormatName == "STEP") {
+    aShape = STEPImport::Import(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
+  if ( aShape.IsNull() ) {
+     const static std::string aShapeError = 
+       "An error occurred while importing " + theFileName + ": " + anError.ToCString();
+     setError(aShapeError);
      return false;
    }
-  //
-   // Pass the results into the model
-   std::string anObjectName = aPath.Name().ToCString();
-   data()->setName(anObjectName);
-   boost::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data());
-   boost::shared_ptr<GeomAPI_Shape> aGeomShape(new GeomAPI_Shape);
-   aGeomShape->setImpl(new TopoDS_Shape(aShape));
 
-   //LoadNamingDS of the imported shape
-   loadNamingDS(aGeomShape, aResultBody);
+  // 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));
 
-   setResult(aResultBody);
+  //LoadNamingDS of the imported shape
+  loadNamingDS(aGeomShape, aResultBody);
 
-   return true;
+  setResult(aResultBody);
+
+  return true;
 }
 
 //============================================================================
 void ExchangePlugin_ImportFeature::loadNamingDS(
-                                   boost::shared_ptr<GeomAPI_Shape> theGeomShape, 
-                                            boost::shared_ptr<ModelAPI_ResultBody> theResultBody)
+                                   std::shared_ptr<GeomAPI_Shape> theGeomShape, 
+                                            std::shared_ptr<ModelAPI_ResultBody> theResultBody)
 {  
   //load result
   theResultBody->store(theGeomShape);
   int aTag(1);
-  theResultBody->loadFirstLevel(theGeomShape, aTag);
-  theResultBody->loadDisconnectedEdges(theGeomShape, aTag);
-  theResultBody->loadDisconnectedVertexes(theGeomShape, aTag); 
-}
-
-LibHandle ExchangePlugin_ImportFeature::loadImportPlugin(const std::string& theFormatName)
-{
-  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;
-  }
-  // 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;
-  }
-  return anImportLib;
+  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); 
 }