]> SALOME platform Git repositories - modules/geom.git/blob - src/BREPPlugin/BREPPlugin_IOperations.cxx
Salome HOME
Prepare OCCT version 7.1.0+ for SALOME-8.2.0
[modules/geom.git] / src / BREPPlugin / BREPPlugin_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 "BREPPlugin_IOperations.hxx"
22 #include "BREPPlugin_ExportDriver.hxx"
23 #include "BREPPlugin_ImportDriver.hxx"
24 #include "BREPPlugin_IExport.hxx"
25 #include "BREPPlugin_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 BREPPlugin_IOperations::BREPPlugin_IOperations( GEOM_Engine* theEngine, int theDocID )
42 : GEOMImpl_IBaseIEOperations( theEngine, theDocID )
43 {
44   MESSAGE( "BREPPlugin_IOperations::BREPPlugin_IOperations" );
45 }
46
47 //=============================================================================
48 /*!
49  *  Destructor
50  */
51 //=============================================================================
52 BREPPlugin_IOperations::~BREPPlugin_IOperations()
53 {
54   MESSAGE( "BREPPlugin_IOperations::~BREPPlugin_IOperations" );
55 }
56
57 //=============================================================================
58 /*!
59  *  ExportBREP
60  *  Export a shape to BREP 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  */
66 //=============================================================================
67 void BREPPlugin_IOperations::ExportBREP( const Handle(GEOM_Object)      theOriginal,
68                                          const TCollection_AsciiString& theFileName )
69 {
70   SetErrorCode(KO);
71   if( theOriginal.IsNull() ) return;
72
73   Handle(GEOM_Function) aRefFunction = theOriginal->GetLastFunction();
74   if( aRefFunction.IsNull() ) return;  //There is no function which creates an object to be exported
75
76   //Add a new result object
77   Handle(GEOM_Object) result = GetEngine()->AddObject( GetDocID(), GEOM_IMPORT);
78
79   //Add an Export function
80   Handle(GEOM_Function) aFunction = result->AddFunction( BREPPlugin_ExportDriver::GetID(), EXPORT_SHAPE );
81   if( aFunction.IsNull() ) return;
82
83   //Check if the function is set correctly
84   if( aFunction->GetDriverGUID() != BREPPlugin_ExportDriver::GetID() ) return;
85
86   //Set parameters
87   BREPPlugin_IExport aCI( aFunction );
88   aCI.SetOriginal( aRefFunction );
89   aCI.SetFileName( theFileName );
90
91   //Perform the Export
92   try {
93     OCC_CATCH_SIGNALS;
94     if( !GetSolver()->ComputeFunction( aFunction ) ) {
95       SetErrorCode( "Not enough space on disk, or you haven't permissions to write this directory" );
96       return;
97     }
98   }
99   catch( Standard_Failure ) {
100     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
101     SetErrorCode( aFail->GetMessageString() );
102     return;
103   }
104
105   //Make a Python command
106   GEOM::TPythonDump(aFunction) << "geompy.ExportBREP(" << theOriginal << ", \""
107     << theFileName.ToCString() << "\" )";
108
109   SetErrorCode(OK);
110 }
111
112 //=============================================================================
113 /*!
114  *  ImportBREP
115  *  Import a shape from BREP format
116  *  \param theFileName The name of the file to import
117  *  \return List of GEOM_Objects, containing the created shape and propagation groups.
118  */
119 //=============================================================================
120 Handle(TColStd_HSequenceOfTransient)
121 BREPPlugin_IOperations::ImportBREP( const TCollection_AsciiString& theFileName )
122 {
123   SetErrorCode(KO);
124   if( theFileName.IsEmpty() ) return NULL;
125
126   //Add a new result object
127   Handle(GEOM_Object) anImported = GetEngine()->AddObject( GetDocID(), GEOM_IMPORT );
128
129   //Add an Import function
130   Handle(GEOM_Function) aFunction =
131     anImported->AddFunction( BREPPlugin_ImportDriver::GetID(), IMPORT_SHAPE);
132   if (aFunction.IsNull()) return NULL;
133
134   //Check if the function is set correctly
135   if (aFunction->GetDriverGUID() != BREPPlugin_ImportDriver::GetID()) return NULL;
136
137   //Set parameters
138   BREPPlugin_IImport aCI( aFunction );
139   aCI.SetFileName( theFileName );
140
141   //Perform the Import
142   Handle(TColStd_HSequenceOfTransient) aSeq = new TColStd_HSequenceOfTransient;
143
144   try {
145     OCC_CATCH_SIGNALS;
146     if( !GetSolver()->ComputeFunction( aFunction ) ) {
147       SetErrorCode( "Import driver failed" );
148       return NULL;
149     }
150     aSeq->Append(anImported);
151
152     // Greate material groups.
153     // MakeMaterialGroups( anImported, aSeq );
154   }
155   catch( Standard_Failure ) {
156     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
157     SetErrorCode( aFail->GetMessageString() );
158     return NULL;
159   }
160
161   //Make a Python command
162   GEOM::TPythonDump pd (aFunction);
163   pd << aSeq << " = geompy.ImportBREP(\"" << theFileName.ToCString() << "\" )";
164   SetErrorCode(OK);
165
166   return aSeq;
167 }