src/HYDRO_tests/test_HYDROGUI_ProfilesDlg.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #undef HYDROGUI_EXPORTS
  20
  21 #include <test_HYDROGUI_ProfilesDlg.h>
  22
  23 #define private public
  24 #include <HYDROGUI_ProfileDlg.h>
  25 #include <CurveCreator_Widget.h>
  26 #undef private
  27
  28 #include <HYDROGUI_CurveCreatorProfile.h>
  29 #include <CurveCreator_TableView.h>
  30 #include <CurveCreator_Displayer.hxx>
  31 #include <CurveCreator_Utils.hxx>
  32
  33 #include <HYDROData_Document.h>
  34 #include <HYDROData_Profile.h>
  35
  36
  37 #include <TestViewer.h>
  38 #include <QApplication>
  39 #include <QAction>
  40 #include <QSplitter>
  41 #include <QTest>
  42
  43 #define CPPUNIT_ASSERT_WIDGET( theWidget, theCase )                           \
  44   {                                                                           \
  45     QPixmap aPixmap = QPixmap::grabWindow( theWidget->winId() );              \
  46     QImage anImage = aPixmap.toImage();                                       \
  47     QString aMessage;                                                         \
  48     if( !TestViewer::AssertImages( aMessage, &anImage, theCase, false ) )     \
  49     {                                                                         \
  50       TestViewer::showColorScale( false );                                    \
  51       std::string aMessageStl = aMessage.toStdString();                       \
  52       CPPUNIT_FAIL( aMessageStl.c_str() );                                    \
  53     }                                                                         \
  54   }                                                                           \
  55
  56
  57 void test_HYDROGUI_ProfilesDlg::test_default_size()
  58 {
  59   HYDROGUI_ProfileDlg* dlg = new HYDROGUI_ProfileDlg( 0, "test", true );
  60   dlg->resize( 320, 640 );
  61   dlg->show();
  62   QTest::qWaitForWindowExposed( dlg );
  63
  64   CPPUNIT_ASSERT_WIDGET( dlg, "profiles_dlg_presentation" );
  65
  66   dlg->deleteLater();
  67   qApp->processEvents();
  68 }
  69
  70 HYDROGUI_CurveCreatorProfile* profile()
  71 {
  72   static HYDROGUI_CurveCreatorProfile* aProfile = 0;
  73   if( aProfile )
  74     return aProfile;
  75
  76   aProfile = new HYDROGUI_CurveCreatorProfile();
  77   CurveCreator::Coordinates coords;
  78   coords.push_back( 0 );
  79   coords.push_back( 20 );
  80   coords.push_back( 30 );
  81   coords.push_back( 0 );
  82   coords.push_back( 60 );
  83   coords.push_back( 0 );
  84   coords.push_back( 90 );
  85   coords.push_back( 20 );
  86   aProfile->addPoints( coords, 0 );
  87   return aProfile;
  88 }
  89
  90 void show_profile( HYDROGUI_ProfileDlg* dlg )
  91 {
  92   Handle(AIS_InteractiveContext) aCtx = dlg->getAISContext();
  93   CurveCreator_Displayer* aDisplayer = new CurveCreator_Displayer( aCtx );
  94   profile()->setDisplayer( aDisplayer );
  95   aDisplayer->display( profile()->getAISObject( true ), true );
  96 }
  97
  98 void select_points( HYDROGUI_ProfileDlg* dlg )
  99 {
 100   CurveCreator_ICurve::SectionToPointList sel;
 101   sel.push_back( CurveCreator_ICurve::SectionToPoint( 0, 0 ) );
 102   sel.push_back( CurveCreator_ICurve::SectionToPoint( 0, 1 ) );
 103   sel.push_back( CurveCreator_ICurve::SectionToPoint( 0, 2 ) );
 104   sel.push_back( CurveCreator_ICurve::SectionToPoint( 0, 3 ) );
 105   dlg->myEditorWidget->getAction( CurveCreator_Widget::ModificationMode )->toggle();
 106   dlg->myEditorWidget->onModificationMode( true );
 107   dlg->myEditorWidget->setSelectedPoints( sel );
 108   dlg->myEditorWidget->updateLocalPointView();
 109 }
 110
 111 std::string coords( HYDROGUI_ProfileDlg* dlg, int col=2 )
 112 {
 113   int n = dlg->myEditorWidget->myLocalPointView->rowCount();
 114   QStringList datas;
 115   for( int r=0; r<n; r++ )
 116   {
 117     QString data = dlg->myEditorWidget->myLocalPointView->item(r, col)->data( Qt::DisplayRole ).toString();
 118     datas.append( data );
 119   }
 120   return datas.join( ", " ).toStdString();
 121 }
 122
 123 void setCoords( HYDROGUI_ProfileDlg* dlg, int theIndex, double theValue )
 124 {
 125   QAbstractItemModel* m = dlg->myEditorWidget->myLocalPointView->model();
 126   QModelIndex index = m->index( theIndex, 2 );
 127
 128   m->setData( index, theValue, Qt::UserRole );
 129   dlg->myEditorWidget->updateLocalPointView();
 130 }
 131
 132 void test_HYDROGUI_ProfilesDlg::test_points_table()
 133 {
 134   HYDROGUI_ProfileDlg* dlg = new HYDROGUI_ProfileDlg( 0, "test", false );
 135   dlg->resize( 320, 800 );
 136   dlg->setProfile( profile() );
 137   dlg->show();
 138   QTest::qWaitForWindowExposed( dlg );
 139
 140   show_profile( dlg );
 141   select_points( dlg );
 142
 143   QList<int> sizes;
 144   sizes.append( 25 );
 145   sizes.append( 100 );
 146   sizes.append( 200 );
 147   sizes.append( 25 );
 148   sizes.append( 25 );
 149   dlg->splitter()->setSizes( sizes );
 150   qApp->processEvents();
 151
 152   CPPUNIT_ASSERT_WIDGET( dlg, "profiles_selected_points" );
 153   CPPUNIT_ASSERT_EQUAL( std::string( "0, 30, 60, 90" ), coords( dlg ) );
 154   CPPUNIT_ASSERT_EQUAL( std::string( "20, 0, 0, 20" ), coords( dlg, 3 ) );
 155
 156   setCoords( dlg, 0, 130 );
 157   CPPUNIT_ASSERT_EQUAL( std::string( "130, 30, 60, 90" ), coords( dlg ) );
 158   CPPUNIT_ASSERT_EQUAL( std::string( "20, 0, 0, 20" ), coords( dlg, 3 ) );
 159
 160   setCoords( dlg, 0, 0 );
 161   CPPUNIT_ASSERT_EQUAL( std::string( "0, 30, 60, 90" ), coords( dlg ) );
 162   CPPUNIT_ASSERT_EQUAL( std::string( "20, 0, 0, 20" ), coords( dlg, 3 ) );
 163
 164   setCoords( dlg, 0, 80 );
 165   dlg->myEditorWidget->myLocalPointView->OnHeaderClick( 2 );
 166   CPPUNIT_ASSERT_EQUAL( std::string( "30, 60, 80, 90" ), coords( dlg ) );
 167   CPPUNIT_ASSERT_EQUAL( std::string( "0, 0, 20, 20" ), coords( dlg, 3 ) );
 168
 169   setCoords( dlg, 2, 0 );
 170   CPPUNIT_ASSERT_EQUAL( std::string( "30, 60, 0, 90" ), coords( dlg ) );
 171   CPPUNIT_ASSERT_EQUAL( std::string( "0, 0, 20, 20" ), coords( dlg, 3 ) );
 172
 173   //qApp->processEvents();
 174   //QTest::qWait( 50000 );
 175
 176   dlg->deleteLater();
 177   qApp->processEvents();
 178 }