Fix regression in unit tests

[modules/shaper.git] / src / GeomAlgoAPI / GeomAlgoAPI_BREPExport.cpp

diff --git a/src/GeomAlgoAPI/GeomAlgoAPI_BREPExport.cpp b/src/GeomAlgoAPI/GeomAlgoAPI_BREPExport.cpp
index d85ae6b4f74de282aad96e9a93839936f1872a80..a9c4fffd2ca63d6727c7213e83297202760e6ff4 100644 (file)
--- a/src/GeomAlgoAPI/GeomAlgoAPI_BREPExport.cpp
+++ b/src/GeomAlgoAPI/GeomAlgoAPI_BREPExport.cpp
@@ -8,6 +8,8 @@
 
 #include "GeomAlgoAPI_Tools.h"
 
+#include <TopoDS_Shape.hxx>
+
 #include <BRepTools.hxx>
 #include <BRep_Builder.hxx>
 
@@ -18,17 +20,22 @@
 //=============================================================================
 bool BREPExport(const std::string& theFileName,
                 const std::string&,
-                const TopoDS_Shape& theShape,
+                const std::shared_ptr<GeomAPI_Shape>& theShape,
                 std::string& theError)
 {
   #ifdef _DEBUG
   std::cout << "Export BREP into file " << theFileName << std::endl;
   #endif
 
+  if (!theShape.get()) {
+    theError = "BREP Export failed: An invalid argument";
+    return false;
+  }
+
   // Set "C" numeric locale to save numbers correctly
   GeomAlgoAPI_Tools::Localizer loc;
 
-  if (!BRepTools::Write(theShape, theFileName.c_str())) {
+  if (!BRepTools::Write(theShape->impl<TopoDS_Shape>(), theFileName.c_str())) {
     theError = "BREP Export failed";
     return false;
   }