2 #include "HYDROData_Stream.h"
4 #include "HYDROData_Document.h"
5 #include "HYDROData_PolylineXY.h"
6 #include "HYDROData_Profile.h"
7 #include "HYDROData_ShapesGroup.h"
8 #include "HYDROData_ShapesTool.h"
9 #include "HYDROData_IAltitudeObject.h"
11 #include <TDataStd_RealArray.hxx>
13 #include <Precision.hxx>
15 #include <NCollection_DataMap.hxx>
17 #include <TColStd_Array1OfReal.hxx>
18 #include <TColStd_ListOfReal.hxx>
19 #include <TColStd_ListIteratorOfListOfReal.hxx>
20 #include <TCollection_CompareOfReal.hxx>
21 #include <TColgp_Array1OfPnt.hxx>
22 #include <TColgp_HArray1OfPnt.hxx>
25 #include <TopoDS_Wire.hxx>
26 #include <TopoDS_Shell.hxx>
27 #include <TopoDS_Face.hxx>
28 #include <TopoDS_Edge.hxx>
29 #include <TopoDS_Vertex.hxx>
31 #include <TopExp_Explorer.hxx>
33 #include <Bnd_Box.hxx>
35 #include <BRep_Builder.hxx>
36 #include <BRepBuilderAPI_MakeEdge.hxx>
37 #include <BRepBuilderAPI_MakeWire.hxx>
38 #include <BRepBuilderAPI_MakeFace.hxx>
40 #include <BRepBndLib.hxx>
41 #include <BRepProj_Projection.hxx>
42 #include <BRepExtrema_ExtCC.hxx>
43 #include <BRepCheck_Analyzer.hxx>
53 #include <GeomAPI_Interpolate.hxx>
54 #include <Geom_BSplineCurve.hxx>
56 #include <TopTools_Array1OfShape.hxx>
58 #include <SortTools_QuickSortOfReal.hxx>
61 #include <QStringList>
63 //#define DEB_STREAM 1
65 //#define DEB_HASINT 1
66 //#define DEB_UPDATE 1
67 #include <BRepTools.hxx>
68 #include <TCollection_AsciiString.hxx>
71 #define PYTHON_STREAM_ID "KIND_STREAM"
73 typedef NCollection_DataMap<Standard_Real, Handle(HYDROData_Profile)> HYDROData_DataMapOfRealOfHDProfile;
75 IMPLEMENT_STANDARD_HANDLE(HYDROData_Stream,HYDROData_NaturalObject)
76 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Stream,HYDROData_NaturalObject)
79 HYDROData_Stream::HYDROData_Stream()
80 : HYDROData_NaturalObject()
84 HYDROData_Stream::~HYDROData_Stream()
88 QStringList HYDROData_Stream::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
92 Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
93 if ( aDocument.IsNull() )
96 QString aDocName = aDocument->GetDocPyName();
97 QString aStreamName = GetName();
99 aResList << QString( "%1 = %2.CreateObject( %3 );" )
100 .arg( aStreamName ).arg( aDocName ).arg( PYTHON_STREAM_ID );
101 aResList << QString( "%1.SetName( \"%2\" );" )
102 .arg( aStreamName ).arg( aStreamName );
103 aResList << QString( "" );
110 HYDROData_SequenceOfObjects HYDROData_Stream::GetAllReferenceObjects() const
112 HYDROData_SequenceOfObjects aResSeq = HYDROData_Object::GetAllReferenceObjects();
114 Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
115 if ( !aHydAxis.IsNull() )
116 aResSeq.Append( aHydAxis );
118 HYDROData_SequenceOfObjects aSeqOfProfiles = GetProfiles();
119 aResSeq.Append( aSeqOfProfiles );
124 TopoDS_Shape HYDROData_Stream::GetTopShape() const
126 return getTopShape();
129 TopoDS_Shape HYDROData_Stream::GetShape3D() const
134 Handle(Geom_BSplineCurve) HYDROData_Stream::buildInterpolationCurve(
135 const Handle(TColgp_HArray1OfPnt)& theArrayOfPnt )
137 Handle(Geom_BSplineCurve) aBSpline;
138 GeomAPI_Interpolate anInterpolator (theArrayOfPnt, Standard_False, 1.0e-5);
139 anInterpolator.Perform() ;
140 if (anInterpolator.IsDone())
141 aBSpline = anInterpolator.Curve();
145 void HYDROData_Stream::Update()
147 updateProfilesOrder();
151 void HYDROData_Stream::UpdatePrs()
153 HYDROData_NaturalObject::Update();
155 Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
156 HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
157 if ( aHydAxis.IsNull() || aRefProfiles.Length() < 2 )
160 bool anIsFirst = true;
161 gp_Pnt aPrevFirstPoint, aPrevLastPoint;
162 Handle(TColgp_HArray1OfPnt) anArrayOfFPnt = new TColgp_HArray1OfPnt(1, aRefProfiles.Length());
163 Handle(TColgp_HArray1OfPnt) anArrayOfLPnt = new TColgp_HArray1OfPnt(1, aRefProfiles.Length());
164 TopTools_Array1OfShape anArrOfProfiles(1, aRefProfiles.Length());
165 TopTools_Array1OfShape anArrOf2DProfiles(1, aRefProfiles.Length());
168 HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
169 for (int i=1 ; anIter.More(); anIter.Next(),i++ )
171 Handle(HYDROData_Profile) aProfile =
172 Handle(HYDROData_Profile)::DownCast( anIter.Value() );
173 if ( aProfile.IsNull() )
175 const TopoDS_Shape& aProf3d = aProfile->GetShape3D();
177 if ( !aProfile->GetLeftPoint( aPnt1 ) || !aProfile->GetRightPoint( aPnt2 ) )
179 anArrOfProfiles.SetValue(i,aProfile->GetShape3D());//aProfile->GetTopShape();
180 anArrOf2DProfiles.SetValue(i,aProfile->GetTopShape());
182 gp_Pnt aCurFirstPoint( aPnt1.X(), aPnt1.Y(), 0 ), aCurFP;
183 gp_Pnt aCurLastPoint( aPnt2.X(), aPnt2.Y(), 0 ), aCurLP;
184 TopoDS_Vertex aV1, aV2;
185 TopExp::Vertices(TopoDS::Wire(aProf3d), aV1, aV2);
186 gp_Pnt aP1 = BRep_Tool::Pnt(aV1);
187 if(aP1.X() == aPnt1.X() && aP1.Y() == aPnt1.Y())
191 aP1 = BRep_Tool::Pnt(aV2);
192 if(aP1.X() == aPnt2.X() && aP1.Y() == aPnt2.Y())
196 anArrayOfFPnt->SetValue(i,aCurFP);
197 anArrayOfLPnt->SetValue(i,aCurLP);
200 // Construct of the 3D presentation
201 Handle(Geom_BSplineCurve) aBSpline = buildInterpolationCurve (anArrayOfFPnt);
202 if(aBSpline.IsNull())
204 TopoDS_Edge anEdgLeft, anEdgRight;
205 BRepBuilderAPI_MakeEdge aMakeEdge(aBSpline);
206 if(aMakeEdge.IsDone())
207 anEdgLeft = aMakeEdge.Edge();
208 if(anEdgLeft.IsNull())
211 aBSpline = buildInterpolationCurve (anArrayOfLPnt);
212 if(aBSpline.IsNull())
214 aMakeEdge.Init(aBSpline);
215 if(aMakeEdge.IsDone())
216 anEdgRight = aMakeEdge.Edge();
217 if(anEdgRight.IsNull())
220 TopoDS_Compound aCmp;
221 aBB.MakeCompound(aCmp);
222 anIter.Init( aRefProfiles );
223 for (int i=1 ; i < anArrOfProfiles.Length() +1; i++ )
224 aBB.Add(aCmp, anArrOfProfiles.Value(i));
225 aBB.Add(aCmp,anEdgLeft);
226 aBB.Add(aCmp,anEdgRight);
227 BRepCheck_Analyzer aCh(aCmp);
232 BRepTools::Write(aCmp, "str3d.brep");
237 // Construct the top presentation
238 for(int i=1;i<= anArrayOfLPnt->Length();i++) {
239 gp_Pnt aPnt = anArrayOfFPnt->Value(i);
240 aPnt.SetZ(.0); // make 2d
241 anArrayOfFPnt->SetValue(i, aPnt);
242 aPnt = anArrayOfLPnt->Value(i);
244 anArrayOfLPnt->SetValue(i, aPnt);
247 aBSpline = buildInterpolationCurve (anArrayOfFPnt);
248 if(aBSpline.IsNull())
250 aMakeEdge.Init(aBSpline);
251 if(aMakeEdge.IsDone())
252 anEdgLeft = aMakeEdge.Edge();
254 aBSpline = buildInterpolationCurve (anArrayOfLPnt);
255 if(aBSpline.IsNull())
257 aMakeEdge.Init(aBSpline);
258 if(aMakeEdge.IsDone())
259 anEdgRight = aMakeEdge.Edge();
260 if(anEdgRight.IsNull())
262 BRepBuilderAPI_MakeEdge aMakeEdge2(anArrayOfFPnt->Value(1),anArrayOfLPnt->Value(1));
263 TopoDS_Edge aBotEdge, aTopEdge;
264 if(aMakeEdge2.IsDone())
265 aBotEdge = aMakeEdge2.Edge();
266 BRepBuilderAPI_MakeEdge aMakeEdge3(anArrayOfFPnt->Value(anArrayOfFPnt->Length()),anArrayOfLPnt->Value(anArrayOfLPnt->Length()));
267 if(aMakeEdge3.IsDone())
268 aTopEdge = aMakeEdge3.Edge();
270 BRepBuilderAPI_MakeWire aMakeWire( aBotEdge, anEdgLeft, aTopEdge,anEdgRight);
271 TopoDS_Wire aSectProfileWire;
272 if(aMakeWire.IsDone())
273 aSectProfileWire = aMakeWire.Wire();
274 BRepBuilderAPI_MakeFace aMakeFace( aSectProfileWire, Standard_True );
277 if( aMakeFace.IsDone() )
278 aFace = aMakeFace.Face();
280 aBB.MakeCompound(aCmp);
282 for(int i=1;i <= anArrOf2DProfiles.Length(); i++)
283 aBB.Add(aCmp,anArrOf2DProfiles.Value(i));
289 BRepTools::Write(aCmp, "str2d.brep");
294 // Create the stream groups
295 createGroupObjects();
297 // This peace of code is for testing of functionality of altitude,
298 // will be removed by adv when testing will be finished
299 Handle(HYDROData_IAltitudeObject) anAltObj = GetAltitudeObject();
300 gp_XY aTestPnt( 5, 0 );
301 anAltObj->GetAltitudeForPoint( aTestPnt );
306 QColor HYDROData_Stream::DefaultFillingColor()
308 return QColor( Qt::green );
311 QColor HYDROData_Stream::DefaultBorderColor()
313 return QColor( Qt::transparent );
316 bool HYDROData_Stream::IsValidAsAxis( const Handle(HYDROData_PolylineXY)& theHydAxis )
318 if ( theHydAxis.IsNull() )
321 TopoDS_Shape aHydraulicShape = theHydAxis->GetShape();
322 if ( aHydraulicShape.IsNull() ||
323 aHydraulicShape.ShapeType() != TopAbs_WIRE ||
324 BRep_Tool::IsClosed( aHydraulicShape ) )
325 return false; // The polyline must be a single not closed wire
330 TopoDS_Shape getShapeFromGroup( const HYDROData_SequenceOfObjects& theGroups,
331 const int theGroupId )
333 TopoDS_Shape aResShape;
334 if ( theGroups.Length() != 4 )
337 Handle(HYDROData_ShapesGroup) aGroup =
338 Handle(HYDROData_ShapesGroup)::DownCast( theGroups.Value( theGroupId ) );
339 if ( aGroup.IsNull() )
342 TopTools_SequenceOfShape aGroupShapes;
343 aGroup->GetShapes( aGroupShapes );
345 if ( !aGroupShapes.IsEmpty() )
346 aResShape = aGroupShapes.First();
351 TopoDS_Shape HYDROData_Stream::GetLeftShape() const
353 HYDROData_SequenceOfObjects aGroups = GetGroups();
354 return getShapeFromGroup( aGroups, 1 );
357 TopoDS_Shape HYDROData_Stream::GetRightShape() const
359 HYDROData_SequenceOfObjects aGroups = GetGroups();
360 return getShapeFromGroup( aGroups, 2 );
363 TopoDS_Shape HYDROData_Stream::GetInletShape() const
365 HYDROData_SequenceOfObjects aGroups = GetGroups();
366 return getShapeFromGroup( aGroups, 3 );
369 TopoDS_Shape HYDROData_Stream::GetOutletShape() const
371 HYDROData_SequenceOfObjects aGroups = GetGroups();
372 return getShapeFromGroup( aGroups, 4 );
375 QColor HYDROData_Stream::getDefaultFillingColor() const
377 return DefaultFillingColor();
380 QColor HYDROData_Stream::getDefaultBorderColor() const
382 return DefaultBorderColor();
385 bool HYDROData_Stream::SetHydraulicAxis( const Handle(HYDROData_PolylineXY)& theAxis )
387 if ( !IsValidAsAxis( theAxis ) )
390 Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
391 if ( IsEqual( aPrevAxis, theAxis ) )
394 SetReferenceObject( theAxis, DataTag_HydraulicAxis );
396 // Update the order of profiles
397 updateProfilesOrder();
399 // Indicate model of the need to update the stream presentation
405 Handle(HYDROData_PolylineXY) HYDROData_Stream::GetHydraulicAxis() const
407 return Handle(HYDROData_PolylineXY)::DownCast(
408 GetReferenceObject( DataTag_HydraulicAxis ) );
411 void HYDROData_Stream::RemoveHydraulicAxis()
413 Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
414 if ( aPrevAxis.IsNull() )
417 ClearReferenceObjects( DataTag_HydraulicAxis );
419 // We remove the reference profiles
422 // Indicate model of the need to update the stream presentation
426 bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theHydAxis,
427 const Handle(HYDROData_Profile)& theProfile,
428 const TopoDS_Face& thePlane,
429 Standard_Real& outPar)
431 if ( theProfile.IsNull() || !IsValidAsAxis( theHydAxis ) )
434 TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() ); //guide line
435 TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
436 if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
439 BRepProj_Projection aProjector (aProfileWire, thePlane, gp::OZ().Direction());
440 if(!aProjector.IsDone())
442 TopoDS_Shape aPrjProfile = aProjector.Shape();
443 if(aPrjProfile.IsNull())
445 TopoDS_Vertex aV1, aV2;
446 if(aPrjProfile.ShapeType() == TopAbs_EDGE)
447 TopExp::Vertices(TopoDS::Edge(aPrjProfile), aV1, aV2);
448 else if(aPrjProfile.ShapeType() == TopAbs_WIRE)
449 TopExp::Vertices(TopoDS::Wire(aPrjProfile), aV1, aV2);
450 else if(aPrjProfile.ShapeType() == TopAbs_COMPOUND){
451 TopExp_Explorer anExp(aPrjProfile, TopAbs_WIRE);
453 TopExp::Vertices(TopoDS::Wire(anExp.Current()), aV1, aV2);
455 anExp.Init(aPrjProfile, TopAbs_EDGE);
457 TopExp::Vertices(TopoDS::Edge(anExp.Current()), aV1, aV2);
461 if(aV1.IsNull() || aV2.IsNull())
463 gp_Pnt aPnt1 = BRep_Tool::Pnt(aV1);
464 gp_Pnt aPnt2 = BRep_Tool::Pnt(aV2);
467 BRepBuilderAPI_MakeEdge aMk(aPnt1, aPnt2);
470 const TopoDS_Edge& anEdg2 = aMk.Edge();//Section edge
471 Standard_Integer aNum(0);
473 TopExp_Explorer anExplo(aHydraulicWire, TopAbs_EDGE);
474 for(;anExplo.More();anExplo.Next()) aNum++;
475 // check for self-intersection
476 const Standard_Real SquareTolerance = Precision::Confusion()*Precision::Confusion();
477 Standard_Boolean hasInt(false);
478 Standard_Real aSqDist(DBL_MAX);
479 Standard_Integer anIndx(0);
480 BRepExtrema_ExtCC aCC;
481 aCC.Initialize(anEdg2);
483 anExplo.Init(aHydraulicWire, TopAbs_EDGE);
484 for(Standard_Integer j=1;anExplo.More();anExplo.Next(),j++) {
485 const TopoDS_Edge& anEdg1 = TopoDS::Edge(anExplo.Current());
488 Standard_Boolean hasSol(false);
492 for(Standard_Integer i=1; i<= aCC.NbExt();i++)
493 if(aCC.SquareDistance(i) < aSqDist) {
494 aSqDist = aCC.SquareDistance(i);
500 if(aSqDist <= SquareTolerance) { // hasInt
501 const gp_Pnt& aPnt = aCC.PointOnE1(anIndx);
503 TopExp::Vertices(anEdg1, aV1, aV2, Standard_True);
504 outPar += BRep_Tool::Pnt(aV1).Distance(aPnt);
506 Standard_Real aPar = aCC.ParameterOnE1(anIndx);
514 TopExp::Vertices(anEdg1, aV1, aV2);
515 outPar += BRep_Tool::Pnt(aV1).Distance(BRep_Tool::Pnt(aV2));
518 } else if(aNum > 1) {
519 TopExp::Vertices(anEdg1, aV1, aV2);
520 outPar += BRep_Tool::Pnt(aV1).Distance(BRep_Tool::Pnt(aV2));
528 bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_Profile)& theProfile, const TopoDS_Face& thePlane,
529 Standard_Real& outPar ) const
531 Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
533 return HasIntersection( aHydAxis, theProfile, thePlane, outPar );
537 bool HYDROData_Stream::AddProfile( const Handle(HYDROData_Profile)& theProfile )
539 if ( theProfile.IsNull() )
542 Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
543 if ( aHydAxis.IsNull() )
547 if(!BuildFace(aHydAxis, aPlane))
550 Standard_Real aPar(.0);
551 if ( HasReference( theProfile, DataTag_Profile ) || !HasIntersection( theProfile, aPlane, aPar ) )
552 return false; // Object is already in reference list or it has no intersection
554 int aProfileIndex = insertParameter( aPar );
555 insertProfileInToOrder( theProfile, aProfileIndex );
557 // Indicate model of the need to update the stream presentation
563 HYDROData_SequenceOfObjects HYDROData_Stream::GetProfiles() const
565 return GetReferenceObjects( DataTag_Profile );
568 bool HYDROData_Stream::RemoveProfile( const Handle(HYDROData_Profile)& theProfile )
570 if ( theProfile.IsNull() )
573 int aProfileIndex = -1;
575 HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
576 HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
577 for ( int i = 0 ; anIter.More(); anIter.Next(), ++i )
579 Handle(HYDROData_Profile) aProfile =
580 Handle(HYDROData_Profile)::DownCast( anIter.Value() );
581 if ( aProfile.IsNull() )
584 if ( IsEqual( theProfile, aProfile ) )
591 if ( aProfileIndex == -1 )
594 RemoveReferenceObject( theProfile->Label(), DataTag_Profile );
596 // Remove parameter for removed profile
597 removeParameter( aProfileIndex );
599 // Indicate model of the need to update the stream presentation
605 void HYDROData_Stream::RemoveProfiles()
607 bool anIsToUpdate = IsMustBeUpdated() || NbReferenceObjects( DataTag_Profile ) > 0;
609 ClearReferenceObjects( DataTag_Profile );
611 // Remove the parameters array
612 removeParametersArray();
614 // Indicate model of the need to update the stream presentation
615 SetToUpdate( anIsToUpdate );
618 void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)& theProfile,
619 const int theBeforeIndex )
621 Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
622 if ( theProfile.IsNull() || aHydAxis.IsNull() )
625 TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() );
626 TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
627 if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
630 if ( theBeforeIndex == -1 )
631 AddReferenceObject( theProfile, DataTag_Profile );
633 InsertReferenceObject( theProfile, DataTag_Profile, theBeforeIndex );
636 bool HYDROData_Stream::BuildFace( const Handle(HYDROData_PolylineXY)& theHydAxis,
637 TopoDS_Face& thePlane ) const
639 if ( !IsValidAsAxis( theHydAxis ) )
642 TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() );
644 gp_Ax2 aX2(gp::XOY());
648 BRepBndLib::Add(aHydraulicWire,B);
649 Standard_Real axmin,aymin,azmin,axmax,aymax,azmax;
650 B.Get(axmin,aymin,azmin,axmax,aymax,azmax);
651 BRepBuilderAPI_MakeFace aMkr(aPln, axmin-500., axmax+500., aymin-500., aymax+500.); // to be tuned later according max/ Profile deviation
652 if(!aMkr.IsDone() || aMkr.Shape().IsNull()) return false;
653 thePlane = TopoDS::Face(aMkr.Shape());
657 void HYDROData_Stream::updateProfilesOrder()
659 HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
660 if ( aRefProfiles.IsEmpty() )
663 // At first we remove all profiles from order
666 Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
667 if ( aHydAxis.IsNull() )
671 if ( !BuildFace( aHydAxis, aPlane ) )
674 Standard_Real aPar( .0 );
678 TopoDS_Compound aCmp;
679 aBB.MakeCompound(aCmp);
682 HYDROData_DataMapOfRealOfHDProfile aDM;
683 TColStd_ListOfReal aList;
684 HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
685 for (int i = 1 ; anIter.More(); anIter.Next(), i++ )
687 Handle(HYDROData_Profile) aProfile =
688 Handle(HYDROData_Profile)::DownCast( anIter.Value() );
690 TopoDS_Wire aProfileWire = TopoDS::Wire( aProfile->GetTopShape() );
691 aBB.Add( aCmp, aProfileWire );
693 if ( aProfile.IsNull() || !HasIntersection( aProfile, aPlane, aPar ) )
696 aDM.Bind( aPar, aProfile );
697 aList.Append( aPar );
700 if ( aList.IsEmpty() )
703 TColStd_Array1OfReal anArr( 1, aList.Extent() );
705 TColStd_ListIteratorOfListOfReal it( aList );
706 for ( int j = 1; it.More(); it.Next(), j++ )
707 anArr( j ) = it.Value();
710 if ( aList.Extent() > 1 )
712 TCollection_CompareOfReal Compar;
713 SortTools_QuickSortOfReal::Sort( anArr, Compar );
715 for (int j = 1; j <= anArr.Length(); j++) {
716 const Standard_Real aKey = anArr(j);
717 const Handle(HYDROData_Profile)& aProfile = aDM.Find(aKey);
718 insertProfileInToOrder( aProfile );
720 } else if ( aList.Extent() == 1 ) {
721 const Standard_Real aKey = aList.Last();
722 const Handle(HYDROData_Profile)& aProfile = aDM.Find(aKey);
723 insertProfileInToOrder( aProfile );
726 setParametersArray( anArr );
729 TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() );
730 BRepTools::Write(aHydraulicWire, "Path.brep");
731 BRepTools::Write(aCmp, "Prof.brep");
735 void HYDROData_Stream::createGroupObjects()
737 TopoDS_Shape aStreamShape = GetTopShape();
738 if ( aStreamShape.IsNull() )
741 HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
742 if ( aRefProfiles.Length() < 2 )
745 Handle(HYDROData_Profile) aFirstProfile =
746 Handle(HYDROData_Profile)::DownCast( aRefProfiles.First() );
747 Handle(HYDROData_Profile) aLastProfile =
748 Handle(HYDROData_Profile)::DownCast( aRefProfiles.Last() );
749 if ( aFirstProfile.IsNull() || aLastProfile.IsNull() )
752 TopoDS_Shape aBotEdgeShape = aFirstProfile->GetTopShape();
753 TopoDS_Shape aTopEdgeShape = aLastProfile->GetTopShape();
754 if ( aBotEdgeShape.IsNull() || aBotEdgeShape.ShapeType() != TopAbs_WIRE ||
755 aTopEdgeShape.IsNull() || aTopEdgeShape.ShapeType() != TopAbs_WIRE )
758 TopExp_Explorer aBotEdgeShapeExp( aBotEdgeShape, TopAbs_EDGE );
759 TopExp_Explorer aTopEdgeShapeExp( aTopEdgeShape, TopAbs_EDGE );
760 if ( !aBotEdgeShapeExp.More() || !aTopEdgeShapeExp.More() )
763 TopoDS_Edge aBotEdge = TopoDS::Edge( aBotEdgeShapeExp.Current() );
764 TopoDS_Edge aTopEdge = TopoDS::Edge( aTopEdgeShapeExp.Current() );
766 TopoDS_Vertex aBotFirstVert, aBotLastVert, aTopFirstVert, aTopLastVert;
767 TopExp::Vertices( aBotEdge, aBotFirstVert, aBotLastVert, true );
768 TopExp::Vertices( aTopEdge, aTopFirstVert, aTopLastVert, true );
770 TopExp_Explorer aStreamFaceExp( aStreamShape, TopAbs_FACE );
771 if ( !aStreamFaceExp.More() )
774 // Get only face because of 2d profile wires is in compound
775 TopoDS_Face aStreamFace = TopoDS::Face( aStreamFaceExp.Current() );
777 TopTools_SequenceOfShape aStreamWires;
778 HYDROData_ShapesTool::ExploreShapeToShapes( aStreamFace, TopAbs_EDGE, aStreamWires );
780 bool anIsLeft = false;
781 bool anIsRight = false;
783 TopTools_SequenceOfShape aLeftEdges, aRightEdges;
784 for ( int i = 1, n = aStreamWires.Length(); i <= n; ++i )
786 TopoDS_Edge anEdge = TopoDS::Edge( aStreamWires.Value( i ) );
787 if ( HYDROData_ShapesTool::IsEdgesEquals( anEdge, aBotEdge ) ||
788 HYDROData_ShapesTool::IsEdgesEquals( anEdge, aTopEdge ) )
795 if ( !anIsLeft && !anIsRight )
797 TopoDS_Vertex anEdgeFirstVert, anEdgeLastVert;
798 TopExp::Vertices( anEdge, anEdgeFirstVert, anEdgeLastVert, true );
800 anIsLeft = HYDROData_ShapesTool::IsVerticesEquals( anEdgeFirstVert, aBotFirstVert ) ||
801 HYDROData_ShapesTool::IsVerticesEquals( anEdgeLastVert, aTopFirstVert );
803 anIsRight = HYDROData_ShapesTool::IsVerticesEquals( anEdgeFirstVert, aTopLastVert ) ||
804 HYDROData_ShapesTool::IsVerticesEquals( anEdgeLastVert, aBotLastVert );
809 aLeftEdges.Append( anEdge );
811 else if ( anIsRight )
813 aRightEdges.Append( anEdge );
817 // Create edges groups
818 QString aLeftGroupName = GetName() + "_Left_Bank";
820 Handle(HYDROData_ShapesGroup) aLeftGroup = createGroupObject();
821 aLeftGroup->SetName( aLeftGroupName );
822 aLeftGroup->SetShapes( aLeftEdges );
824 QString aRightGroupName = GetName() + "_Right_Bank";
826 Handle(HYDROData_ShapesGroup) aRightGroup = createGroupObject();
827 aRightGroup->SetName( aRightGroupName );
828 aRightGroup->SetShapes( aRightEdges );
830 QString anInGroupName = GetName() + "_Inlet";
832 Handle(HYDROData_ShapesGroup) anInGroup = createGroupObject();
833 anInGroup->SetName( anInGroupName );
834 anInGroup->AddShape( aBotEdge );
836 QString anOutGroupName = GetName() + "_Outlet";
838 Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
839 anOutGroup->SetName( anOutGroupName );
840 anOutGroup->AddShape( aTopEdge );
843 ObjectKind HYDROData_Stream::getAltitudeObjectType() const
845 return KIND_STREAM_ALTITUDE;
848 void HYDROData_Stream::setParametersArray( const TColStd_Array1OfReal& theArray )
850 if ( theArray.Length() == 0 )
852 removeParametersArray();
856 TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray );
858 Handle(TDataStd_RealArray) aParamsArray =
859 TDataStd_RealArray::Set( aLabel, theArray.Lower(), theArray.Upper() );
861 for ( int i = theArray.Lower(), n = theArray.Upper(); i <= n; ++i )
863 const Standard_Real& aParam = theArray( i );
864 aParamsArray->SetValue( i, aParam );
868 TColStd_Array1OfReal* HYDROData_Stream::getParametersArray() const
870 TColStd_Array1OfReal* anArray = NULL;
872 TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray, false );
873 if ( !aLabel.IsNull() )
875 Handle(TDataStd_RealArray) aParamsArray;
876 if ( aLabel.FindAttribute( TDataStd_RealArray::GetID(), aParamsArray ) )
878 anArray = new TColStd_Array1OfReal( aParamsArray->Lower(), aParamsArray->Upper() );
879 for ( int i = aParamsArray->Lower(), n = aParamsArray->Upper(); i <= n; ++i )
881 const Standard_Real& aParam = aParamsArray->Value( i );
882 anArray->SetValue( i, aParam );
890 void HYDROData_Stream::removeParametersArray()
892 TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray, false );
893 if ( !aLabel.IsNull() )
894 aLabel.ForgetAllAttributes();
897 int HYDROData_Stream::insertParameter( const Standard_Real& theParam )
901 TColStd_Array1OfReal* anArr = getParametersArray();
906 TColStd_Array1OfReal aNewArr( anArr->Lower(), anArr->Upper() + 1 );
908 bool isInserted = false;
909 for ( int i = anArr->Lower(), j = i, n = anArr->Upper(); i <= n; ++i, ++j )
911 const Standard_Real& aStoredParam = anArr->Value( i );
914 if ( theParam > aStoredParam )
920 aNewArr( j ) = theParam;
926 aNewArr( j ) = aStoredParam;
932 aNewArr( aNewArr.Upper() ) = theParam;
935 setParametersArray( aNewArr );
940 TColStd_Array1OfReal aNewArr( 1, 1 );
941 aNewArr.SetValue( 1, theParam );
942 setParametersArray( aNewArr );
948 void HYDROData_Stream::removeParameter( const int& theIndex )
950 TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray, false );
951 if ( aLabel.IsNull() )
954 Handle(TDataStd_RealArray) aParamsArray;
955 if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), aParamsArray ) )
958 if ( aParamsArray->Length() == 1 )
960 removeParametersArray();
964 TColStd_Array1OfReal aNewArr( aParamsArray->Lower(), aParamsArray->Upper() - 1 );
966 for ( int i = aParamsArray->Lower(), j = i, k = 0, n = aParamsArray->Upper(); i <= n; ++i, ++k )
968 const Standard_Real& aStoredParam = aParamsArray->Value( i );
972 aNewArr.SetValue( j, aStoredParam );
976 setParametersArray( aNewArr );