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_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>
  31
  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>
  37 #include <QColor>
  38 #include <QList>
  39 #include <QPointF>
  40
  41 #include <TestShape.h>
  42 #include <TestViewer.h>
  43 #include <TopoDS_Edge.hxx>
  44 #include <TopoDS_Vertex.hxx>
  45 #include <TopoDS_Wire.hxx>
  46 #include <gp_XY.hxx>
  47
  48 void test_HYDROData_PolylineXY::testPolyline()
  49 {
  50   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
  51
  52   Handle(HYDROData_PolylineXY) aPolyline =
  53     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
  54
  55   aPolyline->AddSection( "Section_1", HYDROData_PolylineXY::SECTION_POLYLINE, false );
  56   aPolyline->AddSection( "Section_2", HYDROData_PolylineXY::SECTION_SPLINE, true );
  57
  58   int aNbSections = aPolyline->NbSections();
  59   CPPUNIT_ASSERT( aNbSections == 2 );
  60
  61   NCollection_Sequence<TCollection_AsciiString>           aSectNames;
  62   NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
  63   NCollection_Sequence<bool>                              aSectClosures;
  64   aPolyline->GetSections( aSectNames, aSectTypes, aSectClosures );
  65
  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 );
  70
  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 );
  74
  75   aDoc->Close();
  76 }
  77
  78
  79 void test_HYDROData_PolylineXY::testCopy()
  80 {
  81   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  82   Handle(HYDROData_PolylineXY) aPolyline1 =
  83     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
  84
  85
  86 //  aPolyline1->setPoints(aPoints);
  87
  88   Handle(HYDROData_PolylineXY) aPolyline2 =
  89     Handle(HYDROData_PolylineXY)::DownCast(aDoc->CreateObject(KIND_POLYLINEXY));
  90
  91   aPolyline1->CopyTo(aPolyline2, true);
  92
  93
  94   aDoc->Close();
  95 }
  96
  97 void test_HYDROData_PolylineXY::testSplit_refs_624()
  98 {
  99   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
 100
 101   Handle(HYDROData_PolylineXY) aPolyline =
 102     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
 103   aPolyline->SetName( "test" );
 104
 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 );
 108
 109   gp_Pnt2d aPnt( 20, 20 );
 110
 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
 114
 115   HYDROData_PolylineOperator anOp;
 116   CPPUNIT_ASSERT_EQUAL( true, anOp.Split( aDoc, aPolyline, aPnt, 1E-3 ) );
 117
 118   HYDROData_Iterator anIt( aDoc, KIND_POLYLINEXY );
 119   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
 120   CPPUNIT_ASSERT_EQUAL( QString( "test" ), anIt.Current()->GetName() );
 121   anIt.Next();
 122   CPPUNIT_ASSERT_EQUAL( true, anIt.More() );
 123   CPPUNIT_ASSERT_EQUAL( QString( "test_1" ), anIt.Current()->GetName() );
 124   anIt.Next();
 125   CPPUNIT_ASSERT_EQUAL( false, anIt.More() );
 126
 127   aDoc->Close();
 128 }
 129
 130 void test_HYDROData_PolylineXY::test_extraction_immersible_zone()
 131 {
 132   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
 133
 134   Handle(HYDROData_PolylineXY) aPolyline =
 135     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
 136   aPolyline->SetName( "test" );
 137
 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 );
 141
 142   Handle(HYDROData_ImmersibleZone) aZone =
 143     Handle(HYDROData_ImmersibleZone)::DownCast( aDoc->CreateObject( KIND_IMMERSIBLE_ZONE ) );
 144   aZone->SetName( "zone" );
 145   aZone->SetPolyline( aPolyline );
 146   aZone->Update();
 147
 148   CPPUNIT_ASSERT_EQUAL( false, (bool)aZone->GetTopShape().IsNull() );
 149
 150   HYDROData_PolylineOperator anOp;
 151   CPPUNIT_ASSERT_EQUAL( true, anOp.Extract( aDoc, aZone ) );
 152
 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() );
 156
 157   TestViewer::show( aZone->GetTopShape(), 1, true, "Extraction_ImmZone" );
 158   TestViewer::show( anOuter->GetShape(), 0, true, Qt::red );
 159   CPPUNIT_ASSERT_IMAGES
 160
 161   aDoc->Close();
 162 }
 163
 164 void test_HYDROData_PolylineXY::test_extraction_channel_refs_611()
 165 {
 166   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 167
 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 );
 172
 173   Handle(HYDROData_Polyline3D) aPolyline3d =
 174     Handle(HYDROData_Polyline3D)::DownCast( aDoc->CreateObject( KIND_POLYLINE ) );
 175   aPolyline3d->SetName( "polyline3d_1" );
 176   aPolyline3d->SetPolylineXY( aPolyline2d );
 177
 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 );
 183
 184   Handle(HYDROData_Profile) aProfile =
 185     Handle(HYDROData_Profile)::DownCast( aDoc->CreateObject( KIND_PROFILE ) );
 186   aProfile->SetName( "profile_1" );
 187
 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 );
 192
 193
 194   Handle(HYDROData_Channel) aChannel =
 195     Handle(HYDROData_Channel)::DownCast( aDoc->CreateObject( KIND_CHANNEL ) );
 196   aChannel->SetName( "channel_1" );
 197
 198   aChannel->SetGuideLine( aPolyline3d );
 199   aChannel->SetProfile( aProfile );
 200   aChannel->Update();
 201   CPPUNIT_ASSERT_EQUAL( false, (bool)aChannel->GetTopShape().IsNull() );
 202   CPPUNIT_ASSERT_EQUAL( false, (bool)aChannel->GetShape3D().IsNull() );
 203
 204   HYDROData_PolylineOperator anOp;
 205   CPPUNIT_ASSERT_EQUAL( true, anOp.Extract( aDoc, aChannel ) );
 206
 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() );
 212
 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
 217
 218   aDoc->Close();
 219 }
 220
 221 void test_HYDROData_PolylineXY::test_presentation()
 222 {
 223   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 224
 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 );
 229
 230   QList<gp_XY> aPoints = QList<gp_XY>() << gp_XY(  0,  0 )
 231                                         << gp_XY( 10, 10 )
 232                                         << gp_XY( 20, 40 )
 233                                         << gp_XY( 30, 10 )
 234                                         << gp_XY( 40, 50 )
 235                                         << gp_XY( 50, 60 )
 236                                         << gp_XY( -10, 40 )
 237                                         << gp_XY( -9, 39 )
 238                                         << gp_XY( -8, 38 )
 239                                         << gp_XY(  0, 20 );
 240   Handle(TColgp_HArray1OfPnt) aPnts = new TColgp_HArray1OfPnt( 1, aPoints.size() );
 241   int i = 1;
 242   foreach( gp_XY aPoint, aPoints )
 243   {
 244     aPolyline2d->AddPoint( 0, aPoint );
 245     aPnts->SetValue( i, gp_Pnt( aPoint.X(), aPoint.Y(), 0 ) );
 246     i++;
 247   }
 248   aPolyline2d->Update();
 249
 250   CPPUNIT_ASSERT_EQUAL( false, (bool)aPolyline2d->GetShape().IsNull() );
 251
 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 );
 256
 257   Handle(HYDROGUI_Polyline) aNewPolylinePrs = new HYDROGUI_Polyline( aPolyline2d->GetShape() );
 258   aNewPolylinePrs->SetColor( Quantity_NOC_DARKGREEN );
 259
 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
 264
 265   aDoc->Close();
 266 }
 267
 268 void test_HYDROData_PolylineXY::testSplit_refs_627()
 269 {
 270   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
 271
 272   Handle(HYDROData_PolylineXY) aPolyline =
 273     Handle(HYDROData_PolylineXY)::DownCast( aDoc->CreateObject( KIND_POLYLINEXY ) );
 274   aPolyline->SetName( "test" );
 275
 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 );
 280
 281   gp_Pnt2d aPnt( 20, 20 );
 282
 283   HYDROData_PolylineOperator anOp;
 284   CPPUNIT_ASSERT_EQUAL( true, anOp.Split( aDoc, aPolyline, aPnt, 1E-3 ) );
 285
 286   TestViewer::show( TopoDS_Shape(), 0, true, "Split_Polylines_Colors" );
 287   HYDROData_Iterator anIt( aDoc, KIND_POLYLINEXY );
 288   for( ; anIt.More(); anIt.Next() )
 289   {
 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() );
 293   }
 294   CPPUNIT_ASSERT_IMAGES
 295
 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() );
 301   anIt2.Next();
 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() );
 306   anIt2.Next();
 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() );
 311   anIt2.Next();
 312   CPPUNIT_ASSERT_EQUAL( false, anIt2.More() );
 313   anIt2.Next();
 314
 315   aDoc->Close();
 316 }
 317
 318 void test_HYDROData_PolylineXY::test_custom_polylines()
 319 {
 320   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( 1 );
 321
 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 ) );
 328
 329   aPolyline1->SetName( "test1" );
 330   aPolyline2->SetName( "test2" );
 331   aPolyline3->SetName( "test3" );
 332
 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() );
 340
 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 );
 345
 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 ) );
 352
 353
 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 );
 358
 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 ) );
 370
 371   aDoc->Close();
 372 }