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 0d9a66680f6d0371c4d5ebf1b93d56b8176af5c6..6cf7242fa3f7666d7884c5f33df9583cf15a8e84 100644 (file)
--- a/src/HYDRO_tests/TestViewer.cxx
+++ b/src/HYDRO_tests/TestViewer.cxx
@@ -1,7 +1,9 @@
 
 #include <TestViewer.h>
+#include <HYDROData_Tool.h>
 #include <random.h>
 #include <OCCViewer_ViewManager.h>
+#include <OCCViewer_ViewPort3d.h>
 #ifdef WIN32
   #pragma warning ( disable: 4251 )
 #endif
@@ -15,6 +17,7 @@
 #endif
 #include <AIS_InteractiveContext.hxx>
 #include <AIS_Shape.hxx>
+#include <Aspect_ColorScale.hxx>
 #include <TopoDS_Iterator.hxx>
 
 #include <QDir>
@@ -69,19 +72,31 @@ QColor randomColor()
 {
   int r = test_rand();
   int aHue = r%255;
+  //std::cout << "hue: " << aHue << std::endl;
   QColor aColor = QColor::fromHsl( aHue, 255, 128 );
   return aColor;
 }
 
-void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const QColor& theColor )
+void TestViewer::show( const Handle(AIS_InteractiveObject)& theObject,
+                       int theMode, int theSelectionMode, bool isFitAll, const char* theKey )
 {
-  double r = theColor.red() / 255.0;
-  double g = theColor.green() / 255.0;
-  double b = theColor.blue() / 255.0;
+  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 );
   aShape->SetMaterial( Graphic3d_NOM_PLASTIC );
-  aShape->SetColor( Quantity_Color( r, g, b, Quantity_TOC_RGB ) );
+  aShape->SetColor( HYDROData_Tool::toOccColor( theColor ) );
   context()->Display( aShape, theMode, 0, Standard_False );
 
   if( isFitAll )
@@ -91,9 +106,9 @@ void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll,
   }
 }
 
-void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, char* theKey )
+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 );
@@ -152,6 +167,7 @@ bool TestViewer::AssertImages( QString& theMessage )
 
   QString aPath = QDir::tempPath() + "/" + myKey + ".png";
   anActualImage.save( aPath );
+  //std::cout << anActualImage.width() << "x" << anActualImage.height() << std::endl;
   theMessage = "The viewer contents does not correspond to the reference image: " + myKey;
   
   QImage aDiff( anExpectedRefImage.width(), anExpectedRefImage.height(), QImage::Format_ARGB32 );
@@ -165,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;
+}