src/IGESPlugin/IGESPlugin_IOperations.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 "IGESPlugin_IOperations.hxx"
  22 #include "IGESPlugin_ExportDriver.hxx"
  23 #include "IGESPlugin_ImportDriver.hxx"
  24 #include "IGESPlugin_IExport.hxx"
  25 #include "IGESPlugin_IImport.hxx"
  26
  27 // KERNEL includes
  28 #include <utilities.h>
  29
  30 // GEOM includes
  31 #include "GEOM_PythonDump.hxx"
  32 #include "GEOMImpl_Types.hxx"
  33
  34 #include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic : see Lucien PIGNOLONI / OCC
  35
  36 //=============================================================================
  37 /*!
  38  *  Constructor
  39  */
  40 //=============================================================================
  41 IGESPlugin_IOperations::IGESPlugin_IOperations( GEOM_Engine* theEngine )
  42 : GEOMImpl_IBaseIEOperations( theEngine )
  43 {
  44   MESSAGE( "IGESPlugin_IOperations::IGESPlugin_IOperations" );
  45 }
  46
  47 //=============================================================================
  48 /*!
  49  *  Destructor
  50  */
  51 //=============================================================================
  52 IGESPlugin_IOperations::~IGESPlugin_IOperations()
  53 {
  54   MESSAGE( "IGESPlugin_IOperations::~IGESPlugin_IOperations" );
  55 }
  56
  57 //=============================================================================
  58 /*!
  59  *  ExportIGES
  60  *  Export a shape to IGES format
  61  *  \param theOriginal The shape to export
  62  *  \param theFileName The name of the file to exported
  63  *  \param theIsASCII The format of the exported file (ASCII or Binary)
  64  *  \param theDeflection The deflection of the shape to exported
  65  *  \param theVersion The version of IGES format which defines, whether to write
  66  *                    only faces (5.1 IGES format) or shells and solids also (5.3 IGES format).
  67  */
  68 //=============================================================================
  69 void IGESPlugin_IOperations::ExportIGES( const Handle(GEOM_Object)      theOriginal,
  70                                                  const TCollection_AsciiString& theFileName,
  71                                                  const TCollection_AsciiString& theVersion )
  72 {
  73   SetErrorCode(KO);
  74   if( theOriginal.IsNull() ) return;
  75
  76   Handle(GEOM_Function) aRefFunction = theOriginal->GetLastFunction();
  77   if( aRefFunction.IsNull() ) return;  //There is no function which creates an object to be exported
  78
  79   //Add a new result object
  80   Handle(GEOM_Object) result = GetEngine()->AddObject( GEOM_IMPORT);
  81
  82   //Add an Export function
  83   Handle(GEOM_Function) aFunction = result->AddFunction( IGESPlugin_ExportDriver::GetID(), EXPORT_SHAPE );
  84   if( aFunction.IsNull() ) return;
  85
  86   //Check if the function is set correctly
  87   if( aFunction->GetDriverGUID() != IGESPlugin_ExportDriver::GetID() ) return;
  88
  89   //Set parameters
  90   IGESPlugin_IExport aCI( aFunction );
  91   aCI.SetOriginal( aRefFunction );
  92   aCI.SetFileName( theFileName );
  93   aCI.SetVersion( theVersion );
  94
  95   //Perform the Export
  96   try {
  97     OCC_CATCH_SIGNALS;
  98     if( !GetSolver()->ComputeFunction( aFunction ) ) {
  99       SetErrorCode( "Not enough space on disk, or you haven't permissions to write this directory" );
 100       return;
 101     }
 102   }
 103   catch( Standard_Failure& aFail ) {
 104     SetErrorCode( aFail.GetMessageString() );
 105     return;
 106   }
 107
 108   //Make a Python command
 109   GEOM::TPythonDump(aFunction) << "geompy.ExportIGES(" << theOriginal << ", \""
 110     << theFileName.ToCString() << "\", \"" << theVersion.ToCString() << "\" )";
 111
 112   SetErrorCode(OK);
 113 }
 114
 115 //=============================================================================
 116 /*!
 117  *  ImportIGES
 118  *  Import a shape from IGES format
 119  *  \param theFileName The name of the file to import
 120  *  \return List of GEOM_Objects, containing the created shape and propagation groups.
 121  */
 122 //=============================================================================
 123 Handle(TColStd_HSequenceOfTransient)
 124 IGESPlugin_IOperations::ImportIGES( const TCollection_AsciiString& theFileName,
 125                                             const bool theIsIgnoreUnits )
 126 {
 127   SetErrorCode(KO);
 128   if( theFileName.IsEmpty() ) return NULL;
 129
 130   //Add a new result object
 131   Handle(GEOM_Object) anImported = GetEngine()->AddObject( GEOM_IMPORT );
 132
 133   //Add an Import function
 134   Handle(GEOM_Function) aFunction =
 135     anImported->AddFunction( IGESPlugin_ImportDriver::GetID(), IMPORT_SHAPE);
 136   if (aFunction.IsNull()) return NULL;
 137
 138   //Check if the function is set correctly
 139   if (aFunction->GetDriverGUID() != IGESPlugin_ImportDriver::GetID()) return NULL;
 140
 141   //Set parameters
 142   IGESPlugin_IImport aCI( aFunction );
 143   aCI.SetFileName( theFileName );
 144   aCI.SetIsIgnoreUnits( theIsIgnoreUnits );
 145
 146   //Perform the Import
 147   Handle(TColStd_HSequenceOfTransient) aSeq = new TColStd_HSequenceOfTransient;
 148
 149   try {
 150     OCC_CATCH_SIGNALS;
 151     if( !GetSolver()->ComputeFunction( aFunction ) ) {
 152       SetErrorCode( "Import driver failed" );
 153       return NULL;
 154     }
 155     aSeq->Append(anImported);
 156
 157     // Create material groups.
 158     // MakeMaterialGroups( anImported, aSeq );
 159   }
 160   catch( Standard_Failure& aFail ) {
 161     SetErrorCode( aFail.GetMessageString() );
 162     return NULL;
 163   }
 164
 165   //Make a Python command
 166   GEOM::TPythonDump pd (aFunction);
 167   if( theIsIgnoreUnits )
 168     pd << aSeq << " = geompy.ImportIGES(\"" << theFileName.ToCString() << "\", True)";
 169   else
 170     pd << aSeq << " = geompy.ImportIGES(\"" << theFileName.ToCString() << "\")";
 171   SetErrorCode(OK);
 172
 173   return aSeq;
 174 }
 175
 176 //=============================================================================
 177 /*!
 178  *  ReadValue
 179  */
 180 //=============================================================================
 181 TCollection_AsciiString IGESPlugin_IOperations::ReadValue
 182                                  ( const TCollection_AsciiString& theFileName,
 183                                    const TCollection_AsciiString& theParameterName )
 184 {
 185   SetErrorCode(KO);
 186
 187   TCollection_AsciiString aValue, anError;
 188
 189   if (theFileName.IsEmpty() || theParameterName.IsEmpty()) return aValue;
 190
 191   aValue = IGESPlugin_ImportDriver::GetValue( theFileName, theParameterName, anError );
 192
 193   if( aValue.IsEmpty() ) {
 194     if( anError.IsEmpty() )
 195       anError = theFileName + " doesn't contain requested parameter";
 196     return aValue;
 197   }
 198   if (anError.IsEmpty())
 199     SetErrorCode(OK);
 200   else
 201     SetErrorCode(anError.ToCString());
 202
 203   return aValue;
 204 }