2 #include <HYDROData_DTM.h>
3 #include <HYDROData_Profile.h>
5 #include <Geom2d_BSplineCurve.hxx>
6 #include <Geom2dAPI_Interpolate.hxx>
7 #include <TColgp_HArray1OfPnt2d.hxx>
8 #include <TColgp_Array1OfVec2d.hxx>
9 #include <TColStd_HArray1OfBoolean.hxx>
11 #include <TopoDS_Edge.hxx>
12 #include <TopoDS_Wire.hxx>
13 #include <TopExp_Explorer.hxx>
14 #include <BRep_Tool.hxx>
16 #include <Geom_Line.hxx>
17 #include <Geom2d_Line.hxx>
18 #include <Geom2d_TrimmedCurve.hxx>
19 #include <Geom_BSplineCurve.hxx>
20 #include <Geom2d_BSplineCurve.hxx>
21 #include <GeomAPI_Interpolate.hxx>
22 #include <TColStd_Array1OfReal.hxx>
23 #include <TColStd_Array1OfInteger.hxx>
24 #include <TColgp_Array1OfPnt.hxx>
25 #include <TColgp_Array1OfVec.hxx>
26 #include <TColgp_HArray1OfPnt.hxx>
27 #include <Geom2dAPI_InterCurveCurve.hxx>
28 #include <Geom2dAPI_ProjectPointOnCurve.hxx>
29 #include <Geom2dAdaptor_Curve.hxx>
30 #include <GCPnts_AbscissaPoint.hxx>
31 #include <BRepBuilderAPI_MakeEdge.hxx>
34 IMPLEMENT_STANDARD_HANDLE( HYDROData_DTM, HYDROData_Bathymetry )
35 IMPLEMENT_STANDARD_RTTIEXT( HYDROData_DTM, HYDROData_Bathymetry )
38 HYDROData_DTM::CurveUZ::CurveUZ( double theXCurv )
43 HYDROData_DTM::CurveUZ::~CurveUZ()
47 double HYDROData_DTM::CurveUZ::Xcurv() const
52 HYDROData_DTM::CurveUZ HYDROData_DTM::CurveUZ::operator + ( const CurveUZ& c ) const
54 HYDROData_DTM::CurveUZ res( Xcurv() + c.Xcurv() );
57 for( int i=0; i<n; i++ )
60 p.U = operator[]( i ).U + c[i].U;
61 p.Z = operator[]( i ).Z;
67 HYDROData_DTM::CurveUZ HYDROData_DTM::CurveUZ::operator * ( double d ) const
69 HYDROData_DTM::CurveUZ res( Xcurv()*d );
72 for( int i=0; i<n; i++ )
75 p.U = operator[]( i ).U * d;
76 p.Z = operator[]( i ).Z;
82 void HYDROData_DTM::Bank::reserve( int theNbPoints )
84 myPoints.reserve( theNbPoints );
85 myDirs.reserve( theNbPoints );
88 void HYDROData_DTM::Bank::push_back( const gp_Pnt& thePnt, const gp_Dir& theTangent )
90 myPoints.push_back( thePnt );
91 myDirs.push_back( theTangent );
94 void HYDROData_DTM::Bank::clear()
100 TopoDS_Edge HYDROData_DTM::Bank::createEdge3d() const
102 size_t n = myPoints.size();
104 return TopoDS_Edge();
106 Handle_Geom_BSplineCurve aCurve;
108 Handle(TColgp_HArray1OfPnt) points = new TColgp_HArray1OfPnt( 1, (int)n );
109 TColgp_Array1OfVec tangents( 1, (int)n );
110 Handle(TColStd_HArray1OfBoolean) flags = new TColStd_HArray1OfBoolean( 1, (int)n );
112 for( size_t i = 1; i <= n; i++ )
114 gp_Pnt aPnt = myPoints[i-1];
115 gp_Vec aVec = myDirs[i-1];
116 points->SetValue( (int)i, aPnt );
117 tangents.SetValue( (int)i, aVec );
118 flags->SetValue( (int)i, Standard_True );
121 GeomAPI_Interpolate anInterpolator( points, Standard_False, Standard_False );
122 anInterpolator.Load( tangents, flags );
123 anInterpolator.Perform();
124 if( anInterpolator.IsDone() )
126 aCurve = anInterpolator.Curve();
127 return BRepBuilderAPI_MakeEdge( aCurve ).Edge();
130 return TopoDS_Edge();
135 HYDROData_DTM::HYDROData_DTM()
139 HYDROData_DTM::~HYDROData_DTM()
143 HYDROData_SequenceOfObjects HYDROData_DTM::GetProfiles() const
145 return GetReferenceObjects( DataTag_Profiles );
148 void HYDROData_DTM::SetProfiles( const HYDROData_SequenceOfObjects& theProfiles )
150 SetReferenceObjects( theProfiles, DataTag_Profiles );
153 double HYDROData_DTM::GetDDZ() const
155 return GetDouble( DataTag_DDZ );
158 void HYDROData_DTM::SetDDZ( double theDDZ )
160 SetDouble( DataTag_DDZ, theDDZ );
163 double HYDROData_DTM::GetSpatialStep() const
165 return GetDouble( DataTag_SpatialStep );
168 void HYDROData_DTM::SetSpatialStep( double theSpatialStep )
170 SetDouble( DataTag_SpatialStep, theSpatialStep );
173 void HYDROData_DTM::Update()
175 HYDROData_SequenceOfObjects objs = GetProfiles();
176 int aLower = objs.Lower(), anUpper = objs.Upper();
177 size_t n = anUpper-aLower+1;
179 std::vector<Handle_HYDROData_Profile> profiles;
180 profiles.reserve( n );
181 for( int i=aLower; i<=anUpper; i++ )
183 Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( objs.Value( i ) );
184 if( !aProfile.IsNull() )
185 profiles.push_back( aProfile );
188 double ddz = GetDDZ();
189 double step = GetSpatialStep();
190 const double EPS = 1E-3;
191 AltitudePoints points;
195 if( ddz>EPS && step>EPS )
196 points = Interpolate( profiles, ddz, step, &myLeft, &myRight );
197 SetAltitudePoints( points );
209 void HYDROData_DTM::GetProperties( const Handle_HYDROData_Profile& theProfile,
210 gp_Pnt& theLowestPoint, gp_Vec2d& theDir,
212 double& theZMin, double& theZMax )
214 theLowestPoint = theProfile->GetBottomPoint();
217 theProfile->GetLeftPoint( aLeft, true, true );
218 theProfile->GetRightPoint( aRight, true, true );
219 double x = aRight.X()-aLeft.X();
220 double y = aRight.Y()-aLeft.Y();
222 theDir = gp_Vec2d( -y, x );
224 theDir = gp_Vec2d( x, y );
226 HYDROData_Profile::ProfilePoints points = theProfile->GetProfilePoints();
227 int lo = points.Lower();
228 int up = points.Upper();
229 theZMin = std::numeric_limits<double>::max();
231 for( int i=lo; i<=up; i++ )
233 double z = points.Value( i ).Z();
241 inline gp_Pnt2d To2D( const gp_Pnt& thePnt, const gp_Trsf& theTr,
242 double& theUMin, double& theUMax )
244 gp_Pnt p = thePnt.Transformed( theTr );
251 return gp_Pnt2d( u, z );
254 Handle(TColgp_HArray1OfPnt2d) To2D( const TColgp_Array1OfPnt& thePoints,
255 const gp_Trsf& theTr,
256 double& theUMin, double& theUMax )
258 int low = thePoints.Lower(), up = thePoints.Upper();
259 Handle(TColgp_HArray1OfPnt2d) points = new TColgp_HArray1OfPnt2d( low, up );
260 for( int i=low; i<=up; i++ )
261 points->SetValue( i, To2D( thePoints.Value( i ), theTr, theUMin, theUMax ) );
265 Handle(Geom2d_Curve) CurveTo2D( const Handle(Geom_Curve)& theCurve,
266 Standard_Real theFirst, Standard_Real theLast,
267 const gp_Trsf& theTr,
268 double& theUMin, double& theUMax )
270 if( theCurve->IsKind( STANDARD_TYPE( Geom_Line ) ) )
272 gp_Pnt aFirstPnt, aLastPnt;
273 theCurve->D0( theFirst, aFirstPnt );
274 theCurve->D0( theLast, aLastPnt );
277 aFirst2d = To2D( aFirstPnt, theTr, theUMin, theUMax ),
278 aLast2d = To2D( aLastPnt, theTr, theUMin, theUMax );
280 gp_Vec2d dir( aFirst2d, aLast2d );
281 Handle_Geom2d_Line aLine2d = new Geom2d_Line( aFirst2d, gp_Dir2d( dir.X(), dir.Y() ) );
282 return new Geom2d_TrimmedCurve( aLine2d, 0, aLast2d.Distance( aFirst2d ) );
285 if( theCurve->IsKind( STANDARD_TYPE( Geom_BSplineCurve ) ) )
287 Handle(Geom_BSplineCurve) aSpline = Handle(Geom_BSplineCurve)::DownCast( theCurve );
289 Handle(TColgp_HArray1OfPnt2d) poles = To2D( aSpline->Poles(), theTr, theUMin, theUMax );
290 const TColStd_Array1OfReal& knots = aSpline->Knots();
291 const TColStd_Array1OfInteger& multiplicities = aSpline->Multiplicities();
292 int aDegree = aSpline->Degree();
294 return new Geom2d_BSplineCurve( poles->Array1(), knots, multiplicities, aDegree );
297 return Handle(Geom2d_Curve)();
300 Handle_Geom2d_BSplineCurve HYDROData_DTM::CreateHydraulicAxis(
301 const std::vector<Handle_HYDROData_Profile>& theProfiles,
302 std::vector<double>& theDistances )
304 size_t n = theProfiles.size();
305 Handle_Geom2d_BSplineCurve aResult;
307 Handle(TColgp_HArray1OfPnt2d) points = new TColgp_HArray1OfPnt2d( 1, (int)n );
308 TColgp_Array1OfVec2d tangents( 1, (int)n );
309 Handle(TColStd_HArray1OfBoolean) flags = new TColStd_HArray1OfBoolean( 1, (int)n );
311 for( size_t i = 1; i <= n; i++ )
313 Handle_HYDROData_Profile aProfile = theProfiles[i-1];
319 GetProperties( aProfile, aLowest, aTangent, true, zmin, zmax );
320 aTangent.Normalize();
322 points->SetValue( (int)i, gp_Pnt2d( aLowest.X(), aLowest.Y() ) );
323 tangents.SetValue( (int)i, aTangent );
324 flags->SetValue( (int)i, Standard_True );
327 Geom2dAPI_Interpolate anInterpolator( points, Standard_False, Standard_False );
328 anInterpolator.Load( tangents, flags );
329 anInterpolator.Perform();
330 if( anInterpolator.IsDone() )
332 aResult = anInterpolator.Curve();
334 //fill the distances vector
335 Geom2dAdaptor_Curve anAdaptor( aResult );
337 theDistances.clear();
338 theDistances.reserve( n );
339 Standard_Real aParamFirst = anAdaptor.FirstParameter(), aParamLast = anAdaptor.LastParameter();
340 for( size_t i = 1; i <= n; i++ )
342 gp_Pnt2d aPnt = points->Value( (Standard_Integer)i );
343 Geom2dAPI_ProjectPointOnCurve aProject( aPnt, aResult );
344 Standard_Real aParam = aProject.LowerDistanceParameter();
345 double aDistance = GCPnts_AbscissaPoint::Length( anAdaptor, aParamFirst, aParam );
346 theDistances.push_back( aDistance );
352 std::vector<Handle_Geom2d_Curve> HYDROData_DTM::ProfileToParametric(
353 const Handle_HYDROData_Profile& theProfile,
354 double& theUMin, double& theUMax )
356 std::vector<Handle_Geom2d_Curve> curves;
358 // Transformation of the coordinate systems
362 GetProperties( theProfile, aLowest, aDir, false, zmin, zmax );
364 gp_Ax3 aStd3d( gp_Pnt( 0, 0, 0 ), gp_Dir( 0, 0, 1 ), gp_Dir( 1, 0, 0 ) );
365 gp_Ax3 aLocal( aLowest, gp_Dir( 0, 0, 1 ), gp_Dir( aDir.X(), aDir.Y(), 0 ) );
368 aTransf.SetTransformation( aStd3d, aLocal );
370 // Iteration via edges
371 TopoDS_Wire aWire = TopoDS::Wire( theProfile->GetShape3D() );
372 TopExp_Explorer anExp( aWire, TopAbs_EDGE );
373 for( ; anExp.More(); anExp.Next() )
375 // Extract an edge from wire
376 TopoDS_Edge anEdge = TopoDS::Edge( anExp.Current() );
378 // Extract a curve corresponding to the edge
379 TopLoc_Location aLoc;
380 Standard_Real aFirst, aLast;
381 Handle(Geom_Curve) aCurve = BRep_Tool::Curve( anEdge, aLoc, aFirst, aLast );
383 // Convert the curve to 2d CS
384 Handle(Geom2d_Curve) aCurve2d = CurveTo2D( aCurve, aFirst, aLast, aTransf, theUMin, theUMax );
385 if( !aCurve2d.IsNull() )
386 curves.push_back( aCurve2d );
392 double CalcGC( const std::vector<gp_Pnt2d>& intersections )
395 size_t n = intersections.size();
396 for( size_t i = 0; i < n; i++ )
398 u += intersections[i].X();
404 double CalcWidth( const std::vector<gp_Pnt2d>& intersections )
406 double umin = std::numeric_limits<double>::max(),
409 size_t n = intersections.size();
413 for( size_t i = 0; i < n; i++ )
415 double u = intersections[i].X();
424 void HYDROData_DTM::ProfileDiscretization( const Handle_HYDROData_Profile& theProfile,
425 double theXCurv, double theMinZ, double theMaxZ, double theDDZ,
426 CurveUZ& theMidPointCurve,
427 CurveUZ& theWidthCurve,
428 double theTolerance )
430 double aDblMax = std::numeric_limits<double>::max(),
435 std::vector<Handle_Geom2d_Curve> curves = ProfileToParametric( theProfile, aUMin, aUMax );
436 size_t n = curves.size();
441 // we add the "virtual" vertical lines to simulate the intersection with profile
442 gp_Pnt2d aFirst, aLast;
443 curves[0]->D0( curves[0]->FirstParameter(), aFirst );
444 curves[n-1]->D0( curves[n-1]->LastParameter(), aLast );
445 Handle(Geom2d_Line) aV1 = new Geom2d_Line( aFirst, gp_Dir2d( 0, 1 ) );
446 Handle(Geom2d_TrimmedCurve) aT1 = new Geom2d_TrimmedCurve( aV1, 0.0, aVMax );
448 Handle(Geom2d_Line) aV2 = new Geom2d_Line( aLast, gp_Dir2d( 0, 1 ) );
449 Handle(Geom2d_TrimmedCurve) aT2 = new Geom2d_TrimmedCurve( aV2, 0.0, aVMax );
451 curves.push_back( aT1 );
452 curves.push_back( aT2 );
454 int psize = ( int )( ( theMaxZ-theMinZ ) / theDDZ + 1 );
455 theMidPointCurve = CurveUZ( theXCurv );
456 theMidPointCurve.reserve( psize );
457 theWidthCurve = CurveUZ( theXCurv );
458 theWidthCurve.reserve( psize );
461 // for each discrete value of z we search intersection with profile
462 for( double z = theMinZ; z <= theMaxZ; z += theDDZ )
464 Handle(Geom2d_Line) aLine = new Geom2d_Line( gp_Pnt2d( 0, z ), gp_Dir2d( 1, 0 ) );
465 std::vector<gp_Pnt2d> intersections;
466 for( size_t i = 0; i < n; i++ )
468 Handle_Geom2d_Curve aCurve = curves[i];
469 Geom2dAPI_InterCurveCurve anIntersect( aCurve, aLine, theTolerance );
470 for( int k=1, m=anIntersect.NbPoints(); k<=m; k++ )
471 intersections.push_back( anIntersect.Point( k ) );
474 if( intersections.size() >= 2 )
476 double u_mid = CalcGC( intersections );
480 theMidPointCurve.push_back( p_mid );
482 double u_wid = CalcWidth( intersections );
486 theWidthCurve.push_back( p_wid );
491 void HYDROData_DTM::Interpolate( const CurveUZ& theCurveA, const CurveUZ& theCurveB,
492 int theNbSteps, std::vector<CurveUZ>& theInterpolation,
495 theInterpolation.clear();
496 int d = isAddSecond ? 2 : 1;
497 theInterpolation.reserve( theNbSteps+d );
498 double dt = 1.0 / double( theNbSteps + 1 );
500 theInterpolation.push_back( theCurveA );
501 for( int i=0; i<theNbSteps; i++, t+=dt )
503 CurveUZ anInterp = theCurveA*(1-t) + theCurveB*t;
504 theInterpolation.push_back( anInterp );
507 theInterpolation.push_back( theCurveB );
510 void HYDROData_DTM::CurveTo3D( const Handle_Geom2d_BSplineCurve& theHydraulicAxis,
511 const CurveUZ& theMidCurve, const CurveUZ& theWidthCurve,
512 AltitudePoints& thePoints,
513 Bank* theLeft, Bank* theRight, double dz )
515 Geom2dAdaptor_Curve anAdaptor( theHydraulicAxis );
516 GCPnts_AbscissaPoint ap( anAdaptor, theMidCurve.Xcurv(), anAdaptor.FirstParameter() );
517 double aParam = ap.Parameter();
520 gp_Vec2d tangent, profile_dir;
521 anAdaptor.D1( aParam, point, tangent );
522 profile_dir.SetCoord( tangent.Y(), -tangent.X() );
523 profile_dir.Normalize();
524 gp_Dir tangent_n( tangent.X(), tangent.Y(), dz );
526 size_t n = theMidCurve.size();
527 double min_param = 1E+15;
528 double max_param = -1E+15;
530 for( size_t i=0; i<n; i++ )
532 double param1 = theMidCurve[i].U - theWidthCurve[i].U / 2;
533 double param2 = theMidCurve[i].U + theWidthCurve[i].U / 2;
535 gp_Pnt2d p1 = point.Translated( param1 * profile_dir / 2 );
536 gp_Pnt2d p2 = point.Translated( param2 * profile_dir / 2 );
538 double z = theMidCurve[i].Z;
540 if( param1 < min_param )
545 if( param2 < min_param )
550 if( param1 > max_param )
555 if( param2 > max_param )
561 AltitudePoint p3d_1( p1.X(), p1.Y(), z ), p3d_2( p2.X(), p2.Y(), z );
562 thePoints.push_back( p3d_1 );
563 thePoints.push_back( p3d_2 );
568 gp_Pnt2d left2d = point.Translated( min_param * profile_dir / 2 );
569 gp_Pnt left( left2d.X(), left2d.Y(), z1 );
570 theLeft->push_back( left, tangent_n );
574 gp_Pnt2d right2d = point.Translated( max_param * profile_dir / 2 );
575 gp_Pnt right( right2d.X(), right2d.Y(), z2 );
576 theRight->push_back( right, tangent_n );
580 inline double max( double a, double b )
588 HYDROData_Bathymetry::AltitudePoints HYDROData_DTM::Interpolate
589 ( const Handle_Geom2d_BSplineCurve& theHydraulicAxis,
590 const Handle_HYDROData_Profile& theProfileA,
592 const Handle_HYDROData_Profile& theProfileB,
594 double theDDZ, int theNbSteps, bool isAddSecond,
595 Bank* theLeft, Bank* theRight )
597 double zminA, zmaxA, zminB, zmaxB;
598 gp_Pnt lowestA, lowestB;
601 GetProperties( theProfileA, lowestA, dirA, false, zminA, zmaxA );
602 GetProperties( theProfileB, lowestB, dirB, false, zminB, zmaxB );
604 double dz = zminB - zminA;
606 double zmin = max( zminA, zminB );
607 double zmax = max( zmaxA, zmaxB );
609 CurveUZ midA(0), midB(0);
610 CurveUZ widA(0), widB(0);
612 ProfileDiscretization( theProfileA, theXCurvA, zmin, zmax, theDDZ, midA, widA );
613 ProfileDiscretization( theProfileB, theXCurvB, zmin, zmax, theDDZ, midB, widB );
615 std::vector<CurveUZ> mid, wid;
616 Interpolate( midA, midB, theNbSteps, mid, isAddSecond );
617 Interpolate( widA, widB, theNbSteps, wid, isAddSecond );
619 size_t p = mid.size();
620 size_t q = p>0 ? 2*mid[0].size() : 1;
621 AltitudePoints points;
622 points.reserve( p*q );
623 for( size_t i=0; i<p; i++ )
624 CurveTo3D( theHydraulicAxis, mid[i], wid[i], points, theLeft, theRight, dz );
629 HYDROData_Bathymetry::AltitudePoints HYDROData_DTM::Interpolate
630 ( const std::vector<Handle_HYDROData_Profile>& theProfiles,
631 double theDDZ, double theSpatialStep,
635 AltitudePoints points;
636 size_t n = theProfiles.size();
640 std::vector<double> distances;
641 Handle_Geom2d_BSplineCurve aHydraulicAxis = CreateHydraulicAxis( theProfiles, distances );
642 if( aHydraulicAxis.IsNull() )
645 int aNbStepsComplete = 0;
646 for( size_t i=0, n1=n-1; i<n1; i++ )
648 double aDistance = distances[i+1]-distances[i];
649 aNbStepsComplete += ( int(aDistance/theSpatialStep) + 1 );
652 theLeft->reserve( aNbStepsComplete );
654 theRight->reserve( aNbStepsComplete );
656 for( size_t i=0, n1=n-1; i<n1; i++ )
658 double aDistance = distances[i+1]-distances[i];
659 int aNbSteps = int(aDistance/theSpatialStep) + 1;
660 bool isAddSecond = i==n1-1;
662 AltitudePoints local_points = Interpolate( aHydraulicAxis, theProfiles[i], distances[i],
663 theProfiles[i+1], distances[i+1], theDDZ, aNbSteps, isAddSecond, theLeft, theRight );
666 points.reserve( local_points.size() * ( n-1 ) );
668 for( size_t j=0, m=local_points.size(); j<m; j++ )
669 points.push_back( local_points[j] );
674 void HYDROData_DTM::CreateBankShapes( TopoDS_Edge& theLeft, TopoDS_Edge& theRight ) const
676 theLeft = myLeft.createEdge3d();
677 theRight = myRight.createEdge3d();