Salome HOME
refs #662: the tests for land cover presentations
[modules/hydro.git] / src / HYDRO_tests / TestViewer.cxx
index 02516f517c1749ba779745ee138043a457f16c47..6cf7242fa3f7666d7884c5f33df9583cf15a8e84 100644 (file)
@@ -3,6 +3,7 @@
 #include <HYDROData_Tool.h>
 #include <random.h>
 #include <OCCViewer_ViewManager.h>
+#include <OCCViewer_ViewPort3d.h>
 #ifdef WIN32
   #pragma warning ( disable: 4251 )
 #endif
@@ -16,6 +17,7 @@
 #endif
 #include <AIS_InteractiveContext.hxx>
 #include <AIS_Shape.hxx>
+#include <Aspect_ColorScale.hxx>
 #include <TopoDS_Iterator.hxx>
 
 #include <QDir>
@@ -76,11 +78,13 @@ QColor randomColor()
 }
 
 void TestViewer::show( const Handle(AIS_InteractiveObject)& theObject,
-                       int theMode, int theSelectionMode, bool isFitAll )
+                       int theMode, int theSelectionMode, bool isFitAll, const char* theKey )
 {
   context()->EraseAll( Standard_False );
   context()->Display( theObject, theMode, theSelectionMode );
 
+  myKey = theKey;
+
   if( isFitAll )
   {
     viewWindow()->onTopView();
@@ -177,3 +181,36 @@ bool TestViewer::AssertImages( QString& theMessage )
 
   return false;
 }
+
+Handle_Aspect_ColorScale TestViewer::showColorScale()
+{
+  Handle(V3d_View) aView = myViewWindow->getViewPort()->getView();
+  if( aView.IsNull() )
+    return Handle(Aspect_ColorScale)();
+
+  Handle(Aspect_ColorScale) aColorScale = aView->ColorScale();
+  if( aColorScale.IsNull() )
+    return aColorScale;
+
+  Standard_Real anXPos = 0.05; //TODO
+  Standard_Real anYPos = 0.1; //TODO
+  Standard_Real aWidth = 0.2; //TODO
+  Standard_Real aHeight = 0.5; //TODO
+  Standard_Integer aTextHeight = 14; //TODO
+  Standard_Integer aNbIntervals = 30; //TODO
+
+  aColorScale->SetXPosition( anXPos );
+  aColorScale->SetYPosition( anYPos );
+  aColorScale->SetWidth( aWidth );
+  aColorScale->SetHeight( aHeight );
+  aColorScale->SetTextHeight( aTextHeight );
+  aColorScale->SetNumberOfIntervals( aNbIntervals );
+
+  aColorScale->SetTitle( "test" );
+  aColorScale->SetRange( 0, 1 );
+
+  if( !aView->ColorScaleIsDisplayed() )
+    aView->ColorScaleDisplay();
+
+  return aColorScale;
+}