Exlude the user input during process events by application (Bug #325).

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_VTKPrsBathymetry.h

diff --git a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.h b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.h
index 3a4098c718cb11fa05b455999ee4ef5c2cf75ab2..8b8cd54f53274b6943aae63bde85ebc3b7cc31e7 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.h
+++ b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.h
@@ -27,6 +27,11 @@
 
 #include <HYDROData_Bathymetry.h>
 
+#include <vtkScalarsToColors.h>
+#include <vtkWeakPointer.h>
+#include <vtkNew.h>
+#include <vtkPolyDataMapper.h>
+
 /*
   Class       : HYDROGUI_VTKPrsBathymetry
   Description : Presentation for Bathymetry object
@@ -37,7 +42,18 @@ public:
   HYDROGUI_VTKPrsBathymetry( const Handle(HYDROData_Bathymetry)& theObject );
   virtual ~HYDROGUI_VTKPrsBathymetry();
 
-  virtual void compute();
+  virtual void                     compute();
+  virtual bool                     needScalarBar() { return true; }
+
+  //! Get the range of colored 
+  void setLookupTable( vtkScalarsToColors* theTable ) { myLookupTable = theTable; }
+
+protected:
+  virtual vtkMapper*               mapper() { return myMapper.GetPointer(); }
+
+private:
+  vtkWeakPointer< vtkScalarsToColors > myLookupTable;
+  vtkNew< vtkPolyDataMapper >          myMapper;
 };
 
 #endif