Salome HOME
HYDRO_tests reference data on install directory
[modules/hydro.git] / src / HYDRO_tests / test_HYDROData_LandCoverMap.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_LandCoverMap.h>
20 #include <HYDROData_CalculationCase.h>
21 #include <HYDROData_Document.h>
22 #include <HYDROData_LandCoverMap.h>
23 #include <HYDROData_ImmersibleZone.h>
24 #include <HYDROData_PolylineXY.h>
25 #include <HYDROData_StricklerTable.h>
26 #include <HYDROData_Tool.h>
27 #include <HYDROData_ShapeFile.h>
28 #include <HYDROGUI_LandCoverMapPrs.h>
29 #include <TopoDS_Edge.hxx>
30 #include <TopoDS_Wire.hxx>
31 #include <TopoDS_Face.hxx>
32 #include <TopoDS.hxx>
33 #include <TestViewer.h>
34 #include <TestShape.h>
35 #include <TopTools_ListOfShape.hxx>
36 #include <AIS_DisplayMode.hxx>
37 #include <Aspect_ColorScale.hxx>
38 #include <BRepMesh_IncrementalMesh.hxx>
39 #include <QString>
40 #include <QColor>
41 #include <QMap>
42 #include <QDir>
43 #include <BRepTools.hxx>
44 #include <BRep_Builder.hxx>
45 #include <BRepCheck_Analyzer.hxx>
46
47 #define _DEVDEBUG_
48 #include "HYDRO_trace.hxx"
49
50 const QString REF_DATA_PATH = qgetenv( "HYDRO_ROOT_DIR" ) + "/bin/salome/test/HYDRO";
51 const QString DEF_STR_PATH = qgetenv( "HYDRO_ROOT_DIR" ) + "/share/salome/resources/hydro/def_strickler_table.txt";
52
53 void test_HYDROData_LandCoverMap::test_add_2_objects()
54 {
55   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
56
57   Handle(HYDROData_LandCoverMap) aMap =
58     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
59
60   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
61
62   TopoDS_Face aLC1 = Face( QList<double>() << 10 << 10 << 50 << 20 << 30 << 50 << 15 << 30 );
63   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "test1" ) );
64
65   TopoDS_Face aLC2 = Face( QList<double>() << 30 << 20 << 60 << 10 << 70 << 35 << 40 << 40 );
66   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "test2" ) );
67
68   TestViewer::show( aMap->GetShape(), AIS_Shaded, true, "LandCoverMap_Add_2_Objects" );
69   CPPUNIT_ASSERT_IMAGES
70
71   HYDROData_LandCoverMap::Explorer anIt( aMap );
72   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
73   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
74   anIt.Next();
75   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
76   CPPUNIT_ASSERT_EQUAL( QString( "test2" ), anIt.StricklerType() );
77   anIt.Next();
78   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
79
80   aDoc->Close();
81 }
82
83 void test_HYDROData_LandCoverMap::test_split()
84 {
85   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
86
87   Handle(HYDROData_LandCoverMap) aMap =
88     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
89
90   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
91
92   TopoDS_Face aLC = Face( QList<double>() << 10 << 10 << 50 << 20 << 30 << 50 << 15 << 30 );
93   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC, "test1" ) );
94
95   Handle(HYDROData_PolylineXY) aPolyline =
96     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
97   TopoDS_Wire aWire = Wire( QList<double>() << 10 << 40 << 30 << 10 << 40 << 10 << 60 << 10, false );
98   aPolyline->SetShape( aWire );
99
100   CPPUNIT_ASSERT_EQUAL( true, aMap->Split( aPolyline ) );
101
102   TestViewer::show( aMap->GetShape(), AIS_Shaded, true, "LandCoverMap_Split_1" );
103   TestViewer::show( aWire, 0, true, Qt::blue );
104   CPPUNIT_ASSERT_IMAGES
105
106   HYDROData_LandCoverMap::Explorer anIt( aMap );
107   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
108   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
109   anIt.Next();
110   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
111   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
112   anIt.Next();
113   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
114
115   aDoc->Close();
116 }
117
118 void test_HYDROData_LandCoverMap::test_incomplete_split()
119 {
120   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
121
122   Handle(HYDROData_LandCoverMap) aMap =
123     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
124
125   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
126
127   TopoDS_Face aLC = Face( QList<double>() << 10 << 10 << 50 << 20 << 30 << 50 << 15 << 30 );
128   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC, "test1" ) );
129
130   Handle(HYDROData_PolylineXY) aPolyline =
131     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
132   TopoDS_Wire aWire = Wire( QList<double>() << 10 << 40 << 30 << 10 << 40 << 10, false );
133   aPolyline->SetShape( aWire );
134
135   CPPUNIT_ASSERT_EQUAL( true, aMap->Split( aPolyline ) );
136
137   TestViewer::show( aMap->GetShape(), AIS_Shaded, true, "LandCoverMap_Split_2" );
138   TestViewer::show( aWire, 0, true, Qt::green );
139   CPPUNIT_ASSERT_IMAGES
140
141   HYDROData_LandCoverMap::Explorer anIt( aMap );
142   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
143   CPPUNIT_ASSERT_EQUAL( QString( "test1" ), anIt.StricklerType() );
144   anIt.Next();
145   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
146
147   aDoc->Close();
148 }
149
150 void test_HYDROData_LandCoverMap::test_merge()
151 {
152   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
153
154   Handle(HYDROData_LandCoverMap) aMap =
155     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
156
157   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
158
159   TopoDS_Face aLC1 = Face( QList<double>() << 12 << 19 << 82 << 9 << 126 << 53 << 107 << 80 << 29 << 75 );
160
161   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "test1" ) );
162
163   TopoDS_Face aLC2 = Face( QList<double>() << 21 << 34 << 24 << 25 << 37   << 37 << 40  << 61 <<
164                                               44 << 95 << 85 << 100 << 104 << 66 << 107 << 33 <<
165                                              128 << 18 << 140 << 50 << 131 << 89 << 104 << 111 <<
166                                               31 << 114 );
167   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "test2" ) );
168
169   TopoDS_Face aLC3 = Face( QList<double>() << 4 << 54 << 1   << 47 << 51  << 45 <<
170                                             127 << 42 << 145 << 43 << 148 << 60 << 90 << 65 );
171   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC3, "test3" ) );
172
173   //TestViewer::show( aMap->GetShape(), AIS_Shaded, true );
174   //TestViewer::AssertEqual( "LandCoverMap_Before_Merge" );
175
176   //BRepTools::Write( aMap->GetShape(), "c:\\map.brep" );
177
178   QString aType1, aType2;
179   gp_Pnt2d aPnt1( 25, 35 );
180   gp_Pnt2d aPnt2( 45, 55 );
181   //gp_Pnt2d aPnt3( 45, 10 );
182   //gp_Pnt2d aPnt4( 45, 80 );
183   TopTools_ListOfShape aList;
184   aList.Append( aMap->FindByPoint( aPnt1, aType1 ) );
185   aList.Append( aMap->FindByPoint( aPnt2, aType2 ) );
186   CPPUNIT_ASSERT_EQUAL( true, aMap->Merge( aList, "new" ) );
187
188   TestViewer::show( aMap->GetShape(), AIS_Shaded, true, "LandCoverMap_Merge_1" );
189   //TestViewer::show( BRepBuilderAPI_MakeEdge( gp_Pnt(aPnt1.X(), aPnt1.Y(), 0), gp_Pnt(aPnt2.X(), aPnt2.Y(), 0) ).Edge(), QColor( Qt::blue ), AIS_Shaded );
190   CPPUNIT_ASSERT_IMAGES
191
192   aDoc->Close();
193 }
194
195 void test_HYDROData_LandCoverMap::test_merge_refs_691()
196 {
197   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
198
199   Handle(HYDROData_PolylineXY) Polyline_4 =
200     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
201   Polyline_4->SetName( "test_Polyline_4" );
202   Polyline_4->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
203   Polyline_4->AddPoint( 0, gp_XY( -242.741935, -16.129032 ) );
204   Polyline_4->AddPoint( 0, gp_XY( -339.516129, -95.161290 ) );
205   Polyline_4->AddPoint( 0, gp_XY( -204.032258, -167.740000 ) );
206   Polyline_4->AddPoint( 0, gp_XY( -86.290000, -158.060000 ) );
207   Polyline_4->AddPoint( 0, gp_XY( -83.064516, -37.096774 ) );
208   Polyline_4->AddPoint( 0, gp_XY( -116.935484, 17.741935 ) );
209   Polyline_4->AddSection( "Section_2", HYDROData_PolylineXY::SECTION_POLYLINE, true );
210   Polyline_4->AddPoint( 1, gp_XY( -239.516129, -295.161290 ) );
211   Polyline_4->AddPoint( 1, gp_XY( -204.030000, -167.740000 ) );
212   Polyline_4->AddPoint( 1, gp_XY( -86.290000, -158.060000 ) );
213   Polyline_4->AddPoint( 1, gp_XY( -2.419355, -288.709677 ) );
214   Polyline_4->AddPoint( 1, gp_XY( -165.322581, -351.612903 ) );
215   Polyline_4->Update();
216
217   Handle(HYDROData_LandCoverMap) aMap =
218     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
219
220   CPPUNIT_ASSERT_EQUAL( true, aMap->Add( Polyline_4, "test" ) );
221
222   TopTools_ListOfShape aList;
223   HYDROData_LandCoverMap::Explorer anIt( aMap );
224   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
225   CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt.StricklerType() );
226   aList.Append( anIt.Face() );
227   anIt.Next();
228   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
229   CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt.StricklerType() );
230   aList.Append( anIt.Face() );
231   anIt.Next();
232   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
233
234   BRepCheck_Analyzer aCheck1( aMap->GetShape() );
235   CPPUNIT_ASSERT_EQUAL( true, (bool)aCheck1.IsValid() );
236   
237   CPPUNIT_ASSERT_EQUAL( true, aMap->Merge( aList, "test" ) );
238
239   BRepCheck_Analyzer aCheck2( aMap->GetShape() );
240   CPPUNIT_ASSERT_EQUAL( true, (bool)aCheck2.IsValid() );
241
242   // we try construct the triangulation to check that it is correct
243   BRepMesh_IncrementalMesh anIncMesh( aMap->GetShape(), 0.01 );
244   CPPUNIT_ASSERT_EQUAL( true, (bool)anIncMesh.IsDone() );
245   BRepCheck_Analyzer aCheck3( aMap->GetShape() );
246   CPPUNIT_ASSERT_EQUAL( true, (bool)aCheck3.IsValid() );
247
248   Handle(HYDROGUI_LandCoverMapPrs) aPrs = new HYDROGUI_LandCoverMapPrs( aMap );
249   TestViewer::show( aPrs, AIS_Shaded, 0, true, "LandCoverMap_Merge_2" );
250   CPPUNIT_ASSERT_IMAGES
251
252   aDoc->Close();
253 }
254
255 void test_HYDROData_LandCoverMap::test_remove()
256 {
257   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
258
259   Handle(HYDROData_LandCoverMap) aMap =
260     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
261
262   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
263
264   TopoDS_Face aLC1 = Face( QList<double>() << 12 << 19 << 82 << 9 << 126 << 53 << 107 << 80 << 29 << 75 );
265
266   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "test1" ) );
267
268   TopoDS_Face aLC2 = Face( QList<double>() << 21 << 34 << 24 << 25 << 37   << 37 << 40  << 61 <<
269                                               44 << 95 << 85 << 100 << 104 << 66 << 107 << 33 <<
270                                              128 << 18 << 140 << 50 << 131 << 89 << 104 << 111 <<
271                                               31 << 114 );
272   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "test2" ) );
273
274   TopoDS_Face aLC3 = Face( QList<double>() << 4 << 54 << 1   << 47 << 51  << 45 <<
275                                             127 << 42 << 145 << 43 << 148 << 60 << 90 << 65 );
276   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC3, "test3" ) );
277
278   QString aType1, aType2;
279   gp_Pnt2d aPnt1( 25, 35 );
280   gp_Pnt2d aPnt2( 45, 55 );
281   TopTools_ListOfShape aList;
282   aList.Append( aMap->FindByPoint( aPnt1, aType1 ) );
283   aList.Append( aMap->FindByPoint( aPnt2, aType2 ) );
284   CPPUNIT_ASSERT_EQUAL( true, aMap->Remove( aList ) );
285
286   TestViewer::show( aMap->GetShape(), AIS_Shaded, true, "LandCoverMap_Remove_1" );
287   CPPUNIT_ASSERT_IMAGES
288
289   aDoc->Close();
290 }
291
292 void test_HYDROData_LandCoverMap::test_merge_faces_boxes()
293 {
294   TopoDS_Shape pp1, pp2, pp3, pp4;
295   BRep_Builder BB;
296   BRepTools::Read(pp1, (REF_DATA_PATH + "/pp1.brep").toStdString().c_str(), BB);
297   BRepTools::Read(pp2, (REF_DATA_PATH + "/pp2.brep").toStdString().c_str(), BB);
298   BRepTools::Read(pp3, (REF_DATA_PATH + "/pp3.brep").toStdString().c_str(), BB);
299   BRepTools::Read(pp4, (REF_DATA_PATH + "/pp4.brep").toStdString().c_str(), BB);
300
301   CPPUNIT_ASSERT(!pp1.IsNull());
302   CPPUNIT_ASSERT(!pp2.IsNull());
303   CPPUNIT_ASSERT(!pp3.IsNull());
304   CPPUNIT_ASSERT(!pp4.IsNull());
305
306   //Test mergeFaces() func // boxes // USD == true
307   {
308     TopTools_ListOfShape Faces;
309     Faces.Append(pp1);
310     Faces.Append(pp2);
311     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
312     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_11" );
313     CPPUNIT_ASSERT_IMAGES
314     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_FACE);
315     BRepCheck_Analyzer aBCA(aMergedFace);
316     CPPUNIT_ASSERT(aBCA.IsValid());
317   }
318   {
319     TopTools_ListOfShape Faces;
320     Faces.Append(pp1);
321     Faces.Append(pp3);
322     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
323     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_12" );
324     CPPUNIT_ASSERT_IMAGES
325     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
326     BRepCheck_Analyzer aBCA(aMergedFace);
327     CPPUNIT_ASSERT(aBCA.IsValid());
328   }
329   {
330     TopTools_ListOfShape Faces;
331     Faces.Append(pp1);
332     Faces.Append(pp2);
333     Faces.Append(pp4);
334     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
335     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_13" );
336     CPPUNIT_ASSERT_IMAGES
337     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
338     BRepCheck_Analyzer aBCA(aMergedFace);
339     CPPUNIT_ASSERT(aBCA.IsValid());
340   }
341   {
342     TopTools_ListOfShape Faces;
343     Faces.Append(pp1);
344     Faces.Append(pp3);
345     Faces.Append(pp4);
346     Faces.Append(pp2);
347     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
348     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_14" );
349     CPPUNIT_ASSERT_IMAGES
350     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_FACE);
351     BRepCheck_Analyzer aBCA(aMergedFace);
352     CPPUNIT_ASSERT(aBCA.IsValid());
353   }
354   //
355
356   //Test mergeFaces() func // boxes // USD == false
357   {
358     TopTools_ListOfShape Faces;
359     Faces.Append(pp1);
360     Faces.Append(pp2);
361     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
362     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_21" );
363     CPPUNIT_ASSERT_IMAGES
364     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_SHELL);
365     BRepCheck_Analyzer aBCA(aMergedFace);
366     CPPUNIT_ASSERT(aBCA.IsValid());
367   }
368   {
369     TopTools_ListOfShape Faces;
370     Faces.Append(pp1);
371     Faces.Append(pp3);
372     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
373     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_22" );
374     CPPUNIT_ASSERT_IMAGES
375     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
376     BRepCheck_Analyzer aBCA(aMergedFace);
377     CPPUNIT_ASSERT(aBCA.IsValid());
378   }
379   {
380     TopTools_ListOfShape Faces;
381     Faces.Append(pp1);
382     Faces.Append(pp2);
383     Faces.Append(pp4);
384     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
385     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_23" );
386     CPPUNIT_ASSERT_IMAGES
387     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
388     BRepCheck_Analyzer aBCA(aMergedFace);
389     CPPUNIT_ASSERT(aBCA.IsValid());
390   }
391   {
392     TopTools_ListOfShape Faces;
393     Faces.Append(pp1);
394     Faces.Append(pp3);
395     Faces.Append(pp4);
396     Faces.Append(pp2);
397     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
398     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_boxes_24" );
399     CPPUNIT_ASSERT_IMAGES
400     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_SHELL);
401     BRepCheck_Analyzer aBCA(aMergedFace);
402     CPPUNIT_ASSERT(aBCA.IsValid());
403   }
404 }
405
406
407 void test_HYDROData_LandCoverMap::test_merge_faces_circles()
408 {
409   TopoDS_Shape ff1, ff2, ff3;
410   BRep_Builder BB;
411   BRepTools::Read(ff1, (REF_DATA_PATH + "/ff1.brep").toStdString().c_str(), BB);
412   BRepTools::Read(ff2, (REF_DATA_PATH + "/ff2.brep").toStdString().c_str(), BB);
413   BRepTools::Read(ff3, (REF_DATA_PATH + "/ff3.brep").toStdString().c_str(), BB);
414
415   CPPUNIT_ASSERT(!ff1.IsNull());
416   CPPUNIT_ASSERT(!ff2.IsNull());
417   CPPUNIT_ASSERT(!ff3.IsNull());
418
419   //Test mergeFaces() func // circles // USD == true
420   { 
421     TopTools_ListOfShape Faces;
422     Faces.Append(ff1);
423     Faces.Append(ff2);
424     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
425     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_11" );
426     CPPUNIT_ASSERT_IMAGES
427     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_FACE);
428     BRepCheck_Analyzer aBCA(aMergedFace);
429     CPPUNIT_ASSERT(aBCA.IsValid());
430   }
431   {
432     TopTools_ListOfShape Faces;
433     Faces.Append(ff1);
434     Faces.Append(ff3);
435     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
436     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_12" );
437     CPPUNIT_ASSERT_IMAGES
438     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
439     BRepCheck_Analyzer aBCA(aMergedFace);
440     CPPUNIT_ASSERT(aBCA.IsValid());
441   }
442   {
443     TopTools_ListOfShape Faces;
444     Faces.Append(ff1);
445     Faces.Append(ff2);
446     Faces.Append(ff3);
447     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
448     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_13" );
449     CPPUNIT_ASSERT_IMAGES
450     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
451     BRepCheck_Analyzer aBCA(aMergedFace);
452     CPPUNIT_ASSERT(aBCA.IsValid());
453   }
454   {
455     TopTools_ListOfShape Faces;
456     Faces.Append(ff1);
457     Faces.Append(ff3);
458     Faces.Append(ff2);
459     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, true );
460     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_14" );
461     CPPUNIT_ASSERT_IMAGES
462     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
463     BRepCheck_Analyzer aBCA(aMergedFace);
464     CPPUNIT_ASSERT(aBCA.IsValid());
465   }
466   //
467
468   //Test mergeFaces() func // circles // USD == false
469   {
470     TopTools_ListOfShape Faces;
471     Faces.Append(ff1);
472     Faces.Append(ff2);
473     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
474     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_21" );
475     CPPUNIT_ASSERT_IMAGES
476     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_SHELL);
477     BRepCheck_Analyzer aBCA(aMergedFace);
478     CPPUNIT_ASSERT(aBCA.IsValid());
479   }
480   {
481     TopTools_ListOfShape Faces;
482     Faces.Append(ff1);
483     Faces.Append(ff3);
484     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
485     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_22" );
486     CPPUNIT_ASSERT_IMAGES
487     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
488     BRepCheck_Analyzer aBCA(aMergedFace);
489     CPPUNIT_ASSERT(aBCA.IsValid());
490   }
491   {
492     TopTools_ListOfShape Faces;
493     Faces.Append(ff1);
494     Faces.Append(ff2);
495     Faces.Append(ff3);
496     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
497     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_23" );
498     CPPUNIT_ASSERT_IMAGES
499     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
500     BRepCheck_Analyzer aBCA(aMergedFace);
501     CPPUNIT_ASSERT(aBCA.IsValid());
502   }
503   {
504     TopTools_ListOfShape Faces;
505     Faces.Append(ff1);
506     Faces.Append(ff3);
507     Faces.Append(ff2);
508     TopoDS_Shape aMergedFace = HYDROData_LandCoverMap::MergeFaces( Faces, false );
509     TestViewer::show( aMergedFace, AIS_Shaded, true, "merge_faces_circles_24" );
510     CPPUNIT_ASSERT_IMAGES
511     CPPUNIT_ASSERT(aMergedFace.ShapeType() == TopAbs_COMPOUND);
512     BRepCheck_Analyzer aBCA(aMergedFace);
513     CPPUNIT_ASSERT(aBCA.IsValid());
514   }
515 }
516
517 void test_HYDROData_LandCoverMap::test_import_dbf()
518 {
519   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
520
521   Handle(HYDROData_LandCoverMap) aMap =
522     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
523
524   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
525   QString aFileName = REF_DATA_PATH + "/t1.dbf";
526   QStringList aDBFV;
527   QStringList aST;
528   aDBFV.append("100");
529   aDBFV.append("200");
530   aDBFV.append("300");
531   aST.append("water");
532   aST.append("forest");
533   aST.append("road");
534   // aST.size() == aDBFV.size()!!
535   Handle(HYDROData_PolylineXY) aPolyline =
536     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
537   TopoDS_Wire aWire = Wire( QList<double>() << 10 << 40 << 30 << 10 << 40 << 10 << 60 << 10 );
538   aPolyline->SetShape( aWire );
539   for (int i = 0; i < 3; i++)
540     aMap->Add(aPolyline, "");
541   TopoDS_Face aLC1 = Face( QList<double>() << 10 << 10 << 50 << 20 << 30 << 50 << 15 << 30 );
542   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "test1" ) );
543   
544   CPPUNIT_ASSERT_EQUAL( true, aMap->Split( aPolyline ) );
545   QList<int> Inds = QList<int>() << 1 << 2 << 3;
546   aMap->ImportDBF(aFileName, "TESTFIELD1", aDBFV, aST, Inds);
547   HYDROData_LandCoverMap::Explorer anIt( aMap );
548   CPPUNIT_ASSERT_EQUAL( QString( "forest" ), anIt.StricklerType() );
549   anIt.Next();
550   CPPUNIT_ASSERT_EQUAL( QString( "road" ), anIt.StricklerType() );
551
552   aDoc->Close();
553 }
554
555 void test_HYDROData_LandCoverMap::test_land_cover_prs_by_types()
556 {
557   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
558
559   Handle(HYDROData_StricklerTable) aTable =
560     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
561   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
562
563   Handle(HYDROData_LandCoverMap) aMap =
564     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
565
566   TopoDS_Face aLC1 = Face( QList<double>() << 12 << 19 << 82 << 9 << 126 << 53 << 107 << 80 << 29 << 75 );
567   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "Zones de champs cultivé à végétation basse" ) );
568
569   TopoDS_Face aLC2 = Face( QList<double>() << 21 << 34 << 24 << 25 << 37   << 37 << 40  << 61 <<
570                                               44 << 95 << 85 << 100 << 104 << 66 << 107 << 33 <<
571                                              128 << 18 << 140 << 50 << 131 << 89 << 104 << 111 <<
572                                               31 << 114 );
573   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "Zones de champs cultivé à végétation haute" ) );
574
575   TopoDS_Face aLC3 = Face( QList<double>() << 4 << 54 << 1   << 47 << 51  << 45 <<
576                                             127 << 42 << 145 << 43 << 148 << 60 << 90 << 65 );
577   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC3, "Zones de champs, prairies, sans cultures" ) );
578
579   // build presentation object 
580   Handle(HYDROGUI_LandCoverMapPrs) aPrs = new HYDROGUI_LandCoverMapPrs( aMap );
581   aPrs->SetTable( aTable );
582   // show presentation in viewer
583   TestViewer::show( aPrs, AIS_Shaded, 4, true, "LandCoverMap_PrsByTypes" );
584   // select one of faces (first)
585   TestViewer::select( 200, 300 );
586   CPPUNIT_ASSERT_IMAGES
587
588   aDoc->Close();
589 }
590
591 void test_HYDROData_LandCoverMap::test_land_cover_prs_by_coeff()
592 {
593   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
594
595   Handle(HYDROData_StricklerTable) aTable =
596     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
597   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
598
599   Handle(HYDROData_LandCoverMap) aMap =
600     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
601
602   TopoDS_Face aLC1 = Face( QList<double>() << 12 << 19 << 82 << 9 << 126 << 53 << 107 << 80 << 29 << 75 );
603   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "Zones de champs cultivé à végétation basse" ) );
604
605   TopoDS_Face aLC2 = Face( QList<double>() << 21 << 34 << 24 << 25 << 37   << 37 << 40  << 61 <<
606                                               44 << 95 << 85 << 100 << 104 << 66 << 107 << 33 <<
607                                              128 << 18 << 140 << 50 << 131 << 89 << 104 << 111 <<
608                                               31 << 114 );
609   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "Zones de champs cultivé à végétation haute" ) );
610
611   TopoDS_Face aLC3 = Face( QList<double>() << 4 << 54 << 1   << 47 << 51  << 45 <<
612                                             127 << 42 << 145 << 43 << 148 << 60 << 90 << 65 );
613   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC3, "Zones de champs, prairies, sans cultures" ) );
614
615   Handle(HYDROGUI_LandCoverMapPrs) aPrs = new HYDROGUI_LandCoverMapPrs( aMap );
616   aPrs->SetTable( aTable );
617   TestViewer::showColorScale( true );
618   aPrs->SetColorScale( TestViewer::colorScale() );
619   TestViewer::show( aPrs, AIS_Shaded, 4, true, "LandCoverMap_PrsByCoeff" );
620   CPPUNIT_ASSERT_IMAGES
621
622   TestViewer::showColorScale( false );
623   aDoc->Close();
624 }
625
626 void test_HYDROData_LandCoverMap::test_dump_python()
627 {
628   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
629
630   Handle(HYDROData_StricklerTable) aTable =
631     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
632   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
633   aTable->SetName( "DefStr" );
634
635   Handle(HYDROData_LandCoverMap) aMap =
636     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
637   aMap->SetName( "test_LCM" );
638
639   CPPUNIT_ASSERT_EQUAL( KIND_LAND_COVER_MAP, aMap->GetKind() );
640
641   TopoDS_Face aLC1 = Face( QList<double>() << 12 << 19 << 82 << 9 << 126 << 53 << 107 << 80 << 29 << 75 );
642
643   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "test1" ) );
644
645   TopoDS_Face aLC2 = Face( QList<double>() << 21 << 34 << 24 << 25 << 37   << 37 << 40  << 61 <<
646                                               44 << 95 << 85 << 100 << 104 << 66 << 107 << 33 <<
647                                              128 << 18 << 140 << 50 << 131 << 89 << 104 << 111 <<
648                                               31 << 114 );
649   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "test2" ) );
650
651   TopoDS_Face aLC3 = Face( QList<double>() << 4 << 54 << 1   << 47 << 51  << 45 <<
652                                             127 << 42 << 145 << 43 << 148 << 60 << 90 << 65 );
653   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC3, "test3" ) );
654
655   QString aBaseName = "lc_dump.py";
656   QString aBaseShp = "lc_dump.shp";
657   QString aBaseDbf = "lc_dump.dbf";
658   QString aTmpPath = QDir::tempPath() + "/" + aBaseName;
659   CPPUNIT_ASSERT_EQUAL( true, aDoc->DumpToPython( aTmpPath, false ) );
660
661   //TODO: CPPUNIT_ASSERT_SCRIPTS_EQUAL( aBaseName, true, false, 20 );
662   //TODO: CPPUNIT_ASSERT_FILES_EQUAL( aBaseShp );
663   //TODO: CPPUNIT_ASSERT_FILES_EQUAL( aBaseDbf );
664
665   aDoc->Close();
666 }
667
668 void test_HYDROData_LandCoverMap::test_transparent_prs()
669 {
670   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
671
672   Handle(HYDROData_StricklerTable) aTable =
673     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
674   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
675
676   Handle(HYDROData_PolylineXY) aPoly =
677     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
678   aPoly->AddSection( "", HYDROData_PolylineXY::SECTION_SPLINE, true );
679   aPoly->AddPoint( 0, gp_XY( 0, 0 ) );
680   aPoly->AddPoint( 0, gp_XY( 20, 0 ) );
681   aPoly->AddPoint( 0, gp_XY( 10, 10 ) );
682   aPoly->Update();
683
684   Handle(HYDROData_ImmersibleZone) aZone =
685     Handle(HYDROData_ImmersibleZone)::DownCast( aDoc->CreateObject( KIND_IMMERSIBLE_ZONE ) );
686   aZone->SetPolyline( aPoly );
687   aZone->Update();
688
689   Handle(HYDROData_LandCoverMap) aMap =
690     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
691   aMap->LocalPartition( Face( QList<double>() << 1 << 1 << 10 << 10 << 10 << 20 ), "Zones de champs cultivé à végétation haute" );
692   aMap->LocalPartition( Face( QList<double>() << 5 << 5 << 10 << 5 << 10 << 8 << 5 << 12 << 5 << 8 ), "Zones de champs cultivé à végétation haute" );
693   aMap->SetName( "test_LCM" );
694
695   TestViewer::show( aZone->GetTopShape(), AIS_Shaded, true, "LandCoverMap_TransparentPrs" );
696
697   aMap->SetTransparency( 0.5 );
698   CPPUNIT_ASSERT_DOUBLES_EQUAL( 0.5, aMap->GetTransparency(), 1E-6 );
699
700   Handle(HYDROGUI_LandCoverMapPrs) aPrs = new HYDROGUI_LandCoverMapPrs( aMap );
701   aPrs->SetTable( aTable );
702   TestViewer::show( aPrs, AIS_Shaded, 0, true, "" );
703
704   CPPUNIT_ASSERT_IMAGES
705   aDoc->Close();
706 }
707
708 void test_HYDROData_LandCoverMap::test_add_triangles()
709 {
710   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
711
712   Handle(HYDROData_PolylineXY) Polyline_1 =
713     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
714
715   Polyline_1->SetName( "Polyline_1" );
716   Polyline_1->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
717   Polyline_1->AddPoint( 0, gp_XY( -264.84, 323.44 ) );
718   Polyline_1->AddPoint( 0, gp_XY( 254.45, 301.19 ) );
719   Polyline_1->AddPoint( 0, gp_XY( -291.54, -47.48 ) );
720   Polyline_1->Update();
721
722   Handle(HYDROData_PolylineXY) Polyline_2 =
723     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
724   Polyline_2->SetName( "Polyline_2" );
725   Polyline_2->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, false );
726   Polyline_2->AddPoint( 0, gp_XY( -42.28, -351.63 ) );
727   Polyline_2->AddPoint( 0, gp_XY( 218.84, -146.88 ) );
728   Polyline_2->AddPoint( 0, gp_XY( 413.20, -235.91 ) );
729   Polyline_2->AddPoint( 0, gp_XY( 466.62, -388.72 ) );
730   Polyline_2->Update();
731
732   Handle(HYDROData_PolylineXY) Polyline_3 =
733     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
734   Polyline_3->SetName( "Polyline_3" );
735   Polyline_3->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
736   Polyline_3->AddPoint( 0, gp_XY( -108.31, 213.65 ) );
737   Polyline_3->AddPoint( 0, gp_XY( 127.60, 382.79 ) );
738   Polyline_3->AddPoint( 0, gp_XY( 192.88, -91.99 ) );
739   Polyline_3->Update();
740
741   Handle(HYDROData_PolylineXY) Polyline_4 =
742     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
743   Polyline_4->SetName( "Polyline_4" );
744   Polyline_4->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
745   Polyline_4->AddPoint( 0, gp_XY( 215.52, 344.61 ) );
746   Polyline_4->AddPoint( 0, gp_XY( 68.25, 258.52 ) );
747   Polyline_4->AddPoint( 0, gp_XY( 200.58, 154.65 ) );
748   Polyline_4->Update();
749
750   Handle(HYDROData_PolylineXY) Polyline_5 =
751     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
752   Polyline_5->SetName( "Polyline_5" );
753   Polyline_5->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, false );
754   Polyline_5->AddPoint( 0, gp_XY( 16.15, -58.08 ) );
755   Polyline_5->AddPoint( 0, gp_XY( 116.47, 84.21 ) );
756   Polyline_5->AddPoint( 0, gp_XY( 266.59, 29.43 ) );
757   Polyline_5->Update();
758
759   Handle(HYDROData_LandCoverMap) aMap =
760     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
761
762   CPPUNIT_ASSERT_EQUAL( true, aMap->Add( Polyline_1, "" ) );
763   CPPUNIT_ASSERT_EQUAL( true, aMap->Add( Polyline_3, "" ) );
764   CPPUNIT_ASSERT_EQUAL( true, aMap->Add( Polyline_4, "" ) );
765
766   Handle(HYDROGUI_LandCoverMapPrs) aPrs = new HYDROGUI_LandCoverMapPrs( aMap );
767   TopoDS_Shape aSh = aMap->GetShape();
768   TestViewer::show( aPrs, AIS_Shaded, 0, true, "LandCover_Triangles" );
769   CPPUNIT_ASSERT_IMAGES
770
771   CPPUNIT_ASSERT_EQUAL( true, aMap->Split( Polyline_5 ) );
772   Handle(HYDROGUI_LandCoverMapPrs) aPrs2 = new HYDROGUI_LandCoverMapPrs( aMap );
773   TestViewer::show( aPrs2, AIS_Shaded, 0, true, "LandCover_Triangles_Split" );
774   CPPUNIT_ASSERT_IMAGES
775
776   aDoc->Close();
777 }
778
779 void test_HYDROData_LandCoverMap::test_assign_to_calc_case()
780 {
781   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
782
783   Handle(HYDROData_LandCoverMap) aMap =
784     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
785   Handle(HYDROData_CalculationCase) aCalcCase =
786     Handle(HYDROData_CalculationCase)::DownCast( aDoc->CreateObject( KIND_CALCULATION ) );
787
788   aCalcCase->SetLandCoverMap( aMap );
789   CPPUNIT_ASSERT_EQUAL( aMap->myLab, aCalcCase->GetLandCoverMap()->myLab );
790
791   aDoc->Close();
792 }
793
794 void test_HYDROData_LandCoverMap::test_shp_import_cyp()
795 {
796   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
797   QString aFileName = REF_DATA_PATH + "/cyprus_natural.shp";
798   HYDROData_ShapeFile anImporter;
799   QStringList PolygonList;
800   TopTools_SequenceOfShape PolygonFaces;
801   int Type = -1;
802   int aStat = anImporter.ImportPolygons(aDoc, aFileName, PolygonList, PolygonFaces, Type);
803   CPPUNIT_ASSERT(aStat == 1);
804   CPPUNIT_ASSERT_EQUAL(5, Type);
805   CPPUNIT_ASSERT_EQUAL(268, PolygonFaces.Length());
806   
807   Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
808   HYDROData_MapOfFaceToStricklerType aMapFace2ST;
809
810   for ( int i = 1; i <= PolygonFaces.Length(); i++ )
811   {
812     TopoDS_Shape aShape = PolygonFaces(i);
813     if ( aShape.IsNull() ) 
814       continue;
815     aMapFace2ST.Add( TopoDS::Face( aShape ), "" );
816   }
817
818   LCM->StoreLandCovers(aMapFace2ST);
819   TestViewer::show( LCM->GetShape(), AIS_Shaded, true, "cyprus_natural_all" );
820   CPPUNIT_ASSERT_IMAGES
821   aDoc->Close();
822 }
823
824 void test_HYDROData_LandCoverMap::test_shp_import_clc_dec()
825 {
826   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
827   QString aFileName = REF_DATA_PATH + "/CLC_decoupe.shp";
828   HYDROData_ShapeFile anImporter;
829   QStringList PolygonList;
830   TopTools_SequenceOfShape PolygonFaces;
831   int Type = -1;
832   int aStat = anImporter.ImportPolygons(aDoc, aFileName, PolygonList, PolygonFaces, Type);
833   CPPUNIT_ASSERT(aStat == 1);
834   CPPUNIT_ASSERT_EQUAL(5, Type);
835   CPPUNIT_ASSERT_EQUAL(625, PolygonFaces.Length());
836   
837   Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
838   HYDROData_MapOfFaceToStricklerType aMapFace2ST;
839
840   for ( int i = 150; i <= 350; i++ )
841   {
842     TopoDS_Shape aShape = PolygonFaces(i);
843     if ( aShape.IsNull() ) 
844       continue;
845     aMapFace2ST.Add( TopoDS::Face( aShape ), "" );
846   }
847
848   LCM->StoreLandCovers(aMapFace2ST);
849   TestViewer::show( LCM->GetShape(), AIS_Shaded, true, "clc_dec_150_350" );
850   CPPUNIT_ASSERT_IMAGES
851   aDoc->Close();
852 }
853
854 void test_HYDROData_LandCoverMap::test_shp_import_nld_areas()
855 {
856   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
857   QString aFileName = REF_DATA_PATH + "/NLD_water_areas_dcw.shp";
858   HYDROData_ShapeFile anImporter;
859   QStringList PolygonList;
860   TopTools_SequenceOfShape PolygonFaces;
861   int Type = -1;
862   int aStat = anImporter.ImportPolygons(aDoc, aFileName, PolygonList, PolygonFaces, Type);
863   CPPUNIT_ASSERT(aStat == 1);
864   CPPUNIT_ASSERT_EQUAL(5, Type);
865   CPPUNIT_ASSERT_EQUAL(127, PolygonFaces.Length());
866   
867   Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
868   HYDROData_MapOfFaceToStricklerType aMapFace2ST;
869
870   for ( int i = 1; i <= PolygonFaces.Length(); i++ )
871   {
872     TopoDS_Shape aShape = PolygonFaces(i);
873     if ( aShape.IsNull() ) 
874       continue;
875     aMapFace2ST.Add( TopoDS::Face( aShape ), "" );
876   }
877
878   LCM->StoreLandCovers(aMapFace2ST);
879   TestViewer::show( LCM->GetShape(), AIS_Shaded, true, "NLD_water_areas_dcw" );
880   CPPUNIT_ASSERT_IMAGES
881   aDoc->Close();
882 }
883
884 void test_HYDROData_LandCoverMap::test_shp_loop_back()
885 {
886   QString aFFileName = REF_DATA_PATH + "/CLC06-cut_1.shp";
887   QString aSFileName = REF_DATA_PATH + "/CLC06-cut_1_res.shp";
888
889   //This test verify only some geom data (shp+shx) without any dbf reading/writing 
890   {
891     DEBTRACE("aFFileName: " << aFFileName.toStdString());
892     DEBTRACE("aSFileName: " << aSFileName.toStdString());
893     Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
894     HYDROData_ShapeFile anImporter;
895     QStringList PolygonList;
896     TopTools_SequenceOfShape PolygonFaces;
897     int Type = -1;
898     DEBTRACE("---");
899
900     //import LCM from file (#2-4; #12-14 polygons)
901     CPPUNIT_ASSERT( anImporter.ImportPolygons(aDoc, aFFileName, PolygonList, PolygonFaces, Type));
902     Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
903     HYDROData_MapOfFaceToStricklerType aMapFace2ST;
904
905     aMapFace2ST.Add( TopoDS::Face( PolygonFaces(2) ), "" );
906     aMapFace2ST.Add( TopoDS::Face( PolygonFaces(3) ), "" );
907     aMapFace2ST.Add( TopoDS::Face( PolygonFaces(4) ), "" );
908     aMapFace2ST.Add( TopoDS::Face( PolygonFaces(12) ), "" );
909     aMapFace2ST.Add( TopoDS::Face( PolygonFaces(13) ), "" );
910     aMapFace2ST.Add( TopoDS::Face( PolygonFaces(14) ), "" );
911     DEBTRACE("---");
912
913     LCM->StoreLandCovers(aMapFace2ST);
914     LCM->SetName("lcm_1");
915
916     TopoDS_Shape aSh = LCM->GetShape();
917     BRepCheck_Analyzer aBCA(aSh);
918     CPPUNIT_ASSERT(aBCA.IsValid());
919
920     //export lcm
921     HYDROData_ShapeFile anExporter;
922     QStringList aNonExpList;
923     anExporter.Export(aSFileName, LCM, aNonExpList);
924     CPPUNIT_ASSERT (aNonExpList.empty());
925     DEBTRACE("---");
926
927     aDoc->Close();
928   }
929   {
930     Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
931     HYDROData_ShapeFile anImporter;
932     QStringList PolygonList;
933     TopTools_SequenceOfShape PolygonFaces;
934     int Type = -1;
935
936     //import all
937     CPPUNIT_ASSERT( anImporter.ImportPolygons(aDoc, aSFileName, PolygonList, PolygonFaces, Type));
938     Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
939     HYDROData_MapOfFaceToStricklerType aMapFace2ST;
940     CPPUNIT_ASSERT_EQUAL(PolygonFaces.Length(), 6);
941     for ( int i = 1; i <= PolygonFaces.Length(); i++ )
942       aMapFace2ST.Add( TopoDS::Face( PolygonFaces(i) ), "" );
943
944     LCM->StoreLandCovers(aMapFace2ST);
945     LCM->SetName("lcm_2");
946
947     TopoDS_Shape OutSh = LCM->GetShape();
948     BRepCheck_Analyzer aBCA(OutSh);
949     CPPUNIT_ASSERT(aBCA.IsValid());
950
951     TestViewer::show( OutSh, AIS_Shaded, true, "CLC06-cut_1_res" );
952     CPPUNIT_ASSERT_IMAGES
953
954     aDoc->Close();
955   }
956   CPPUNIT_ASSERT_EQUAL(0, remove(aSFileName.toStdString().c_str()));
957 }
958
959 void test_HYDROData_LandCoverMap::test_add_2_section_poly()
960 {
961   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
962
963   Handle(HYDROData_PolylineXY) Polyline_12 =
964     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
965   Polyline_12->SetName( "Polyline_12" );
966
967   Polyline_12->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
968   Polyline_12->AddPoint( 0, gp_XY( -148.70, 145.68 ) );
969   Polyline_12->AddPoint( 0, gp_XY( -81.71, 210.97 ) );
970   Polyline_12->AddPoint( 0, gp_XY( -21.09, 122.79 ) );
971   Polyline_12->AddPoint( 0, gp_XY( -106.30, 102.44 ) );
972   Polyline_12->AddSection( "Section_2", HYDROData_PolylineXY::SECTION_POLYLINE, true );
973   Polyline_12->AddPoint( 1, gp_XY( -7.09, 227.08 ) );
974   Polyline_12->AddPoint( 1, gp_XY( 46.32, 228.78 ) );
975   Polyline_12->AddPoint( 1, gp_XY( 69.64, 165.61 ) );
976   Polyline_12->AddPoint( 1, gp_XY( 13.68, 156.28 ) );
977
978   Polyline_12->Update();
979
980   Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
981   CPPUNIT_ASSERT_EQUAL( true, LCM->Add( Polyline_12, "" ) );
982
983   TestViewer::show( LCM->GetShape(), AIS_Shaded, true, "lcm_poly_2_sections" );
984   CPPUNIT_ASSERT_IMAGES
985
986   aDoc->Close();
987 }
988
989 void test_HYDROData_LandCoverMap::test_split_2()
990 {
991   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
992
993   Handle(HYDROData_PolylineXY) Polyline_1 =
994     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
995   Polyline_1->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
996   Polyline_1->AddPoint( 0, gp_XY( -264.84, 323.44 ) );
997   Polyline_1->AddPoint( 0, gp_XY( 254.45, 301.19 ) );
998   Polyline_1->AddPoint( 0, gp_XY( -291.54, -47.48 ) );
999   Polyline_1->Update();
1000
1001   Handle(HYDROData_PolylineXY) Polyline_6 =
1002     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
1003   Polyline_6->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
1004   Polyline_6->AddPoint( 0, gp_XY( -155.15, 199.47 ) );
1005   Polyline_6->AddPoint( 0, gp_XY( -71.19, 289.12 ) );
1006   Polyline_6->AddPoint( 0, gp_XY( 22.01, 211.57 ) );
1007   Polyline_6->AddPoint( 0, gp_XY( -84.00, 134.73 ) );
1008   Polyline_6->Update();
1009
1010   Handle(HYDROData_PolylineXY) Polyline_7 =
1011     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
1012   Polyline_7->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
1013   Polyline_7->AddPoint( 0, gp_XY( -50.56, 232.20 ) );
1014   Polyline_7->AddPoint( 0, gp_XY( 40.51, 286.98 ) );
1015   Polyline_7->AddPoint( 0, gp_XY( 118.77, 256.39 ) );
1016   Polyline_7->AddPoint( 0, gp_XY( 72.52, 114.10 ) );
1017   Polyline_7->Update();
1018
1019   Handle(HYDROData_PolylineXY) Polyline_8 =
1020     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
1021   Polyline_8->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, true );
1022   Polyline_8->AddPoint( 0, gp_XY( -44.16, 190.93 ) );
1023   Polyline_8->AddPoint( 0, gp_XY( 9.91, 226.51 ) );
1024   Polyline_8->AddPoint( 0, gp_XY( 35.53, 175.99 ) );
1025   Polyline_8->AddPoint( 0, gp_XY( -9.30, 157.50 ) );
1026   Polyline_8->Update();
1027
1028   Handle(HYDROData_PolylineXY) Polyline_9 =
1029     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
1030   Polyline_9->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, false );
1031   Polyline_9->AddPoint( 0, gp_XY( -103.92, 288.40 ) );
1032   Polyline_9->AddPoint( 0, gp_XY( -100.09, 285.22 ) );
1033   Polyline_9->AddPoint( 0, gp_XY( -96.10, 281.90 ) );
1034   Polyline_9->AddPoint( 0, gp_XY( -90.43, 277.19 ) );
1035   Polyline_9->AddPoint( 0, gp_XY( -53.41, 246.43 ) );
1036   Polyline_9->AddPoint( 0, gp_XY( 10.62, 245.01 ) );
1037
1038
1039   Polyline_9->Update();
1040
1041
1042   Handle(HYDROData_LandCoverMap) LCM = Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
1043   CPPUNIT_ASSERT_EQUAL( true, LCM->Add( Polyline_1, "" ) );
1044   CPPUNIT_ASSERT_EQUAL( true, LCM->Add( Polyline_6, "" ) );
1045   CPPUNIT_ASSERT_EQUAL( true, LCM->Add( Polyline_7, "" ) );
1046
1047   CPPUNIT_ASSERT_EQUAL( true, LCM->Split( Polyline_9 ) );
1048
1049   //Handle(HYDROGUI_LandCoverMapPrs) aPrs1 = new HYDROGUI_LandCoverMapPrs( LCM );
1050   TestViewer::show( LCM->GetShape(), AIS_Shaded, /*0,*/ true, "LCM_split_2a" );
1051   CPPUNIT_ASSERT_IMAGES
1052
1053   CPPUNIT_ASSERT_EQUAL( true, LCM->Split( Polyline_8 ) );
1054
1055   // Handle(HYDROGUI_LandCoverMapPrs) aPrs2 = new HYDROGUI_LandCoverMapPrs( LCM );
1056   TestViewer::show( LCM->GetShape(), AIS_Shaded, /*0,*/ true, "LCM_split_2b" );
1057   CPPUNIT_ASSERT_IMAGES
1058
1059   aDoc->Close();
1060 }
1061
1062 void test_HYDROData_LandCoverMap::test_export_telemac()
1063 {
1064   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
1065
1066   Handle(HYDROData_StricklerTable) aTable =
1067     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
1068   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
1069
1070   Handle(HYDROData_LandCoverMap) aMap =
1071     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
1072
1073   TopoDS_Face aLC1 = Face( QList<double>() << 12 << 19 << 82 << 9 << 126 << 53 << 107 << 80 << 29 << 75 );
1074   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "Zones de champs cultivé à végétation basse" ) );
1075
1076   TopoDS_Face aLC2 = Face( QList<double>() << 21 << 34 << 24 << 25 << 37   << 37 << 40  << 61 <<
1077                                               44 << 95 << 85 << 100 << 104 << 66 << 107 << 33 <<
1078                                              128 << 18 << 140 << 50 << 131 << 89 << 104 << 111 <<
1079                                               31 << 114 );
1080   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "Zones de champs cultivé à végétation haute" ) );
1081
1082   TopoDS_Face aLC3 = Face( QList<double>() << 4 << 54 << 1   << 47 << 51  << 45 <<
1083                                             127 << 42 << 145 << 43 << 148 << 60 << 90 << 65 );
1084   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC3, "Zones de champs, prairies, sans cultures" ) );
1085
1086   QString aTmpFileName = "test.telemac";
1087   QString aTmpPath = QDir::tempPath() + "/" + aTmpFileName;
1088   CPPUNIT_ASSERT_EQUAL( true, aMap->ExportTelemac( aTmpPath, 1E-4, aTable ) );
1089   CPPUNIT_ASSERT_SCRIPTS_EQUAL( aTmpFileName, true, true, 0 );
1090
1091   aDoc->Close();
1092 }
1093
1094 void test_HYDROData_LandCoverMap::test_copy()
1095 {
1096   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
1097
1098   Handle(HYDROData_LandCoverMap) aMap =
1099     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
1100   aMap->SetName( "map_1" );
1101
1102   TopoDS_Face aLC1 = Face( QList<double>() << 10 << 10 << 30 << 10 << 20 << 20 );
1103   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC1, "Zones de champs cultivé à végétation basse" ) );
1104
1105   TopoDS_Face aLC2 = Face( QList<double>() << 110 << 10 << 130 << 10 << 120 << 20 );
1106   CPPUNIT_ASSERT_EQUAL( true, aMap->LocalPartition( aLC2, "Zones de champs cultivé à végétation haute" ) );
1107
1108   Handle(HYDROData_LandCoverMap) aMap2 =
1109     Handle(HYDROData_LandCoverMap)::DownCast( aDoc->CreateObject( KIND_LAND_COVER_MAP ) );
1110   aMap->CopyTo( aMap2, true );
1111
1112   CPPUNIT_ASSERT_EQUAL( QString( "map_2" ), aMap2->GetName() );
1113   HYDROData_LandCoverMap::Explorer anIt( aMap2 );
1114   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
1115   CPPUNIT_ASSERT_EQUAL( QString( "Zones de champs cultivé à végétation basse" ), anIt.StricklerType() );
1116   CPPUNIT_ASSERT( anIt.Face().TShape()!=aLC1.TShape() );
1117   anIt.Next();
1118   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
1119   CPPUNIT_ASSERT_EQUAL( QString( "Zones de champs cultivé à végétation haute" ), anIt.StricklerType() );
1120   CPPUNIT_ASSERT( anIt.Face().TShape()!=aLC2.TShape() );
1121   anIt.Next();
1122   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
1123
1124   TopoDS_Shape aShape1 = aMap->GetShape();
1125   TopoDS_Shape aShape2 = aMap2->GetShape();
1126   gp_Trsf aTr;
1127   aTr.SetTranslation( gp_Vec( 0, 50, 0 ) );
1128   aShape2.Move( TopLoc_Location( aTr ) );
1129   TestViewer::show( aShape1, AIS_Shaded, /*0,*/ true, "LCM_copied" );
1130   TestViewer::show( aShape2, AIS_Shaded, /*0,*/ true, "" );
1131   CPPUNIT_ASSERT_IMAGES
1132
1133   aDoc->Close();
1134 }