Salome HOME
c71ab52ed387357a04ee263c94b3dda82ce74b48
[modules/geom.git] / src / VTKPlugin / VTKPlugin_OperationsCreator.cxx
1 // Copyright (C) 2014  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 "VTKPlugin_OperationsCreator.hxx"
22 #include "VTKPlugin_IOperations_i.hh"
23 #include "VTKPlugin_IOperations.hxx"
24 #include "VTKPlugin_ExportDriver.hxx"
25 #include "VTKPlugin_IECallBack.hxx"
26
27 // KERNEL includes
28 #include <Utils_ExceptHandlers.hxx>
29 #include <utilities.h>
30
31 // OCCT includes
32 #include <TFunction_DriverTable.hxx>
33
34 std::map <int, VTKPlugin_IOperations*> VTKPlugin_OperationsCreator::_mapOfOperations;
35
36 VTKPlugin_OperationsCreator::VTKPlugin_OperationsCreator()
37 {
38   // Register drivers
39   TFunction_DriverTable::Get()->AddDriver( VTKPlugin_ExportDriver::GetID(),
40                                            new VTKPlugin_ExportDriver() );
41   
42   // Register callback
43   VTKPlugin_IECallBack* callback = new VTKPlugin_IECallBack();
44   GEOMImpl_IECallBack::Register( "VTK", callback );
45 }
46
47 VTKPlugin_OperationsCreator::~VTKPlugin_OperationsCreator()
48 {
49 }
50
51 GEOM_IOperations_i* VTKPlugin_OperationsCreator::Create( PortableServer::POA_ptr thePOA,
52                                                          int                     theStudyId,
53                                                          GEOM::GEOM_Gen_ptr      theEngine,
54                                                          ::GEOMImpl_Gen*         theGenImpl )
55 {
56   Unexpect aCatch( SALOME_SalomeException );
57   MESSAGE( "VTKPlugin_OperationsCreator::Create" );
58   return new VTKPlugin_IOperations_i( thePOA, theEngine, get( theGenImpl, theStudyId ) );
59 }
60
61 VTKPlugin_IOperations* VTKPlugin_OperationsCreator::get( ::GEOMImpl_Gen* theGenImpl,
62                                                          int             theStudyId )
63 {
64   if (_mapOfOperations.find( theStudyId ) == _mapOfOperations.end() )
65     _mapOfOperations[theStudyId] = new VTKPlugin_IOperations( theGenImpl, theStudyId );
66   return _mapOfOperations[theStudyId];
67 }