Salome HOME
correct previous integration (Porting to Python 2.6)
[modules/smesh.git] / src / OBJECT / SMESH_DeviceActor.h
1 //  Copyright (C) 2007-2008  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 //  SMESH OBJECT : interactive object for SMESH visualization
23 //  File   : SMESH_DeviceActor.h
24 //  Author : Nicolas REJNERI
25 //  Module : SMESH
26 //  $Header$
27 //
28 #ifndef SMESH_DEVICE_ACTOR_H
29 #define SMESH_DEVICE_ACTOR_H
30
31 #include <VTKViewer_GeometryFilter.h>
32 #include "SMESH_Controls.hxx"
33 #include "SMESH_Object.h"
34
35 #include <vtkLODActor.h>
36 #include <vtkSmartPointer.h>
37
38 class vtkCell;
39 class vtkProperty;
40 class vtkMergeFilter;
41 class vtkShrinkFilter;
42 class vtkPolyDataMapper;
43 class vtkUnstructuredGrid;
44 class vtkScalarBarActor;
45 class vtkLookupTable;
46 class vtkImplicitBoolean;
47 class vtkPassThroughFilter;
48
49 class VTKViewer_Transform;
50 class VTKViewer_TransformFilter;
51 class VTKViewer_ExtractUnstructuredGrid;
52
53 class SMESH_ExtractGeometry;
54 class SMESH_FaceOrientationFilter;
55
56
57 class SMESHOBJECT_EXPORT SMESH_DeviceActor: public vtkLODActor{
58   friend class SMESH_ActorDef;
59
60  public:
61   vtkTypeMacro(SMESH_DeviceActor,vtkLODActor);
62   static SMESH_DeviceActor* New();
63
64   void SetStoreClippingMapping(bool theStoreMapping);
65   void SetStoreGemetryMapping(bool theStoreMapping);
66   void SetStoreIDMapping(bool theStoreMapping);
67
68   virtual int GetNodeObjId(int theVtkID);
69   virtual vtkFloatingPointType* GetNodeCoord(int theObjID);
70
71   virtual int GetElemObjId(int theVtkID);
72   virtual vtkCell* GetElemCell(int theObjID);
73
74   virtual void SetTransform(VTKViewer_Transform* theTransform); 
75   virtual unsigned long int GetMTime();
76
77   virtual void SetFacesOriented(bool theIsFacesOriented);
78   virtual bool GetFacesOriented() { return myIsFacesOriented; }
79
80   //----------------------------------------------------------------------------
81   //! Setting for displaying quadratic elements
82   virtual void SetQuadraticArcMode(bool theFlag);
83   virtual bool GetQuadraticArcMode();
84   
85   virtual void SetQuadraticArcAngle(vtkFloatingPointType theMaxAngle);
86   virtual vtkFloatingPointType GetQuadraticArcAngle();
87   
88   void UpdateFaceOrientation();
89
90   vtkFloatingPointType GetShrinkFactor();
91   void  SetShrinkFactor(vtkFloatingPointType value);
92
93   bool IsShrunkable() { return myIsShrinkable;}
94   bool IsShrunk() { return myIsShrunk;}
95   void SetShrink(); 
96   void UnShrink(); 
97
98   enum EReperesent { ePoint, eWireframe, eSurface, eInsideframe};
99   EReperesent GetRepresentation(){ return myRepresentation;}
100   void SetRepresentation(EReperesent theMode);
101
102   virtual void SetVisibility(int theMode);
103   virtual int GetVisibility();
104
105   virtual void AddToRender(vtkRenderer* theRenderer); 
106   virtual void RemoveFromRender(vtkRenderer* theRenderer);
107
108   VTKViewer_ExtractUnstructuredGrid* GetExtractUnstructuredGrid();
109   vtkUnstructuredGrid* GetUnstructuredGrid();
110
111   void SetControlMode(SMESH::Controls::FunctorPtr theFunctor,
112                       vtkScalarBarActor* theScalarBarActor,
113                       vtkLookupTable* theLookupTable);
114   void SetExtControlMode(SMESH::Controls::FunctorPtr theFunctor,
115                          vtkScalarBarActor* theScalarBarActor,
116                          vtkLookupTable* theLookupTable);
117   void SetExtControlMode(SMESH::Controls::FunctorPtr theFunctor);
118
119   bool IsHighlited() { return myIsHighlited;}
120   void SetHighlited(bool theIsHighlited);
121
122   virtual void Render(vtkRenderer *, vtkMapper *);
123
124   void SetImplicitFunctionUsed(bool theIsImplicitFunctionUsed);
125   bool IsImplicitFunctionUsed() const{ return myIsImplicitFunctionUsed;}
126
127  protected:
128   void Init(TVisualObjPtr theVisualObj, vtkImplicitBoolean* theImplicitBoolean);
129   void SetUnstructuredGrid(vtkUnstructuredGrid* theGrid);
130
131   vtkPolyDataMapper *myMapper;
132   TVisualObjPtr myVisualObj;
133
134   vtkProperty *myProperty;
135   EReperesent myRepresentation;
136
137   SMESH_ExtractGeometry* myExtractGeometry;
138   bool myIsImplicitFunctionUsed;
139
140   vtkMergeFilter* myMergeFilter;
141   VTKViewer_ExtractUnstructuredGrid* myExtractUnstructuredGrid;
142
143   bool myIsFacesOriented;
144   SMESH_FaceOrientationFilter* myFaceOrientationFilter;
145   vtkPolyDataMapper* myFaceOrientationDataMapper;
146   vtkActor* myFaceOrientation;
147
148   bool myStoreClippingMapping;
149   VTKViewer_GeometryFilter *myGeomFilter;
150   VTKViewer_TransformFilter *myTransformFilter;
151   std::vector<vtkPassThroughFilter*> myPassFilter;
152
153   vtkShrinkFilter* myShrinkFilter;
154   bool myIsShrinkable;
155   bool myIsShrunk;
156   
157   bool myIsHighlited;
158
159   vtkFloatingPointType myPolygonOffsetFactor;
160   vtkFloatingPointType myPolygonOffsetUnits;
161
162   void
163   SetPolygonOffsetParameters(vtkFloatingPointType factor, 
164                              vtkFloatingPointType units);
165
166   void
167   GetPolygonOffsetParameters(vtkFloatingPointType& factor, 
168                              vtkFloatingPointType& units)
169   {
170     factor = myPolygonOffsetFactor;
171     units = myPolygonOffsetUnits;
172   }
173
174   SMESH_DeviceActor();
175   ~SMESH_DeviceActor();
176   SMESH_DeviceActor(const SMESH_DeviceActor&);
177   void operator=(const SMESH_DeviceActor&);
178
179 };
180
181
182 #endif //SMESH_DEVICE_ACTOR_H