Salome HOME
Merge from V6_3_BR 06/06/2011
[modules/smesh.git] / src / OBJECT / SMESH_ActorDef.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 //  SMESH OBJECT : interactive object for SMESH visualization
24 //  File   : SMESH_ActorDef.h
25 //  Author : Nicolas REJNERI
26 //  Module : SMESH
27 //
28 #ifndef SMESH_ACTORDEF_H
29 #define SMESH_ACTORDEF_H
30
31 #include "SMESH_Actor.h"
32 #include "SMESH_Object.h"
33
34 #include <vtkSmartPointer.h>
35
36 template <class T>
37 class TVTKSmartPtr: public vtkSmartPointer<T>
38 {
39 public:
40   TVTKSmartPtr() {}
41   TVTKSmartPtr(T* r, bool theIsOwner = false): vtkSmartPointer<T>(r) { 
42     if(r && theIsOwner) 
43       r->Delete();
44   }
45   TVTKSmartPtr& operator()(T* r, bool theIsOwner = false){ 
46     vtkSmartPointer<T>::operator=(r); 
47     if(r && theIsOwner) 
48       r->Delete();
49     return *this;
50   }
51   TVTKSmartPtr& operator=(T* r){ vtkSmartPointer<T>::operator=(r); return *this;}
52   T* Get() const { return this->GetPointer();}
53 };
54
55
56 class vtkProperty;
57 class vtkShrinkFilter;
58 class vtkPolyDataMapper;
59 class vtkUnstructuredGrid;
60 class vtkMergeFilter;
61 class vtkPolyData;
62 class vtkMapper;
63 class vtkActor2D;
64 class vtkMaskPoints;
65 class vtkLabeledDataMapper;
66 class vtkSelectVisiblePoints;
67 class vtkLookupTable;
68 class vtkPlane;
69 class vtkImplicitBoolean;
70 class vtkTimeStamp;
71
72 class VTKViewer_CellCenters;
73
74 class SMESH_DeviceActor;
75 class SMESH_ScalarBarActor;
76
77 #ifndef DISABLE_PLOT2DVIEWER
78 class SPlot2d_Histogram;
79 #endif
80
81
82 class SMESH_ActorDef : public SMESH_Actor
83 {
84   friend class SMESH_VisualObj;
85   friend class SMESH_Actor;
86
87  public:
88   vtkTypeMacro(SMESH_ActorDef,SMESH_Actor);
89   
90   virtual void ReleaseGraphicsResources(vtkWindow *renWin);
91   virtual int RenderOpaqueGeometry(vtkViewport *viewport);
92   virtual int RenderTranslucentGeometry(vtkViewport *viewport);
93   virtual void Render(vtkRenderer *ren);
94
95   virtual void AddToRender(vtkRenderer* theRenderer); 
96   virtual void RemoveFromRender(vtkRenderer* theRenderer);
97
98   virtual bool hasHighlight() { return true; }  
99   virtual void highlight(bool theHighlight);  
100   virtual void SetPreSelected(bool thePreselect = false);
101
102   virtual bool IsInfinitive();  
103
104   virtual void SetOpacity(vtkFloatingPointType theValue);
105   virtual vtkFloatingPointType GetOpacity();
106
107   virtual void SetSufaceColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
108   virtual void GetSufaceColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
109
110   virtual void SetBackSufaceColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
111   virtual void GetBackSufaceColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
112
113   virtual void SetEdgeColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
114   virtual void GetEdgeColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
115
116   virtual void SetNodeColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
117   virtual void GetNodeColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
118
119   virtual void Set0DColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
120   virtual void Get0DColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
121
122   virtual void SetHighlightColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
123   virtual void GetHighlightColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
124
125   virtual void SetPreHighlightColor(vtkFloatingPointType r,vtkFloatingPointType g,vtkFloatingPointType b);
126   virtual void GetPreHighlightColor(vtkFloatingPointType& r,vtkFloatingPointType& g,vtkFloatingPointType& b);
127  
128   virtual vtkFloatingPointType GetLineWidth();
129   virtual void SetLineWidth(vtkFloatingPointType theVal);
130
131   virtual void Set0DSize(vtkFloatingPointType size);
132   virtual vtkFloatingPointType Get0DSize();
133
134   virtual int GetNodeObjId(int theVtkID);
135   virtual vtkFloatingPointType* GetNodeCoord(int theObjID);
136
137   virtual int GetElemObjId(int theVtkID);
138   virtual vtkCell* GetElemCell(int theObjID);
139
140   virtual int GetObjDimension( const int theObjId );
141
142   virtual void SetVisibility(int theMode);
143   void SetVisibility(int theMode, bool theIsUpdateRepersentation);
144
145   virtual void SetRepresentation(int theMode);
146   
147   virtual unsigned int GetEntityMode() const { return myEntityMode;}
148   virtual void SetEntityMode(unsigned int theMode);
149
150   virtual void SetPointRepresentation(bool theIsPointsVisible);
151   virtual bool GetPointRepresentation();
152
153   virtual vtkFloatingPointType* GetBounds();
154   virtual void SetTransform(VTKViewer_Transform* theTransform); 
155
156   virtual vtkUnstructuredGrid* GetUnstructuredGrid();
157   virtual vtkDataSet* GetInput();
158   virtual vtkMapper* GetMapper();
159
160   virtual vtkFloatingPointType GetShrinkFactor();
161   virtual void SetShrinkFactor(vtkFloatingPointType theValue);
162
163   virtual bool IsShrunkable() { return myIsShrinkable;}
164   virtual bool IsShrunk() { return myIsShrunk;}
165   virtual void SetShrink(); 
166   virtual void UnShrink(); 
167
168   virtual void SetPointsLabeled(bool theIsPointsLabeled);
169   virtual bool GetPointsLabeled(){ return myIsPointsLabeled;}
170
171   virtual void SetCellsLabeled(bool theIsCellsLabeled);
172   virtual bool GetCellsLabeled(){ return myIsCellsLabeled;}
173
174   virtual void SetFacesOriented(bool theIsFacesOriented);
175   virtual bool GetFacesOriented();
176
177   virtual void SetFacesOrientationColor(vtkFloatingPointType theColor[3]);
178   virtual void GetFacesOrientationColor(vtkFloatingPointType theColor[3]);
179
180   virtual void SetFacesOrientationScale(vtkFloatingPointType theScale);
181   virtual vtkFloatingPointType GetFacesOrientationScale();
182
183   virtual void SetFacesOrientation3DVectors(bool theState);
184   virtual bool GetFacesOrientation3DVectors();
185
186   virtual void SetControlMode(eControl theMode);
187   virtual eControl GetControlMode(){ return myControlMode;}
188   virtual SMESH::Controls::FunctorPtr GetFunctor() { return myFunctor; }
189
190   virtual SMESH_ScalarBarActor* GetScalarBarActor(){ return myScalarBarActor;}
191
192   virtual void RemoveAllClippingPlanes();
193   virtual vtkIdType GetNumberOfClippingPlanes();
194   virtual vtkPlane* GetClippingPlane(vtkIdType theID);
195   virtual vtkIdType AddClippingPlane(vtkPlane* thePlane); 
196
197   virtual TVisualObjPtr GetObject() { return myVisualObj;}
198
199   virtual void SetControlsPrecision( const long p ) { myControlsPrecision = p; }
200   virtual long GetControlsPrecision() const { return myControlsPrecision; }
201
202   virtual void UpdateScalarBar();
203   virtual void UpdateDistribution();
204
205 #ifndef DISABLE_PLOT2DVIEWER
206   virtual SPlot2d_Histogram* GetPlot2Histogram() { return my2dHistogram; }
207   virtual SPlot2d_Histogram* UpdatePlot2Histogram();
208 #endif
209
210
211   virtual void SetQuadratic2DRepresentation(EQuadratic2DRepresentation);
212   virtual EQuadratic2DRepresentation GetQuadratic2DRepresentation();
213   
214   virtual void SetMarkerStd( VTK::MarkerType, VTK::MarkerScale );
215   virtual void SetMarkerTexture( int, VTK::MarkerTexture );
216
217  protected:
218   void SetControlMode(eControl theMode, bool theCheckEntityMode);
219   void SetImplicitFunctionUsed(bool theIsImplicitFunctionUsed);
220   bool IsImplicitFunctionUsed() const;
221
222   TVisualObjPtr myVisualObj;
223   vtkTimeStamp* myTimeStamp;
224
225   SMESH_ScalarBarActor* myScalarBarActor;
226   vtkLookupTable* myLookupTable;
227
228   vtkProperty* mySurfaceProp;
229   vtkProperty* myBackSurfaceProp;
230   vtkProperty* myEdgeProp;
231   vtkProperty* myNodeProp;
232
233   SMESH_DeviceActor* myBaseActor;
234   SMESH_DeviceActor* myNodeActor;
235   SMESH_DeviceActor* myPickableActor;
236
237   vtkProperty* myHighlightProp;
238   vtkProperty* myPreselectProp;
239   SMESH_DeviceActor* myHighlitableActor;
240
241   eControl myControlMode;
242   SMESH::Controls::FunctorPtr myFunctor;
243   vtkProperty* my2DExtProp;
244   SMESH_DeviceActor* my2DActor;
245   SMESH_DeviceActor* my2DExtActor;
246   SMESH_DeviceActor* my3DActor;
247   SMESH_DeviceActor* my3DExtActor;
248   SMESH_DeviceActor* myControlActor;
249
250   vtkProperty* myNodeExtProp;
251   SMESH_DeviceActor* myNodeExtActor;
252
253   vtkProperty* my1DProp;
254   SMESH_DeviceActor* my1DActor;
255   vtkProperty* my1DExtProp;
256   SMESH_DeviceActor* my1DExtActor;
257
258   vtkProperty* my0DProp;
259   SMESH_DeviceActor* my0DActor;
260   vtkProperty* my0DExtProp;
261   SMESH_DeviceActor* my0DExtActor;
262
263   unsigned int myEntityMode;
264   unsigned int myEntityState;
265   bool myIsPointsVisible;
266
267   bool myIsShrinkable;
268   bool myIsShrunk;
269   
270   bool myIsPointsLabeled;
271   vtkUnstructuredGrid* myPointsNumDataSet;
272   vtkActor2D *myPointLabels;
273   vtkMaskPoints* myPtsMaskPoints;
274   vtkLabeledDataMapper* myPtsLabeledDataMapper;
275   vtkSelectVisiblePoints* myPtsSelectVisiblePoints;
276
277   bool myIsCellsLabeled;
278   vtkUnstructuredGrid* myCellsNumDataSet;
279   vtkActor2D *myCellsLabels;
280   vtkMaskPoints* myClsMaskPoints;
281   VTKViewer_CellCenters* myCellCenters;
282   vtkLabeledDataMapper* myClsLabeledDataMapper;
283   vtkSelectVisiblePoints* myClsSelectVisiblePoints;
284
285   vtkImplicitBoolean* myImplicitBoolean;
286   typedef TVTKSmartPtr<vtkPlane> TPlanePtr;
287   typedef std::vector<TPlanePtr> TCippingPlaneCont;
288   TCippingPlaneCont myCippingPlaneCont;
289   long myControlsPrecision;
290
291 #ifndef DISABLE_PLOT2DVIEWER
292   SPlot2d_Histogram* my2dHistogram;
293 #endif
294
295   bool myIsFacesOriented;
296
297   VTK::MarkerTexture myMarkerTexture;
298
299   SMESH_ActorDef();
300   ~SMESH_ActorDef();
301
302   bool Init(TVisualObjPtr theVisualObj, 
303             const char* theEntry, 
304             const char* theName,
305             int theIsClear);
306
307   void SetIsShrunkable(bool theShrunkable);
308   void UpdateHighlight();
309   void Update();
310
311  private:
312   // hide the two parameter Render() method from the user and the compiler.
313   virtual void Render(vtkRenderer *, vtkMapper *) {};
314   virtual void ShallowCopy(vtkProp *prop);
315   virtual void SetMapper(vtkMapper *);
316   static SMESH_ActorDef* New();
317
318   // Not implemented.
319   SMESH_ActorDef(const SMESH_ActorDef&);  
320   void operator=(const SMESH_ActorDef&);
321 };
322
323
324 #endif //SMESH_ACTORDEF_H