src/HYDRO_tests/test_HYDROData_StricklerTable.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 #include <test_HYDROData_StricklerTable.h>
  20 #include <HYDROData_Document.h>
  21 #include <HYDROData_Iterator.h>
  22 #include <HYDROData_StricklerTable.h>
  23 #include <HYDROData_Tool.h>
  24 #include <HYDROGUI_StricklerTableDlg.h>
  25 #include <TestViewer.h>
  26 #include <QColor>
  27 #include <QDir>
  28 #include <QStringList>
  29 #include <QTableWidget>
  30
  31 const QString DEF_STR_PATH = qgetenv( "HYDRO_ROOT_DIR" ) + "/share/salome/resources/hydro/def_strickler_table_06.txt";
  32
  33 void test_HYDROData_StricklerTable::test_import()
  34 {
  35   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  36
  37   Handle(HYDROData_StricklerTable) aTable =
  38     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  39
  40   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  41
  42   QStringList aTypes = aTable->GetTypes();
  43   CPPUNIT_ASSERT_EQUAL( 44, aTypes.size() );
  44   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable->GetAttrName() );
  45
  46   QString aType = QString::fromUtf8("Périmètres irrigués en permanence");
  47   CPPUNIT_ASSERT_EQUAL( aType, aTypes[25] );
  48   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable->GetColor( aType ) );
  49   CPPUNIT_ASSERT_EQUAL( QString( "212" ), aTable->GetAttrValue( aType ) );
  50
  51   aType = QString::fromUtf8("Végétation clairsemée");
  52   CPPUNIT_ASSERT_EQUAL( aType, aTypes[38] );
  53   CPPUNIT_ASSERT_EQUAL( QColor( 204, 255, 204 ), aTable->GetColor( aType ) );
  54   CPPUNIT_ASSERT_EQUAL( QString( "333" ), aTable->GetAttrValue( aType ) );
  55
  56   aDoc->Close();
  57 }
  58
  59 void test_HYDROData_StricklerTable::test_import_export_equivalence()
  60 {
  61   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  62
  63   Handle(HYDROData_StricklerTable) aTable =
  64     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  65
  66   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  67   QString aTmpPath = QDir::tempPath() + "/stricker.txt";
  68   CPPUNIT_ASSERT_EQUAL( true, aTable->Export( aTmpPath ) );
  69
  70   QFile aRefFile( DEF_STR_PATH ), aTmpFile( aTmpPath );
  71   CPPUNIT_ASSERT_EQUAL( true, aRefFile.open( QFile::ReadOnly ) );
  72   CPPUNIT_ASSERT_EQUAL( true, aTmpFile.open( QFile::ReadOnly ) );
  73
  74   QByteArray aRefContents = aRefFile.readAll();
  75   QByteArray aTmpContents = aTmpFile.readAll();
  76
  77 //  bool isEqual = aRefContents.size()==aTmpContents.size();
  78 //  CPPUNIT_ASSERT_EQUAL( true, isEqual );
  79 //  for( int i=0, n=aRefContents.size(); isEqual && i<n; i++ )
  80 //    if( aRefContents[i]!=aTmpContents[i] )
  81 //      isEqual = false;
  82 //
  83 //  CPPUNIT_ASSERT_EQUAL( true, isEqual );
  84
  85   aRefFile.close();
  86   aTmpFile.close();
  87   aDoc->Close();
  88 }
  89
  90 void test_HYDROData_StricklerTable::test_type_by_attr()
  91 {
  92   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  93
  94   Handle(HYDROData_StricklerTable) aTable =
  95     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  96
  97   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  98
  99   CPPUNIT_ASSERT_EQUAL( QString( "Plans d'eau" ), aTable->GetType( "512" ) );
 100   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "125" ) );
 101   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "" ) );
 102
 103   aDoc->Close();
 104 }
 105
 106 void test_HYDROData_StricklerTable::test_unique_attr_name()
 107 {
 108   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 109
 110   Handle(HYDROData_StricklerTable) aTable1 =
 111     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 112   Handle(HYDROData_StricklerTable) aTable2 =
 113     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 114
 115   CPPUNIT_ASSERT_EQUAL( true, aTable1->SetAttrName( "CODE_06" ) );
 116   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 117   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
 118
 119   CPPUNIT_ASSERT_EQUAL( false, aTable2->SetAttrName( "CODE_06" ) );
 120   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 121   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
 122
 123   CPPUNIT_ASSERT_EQUAL( true, aTable2->SetAttrName( "CODE_07" ) );
 124   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 125   CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
 126
 127   CPPUNIT_ASSERT_EQUAL( false, aTable1->SetAttrName( "CODE_07" ) );
 128   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 129   CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
 130
 131   aDoc->Close();
 132 }
 133
 134 void test_HYDROData_StricklerTable::test_colors_sync()
 135 {
 136   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 137
 138   Handle(HYDROData_StricklerTable) aTable1 =
 139     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 140   Handle(HYDROData_StricklerTable) aTable2 =
 141     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 142
 143   aTable1->Set( "test1", 10 );
 144   aTable1->Set( "test2", 20 );
 145   aTable1->SetColor( "test1", QColor( 255, 0, 0 ) );
 146   aTable1->SetColor( "test2", QColor( 255, 0, 1 ) );
 147   CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 0 ), aTable1->GetColor( "test1" ) );
 148   CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 1 ), aTable1->GetColor( "test2" ) );
 149
 150   aTable2->Set( "test1", 15 );
 151   aTable2->SetColor( "test1", QColor( 255, 255, 0 ) );
 152   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable1->GetColor( "test1" ) );
 153   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable2->GetColor( "test1" ) );
 154
 155   aDoc->Close();
 156 }
 157
 158 void test_HYDROData_StricklerTable::test_duplication_refs_613()
 159 {
 160   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 161
 162   Handle(HYDROData_StricklerTable) aTable1 =
 163     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 164   aTable1->Set( "type1", 1.5 );
 165   aTable1->SetColor( "type1", QColor( 10, 20, 30 ) );
 166   aTable1->SetName( "DefStrickler_1" );
 167
 168   Handle(HYDROData_StricklerTable) aTable2 =
 169     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 170   aTable1->CopyTo( aTable2, true );
 171
 172   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_2" ), aTable2->GetName() );
 173   CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.5, aTable2->Get( "type1", 0.0 ), 1E-5 );
 174   CPPUNIT_ASSERT_EQUAL( QColor( 10, 20, 30 ), aTable2->GetColor( "type1" ) );
 175
 176   aDoc->StartOperation();
 177   aTable1->Remove();
 178   aTable2->Remove();
 179   Handle(HYDROData_StricklerTable) aTable3 =
 180     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 181   aTable3->SetName( "DefStrickler_1" );
 182   aDoc->CommitOperation();
 183
 184   HYDROData_Iterator anIt1( aDoc, KIND_STRICKLER_TABLE );
 185   CPPUNIT_ASSERT_EQUAL( true, anIt1.More() );
 186   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_1" ), anIt1.Current()->GetName() );
 187   anIt1.Next();
 188   CPPUNIT_ASSERT_EQUAL( false, anIt1.More() );
 189
 190   aDoc->Undo();
 191
 192   HYDROData_Iterator anIt2( aDoc, KIND_STRICKLER_TABLE );
 193   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
 194   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_1" ), anIt2.Current()->GetName() );
 195   anIt2.Next();
 196   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
 197   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_2" ), anIt2.Current()->GetName() );
 198   anIt2.Next();
 199   CPPUNIT_ASSERT_EQUAL( false, anIt2.More() );
 200
 201   aDoc->Close();
 202 }
 203
 204 void test_HYDROData_StricklerTable::test_dump_python()
 205 {
 206   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 207
 208   Handle(HYDROData_StricklerTable) aTable =
 209     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 210   aTable->SetName( "ST" );
 211   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
 212
 213 #ifdef WIN32
 214   QString aTmpPath = QDir::tempPath() + "/st_dump.py";
 215 #else
 216   QString aTmpPath = QDir::tempPath() + "/hydro/st_dump.py";
 217 #endif
 218
 219   CPPUNIT_ASSERT_EQUAL( true, aDoc->DumpToPython( aTmpPath, false ) );
 220
 221   CPPUNIT_ASSERT_SCRIPTS_EQUAL( "st_dump.py", true, true, 0 );
 222
 223   aDoc->Close();
 224 }
 225
 226 void test_HYDROData_StricklerTable::test_add_row_in_gui_refs_717()
 227 {
 228   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 229
 230   Handle(HYDROData_StricklerTable) aTable =
 231     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 232
 233   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
 234
 235   HYDROGUI_StricklerTableDlg* aDlg = new HYDROGUI_StricklerTableDlg( 0, "", 0 );
 236   aDlg->setGuiData( aTable );
 237   CPPUNIT_ASSERT_EQUAL( 44, aDlg->myTable->rowCount() );
 238   aDlg->onAddCoefficient();
 239   aDlg->getGuiData( aTable );
 240
 241   aDlg->deleteLater();
 242   aDoc->Close();
 243 }