X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_PolylineXY.cxx;h=d418144a2e1865ef4b2fa25d71411bc399ed1274;hb=4a4e234c1bde9db8ac7e1020842ab355e03335b0;hp=f651ac2a4c33e3cce6c7bbca930023c275cd9a96;hpb=deed826b2d6c39ba2ed410108cdf54d64cded321;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_PolylineXY.cxx b/src/HYDROData/HYDROData_PolylineXY.cxx index f651ac2a..d418144a 100755 --- a/src/HYDROData/HYDROData_PolylineXY.cxx +++ b/src/HYDROData/HYDROData_PolylineXY.cxx @@ -1,16 +1,96 @@ #include "HYDROData_PolylineXY.h" +#include "HYDROData_BSplineOperation.h" +#include "HYDROData_Document.h" +#include "HYDROData_ShapesTool.h" #include "HYDROData_Tool.h" +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include + +#include + +#include + +#include #include +#include + +#include + +#include +#include +#include #include +#include + +#include +#include +#include +#include #include +#include + +#include +#include +#include +#include +#include +#include + +#include +#include +#include -#include +static const Standard_GUID GUID_IS_UNEDITABLE("e5799736-9030-4051-91a4-2e58321fa153"); +const double LOCAL_SELECTION_TOLERANCE = 0.0001; + +TCollection_AsciiString getUniqueSectionName( const NCollection_Sequence& theNamesSeq ) +{ + NCollection_Map aNamesMap; + + for ( int i = 1, n = theNamesSeq.Size(); i <= n; ++i ) + { + const TCollection_AsciiString& aSectName = theNamesSeq.Value( i ); + aNamesMap.Add( aSectName ); + } + + TCollection_AsciiString aResName; + + int aPrefIdx = 1; + do + { + aResName = TCollection_AsciiString( "Section_" ) + aPrefIdx; + ++aPrefIdx; + } + while ( aNamesMap.Contains( aResName ) ); + + return aResName; +} + +TCollection_AsciiString getUniqueSectionName( const Handle(TDataStd_ExtStringList)& theNamesList ) +{ + NCollection_Sequence aNamesSeq; + + TDataStd_ListIteratorOfListOfExtendedString aNamesIter( theNamesList->List() ); + for ( ; aNamesIter.More(); aNamesIter.Next() ) + aNamesSeq.Append( aNamesIter.Value() ); + + return getUniqueSectionName( aNamesSeq ); +} IMPLEMENT_STANDARD_HANDLE(HYDROData_PolylineXY, HYDROData_IPolyline) IMPLEMENT_STANDARD_RTTIEXT(HYDROData_PolylineXY, HYDROData_IPolyline) @@ -24,155 +104,929 @@ HYDROData_PolylineXY::~HYDROData_PolylineXY() { } -TopoDS_Wire HYDROData_PolylineXY::GetWire() const +QStringList HYDROData_PolylineXY::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const { - // TODO - return TopoDS_Wire(); + QStringList aResList = dumpObjectCreation( theTreatedObjects ); + QString aPolylineName = GetObjPyName(); + + // Set polilyne data + NCollection_Sequence aSectNames; + NCollection_Sequence aSectTypes; + NCollection_Sequence aSectClosures; + GetSections( aSectNames, aSectTypes, aSectClosures ); + + for ( int i = 1, n = aSectNames.Size(); i <= n; ++i ) + { + const TCollection_AsciiString& aSectName = aSectNames.Value( i ); + const SectionType& aSectType = aSectTypes.Value( i ); + bool aSectClosure = aSectClosures.Value( i ); + + aResList << QString( "%1.AddSection( \"%2\", %3, %4 );" ).arg( aPolylineName ) + .arg( aSectName.ToCString() ).arg( aSectType ).arg( aSectClosure ); + + HYDROData_IPolyline::PointsList aSectPointsList = GetPoints( i - 1 ); + for ( int k = 1, aNbPoints = aSectPointsList.Size(); k <= aNbPoints; ++k ) + { + const Point& aSectPoint = aSectPointsList.Value( k ); + + aResList << QString( "%1.AddPoint( %2, gp_XY( %3, %4 ) );" ).arg( aPolylineName ) + .arg( i - 1 ).arg( aSectPoint.X() ).arg( aSectPoint.Y() ); + } + } + + aResList << QString( "" ); + aResList << QString( "%1.Update();" ).arg( aPolylineName ); + aResList << QString( "" ); + + return aResList; } -int HYDROData_PolylineXY::NbSections() const +QVariant HYDROData_PolylineXY::GetDataVariant() { - return 1; + QPainterPath aPath = GetPainterPath(); + + QVariant aVarData; + aVarData.setValue( aPath ); + + return aVarData; } -void HYDROData_PolylineXY::AddSection( const bool /*theIsClosed*/ ) +QColor HYDROData_PolylineXY::DefaultWireColor() { + return QColor( Qt::red ); } -bool HYDROData_PolylineXY::IsClosedSection( const int /*theSectionIndex*/ ) const +TopoDS_Shape HYDROData_PolylineXY::GetShape() const { - return false; + return getPolylineShape(); } -void HYDROData_PolylineXY::RemoveSection( const int /*theSectionIndex*/ ) +bool convertEdgeToSection( const TopoDS_Edge& theEdge, + NCollection_Sequence& theSectNames, + NCollection_Sequence& theSectTypes, + NCollection_Sequence& theSectClosures, + NCollection_Sequence& theSectPoints, + const bool theIsCanBeClosed ) { + Standard_Real aFirst = 0.0, aLast = 0.0; + Handle(Geom_Curve) anEdgeGeomCurve = BRep_Tool::Curve( theEdge, aFirst, aLast ); + if ( anEdgeGeomCurve.IsNull() ) + return false; + + TCollection_AsciiString aSectName = getUniqueSectionName( theSectNames ); + bool anIsEdgeClosed = anEdgeGeomCurve->IsClosed(); + + HYDROData_PolylineXY::SectionType aSectionType = HYDROData_PolylineXY::SECTION_POLYLINE; + HYDROData_PolylineXY::PointsList aPointsList; + + if ( anEdgeGeomCurve->IsKind( STANDARD_TYPE(Geom_Line) ) ) + { + Handle(Geom_Line) aGeomLine = Handle(Geom_Line)::DownCast( anEdgeGeomCurve ); + + gp_Pnt aFirstPoint, aLastPoint; + aGeomLine->D0( aFirst, aFirstPoint ); + aGeomLine->D0( aLast, aLastPoint ); + + HYDROData_PolylineXY::Point aSectFirstPoint( aFirstPoint.X(), aFirstPoint.Y() ); + aPointsList.Append( aSectFirstPoint ); + + HYDROData_PolylineXY::Point aSectLastPoint( aLastPoint.X(), aLastPoint.Y() ); + aPointsList.Append( aSectLastPoint ); + } + else if ( anEdgeGeomCurve->IsKind( STANDARD_TYPE(Geom_BSplineCurve) ) ) + { + aSectionType = HYDROData_PolylineXY::SECTION_SPLINE; + + Handle(Geom_BSplineCurve) aGeomSpline = + Handle(Geom_BSplineCurve)::DownCast( anEdgeGeomCurve ); + + int aNbKnots = aGeomSpline->NbKnots(); + + TColStd_Array1OfReal aSplineKnots( 1, aNbKnots ); + aGeomSpline->Knots( aSplineKnots ); + + // Decrease the number of imported knots because of last one + // knot is the closing point which are the start point + if ( anIsEdgeClosed ) aNbKnots--; + + for ( int i = 1; i <= aNbKnots; ++i ) + { + const Standard_Real& aKnot = aSplineKnots.Value( i ); + + gp_Pnt aPoint; + aGeomSpline->D0( aKnot, aPoint ); + + HYDROData_PolylineXY::Point aSectPoint( aPoint.X(), aPoint.Y() ); + aPointsList.Append( aSectPoint ); + } + } + else + { + // Other curve types are not supported + return false; + } + + if ( aPointsList.IsEmpty() ) + return false; + + theSectNames.Append( aSectName ); + theSectTypes.Append( aSectionType ); + theSectClosures.Append( anIsEdgeClosed ); + theSectPoints.Append( aPointsList ); + + return true; +} + +bool HYDROData_PolylineXY::ImportShape( const TopoDS_Shape& theShape ) +{ + if ( theShape.IsNull() ) + return false; + RemoveSections(); + + bool anIsCanBeImported = false; + + NCollection_Sequence aSectNames; + NCollection_Sequence aSectTypes; + NCollection_Sequence aSectClosures; + NCollection_Sequence aSectPoints; + + if ( theShape.ShapeType() == TopAbs_EDGE ) + { + TopoDS_Edge anEdge = TopoDS::Edge( theShape ); + anIsCanBeImported = convertEdgeToSection( anEdge, aSectNames, aSectTypes, aSectClosures, aSectPoints, true ); + } + else if ( theShape.ShapeType() == TopAbs_WIRE ) + { + TopTools_SequenceOfShape anEdges; + HYDROData_ShapesTool::ExploreShapeToShapes( theShape, TopAbs_EDGE, anEdges ); + + anIsCanBeImported = !anEdges.IsEmpty(); + for ( int i = 1, n = anEdges.Length(); i <= n && anIsCanBeImported; ++i ) + { + TopoDS_Edge aWireEdge = TopoDS::Edge( anEdges.Value( i ) ); + anIsCanBeImported = convertEdgeToSection( aWireEdge, aSectNames, aSectTypes, aSectClosures, aSectPoints, false ); + } + } + + if ( anIsCanBeImported ) + { + for ( int i = 1, n = aSectNames.Length(); i <= n; ++i ) + { + const TCollection_AsciiString& aSectName = aSectNames.Value( i ); + const SectionType& aSectType = aSectTypes.Value( i ); + bool anIsSectionClosed = aSectClosures.Value( i ); + const PointsList& aSectPointsList = aSectPoints( i ); + + AddSection( aSectName, aSectType, anIsSectionClosed ); + SetPoints( i - 1, aSectPointsList ); + } + } + else + { + TopoDS_Shape aShape = theShape; + + if ( theShape.ShapeType() == TopAbs_EDGE ) + { + // We make the wire from incoming edge because of other algorithms + // are waiting at least the wire from polyline + TopoDS_Edge anEdge = TopoDS::Edge( theShape ); + BRepBuilderAPI_MakeWire aMakeWire( anEdge ); + aMakeWire.Build(); + if ( aMakeWire.IsDone() ) + aShape = aMakeWire.Wire(); + } + + gp_Pln aPlane( gp_Pnt( 0, 0, 0 ), gp_Dir( 0, 0, 1 ) ); + BRepBuilderAPI_MakeFace aMakeFace( aPlane ); + aMakeFace.Build(); + BRepOffsetAPI_NormalProjection aProj( aMakeFace.Face() ); + aProj.Add( aShape ); + aProj.Build(); + TopoDS_Shape aResult; + if( aProj.IsDone() ) + aResult = aProj.Shape(); + + setPolylineShape( aResult ); + } + + setEditable( anIsCanBeImported ); + + return true; } -void HYDROData_PolylineXY::RemoveSections() +TopoDS_Wire HYDROData_PolylineXY::BuildWire( const SectionType& theType, + const bool& theIsClosed, + const NCollection_Sequence& thePoints ) { - removePointsLists( 0 ); + TopoDS_Wire aWire; + if( theType == SECTION_POLYLINE ) + { + BRepBuilderAPI_MakePolygon aMakeWire; + for ( int i = 1, n = thePoints.Length(); i <= n ; ++i ) + { + gp_XYZ aPoint = thePoints.Value( i ); + gp_Pnt aPnt( aPoint.X(), aPoint.Y(), aPoint.Z() ); + aMakeWire.Add( aPnt ); + } + if( theIsClosed ) + aMakeWire.Close(); + + if ( aMakeWire.IsDone() ) + aWire = aMakeWire.Wire(); + } + else //if( theType == PolylineSection::SECTION_SPLINE ) + { + BRepBuilderAPI_MakeWire aMakeWire; + + if ( thePoints.Size() > 1 ) + { + HYDROData_BSplineOperation aBSpline( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE ); + + TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge( aBSpline.Curve() ).Edge(); + aMakeWire.Add( anEdge ); + } + aMakeWire.Build(); + if ( aMakeWire.IsDone() ) + aWire = aMakeWire; + } + + return aWire; } -void HYDROData_PolylineXY::AddPoint( const int /*theSectionIndex*/, - const Point& thePoint, - const int thePointIndex ) +void HYDROData_PolylineXY::BuildPainterPath( QPainterPath& thePath, + const SectionType& theType, + const bool& theIsClosed, + const NCollection_Sequence& thePoints ) { - double aNewCoordU = thePoint.X(); - double aNewCoordZ = thePoint.Y(); + if ( thePoints.IsEmpty() ) + return; - Handle(TDataStd_RealList) aListU, aListZ; - getPointsLists( 0, aListU, aListZ ); + if ( theType == SECTION_POLYLINE ) + { + const gp_XYZ& aFirstPoint = thePoints.Value( 1 ); + thePath.moveTo( aFirstPoint.X(), aFirstPoint.Y() ); + + for( int i = 2, n = thePoints.Size(); i <= n; ++i ) + { + const gp_XYZ& aSectPoint = thePoints.Value( i ); + + thePath.lineTo( aSectPoint.X(), aSectPoint.Y() ); + } - if ( aListU->IsEmpty() || aNewCoordU > aListU->Last() ) + if( theIsClosed ) + thePath.closeSubpath(); + } + else { - aListU->Append( aNewCoordU ); - aListZ->Append( aNewCoordZ ); - return; + HYDROData_BSplineOperation aBSpline( thePoints, theIsClosed, LOCAL_SELECTION_TOLERANCE ); + aBSpline.ComputePath( thePath ); } - else if ( aNewCoordU < aListU->First() ) +} + +void HYDROData_PolylineXY::Update() +{ + if ( !IsEditable() ) { - aListU->Prepend( aNewCoordU ); - aListZ->Prepend( aNewCoordZ ); + // If polyline is not editable we no need to update it wire + SetToUpdate( false ); return; } - TColStd_ListOfReal anOldListU; - anOldListU = aListU->List(); + HYDROData_IPolyline::Update(); + + NCollection_Sequence aSectNames; + NCollection_Sequence aSectTypes; + NCollection_Sequence aSectClosures; + GetSections( aSectNames, aSectTypes, aSectClosures ); + + BRepBuilderAPI_MakeWire aMakeWire; + + TopTools_ListOfShape aSectionWiresList; + + for ( int aSectionId = 1, aNbSects = aSectNames.Size(); aSectionId <= aNbSects; aSectionId++ ) + { + TCollection_AsciiString aSectName = aSectNames.Value( aSectionId ); + SectionType aSectionType = aSectTypes.Value( aSectionId ); + bool anIsSectionClosed = aSectClosures.Value( aSectionId ); + + PointsList aSectPointsList = GetPoints( aSectionId - 1 ); + if ( aSectPointsList.IsEmpty() ) + continue; + + NCollection_Sequence aPoints; + for( int i = 1, n = aSectPointsList.Size(); i <= n; ++i ) + { + const Point& aSectPoint = aSectPointsList.Value( i ); - TColStd_ListOfReal anOldListZ; - anOldListZ = aListZ->List(); + gp_XYZ aPoint( aSectPoint.X(), aSectPoint.Y(), 0.0 ); + aPoints.Append( aPoint ); + } - // Crsat new lists - removePointsLists( 0 ); - getPointsLists( 0, aListU, aListZ ); + TopoDS_Wire aSectionWire = BuildWire( aSectionType, anIsSectionClosed, aPoints ); + if ( !aSectionWire.IsNull() ) { + aSectionWiresList.Append( aSectionWire ); + aMakeWire.Add( aSectionWire ); + } + } +// all input wires in the + Handle(TopTools_HSequenceOfShape) aSeqWires = new TopTools_HSequenceOfShape; + Handle(TopTools_HSequenceOfShape) aSeqEdges = new TopTools_HSequenceOfShape; + TopTools_ListIteratorOfListOfShape it(aSectionWiresList); + for(;it.More();it.Next()) + { + TopExp_Explorer it2(it.Value(), TopAbs_EDGE); + for(;it2.More();it2.Next()) + aSeqEdges->Append(it2.Current()); + } - bool anIsInserted = false; - TColStd_ListIteratorOfListOfReal anIterU( anOldListU ); - TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ ); - for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() ) + BRep_Builder aBB; + TopoDS_Compound aCmp; + TopoDS_Shape aResult; + aBB.MakeCompound(aCmp); + if(aSeqEdges->Length() >1) { - double aCoordU = anIterU.Value(); - double aCoordZ = anIterZ.Value(); + ShapeAnalysis_FreeBounds::ConnectEdgesToWires(aSeqEdges,1E-5,Standard_False,aSeqWires); - if ( !anIsInserted ) + if( aSeqWires->Length()==1 ) + aResult = aSeqWires->Value( 1 ); + else { - if ( ValuesEquals( aNewCoordU, aCoordU ) ) + for (Standard_Integer i = 1; i <= aSeqWires->Length();i++) { - // Just update Z value - aCoordZ = aNewCoordZ; - anIsInserted = true; + const TopoDS_Shape& aS1 = aSeqWires->Value(i); + aBB.Add(aCmp, aS1); } - else if ( aNewCoordU < aCoordU ) + aResult = aCmp; + } + } + else if (aSeqEdges->Length() == 1) + { + BRepBuilderAPI_MakeWire mkWire (TopoDS::Edge(aSeqEdges->Value(1))); + if (mkWire.IsDone()) + aResult = mkWire.Wire(); + } + + setPolylineShape( aResult ); +} + +bool HYDROData_PolylineXY::IsEditable() const +{ + return !myLab.IsAttribute( GUID_IS_UNEDITABLE ); +} + +void HYDROData_PolylineXY::setEditable( const bool theIsEditable ) +{ + if ( !theIsEditable ) + TDataStd_UAttribute::Set( myLab, GUID_IS_UNEDITABLE ); + else + myLab.ForgetAttribute( GUID_IS_UNEDITABLE ); +} + +/** + * Returns true if polyline is closed + */ +bool HYDROData_PolylineXY::IsClosed( const bool theIsSimpleCheck ) const +{ + bool anIsClosed = false; + + TopoDS_Shape aShape = GetShape(); + if ( aShape.IsNull() ) + return anIsClosed; + + TopTools_SequenceOfShape aWires; + HYDROData_ShapesTool::ExploreShapeToShapes( aShape, TopAbs_WIRE, aWires ); + + int aNbWires = aWires.Length(); + if ( theIsSimpleCheck ) + { + anIsClosed = aNbWires > 0; + for ( int i = 1; i <= aNbWires && anIsClosed; ++i ) + { + const TopoDS_Shape& aWire = aWires.Value( i ); + anIsClosed = BRep_Tool::IsClosed( aWire ); + } + } + else + { + anIsClosed = aNbWires == 1 && BRep_Tool::IsClosed( aWires.First() ); + } + + return anIsClosed; +} + +double HYDROData_PolylineXY::GetDistance( const int theSectionIndex, + const int thePointIndex ) const +{ + double aResDistance = -1; + if ( theSectionIndex < 0 || theSectionIndex >= NbSections() ) + return aResDistance; + + if ( thePointIndex == 0 ) + return 0.0; + + SectionType aSectionType = GetSectionType( theSectionIndex ); + bool anIsSectionClosed = IsClosedSection( theSectionIndex ); + PointsList aSectPointsList = GetPoints( theSectionIndex ); + if ( thePointIndex < 0 || thePointIndex >= aSectPointsList.Size() ) + return aResDistance; + + if ( aSectionType == SECTION_POLYLINE ) + { + aResDistance = 0.0; + + Point aPrevPoint = aSectPointsList.Value( 1 ); + for ( int i = 2, aNbPoints = aSectPointsList.Size(); i <= aNbPoints; ++i ) + { + const Point& aSectPoint = aSectPointsList.Value( i ); + aResDistance += gp_Pnt2d( aPrevPoint ).Distance( aSectPoint ); + aPrevPoint = aSectPoint; + + if ( thePointIndex == i - 1 ) + break; + } + } + else + { + gp_XYZ aPointToTest; + + int aSectNbPoints = aSectPointsList.Size(); + NCollection_Sequence aPoints; + for( int i = 1 ; i <= aSectNbPoints; ++i ) + { + const Point& aSectPoint = aSectPointsList.Value( i ); + + gp_XYZ aPoint( aSectPoint.X(), aSectPoint.Y(), 0.0 ); + aPoints.Append( aPoint ); + + if ( thePointIndex == i - 1 ) + aPointToTest = aPoint; + } + + HYDROData_BSplineOperation aBSpline( aPoints, anIsSectionClosed, LOCAL_SELECTION_TOLERANCE ); + + Quantity_Parameter aFirstParam = aBSpline.Curve()->FirstParameter(); + Quantity_Parameter aSecondParam = aBSpline.Curve()->LastParameter(); + + if ( thePointIndex != aSectNbPoints - 1 ) + { + GeomAPI_ProjectPointOnCurve aProject( aPointToTest, aBSpline.Curve() ); + aSecondParam = aProject.LowerDistanceParameter(); + } + + GeomAdaptor_Curve anAdap( aBSpline.Curve() ); + + aResDistance = GCPnts_AbscissaPoint::Length( anAdap, aFirstParam, aSecondParam ); + } + + return aResDistance; +} + +int HYDROData_PolylineXY::NbSections() const +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + + return !aClosuresList.IsNull() ? aClosuresList->Extent() : 0; +} + +void HYDROData_PolylineXY::AddSection( const TCollection_AsciiString& theSectName, + const SectionType theSectionType, + const bool theIsClosed ) +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList ); + + TCollection_ExtendedString aSectName( theSectName ); + if ( aSectName.Length() <= 0 ) + aSectName = getUniqueSectionName( aNamesList ); + + aNamesList->Append( aSectName ); + aTypesList->Append( theSectionType ); + aClosuresList->Append( theIsClosed ); + + SetToUpdate( true ); +} + +TCollection_AsciiString HYDROData_PolylineXY::GetSectionName( const int theSectionIndex ) const +{ + TCollection_AsciiString aResName; + + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aNamesList.IsNull() || theSectionIndex >= aNamesList->Extent() ) + return aResName; + + TDataStd_ListIteratorOfListOfExtendedString aNamesIter( aNamesList->List() ); + for ( int i = 0; aNamesIter.More() && i != theSectionIndex; aNamesIter.Next(), ++i ); + + if ( aNamesIter.More() ) + aResName = aNamesIter.Value(); + + return aResName; +} + +void HYDROData_PolylineXY::SetSectionName( const int theSectionIndex, + const TCollection_AsciiString& theSectionName ) +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aNamesList.IsNull() || theSectionIndex >= aNamesList->Extent() ) + return; + + TDataStd_ListOfExtendedString anOldNamesList; + anOldNamesList = aNamesList->List(); + + // Refill the existing list + aNamesList->Clear(); + + TCollection_ExtendedString aNewSectName = theSectionName; + + TDataStd_ListIteratorOfListOfExtendedString aNamesIter( anOldNamesList ); + for ( int i = 0; aNamesIter.More(); aNamesIter.Next(), ++i ) + aNamesList->Append( i == theSectionIndex ? aNewSectName : aNamesIter.Value() ); + + SetToUpdate( true ); +} + +HYDROData_PolylineXY::SectionType HYDROData_PolylineXY::GetSectionType( const int theSectionIndex ) const +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aTypesList.IsNull() || theSectionIndex >= aTypesList->Extent() ) + return SECTION_POLYLINE; + + TColStd_ListIteratorOfListOfInteger aTypesIter( aTypesList->List() ); + for ( int i = 0; aTypesIter.More() && i != theSectionIndex; aTypesIter.Next(), ++i ); + + return aTypesIter.More() ? (SectionType)aTypesIter.Value() : SECTION_POLYLINE; +} + +void HYDROData_PolylineXY::SetSectionType( const int theSectionIndex, + const SectionType theSectionType ) +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aTypesList.IsNull() || theSectionIndex >= aTypesList->Extent() ) + return; + + TColStd_ListOfInteger anOldTypesList; + anOldTypesList = aTypesList->List(); + + // Refill the existing list + aTypesList->Clear(); + + TColStd_ListIteratorOfListOfInteger aTypesIter( anOldTypesList ); + for ( int i = 0; aTypesIter.More(); aTypesIter.Next(), ++i ) + aTypesList->Append( i == theSectionIndex ? theSectionType : aTypesIter.Value() ); + + SetToUpdate( true ); +} + +bool HYDROData_PolylineXY::IsClosedSection( const int theSectionIndex ) const +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aClosuresList.IsNull() || theSectionIndex >= aClosuresList->Extent() ) + return false; + + TDataStd_ListIteratorOfListOfByte aClosuresIter( aClosuresList->List() ); + for ( int i = 0; aClosuresIter.More() && i != theSectionIndex; aClosuresIter.Next(), ++i ); + + return aClosuresIter.More() ? (bool)aClosuresIter.Value() : false; +} + +void HYDROData_PolylineXY::SetSectionClosed( const int theSectionIndex, + const bool theIsClosed ) +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aClosuresList.IsNull() || theSectionIndex >= aClosuresList->Extent() ) + return; + + TDataStd_ListOfByte anOldClosuresList; + anOldClosuresList = aClosuresList->List(); + + // Refill the existing list + aClosuresList->Clear(); + + TDataStd_ListIteratorOfListOfByte aClosuresIter( anOldClosuresList ); + for ( int i = 0; aClosuresIter.More(); aClosuresIter.Next(), ++i ) + aClosuresList->Append( i == theSectionIndex ? theIsClosed : (bool)aClosuresIter.Value() ); + + SetToUpdate( true ); +} + +void HYDROData_PolylineXY::GetSections( NCollection_Sequence& theSectNames, + NCollection_Sequence& theSectTypes, + NCollection_Sequence& theSectClosures ) const +{ + theSectNames.Clear(); + theSectTypes.Clear(); + theSectClosures.Clear(); + + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aNamesList.IsNull() || aTypesList.IsNull() || aClosuresList.IsNull() ) + return; + + TDataStd_ListIteratorOfListOfExtendedString aNamesIter( aNamesList->List() ); + TColStd_ListIteratorOfListOfInteger aTypesIter( aTypesList->List() ); + TDataStd_ListIteratorOfListOfByte aClosuresIter( aClosuresList->List() ); + for ( ; aNamesIter.More() && aTypesIter.More() && aClosuresIter.More(); + aNamesIter.Next(), aTypesIter.Next(), aClosuresIter.Next() ) + { + const TCollection_ExtendedString& aSectName = aNamesIter.Value(); + SectionType aSectType = (SectionType)aTypesIter.Value(); + bool aSectClosures = aClosuresIter.Value(); + + theSectNames.Append( aSectName ); + theSectTypes.Append( aSectType ); + theSectClosures.Append( aSectClosures ); + } +} + +void HYDROData_PolylineXY::RemoveSection( const int theSectionIndex ) +{ + Handle(TDataStd_ExtStringList) aNamesList; + Handle(TDataStd_IntegerList) aTypesList; + Handle(TDataStd_BooleanList) aClosuresList; + getSectionsLists( aNamesList, aTypesList, aClosuresList, false ); + if ( aNamesList.IsNull() || theSectionIndex >= aNamesList->Extent() ) + return; + + if ( aNamesList->Extent() == 1 ) + { + removeSectionsLists(); + removePointsLists(); + } + else + { + TDataStd_ListOfExtendedString anOldNamesList; + anOldNamesList = aNamesList->List(); + + TColStd_ListOfInteger anOldTypesList; + anOldTypesList = aTypesList->List(); + + TDataStd_ListOfByte anOldClosuresList; + anOldClosuresList = aClosuresList->List(); + + // Refill the existing lists + aNamesList->Clear(); + aTypesList->Clear(); + aClosuresList->Clear(); + + TDataStd_ListIteratorOfListOfExtendedString aNamesIter( anOldNamesList ); + TColStd_ListIteratorOfListOfInteger aTypesIter( anOldTypesList ); + TDataStd_ListIteratorOfListOfByte aClosuresIter( anOldClosuresList ); + for ( int i = 0; aNamesIter.More() && aTypesIter.More() && aClosuresIter.More(); + aNamesIter.Next(), aTypesIter.Next(), aClosuresIter.Next(), ++i ) + { + if ( i == theSectionIndex ) + continue; // skip index to remove + + aNamesList->Append( aNamesIter.Value() ); + aTypesList->Append( aTypesIter.Value() ); + aClosuresList->Append( (bool)aClosuresIter.Value() ); + } + + // Remove points that belongs to removed section + removePointsLists( theSectionIndex ); + } + + SetToUpdate( true ); +} + +void HYDROData_PolylineXY::RemoveSections() +{ + removeSectionsLists(); + removePointsLists(); + SetToUpdate( true ); +} + +void HYDROData_PolylineXY::AddPoint( const int theSectionIndex, + const Point& thePoint, + const int thePointIndex ) +{ + Handle(TDataStd_RealList) aListX, aListY; + getPointsLists( theSectionIndex, aListX, aListY ); + + if ( thePointIndex < 0 || thePointIndex >= aListX->Extent() ) + { + aListX->Append( thePoint.X() ); + aListY->Append( thePoint.Y() ); + } + else + { + TColStd_ListOfReal anOldListX; + anOldListX = aListX->List(); + + TColStd_ListOfReal anOldListY; + anOldListY = aListY->List(); + + // Refill the existing lists + aListX->Clear(); + aListY->Clear(); + + TColStd_ListIteratorOfListOfReal anIterX( anOldListX ); + TColStd_ListIteratorOfListOfReal anIterY( anOldListY ); + for ( int i = 0; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next(), ++i ) + { + double aCoordX = anIterX.Value(); + double aCoordY = anIterY.Value(); + + if ( i == thePointIndex ) { - // Insert new point - aListU->Append( aNewCoordU ); - aListZ->Append( aNewCoordZ ); - anIsInserted = true; + // Insert our new point + aListX->Append( thePoint.X() ); + aListY->Append( thePoint.Y() ); } - } - aListU->Append( aCoordU ); - aListZ->Append( aCoordZ ); + aListX->Append( aCoordX ); + aListY->Append( aCoordY ); + } } + + SetToUpdate( true ); } void HYDROData_PolylineXY::SetPoint( const int theSectionIndex, - const int /*thePointIndex*/, - const Point& thePoint ) + const Point& thePoint, + const int thePointIndex ) +{ + Handle(TDataStd_RealList) aListX, aListY; + getPointsLists( theSectionIndex, aListX, aListY ); + + if ( thePointIndex < 0 ) + { + aListX->Prepend( thePoint.X() ); + aListY->Prepend( thePoint.Y() ); + } + else if ( thePointIndex >= aListX->Extent() ) + { + aListX->Append( thePoint.X() ); + aListY->Append( thePoint.Y() ); + } + else + { + TColStd_ListOfReal anOldListX; + anOldListX = aListX->List(); + + TColStd_ListOfReal anOldListY; + anOldListY = aListY->List(); + + // Refill the existing lists + aListX->Clear(); + aListY->Clear(); + + TColStd_ListIteratorOfListOfReal anIterX( anOldListX ); + TColStd_ListIteratorOfListOfReal anIterY( anOldListY ); + for ( int i = 0; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next(), ++i ) + { + double aCoordX = anIterX.Value(); + double aCoordY = anIterY.Value(); + + if ( i == thePointIndex ) + { + // Insert our new point instead of old one + aCoordX = thePoint.X(); + aCoordY = thePoint.Y(); + } + + aListX->Append( aCoordX ); + aListY->Append( aCoordY ); + } + } + + SetToUpdate( true ); +} + +void HYDROData_PolylineXY::SetPoints( const int theSectionIndex, + const PointsList& thePoints ) { - AddPoint( theSectionIndex, thePoint ); + Handle(TDataStd_RealList) aListX, aListY; + getPointsLists( theSectionIndex, aListX, aListY ); + + aListX->Clear(); + aListY->Clear(); + + for ( int i = 1, n = thePoints.Length(); i <= n; ++i ) + { + const Point& aPoint = thePoints.Value( i ); + aListX->Append( aPoint.X() ); + aListY->Append( aPoint.Y() ); + } } -void HYDROData_PolylineXY::RemovePoint( const int /*theSectionIndex*/, - const int thePointIndex ) +void HYDROData_PolylineXY::RemovePoint( const int theSectionIndex, + const int thePointIndex ) { - Handle(TDataStd_RealList) aListU, aListZ; - getPointsLists( 0, aListU, aListZ, false ); - if ( aListU.IsNull() || aListZ.IsNull() || aListU->IsEmpty() ) + Handle(TDataStd_RealList) aListX, aListY; + getPointsLists( theSectionIndex, aListX, aListY, false ); + if ( aListX.IsNull() || aListY.IsNull() || aListX->IsEmpty() ) return; - TColStd_ListOfReal anOldListU; - anOldListU = aListU->List(); + if ( aListX->Extent() == 1 ) + { + removePointsLists( theSectionIndex ); + } + else + { + TColStd_ListOfReal anOldListX; + anOldListX = aListX->List(); - TColStd_ListOfReal anOldListZ; - anOldListZ = aListZ->List(); + TColStd_ListOfReal anOldListY; + anOldListY = aListY->List(); - // Creat new lists - removePointsLists( 0 ); - getPointsLists( 0, aListU, aListZ ); + // Refill the existing lists + aListX->Clear(); + aListY->Clear(); - bool anIsInserted = false; - TColStd_ListIteratorOfListOfReal anIterU( anOldListU ); - TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ ); - for ( int i = 0; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next(), ++i ) - { - if ( i == thePointIndex ) - continue; // skip index to remove + TColStd_ListIteratorOfListOfReal anIterX( anOldListX ); + TColStd_ListIteratorOfListOfReal anIterY( anOldListY ); + for ( int i = 0; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next(), ++i ) + { + if ( i == thePointIndex ) + continue; // skip index to remove - aListU->Append( anIterU.Value() ); - aListZ->Append( anIterZ.Value() ); + aListX->Append( anIterX.Value() ); + aListY->Append( anIterY.Value() ); + } } + + SetToUpdate( true ); } -HYDROData_PolylineXY::PointsList HYDROData_PolylineXY::GetPoints( const int /*theSectionIndex*/ ) const +HYDROData_PolylineXY::PointsList HYDROData_PolylineXY::GetPoints( const int theSectionIndex ) const { PointsList aResList; - Handle(TDataStd_RealList) aListU, aListZ; - getPointsLists( 0, aListU, aListZ, false ); - if ( aListU.IsNull() || aListZ.IsNull() ) + Handle(TDataStd_RealList) aListX, aListY; + getPointsLists( theSectionIndex, aListX, aListY, false ); + if ( aListX.IsNull() || aListY.IsNull() || aListX->IsEmpty() ) return aResList; - TColStd_ListIteratorOfListOfReal anIterU( aListU->List() ); - TColStd_ListIteratorOfListOfReal anIterZ( aListZ->List() ); - for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() ) + TColStd_ListIteratorOfListOfReal anIterX( aListX->List() ); + TColStd_ListIteratorOfListOfReal anIterY( aListY->List() ); + for ( ; anIterX.More() && anIterY.More(); anIterX.Next(), anIterY.Next() ) { - Point aPoint( anIterU.Value(), anIterZ.Value() ); + Point aPoint( anIterX.Value(), anIterY.Value() ); aResList.Append( aPoint ); } return aResList; } +QPainterPath HYDROData_PolylineXY::GetPainterPath() const +{ + QPainterPath aPath; + + NCollection_Sequence aSectNames; + NCollection_Sequence aSectTypes; + NCollection_Sequence aSectClosures; + GetSections( aSectNames, aSectTypes, aSectClosures ); + + for ( int aSectionId = 1, aNbSects = aSectNames.Size(); aSectionId <= aNbSects; aSectionId++ ) + { + TCollection_AsciiString aSectName = aSectNames.Value( aSectionId ); + SectionType aSectionType = aSectTypes.Value( aSectionId ); + bool anIsSectionClosed = aSectClosures.Value( aSectionId ); + + PointsList aSectPointsList = GetPoints( aSectionId - 1 ); + if ( aSectPointsList.IsEmpty() ) + continue; + + NCollection_Sequence aPoints; + for( int i = 1, n = aSectPointsList.Size(); i <= n; ++i ) + { + const Point& aSectPoint = aSectPointsList.Value( i ); + + gp_XYZ aPoint( aSectPoint.X(), aSectPoint.Y(), 0.0 ); + aPoints.Append( aPoint ); + } + + BuildPainterPath( aPath, aSectionType, anIsSectionClosed, aPoints ); + } + + return aPath; +} +