+Handle(Geom_BSplineCurve) HYDROData_Stream::buildInterpolationCurve(
+ const Handle(TColgp_HArray1OfPnt)& theArrayOfPnt )
+{
+ Handle(Geom_BSplineCurve) aBSpline;
+ GeomAPI_Interpolate anInterpolator (theArrayOfPnt, Standard_False, 1.0e-5);
+ anInterpolator.Perform() ;
+ if (anInterpolator.IsDone())
+ aBSpline = anInterpolator.Curve();
+ return aBSpline;
+}
+
+void HYDROData_Stream::Update()
+{
+ updateProfilesOrder();
+ UpdatePrs();
+}
+
+void HYDROData_Stream::UpdatePrs()
+{
+ HYDROData_NaturalObject::Update();
+
+ Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+ HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
+ if ( aHydAxis.IsNull() || aRefProfiles.Length() < 2 )
+ return;
+
+ bool anIsFirst = true;
+ gp_Pnt aPrevFirstPoint, aPrevLastPoint;
+ Handle(TColgp_HArray1OfPnt) anArrayOfFPnt = new TColgp_HArray1OfPnt(1, aRefProfiles.Length());
+ Handle(TColgp_HArray1OfPnt) anArrayOfLPnt = new TColgp_HArray1OfPnt(1, aRefProfiles.Length());
+ TopTools_Array1OfShape anArrOfProfiles(1, aRefProfiles.Length());
+ TopTools_Array1OfShape anArrOf2DProfiles(1, aRefProfiles.Length());
+
+ // Pre-processing
+ HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
+ for (int i=1 ; anIter.More(); anIter.Next(),i++ )
+ {
+ Handle(HYDROData_Profile) aProfile =
+ Handle(HYDROData_Profile)::DownCast( anIter.Value() );
+ if ( aProfile.IsNull() )
+ continue;
+ const TopoDS_Shape& aProf3d = aProfile->GetShape3D();
+ gp_XY aPnt1, aPnt2;
+ if ( !aProfile->GetLeftPoint( aPnt1 ) || !aProfile->GetRightPoint( aPnt2 ) )
+ continue;
+ anArrOfProfiles.SetValue(i,aProfile->GetShape3D());//aProfile->GetTopShape();
+ anArrOf2DProfiles.SetValue(i,aProfile->GetTopShape());
+
+ gp_Pnt aCurFirstPoint( aPnt1.X(), aPnt1.Y(), 0 ), aCurFP;
+ gp_Pnt aCurLastPoint( aPnt2.X(), aPnt2.Y(), 0 ), aCurLP;
+ TopoDS_Vertex aV1, aV2;
+ TopExp::Vertices(TopoDS::Wire(aProf3d), aV1, aV2);
+ gp_Pnt aP1 = BRep_Tool::Pnt(aV1);
+ if(aP1.X() == aPnt1.X() && aP1.Y() == aPnt1.Y())
+ aCurFP = aP1;
+ else
+ aCurLP = aP1;
+ aP1 = BRep_Tool::Pnt(aV2);
+ if(aP1.X() == aPnt2.X() && aP1.Y() == aPnt2.Y())
+ aCurLP = aP1;
+ else
+ aCurFP = aP1;
+ anArrayOfFPnt->SetValue(i,aCurFP);
+ anArrayOfLPnt->SetValue(i,aCurLP);
+ }
+
+ // Construct of the 3D presentation
+ Handle(Geom_BSplineCurve) aBSpline = buildInterpolationCurve (anArrayOfFPnt);
+ if(aBSpline.IsNull())
+ return;
+ TopoDS_Edge anEdgLeft, anEdgRight;
+ BRepBuilderAPI_MakeEdge aMakeEdge(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgLeft = aMakeEdge.Edge();
+ if(anEdgLeft.IsNull())
+ return;
+ aBSpline.Nullify();
+ aBSpline = buildInterpolationCurve (anArrayOfLPnt);
+ if(aBSpline.IsNull())
+ return;
+ aMakeEdge.Init(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgRight = aMakeEdge.Edge();
+ if(anEdgRight.IsNull())
+ return;
+ BRep_Builder aBB;
+ TopoDS_Compound aCmp;
+ aBB.MakeCompound(aCmp);
+ anIter.Init( aRefProfiles );
+ for (int i=1 ; i < anArrOfProfiles.Length() +1; i++ )
+ aBB.Add(aCmp, anArrOfProfiles.Value(i));
+ aBB.Add(aCmp,anEdgLeft);
+ aBB.Add(aCmp,anEdgRight);
+ BRepCheck_Analyzer aCh(aCmp);
+ if(aCh.IsValid())
+ SetShape3D(aCmp);
+#ifdef DEB_UPDATE
+ else {
+ BRepTools::Write(aCmp, "str3d.brep");
+ SetShape3D(aCmp);
+ }
+#endif
+
+ // Construct the top presentation
+ for(int i=1;i<= anArrayOfLPnt->Length();i++) {
+ gp_Pnt aPnt = anArrayOfFPnt->Value(i);
+ aPnt.SetZ(.0); // make 2d
+ anArrayOfFPnt->SetValue(i, aPnt);
+ aPnt = anArrayOfLPnt->Value(i);
+ aPnt.SetZ(.0);
+ anArrayOfLPnt->SetValue(i, aPnt);
+ }
+ aBSpline.Nullify();
+ aBSpline = buildInterpolationCurve (anArrayOfFPnt);
+ if(aBSpline.IsNull())
+ return;
+ aMakeEdge.Init(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgLeft = aMakeEdge.Edge();
+ aBSpline.Nullify();
+ aBSpline = buildInterpolationCurve (anArrayOfLPnt);
+ if(aBSpline.IsNull())
+ return;
+ aMakeEdge.Init(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgRight = aMakeEdge.Edge();
+ if(anEdgRight.IsNull())
+ return;
+ BRepBuilderAPI_MakeEdge aMakeEdge2(anArrayOfFPnt->Value(1),anArrayOfLPnt->Value(1));
+ TopoDS_Edge aBotEdge, aTopEdge;
+ if(aMakeEdge2.IsDone())
+ aBotEdge = aMakeEdge2.Edge();
+ BRepBuilderAPI_MakeEdge aMakeEdge3(anArrayOfFPnt->Value(anArrayOfFPnt->Length()),anArrayOfLPnt->Value(anArrayOfLPnt->Length()));
+ if(aMakeEdge3.IsDone())
+ aTopEdge = aMakeEdge3.Edge();
+
+ // Make wire for 2D presentation with updating of corresponding edges
+ BRepBuilderAPI_MakeWire aMakeWire;
+
+ aMakeWire.Add( aBotEdge );
+ aBotEdge = aMakeWire.Edge();
+
+ aMakeWire.Add( anEdgLeft );
+ anEdgLeft = aMakeWire.Edge();
+
+ aMakeWire.Add( aTopEdge );
+ aTopEdge = aMakeWire.Edge();
+
+ aMakeWire.Add( anEdgRight );
+ anEdgRight = aMakeWire.Edge();
+
+ TopoDS_Wire aSectProfileWire;
+ if(aMakeWire.IsDone())
+ aSectProfileWire = aMakeWire.Wire();
+
+ BRepBuilderAPI_MakeFace aMakeFace( aSectProfileWire, Standard_True );
+ TopoDS_Face aFace;
+ aMakeFace.Build();
+ if( aMakeFace.IsDone() )
+ aFace = aMakeFace.Face();
+
+ aCmp.Nullify();
+ aBB.MakeCompound(aCmp);
+ aBB.Add(aCmp,aFace);
+ for(int i=1;i <= anArrOf2DProfiles.Length(); i++)
+ aBB.Add(aCmp,anArrOf2DProfiles.Value(i));
+ aCh.Init(aCmp);
+ if(aCh.IsValid())
+ SetTopShape(aCmp);
+#ifdef DEB_UPDATE
+ else {
+ BRepTools::Write(aCmp, "str2d.brep");
+ SetTopShape(aCmp);
+ }
+#endif
+
+ // Create the stream groups
+ QString aLeftGroupName = GetName() + "_Left_Bank";
+
+ Handle(HYDROData_ShapesGroup) aLeftGroup = createGroupObject();
+ aLeftGroup->SetName( aLeftGroupName );
+ aLeftGroup->AddShape( anEdgLeft );
+
+ QString aRightGroupName = GetName() + "_Right_Bank";
+
+ Handle(HYDROData_ShapesGroup) aRightGroup = createGroupObject();
+ aRightGroup->SetName( aRightGroupName );
+ aRightGroup->AddShape( anEdgRight );
+
+ QString anInGroupName = GetName() + "_Inlet";
+
+ Handle(HYDROData_ShapesGroup) anInGroup = createGroupObject();
+ anInGroup->SetName( anInGroupName );
+ anInGroup->AddShape( aBotEdge );
+
+ QString anOutGroupName = GetName() + "_Outlet";
+
+ Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
+ anOutGroup->SetName( anOutGroupName );
+ anOutGroup->AddShape( aTopEdge );
+
+ // This peace of code is for testing of functionality of altitude,
+ // will be removed by adv when testing will be finished
+ Handle(HYDROData_IAltitudeObject) anAltObj = GetAltitudeObject();
+ gp_XY aTestPnt( 5, 0 );
+ anAltObj->GetAltitudeForPoint( aTestPnt );
+ // End of test code
+
+}
+
+QColor HYDROData_Stream::DefaultFillingColor()
+{
+ return QColor( Qt::green );
+}
+
+QColor HYDROData_Stream::DefaultBorderColor()
+{
+ return QColor( Qt::transparent );
+}
+
+bool HYDROData_Stream::IsValidAsAxis( const Handle(HYDROData_PolylineXY)& theHydAxis )
+{
+ if ( theHydAxis.IsNull() )
+ return false;
+
+ TopoDS_Shape aHydraulicShape = theHydAxis->GetShape();
+ if ( aHydraulicShape.IsNull() ||
+ aHydraulicShape.ShapeType() != TopAbs_WIRE ||
+ BRep_Tool::IsClosed( aHydraulicShape ) )
+ return false; // The polyline must be a single not closed wire
+
+ return true;
+}
+
+TopoDS_Shape getShapeFromGroup( const HYDROData_SequenceOfObjects& theGroups,
+ const int theGroupId )
+{
+ TopoDS_Shape aResShape;
+ if ( theGroups.Length() != 4 )
+ return aResShape;
+
+ Handle(HYDROData_ShapesGroup) aGroup =
+ Handle(HYDROData_ShapesGroup)::DownCast( theGroups.Value( theGroupId ) );
+ if ( aGroup.IsNull() )
+ return aResShape;
+
+ TopTools_SequenceOfShape aGroupShapes;
+ aGroup->GetShapes( aGroupShapes );
+
+ if ( !aGroupShapes.IsEmpty() )
+ aResShape = aGroupShapes.First();
+
+ return aResShape;
+}
+
+TopoDS_Shape HYDROData_Stream::GetLeftShape() const
+{
+ HYDROData_SequenceOfObjects aGroups = GetGroups();
+ return getShapeFromGroup( aGroups, 1 );
+}
+
+TopoDS_Shape HYDROData_Stream::GetRightShape() const
+{
+ HYDROData_SequenceOfObjects aGroups = GetGroups();
+ return getShapeFromGroup( aGroups, 2 );
+}
+
+TopoDS_Shape HYDROData_Stream::GetInletShape() const
+{
+ HYDROData_SequenceOfObjects aGroups = GetGroups();
+ return getShapeFromGroup( aGroups, 3 );
+}
+
+TopoDS_Shape HYDROData_Stream::GetOutletShape() const
+{
+ HYDROData_SequenceOfObjects aGroups = GetGroups();
+ return getShapeFromGroup( aGroups, 4 );
+}
+
+QColor HYDROData_Stream::getDefaultFillingColor() const
+{
+ return DefaultFillingColor();
+}
+
+QColor HYDROData_Stream::getDefaultBorderColor() const
+{
+ return DefaultBorderColor();
+}
+
+bool HYDROData_Stream::SetHydraulicAxis( const Handle(HYDROData_PolylineXY)& theAxis )
+{
+ if ( !IsValidAsAxis( theAxis ) )
+ return false;
+
+ Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
+ if ( IsEqual( aPrevAxis, theAxis ) )
+ return true;
+
+ SetReferenceObject( theAxis, DataTag_HydraulicAxis );
+
+ // Update the order of profiles
+ updateProfilesOrder();
+
+ // Indicate model of the need to update the stream presentation
+ SetToUpdate( true );
+
+ return true;
+}
+
+Handle(HYDROData_PolylineXY) HYDROData_Stream::GetHydraulicAxis() const
+{
+ return Handle(HYDROData_PolylineXY)::DownCast(
+ GetReferenceObject( DataTag_HydraulicAxis ) );
+}
+
+void HYDROData_Stream::RemoveHydraulicAxis()
+{
+ Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
+ if ( aPrevAxis.IsNull() )
+ return;
+
+ ClearReferenceObjects( DataTag_HydraulicAxis );
+
+ // We remove the reference profiles
+ RemoveProfiles();
+
+ // Indicate model of the need to update the stream presentation
+ SetToUpdate( true );
+}
+
+bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theHydAxis,
+ const Handle(HYDROData_Profile)& theProfile,
+ const TopoDS_Face& thePlane,
+ Standard_Real& outPar)
+{
+ if ( theProfile.IsNull() || !IsValidAsAxis( theHydAxis ) )
+ return false;
+
+ TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() ); //guide line
+ TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
+ if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
+ return false;
+
+ BRepProj_Projection aProjector (aProfileWire, thePlane, gp::OZ().Direction());
+ if(!aProjector.IsDone())
+ return false;
+ TopoDS_Shape aPrjProfile = aProjector.Shape();
+ if(aPrjProfile.IsNull())
+ return false;
+ TopoDS_Vertex aV1, aV2;
+ if(aPrjProfile.ShapeType() == TopAbs_EDGE)
+ TopExp::Vertices(TopoDS::Edge(aPrjProfile), aV1, aV2);
+ else if(aPrjProfile.ShapeType() == TopAbs_WIRE)
+ TopExp::Vertices(TopoDS::Wire(aPrjProfile), aV1, aV2);
+ else if(aPrjProfile.ShapeType() == TopAbs_COMPOUND){
+ TopExp_Explorer anExp(aPrjProfile, TopAbs_WIRE);
+ if(anExp.More()) {
+ TopExp::Vertices(TopoDS::Wire(anExp.Current()), aV1, aV2);
+ } else {
+ anExp.Init(aPrjProfile, TopAbs_EDGE);
+ if(anExp.More()) {
+ TopExp::Vertices(TopoDS::Edge(anExp.Current()), aV1, aV2);
+ }
+ }
+ }
+ if(aV1.IsNull() || aV2.IsNull())
+ return false;
+ gp_Pnt aPnt1 = BRep_Tool::Pnt(aV1);
+ gp_Pnt aPnt2 = BRep_Tool::Pnt(aV2);
+ aPnt1.SetZ(0.0);
+ aPnt2.SetZ(0.0);
+ BRepBuilderAPI_MakeEdge aMk(aPnt1, aPnt2);
+ if(!aMk.IsDone())
+ return false;
+ const TopoDS_Edge& anEdg2 = aMk.Edge();//Section edge
+ Standard_Integer aNum(0);
+
+ TopExp_Explorer anExplo(aHydraulicWire, TopAbs_EDGE);
+ for(;anExplo.More();anExplo.Next()) aNum++;
+ // check for self-intersection
+ const Standard_Real SquareTolerance = Precision::Confusion()*Precision::Confusion();
+ Standard_Boolean hasInt(false);
+ Standard_Real aSqDist(DBL_MAX);
+ Standard_Integer anIndx(0);
+ BRepExtrema_ExtCC aCC;
+ aCC.Initialize(anEdg2);
+ outPar = 0.0;
+ anExplo.Init(aHydraulicWire, TopAbs_EDGE);
+ for(Standard_Integer j=1;anExplo.More();anExplo.Next(),j++) {
+ const TopoDS_Edge& anEdg1 = TopoDS::Edge(anExplo.Current());
+ if(anEdg1.IsNull())
+ continue;
+ Standard_Boolean hasSol(false);
+ aCC.Perform(anEdg1);
+ if(aCC.IsDone()) {
+ // find minimal dist
+ for(Standard_Integer i=1; i<= aCC.NbExt();i++)
+ if(aCC.SquareDistance(i) < aSqDist) {
+ aSqDist = aCC.SquareDistance(i);
+ anIndx = i;
+ hasSol = true;
+ }
+ }
+ if(hasSol) {
+ if(aSqDist <= SquareTolerance) { // hasInt
+ const gp_Pnt& aPnt = aCC.PointOnE1(anIndx);
+ if(aNum > 1) {
+ TopExp::Vertices(anEdg1, aV1, aV2, Standard_True);
+ outPar += BRep_Tool::Pnt(aV1).Distance(aPnt);
+ } else {
+ Standard_Real aPar = aCC.ParameterOnE1(anIndx);
+ outPar = aPar;
+ }
+ hasInt = true;
+ break;
+ } else {
+ // no ints-n
+ if(aNum > 1) {
+ TopExp::Vertices(anEdg1, aV1, aV2);
+ outPar += BRep_Tool::Pnt(aV1).Distance(BRep_Tool::Pnt(aV2));
+ }
+ }
+ } else if(aNum > 1) {
+ TopExp::Vertices(anEdg1, aV1, aV2);
+ outPar += BRep_Tool::Pnt(aV1).Distance(BRep_Tool::Pnt(aV2));
+ }
+ }
+ if(hasInt)
+ return true;
+ return false;
+}
+
+bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_Profile)& theProfile, const TopoDS_Face& thePlane,
+ Standard_Real& outPar ) const
+{
+ Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+
+ return HasIntersection( aHydAxis, theProfile, thePlane, outPar );
+}
+
+
+bool HYDROData_Stream::AddProfile( const Handle(HYDROData_Profile)& theProfile )
+{
+ if ( theProfile.IsNull() )
+ return false;
+
+ Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+ if ( aHydAxis.IsNull() )
+ return false;
+
+ TopoDS_Face aPlane;
+ if(!BuildFace(aHydAxis, aPlane))
+ return false;
+
+ Standard_Real aPar(.0);
+ if ( HasReference( theProfile, DataTag_Profile ) || !HasIntersection( theProfile, aPlane, aPar ) )
+ return false; // Object is already in reference list or it has no intersection
+
+ int aProfileIndex = insertParameter( aPar );
+ insertProfileInToOrder( theProfile, aProfileIndex );
+
+ // Indicate model of the need to update the stream presentation
+ SetToUpdate( true );
+
+ return true;
+}
+
+HYDROData_SequenceOfObjects HYDROData_Stream::GetProfiles() const
+{
+ return GetReferenceObjects( DataTag_Profile );
+}
+
+bool HYDROData_Stream::RemoveProfile( const Handle(HYDROData_Profile)& theProfile )
+{
+ if ( theProfile.IsNull() )
+ return false;
+
+ int aProfileIndex = -1;
+
+ HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
+ HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
+ for ( int i = 0 ; anIter.More(); anIter.Next(), ++i )
+ {
+ Handle(HYDROData_Profile) aProfile =
+ Handle(HYDROData_Profile)::DownCast( anIter.Value() );
+ if ( aProfile.IsNull() )
+ continue;
+
+ if ( IsEqual( theProfile, aProfile ) )
+ {
+ aProfileIndex = i;
+ break;
+ }
+ }
+
+ if ( aProfileIndex == -1 )
+ return false;
+
+ RemoveReferenceObject( theProfile->Label(), DataTag_Profile );
+
+ // Remove parameter for removed profile
+ removeParameter( aProfileIndex );
+
+ // Indicate model of the need to update the stream presentation
+ SetToUpdate( true );
+
+ return true;
+}
+
+void HYDROData_Stream::RemoveProfiles()
+{
+ bool anIsToUpdate = IsMustBeUpdated() || NbReferenceObjects( DataTag_Profile ) > 0;
+
+ ClearReferenceObjects( DataTag_Profile );
+
+ // Remove the parameters array
+ removeParametersArray();
+
+ // Indicate model of the need to update the stream presentation
+ SetToUpdate( anIsToUpdate );
+}
+
+void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)& theProfile,
+ const int theBeforeIndex )
+{
+ Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+ if ( theProfile.IsNull() || aHydAxis.IsNull() )
+ return;
+
+ TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() );
+ TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
+ if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
+ return;
+
+ if ( theBeforeIndex == -1 )
+ AddReferenceObject( theProfile, DataTag_Profile );
+ else
+ InsertReferenceObject( theProfile, DataTag_Profile, theBeforeIndex );
+}
+
+bool HYDROData_Stream::BuildFace( const Handle(HYDROData_PolylineXY)& theHydAxis,
+ TopoDS_Face& thePlane ) const
+{
+ if ( !IsValidAsAxis( theHydAxis ) )
+ return false;