src/STLPlugin/STLPlugin_ExportDriver.cxx

   1 // Copyright (C) 2014-2016  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // internal includes
  21 #include "STLPlugin_ExportDriver.hxx"
  22 #include "STLPlugin_IExport.hxx"
  23
  24 // KERNEL includes
  25 #include <utilities.h>
  26 #include <Basics_Utils.hxx>
  27 #include <Basics_OCCTVersion.hxx>
  28
  29 // GEOM includes
  30 #include "GEOM_Function.hxx"
  31
  32 // OOCT includes
  33 #include <BRepBuilderAPI_Copy.hxx>
  34 #include <StlAPI_Writer.hxx>
  35 #include <TopoDS_Shape.hxx>
  36 #include <Bnd_Box.hxx>
  37 #include <BRepBndLib.hxx>
  38 #include <BRepTools.hxx>
  39 #include <BRepMesh_IncrementalMesh.hxx>
  40
  41 #define MAX2(X, Y)      ( Abs(X) > Abs(Y) ? Abs(X) : Abs(Y) )
  42 #define MAX3(X, Y, Z)   ( MAX2 ( MAX2(X, Y) , Z ) )
  43
  44 //=======================================================================
  45 //function : GetID
  46 //purpose  :
  47 //=======================================================================
  48 const Standard_GUID& STLPlugin_ExportDriver::GetID()
  49 {
  50   static Standard_GUID aGUID("88678a6a-885c-477c-b90b-51934572c61d");
  51   return aGUID;
  52 }
  53
  54 //=======================================================================
  55 //function : STLPlugin_ExportDriver
  56 //purpose  :
  57 //=======================================================================
  58 STLPlugin_ExportDriver::STLPlugin_ExportDriver()
  59 {
  60 }
  61
  62 //=======================================================================
  63 //function : Execute
  64 //purpose  :
  65 //=======================================================================
  66 Standard_Integer STLPlugin_ExportDriver::Execute(LOGBOOK& log) const
  67 {
  68   if (Label().IsNull()) return 0;
  69   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction( Label() );
  70
  71   STLPlugin_IExport aData (aFunction);
  72
  73   // retrieve the being exported shape
  74   TopoDS_Shape aShape;
  75   Handle(GEOM_Function) aRefFunction = aData.GetOriginal();
  76   if( aRefFunction.IsNull() ) return 0;
  77   aShape = aRefFunction->GetValue();
  78   if( aShape.IsNull() ) return 0;
  79   // set the result of function to be used by next operations
  80   aFunction->SetValue( aShape );
  81
  82   TCollection_AsciiString aFileName = aData.GetFileName();
  83   double aDeflection = aData.GetDeflection();
  84   bool anIsASCII = aData.GetIsASCII();
  85   bool anIsRelative = aData.GetIsRelative();
  86
  87   MESSAGE( "Export STL into file " << aFileName );
  88
  89   // Set "C" numeric locale to save numbers correctly
  90   Kernel_Utils::Localizer loc;
  91
  92   try
  93   {
  94     StlAPI_Writer aWriter;
  95     // copy source shape
  96     BRepBuilderAPI_Copy aCopy( aShape, Standard_False );
  97     TopoDS_Shape aCopyShape = aCopy.Shape();
  98     // ASCII mode
  99     aWriter.ASCIIMode() = anIsASCII;
 100 #if OCC_VERSION_LARGE > 0x06080000
 101     if ( anIsRelative ) {
 102       Standard_Real aXmin, aYmin, aZmin, aXmax, aYmax, aZmax;
 103       Bnd_Box bndBox;
 104       BRepBndLib::Add( aShape, bndBox );
 105       bndBox.Get( aXmin, aYmin, aZmin, aXmax, aYmax, aZmax );
 106       aDeflection = MAX3( aXmax-aXmin, aYmax-aYmin, aZmax-aZmin ) * aDeflection;
 107     }
 108     //Compute triangulation
 109     BRepTools::Clean( aCopyShape );
 110     BRepMesh_IncrementalMesh aMesh( aCopyShape, aDeflection );
 111 #else
 112     // set relative mode on false for using custom deflection coefficient
 113     aWriter.RelativeMode( ) = anIsRelative;
 114     if( anIsRelative )
 115       aWriter.SetCoefficient( aDeflection );
 116     else
 117       aWriter.SetDeflection( aDeflection );
 118 #endif
 119     aWriter.Write( aCopyShape, aFileName.ToCString() );
 120 #if OCC_VERSION_MAJOR < 7
 121     log.SetTouched(Label());
 122 #else
 123     log->SetTouched(Label());
 124 #endif
 125     return 1;
 126   }
 127   catch( Standard_Failure )
 128   {
 129     //THROW_SALOME_CORBA_EXCEPTION("Exception catched in ExportSTL", SALOME::BAD_PARAM);
 130   }
 131   return 0;
 132 }
 133
 134 //================================================================================
 135 /*!
 136  * \brief Returns a name of creation operation and names and values of creation parameters
 137  */
 138 //================================================================================
 139 bool STLPlugin_ExportDriver::
 140 GetCreationInformation( std::string&             theOperationName,
 141                         std::vector<GEOM_Param>& theParams )
 142 {
 143   return false;
 144 }
 145
 146 OCCT_IMPLEMENT_STANDARD_RTTIEXT( STLPlugin_ExportDriver,GEOM_BaseDriver );