src/HYDROData/HYDROData_PolylineXY.cxx

   1
   2 #include "HYDROData_PolylineXY.h"
   3
   4 #include "HYDROData_BSplineOperation.h"
   5 #include "HYDROData_Document.h"
   6 #include "HYDROData_ShapesTool.h"
   7 #include "HYDROData_Tool.h"
   8
   9 #include <BRep_Builder.hxx>
  10 #include <BRepBuilderAPI_MakeEdge.hxx>
  11 #include <BRepBuilderAPI_MakeWire.hxx>
  12 #include <BRepBuilderAPI_MakePolygon.hxx>
  13 #include <BRepBuilderAPI_MakeFace.hxx>
  14 #include <BRepOffsetAPI_NormalProjection.hxx>
  15
  16 #include <GEOMBase.h>
  17
  18 #include <GeomAPI_ProjectPointOnCurve.hxx>
  19 #include <GeomAdaptor_Curve.hxx>
  20 #include <Geom_Line.hxx>
  21 #include <Geom_BSplineCurve.hxx>
  22
  23 #include <GCPnts_AbscissaPoint.hxx>
  24
  25 #include <ImageComposer_MetaTypes.h>
  26
  27 #include <gp_Pnt.hxx>
  28 #include <gp_XY.hxx>
  29 #include <gp_Pln.hxx>
  30
  31 #include <NCollection_Map.hxx>
  32
  33 #include <TCollection_ExtendedString.hxx>
  34
  35 #include <TDataStd_ListIteratorOfListOfByte.hxx>
  36 #include <TColStd_ListIteratorOfListOfInteger.hxx>
  37 #include <TColStd_ListIteratorOfListOfReal.hxx>
  38
  39 #include <TColStd_Array1OfReal.hxx>
  40
  41 #include <TDataStd_AsciiString.hxx>
  42 #include <TDataStd_BooleanList.hxx>
  43 #include <TDataStd_ExtStringList.hxx>
  44 #include <TDataStd_IntegerList.hxx>
  45 #include <TDataStd_ListIteratorOfListOfExtendedString.hxx>
  46 #include <TDataStd_RealList.hxx>
  47 #include <TDataStd_UAttribute.hxx>
  48
  49 #include <TopoDS_Iterator.hxx>
  50 #include <TopTools_ListIteratorOfListOfShape.hxx>
  51 #include <TopTools_HSequenceOfShape.hxx>
  52 #include <TopExp_Explorer.hxx>
  53 #include <ShapeAnalysis_FreeBounds.hxx>
  54 #include <TopoDS.hxx>
  55
  56 #include <QColor>
  57 #include <QPainterPath>
  58 #include <QVariant>
  59
  60 static const Standard_GUID GUID_IS_UNEDITABLE("e5799736-9030-4051-91a4-2e58321fa153");
  61
  62 const double LOCAL_SELECTION_TOLERANCE = 0.0001;
  63
  64 TCollection_AsciiString getUniqueSectionName( const NCollection_Sequence<TCollection_AsciiString>& theNamesSeq )
  65 {
  66   NCollection_Map<TCollection_AsciiString> aNamesMap;
  67
  68   for ( int i = 1, n = theNamesSeq.Size(); i <= n; ++i )
  69   {
  70     const TCollection_AsciiString& aSectName = theNamesSeq.Value( i );
  71     aNamesMap.Add( aSectName );
  72   }
  73
  74   TCollection_AsciiString aResName;
  75
  76   int aPrefIdx = 1;
  77   do
  78   {
  79     aResName = TCollection_AsciiString( "Section_" ) + aPrefIdx;
  80     ++aPrefIdx;
  81   }
  82   while ( aNamesMap.Contains( aResName ) );
  83
  84   return aResName;
  85 }
  86
  87 TCollection_AsciiString getUniqueSectionName( const Handle(TDataStd_ExtStringList)& theNamesList )
  88 {
  89   NCollection_Sequence<TCollection_AsciiString> aNamesSeq;
  90
  91   TDataStd_ListIteratorOfListOfExtendedString aNamesIter( theNamesList->List() );
  92   for ( ; aNamesIter.More(); aNamesIter.Next() )
  93     aNamesSeq.Append( aNamesIter.Value() );
  94
  95   return getUniqueSectionName( aNamesSeq );
  96 }
  97
  98 IMPLEMENT_STANDARD_HANDLE(HYDROData_PolylineXY, HYDROData_IPolyline)
  99 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_PolylineXY, HYDROData_IPolyline)
 100
 101 HYDROData_PolylineXY::HYDROData_PolylineXY()
 102 : HYDROData_IPolyline()
 103 {
 104 }
 105
 106 HYDROData_PolylineXY::~HYDROData_PolylineXY()
 107 {
 108 }
 109
 110 QStringList HYDROData_PolylineXY::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
 111 {
 112   QStringList aResList = dumpObjectCreation( theTreatedObjects );
 113   QString aPolylineName = GetObjPyName();
 114
 115   // Set the wire color
 116   QStringList aWireColorDef;
 117
 118   QColor aWireColor = GetWireColor();
 119   setPythonObjectColor( aWireColorDef, aWireColor, DefaultWireColor(), "SetWireColor" );
 120
 121   if ( !aWireColorDef.isEmpty() )
 122   {
 123     aResList << aWireColorDef;
 124     aResList << QString( "" );
 125   }
 126
 127   bool anIsEditable = IsEditable();
 128   if ( !anIsEditable )
 129   {
 130     // If polyline is not editable we try to import the shape from geom
 131     TCollection_AsciiString aGeomObjectEntry = GetGeomObjectEntry();
 132     if ( !aGeomObjectEntry.IsEmpty() )
 133     {
 134       QString aSalomeObjName = HYDROData_Tool::GenerateNameForPython( theTreatedObjects, "polyline_sobj" );
 135       aResList << QString( "%1 = salome.myStudy.FindObjectID( \"%2\" );" )
 136                   .arg( aSalomeObjName ).arg( aGeomObjectEntry.ToCString() );
 137
 138       aResList << QString( "%1.ImportFromGeomIOR( %2.GetIOR() );" )
 139                   .arg( aPolylineName ).arg( aSalomeObjName );
 140
 141       aResList << QString( "%1.SetGeomObjectEntry( \"%2\" );" )
 142                   .arg( aPolylineName ).arg( aGeomObjectEntry.ToCString() );
 143     }
 144   }
 145   else
 146   {
 147     // Set polilyne data
 148     NCollection_Sequence<TCollection_AsciiString>           aSectNames;
 149     NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
 150     NCollection_Sequence<bool>                              aSectClosures;
 151     GetSections( aSectNames, aSectTypes, aSectClosures );
 152
 153     for ( int i = 1, n = aSectNames.Size(); i <= n; ++i )
 154     {
 155       const TCollection_AsciiString& aSectName = aSectNames.Value( i );
 156       const SectionType& aSectType = aSectTypes.Value( i );
 157       bool aSectClosure = aSectClosures.Value( i );
 158
 159       aResList << QString( "%1.AddSection( \"%2\", %3, %4 );" ).arg( aPolylineName )
 160                   .arg( aSectName.ToCString() ).arg( aSectType ).arg( aSectClosure );
 161
 162       HYDROData_IPolyline::PointsList aSectPointsList = GetPoints( i - 1 );
 163       for ( int k = 1, aNbPoints = aSectPointsList.Size(); k <= aNbPoints; ++k )
 164       {
 165         const Point& aSectPoint = aSectPointsList.Value( k );
 166
 167         aResList << QString( "%1.AddPoint( %2, gp_XY( %3, %4 ) );" ).arg( aPolylineName )
 168           .arg( i - 1 ).arg( aSectPoint.X() ).arg( aSectPoint.Y() );
 169       }
 170     }
 171   }
 172   aResList << QString( "" );
 173   aResList << QString( "%1.Update();" ).arg( aPolylineName );
 174   aResList << QString( "" );
 175
 176   return aResList;
 177 }
 178
 179 QVariant HYDROData_PolylineXY::GetDataVariant()
 180 {
 181   QPainterPath aPath = GetPainterPath();
 182
 183   QVariant aVarData;
 184   aVarData.setValue<QPainterPath>( aPath );
 185
 186   return aVarData;
 187 }
 188
 189 QColor HYDROData_PolylineXY::DefaultWireColor()
 190 {
 191   return QColor( Qt::red );
 192 }
 193
 194 bool HYDROData_PolylineXY::ImportFromGeomIOR( const TCollection_AsciiString& theIOR )
 195 {
 196   if ( theIOR.IsEmpty() )
 197     return false;
 198
 199   TopoDS_Shape aShape = GEOMBase::GetShapeFromIOR( theIOR.ToCString() );
 200   if ( aShape.IsNull() )
 201     return false;
 202
 203   return ImportShape( aShape );
 204 }
 205
 206 void HYDROData_PolylineXY::SetGeomObjectEntry( const TCollection_AsciiString& theEntry )
 207 {
 208   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_GeomObjectEntry ), theEntry );
 209 }
 210
 211 TCollection_AsciiString HYDROData_PolylineXY::GetGeomObjectEntry() const
 212 {
 213   TCollection_AsciiString aRes;
 214
 215   TDF_Label aLabel = myLab.FindChild( DataTag_GeomObjectEntry, false );
 216   if ( !aLabel.IsNull() )
 217   {
 218     Handle(TDataStd_AsciiString) anAsciiStr;
 219     if ( aLabel.FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
 220       aRes = anAsciiStr->Get();
 221   }
 222
 223   return aRes;
 224 }
 225
 226 TopoDS_Shape HYDROData_PolylineXY::GetShape() const
 227 {
 228   return getPolylineShape();
 229 }
 230
 231 bool convertEdgeToSection( const TopoDS_Edge&                                       theEdge,
 232                            NCollection_Sequence<TCollection_AsciiString>&           theSectNames,
 233                            NCollection_Sequence<HYDROData_PolylineXY::SectionType>& theSectTypes,
 234                            NCollection_Sequence<bool>&                              theSectClosures,
 235                            NCollection_Sequence<HYDROData_PolylineXY::PointsList>&  theSectPoints,
 236                            const bool                                               theIsCanBeClosed )
 237 {
 238   Standard_Real aFirst = 0.0, aLast = 0.0;
 239   Handle(Geom_Curve) anEdgeGeomCurve = BRep_Tool::Curve( theEdge, aFirst, aLast );
 240   if ( anEdgeGeomCurve.IsNull() )
 241     return false;
 242
 243   TCollection_AsciiString aSectName = getUniqueSectionName( theSectNames );
 244   bool anIsEdgeClosed = anEdgeGeomCurve->IsClosed();
 245
 246   HYDROData_PolylineXY::SectionType aSectionType = HYDROData_PolylineXY::SECTION_POLYLINE;
 247   HYDROData_PolylineXY::PointsList aPointsList;
 248
 249   if ( anEdgeGeomCurve->IsKind( STANDARD_TYPE(Geom_Line) ) )
 250   {
 251     Handle(Geom_Line) aGeomLine = Handle(Geom_Line)::DownCast( anEdgeGeomCurve );
 252
 253     gp_Pnt aFirstPoint, aLastPoint;
 254     aGeomLine->D0( aFirst, aFirstPoint );
 255     aGeomLine->D0( aLast, aLastPoint );
 256
 257     HYDROData_PolylineXY::Point aSectFirstPoint( aFirstPoint.X(), aFirstPoint.Y() );
 258     aPointsList.Append( aSectFirstPoint );
 259
 260     HYDROData_PolylineXY::Point aSectLastPoint( aLastPoint.X(), aLastPoint.Y() );
 261     aPointsList.Append( aSectLastPoint );
 262   }
 263   else if ( anEdgeGeomCurve->IsKind( STANDARD_TYPE(Geom_BSplineCurve) ) )
 264   {
 265     aSectionType = HYDROData_PolylineXY::SECTION_SPLINE;
 266
 267     Handle(Geom_BSplineCurve) aGeomSpline =
 268       Handle(Geom_BSplineCurve)::DownCast( anEdgeGeomCurve );
 269
 270     int aNbKnots = aGeomSpline->NbKnots();
 271
 272     TColStd_Array1OfReal aSplineKnots( 1, aNbKnots );
 273     aGeomSpline->Knots( aSplineKnots );
 274
 275     // Decrease the number of imported knots because of last one
 276     // knot is the closing point which are the start point
 277     if ( anIsEdgeClosed ) aNbKnots--;
 278
 279     for ( int i = 1; i <= aNbKnots; ++i )
 280     {
 281       const Standard_Real& aKnot = aSplineKnots.Value( i );
 282
 283       gp_Pnt aPoint;
 284       aGeomSpline->D0( aKnot, aPoint );
 285
 286       HYDROData_PolylineXY::Point aSectPoint( aPoint.X(), aPoint.Y() );
 287       aPointsList.Append( aSectPoint );
 288     }
 289   }
 290   else
 291   {
 292     // Other curve types are not supported
 293     return false;
 294   }
 295
 296   if ( aPointsList.IsEmpty() )
 297     return false;
 298
 299   theSectNames.Append( aSectName );
 300   theSectTypes.Append( aSectionType );
 301   theSectClosures.Append( anIsEdgeClosed );
 302   theSectPoints.Append( aPointsList );
 303
 304   return true;
 305 }
 306
 307 bool HYDROData_PolylineXY::ImportShape( const TopoDS_Shape& theShape )
 308 {
 309   if ( theShape.IsNull() )
 310     return false;
 311
 312   RemoveSections();
 313
 314   bool anIsCanBeImported = false;
 315
 316   NCollection_Sequence<TCollection_AsciiString> aSectNames;
 317   NCollection_Sequence<SectionType>             aSectTypes;
 318   NCollection_Sequence<bool>                    aSectClosures;
 319   NCollection_Sequence<PointsList>              aSectPoints;
 320
 321   if ( theShape.ShapeType() == TopAbs_EDGE )
 322   {
 323     TopoDS_Edge anEdge = TopoDS::Edge( theShape );
 324     anIsCanBeImported = convertEdgeToSection( anEdge, aSectNames, aSectTypes, aSectClosures, aSectPoints, true );
 325   }
 326   else if ( theShape.ShapeType() == TopAbs_WIRE )
 327   {
 328     TopTools_SequenceOfShape anEdges;
 329     HYDROData_ShapesTool::ExploreShapeToShapes( theShape, TopAbs_EDGE, anEdges );
 330
 331     anIsCanBeImported = !anEdges.IsEmpty();
 332     for ( int i = 1, n = anEdges.Length(); i <= n && anIsCanBeImported; ++i )
 333     {
 334       TopoDS_Edge aWireEdge = TopoDS::Edge( anEdges.Value( i ) );
 335       anIsCanBeImported = convertEdgeToSection( aWireEdge, aSectNames, aSectTypes, aSectClosures, aSectPoints, false );
 336     }
 337   }
 338
 339   if ( anIsCanBeImported )
 340   {
 341     for ( int i = 1, n = aSectNames.Length(); i <= n; ++i )
 342     {
 343       const TCollection_AsciiString& aSectName = aSectNames.Value( i );
 344       const SectionType& aSectType = aSectTypes.Value( i );
 345       bool anIsSectionClosed = aSectClosures.Value( i );
 346       const PointsList& aSectPointsList = aSectPoints( i );
 347
 348       AddSection( aSectName, aSectType, anIsSectionClosed );
 349       SetPoints( i - 1, aSectPointsList );
 350     }
 351   }
 352   else
 353   {
 354     TopoDS_Shape aShape = theShape;
 355
 356     if ( theShape.ShapeType() == TopAbs_EDGE )
 357     {
 358       // We make the wire from incoming edge because of other algorithms
 359       // are waiting at least the wire from polyline
 360       TopoDS_Edge anEdge = TopoDS::Edge( theShape );
 361       BRepBuilderAPI_MakeWire aMakeWire( anEdge );
 362       aMakeWire.Build();
 363       if ( aMakeWire.IsDone() )
 364         aShape = aMakeWire.Wire();
 365     }
 366
 367     gp_Pln aPlane( gp_Pnt( 0, 0, 0 ), gp_Dir( 0, 0, 1 ) );
 368     BRepBuilderAPI_MakeFace aMakeFace( aPlane );
 369     aMakeFace.Build();
 370     BRepOffsetAPI_NormalProjection aProj( aMakeFace.Face() );
 371     aProj.Add( aShape );
 372     aProj.Build();
 373     TopoDS_Shape aResult;
 374     if( aProj.IsDone() )
 375       aResult = aProj.Shape();
 376
 377     setPolylineShape( aResult );
 378   }
 379
 380   setEditable( anIsCanBeImported );
 381
 382   return true;
 383 }
 384
 385 TopoDS_Wire HYDROData_PolylineXY::BuildWire( const SectionType&                  theType,
 386                                              const bool&                         theIsClosed,
 387                                              const NCollection_Sequence<gp_XYZ>& thePoints )
 388 {
 389   TopoDS_Wire aWire;
 390   if( theType == SECTION_POLYLINE )
 391   {
 392     int aNbPoints = thePoints.Length();
 393     BRepBuilderAPI_MakePolygon aMakeWire;
 394     for ( int i = 1, n = aNbPoints; i <= n ; ++i )
 395     {
 396       gp_XYZ aPoint = thePoints.Value( i );
 397       gp_Pnt aPnt( aPoint.X(), aPoint.Y(), aPoint.Z() );
 398       aMakeWire.Add( aPnt );
 399     }
 400     if( theIsClosed && ( aNbPoints > 2 ) )
 401       aMakeWire.Close();
 402
 403     if ( aMakeWire.IsDone() )
 404       aWire = aMakeWire.Wire();
 405   }
 406   else //if( theType == PolylineSection::SECTION_SPLINE )
 407   {
 408     BRepBuilderAPI_MakeWire aMakeWire;
 409
 410     if ( thePoints.Size() > 1 )
 411     {
 412       HYDROData_BSplineOperation aBSpline( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE );
 413
 414       TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge( aBSpline.Curve() ).Edge();
 415       aMakeWire.Add( anEdge );
 416     }
 417     aMakeWire.Build();
 418     if ( aMakeWire.IsDone() )
 419       aWire = aMakeWire;
 420   }
 421
 422   return aWire;
 423 }
 424
 425 void HYDROData_PolylineXY::BuildPainterPath( QPainterPath&                       thePath,
 426                                              const SectionType&                  theType,
 427                                              const bool&                         theIsClosed,
 428                                              const NCollection_Sequence<gp_XYZ>& thePoints )
 429 {
 430   if ( thePoints.IsEmpty() )
 431     return;
 432
 433   if ( theType == SECTION_POLYLINE )
 434   {
 435     const gp_XYZ& aFirstPoint = thePoints.Value( 1 );
 436     thePath.moveTo( aFirstPoint.X(), aFirstPoint.Y() );
 437
 438     for( int i = 2, n = thePoints.Size(); i <= n; ++i )
 439     {
 440       const gp_XYZ& aSectPoint = thePoints.Value( i );
 441
 442       thePath.lineTo( aSectPoint.X(), aSectPoint.Y() );
 443     }
 444
 445     if( theIsClosed )
 446       thePath.closeSubpath();
 447   }
 448   else
 449   {
 450     HYDROData_BSplineOperation aBSpline( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE );
 451     aBSpline.ComputePath( thePath );
 452   }
 453 }
 454
 455 void HYDROData_PolylineXY::Update()
 456 {
 457   if ( !IsEditable() )
 458   {
 459     // If polyline is not editable we no need to update it wire
 460     SetToUpdate( false );
 461     return;
 462   }
 463
 464   HYDROData_IPolyline::Update();
 465
 466   NCollection_Sequence<TCollection_AsciiString>           aSectNames;
 467   NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
 468   NCollection_Sequence<bool>                              aSectClosures;
 469   GetSections( aSectNames, aSectTypes, aSectClosures );
 470
 471   BRepBuilderAPI_MakeWire aMakeWire;
 472
 473   TopTools_ListOfShape aSectionWiresList;
 474
 475   for ( int aSectionId = 1, aNbSects = aSectNames.Size(); aSectionId <= aNbSects; aSectionId++ )
 476   {
 477     TCollection_AsciiString aSectName = aSectNames.Value( aSectionId );
 478     SectionType aSectionType = aSectTypes.Value( aSectionId );
 479     bool anIsSectionClosed = aSectClosures.Value( aSectionId );
 480
 481     PointsList aSectPointsList = GetPoints( aSectionId - 1 );
 482     if ( aSectPointsList.IsEmpty() )
 483       continue;
 484
 485     NCollection_Sequence<gp_XYZ> aPoints;
 486     for( int i = 1, n = aSectPointsList.Size(); i <= n; ++i )
 487     {
 488       const Point& aSectPoint = aSectPointsList.Value( i );
 489
 490       gp_XYZ aPoint( aSectPoint.X(), aSectPoint.Y(), 0.0 );
 491       aPoints.Append( aPoint );
 492     }
 493
 494     TopoDS_Wire aSectionWire = BuildWire( aSectionType, anIsSectionClosed, aPoints );
 495     if ( !aSectionWire.IsNull() ) {
 496       aSectionWiresList.Append( aSectionWire );
 497       aMakeWire.Add( aSectionWire );
 498     }
 499   }
 500 // all input wires in the <aSectionWiresList>
 501   Handle(TopTools_HSequenceOfShape) aSeqWires = new TopTools_HSequenceOfShape;
 502   Handle(TopTools_HSequenceOfShape) aSeqEdges = new TopTools_HSequenceOfShape;
 503   TopTools_ListIteratorOfListOfShape it(aSectionWiresList);
 504   for(;it.More();it.Next())
 505   {
 506     TopExp_Explorer it2(it.Value(), TopAbs_EDGE);
 507     for(;it2.More();it2.Next())
 508       aSeqEdges->Append(it2.Current());
 509   }
 510
 511   BRep_Builder aBB;
 512   TopoDS_Compound aCmp;
 513   TopoDS_Shape aResult;
 514   aBB.MakeCompound(aCmp);
 515   if(aSeqEdges->Length() >1)
 516   {
 517     ShapeAnalysis_FreeBounds::ConnectEdgesToWires(aSeqEdges,1E-5,Standard_False,aSeqWires);
 518
 519     if( aSeqWires->Length()==1 )
 520       aResult = aSeqWires->Value( 1 );
 521     else
 522     {
 523       for (Standard_Integer i = 1; i <= aSeqWires->Length();i++)
 524       {
 525         const TopoDS_Shape& aS1 = aSeqWires->Value(i);
 526         aBB.Add(aCmp, aS1);
 527       }
 528       aResult = aCmp;
 529     }
 530   }
 531   else if (aSeqEdges->Length() == 1)
 532   {
 533     BRepBuilderAPI_MakeWire mkWire (TopoDS::Edge(aSeqEdges->Value(1)));
 534     if (mkWire.IsDone())
 535       aResult = mkWire.Wire();
 536   }
 537
 538   setPolylineShape( aResult );
 539 }
 540
 541 bool HYDROData_PolylineXY::IsHas2dPrs() const
 542 {
 543   return true;
 544 }
 545
 546 bool HYDROData_PolylineXY::IsEditable() const
 547 {
 548   return !myLab.IsAttribute( GUID_IS_UNEDITABLE );
 549 }
 550
 551 void HYDROData_PolylineXY::setEditable( const bool theIsEditable )
 552 {
 553   if ( !theIsEditable )
 554     TDataStd_UAttribute::Set( myLab, GUID_IS_UNEDITABLE );
 555   else
 556     myLab.ForgetAttribute( GUID_IS_UNEDITABLE );
 557 }
 558
 559 /**
 560  * Returns true if polyline is closed
 561  */
 562 bool HYDROData_PolylineXY::IsClosed( const bool theIsSimpleCheck ) const
 563 {
 564   bool anIsClosed = false;
 565
 566   TopoDS_Shape aShape = GetShape();
 567   if ( aShape.IsNull() )
 568     return anIsClosed;
 569
 570   TopTools_SequenceOfShape aWires;
 571   HYDROData_ShapesTool::ExploreShapeToShapes( aShape, TopAbs_WIRE, aWires );
 572
 573   int aNbWires = aWires.Length();
 574   if ( theIsSimpleCheck )
 575   {
 576     anIsClosed = aNbWires > 0;
 577     for ( int i = 1; i <= aNbWires && anIsClosed; ++i )
 578     {
 579       const TopoDS_Shape& aWire = aWires.Value( i );
 580       anIsClosed = BRep_Tool::IsClosed( aWire );
 581     }
 582   }
 583   else
 584   {
 585     anIsClosed = aNbWires == 1 && BRep_Tool::IsClosed( aWires.First() );
 586   }
 587
 588   return anIsClosed;
 589 }
 590
 591 double HYDROData_PolylineXY::GetDistance( const int theSectionIndex,
 592                                           const int thePointIndex ) const
 593 {
 594   double aResDistance = -1;
 595   if ( theSectionIndex < 0 || theSectionIndex >= NbSections() )
 596     return aResDistance;
 597
 598   if ( thePointIndex == 0 )
 599     return 0.0;
 600
 601   SectionType aSectionType = GetSectionType( theSectionIndex );
 602   bool anIsSectionClosed = IsClosedSection( theSectionIndex );
 603   PointsList aSectPointsList = GetPoints( theSectionIndex );
 604   if ( thePointIndex < 0 || thePointIndex >= aSectPointsList.Size()  )
 605     return aResDistance;
 606
 607   if ( aSectionType == SECTION_POLYLINE )
 608   {
 609     aResDistance = 0.0;
 610
 611     Point aPrevPoint = aSectPointsList.Value( 1 );
 612     for ( int i = 2, aNbPoints = aSectPointsList.Size(); i <= aNbPoints; ++i )
 613     {
 614       const Point& aSectPoint = aSectPointsList.Value( i );
 615       aResDistance += gp_Pnt2d( aPrevPoint ).Distance( aSectPoint );
 616       aPrevPoint = aSectPoint;
 617
 618       if ( thePointIndex == i - 1 )
 619         break;
 620     }
 621   }
 622   else
 623   {
 624     gp_XYZ aPointToTest;
 625
 626     int aSectNbPoints = aSectPointsList.Size();
 627     NCollection_Sequence<gp_XYZ> aPoints;
 628     for( int i = 1 ; i <= aSectNbPoints; ++i )
 629     {
 630       const Point& aSectPoint = aSectPointsList.Value( i );
 631
 632       gp_XYZ aPoint( aSectPoint.X(), aSectPoint.Y(), 0.0 );
 633       aPoints.Append( aPoint );
 634
 635       if ( thePointIndex == i - 1 )
 636         aPointToTest = aPoint;
 637     }
 638
 639     HYDROData_BSplineOperation aBSpline( aPoints, anIsSectionClosed, LOCAL_SELECTION_TOLERANCE );
 640
 641     Quantity_Parameter aFirstParam = aBSpline.Curve()->FirstParameter();
 642     Quantity_Parameter aSecondParam = aBSpline.Curve()->LastParameter();
 643
 644     if ( thePointIndex != aSectNbPoints - 1 )
 645     {
 646       GeomAPI_ProjectPointOnCurve aProject( aPointToTest, aBSpline.Curve() );
 647       aSecondParam = aProject.LowerDistanceParameter();
 648     }
 649
 650     GeomAdaptor_Curve anAdap( aBSpline.Curve() );
 651
 652     aResDistance = GCPnts_AbscissaPoint::Length( anAdap, aFirstParam, aSecondParam );
 653   }
 654
 655   return aResDistance;
 656 }
 657
 658 int HYDROData_PolylineXY::NbSections() const
 659 {
 660   Handle(TDataStd_ExtStringList) aNamesList;
 661   Handle(TDataStd_IntegerList)   aTypesList;
 662   Handle(TDataStd_BooleanList)   aClosuresList;
 663   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 664
 665   return !aClosuresList.IsNull() ? aClosuresList->Extent() : 0;
 666 }
 667
 668 void HYDROData_PolylineXY::AddSection( const TCollection_AsciiString& theSectName,
 669                                        const SectionType              theSectionType,
 670                                        const bool                     theIsClosed )
 671 {
 672   Handle(TDataStd_ExtStringList) aNamesList;
 673   Handle(TDataStd_IntegerList)   aTypesList;
 674   Handle(TDataStd_BooleanList)   aClosuresList;
 675   getSectionsLists( aNamesList, aTypesList, aClosuresList );
 676
 677   TCollection_ExtendedString aSectName( theSectName );
 678   if ( aSectName.Length() <= 0 )
 679     aSectName = getUniqueSectionName( aNamesList );
 680
 681   aNamesList->Append( aSectName );
 682   aTypesList->Append( theSectionType );
 683   aClosuresList->Append( theIsClosed );
 684
 685   SetToUpdate( true );
 686 }
 687
 688 TCollection_AsciiString HYDROData_PolylineXY::GetSectionName( const int theSectionIndex ) const
 689 {
 690   TCollection_AsciiString aResName;
 691
 692   Handle(TDataStd_ExtStringList) aNamesList;
 693   Handle(TDataStd_IntegerList) aTypesList;
 694   Handle(TDataStd_BooleanList) aClosuresList;
 695   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 696   if ( aNamesList.IsNull() || theSectionIndex >= aNamesList->Extent() )
 697     return aResName;
 698
 699   TDataStd_ListIteratorOfListOfExtendedString aNamesIter( aNamesList->List() );
 700   for ( int i = 0; aNamesIter.More() && i != theSectionIndex; aNamesIter.Next(), ++i );
 701
 702   if ( aNamesIter.More() )
 703     aResName = aNamesIter.Value();
 704
 705   return aResName;
 706 }
 707
 708 void HYDROData_PolylineXY::SetSectionName( const int                      theSectionIndex,
 709                                            const TCollection_AsciiString& theSectionName )
 710 {
 711   Handle(TDataStd_ExtStringList) aNamesList;
 712   Handle(TDataStd_IntegerList) aTypesList;
 713   Handle(TDataStd_BooleanList) aClosuresList;
 714   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 715   if ( aNamesList.IsNull() || theSectionIndex >= aNamesList->Extent() )
 716     return;
 717
 718   TDataStd_ListOfExtendedString anOldNamesList;
 719   anOldNamesList = aNamesList->List();
 720
 721   // Refill the existing list
 722   aNamesList->Clear();
 723
 724   TCollection_ExtendedString aNewSectName = theSectionName;
 725
 726   TDataStd_ListIteratorOfListOfExtendedString aNamesIter( anOldNamesList );
 727   for ( int i = 0; aNamesIter.More(); aNamesIter.Next(), ++i )
 728     aNamesList->Append( i == theSectionIndex ? aNewSectName : aNamesIter.Value() );
 729
 730   SetToUpdate( true );
 731 }
 732
 733 HYDROData_PolylineXY::SectionType HYDROData_PolylineXY::GetSectionType( const int theSectionIndex ) const
 734 {
 735   Handle(TDataStd_ExtStringList) aNamesList;
 736   Handle(TDataStd_IntegerList) aTypesList;
 737   Handle(TDataStd_BooleanList) aClosuresList;
 738   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 739   if ( aTypesList.IsNull() || theSectionIndex >= aTypesList->Extent() )
 740     return SECTION_POLYLINE;
 741
 742   TColStd_ListIteratorOfListOfInteger aTypesIter( aTypesList->List() );
 743   for ( int i = 0; aTypesIter.More() && i != theSectionIndex; aTypesIter.Next(), ++i );
 744
 745   return aTypesIter.More() ? (SectionType)aTypesIter.Value() : SECTION_POLYLINE;
 746 }
 747
 748 void HYDROData_PolylineXY::SetSectionType( const int         theSectionIndex,
 749                                            const SectionType theSectionType )
 750 {
 751   Handle(TDataStd_ExtStringList) aNamesList;
 752   Handle(TDataStd_IntegerList) aTypesList;
 753   Handle(TDataStd_BooleanList) aClosuresList;
 754   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 755   if ( aTypesList.IsNull() || theSectionIndex >= aTypesList->Extent() )
 756     return;
 757
 758   TColStd_ListOfInteger anOldTypesList;
 759   anOldTypesList = aTypesList->List();
 760
 761   // Refill the existing list
 762   aTypesList->Clear();
 763
 764   TColStd_ListIteratorOfListOfInteger aTypesIter( anOldTypesList );
 765   for ( int i = 0; aTypesIter.More(); aTypesIter.Next(), ++i )
 766     aTypesList->Append( i == theSectionIndex ? theSectionType : aTypesIter.Value() );
 767
 768   SetToUpdate( true );
 769 }
 770
 771 bool HYDROData_PolylineXY::IsClosedSection( const int theSectionIndex ) const
 772 {
 773   Handle(TDataStd_ExtStringList) aNamesList;
 774   Handle(TDataStd_IntegerList) aTypesList;
 775   Handle(TDataStd_BooleanList) aClosuresList;
 776   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 777   if ( aClosuresList.IsNull() || theSectionIndex >= aClosuresList->Extent() )
 778     return false;
 779
 780   TDataStd_ListIteratorOfListOfByte aClosuresIter( aClosuresList->List() );
 781   for ( int i = 0; aClosuresIter.More() && i != theSectionIndex; aClosuresIter.Next(), ++i );
 782
 783   return aClosuresIter.More() ? (bool)aClosuresIter.Value() : false;
 784 }
 785
 786 void HYDROData_PolylineXY::SetSectionClosed( const int  theSectionIndex,
 787                                              const bool theIsClosed )
 788 {
 789   Handle(TDataStd_ExtStringList) aNamesList;
 790   Handle(TDataStd_IntegerList) aTypesList;
 791   Handle(TDataStd_BooleanList) aClosuresList;
 792   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 793   if ( aClosuresList.IsNull() || theSectionIndex >= aClosuresList->Extent() )
 794     return;
 795
 796   TDataStd_ListOfByte anOldClosuresList;
 797   anOldClosuresList = aClosuresList->List();
 798
 799   // Refill the existing list
 800   aClosuresList->Clear();
 801
 802   TDataStd_ListIteratorOfListOfByte aClosuresIter( anOldClosuresList );
 803   for ( int i = 0; aClosuresIter.More(); aClosuresIter.Next(), ++i )
 804     aClosuresList->Append( i == theSectionIndex ? theIsClosed : (bool)aClosuresIter.Value() );
 805
 806   SetToUpdate( true );
 807 }
 808
 809 void HYDROData_PolylineXY::GetSections( NCollection_Sequence<TCollection_AsciiString>& theSectNames,
 810                                         NCollection_Sequence<SectionType>&             theSectTypes,
 811                                         NCollection_Sequence<bool>&                    theSectClosures ) const
 812 {
 813   theSectNames.Clear();
 814   theSectTypes.Clear();
 815   theSectClosures.Clear();
 816
 817   Handle(TDataStd_ExtStringList) aNamesList;
 818   Handle(TDataStd_IntegerList) aTypesList;
 819   Handle(TDataStd_BooleanList) aClosuresList;
 820   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 821   if ( aNamesList.IsNull() || aTypesList.IsNull() || aClosuresList.IsNull() )
 822     return;
 823
 824   TDataStd_ListIteratorOfListOfExtendedString aNamesIter( aNamesList->List() );
 825   TColStd_ListIteratorOfListOfInteger aTypesIter( aTypesList->List() );
 826   TDataStd_ListIteratorOfListOfByte aClosuresIter( aClosuresList->List() );
 827   for ( ; aNamesIter.More() && aTypesIter.More() && aClosuresIter.More();
 828           aNamesIter.Next(), aTypesIter.Next(), aClosuresIter.Next() )
 829   {
 830     const TCollection_ExtendedString& aSectName = aNamesIter.Value();
 831     SectionType aSectType = (SectionType)aTypesIter.Value();
 832     bool aSectClosures = aClosuresIter.Value();
 833
 834     theSectNames.Append( aSectName );
 835     theSectTypes.Append( aSectType );
 836     theSectClosures.Append( aSectClosures );
 837   }
 838 }
 839
 840 void HYDROData_PolylineXY::RemoveSection( const int theSectionIndex )
 841 {
 842   Handle(TDataStd_ExtStringList) aNamesList;
 843   Handle(TDataStd_IntegerList)   aTypesList;
 844   Handle(TDataStd_BooleanList)   aClosuresList;
 845   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
 846   if ( aNamesList.IsNull() || theSectionIndex >= aNamesList->Extent() )
 847     return;
 848
 849   if ( aNamesList->Extent() == 1 )
 850   {
 851     removeSectionsLists();
 852     removePointsLists();
 853   }
 854   else
 855   {
 856     TDataStd_ListOfExtendedString anOldNamesList;
 857     anOldNamesList = aNamesList->List();
 858
 859     TColStd_ListOfInteger anOldTypesList;
 860     anOldTypesList = aTypesList->List();
 861
 862     TDataStd_ListOfByte anOldClosuresList;
 863     anOldClosuresList = aClosuresList->List();
 864
 865     // Refill the existing lists
 866     aNamesList->Clear();
 867     aTypesList->Clear();
 868     aClosuresList->Clear();
 869
 870     TDataStd_ListIteratorOfListOfExtendedString aNamesIter( anOldNamesList );
 871     TColStd_ListIteratorOfListOfInteger aTypesIter( anOldTypesList );
 872     TDataStd_ListIteratorOfListOfByte aClosuresIter( anOldClosuresList );
 873     for ( int i = 0; aNamesIter.More() && aTypesIter.More() && aClosuresIter.More();
 874                      aNamesIter.Next(), aTypesIter.Next(), aClosuresIter.Next(), ++i )
 875     {
 876       if ( i == theSectionIndex )
 877         continue; // skip index to remove
 878
 879       aNamesList->Append( aNamesIter.Value() );
 880       aTypesList->Append( aTypesIter.Value() );
 881       aClosuresList->Append( (bool)aClosuresIter.Value() );
 882     }
 883
 884     // Remove points that belongs to removed section
 885     removePointsLists( theSectionIndex );
 886   }
 887
 888   SetToUpdate( true );
 889 }
 890
 891 void HYDROData_PolylineXY::RemoveSections()
 892 {
 893   removeSectionsLists();
 894   removePointsLists();
 895   SetToUpdate( true );
 896 }
 897
 898 void HYDROData_PolylineXY::AddPoint( const int    theSectionIndex,
 899                                      const Point& thePoint,
 900                                      const int    thePointIndex )
 901 {
 902   Handle(TDataStd_RealList) aListX, aListY;
 903   getPointsLists( theSectionIndex, aListX, aListY );
 904
 905   if ( thePointIndex < 0 || thePointIndex >= aListX->Extent() )
 906   {
 907     aListX->Append( thePoint.X() );
 908     aListY->Append( thePoint.Y() );
 909   }
 910   else
 911   {
 912     TColStd_ListOfReal anOldListX;
 913     anOldListX = aListX->List();
 914
 915     TColStd_ListOfReal anOldListY;
 916     anOldListY = aListY->List();
 917
 918     // Refill the existing lists
 919     aListX->Clear();
 920     aListY->Clear();
 921
 922     TColStd_ListIteratorOfListOfReal anIterX( anOldListX );
 923     TColStd_ListIteratorOfListOfReal anIterY( anOldListY );
 924     for ( int i = 0; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next(), ++i )
 925     {
 926       double aCoordX = anIterX.Value();
 927       double aCoordY = anIterY.Value();
 928
 929       if ( i == thePointIndex )
 930       {
 931         // Insert our new point
 932         aListX->Append( thePoint.X() );
 933         aListY->Append( thePoint.Y() );
 934       }
 935
 936       aListX->Append( aCoordX );
 937       aListY->Append( aCoordY );
 938     }
 939   }
 940
 941   SetToUpdate( true );
 942 }
 943
 944 void HYDROData_PolylineXY::SetPoint( const int    theSectionIndex,
 945                                      const Point& thePoint,
 946                                      const int    thePointIndex )
 947 {
 948   Handle(TDataStd_RealList) aListX, aListY;
 949   getPointsLists( theSectionIndex, aListX, aListY );
 950
 951   if ( thePointIndex < 0 )
 952   {
 953     aListX->Prepend( thePoint.X() );
 954     aListY->Prepend( thePoint.Y() );
 955   }
 956   else if ( thePointIndex >= aListX->Extent() )
 957   {
 958     aListX->Append( thePoint.X() );
 959     aListY->Append( thePoint.Y() );
 960   }
 961   else
 962   {
 963     TColStd_ListOfReal anOldListX;
 964     anOldListX = aListX->List();
 965
 966     TColStd_ListOfReal anOldListY;
 967     anOldListY = aListY->List();
 968
 969     // Refill the existing lists
 970     aListX->Clear();
 971     aListY->Clear();
 972
 973     TColStd_ListIteratorOfListOfReal anIterX( anOldListX );
 974     TColStd_ListIteratorOfListOfReal anIterY( anOldListY );
 975     for ( int i = 0; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next(), ++i )
 976     {
 977       double aCoordX = anIterX.Value();
 978       double aCoordY = anIterY.Value();
 979
 980       if ( i == thePointIndex )
 981       {
 982         // Insert our new point instead of old one
 983         aCoordX = thePoint.X();
 984         aCoordY = thePoint.Y();
 985       }
 986
 987       aListX->Append( aCoordX );
 988       aListY->Append( aCoordY );
 989     }
 990   }
 991
 992   SetToUpdate( true );
 993 }
 994
 995 void HYDROData_PolylineXY::SetPoints( const int         theSectionIndex,
 996                                       const PointsList& thePoints )
 997 {
 998   Handle(TDataStd_RealList) aListX, aListY;
 999   getPointsLists( theSectionIndex, aListX, aListY );
1000
1001   aListX->Clear();
1002   aListY->Clear();
1003
1004   for ( int i = 1, n = thePoints.Length(); i <= n; ++i )
1005   {
1006     const Point& aPoint = thePoints.Value( i );
1007     aListX->Append( aPoint.X() );
1008     aListY->Append( aPoint.Y() );
1009   }
1010 }
1011
1012 void HYDROData_PolylineXY::RemovePoint( const int theSectionIndex,
1013                                         const int thePointIndex )
1014 {
1015   Handle(TDataStd_RealList) aListX, aListY;
1016   getPointsLists( theSectionIndex, aListX, aListY, false );
1017   if ( aListX.IsNull() || aListY.IsNull() || aListX->IsEmpty() )
1018     return;
1019
1020   if ( aListX->Extent() == 1 )
1021   {
1022     removePointsLists( theSectionIndex );
1023   }
1024   else
1025   {
1026     TColStd_ListOfReal anOldListX;
1027     anOldListX = aListX->List();
1028
1029     TColStd_ListOfReal anOldListY;
1030     anOldListY = aListY->List();
1031
1032     // Refill the existing lists
1033     aListX->Clear();
1034     aListY->Clear();
1035
1036     TColStd_ListIteratorOfListOfReal anIterX( anOldListX );
1037     TColStd_ListIteratorOfListOfReal anIterY( anOldListY );
1038     for ( int i = 0; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next(), ++i )
1039     {
1040       if ( i == thePointIndex )
1041         continue; // skip index to remove
1042
1043       aListX->Append( anIterX.Value() );
1044       aListY->Append( anIterY.Value() );
1045     }
1046   }
1047
1048   SetToUpdate( true );
1049 }
1050
1051 HYDROData_PolylineXY::PointsList HYDROData_PolylineXY::GetPoints( const int theSectionIndex ) const
1052 {
1053   PointsList aResList;
1054
1055   Handle(TDataStd_RealList) aListX, aListY;
1056   getPointsLists( theSectionIndex, aListX, aListY, false );
1057   if ( aListX.IsNull() || aListY.IsNull() || aListX->IsEmpty() )
1058     return aResList;
1059
1060   TColStd_ListIteratorOfListOfReal anIterX( aListX->List() );
1061   TColStd_ListIteratorOfListOfReal anIterY( aListY->List() );
1062   for ( ; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next() )
1063   {
1064     Point aPoint( anIterX.Value(), anIterY.Value() );
1065     aResList.Append( aPoint );
1066   }
1067
1068   return aResList;
1069 }
1070
1071 QPainterPath HYDROData_PolylineXY::GetPainterPath() const
1072 {
1073   QPainterPath aPath;
1074
1075   NCollection_Sequence<TCollection_AsciiString>           aSectNames;
1076   NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
1077   NCollection_Sequence<bool>                              aSectClosures;
1078   GetSections( aSectNames, aSectTypes, aSectClosures );
1079
1080   for ( int aSectionId = 1, aNbSects = aSectNames.Size(); aSectionId <= aNbSects; aSectionId++ )
1081   {
1082     TCollection_AsciiString aSectName = aSectNames.Value( aSectionId );
1083     SectionType aSectionType = aSectTypes.Value( aSectionId );
1084     bool anIsSectionClosed = aSectClosures.Value( aSectionId );
1085
1086     PointsList aSectPointsList = GetPoints( aSectionId - 1 );
1087     if ( aSectPointsList.IsEmpty() )
1088       continue;
1089
1090     NCollection_Sequence<gp_XYZ> aPoints;
1091     for( int i = 1, n = aSectPointsList.Size(); i <= n; ++i )
1092     {
1093       const Point& aSectPoint = aSectPointsList.Value( i );
1094
1095       gp_XYZ aPoint( aSectPoint.X(), aSectPoint.Y(), 0.0 );
1096       aPoints.Append( aPoint );
1097     }
1098
1099     BuildPainterPath( aPath, aSectionType, anIsSectionClosed, aPoints );
1100   }
1101
1102   return aPath;
1103 }
1104
1105 void HYDROData_PolylineXY::UpdateLocalCS( double theDx, double theDy )
1106 {
1107   NCollection_Sequence<TCollection_AsciiString>           aSectNames;
1108   NCollection_Sequence<HYDROData_PolylineXY::SectionType> aSectTypes;
1109   NCollection_Sequence<bool>                              aSectClosures;
1110   GetSections( aSectNames, aSectTypes, aSectClosures );
1111
1112   gp_XY aDelta( theDx, theDy );
1113   for ( int i = 0, aNbSects = aSectNames.Size(); i < aNbSects; i++ )
1114   {
1115     PointsList aPoints = GetPoints( i );
1116     for( int j = 1, n = aPoints.Size(); j <= n; ++j )
1117     {
1118       Point& aPoint = aPoints.ChangeValue( j );
1119       aPoint += aDelta;
1120     }
1121     SetPoints( i, aPoints );
1122   }
1123   SetToUpdate( true );
1124 }
1125
1126
1127