refs #662: the tests for land cover presentations

[modules/hydro.git] / src / HYDRO_tests / TestViewer.cxx

diff --git a/src/HYDRO_tests/TestViewer.cxx b/src/HYDRO_tests/TestViewer.cxx
index 5b5de3d225e930d7a4f1def28d89c9961a8c09df..6cf7242fa3f7666d7884c5f33df9583cf15a8e84 100644 (file)
--- a/src/HYDRO_tests/TestViewer.cxx
+++ b/src/HYDRO_tests/TestViewer.cxx
@@ -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>
@@ -75,6 +77,21 @@ QColor randomColor()
   return aColor;
 }
 
+void TestViewer::show( const Handle(AIS_InteractiveObject)& theObject,
+                       int theMode, int theSelectionMode, bool isFitAll, const char* theKey )
+{
+  context()->EraseAll( Standard_False );
+  context()->Display( theObject, theMode, theSelectionMode );
+
+  myKey = theKey;
+
+  if( isFitAll )
+  {
+    viewWindow()->onTopView();
+    viewWindow()->onFitAll();
+  }
+}
+
 void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const QColor& theColor )
 {
   Handle(AIS_Shape) aShape = new AIS_Shape( theShape );
@@ -91,7 +108,7 @@ void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll,
 
 void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const char* theKey )
 {
-  context()->EraseAll();
+  context()->EraseAll( Standard_False );
   
   myKey = theKey;
   test_srand( 0 );
@@ -164,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;
+}