#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 <BRep_Builder.hxx>
-#include <BRepBuilderAPI_MakeEdge.hxx>
-#include <BRepBuilderAPI_MakeWire.hxx>
-#include <BRepBuilderAPI_MakeFace.hxx>
+#include <TDataStd_RealArray.hxx>
+
+#include <Precision.hxx>
+
+#include <NCollection_DataMap.hxx>
+
+#include <TColStd_Array1OfReal.hxx>
+#include <TColStd_ListOfReal.hxx>
+#include <TColStd_ListIteratorOfListOfReal.hxx>
+#include <TCollection_CompareOfReal.hxx>
+#include <TColgp_Array1OfPnt.hxx>
+#include <TColgp_HArray1OfPnt.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Wire.hxx>
#include <TopoDS_Vertex.hxx>
#include <TopExp.hxx>
#include <TopExp_Explorer.hxx>
+
+#include <Bnd_Box.hxx>
+
+#include <BRep_Builder.hxx>
+#include <BRepBuilderAPI_MakeEdge.hxx>
+#include <BRepBuilderAPI_MakeWire.hxx>
+#include <BRepBuilderAPI_MakeFace.hxx>
+
+#include <BRepBndLib.hxx>
#include <BRepProj_Projection.hxx>
#include <BRepExtrema_ExtCC.hxx>
+#include <BRepCheck_Analyzer.hxx>
+
+#include <gp.hxx>
#include <gp_Ax1.hxx>
#include <gp_Ax2.hxx>
#include <gp_Ax3.hxx>
#include <gp_Vec.hxx>
#include <gp_Pnt.hxx>
#include <gp_Pln.hxx>
-#include <gp.hxx>
-#include <Bnd_Box.hxx>
-#include <BRepBndLib.hxx>
-#include <TColStd_Array1OfReal.hxx>
-#include <Precision.hxx>
-#include <QStringList>
-#include <NCollection_DataMap.hxx>
-typedef NCollection_DataMap<Standard_Real, Handle(HYDROData_Profile)> HYDROData_DataMapOfRealOfHDProfile;
-//typedef HYDROData_DataMapOfRealOfHDProfile::Iterator HYDROData_DataMapIteratorOfDataMapOfRealOfHDProfile;
-#include <TColStd_ListOfReal.hxx>
-#include <TColStd_ListIteratorOfListOfReal.hxx>
-#include <TCollection_CompareOfReal.hxx>
+#include <GeomAPI_Interpolate.hxx>
+#include <Geom_BSplineCurve.hxx>
+
+#include <TopTools_Array1OfShape.hxx>
+
#include <SortTools_QuickSortOfReal.hxx>
+
+#include <QColor>
+#include <QStringList>
+
+//#define DEB_STREAM 1
+#ifdef DEB_STREAM
//#define DEB_HASINT 1
-#ifdef DEB_HASINT
+//#define DEB_UPDATE 1
#include <BRepTools.hxx>
#include <TCollection_AsciiString.hxx>
-#include <BRep_Builder.hxx>
#endif
-#define PYTHON_STREAM_ID "KIND_STREAM"
+typedef NCollection_DataMap<Standard_Real, Handle(HYDROData_Profile)> HYDROData_DataMapOfRealOfHDProfile;
IMPLEMENT_STANDARD_HANDLE(HYDROData_Stream,HYDROData_NaturalObject)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Stream,HYDROData_NaturalObject)
QStringList HYDROData_Stream::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
{
- QStringList aResList;
+ QStringList aResList = dumpObjectCreation( theTreatedObjects );
+ QString aName = GetObjPyName();
- Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
- if ( aDocument.IsNull() )
- return aResList;
+ Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+ setPythonReferenceObject( theTreatedObjects, aResList, aHydAxis, "SetHydraulicAxis" );
- QString aDocName = aDocument->GetDocPyName();
- QString aStreamName = GetName();
+ HYDROData_SequenceOfObjects aSeqOfProfiles = GetProfiles();
+ for ( int i = 1, aNb = aSeqOfProfiles.Size(); i <= aNb; ++i )
+ {
+ const Handle(HYDROData_Entity) aProfile = aSeqOfProfiles.Value( i );
+ setPythonReferenceObject( theTreatedObjects, aResList, aProfile, "AddProfile" );
+ }
- 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( "" );
-
- // TODO
+ aResList << QString( "%1.Update();" ).arg( aName );
+ aResList << QString( "" );
return aResList;
}
return getShape3D();
}
+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()
{
- HYDROData_NaturalObject::Update();
+ updateProfilesOrder();
+ UpdatePrs();
+}
- Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
- HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
- if ( aHydAxis.IsNull() || aRefProfiles.IsEmpty() )
- return;
+bool HYDROData_Stream::IsHas2dPrs() const
+{
+ return true;
+}
- TopoDS_Shell a2dShell;
- BRep_Builder a2dShellBuilder;
- a2dShellBuilder.MakeShell( a2dShell );
+bool HYDROData_Stream::CreatePresentations( const Handle(HYDROData_PolylineXY)& theHydAxis,
+ const HYDROData_SequenceOfObjects& theProfiles,
+ PrsDefinition& thePrs )
+{
+ if ( theHydAxis.IsNull() || theProfiles.Length() < 2 )
+ return false;
- bool anIsFirst = true;
gp_Pnt aPrevFirstPoint, aPrevLastPoint;
-
- // Construct the top presentation
- HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
- for ( ; anIter.More(); anIter.Next() )
+ Handle(TColgp_HArray1OfPnt) anArrayOfFPnt = new TColgp_HArray1OfPnt(1, theProfiles.Length());
+ Handle(TColgp_HArray1OfPnt) anArrayOfLPnt = new TColgp_HArray1OfPnt(1, theProfiles.Length());
+ TopTools_Array1OfShape anArrOfProfiles(1, theProfiles.Length());
+ TopTools_Array1OfShape anArrOf2DProfiles(1, theProfiles.Length());
+
+ // Pre-processing
+ HYDROData_SequenceOfObjects::Iterator anIter( theProfiles );
+ 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->GetFirstPoint( aPnt1 ) || !aProfile->GetLastPoint( aPnt2 ) )
+ if ( !aProfile->GetLeftPoint( aPnt1 ) || !aProfile->GetRightPoint( aPnt2 ) )
continue;
-
- gp_Pnt aCurFirstPoint( aPnt1.X(), aPnt1.Y(), 0 );
- gp_Pnt aCurLastPoint( aPnt2.X(), aPnt2.Y(), 0 );
- if ( anIsFirst )
- {
- aPrevFirstPoint = aCurFirstPoint;
- aPrevLastPoint = aCurLastPoint;
- anIsFirst = false;
- 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);
+ }
- BRepBuilderAPI_MakeEdge aFirstEdge( aPrevFirstPoint, aPrevLastPoint );
- BRepBuilderAPI_MakeEdge aSecondEdge( aPrevLastPoint, aCurLastPoint );
- BRepBuilderAPI_MakeEdge aThirdEdge( aCurLastPoint, aCurFirstPoint );
- BRepBuilderAPI_MakeEdge aFourthEdge( aCurFirstPoint, aPrevFirstPoint );
+ // Construct of the 3D presentation
+ Handle(Geom_BSplineCurve) aBSpline = buildInterpolationCurve (anArrayOfFPnt);
+ if(aBSpline.IsNull())
+ return false;
- BRepBuilderAPI_MakeWire aMakeWire( aFirstEdge.Edge(), aSecondEdge.Edge(),
- aThirdEdge.Edge(), aFourthEdge.Edge() );
-
- TopoDS_Wire aSectProfileWire = aMakeWire.Wire();
-
- BRepBuilderAPI_MakeFace aMakeFace( aSectProfileWire, Standard_True );
- aMakeFace.Build();
- if( aMakeFace.IsDone() )
- {
- a2dShellBuilder.Add( a2dShell, aMakeFace.Face() );
- }
+ TopoDS_Edge anEdgLeft, anEdgRight;
+
+ BRepBuilderAPI_MakeEdge aMakeEdge(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgLeft = aMakeEdge.Edge();
- aPrevFirstPoint = aCurFirstPoint;
- aPrevLastPoint = aCurLastPoint;
+ if(anEdgLeft.IsNull())
+ return false;
+
+ aBSpline.Nullify();
+ aBSpline = buildInterpolationCurve (anArrayOfLPnt);
+ if(aBSpline.IsNull())
+ return false;
+
+ aMakeEdge.Init(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgRight = aMakeEdge.Edge();
+
+ if(anEdgRight.IsNull())
+ return false;
+
+ BRep_Builder aBB;
+ TopoDS_Compound aCmp;
+ aBB.MakeCompound(aCmp);
+ anIter.Init( theProfiles );
+ 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())
+ thePrs.myPrs3D = aCmp;
+#ifdef DEB_UPDATE
+ else {
+ BRepTools::Write(aCmp, "str3d.brep");
+ thePrs.myPrs3D = 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 false;
+
+ aMakeEdge.Init(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgLeft = aMakeEdge.Edge();
+
+ aBSpline.Nullify();
+ aBSpline = buildInterpolationCurve (anArrayOfLPnt);
+ if(aBSpline.IsNull())
+ return false;
- SetTopShape( a2dShell );
+ aMakeEdge.Init(aBSpline);
+ if(aMakeEdge.IsDone())
+ anEdgRight = aMakeEdge.Edge();
+ if(anEdgRight.IsNull())
+ return false;
+
+ 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 );
+ thePrs.myInlet = aMakeWire.Edge();
+
+ aMakeWire.Add( anEdgLeft );
+ thePrs.myLeftBank = aMakeWire.Edge();
+
+ aMakeWire.Add( aTopEdge );
+ thePrs.myOutlet = aMakeWire.Edge();
+
+ aMakeWire.Add( anEdgRight );
+ thePrs.myRightBank = 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();
- // Construct the 3D presentation
- /// TODO
+ 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())
+ thePrs.myPrs2D = aCmp;
+#ifdef DEB_UPDATE
+ else {
+ BRepTools::Write(aCmp, "str2d.brep");
+ thePrs.myPrs2D = aCmp;
+ }
+#endif
+
+ return true;
}
-bool HYDROData_Stream::SetHydraulicAxis( const Handle(HYDROData_PolylineXY)& theAxis )
+void HYDROData_Stream::UpdatePrs()
{
- Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
+ HYDROData_NaturalObject::Update();
- if ( theAxis.IsNull() )
- {
- RemoveHydraulicAxis();
- return !aPrevAxis.IsNull();
- }
+ Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+ HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
- if ( IsEqual( aPrevAxis, theAxis ) )
+ PrsDefinition aResultPrs;
+ if ( !CreatePresentations( aHydAxis, aRefProfiles, aResultPrs ) )
+ return;
+
+ SetShape3D( aResultPrs.myPrs3D );
+ SetTopShape( aResultPrs.myPrs2D );
+
+ // Create the stream groups
+ QString aLeftGroupName = GetName() + "_Left_Bank";
+
+ Handle(HYDROData_ShapesGroup) aLeftGroup = createGroupObject();
+ aLeftGroup->SetName( aLeftGroupName );
+ aLeftGroup->AddShape( aResultPrs.myLeftBank );
+
+ QString aRightGroupName = GetName() + "_Right_Bank";
+
+ Handle(HYDROData_ShapesGroup) aRightGroup = createGroupObject();
+ aRightGroup->SetName( aRightGroupName );
+ aRightGroup->AddShape( aResultPrs.myRightBank );
+
+ QString anInGroupName = GetName() + "_Inlet";
+
+ Handle(HYDROData_ShapesGroup) anInGroup = createGroupObject();
+ anInGroup->SetName( anInGroupName );
+ anInGroup->AddShape( aResultPrs.myInlet );
+
+ QString anOutGroupName = GetName() + "_Outlet";
+
+ Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
+ anOutGroup->SetName( anOutGroupName );
+ anOutGroup->AddShape( aResultPrs.myOutlet );
+}
+
+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;
- 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 );
SetToUpdate( true );
}
-bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_Profile)& theProfile, const TopoDS_Face& thePlane,
- Standard_Real& outPar ) const
+bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_Profile)& theProfile,
+ const TopoDS_Face& thePlane,
+ Standard_Real& theOutPar ) const
{
Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
- if ( theProfile.IsNull() || aHydAxis.IsNull() )
+ return HasIntersection( aHydAxis, theProfile, thePlane, theOutPar );
+}
+
+bool HYDROData_Stream::HasIntersection( const Handle(HYDROData_PolylineXY)& theHydAxis,
+ const Handle(HYDROData_Profile)& theProfile,
+ const TopoDS_Face& thePlane,
+ Standard_Real& theOutPar )
+{
+ if ( theProfile.IsNull() || !IsValidAsAxis( theHydAxis ) )
return false;
- TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() ); //guide line
+ TopoDS_Wire aHydraulicWire = TopoDS::Wire( theHydAxis->GetShape() ); //guide line
TopoDS_Wire aProfileWire = TopoDS::Wire( theProfile->GetTopShape() );
if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
return false;
- // TODO
- //return true; // temporary
BRepProj_Projection aProjector (aProfileWire, thePlane, gp::OZ().Direction());
if(!aProjector.IsDone())
return false;
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);
Standard_Integer anIndx(0);
BRepExtrema_ExtCC aCC;
aCC.Initialize(anEdg2);
- outPar = 0.0;
+ theOutPar = 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 {
+ theOutPar += BRep_Tool::Pnt(aV1).Distance(aPnt);
+ } else {
Standard_Real aPar = aCC.ParameterOnE1(anIndx);
- outPar = aPar;
- }
- hasInt = true;
- break;
- } else {
- // no ints-n
+ theOutPar = 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));
- }
+ theOutPar += BRep_Tool::Pnt(aV1).Distance(BRep_Tool::Pnt(aV2));
+ }
+ }
+ } else if(aNum > 1) {
+ TopExp::Vertices(anEdg1, aV1, aV2);
+ theOutPar += BRep_Tool::Pnt(aV1).Distance(BRep_Tool::Pnt(aV2));
+ }
}
if(hasInt)
return true;
bool HYDROData_Stream::AddProfile( const Handle(HYDROData_Profile)& theProfile )
{
if ( theProfile.IsNull() )
- return false;
+ return false;
+
Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
if ( aHydAxis.IsNull() )
- return false;
+ 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
- insertProfileInToOrder( theProfile );
+ int aProfileIndex = insertParameter( aPar );
+ insertProfileInToOrder( theProfile, aProfileIndex );
// Indicate model of the need to update the stream presentation
SetToUpdate( true );
return true;
}
+bool HYDROData_Stream::SetProfiles( const HYDROData_SequenceOfObjects& theProfiles,
+ const bool& theIsToOrder )
+{
+ if ( theIsToOrder )
+ {
+ for ( int i = 1; i <= theProfiles.Length(); ++i )
+ {
+ Handle(HYDROData_Profile) aProfile =
+ Handle(HYDROData_Profile)::DownCast( theProfiles.Value( i ) );
+ if ( aProfile.IsNull() )
+ continue;
+
+ if ( !AddProfile( aProfile ) )
+ return false;
+ }
+ }
+ else // Just store the sequence of objects as is
+ {
+ bool anIsToUpdate = true;
+
+ HYDROData_SequenceOfObjects anOldProfiles = GetProfiles();
+ if ( anOldProfiles.Length() == theProfiles.Length() )
+ {
+ anIsToUpdate = false;
+
+ for ( int i = 1; i <= theProfiles.Length(); ++i )
+ {
+ Handle(HYDROData_Entity) anOldProfile = anOldProfiles.Value( i );
+ Handle(HYDROData_Entity) aNewProfile = theProfiles.Value( i );
+ if ( !IsEqual( anOldProfile, aNewProfile ) )
+ {
+ anIsToUpdate = true;
+ break;
+ }
+ }
+ }
+
+ SetReferenceObjects( theProfiles, DataTag_Profile );
+
+ if ( anIsToUpdate )
+ 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() || !HasReference( theProfile, DataTag_Profile ) )
+ 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 );
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 )
+void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)& theProfile,
+ const int theBeforeIndex )
{
Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
if ( theProfile.IsNull() || aHydAxis.IsNull() )
if ( aHydraulicWire.IsNull() || aProfileWire.IsNull() )
return;
- // TODO
- AddReferenceObject( theProfile, DataTag_Profile ); // temporary for testing only
+ 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
+bool HYDROData_Stream::BuildFace( const Handle(HYDROData_PolylineXY)& theHydAxis,
+ TopoDS_Face& thePlane )
{
- 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);
thePlane = TopoDS::Face(aMkr.Shape());
return true;
}
+
void HYDROData_Stream::updateProfilesOrder()
{
HYDROData_SequenceOfObjects aRefProfiles = GetProfiles();
Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
if ( aHydAxis.IsNull() )
return;
+
TopoDS_Face aPlane;
- if(!BuildFace(aHydAxis, aPlane))
+ if ( !BuildFace( aHydAxis, aPlane ) )
return;
- Standard_Real aPar(.0);
+
+ Standard_Real aPar( .0 );
+
#ifdef DEB_HASINT
BRep_Builder aBB;
TopoDS_Compound aCmp;
aBB.MakeCompound(aCmp);
#endif
+
HYDROData_DataMapOfRealOfHDProfile aDM;
TColStd_ListOfReal aList;
HYDROData_SequenceOfObjects::Iterator anIter( aRefProfiles );
Handle(HYDROData_Profile)::DownCast( anIter.Value() );
#ifdef DEB_HASINT
TopoDS_Wire aProfileWire = TopoDS::Wire( aProfile->GetTopShape() );
- aBB.Add( aCmp, aProfileWire);
+ aBB.Add( aCmp, aProfileWire );
#endif
if ( aProfile.IsNull() || !HasIntersection( aProfile, aPlane, aPar ) )
continue;
- aDM.Bind(aPar, aProfile);
- aList.Append(aPar);
- //insertProfileInToOrder( aProfile );
+
+ aDM.Bind( aPar, aProfile );
+ aList.Append( aPar );
}
- // sorting
- if(aList.Extent() > 1) {
- TColStd_Array1OfReal anArr(1, aList.Extent());
- TColStd_ListIteratorOfListOfReal it(aList);
- for (int j=1;it.More();it.Next(), j++)
- anArr(j) = it.Value();
- TCollection_CompareOfReal Compar;
- SortTools_QuickSortOfReal::Sort( anArr, Compar);
- for (int j = 1; j <= anArr.Length(); j++) {
- const Standard_Real aKey = anArr(j);
- const Handle(HYDROData_Profile)& aProfile = aDM.Find(aKey);
- insertProfileInToOrder( aProfile );
- }
- } else if(aList.Extent() == 1) {
+
+ if ( aList.IsEmpty() )
+ return;
+
+ TColStd_Array1OfReal anArr( 1, aList.Extent() );
+
+ TColStd_ListIteratorOfListOfReal it( aList );
+ for ( int j = 1; it.More(); it.Next(), j++ )
+ anArr( j ) = it.Value();
+
+ // sorting
+ if ( aList.Extent() > 1 )
+ {
+ TCollection_CompareOfReal Compar;
+ SortTools_QuickSortOfReal::Sort( anArr, Compar );
+
+ for (int j = 1; j <= anArr.Length(); j++) {
+ const Standard_Real aKey = anArr(j);
+ const Handle(HYDROData_Profile)& aProfile = aDM.Find(aKey);
+ insertProfileInToOrder( aProfile );
+ }
+ } else if ( aList.Extent() == 1 ) {
const Standard_Real aKey = aList.Last();
const Handle(HYDROData_Profile)& aProfile = aDM.Find(aKey);
insertProfileInToOrder( aProfile );
}
+ setParametersArray( anArr );
+
#ifdef DEB_HASINT
TopoDS_Wire aHydraulicWire = TopoDS::Wire( aHydAxis->GetShape() );
BRepTools::Write(aHydraulicWire, "Path.brep");
#endif
}
+ObjectKind HYDROData_Stream::getAltitudeObjectType() const
+{
+ return KIND_STREAM_ALTITUDE;
+}
+
+void HYDROData_Stream::setParametersArray( const TColStd_Array1OfReal& theArray )
+{
+ if ( theArray.Length() == 0 )
+ {
+ removeParametersArray();
+ return;
+ }
+
+ TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray );
+
+ Handle(TDataStd_RealArray) aParamsArray =
+ TDataStd_RealArray::Set( aLabel, theArray.Lower(), theArray.Upper() );
+
+ for ( int i = theArray.Lower(), n = theArray.Upper(); i <= n; ++i )
+ {
+ const Standard_Real& aParam = theArray( i );
+ aParamsArray->SetValue( i, aParam );
+ }
+}
+
+TColStd_Array1OfReal* HYDROData_Stream::getParametersArray() const
+{
+ TColStd_Array1OfReal* anArray = NULL;
+
+ TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray, false );
+ if ( !aLabel.IsNull() )
+ {
+ Handle(TDataStd_RealArray) aParamsArray;
+ if ( aLabel.FindAttribute( TDataStd_RealArray::GetID(), aParamsArray ) )
+ {
+ anArray = new TColStd_Array1OfReal( aParamsArray->Lower(), aParamsArray->Upper() );
+ for ( int i = aParamsArray->Lower(), n = aParamsArray->Upper(); i <= n; ++i )
+ {
+ const Standard_Real& aParam = aParamsArray->Value( i );
+ anArray->SetValue( i, aParam );
+ }
+ }
+ }
+
+ return anArray;
+}
+
+void HYDROData_Stream::removeParametersArray()
+{
+ TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray, false );
+ if ( !aLabel.IsNull() )
+ aLabel.ForgetAllAttributes();
+}
+
+int HYDROData_Stream::insertParameter( const Standard_Real& theParam )
+{
+ int aResIndex = -1;
+
+ TColStd_Array1OfReal* anArr = getParametersArray();
+ if ( anArr )
+ {
+ aResIndex = 0;
+
+ TColStd_Array1OfReal aNewArr( anArr->Lower(), anArr->Upper() + 1 );
+
+ bool isInserted = false;
+ for ( int i = anArr->Lower(), j = i, n = anArr->Upper(); i <= n; ++i, ++j )
+ {
+ const Standard_Real& aStoredParam = anArr->Value( i );
+ if ( !isInserted )
+ {
+ if ( theParam > aStoredParam )
+ {
+ aResIndex++;
+ }
+ else
+ {
+ aNewArr( j ) = theParam;
+ isInserted = true;
+ ++j;
+ }
+ }
+
+ aNewArr( j ) = aStoredParam;
+ }
+
+ if ( !isInserted )
+ {
+ aResIndex = -1;
+ aNewArr( aNewArr.Upper() ) = theParam;
+ }
+
+ setParametersArray( aNewArr );
+ delete anArr;
+ }
+ else
+ {
+ TColStd_Array1OfReal aNewArr( 1, 1 );
+ aNewArr.SetValue( 1, theParam );
+ setParametersArray( aNewArr );
+ }
+
+ return aResIndex;
+}
+
+void HYDROData_Stream::removeParameter( const int& theIndex )
+{
+ TDF_Label aLabel = myLab.FindChild( DataTag_ParamsArray, false );
+ if ( aLabel.IsNull() )
+ return;
+
+ Handle(TDataStd_RealArray) aParamsArray;
+ if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), aParamsArray ) )
+ return;
+
+ if ( aParamsArray->Length() == 1 )
+ {
+ removeParametersArray();
+ return;
+ }
+
+ TColStd_Array1OfReal aNewArr( aParamsArray->Lower(), aParamsArray->Upper() - 1 );
+
+ for ( int i = aParamsArray->Lower(), j = i, k = 0, n = aParamsArray->Upper(); i <= n; ++i, ++k )
+ {
+ const Standard_Real& aStoredParam = aParamsArray->Value( i );
+ if ( k == theIndex )
+ continue;
+
+ aNewArr.SetValue( j, aStoredParam );
+ ++j;
+ }
+
+ setParametersArray( aNewArr );
+}