Salome HOME
ee08e8bb0efb9bb80f8e64287b5c96f795869ab9
[modules/hydro.git] / src / HYDRO_tests / test_HYDROData_PolylineXY.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_PolylineXY.h>
20
21 #include <HYDROData_Document.h>
22 #include <HYDROData_PolylineXY.h>
23 #include <HYDROData_PolylineOperator.h>
24 #include <HYDROData_Iterator.h>
25 #include <HYDROData_Tool.h>
26
27 #include <QColor>
28 #include <QList>
29 #include <QPointF>
30
31 #include <TestViewer.h>
32 #include <BRepBuilderAPI_MakeVertex.hxx>
33 #include <BRepBuilderAPI_MakeWire.hxx>
34 #include <TopoDS_Edge.hxx>
35 #include <TopoDS_Vertex.hxx>
36 #include <TopoDS_Wire.hxx>
37
38 TopoDS_Edge Spline( const QList<double>& theXYList, bool isClosed = false );
39
40
41
42
43 void test_HYDROData_PolylineXY::testPolyline()
44 {
45   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
46
47   Handle(HYDROData_PolylineXY) aPolyline = 
48     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
49
50   aPolyline->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, false );
51   aPolyline->AddSection( "Section_2", HYDROData_PolylineXY::SECTION_SPLINE, true );
52
53   int aNbSections = aPolyline->NbSections();
54   CPPUNIT_ASSERT( aNbSections == 2 );
55   
56   NCollection_Sequence<TCollection_AsciiString>           aSectNames;
57   NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
58   NCollection_Sequence<bool>                              aSectClosures;
59   aPolyline->GetSections( aSectNames, aSectTypes, aSectClosures );
60
61   CPPUNIT_ASSERT_EQUAL( 2, aSectNames.Size() );
62   CPPUNIT_ASSERT( aSectNames.Value( 1 ) == "Section_1" );
63   CPPUNIT_ASSERT( aSectTypes.Value( 1 ) == HYDROData_PolylineXY::SECTION_POLYLINE );
64   CPPUNIT_ASSERT( aSectClosures.Value( 1 ) == false );
65
66   CPPUNIT_ASSERT( aSectNames.Value( 2 ) == "Section_2" );
67   CPPUNIT_ASSERT( aSectTypes.Value( 2 ) == HYDROData_PolylineXY::SECTION_SPLINE );
68   CPPUNIT_ASSERT( aSectClosures.Value( 2 ) == true );
69
70   aDoc->Close();
71 }
72
73
74 void test_HYDROData_PolylineXY::testCopy()
75 {
76   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
77   Handle(HYDROData_PolylineXY) aPolyline1 = 
78     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
79
80
81 //  aPolyline1->setPoints(aPoints);
82
83   Handle(HYDROData_PolylineXY) aPolyline2 = 
84     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
85
86   aPolyline1->CopyTo(aPolyline2, true);
87
88
89   aDoc->Close();
90 }
91
92 void test_HYDROData_PolylineXY::testSplit_refs_624()
93 {
94   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
95
96   Handle(HYDROData_PolylineXY) aPolyline = 
97     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
98   aPolyline->SetName( "test" );
99
100   QList<double> aCoords = QList<double>() << 10 << 10 << 20 << 10 << 20 << 20 << 10 << 20;
101   TopoDS_Wire aWire = BRepBuilderAPI_MakeWire( Spline( aCoords, true ) ).Wire();
102   aPolyline->SetShape( aWire );
103
104   gp_Pnt2d aPnt( 20, 20 );
105
106   TestViewer::show( aPolyline->GetShape(), 0, true, "LandCoverMap_Split_Polyline" );
107   //TestViewer::show( BRepBuilderAPI_MakeVertex( aPnt ).Vertex(), 1, true, Qt::green );
108   CPPUNIT_ASSERT_IMAGES
109
110   HYDROData_PolylineOperator anOp;
111   CPPUNIT_ASSERT_EQUAL( true, anOp.Split( aDoc, aPolyline, aPnt, 1E-3 ) );
112
113   HYDROData_Iterator anIt( aDoc, KIND_POLYLINEXY );
114   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
115   CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt.Current()->GetName() );
116   anIt.Next();
117   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
118   CPPUNIT_ASSERT_EQUAL( QString( "test_1" ), anIt.Current()->GetName() );
119   anIt.Next();
120   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
121
122   aDoc->Close();
123 }
124