The bathymetry is changed to their base altitude class for geometry objects (Bug...

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

diff --git a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetryDriver.h b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetryDriver.h
index 37bee6935a312dde0aae7185a814399e6e1b25b1..a5b23b3aaa638e0e577094015c6680589f1a7c4c 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetryDriver.h
+++ b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetryDriver.h
@@ -23,7 +23,9 @@
 #ifndef HYDROGUI_VTKPrsBathymetryDRIVER_H
 #define HYDROGUI_VTKPrsBathymetryDRIVER_H
 
-#include <HYDROGUI_VTKPrsDriver.h>
+#include "HYDROGUI_VTKPrsDriver.h"
+
+#include <vtkScalarBarActor.h>
 
 /**
  * \class HYDROGUI_VTKPrsBathymetryDriver
@@ -35,7 +37,7 @@ public:
   /**
    * \brief Constructor.
    */
-  HYDROGUI_VTKPrsBathymetryDriver();
+  HYDROGUI_VTKPrsBathymetryDriver( vtkScalarBarActor* theScalarBar );
 
   /**
    * \brief Destructor.
@@ -51,6 +53,9 @@ public:
    */
   virtual bool Update( const Handle(HYDROData_Entity)& theObj,
                        HYDROGUI_VTKPrs*& thePrs );
+
+private:
+  vtkScalarBarActor* myScalarBar;
 };
 
 #endif