Merge branch 'BR_1328' into BR_DEMO

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

diff --git a/src/HYDROGUI/HYDROGUI_OCCDisplayer.h b/src/HYDROGUI/HYDROGUI_OCCDisplayer.h
index 968e051ec2ec032fccc3c74ba2c47980fb0ba58d..275954472bcd21b65b54531eab4ac330fc806517 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_OCCDisplayer.h
+++ b/src/HYDROGUI/HYDROGUI_OCCDisplayer.h
@@ -23,6 +23,7 @@
 
 class HYDROGUI_Shape;
 class AIS_InteractiveContext;
+class AIS_ColorScale;
 class OCCViewer_ViewManager;
 class OCCViewer_Viewer;
 
@@ -74,6 +75,15 @@ 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 );
+
+  void UpdateColorScale( const OCCViewer_Viewer* );
+
 protected:
   /**
    * \brief Erase all viewer objects.
@@ -100,15 +110,11 @@ protected:
                                            const int theViewerId,
                                            const bool theIsForced,
                                            const bool theDoFitAll );
-
-protected:
   /**
    * \brief Purge all invalid objects in the viewer.
    * \param theViewerId viewer identifier
    */
   void                            purgeObjects( const int theViewerId );
-
-  void UpdateColorScale( const OCCViewer_Viewer* );
   
 private:
   /**
@@ -145,6 +151,8 @@ private:
 
 private:
   bool myToUpdateColorScale;
+  
+  QMap<int, Handle(AIS_ColorScale)> myColorScales;
 };
 
 #endif