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