X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDRO_tests%2FTestViewer.cxx;h=3f886a5c025c8f8ac29d17457dead53007e7967c;hb=ddc8b027120ad163e63fd1a6b9828aeec3aff79b;hp=f3bd5105943fd80eb84d17a7ab30e685c943d209;hpb=7f6a67a5b0ca7be95649a25ed3632217ac9fd35d;p=modules%2Fhydro.git diff --git a/src/HYDRO_tests/TestViewer.cxx b/src/HYDRO_tests/TestViewer.cxx index f3bd5105..3f886a5c 100644 --- a/src/HYDRO_tests/TestViewer.cxx +++ b/src/HYDRO_tests/TestViewer.cxx @@ -1,3 +1,20 @@ +// Copyright (C) 2014-2015 EDF-R&D +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include #include @@ -13,29 +30,39 @@ #ifdef WIN32 #pragma warning ( disable: 4251 ) #endif -#include +#include #ifdef WIN32 #pragma warning ( disable: 4251 ) #endif -#include #include -#include +#include #include #include #include #include #include +#include +#include +#include + +#include +#include +#include #ifdef WIN32 #pragma warning ( default: 4251 ) #endif #include +#include OCCViewer_ViewManager* TestViewer::myViewManager = 0; -OCCViewer_ViewWindow* TestViewer::myViewWindow = 0; +OCCViewer_ViewFrame* TestViewer::myViewWindow = 0; QString TestViewer::myKey = ""; +extern QString REF_DATA_PATH; +extern QString TMP_DIR; + OCCViewer_ViewManager* TestViewer::viewManager() { if( myViewManager ) @@ -44,13 +71,13 @@ OCCViewer_ViewManager* TestViewer::viewManager() myViewManager = new OCCViewer_ViewManager( 0, 0 ); OCCViewer_Viewer* aViewer = new OCCViewer_Viewer( true ); - aViewer->setTrihedronSize( 100, true ); + aViewer->setTrihedronSize( 0, true ); aViewer->setInteractionStyle( 0 ); aViewer->setZoomingStyle( 1 ); myViewManager->setViewModel( aViewer ); - myViewWindow = dynamic_cast( myViewManager->createViewWindow() ); - + myViewWindow = dynamic_cast( myViewManager->createViewWindow() ); + aViewer->setTrihedronShown( false ); return myViewManager; } @@ -59,15 +86,15 @@ OCCViewer_Viewer* TestViewer::viewer() return dynamic_cast( viewManager()->getViewModel() ); } -OCCViewer_ViewWindow* TestViewer::viewWindow() +OCCViewer_ViewFrame* TestViewer::viewWindow() { viewManager(); //to create the view if it was not created earlier return myViewWindow; } -Handle(AIS_InteractiveContext) context() +Handle(AIS_InteractiveContext) TestViewer::context() { - return TestViewer::viewer()->getAISContext(); + return viewer()->getAISContext(); } QColor TestViewer::GetColor(int i) @@ -83,67 +110,156 @@ QColor TestViewer::GetColor(int i) << QColor(100,100,20) << QColor(10,100,150); } - return aCV[i]; + if (i < aCV.size()) + return aCV[i]; + else + { + QColor TestColor = aCV[i % aCV.size()]; + QColor NewColor((TestColor.red() + i * 41) % 256, + (TestColor.green() + i * 13) % 256, + (TestColor.blue() + i * 23) % 256); + return NewColor; + } +} + +void TestViewer::eraseAll( bool isUpdate, bool eraseStructures ) +{ + context()->CloseLocalContext( -1, Standard_False ); + context()->EraseAll( isUpdate ); + if( eraseStructures ) + { + Graphic3d_MapOfStructure GmapS; + viewer()->getViewer3d()->StructureManager()->DisplayedStructures(GmapS); + for (Graphic3d_MapOfStructure::Iterator it(GmapS); it.More(); it.Next()) + { + Handle(Graphic3d_Structure) GS = it.Key(); + GS->Erase(); + } + } + viewer()->setTrihedronShown( false ); + qApp->processEvents(); } void TestViewer::show( const Handle(AIS_InteractiveObject)& theObject, int theMode, int theSelectionMode, bool isFitAll, const char* theKey ) { - context()->CloseLocalContext(); - context()->EraseAll( Standard_False ); + QString aNewKey = theKey; + if( !aNewKey.isEmpty() ) + { + myKey = aNewKey; + eraseAll( false, true ); + } + if( theSelectionMode > 0 ) { - context()->Display( theObject, theMode, theSelectionMode ); context()->OpenLocalContext(); + context()->Display( theObject, theMode, theSelectionMode, true ); context()->Activate( theObject, theSelectionMode, Standard_True ); } - - myKey = theKey; + else + context()->Display( theObject, theMode, theSelectionMode, true ); if( isFitAll ) - { - viewWindow()->onTopView(); - viewWindow()->onFitAll(); - } + fitAll(); } -void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const QColor& theColor ) +void TestViewer::fitAll() +{ + viewWindow()->onTopView(); + viewWindow()->onFitAll(); +} + +void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const QColor& theColor, + int theUIANb, int theVIANb) { Handle(AIS_Shape) aShape = new AIS_Shape( theShape ); + cout <Attributes()->PointAspect()->SetTypeOfMarker( Aspect_TOM_X ); + if (theShape.ShapeType()==TopAbs_FACE) + { + context()->DefaultDrawer()->UIsoAspect()->SetNumber(theUIANb); + context()->DefaultDrawer()->VIsoAspect()->SetNumber(theVIANb); + Handle_Prs3d_Drawer aDrawer = aShape->Attributes(); + aDrawer->UIsoAspect()->SetNumber(theUIANb); + aDrawer->VIsoAspect()->SetNumber(theVIANb); + } aShape->SetMaterial( Graphic3d_NOM_PLASTIC ); aShape->SetColor( HYDROData_Tool::toOccColor( theColor ) ); context()->Display( aShape, theMode, 0, Standard_False ); if( isFitAll ) + fitAll(); +} + +void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const char* theKey, + int theUIANb, int theVIANb) +{ + QString aNewKey = theKey; + if( !aNewKey.isEmpty() ) { - viewWindow()->onTopView(); - viewWindow()->onFitAll(); + eraseAll( false ); + myKey = aNewKey; } + + if( theShape.IsNull() ) + return; + + int i = 0; + //show all faces first + TopTools_ListOfShape aListOfFaces; + TopExp_Explorer aFE( theShape, TopAbs_FACE ); + for( ; aFE.More(); aFE.Next() ) + aListOfFaces.Append(aFE.Current()); + GEOMUtils::SortShapes(aListOfFaces); + TopTools_ListIteratorOfListOfShape aLF(aListOfFaces); + for( ; aLF.More(); aLF.Next(), i++) + show( aLF.Value(), theMode, false, GetColor(i), theUIANb, theVIANb ); + + //show all independent wires + TopTools_ListOfShape aListOfWires; + TopExp_Explorer aWE( theShape, TopAbs_WIRE, TopAbs_FACE ); + for( ; aWE.More(); aWE.Next() ) + aListOfWires.Append(aWE.Current()); + GEOMUtils::SortShapes(aListOfWires); + TopTools_ListIteratorOfListOfShape aLW(aListOfWires); + for( ; aLW.More(); aLW.Next(), i++) + show( aLW.Value(), theMode, false, GetColor(i) ); + + //show all independent edges + TopTools_ListOfShape aListOfEdges; + TopExp_Explorer anEE( theShape, TopAbs_EDGE, TopAbs_WIRE ); + for( ; anEE.More(); anEE.Next()) + aListOfEdges.Append(anEE.Current()); + GEOMUtils::SortShapes(aListOfEdges); + TopTools_ListIteratorOfListOfShape aLE(aListOfEdges); + for( ; aLE.More(); aLE.Next(), i++) + show( aLE.Value(), theMode, false, GetColor(i) ); + + if( isFitAll ) + fitAll(); } -void TestViewer::show( const TopoDS_Shape& theShape, int theMode, bool isFitAll, const char* theKey ) +/*void TestViewer::ShowShape(const TopoDS_Shape& theShape, int theMode, int& i) { - context()->EraseAll( Standard_False ); - - myKey = theKey; - int i = 0; if( theShape.ShapeType()==TopAbs_SHELL ) { TopoDS_Iterator anIt( theShape ); - for( ; anIt.More(); anIt.Next(), i++ ) + for( ; anIt.More(); anIt.Next()) + { show( anIt.Value(), theMode, false, GetColor(i) ); + i++; + } } - else - show( theShape, theMode, false, GetColor(0) ); - - if( isFitAll ) + else if (theShape.ShapeType()==TopAbs_FACE || + theShape.ShapeType()==TopAbs_WIRE || + theShape.ShapeType()==TopAbs_EDGE || + theShape.ShapeType()==TopAbs_VERTEX ) { - viewWindow()->onTopView(); - viewWindow()->onFitAll(); + show( theShape, theMode, false, GetColor(i) ); + i++; } -} +}*/ bool AreImagesEqual( const QImage& theImage1, const QImage& theImage2, double theTolerance ) { @@ -167,14 +283,44 @@ bool AreImagesEqual( const QImage& theImage1, const QImage& theImage2, double th return true; } -bool TestViewer::AssertImages( QString& theMessage ) +QImage TestViewer::diff( const QImage& theExpectedRefImage, const QImage& theActualImage ) { - QImage anActualImage = viewWindow()->dumpView(); + const QImage::Format aFormat = QImage::Format_RGB32; + + QImage anExpectedRefImage = theExpectedRefImage.convertToFormat( aFormat ); + QImage anActualImage = theActualImage.convertToFormat( aFormat ); + + QImage aDiff( anExpectedRefImage.width(), anExpectedRefImage.height(), aFormat ); + QPainter aPainter( &aDiff ); + aPainter.drawImage( 0, 0, anExpectedRefImage ); + aPainter.setCompositionMode( QPainter::RasterOp_SourceXorDestination ); + aPainter.drawImage( 0, 0, anActualImage ); + return aDiff; +} + +bool TestViewer::AssertImages( QString& theMessage, const QImage* theImage, const char* theCase, bool swapRGB ) +{ + QImage anActualImage; + if( theImage ) + anActualImage = *theImage; + else + anActualImage = viewWindow()->getView(OCCViewer_ViewFrame::MAIN_VIEW)->dumpView(); + + if( swapRGB ) + { + // A temporary patch for bug in SALOME/OCC dump; the result image contains swapped RGB colors + anActualImage = anActualImage.rgbSwapped(); + } - QString anExpectedRefFilePath = qgetenv( "HYDRO_REFERENCE_DATA" ); + + if( theCase ) + myKey = theCase; + + QString anExpectedRefFilePath = REF_DATA_PATH; anExpectedRefFilePath += "/" + myKey + ".png"; QImage anExpectedRefImage; anExpectedRefImage.load( anExpectedRefFilePath ); + //std::cout << "Expected image loading: " << anExpectedRefFilePath.toStdString() << std::endl; if( AreImagesEqual( anActualImage, anExpectedRefImage, 0.001 ) ) { @@ -182,39 +328,45 @@ bool TestViewer::AssertImages( QString& theMessage ) return true; } - QString aPath = QDir::tempPath() + "/" + myKey + ".png"; + QString aPath = TMP_DIR + "/" + myKey + ".png"; anActualImage.save( aPath ); + //std::cout << "Actual image: " << aPath.toStdString() << std::endl; + //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 ); - QPainter aPainter( &aDiff ); - aPainter.drawImage( 0, 0, anExpectedRefImage ); - aPainter.setCompositionMode( QPainter::RasterOp_SourceXorDestination ); - aPainter.drawImage( 0, 0, anActualImage ); + QImage aDiff = diff( anExpectedRefImage, anActualImage ); - QString aDiffFilePath = QDir::tempPath() + "/" + myKey + "_diff.png"; + QString aDiffFilePath = TMP_DIR + "/" + myKey + "_diff.png"; aDiff.save( aDiffFilePath ); + //std::cout << "Diff image: " << aDiffFilePath.toStdString() << std::endl; + + QString anExpected = TMP_DIR + "/" + myKey + "_1.png"; + //std::cout << "Expected image: " << anExpected.toStdString() << std::endl; + anExpectedRefImage.save( anExpected ); return false; } -Handle_Aspect_ColorScale TestViewer::showColorScale( bool isShow ) +Handle(AIS_ColorScale) TestViewer::colorScale() { - Handle(V3d_View) aView = myViewWindow->getViewPort()->getView(); - if( aView.IsNull() ) - return Handle(Aspect_ColorScale)(); + static Handle(AIS_ColorScale) aColorScale = new AIS_ColorScale(); + + return aColorScale; +} - Handle(Aspect_ColorScale) aColorScale = aView->ColorScale(); +void TestViewer::showColorScale( bool isShow ) +{ + Handle(AIS_ColorScale) aColorScale = colorScale(); if( aColorScale.IsNull() ) - return aColorScale; + return; - 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 + Standard_Real anXPos = 0.05; + Standard_Real anYPos = 0.1; + Standard_Real aWidth = 0.2; + Standard_Real aHeight = 0.5; + Standard_Integer aTextHeight = 14; + Standard_Integer aNbIntervals = 30; aColorScale->SetXPosition( anXPos ); aColorScale->SetYPosition( anYPos ); @@ -226,15 +378,99 @@ Handle_Aspect_ColorScale TestViewer::showColorScale( bool isShow ) aColorScale->SetTitle( "test" ); aColorScale->SetRange( 0, 1 ); + aColorScale->SetToUpdate(); + if( isShow ) { - if( !aView->ColorScaleIsDisplayed() ) - aView->ColorScaleDisplay(); + if( !context()->IsDisplayed( aColorScale ) ) + context()->Display( aColorScale, true ); } else { - if( aView->ColorScaleIsDisplayed() ) - aView->ColorScaleErase(); + if( context()->IsDisplayed( aColorScale ) ) + context()->Erase( aColorScale, true ); } - return aColorScale; +} + +bool TestViewer::ColorScaleIsDisplayed() +{ + return context()->IsDisplayed( colorScale() ); +} + +void TestViewer::select( int theViewX, int theViewY ) +{ + Handle(V3d_View) aView = myViewWindow->getViewPort()->getView(); + context()->MoveTo( theViewX, theViewY, aView, true); + context()->Select(true); + // context()->CloseAllContexts(); +} + +QString GetLine( QFile& theFile, bool isUtf8 ) +{ + QByteArray aLineData = theFile.readLine(); + QString aLine; + if( isUtf8 ) + aLine = QString::fromUtf8( aLineData ); + else + aLine = aLineData; + return aLine; +} + +bool TestViewer::areScriptsEqual( const QString& theBaseName, + bool isExpectedUtf8, + bool isActualUtf8, + int theLinesToOmit, + QString& theMsg ) +{ + QString anExpectedRefFilePath = REF_DATA_PATH; + anExpectedRefFilePath += "/" + theBaseName; + + QString anActualFilePath = TMP_DIR + "/" + theBaseName; + + QFile anExpected( anExpectedRefFilePath ); + QFile anActual( anActualFilePath ); + if( !anExpected.open( QFile::ReadOnly | QFile::Text ) ) + { + theMsg = "Expected file cannot be opened: " + anExpectedRefFilePath; + return false; + } + + if( !anActual.open( QFile::ReadOnly | QFile::Text ) ) + { + theMsg = "Actual file cannot be opened: " + anActualFilePath; + return false; + } + + for( int i=0; i