X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROGUI%2FHYDROGUI_OCCDisplayer.h;h=92e20c815854ade1c62d9322833cf1fb7dc27df6;hb=dfa894b758ac65080dc7185d25b7646dc45a10f4;hp=bd89dfe291095c74e7b7112e713a80642ede679e;hpb=f0688b4c39fcc3e49c2b58a90724b9c1c84f1337;p=modules%2Fhydro.git

diff --git a/src/HYDROGUI/HYDROGUI_OCCDisplayer.h b/src/HYDROGUI/HYDROGUI_OCCDisplayer.h
index bd89dfe2..92e20c81 100644
--- a/src/HYDROGUI/HYDROGUI_OCCDisplayer.h
+++ b/src/HYDROGUI/HYDROGUI_OCCDisplayer.h
@@ -22,7 +22,8 @@
 #include "HYDROGUI_AbstractDisplayer.h"
 
 class HYDROGUI_Shape;
-class Handle(AIS_InteractiveContext);
+class AIS_InteractiveContext;
+class AIS_ColorScale;
 class OCCViewer_ViewManager;
 class OCCViewer_Viewer;
 
@@ -74,6 +75,13 @@ public:
 
   void SetToUpdateColorScale();
 
+  /**
+   * \brief Get color scale for the viewer.
+   * \param theViewerId viewer identifier
+   * \return the color scale 
+   */
+  Handle(AIS_ColorScale)          GetColorScale( const int theViewerId );
+
 protected:
   /**
    * \brief Erase all viewer objects.
@@ -109,7 +117,7 @@ protected:
   void                            purgeObjects( const int theViewerId );
 
   void UpdateColorScale( const OCCViewer_Viewer* );
-
+  
 private:
   /**
    * \brief Creates new OCC shape.
@@ -145,6 +153,8 @@ private:
 
 private:
   bool myToUpdateColorScale;
+  
+  QMap<int, Handle(AIS_ColorScale)> myColorScales;
 };
 
 #endif