Salome HOME
4031dc82b2761fd4a9425a4163610ca9659061f0
[modules/hydro.git] / src / HYDROData / HYDROData_ChannelAltitude.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include "HYDROData_ChannelAltitude.h"
20
21 #include "HYDROData_Document.h"
22 #include "HYDROData_Object.h"
23 #include "HYDROData_Channel.h"
24 #include "HYDROData_Projection.h"
25 #include "HYDROData_Polyline3D.h"
26 #include "HYDROData_Profile.h"
27
28 #define _DEVDEBUG_
29 #include "HYDRO_trace.hxx"
30 #include <QString>
31
32 #include <BRepBuilderAPI_MakeVertex.hxx>
33 #include <BRepExtrema_DistShapeShape.hxx>
34
35 #include <Precision.hxx>
36 #include <TopAbs.hxx>
37 #include <TopExp.hxx>
38 #include <TopoDS.hxx>
39 #include <TopoDS_Shape.hxx>
40 #include <TopoDS_Edge.hxx>
41 #include <TopoDS_Face.hxx>
42 #include <TopoDS_Vertex.hxx>
43 #include <TopoDS_Wire.hxx>
44 #include <TopExp_Explorer.hxx>
45 #include <TopAbs_ShapeEnum.hxx>
46 #include <BRep_Tool.hxx>
47 #include <BRepTools.hxx>
48 #include <Geom_Curve.hxx>
49
50 #include <gp_Trsf.hxx>
51 #include <gp_Pnt.hxx>
52 #include <gp_Vec.hxx>
53
54 IMPLEMENT_STANDARD_HANDLE(HYDROData_ChannelAltitude, HYDROData_IAltitudeObject)
55 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_ChannelAltitude, HYDROData_IAltitudeObject)
56
57 HYDROData_ChannelAltitude::HYDROData_ChannelAltitude()
58 : HYDROData_IAltitudeObject()
59 {
60 }
61
62 HYDROData_ChannelAltitude::~HYDROData_ChannelAltitude()
63 {
64 }
65
66 double HYDROData_ChannelAltitude::GetAltitudeForPoint( const gp_XY& thePoint ) const
67 {
68   DEBTRACE("HYDROData_ChannelAltitude::GetAltitudeForPoint");
69   double aResAltitude = GetInvalidAltitude();
70
71   Handle(HYDROData_Channel) aChannel =
72   Handle(HYDROData_Channel)::DownCast(GetFatherObject());
73   if (aChannel.IsNull())
74     {
75       DEBTRACE("aChannel.IsNull()");
76       return aResAltitude;
77     }
78   DEBTRACE("aChannel: " << aChannel->GetName().toStdString());
79   Handle(HYDROData_Polyline3D) aGuideLine = aChannel->GetGuideLine();
80   if (aGuideLine.IsNull())
81     {
82       DEBTRACE("aGuideLine.IsNull()");
83       return aResAltitude;
84     }
85   DEBTRACE("aGuideLine: " << aGuideLine->GetName().toStdString());
86   Handle (HYDROData_Profile) aProfile = aChannel->GetProfile();
87   if (aProfile.IsNull())
88     {
89       return aResAltitude;
90     }
91   DEBTRACE("aProfile: " << aProfile->GetName().toStdString());
92
93   // --- See GEOMImpl_ProjectionDriver.cxx
94
95   TopoDS_Shape aShape = aGuideLine->GetShape3D();
96   double middleZ = -9999;
97   aGuideLine->GetMiddleZ(middleZ); // use the middle Z value of the 3d line to help the projection.
98   gp_Pnt P1(thePoint.X(), thePoint.Y(), middleZ);
99   TopoDS_Shape aPoint = BRepBuilderAPI_MakeVertex(P1).Shape();
100
101   if (aPoint.IsNull() || aShape.IsNull())
102     {
103       DEBTRACE("aPoint.IsNull() || aShape.IsNull()");
104       return aResAltitude;
105     }
106
107   if (aShape.ShapeType() != TopAbs_EDGE && aShape.ShapeType() != TopAbs_WIRE)
108     {
109       DEBTRACE("Projection aborted : the shape is neither an edge nor a wire");
110       return aResAltitude;
111     }
112
113   // Perform projection.
114   BRepExtrema_DistShapeShape aDistShSh(aPoint, aShape, Extrema_ExtFlag_MIN);
115
116   if (aDistShSh.IsDone() == Standard_False)
117     {
118       DEBTRACE("Projection not done");
119       return aResAltitude;
120     }
121
122   Standard_Boolean hasValidSolution = Standard_False;
123   Standard_Integer aNbSolutions = aDistShSh.NbSolution();
124   Standard_Integer i;
125   double aParam = 0.;
126   Standard_Real aTolConf = BRep_Tool::Tolerance(TopoDS::Vertex(aPoint));
127   Standard_Real aTolAng = 1.e-4;
128
129   for (i = 1; i <= aNbSolutions; i++)
130     {
131       Standard_Boolean isValid = Standard_False;
132       BRepExtrema_SupportType aSupportType = aDistShSh.SupportTypeShape2(i);
133       TopoDS_Shape aSupportShape = aDistShSh.SupportOnShape2(i);
134
135       if (aSupportType == BRepExtrema_IsOnEdge)
136         {
137           // Minimal distance inside edge is really a projection.
138           isValid = Standard_True;
139           aDistShSh.ParOnEdgeS2(i, aParam);
140         }
141       else if (aSupportType == BRepExtrema_IsVertex)
142         {
143           TopExp_Explorer anExp(aShape, TopAbs_EDGE);
144
145           if (aDistShSh.Value() <= aTolConf)
146             {
147               // The point lies on the shape. This means this point
148               // is really a projection.
149               for (; anExp.More() && !isValid; anExp.Next())
150                 {
151                   TopoDS_Edge aCurEdge = TopoDS::Edge(anExp.Current());
152
153                   if (aCurEdge.IsNull() == Standard_False)
154                     {
155                       TopoDS_Vertex aVtx[2];
156
157                       TopExp::Vertices(aCurEdge, aVtx[0], aVtx[1]);
158
159                       for (int j = 0; j < 2; j++)
160                         {
161                           if (aSupportShape.IsSame(aVtx[j]))
162                             {
163                               // The current edge is a projection edge.
164                               isValid = Standard_True;
165                               aSupportShape = aCurEdge;
166                               aParam = BRep_Tool::Parameter(aVtx[j], aCurEdge);
167                               break;
168                             }
169                         }
170                     }
171                 }
172             }
173           else
174             {
175               // Minimal distance to vertex is not always a real projection.
176               gp_Pnt aPnt = BRep_Tool::Pnt(TopoDS::Vertex(aPoint));
177               gp_Pnt aPrjPnt = BRep_Tool::Pnt(TopoDS::Vertex(aSupportShape));
178               gp_Vec aDProjP(aPrjPnt, aPnt);
179
180               for (; anExp.More() && !isValid; anExp.Next())
181                 {
182                   TopoDS_Edge aCurEdge = TopoDS::Edge(anExp.Current());
183
184                   if (aCurEdge.IsNull() == Standard_False)
185                     {
186                       TopoDS_Vertex aVtx[2];
187
188                       TopExp::Vertices(aCurEdge, aVtx[0], aVtx[1]);
189
190                       for (int j = 0; j < 2; j++)
191                         {
192                           if (aSupportShape.IsSame(aVtx[j]))
193                             {
194                               // Check if the point is a projection to the current edge.
195                               Standard_Real anEdgePars[2];
196                               Handle(Geom_Curve) aCurve = BRep_Tool::Curve(aCurEdge, anEdgePars[0], anEdgePars[1]);
197                               gp_Pnt aVal;
198                               gp_Vec aD1;
199
200                               aParam = BRep_Tool::Parameter(aVtx[j], aCurEdge);
201                               aCurve->D1(aParam, aVal, aD1);
202
203                               if (Abs(aD1.Dot(aDProjP)) <= aTolAng)
204                                 {
205                                   // The current edge is a projection edge.
206                                   isValid = Standard_True;
207                                   aSupportShape = aCurEdge;
208                                   break;
209                                 }
210                             }
211                         }
212                     }
213                 }
214             }
215         }
216
217       if (isValid)
218         {
219           if (hasValidSolution)
220             {
221               DEBTRACE("Projection aborted : multiple solutions");
222               return aResAltitude;
223             }
224
225           // Store the valid solution.
226           hasValidSolution = Standard_True;
227
228           // Normalize parameter.
229           TopoDS_Edge aSupportEdge = TopoDS::Edge(aSupportShape);
230           Standard_Real aF, aL;
231
232           BRep_Tool::Range(aSupportEdge, aF, aL);
233
234           if (Abs(aL - aF) <= aTolConf)
235             {
236               DEBTRACE("Projection aborted : degenerated projection edge");
237               return aResAltitude;
238             }
239
240           // Construct a projection vertex.
241           const gp_Pnt &aPntProj = aDistShSh.PointOnShape2(i);
242           TopoDS_Shape aProj = BRepBuilderAPI_MakeVertex(aPntProj).Shape();
243           DEBTRACE("projection: (" << aPntProj.X() << ", " << aPntProj.Y() << ", " << aPntProj.Z() << ")");
244           return aPntProj.Z() + 2.;
245         }
246     }
247
248   if (!hasValidSolution)
249     {
250       DEBTRACE("Projection aborted : no projection");
251       return aResAltitude;
252     }
253
254   return aResAltitude;
255 }
256
257
258
259