X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_StreamAltitude.cxx;h=2360eddd0fce93aa735dd9b99d12bfe530995f6e;hb=e35b4caf3341d2cece2b7921f349acd17fb89351;hp=b701b3af768f9aa793b258a52b8bc1265b5e6bd1;hpb=517590ff502b021ac3b88b59c9f063a1470cc8a9;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_StreamAltitude.cxx b/src/HYDROData/HYDROData_StreamAltitude.cxx index b701b3af..2360eddd 100644 --- a/src/HYDROData/HYDROData_StreamAltitude.cxx +++ b/src/HYDROData/HYDROData_StreamAltitude.cxx @@ -1,13 +1,67 @@ +// Copyright (C) 2014-2015 EDF-R&D +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "HYDROData_StreamAltitude.h" + +#include "HYDROData_Channel.h" #include "HYDROData_Document.h" +#include "HYDROData_Profile.h" +#include "HYDROData_Stream.h" +#include "HYDROData_ShapesTool.h" +#include +#include + +#include +#include +#include + +#include + +#include + +#include + +#include + +#include +#include + +#include +#include +#include + +#include +#include + +#include #include +#include -#define PYTHON_STREAM_ALTITUDE_ID "KIND_STREAM_ALTITUDE" +#define _DEVDEBUG_ +#include "HYDRO_trace.hxx" +#ifdef DEB_CLASS2D +#include +#include +#include +#endif -IMPLEMENT_STANDARD_HANDLE(HYDROData_StreamAltitude, HYDROData_IAltitudeObject) IMPLEMENT_STANDARD_RTTIEXT(HYDROData_StreamAltitude, HYDROData_IAltitudeObject) HYDROData_StreamAltitude::HYDROData_StreamAltitude() @@ -19,31 +73,348 @@ HYDROData_StreamAltitude::~HYDROData_StreamAltitude() { } -QStringList HYDROData_StreamAltitude::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const +bool IsPointBetweenEdges( const gp_Pnt& aFirstPnt1, const gp_Pnt& aLastPnt1, + const gp_Pnt& aFirstPnt2, const gp_Pnt& aLastPnt2, + const gp_Pnt& thePoint) { + BRepBuilderAPI_MakeEdge aLeftMakeEdge( aFirstPnt1, aLastPnt1 ); + BRepBuilderAPI_MakeEdge aBotMakeEdge( aLastPnt1, aLastPnt2 ); + BRepBuilderAPI_MakeEdge aRightMakeEdge( aLastPnt2, aFirstPnt2 ); + BRepBuilderAPI_MakeEdge aTopMakeEdge( aFirstPnt2, aFirstPnt1 ); + + BRepBuilderAPI_MakeWire aMakeWire( aLeftMakeEdge.Edge(), aBotMakeEdge.Edge(), + aRightMakeEdge.Edge(), aTopMakeEdge.Edge() ); + + BRepBuilderAPI_MakeFace aMakeFace( aMakeWire.Wire() ); + + TopoDS_Face aFace = aMakeFace.Face(); +#ifdef DEB_CLASS2D + TopoDS_Compound aCmp; + BRep_Builder aBB; + aBB.MakeCompound(aCmp); + aBB.Add(aCmp, aFace); + BRepBuilderAPI_MakeVertex aMk(thePoint); + aBB.Add(aCmp, aMk.Vertex()); + BRepTools::Write(aCmp, "ProfileFace.brep"); +#endif + + gp_XY anXY( thePoint.X(), thePoint.Y() ); + TopAbs_State aPointState = HYDROData_Tool::ComputePointState(anXY, aFace); + +#ifdef DEB_CLASS2D + cout << "Point status is = " << aPointState <=0) && (z2>=0))) + { + isBetween = false; + } + if (isBetween) + { + double xg = (aFirstPnt1.X() + aLastPnt1.X() + aFirstPnt2.X() + aLastPnt2.X())/4; + double yg = (aFirstPnt1.Y() + aLastPnt1.Y() + aFirstPnt2.Y() + aLastPnt2.Y())/4; + double df1 = (aFirstPnt1.X()-xg)*(aFirstPnt1.X()-xg) + (aFirstPnt1.Y()-yg)*(aFirstPnt1.Y()-yg); + double dl1 = (aLastPnt1.X()-xg)*(aLastPnt1.X()-xg) + (aLastPnt1.Y()-yg)*(aLastPnt1.Y()-yg); + double df2 = (aFirstPnt2.X()-xg)*(aFirstPnt2.X()-xg) + (aFirstPnt2.Y()-yg)*(aFirstPnt2.Y()-yg); + double dl2 = (aLastPnt2.X()-xg)*(aLastPnt2.X()-xg) + (aLastPnt2.Y()-yg)*(aLastPnt2.Y()-yg); + double r2 = std::max(df1,dl1); + r2 = std::max(r2,df2); + r2 = std::max(r2,dl2); + double d2 = (thePoint.X()-xg)*(thePoint.X()-xg) + (thePoint.Y()-yg)*(thePoint.Y()-yg); + if (d2 > r2) + isBetween = false; + } + return isBetween; +} + +Standard_Real getAltitudeFromWire( const TopoDS_Wire& theWire, + const Standard_Real& theLeftDist, + const Standard_Real& theRightDist ) +{ + Standard_Real aResAlt = 0.0; + + gp_XY aFirstPoint, aLastPoint; + TopoDS_Vertex aFirstVertex, aLastVertex; + TopExp::Vertices( theWire, aFirstVertex, aLastVertex ); + + gp_Pnt aPnt1( BRep_Tool::Pnt( aFirstVertex ) ); + aPnt1.SetZ( 0 ); + gp_Pnt aPnt2( BRep_Tool::Pnt( aLastVertex ) ); + aPnt2.SetZ( 0 ); + + Standard_Real aProfileDist = aPnt1.Distance( aPnt2 ); + + Standard_Real aCoeff = aProfileDist / ( theLeftDist + theRightDist ); + + gp_Pnt anIntPoint( aPnt1.XYZ() + ( aCoeff * theLeftDist ) * gp_Dir( gp_Vec( aPnt1, aPnt2 ) ).XYZ() ); + + return HYDROData_Tool::GetAltitudeForWire( theWire, + gp_XY(anIntPoint.X(), anIntPoint.Y()), + 1E-2, + 1E-2, + HYDROData_IAltitudeObject::GetInvalidAltitude() ); +} + +bool HYDROData_StreamAltitude::getBoundaryWiresForPoint( + const gp_XY& thePoint, + TopoDS_Wire& theLeftWire, + TopoDS_Wire& theRightWire ) const { - QStringList aResList; + gp_Pnt aTestPnt( thePoint.X(), thePoint.Y(), 0 ); + + Handle(HYDROData_Object) anObject = + Handle(HYDROData_Object)::DownCast( GetFatherObject() ); + if ( anObject.IsNull() ) { + return false; + } + + if ( anObject->GetKind() == KIND_STREAM ) { + Handle(HYDROData_Stream) aStream = Handle(HYDROData_Stream)::DownCast( anObject ); + if ( aStream.IsNull() ) + return false; + + HYDROData_SequenceOfObjects aStreamProfiles = aStream->GetProfiles(); + if ( aStreamProfiles.Length() < 2 ) + return false; + + Handle(HYDROData_Profile) aPrevProfile; + gp_Pnt aPrevPnt1, aPrevPnt2; + for ( int i = 1, n = aStreamProfiles.Length(); i <= n; ++i ) + { + Handle(HYDROData_Profile) aProfile = + Handle(HYDROData_Profile)::DownCast( aStreamProfiles.Value( i ) ); + if ( aProfile.IsNull() ) + continue; + + gp_XY aFirstPoint, aLastPoint; + if ( !aProfile->GetLeftPoint( aFirstPoint, false ) || + !aProfile->GetRightPoint( aLastPoint, false ) ) + continue; + + gp_Pnt aPnt1( aFirstPoint.X(), aFirstPoint.Y(), 0 ); + gp_Pnt aPnt2( aLastPoint.X(), aLastPoint.Y(), 0 ); + + if ( !aPrevProfile.IsNull() ) + { + if ( IsPointBetweenEdges2( aPrevPnt1, aPrevPnt2, aPnt1, aPnt2, aTestPnt ) ) + { + DEBTRACE("point is between profiles "<< i-1 << " and " <GetShape3D() ); + theRightWire = TopoDS::Wire( aProfile->GetShape3D() ); + break; + } + } - Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab ); - if ( aDocument.IsNull() ) - return aResList; - - QString aDocName = aDocument->GetDocPyName(); - QString anAltitudeName = GetName(); + aPrevProfile = aProfile; + aPrevPnt1 = aPnt1; + aPrevPnt2 = aPnt2; + } + } else if ( anObject->GetKind() == KIND_CHANNEL ) { + Handle(HYDROData_Channel) aChannel = Handle(HYDROData_Channel)::DownCast( anObject ); + if ( aChannel.IsNull() ) + return false; - aResList << QString( "%1 = %2.CreateObject( %3 );" ) - .arg( anAltitudeName ).arg( aDocName ).arg( PYTHON_STREAM_ALTITUDE_ID ); - aResList << QString( "%1.SetName( \"%2\" );" ) - .arg( anAltitudeName ).arg( anAltitudeName ); + TopTools_ListOfShape aWiresList; + TopExp_Explorer anExp( aChannel->GetShape3D(), TopAbs_WIRE ); + for ( ; anExp.More(); anExp.Next() ) { + if(!anExp.Current().IsNull()) { + const TopoDS_Wire& aWire = TopoDS::Wire( anExp.Current() ); + aWiresList.Append( aWire ); + } + } - // TODO + if ( aWiresList.Extent() < 2 ) + return false; - return aResList; + TopoDS_Wire aPrevWire; + gp_Pnt aPrevPnt1, aPrevPnt2; + + TopTools_ListIteratorOfListOfShape anIt( aWiresList ); + for ( ; anIt.More(); anIt.Next() ) { + TopoDS_Wire& aWire = TopoDS::Wire( anIt.Value() ); + if ( aWire.IsNull() ) + continue; + + TopoDS_Vertex aFirstVertex, aLastVertex; + TopExp::Vertices( aWire, aFirstVertex, aLastVertex ); + + gp_Pnt aPnt1( BRep_Tool::Pnt( aFirstVertex ) ); + aPnt1.SetZ( 0 ); + gp_Pnt aPnt2( BRep_Tool::Pnt( aLastVertex ) ); + aPnt2.SetZ( 0 ); + + if ( !aPrevWire.IsNull() ) { + if ( IsPointBetweenEdges2( aPrevPnt1, aPrevPnt2, aPnt1, aPnt2, aTestPnt ) ) { + theLeftWire = aPrevWire; + theRightWire = aWire; + break; + } + } + + aPrevWire = aWire; + aPrevPnt1 = aPnt1; + aPrevPnt2 = aPnt2; + } + } + + return !theLeftWire.IsNull() && !theRightWire.IsNull(); } -double HYDROData_StreamAltitude::GetAltitudeForPoint( const gp_XY& thePoint ) const +double HYDROData_StreamAltitude::GetAltitudeForPoint( const gp_XY& thePoint, + int theMethod) const { + DEBTRACE("GetAltitudeForPoint p(" << thePoint.X() << ", " << thePoint.Y() << ")"); double aResAltitude = GetInvalidAltitude(); + Handle(HYDROData_Object) anObject = + Handle(HYDROData_Object)::DownCast( GetFatherObject() ); + if ( anObject.IsNull() ) + { + DEBTRACE(" --- aStream.IsNull()"); + return aResAltitude; + } + TopoDS_Shape aTopShape = anObject->GetTopShape(); + if ( aTopShape.IsNull() ) + { + DEBTRACE(" --- aTopShape.IsNull()"); + return aResAltitude; + } + + TopExp_Explorer aFaceExp( aTopShape, TopAbs_FACE ); + if ( !aFaceExp.More() ) + { + DEBTRACE(" --- !aFaceExp.More()"); + return aResAltitude; + } + + // Get only face because of 2d profile wires is in compound + TopoDS_Face aFace = TopoDS::Face( aFaceExp.Current() ); + + // Check if point is inside of stream/channel presentation + TopAbs_State aPointState = HYDROData_Tool::ComputePointState(thePoint, aFace); + +#ifdef DEB_CLASS2D + cout << "Point status is = " << aPointState <GetKind() == KIND_STREAM ) { + Handle(HYDROData_Stream) aStream = + Handle(HYDROData_Stream)::DownCast( GetFatherObject() ); + if ( !aStream.IsNull() ) { + aLeftEdge = TopoDS::Edge( aStream->GetLeftShape() ); + aRightEdge = TopoDS::Edge( aStream->GetRightShape() ); + } + } else if ( anObject->GetKind() == KIND_CHANNEL ) { + Handle(HYDROData_Channel) aChannel = + Handle(HYDROData_Channel)::DownCast( GetFatherObject() ); + if ( !aChannel.IsNull() ) { + aLeftEdge = TopoDS::Edge( aChannel->GetLeftShape() ); + aRightEdge = TopoDS::Edge( aChannel->GetRightShape() ); + } + } + + // Find the two profiles between which the point is lies + TopoDS_Wire aLeftWire, aRightWire; + if ( !getBoundaryWiresForPoint( thePoint, aLeftWire, aRightWire ) ) + { + DEBTRACE(" --- !getBoundaryProfilesForPoint( thePoint, aLeftProfile, aRightProfile )"); + return aResAltitude; + } + + // Find the projections of point to borders of stream + gp_XYZ aPointToTest( thePoint.X(), thePoint.Y(), 0.0 ); + + Standard_Real aFirst = 0.0, aLast = 0.0; + Handle(Geom_Curve) anEdgeLeftCurve = BRep_Tool::Curve( aLeftEdge, aFirst, aLast ); + Handle(Geom_Curve) anEdgeRightCurve = BRep_Tool::Curve( aRightEdge, aFirst, aLast ); + + GeomAPI_ProjectPointOnCurve aLeftProject( aPointToTest, anEdgeLeftCurve ); + GeomAPI_ProjectPointOnCurve aRightProject( aPointToTest, anEdgeRightCurve ); + + int aNbPoints1 = aRightProject.NbPoints(); + int aNbPoints2 = aLeftProject.NbPoints(); + if ( aNbPoints1 < 1 || aNbPoints2 < 1) + { + DEBTRACE(" --- projections " << aNbPoints1 << " " << aNbPoints2); + return aResAltitude; + } + Standard_Real aLeftDist = aLeftProject.LowerDistance(); + Standard_Real aRightDist = aRightProject.LowerDistance(); + + // Find the altitude in profiles + Standard_Real aLeftAlt, aRightAlt; + gp_Pnt aLeftProfileP1, aLeftProfileP2, aRightProfileP1, aRightProfileP2; + aLeftAlt = getAltitudeFromWire( aLeftWire, aLeftDist, aRightDist ); + aRightAlt = getAltitudeFromWire( aRightWire, aLeftDist, aRightDist ); + + TopoDS_Vertex aFirstVertex, aLastVertex; + TopExp::Vertices( aLeftWire, aFirstVertex, aLastVertex ); + aLeftProfileP1 = BRep_Tool::Pnt( aFirstVertex ); + aLeftProfileP2 = BRep_Tool::Pnt( aLastVertex ); + + TopExp::Vertices( aRightWire, aFirstVertex, aLastVertex ); + aRightProfileP1 = BRep_Tool::Pnt( aFirstVertex ); + aRightProfileP2 = BRep_Tool::Pnt( aLastVertex ); + + // Interpolate altitudes + // Left profile line ( the segment between the first and the last profile point ) + aLeftProfileP1.SetZ( 0 ); + aLeftProfileP2.SetZ( 0 ); + gp_Vec aLeftProfileVec( aLeftProfileP1, aLeftProfileP2 ); + Handle(Geom_Line) aLeftProfileLine = new Geom_Line( gp_Ax1( aLeftProfileP1, aLeftProfileVec ) ); + // Right profile line + aRightProfileP1.SetZ( 0 ); + aRightProfileP2.SetZ( 0 ); + gp_Vec aRightProfileVec( aRightProfileP1, aRightProfileP2 ); + Handle(Geom_Line) aRightProfileLine = new Geom_Line( gp_Ax1( aRightProfileP1, aRightProfileVec ) ); + // The point projections on the left and right profiles + GeomAPI_ProjectPointOnCurve aLeftProfileProject( aPointToTest, aLeftProfileLine ); + GeomAPI_ProjectPointOnCurve aRightProfileProject( aPointToTest, aRightProfileLine ); + // The point distance to the left and right profiles + Standard_Real aLeftProfileDist = aLeftProfileProject.LowerDistance(); + Standard_Real aRightProfileDist = aRightProfileProject.LowerDistance(); + // The coefficients + Standard_Real aFirstCoeff = aLeftProfileDist / ( aLeftProfileDist + aRightProfileDist ); + Standard_Real aSecCoeff = aRightProfileDist / ( aLeftProfileDist + aRightProfileDist ); + + aResAltitude = aLeftAlt * ( 1 - aFirstCoeff ) + aRightAlt * ( 1 - aSecCoeff ); + DEBTRACE("aResAltitude=" << aResAltitude); return aResAltitude; }