- Bathymethries are colored now by all colors (including red etc.)

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

diff --git a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.h b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.h
index 3a4098c718cb11fa05b455999ee4ef5c2cf75ab2..25e357d53737a283a15b57f60184fa8436d718d1 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
@@ -38,6 +43,16 @@ public:
   virtual ~HYDROGUI_VTKPrsBathymetry();
 
   virtual void compute();
+
+  //! 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