X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Stream.cxx;h=07bf9971d8766716400f9259232a658b484eab71;hb=16cecc72628fbd69c3486834e24c3cde58b85fd1;hp=63e983ec1d35109f89a4b77de997c7387b09dbf7;hpb=fcece92742680e708f87352c32597ec2d84bfe2f;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_Stream.cxx b/src/HYDROData/HYDROData_Stream.cxx index 63e983ec..07bf9971 100644 --- a/src/HYDROData/HYDROData_Stream.cxx +++ b/src/HYDROData/HYDROData_Stream.cxx @@ -4,6 +4,9 @@ #include "HYDROData_Document.h" #include "HYDROData_PolylineXY.h" #include "HYDROData_Profile.h" +#include "HYDROData_ShapesGroup.h" +#include "HYDROData_ShapesTool.h" +#include "HYDROData_IAltitudeObject.h" #include @@ -65,8 +68,6 @@ #include #endif -#define PYTHON_STREAM_ID "KIND_STREAM" - typedef NCollection_DataMap HYDROData_DataMapOfRealOfHDProfile; IMPLEMENT_STANDARD_HANDLE(HYDROData_Stream,HYDROData_NaturalObject) @@ -84,20 +85,7 @@ HYDROData_Stream::~HYDROData_Stream() QStringList HYDROData_Stream::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const { - QStringList aResList; - - Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab ); - if ( aDocument.IsNull() ) - return aResList; - - QString aDocName = aDocument->GetDocPyName(); - QString aStreamName = GetName(); - - aResList << QString( "%1 = %2.CreateObject( %3 );" ) - .arg( aStreamName ).arg( aDocName ).arg( PYTHON_STREAM_ID ); - aResList << QString( "%1.SetName( \"%2\" );" ) - .arg( aStreamName ).arg( aStreamName ); - aResList << QString( "" ); + QStringList aResList = dumpObjectCreation( theTreatedObjects ); // TODO @@ -135,11 +123,17 @@ Handle(Geom_BSplineCurve) HYDROData_Stream::buildInterpolationCurve( GeomAPI_Interpolate anInterpolator (theArrayOfPnt, Standard_False, 1.0e-5); anInterpolator.Perform() ; if (anInterpolator.IsDone()) - aBSpline = anInterpolator.Curve(); + aBSpline = anInterpolator.Curve(); return aBSpline; } void HYDROData_Stream::Update() +{ + updateProfilesOrder(); + UpdatePrs(); +} + +void HYDROData_Stream::UpdatePrs() { HYDROData_NaturalObject::Update(); @@ -162,52 +156,51 @@ void HYDROData_Stream::Update() Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( anIter.Value() ); if ( aProfile.IsNull() ) - continue; - const TopoDS_Shape& aProf3d = aProfile->GetShape3D(); + 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()); + 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); + 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; + return; TopoDS_Edge anEdgLeft, anEdgRight; BRepBuilderAPI_MakeEdge aMakeEdge(aBSpline); if(aMakeEdge.IsDone()) - anEdgLeft = aMakeEdge.Edge(); + anEdgLeft = aMakeEdge.Edge(); if(anEdgLeft.IsNull()) - return; + return; aBSpline.Nullify(); aBSpline = buildInterpolationCurve (anArrayOfLPnt); if(aBSpline.IsNull()) return; aMakeEdge.Init(aBSpline); if(aMakeEdge.IsDone()) - anEdgRight = aMakeEdge.Edge(); + anEdgRight = aMakeEdge.Edge(); if(anEdgRight.IsNull()) - return; + return; BRep_Builder aBB; TopoDS_Compound aCmp; aBB.MakeCompound(aCmp); @@ -222,18 +215,18 @@ void HYDROData_Stream::Update() #ifdef DEB_UPDATE else { BRepTools::Write(aCmp, "str3d.brep"); - SetShape3D(aCmp); + 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); + 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); + aPnt.SetZ(.0); + anArrayOfLPnt->SetValue(i, aPnt); } aBSpline.Nullify(); aBSpline = buildInterpolationCurve (anArrayOfFPnt); @@ -241,48 +234,96 @@ void HYDROData_Stream::Update() return; aMakeEdge.Init(aBSpline); if(aMakeEdge.IsDone()) - anEdgLeft = aMakeEdge.Edge(); + anEdgLeft = aMakeEdge.Edge(); aBSpline.Nullify(); aBSpline = buildInterpolationCurve (anArrayOfLPnt); if(aBSpline.IsNull()) return; aMakeEdge.Init(aBSpline); if(aMakeEdge.IsDone()) - anEdgRight = aMakeEdge.Edge(); + anEdgRight = aMakeEdge.Edge(); if(anEdgRight.IsNull()) - return; + return; BRepBuilderAPI_MakeEdge aMakeEdge2(anArrayOfFPnt->Value(1),anArrayOfLPnt->Value(1)); TopoDS_Edge aBotEdge, aTopEdge; if(aMakeEdge2.IsDone()) - aBotEdge = aMakeEdge2.Edge(); + aBotEdge = aMakeEdge2.Edge(); BRepBuilderAPI_MakeEdge aMakeEdge3(anArrayOfFPnt->Value(anArrayOfFPnt->Length()),anArrayOfLPnt->Value(anArrayOfLPnt->Length())); if(aMakeEdge3.IsDone()) - aTopEdge = aMakeEdge3.Edge(); + 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(); - BRepBuilderAPI_MakeWire aMakeWire( aBotEdge, anEdgLeft, aTopEdge,anEdgRight); TopoDS_Wire aSectProfileWire; if(aMakeWire.IsDone()) - aSectProfileWire = aMakeWire.Wire(); + aSectProfileWire = aMakeWire.Wire(); + BRepBuilderAPI_MakeFace aMakeFace( aSectProfileWire, Standard_True ); TopoDS_Face aFace; aMakeFace.Build(); if( aMakeFace.IsDone() ) - aFace = aMakeFace.Face(); + 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()) + 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); + 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() @@ -295,6 +336,65 @@ 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(); @@ -307,20 +407,12 @@ QColor HYDROData_Stream::getDefaultBorderColor() const bool HYDROData_Stream::SetHydraulicAxis( const Handle(HYDROData_PolylineXY)& theAxis ) { - Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis(); - - if ( theAxis.IsNull() ) - { - RemoveHydraulicAxis(); - return !aPrevAxis.IsNull(); - } - - if ( IsEqual( aPrevAxis, theAxis ) ) + if ( !IsValidAsAxis( theAxis ) ) return false; - TopoDS_Wire aHydraulicWire = TopoDS::Wire( theAxis->GetShape() ); - if ( aHydraulicWire.IsNull() ) - return false; // The polyline must be a single wire + Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis(); + if ( IsEqual( aPrevAxis, theAxis ) ) + return true; SetReferenceObject( theAxis, DataTag_HydraulicAxis ); @@ -357,9 +449,9 @@ void HYDROData_Stream::RemoveHydraulicAxis() bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theHydAxis, const Handle(HYDROData_Profile)& theProfile, const TopoDS_Face& thePlane, - Standard_Real& outPar) + Standard_Real& outPar) { - if ( theProfile.IsNull() || theHydAxis.IsNull() ) + if ( theProfile.IsNull() || !IsValidAsAxis( theHydAxis ) ) return false; TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() ); //guide line @@ -375,22 +467,22 @@ bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theH return false; TopoDS_Vertex aV1, aV2; if(aPrjProfile.ShapeType() == TopAbs_EDGE) - TopExp::Vertices(TopoDS::Edge(aPrjProfile), aV1, aV2); + TopExp::Vertices(TopoDS::Edge(aPrjProfile), aV1, aV2); else if(aPrjProfile.ShapeType() == TopAbs_WIRE) - TopExp::Vertices(TopoDS::Wire(aPrjProfile), aV1, aV2); + 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(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; + return false; gp_Pnt aPnt1 = BRep_Tool::Pnt(aV1); gp_Pnt aPnt2 = BRep_Tool::Pnt(aV2); aPnt1.SetZ(0.0); @@ -413,43 +505,43 @@ bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theH 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); + 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 + // 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 + 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 { + outPar += BRep_Tool::Pnt(aV1).Distance(aPnt); + } else { Standard_Real aPar = aCC.ParameterOnE1(anIndx); - outPar = aPar; - } - hasInt = true; - break; - } else { - // no ints-n + 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)); - } + 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; @@ -457,7 +549,7 @@ bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theH } bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_Profile)& theProfile, const TopoDS_Face& thePlane, - Standard_Real& outPar ) const + Standard_Real& outPar ) const { Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis(); @@ -564,11 +656,14 @@ void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)& InsertReferenceObject( theProfile, DataTag_Profile, theBeforeIndex ); } -bool HYDROData_Stream::BuildFace( const Handle(HYDROData_PolylineXY)& theHydAxis, TopoDS_Face& thePlane) const +bool HYDROData_Stream::BuildFace( const Handle(HYDROData_PolylineXY)& theHydAxis, + TopoDS_Face& thePlane ) const { - if ( theHydAxis.IsNull() ) return false; + if ( !IsValidAsAxis( theHydAxis ) ) + return false; + TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() ); - if(aHydraulicWire.IsNull()) return false; + gp_Ax2 aX2(gp::XOY()); gp_Ax3 aX3(aX2); gp_Pln aPln(aX3); @@ -620,9 +715,9 @@ void HYDROData_Stream::updateProfilesOrder() #endif if ( aProfile.IsNull() || !HasIntersection( aProfile, aPlane, aPar ) ) continue; - + aDM.Bind( aPar, aProfile ); - aList.Append( aPar ); + aList.Append( aPar ); } if ( aList.IsEmpty() ) @@ -660,6 +755,11 @@ void HYDROData_Stream::updateProfilesOrder() #endif } +ObjectKind HYDROData_Stream::getAltitudeObjectType() const +{ + return KIND_STREAM_ALTITUDE; +} + void HYDROData_Stream::setParametersArray( const TColStd_Array1OfReal& theArray ) { if ( theArray.Length() == 0 )