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.
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.
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
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
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>
28 #include <QStringList>
29 #include <QTableWidget>
31 const QString DEF_STR_PATH = qgetenv( "HYDRO_ROOT_DIR" ) + "/share/salome/resources/hydro/def_strickler_table_06.txt";
33 void test_HYDROData_StricklerTable::test_import()
35 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
37 Handle(HYDROData_StricklerTable) aTable =
38 Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
40 CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
42 QStringList aTypes = aTable->GetTypes();
43 CPPUNIT_ASSERT_EQUAL( 44, aTypes.size() );
44 CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable->GetAttrName() );
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 ) );
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 ) );
59 void test_HYDROData_StricklerTable::test_import_export_equivalence()
61 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
63 Handle(HYDROData_StricklerTable) aTable =
64 Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
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 ) );
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 ) );
74 QByteArray aRefContents = aRefFile.readAll();
75 QByteArray aTmpContents = aTmpFile.readAll();
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] )
83 // CPPUNIT_ASSERT_EQUAL( true, isEqual );
90 void test_HYDROData_StricklerTable::test_type_by_attr()
92 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
94 Handle(HYDROData_StricklerTable) aTable =
95 Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
97 CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
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( "" ) );
106 void test_HYDROData_StricklerTable::test_unique_attr_name()
108 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
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 ) );
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() );
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() );
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() );
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() );
134 void test_HYDROData_StricklerTable::test_colors_sync()
136 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
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 ) );
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" ) );
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" ) );
158 void test_HYDROData_StricklerTable::test_duplication_refs_613()
160 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
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" );
168 Handle(HYDROData_StricklerTable) aTable2 =
169 Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
170 aTable1->CopyTo( aTable2, true );
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" ) );
176 aDoc->StartOperation();
179 Handle(HYDROData_StricklerTable) aTable3 =
180 Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
181 aTable3->SetName( "DefStrickler_1" );
182 aDoc->CommitOperation();
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() );
188 CPPUNIT_ASSERT_EQUAL( false, anIt1.More() );
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() );
196 CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
197 CPPUNIT_ASSERT_EQUAL( QString( "DefStrickler_2" ), anIt2.Current()->GetName() );
199 CPPUNIT_ASSERT_EQUAL( false, anIt2.More() );
204 void test_HYDROData_StricklerTable::test_dump_python()
206 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
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 ) );
214 QString aTmpPath = QDir::tempPath() + "/st_dump.py";
216 QString aTmpPath = QDir::tempPath() + "/hydro/st_dump.py";
219 CPPUNIT_ASSERT_EQUAL( true, aDoc->DumpToPython( aTmpPath, false ) );
221 CPPUNIT_ASSERT_SCRIPTS_EQUAL( "st_dump.py", true, true, 0 );
226 void test_HYDROData_StricklerTable::test_add_row_in_gui_refs_717()
228 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
230 Handle(HYDROData_StricklerTable) aTable =
231 Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
233 CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
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 );