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 <TestViewer.h>
  25 #include <QStringList>
  26 #include <QColor>
  27 #include <QDir>
  28
  29 const QString DEF_STR_PATH = qgetenv( "HYDRO_SRC_DIR" ) + "/src/HYDROGUI/resources/def_strickler_table.txt";
  30
  31 void test_HYDROData_StricklerTable::test_import()
  32 {
  33   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  34
  35   Handle(HYDROData_StricklerTable) aTable =
  36     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  37
  38   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  39
  40   QStringList aTypes = aTable->GetTypes();
  41   CPPUNIT_ASSERT_EQUAL( 8, aTypes.size() );
  42   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable->GetAttrName() );
  43
  44   QString aType = "Zones de champs cultivé à végétation basse";
  45   CPPUNIT_ASSERT_EQUAL( aType, aTypes[3] );
  46   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable->GetColor( aType ) );
  47   CPPUNIT_ASSERT_EQUAL( QString( "512" ), aTable->GetAttrValue( aType ) );
  48
  49   aType = "Zones à forte urbanization (agglomération)";
  50   CPPUNIT_ASSERT_EQUAL( aType, aTypes[7] );
  51   CPPUNIT_ASSERT_EQUAL( QColor( 18, 52, 86 ), aTable->GetColor( aType ) );
  52   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetAttrValue( aType ) );
  53
  54   aDoc->Close();
  55 }
  56
  57 void test_HYDROData_StricklerTable::test_import_export_equivalence()
  58 {
  59   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  60
  61   Handle(HYDROData_StricklerTable) aTable =
  62     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  63
  64   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  65   QString aTmpPath = QDir::tempPath() + "/stricker.txt";
  66   CPPUNIT_ASSERT_EQUAL( true, aTable->Export( aTmpPath ) );
  67
  68   QFile aRefFile( DEF_STR_PATH ), aTmpFile( aTmpPath );
  69   CPPUNIT_ASSERT_EQUAL( true, aRefFile.open( QFile::ReadOnly ) );
  70   CPPUNIT_ASSERT_EQUAL( true, aTmpFile.open( QFile::ReadOnly ) );
  71
  72   QByteArray aRefContents = aRefFile.readAll();
  73   QByteArray aTmpContents = aTmpFile.readAll();
  74
  75   bool isEqual = aRefContents.size()==aTmpContents.size();
  76   CPPUNIT_ASSERT_EQUAL( true, isEqual );
  77   for( int i=0, n=aRefContents.size(); isEqual && i<n; i++ )
  78     if( aRefContents[i]!=aTmpContents[i] )
  79       isEqual = false;
  80
  81   CPPUNIT_ASSERT_EQUAL( true, isEqual );
  82
  83   aRefFile.close();
  84   aTmpFile.close();
  85   aDoc->Close();
  86 }
  87
  88 void test_HYDROData_StricklerTable::test_type_by_attr()
  89 {
  90   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  91
  92   Handle(HYDROData_StricklerTable) aTable =
  93     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  94
  95   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  96
  97   CPPUNIT_ASSERT_EQUAL( QString( "Zones de champs cultivé à végétation basse" ), aTable->GetType( "512" ) );
  98   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "123" ) );
  99   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "" ) );
 100
 101   aDoc->Close();
 102 }
 103
 104 void test_HYDROData_StricklerTable::test_unique_attr_name()
 105 {
 106   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 107
 108   Handle(HYDROData_StricklerTable) aTable1 =
 109     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 110   Handle(HYDROData_StricklerTable) aTable2 =
 111     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 112
 113   CPPUNIT_ASSERT_EQUAL( true, aTable1->SetAttrName( "CODE_06" ) );
 114   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 115   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
 116
 117   CPPUNIT_ASSERT_EQUAL( false, aTable2->SetAttrName( "CODE_06" ) );
 118   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 119   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
 120
 121   CPPUNIT_ASSERT_EQUAL( true, aTable2->SetAttrName( "CODE_07" ) );
 122   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 123   CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
 124
 125   CPPUNIT_ASSERT_EQUAL( false, aTable1->SetAttrName( "CODE_07" ) );
 126   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 127   CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
 128
 129   aDoc->Close();
 130 }
 131
 132 void test_HYDROData_StricklerTable::test_colors_sync()
 133 {
 134   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 135
 136   Handle(HYDROData_StricklerTable) aTable1 =
 137     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 138   Handle(HYDROData_StricklerTable) aTable2 =
 139     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 140
 141   aTable1->Set( "test1", 10 );
 142   aTable1->Set( "test2", 20 );
 143   aTable1->SetColor( "test1", QColor( 255, 0, 0 ) );
 144   aTable1->SetColor( "test2", QColor( 255, 0, 1 ) );
 145   CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 0 ), aTable1->GetColor( "test1" ) );
 146   CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 1 ), aTable1->GetColor( "test2" ) );
 147
 148   aTable2->Set( "test1", 15 );
 149   aTable2->SetColor( "test1", QColor( 255, 255, 0 ) );
 150   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable1->GetColor( "test1" ) );
 151   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable2->GetColor( "test1" ) );
 152
 153   aDoc->Close();
 154 }
 155
 156 void test_HYDROData_StricklerTable::test_duplication_refs_613()
 157 {
 158   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 159
 160   Handle(HYDROData_StricklerTable) aTable1 =
 161     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 162   aTable1->Set( "type1", 1.5 );
 163   aTable1->SetColor( "type1", QColor( 10, 20, 30 ) );
 164   aTable1->SetName( "DefStrickler_1" );
 165
 166   Handle(HYDROData_StricklerTable) aTable2 =
 167     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 168   aTable1->CopyTo( aTable2, true );
 169
 170   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_2" ), aTable2->GetName() );
 171   CPPUNIT_ASSERT_DOUBLES_EQUAL( 1.5, aTable2->Get( "type1", 0.0 ), 1E-5 );
 172   CPPUNIT_ASSERT_EQUAL( QColor( 10, 20, 30 ), aTable2->GetColor( "type1" ) );
 173
 174   aDoc->StartOperation();
 175   aTable1->Remove();
 176   aTable2->Remove();
 177   Handle(HYDROData_StricklerTable) aTable3 =
 178     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 179   aTable3->SetName( "DefStrickler_1" );
 180   aDoc->CommitOperation();
 181
 182   HYDROData_Iterator anIt1( aDoc, KIND_STRICKLER_TABLE );
 183   CPPUNIT_ASSERT_EQUAL( true, anIt1.More() );
 184   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_1" ), anIt1.Current()->GetName() );
 185   anIt1.Next();
 186   CPPUNIT_ASSERT_EQUAL( false, anIt1.More() );
 187
 188   aDoc->Undo();
 189
 190   HYDROData_Iterator anIt2( aDoc, KIND_STRICKLER_TABLE );
 191   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
 192   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_1" ), anIt2.Current()->GetName() );
 193   anIt2.Next();
 194   CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
 195   CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_2" ), anIt2.Current()->GetName() );
 196   anIt2.Next();
 197   CPPUNIT_ASSERT_EQUAL( false, anIt2.More() );
 198
 199   aDoc->Close();
 200 }
 201
 202 void test_HYDROData_StricklerTable::test_dump_python()
 203 {
 204   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 205
 206   Handle(HYDROData_StricklerTable) aTable =
 207     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 208   aTable->SetName( "ST" );
 209   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
 210
 211   QString aTmpPath = QDir::tempPath() + "/st_dump.py";
 212   CPPUNIT_ASSERT_EQUAL( true, aDoc->DumpToPython( aTmpPath, false ) );
 213
 214   CPPUNIT_ASSERT_SCRIPTS_EQUAL( "st_dump.py" );
 215
 216   aDoc->Close();
 217 }