src/SMESH/SMESH_DriverShape.cxx

   1 // Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 //  File   : SMESH_DriverShape.cxx
  24 //  Author : Yoann AUDOUIN, EDF
  25 //  Module : SMESH
  26 //
  27
  28 #include <utilities.h>
  29 #include "SMESH_DriverShape.hxx"
  30
  31 // step include
  32 #include <STEPControl_Reader.hxx>
  33 #include <STEPControl_Writer.hxx>
  34 #include <Interface_Static.hxx>
  35
  36 // Brep include
  37 #include <BRepTools.hxx>
  38 #include <BRep_Builder.hxx>
  39
  40 //Occ include
  41 #include <TopoDS.hxx>
  42
  43 #include <boost/filesystem.hpp>
  44 #include <boost/algorithm/string.hpp>
  45 namespace fs = boost::filesystem;
  46
  47 /**
  48  * @brief Import the content of a shape file (STEP) into a TopDS_Shape object
  49  *
  50  * @param shape_file the shape file
  51  * @param aShape the object
  52  *
  53  * @return error code
  54  */
  55 int importSTEPShape(const std::string shape_file, TopoDS_Shape& aShape){
  56
  57   MESSAGE("Importing STEP shape from " << shape_file);
  58   STEPControl_Reader reader;
  59   // Forcing Unit in meter
  60   Interface_Static::SetCVal("xstep.cascade.unit","M");
  61   Interface_Static::SetIVal("read.step.ideas", 1);
  62   Interface_Static::SetIVal("read.step.nonmanifold", 1);
  63   IFSelect_ReturnStatus aStat = reader.ReadFile(shape_file.c_str());
  64   if(aStat != IFSelect_RetDone){
  65     std::cerr << "Reading error for "  << shape_file << std::endl;
  66     return true;
  67   }
  68
  69   int NbTrans = reader.TransferRoots();
  70   // There should be only one shape within the file
  71   assert(NbTrans==1);
  72   aShape = reader.OneShape();
  73
  74   return false;
  75 }
  76
  77 /**
  78  * @brief Export the content of a TopoDS_Shape into a shape file (STEP)
  79  *
  80  * @param shape_file the shape file
  81  * @param aShape the object
  82  *
  83  * @return error code
  84  */
  85 int exportSTEPShape(const std::string shape_file, const TopoDS_Shape& aShape){
  86
  87   MESSAGE("Exporting STEP shape to " << shape_file);
  88
  89   STEPControl_Writer aWriter;
  90   // Forcing Unit in meter
  91   Interface_Static::SetCVal("xstep.cascade.unit","M");
  92   Interface_Static::SetCVal("write.step.unit","M");
  93   Interface_Static::SetIVal("write.step.nonmanifold", 1);
  94
  95   IFSelect_ReturnStatus aStat = aWriter.Transfer(aShape,STEPControl_AsIs);
  96   if(aStat != IFSelect_RetDone){
  97     std::cerr << "Transfer error for "  << shape_file << std::endl;
  98     return true;
  99   }
 100
 101   aStat = aWriter.Write(shape_file.c_str());
 102
 103   if(aStat != IFSelect_RetDone){
 104     std::cerr << "Writing error for "  << shape_file << std::endl;
 105     return true;
 106   }
 107   return aStat;
 108 }
 109
 110 /**
 111  * @brief Import the content of a shape file (BREP) into a TopDS_Shape object
 112  *
 113  * @param shape_file the shape file
 114  * @param aShape the object
 115  *
 116  * @return error code
 117  */
 118 int importBREPShape(const std::string shape_file, TopoDS_Shape& aShape){
 119
 120   MESSAGE("Importing BREP shape from " << shape_file);
 121   BRep_Builder builder;
 122   BRepTools::Read(aShape, shape_file.c_str(), builder);
 123
 124   return false;
 125 }
 126
 127 /**
 128  * @brief Export the content of a TopoDS_Shape into a shape file (BREP)
 129  *
 130  * @param shape_file the shape file
 131  * @param aShape the object
 132  *
 133  * @return error code
 134  */
 135 int exportBREPShape(const std::string shape_file, const TopoDS_Shape& aShape){
 136
 137   MESSAGE("Exporting BREP shape to " << shape_file);
 138   BRepTools::Write(aShape, shape_file.c_str());
 139
 140   return false;
 141 }
 142
 143 /**
 144  * @brief Import the content of a shape file into a TopDS_Shape object
 145  *
 146  * @param shape_file the shape file
 147  * @param aShape the object
 148  *
 149  * @return error code
 150  */
 151 int importShape(const std::string shape_file, TopoDS_Shape& aShape){
 152   std::string type = fs::path(shape_file).extension().string();
 153   boost::algorithm::to_lower(type);
 154   if (type == ".brep"){
 155     return importBREPShape(shape_file, aShape);
 156   } else if (type == ".step"){
 157     return importSTEPShape(shape_file, aShape);
 158   } else {
 159     std::cerr << "Unknow format: " << type << std::endl;
 160     return true;
 161   }
 162 }
 163
 164 /**
 165  * @brief Import the content of a shape file into a TopDS_Shape object
 166  *
 167  * @param shape_file the shape file
 168  * @param aShape the object
 169  *
 170  * @return error code
 171  */
 172 int exportShape(const std::string shape_file, const TopoDS_Shape& aShape){
 173   std::string type = fs::path(shape_file).extension().string();
 174   boost::algorithm::to_lower(type);
 175   if (type == ".brep"){
 176     return exportBREPShape(shape_file, aShape);
 177   } else if (type == ".step"){
 178     return exportSTEPShape(shape_file, aShape);
 179   } else {
 180     std::cerr << "Unknow format: " << type << std::endl;
 181     return true;
 182   }
 183 }