X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_DTM.cxx;h=544dd2e53184269221fe352cacc1f099c5b7fb0f;hb=f9d37ee66fa46871478d806faa54de237225d3c6;hp=a80edf2202d3ddae55e07a21006e11a61d27cc06;hpb=abdcba0f2ee048bf3c1f6073ef7fea4f0b200dcb;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_DTM.cxx b/src/HYDROData/HYDROData_DTM.cxx index a80edf22..544dd2e5 100644 --- a/src/HYDROData/HYDROData_DTM.cxx +++ b/src/HYDROData/HYDROData_DTM.cxx @@ -18,13 +18,125 @@ #include #include #include +#include #include #include #include +#include +#include +#include +#include +#include +#include +#include +#include IMPLEMENT_STANDARD_HANDLE( HYDROData_DTM, HYDROData_Bathymetry ) IMPLEMENT_STANDARD_RTTIEXT( HYDROData_DTM, HYDROData_Bathymetry ) + +HYDROData_DTM::CurveUZ::CurveUZ( double theXCurv, const gp_Vec2d& theProfileDir ) + : myXcurv( theXCurv ), myProfileDir( theProfileDir ) +{ +} + +HYDROData_DTM::CurveUZ::~CurveUZ() +{ +} + +double HYDROData_DTM::CurveUZ::Xcurv() const +{ + return myXcurv; +} + +gp_Vec2d HYDROData_DTM::CurveUZ::ProfileDir() const +{ + return myProfileDir; +} + +HYDROData_DTM::CurveUZ HYDROData_DTM::CurveUZ::operator + ( const CurveUZ& c ) const +{ + HYDROData_DTM::CurveUZ res( Xcurv() + c.Xcurv(), ProfileDir() + c.ProfileDir() ); + size_t n = size(); + res.reserve( n ); + for( int i=0; iSetValue( (int)i, aPnt ); + tangents.SetValue( (int)i, aVec ); + flags->SetValue( (int)i, Standard_True ); + } + + GeomAPI_Interpolate anInterpolator( points, Standard_False, Standard_False ); + anInterpolator.Load( tangents, flags ); + anInterpolator.Perform(); + if( anInterpolator.IsDone() ) + { + aCurve = anInterpolator.Curve(); + return BRepBuilderAPI_MakeEdge( aCurve ).Edge(); + } + else + return TopoDS_Edge(); +} + + + HYDROData_DTM::HYDROData_DTM() { } @@ -33,11 +145,76 @@ HYDROData_DTM::~HYDROData_DTM() { } +HYDROData_SequenceOfObjects HYDROData_DTM::GetProfiles() const +{ + return GetReferenceObjects( DataTag_Profiles ); +} + +void HYDROData_DTM::SetProfiles( const HYDROData_SequenceOfObjects& theProfiles ) +{ + SetReferenceObjects( theProfiles, DataTag_Profiles ); +} + +double HYDROData_DTM::GetDDZ() const +{ + return GetDouble( DataTag_DDZ ); +} + +void HYDROData_DTM::SetDDZ( double theDDZ ) +{ + SetDouble( DataTag_DDZ, theDDZ ); +} + +double HYDROData_DTM::GetSpatialStep() const +{ + return GetDouble( DataTag_SpatialStep ); +} -void GetProperties( const Handle_HYDROData_Profile& theProfile, - gp_Pnt& theLowestPoint, - gp_Vec2d& theDir, - bool isNormalDir ) +void HYDROData_DTM::SetSpatialStep( double theSpatialStep ) +{ + SetDouble( DataTag_SpatialStep, theSpatialStep ); +} + +void HYDROData_DTM::Update() +{ + HYDROData_SequenceOfObjects objs = GetProfiles(); + int aLower = objs.Lower(), anUpper = objs.Upper(); + size_t n = anUpper-aLower+1; + + std::vector profiles; + profiles.reserve( n ); + for( int i=aLower; i<=anUpper; i++ ) + { + Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( objs.Value( i ) ); + if( !aProfile.IsNull() ) + profiles.push_back( aProfile ); + } + + double ddz = GetDDZ(); + double step = GetSpatialStep(); + const double EPS = 1E-3; + AltitudePoints points; + + myLeft.clear(); + myRight.clear(); + if( ddz>EPS && step>EPS ) + points = Interpolate( profiles, ddz, step, &myLeft, &myRight ); + SetAltitudePoints( points ); +} + + + + + + + + + + +void HYDROData_DTM::GetProperties( const Handle_HYDROData_Profile& theProfile, + gp_Pnt& theLowestPoint, gp_Vec2d& theDir, + bool isNormalDir, + double& theZMin, double& theZMax ) { theLowestPoint = theProfile->GetBottomPoint(); @@ -50,26 +227,50 @@ void GetProperties( const Handle_HYDROData_Profile& theProfile, theDir = gp_Vec2d( -y, x ); else theDir = gp_Vec2d( x, y ); + + HYDROData_Profile::ProfilePoints points = theProfile->GetProfilePoints(); + int lo = points.Lower(); + int up = points.Upper(); + theZMin = std::numeric_limits::max(); + theZMax = -theZMin; + for( int i=lo; i<=up; i++ ) + { + double z = points.Value( i ).Z(); + if( z>theZMax ) + theZMax = z; + if( ztheUMax ) + theUMax = u; + return gp_Pnt2d( u, z ); } -Handle(TColgp_HArray1OfPnt2d) To2D( const TColgp_Array1OfPnt& thePoints, const gp_Trsf& theTr ) +Handle(TColgp_HArray1OfPnt2d) To2D( const TColgp_Array1OfPnt& thePoints, + const gp_Trsf& theTr, + double& theUMin, double& theUMax ) { int low = thePoints.Lower(), up = thePoints.Upper(); Handle(TColgp_HArray1OfPnt2d) points = new TColgp_HArray1OfPnt2d( low, up ); for( int i=low; i<=up; i++ ) - points->SetValue( i, To2D( thePoints.Value( i ), theTr ) ); + points->SetValue( i, To2D( thePoints.Value( i ), theTr, theUMin, theUMax ) ); return points; } Handle(Geom2d_Curve) CurveTo2D( const Handle(Geom_Curve)& theCurve, Standard_Real theFirst, Standard_Real theLast, - const gp_Trsf& theTr ) + const gp_Trsf& theTr, + double& theUMin, double& theUMax ) { if( theCurve->IsKind( STANDARD_TYPE( Geom_Line ) ) ) { @@ -78,8 +279,8 @@ Handle(Geom2d_Curve) CurveTo2D( const Handle(Geom_Curve)& theCurve, theCurve->D0( theLast, aLastPnt ); gp_Pnt2d - aFirst2d = To2D( aFirstPnt, theTr ), - aLast2d = To2D( aLastPnt, theTr ); + aFirst2d = To2D( aFirstPnt, theTr, theUMin, theUMax ), + aLast2d = To2D( aLastPnt, theTr, theUMin, theUMax ); gp_Vec2d dir( aFirst2d, aLast2d ); Handle_Geom2d_Line aLine2d = new Geom2d_Line( aFirst2d, gp_Dir2d( dir.X(), dir.Y() ) ); @@ -90,7 +291,7 @@ Handle(Geom2d_Curve) CurveTo2D( const Handle(Geom_Curve)& theCurve, { Handle(Geom_BSplineCurve) aSpline = Handle(Geom_BSplineCurve)::DownCast( theCurve ); - Handle(TColgp_HArray1OfPnt2d) poles = To2D( aSpline->Poles(), theTr ); + Handle(TColgp_HArray1OfPnt2d) poles = To2D( aSpline->Poles(), theTr, theUMin, theUMax ); const TColStd_Array1OfReal& knots = aSpline->Knots(); const TColStd_Array1OfInteger& multiplicities = aSpline->Multiplicities(); int aDegree = aSpline->Degree(); @@ -102,7 +303,8 @@ Handle(Geom2d_Curve) CurveTo2D( const Handle(Geom_Curve)& theCurve, } Handle_Geom2d_BSplineCurve HYDROData_DTM::CreateHydraulicAxis( - const std::vector& theProfiles ) + const std::vector& theProfiles, + std::vector& theDistances ) { size_t n = theProfiles.size(); Handle_Geom2d_BSplineCurve aResult; @@ -114,9 +316,12 @@ Handle_Geom2d_BSplineCurve HYDROData_DTM::CreateHydraulicAxis( for( size_t i = 1; i <= n; i++ ) { Handle_HYDROData_Profile aProfile = theProfiles[i-1]; + aProfile->Update(); + gp_Pnt aLowest; gp_Vec2d aTangent; - GetProperties( aProfile, aLowest, aTangent, true ); + double zmin, zmax; + GetProperties( aProfile, aLowest, aTangent, true, zmin, zmax ); aTangent.Normalize(); points->SetValue( (int)i, gp_Pnt2d( aLowest.X(), aLowest.Y() ) ); @@ -128,22 +333,40 @@ Handle_Geom2d_BSplineCurve HYDROData_DTM::CreateHydraulicAxis( anInterpolator.Load( tangents, flags ); anInterpolator.Perform(); if( anInterpolator.IsDone() ) + { aResult = anInterpolator.Curve(); + + //fill the distances vector + Geom2dAdaptor_Curve anAdaptor( aResult ); + + theDistances.clear(); + theDistances.reserve( n ); + Standard_Real aParamFirst = anAdaptor.FirstParameter(), aParamLast = anAdaptor.LastParameter(); + for( size_t i = 1; i <= n; i++ ) + { + gp_Pnt2d aPnt = points->Value( (Standard_Integer)i ); + Geom2dAPI_ProjectPointOnCurve aProject( aPnt, aResult ); + Standard_Real aParam = aProject.LowerDistanceParameter(); + double aDistance = GCPnts_AbscissaPoint::Length( anAdaptor, aParamFirst, aParam ); + theDistances.push_back( aDistance ); + } + } return aResult; } -std::vector HYDROData_DTM::ProfileToParametric( const Handle_HYDROData_Profile& theProfile ) +std::vector HYDROData_DTM::ProfileToParametric( + const Handle_HYDROData_Profile& theProfile, + double& theUMin, double& theUMax, gp_Vec2d& theDir ) { std::vector curves; - theProfile->Update(); // Transformation of the coordinate systems gp_Pnt aLowest; - gp_Vec2d aDir; - GetProperties( theProfile, aLowest, aDir, false ); + double zmin, zmax; + GetProperties( theProfile, aLowest, theDir, false, zmin, zmax ); gp_Ax3 aStd3d( gp_Pnt( 0, 0, 0 ), gp_Dir( 0, 0, 1 ), gp_Dir( 1, 0, 0 ) ); - gp_Ax3 aLocal( aLowest, gp_Dir( 0, 0, 1 ), gp_Dir( aDir.X(), aDir.Y(), 0 ) ); + gp_Ax3 aLocal( aLowest, gp_Dir( 0, 0, 1 ), gp_Dir( theDir.X(), theDir.Y(), 0 ) ); gp_Trsf aTransf; aTransf.SetTransformation( aStd3d, aLocal ); @@ -162,30 +385,297 @@ std::vector HYDROData_DTM::ProfileToParametric( const Handl Handle(Geom_Curve) aCurve = BRep_Tool::Curve( anEdge, aLoc, aFirst, aLast ); // Convert the curve to 2d CS - Handle(Geom2d_Curve) aCurve2d = CurveTo2D( aCurve, aFirst, aLast, aTransf ); + Handle(Geom2d_Curve) aCurve2d = CurveTo2D( aCurve, aFirst, aLast, aTransf, theUMin, theUMax ); if( !aCurve2d.IsNull() ) curves.push_back( aCurve2d ); } return curves; } + +bool CalcMidWidth( const std::vector& intersections, double& theMid, double& theWid ) +{ + double umin = std::numeric_limits::max(), + umax = -umin; + + size_t n = intersections.size(); + if( n <= 1 ) + return false; + + for( size_t i = 0; i < n; i++ ) + { + double u = intersections[i].X(); + if( uumax ) + umax = u; + } + theMid = ( umin+umax )/2; + theWid = umax-umin; + return true; +} + void HYDROData_DTM::ProfileDiscretization( const Handle_HYDROData_Profile& theProfile, - double theMinZ, double theMaxZ, double theDDZ, + double theXCurv, double theMinZ, double theMaxZ, double theDDZ, CurveUZ& theMidPointCurve, - CurveUZ& theWidthCurve ) + CurveUZ& theWidthCurve, + double theTolerance ) { - /*for( double z = theMinZ; z<=theMaxZ; z += theDDZ ) + double aDblMax = std::numeric_limits::max(), + aUMin = aDblMax, + aUMax = -aUMin, + aVMax = 1000000; + + gp_Vec2d aProfileDir; + std::vector curves = ProfileToParametric( theProfile, aUMin, aUMax, aProfileDir ); + size_t n = curves.size(); + + if( n==0 ) + return; + + // we add the "virtual" vertical lines to simulate the intersection with profile + gp_Pnt2d aFirst, aLast; + curves[0]->D0( curves[0]->FirstParameter(), aFirst ); + curves[n-1]->D0( curves[n-1]->LastParameter(), aLast ); + Handle(Geom2d_Line) aV1 = new Geom2d_Line( aFirst, gp_Dir2d( 0, 1 ) ); + Handle(Geom2d_TrimmedCurve) aT1 = new Geom2d_TrimmedCurve( aV1, 0.0, aVMax ); + + Handle(Geom2d_Line) aV2 = new Geom2d_Line( aLast, gp_Dir2d( 0, 1 ) ); + Handle(Geom2d_TrimmedCurve) aT2 = new Geom2d_TrimmedCurve( aV2, 0.0, aVMax ); + + curves.push_back( aT1 ); + curves.push_back( aT2 ); + + int psize = ( int )( ( theMaxZ-theMinZ ) / theDDZ + 1 ); + theMidPointCurve = CurveUZ( theXCurv, aProfileDir ); + theMidPointCurve.reserve( psize ); + theWidthCurve = CurveUZ( theXCurv, aProfileDir ); + theWidthCurve.reserve( psize ); + + n = curves.size(); + // for each discrete value of z we search intersection with profile + for( double z = theMinZ; z <= theMaxZ; z += theDDZ ) { + Handle(Geom2d_Line) aLine = new Geom2d_Line( gp_Pnt2d( 0, z ), gp_Dir2d( 1, 0 ) ); + std::vector intersections; + for( size_t i = 0; i < n; i++ ) + { + Handle_Geom2d_Curve aCurve = curves[i]; + Geom2dAPI_InterCurveCurve anIntersect( aCurve, aLine, theTolerance ); + for( int k=1, m=anIntersect.NbPoints(); k<=m; k++ ) + intersections.push_back( anIntersect.Point( k ) ); + } + + if( intersections.size() >= 2 ) + { + double u_mid, u_wid; + if( !CalcMidWidth( intersections, u_mid, u_wid ) ) + continue; + + PointUZ p_mid; + p_mid.U = u_mid; + p_mid.Z = z; + theMidPointCurve.push_back( p_mid ); - }*/ + PointUZ p_wid; + p_wid.U = u_wid; + p_wid.Z = z; + theWidthCurve.push_back( p_wid ); + } + } } void HYDROData_DTM::Interpolate( const CurveUZ& theCurveA, const CurveUZ& theCurveB, - int theNbSteps, std::vector& theInterpolation ) + int theNbSteps, std::vector& theInterpolation, + bool isAddSecond ) +{ + theInterpolation.clear(); + int d = isAddSecond ? 2 : 1; + theInterpolation.reserve( theNbSteps+d ); + double dt = 1.0 / double( theNbSteps + 1 ); + double t = dt; + theInterpolation.push_back( theCurveA ); + for( int i=0; i +void HYDROData_DTM::CurveTo3D( const Handle_Geom2d_BSplineCurve& theHydraulicAxis, + const CurveUZ& theMidCurve, const CurveUZ& theWidthCurve, + AltitudePoints& thePoints, + Bank* theLeft, Bank* theRight, double dz ) +{ + Geom2dAdaptor_Curve anAdaptor( theHydraulicAxis ); + TopoDS_Edge E2d = BRepLib_MakeEdge2d(theHydraulicAxis).Edge(); + GCPnts_AbscissaPoint ap( anAdaptor, theMidCurve.Xcurv(), anAdaptor.FirstParameter() ); + double aParam = ap.Parameter(); + + gp_Pnt2d point; + anAdaptor.D0( aParam, point ); + gp_Vec2d profile_dir = theMidCurve.ProfileDir(); + gp_Dir tangent_n( -profile_dir.Y(), profile_dir.X(), dz ); + profile_dir.Normalize(); + + size_t n = theMidCurve.size(); + double min_param = 1E+15; + double max_param = -1E+15; + double z1, z2; + for( size_t i=0; i max_param ) + { + max_param = param1; + z2 = z; + } + if( param2 > max_param ) + { + max_param = param2; + z2 = z; + } + + AltitudePoint p3d_1( p1.X(), p1.Y(), z ), p3d_2( p2.X(), p2.Y(), z ); + thePoints.push_back( p3d_1 ); + thePoints.push_back( p3d_2 ); + } + + if( theLeft ) + { + gp_Pnt2d left2d = point.Translated( min_param * profile_dir ); + gp_Pnt left( left2d.X(), left2d.Y(), z1 ); + theLeft->push_back( left, tangent_n ); + } + if( theRight ) + { + gp_Pnt2d right2d = point.Translated( max_param * profile_dir ); + gp_Pnt right( right2d.X(), right2d.Y(), z2 ); + theRight->push_back( right, tangent_n ); + } +} + +inline double max( double a, double b ) +{ + if( a>b ) + return a; + else + return b; +} + +#include + +HYDROData_Bathymetry::AltitudePoints HYDROData_DTM::Interpolate + ( const Handle_Geom2d_BSplineCurve& theHydraulicAxis, + const Handle_HYDROData_Profile& theProfileA, + double theXCurvA, + const Handle_HYDROData_Profile& theProfileB, + double theXCurvB, + double theDDZ, int theNbSteps, bool isAddSecond, + Bank* theLeft, Bank* theRight ) +{ + double zminA, zmaxA, zminB, zmaxB; + gp_Pnt lowestA, lowestB; + gp_Vec2d dirA, dirB; + + GetProperties( theProfileA, lowestA, dirA, false, zminA, zmaxA ); + GetProperties( theProfileB, lowestB, dirB, false, zminB, zmaxB ); + + double dz = zminB - zminA; + + double zmin = max( zminA, zminB ); + double zmax = max( zmaxA, zmaxB ); + + CurveUZ midA(0, gp_Vec2d()), midB(0, gp_Vec2d()); + CurveUZ widA(0, gp_Vec2d()), widB(0, gp_Vec2d()); + + ProfileDiscretization( theProfileA, theXCurvA, zmin, zmax, theDDZ, midA, widA ); + ProfileDiscretization( theProfileB, theXCurvB, zmin, zmax, theDDZ, midB, widB ); + + std::vector mid, wid; + Interpolate( midA, midB, theNbSteps, mid, isAddSecond ); + Interpolate( widA, widB, theNbSteps, wid, isAddSecond ); + + size_t p = mid.size(); + size_t q = p>0 ? 2*mid[0].size() : 1; + AltitudePoints points; + points.reserve( p*q ); + for( size_t i=0; imyPoints.size() - 1; i++) + WM.Add(BRepLib_MakeEdge(theLeft->myPoints[i], theLeft->myPoints[i+1]).Edge()); + TopoDS_Wire W = WM.Wire(); + return points; +} + +HYDROData_Bathymetry::AltitudePoints HYDROData_DTM::Interpolate + ( const std::vector& theProfiles, + double theDDZ, double theSpatialStep, + Bank* theLeft, + Bank* theRight ) { + AltitudePoints points; + size_t n = theProfiles.size(); + if( n<=1 ) + return points; + + std::vector distances; + Handle_Geom2d_BSplineCurve aHydraulicAxis = CreateHydraulicAxis( theProfiles, distances ); + if( aHydraulicAxis.IsNull() ) + return points; + + int aNbStepsComplete = 0; + for( size_t i=0, n1=n-1; ireserve( aNbStepsComplete ); + if( theRight ) + theRight->reserve( aNbStepsComplete ); + + for( size_t i=0, n1=n-1; i& theInterpolation ) +void HYDROData_DTM::CreateBankShapes( TopoDS_Edge& theLeft, TopoDS_Edge& theRight ) const { + theLeft = myLeft.createEdge3d(); + theRight = myRight.createEdge3d(); }