Salome HOME
e4872babef45a01435994302bd95eb5ab9d1fd3a
[modules/visu.git] / src / PIPELINE / SALOME_ExtractGeometry.h
1 // Copyright (C) 2007-2011  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.
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 #ifndef SALOME_ExtractGeometry_H
24 #define SALOME_ExtractGeometry_H
25
26 #include "VISUPipeline.hxx"
27
28 #include <vtkExtractGeometry.h>
29 #include <vtkSmartPointer.h>
30
31 #include <vector>
32
33 class vtkImplicitBoolean;
34
35 class VISU_PIPELINE_EXPORT SALOME_ExtractGeometry : public vtkExtractGeometry
36 {
37 public:
38   vtkTypeMacro(SALOME_ExtractGeometry, vtkExtractGeometry);
39
40   static 
41   SALOME_ExtractGeometry*
42   New();
43
44   virtual
45   void
46   SetImplicitFunction(vtkImplicitFunction* theImplicitFunction); 
47
48   vtkImplicitBoolean* 
49   GetImplicitBoolean();
50
51   bool 
52   GetStoreMapping() const;
53
54   void
55   SetStoreMapping(bool theStoreMapping);
56
57   virtual 
58   vtkIdType 
59   GetNodeObjId(vtkIdType theID);
60
61   virtual
62   vtkIdType
63   GetElemObjId(vtkIdType theID);
64
65   virtual
66   vtkIdType
67   GetNodeVTKId(vtkIdType theID);
68
69   virtual
70   vtkIdType
71   GetElemVTKId(vtkIdType theID);
72
73 protected:
74   SALOME_ExtractGeometry();
75   ~SALOME_ExtractGeometry();
76
77   // Usual data generation method
78   virtual
79   int
80   RequestData(vtkInformation *, vtkInformationVector **, vtkInformationVector *);
81
82   virtual
83   int
84   RequestData2(vtkInformation *, vtkInformationVector **, vtkInformationVector *);
85
86 private:
87   bool myStoreMapping;
88   bool myIsDoneShallowCopy;
89
90   typedef std::vector<vtkIdType> TVectorId;
91   TVectorId myElemVTK2ObjIds;
92   TVectorId myNodeVTK2ObjIds;
93
94   vtkSmartPointer<vtkImplicitBoolean> myImplicitBoolean;
95
96   SALOME_ExtractGeometry(const SALOME_ExtractGeometry&);  // Not implemented.
97   void operator=(const SALOME_ExtractGeometry&);  // Not implemented.
98 };
99
100 #endif