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_PolylineXY.h>
21 #include <HYDROData_Channel.h>
22 #include <HYDROData_Document.h>
23 #include <HYDROData_PolylineXY.h>
24 #include <HYDROData_Polyline3D.h>
25 #include <HYDROData_PolylineOperator.h>
26 #include <HYDROData_Profile.h>
27 #include <HYDROData_Iterator.h>
28 #include <HYDROData_ImmersibleZone.h>
29 #include <HYDROData_Tool.h>
30 #include <HYDROGUI_Polyline.h>
32 #include <AIS_DisplayMode.hxx>
33 #include <AIS_PointCloud.hxx>
34 #include <Prs3d_LineAspect.hxx>
35 #include <Prs3d_PointAspect.hxx>
36 #include <TColgp_HArray1OfPnt.hxx>
41 #include <TestShape.h>
42 #include <TestViewer.h>
43 #include <TopoDS_Edge.hxx>
44 #include <TopoDS_Vertex.hxx>
45 #include <TopoDS_Wire.hxx>
48 void test_HYDROData_PolylineXY::testPolyline()
50 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
52 Handle(HYDROData_PolylineXY) aPolyline =
53 Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
55 aPolyline->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, false );
56 aPolyline->AddSection( "Section_2", HYDROData_PolylineXY::SECTION_SPLINE, true );
58 int aNbSections = aPolyline->NbSections();
59 CPPUNIT_ASSERT( aNbSections == 2 );
61 NCollection_Sequence<TCollection_AsciiString> aSectNames;
62 NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
63 NCollection_Sequence<bool> aSectClosures;
64 aPolyline->GetSections( aSectNames, aSectTypes, aSectClosures );
66 CPPUNIT_ASSERT_EQUAL( 2, aSectNames.Size() );
67 CPPUNIT_ASSERT( aSectNames.Value( 1 ) == "Section_1" );
68 CPPUNIT_ASSERT( aSectTypes.Value( 1 ) == HYDROData_PolylineXY::SECTION_POLYLINE );
69 CPPUNIT_ASSERT( aSectClosures.Value( 1 ) == false );
71 CPPUNIT_ASSERT( aSectNames.Value( 2 ) == "Section_2" );
72 CPPUNIT_ASSERT( aSectTypes.Value( 2 ) == HYDROData_PolylineXY::SECTION_SPLINE );
73 CPPUNIT_ASSERT( aSectClosures.Value( 2 ) == true );
79 void test_HYDROData_PolylineXY::testCopy()
81 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
82 Handle(HYDROData_PolylineXY) aPolyline1 =
83 Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
86 // aPolyline1->setPoints(aPoints);
88 Handle(HYDROData_PolylineXY) aPolyline2 =
89 Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
91 aPolyline1->CopyTo(aPolyline2, true);
97 void test_HYDROData_PolylineXY::testSplit_refs_624()
99 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
101 Handle(HYDROData_PolylineXY) aPolyline =
102 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
103 aPolyline->SetName( "test" );
105 QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
106 TopoDS_Wire aWire = Wire( aCoords, true );
107 aPolyline->SetShape( aWire );
109 gp_Pnt2d aPnt( 20, 20 );
111 TestViewer::show( aPolyline->GetShape(), 0, true, "LandCoverMap_Split_Polyline" );
112 //TestViewer::show( BRepBuilderAPI_MakeVertex( aPnt ).Vertex(), 1, true, Qt::green );
113 CPPUNIT_ASSERT_IMAGES
115 HYDROData_PolylineOperator anOp;
116 CPPUNIT_ASSERT_EQUAL( true, anOp.Split( aDoc, aPolyline, aPnt, 1E-3 ) );
118 HYDROData_Iterator anIt( aDoc, KIND_POLYLINEXY );
119 CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
120 CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt.Current()->GetName() );
122 CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
123 CPPUNIT_ASSERT_EQUAL( QString( "test_1" ), anIt.Current()->GetName() );
125 CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
130 void test_HYDROData_PolylineXY::test_extraction_immersible_zone()
132 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
134 Handle(HYDROData_PolylineXY) aPolyline =
135 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
136 aPolyline->SetName( "test" );
138 QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
139 TopoDS_Wire aWire = Wire( aCoords, true );
140 aPolyline->SetShape( aWire );
142 Handle(HYDROData_ImmersibleZone) aZone =
143 Handle(HYDROData_ImmersibleZone)::DownCast( aDoc->CreateObject( KIND_IMMERSIBLE_ZONE ) );
144 aZone->SetName( "zone" );
145 aZone->SetPolyline( aPolyline );
148 CPPUNIT_ASSERT_EQUAL( false, (bool)aZone->GetTopShape().IsNull() );
150 HYDROData_PolylineOperator anOp;
151 CPPUNIT_ASSERT_EQUAL( true, anOp.Extract( aDoc, aZone ) );
153 Handle(HYDROData_PolylineXY) anOuter =
154 Handle(HYDROData_PolylineXY)::DownCast( aDoc->FindObjectByName( "zone_Outer_1", KIND_POLYLINEXY ) );
155 CPPUNIT_ASSERT_EQUAL( false, (bool)anOuter.IsNull() );
157 TestViewer::show( aZone->GetTopShape(), 1, true, "Extraction_ImmZone" );
158 TestViewer::show( anOuter->GetShape(), 0, true, Qt::red );
159 CPPUNIT_ASSERT_IMAGES
164 void test_HYDROData_PolylineXY::test_extraction_channel_refs_611()
166 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
168 Handle(HYDROData_PolylineXY) aPolyline2d =
169 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
170 aPolyline2d->SetName( "polyline2d_1" );
171 aPolyline2d->AddSection( "", HYDROData_IPolyline::SECTION_SPLINE, false );
173 Handle(HYDROData_Polyline3D) aPolyline3d =
174 Handle(HYDROData_Polyline3D)::DownCast( aDoc->CreateObject( KIND_POLYLINE ) );
175 aPolyline3d->SetName( "polyline3d_1" );
176 aPolyline3d->SetPolylineXY( aPolyline2d );
178 QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
179 TopoDS_Wire aWire = Wire( aCoords, false );
180 aPolyline2d->SetShape( aWire );
181 aPolyline3d->SetTopShape( aWire );
182 aPolyline3d->SetShape3D( aWire );
184 Handle(HYDROData_Profile) aProfile =
185 Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) );
186 aProfile->SetName( "profile_1" );
188 QList<double> aCoordsPr = QList<double>() << 0.0 << 0.1 << 0.5 << 0.0 << 1.0 << 0.1;
189 TopoDS_Wire aWirePr = Wire( aCoordsPr, false );
190 aProfile->SetTopShape( aWirePr );
191 aProfile->SetShape3D( aWirePr );
194 Handle(HYDROData_Channel) aChannel =
195 Handle(HYDROData_Channel)::DownCast( aDoc->CreateObject( KIND_CHANNEL ) );
196 aChannel->SetName( "channel_1" );
198 aChannel->SetGuideLine( aPolyline3d );
199 aChannel->SetProfile( aProfile );
201 CPPUNIT_ASSERT_EQUAL( false, (bool)aChannel->GetTopShape().IsNull() );
202 CPPUNIT_ASSERT_EQUAL( false, (bool)aChannel->GetShape3D().IsNull() );
204 HYDROData_PolylineOperator anOp;
205 CPPUNIT_ASSERT_EQUAL( true, anOp.Extract( aDoc, aChannel ) );
207 Handle(HYDROData_PolylineXY) aLeft =
208 Handle(HYDROData_PolylineXY)::DownCast( aDoc->FindObjectByName( "channel_1_Left_Bank_1", KIND_POLYLINEXY ) );
209 Handle(HYDROData_PolylineXY) aRight =
210 Handle(HYDROData_PolylineXY)::DownCast( aDoc->FindObjectByName( "channel_1_Right_Bank_1", KIND_POLYLINEXY ) );
211 CPPUNIT_ASSERT_EQUAL( false, (bool)aRight.IsNull() );
213 TestViewer::show( aChannel->GetTopShape(), 1, true, "Extraction_Channel" );
214 TestViewer::show( aLeft->GetShape(), 0, true, Qt::red );
215 TestViewer::show( aRight->GetShape(), 0, true, Qt::red );
216 CPPUNIT_ASSERT_IMAGES
221 void test_HYDROData_PolylineXY::test_presentation()
223 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
225 Handle(HYDROData_PolylineXY) aPolyline2d =
226 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
227 aPolyline2d->SetName( "polyline2d_1" );
228 aPolyline2d->AddSection( "", HYDROData_IPolyline::SECTION_SPLINE, false );
230 QList<gp_XY> aPoints = QList<gp_XY>() << gp_XY( 0, 0 )
240 Handle(TColgp_HArray1OfPnt) aPnts = new TColgp_HArray1OfPnt( 1, aPoints.size() );
242 foreach( gp_XY aPoint, aPoints )
244 aPolyline2d->AddPoint( 0, aPoint );
245 aPnts->SetValue( i, gp_Pnt( aPoint.X(), aPoint.Y(), 0 ) );
248 aPolyline2d->Update();
250 CPPUNIT_ASSERT_EQUAL( false, (bool)aPolyline2d->GetShape().IsNull() );
252 Handle(AIS_PointCloud) aPointsPrs = new AIS_PointCloud();
253 aPointsPrs->SetPoints( aPnts );
254 aPointsPrs->SetColor( Quantity_NOC_BLUE1 );
255 aPointsPrs->Attributes()->PointAspect()->SetTypeOfMarker( Aspect_TOM_O );
257 Handle(HYDROGUI_Polyline) aNewPolylinePrs = new HYDROGUI_Polyline( aPolyline2d->GetShape() );
258 aNewPolylinePrs->SetColor( Quantity_NOC_DARKGREEN );
260 TestViewer::show( aPointsPrs, AIS_PointCloud::DM_Points, 0, true, "Polyline_Presentation" );
261 TestViewer::show( aPolyline2d->GetShape(), 0, true, Qt::red );
262 TestViewer::show( aNewPolylinePrs, AIS_PointCloud::DM_Points, 0, true, "" );
263 CPPUNIT_ASSERT_IMAGES
268 void test_HYDROData_PolylineXY::testSplit_refs_627()
270 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
272 Handle(HYDROData_PolylineXY) aPolyline =
273 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
274 aPolyline->SetName( "test" );
276 QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
277 TopoDS_Wire aWire = Wire( aCoords, false );
278 aPolyline->SetShape( aWire );
279 aPolyline->SetWireColor( Qt::red );
281 gp_Pnt2d aPnt( 20, 20 );
283 HYDROData_PolylineOperator anOp;
284 CPPUNIT_ASSERT_EQUAL( true, anOp.Split( aDoc, aPolyline, aPnt, 1E-3 ) );
286 TestViewer::show( TopoDS_Shape(), 0, true, "Split_Polylines_Colors" );
287 HYDROData_Iterator anIt( aDoc, KIND_POLYLINEXY );
288 for( ; anIt.More(); anIt.Next() )
290 Handle(HYDROData_PolylineXY) anObj = Handle(HYDROData_PolylineXY)::DownCast( anIt.Current() );
291 if( aPolyline->Label() != anObj->Label() )
292 TestViewer::show( anObj->GetShape(), 0, true, anObj->GetWireColor() );
294 CPPUNIT_ASSERT_IMAGES
296 HYDROData_Iterator anIt2( aDoc, KIND_POLYLINEXY );
297 CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
298 CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt2.Current()->GetName() );
299 CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ),
300 Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetWireColor() );
302 CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
303 CPPUNIT_ASSERT_EQUAL( QString( "test_1" ), anIt2.Current()->GetName() );
304 CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ),
305 Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetWireColor() );
307 CPPUNIT_ASSERT_EQUAL( true, anIt2.More() );
308 CPPUNIT_ASSERT_EQUAL( QString( "test_2" ), anIt2.Current()->GetName() );
309 CPPUNIT_ASSERT_EQUAL( QColor( Qt::red ),
310 Handle(HYDROData_PolylineXY)::DownCast( anIt2.Current() )->GetWireColor() );
312 CPPUNIT_ASSERT_EQUAL( false, anIt2.More() );
318 void test_HYDROData_PolylineXY::test_custom_polylines()
320 Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
322 Handle(HYDROData_PolylineXY) aPolyline1 =
323 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
324 Handle(HYDROData_PolylineXY) aPolyline2 =
325 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
326 Handle(HYDROData_PolylineXY) aPolyline3 =
327 Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
329 aPolyline1->SetName( "test1" );
330 aPolyline2->SetName( "test2" );
331 aPolyline3->SetName( "test3" );
333 CPPUNIT_ASSERT_EQUAL( false, aPolyline1->IsCustom() );
334 aPolyline1->AddSection( "", HYDROData_IPolyline::SECTION_SPLINE, false );
335 aPolyline1->Update();
336 CPPUNIT_ASSERT_EQUAL( false, aPolyline1->IsCustom() );
337 aPolyline1->AddPoint( 0, gp_XY( 0, 0 ) );
338 aPolyline1->Update();
339 CPPUNIT_ASSERT_EQUAL( false, aPolyline1->IsCustom() );
341 CPPUNIT_ASSERT_EQUAL( false, aPolyline2->IsCustom() );
342 aPolyline2->SetShape( Wire( QList<double>() << 0 << 0 << 1 << 1 << 2 << 0 ) );
343 CPPUNIT_ASSERT_EQUAL( true, aPolyline2->IsCustom() );
344 HYDROData_PolylineXY::PointsList aPointsList = aPolyline2->GetPoints( 0 );
346 CPPUNIT_ASSERT_EQUAL( false, aPolyline2->IsCustom() );
347 CPPUNIT_ASSERT_EQUAL( 33, aPointsList.Size() );
348 CPPUNIT_ASSERT_EQUAL( gp_XY( 0, 0 ), aPointsList.Value( 1 ) );
349 CPPUNIT_ASSERT_EQUAL( gp_XY( 0.5625, 0.808594 ), aPointsList.Value( 10 ) );
350 CPPUNIT_ASSERT_EQUAL( gp_XY( 1.1875, 0.964844 ), aPointsList.Value( 20 ) );
351 CPPUNIT_ASSERT_EQUAL( gp_XY( 2, 0 ), aPointsList.Value( 33 ) );
354 CPPUNIT_ASSERT_EQUAL( false, aPolyline3->IsCustom() );
355 aPolyline3->SetShape( WireCirc( gp_Pnt(), 1.0 ) );
356 CPPUNIT_ASSERT_EQUAL( true, aPolyline3->IsCustom() );
357 aPointsList = aPolyline3->GetPoints( 0 );
359 CPPUNIT_ASSERT_EQUAL( HYDROData_PolylineXY::SECTION_SPLINE, aPolyline3->GetSectionType( 0 ) );
360 CPPUNIT_ASSERT_EQUAL( true, aPolyline3->IsClosedSection( 0 ) );
361 CPPUNIT_ASSERT_EQUAL( 71, aPointsList.Size() );
362 CPPUNIT_ASSERT_EQUAL( gp_XY( 1, 0 ), aPointsList.Value( 1 ) );
363 CPPUNIT_ASSERT_EQUAL( gp_XY( 0.699242, 0.714885 ), aPointsList.Value( 10 ) );
364 CPPUNIT_ASSERT_EQUAL( gp_XY( -0.110394, 0.993888 ), aPointsList.Value( 20 ) );
365 CPPUNIT_ASSERT_EQUAL( gp_XY( -0.839072, 0.54402 ), aPointsList.Value( 30 ) );
366 CPPUNIT_ASSERT_EQUAL( gp_XY( -0.952415, -0.304806 ), aPointsList.Value( 40 ) );
367 CPPUNIT_ASSERT_EQUAL( gp_XY( -0.367302, -0.930102 ), aPointsList.Value( 50 ) );
368 CPPUNIT_ASSERT_EQUAL( gp_XY( 0.487173, -0.873306 ), aPointsList.Value( 60 ) );
369 CPPUNIT_ASSERT_EQUAL( gp_XY( 0.996087, -0.0883801 ), aPointsList.Value( 71 ) );