]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROData/HYDROData_ProfileUZ.cxx
Salome HOME
Merge remote-tracking branch 'origin/BR_IMPROVEMENTS' into BR_v14_rc
[modules/hydro.git] / src / HYDROData / HYDROData_ProfileUZ.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_ProfileUZ.h"
20
21 #include "HYDROData_Tool.h"
22
23 #include <gp_XY.hxx>
24 #include <gp_Pnt2d.hxx>
25
26 #include <TColStd_ListIteratorOfListOfReal.hxx>
27
28 #include <TDataStd_BooleanList.hxx>
29 #include <TDataStd_ExtStringList.hxx>
30 #include <TDataStd_IntegerList.hxx>
31 #include <TDataStd_RealList.hxx>
32
33 #include <TopoDS_Shape.hxx>
34
35
36 IMPLEMENT_STANDARD_HANDLE(HYDROData_ProfileUZ, HYDROData_IPolyline)
37 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_ProfileUZ, HYDROData_IPolyline)
38
39 HYDROData_ProfileUZ::HYDROData_ProfileUZ()
40 : HYDROData_IPolyline()
41 {
42 }
43
44 HYDROData_ProfileUZ::~HYDROData_ProfileUZ()
45 {
46 }
47
48 TopoDS_Shape HYDROData_ProfileUZ::GetShape() const
49 {
50   return TopoDS_Shape();
51 }
52
53 double HYDROData_ProfileUZ::GetDepthFromDistance( const PointsList& thePoints,
54                                                   const double&     theDistance )
55 {
56   double aResDepth = 0.0;
57
58   int aNbPoints = thePoints.Size();
59   if ( aNbPoints < 2 )
60     return aResDepth;
61
62   double aCompDist = 0.0;
63   HYDROData_IPolyline::Point aPrevPoint = thePoints.First();
64   for ( int i = 2; i <= aNbPoints; ++i )
65   {
66     const Point& aCurPoint = thePoints.Value( i );
67
68     double aPntDist = gp_Pnt2d( aPrevPoint.X(), 0 ).Distance( gp_Pnt2d( aCurPoint.X(), 0 ) );
69
70     aCompDist += aPntDist;
71
72     if ( theDistance < aCompDist )
73     {
74       double aComPntDist = gp_Pnt2d( thePoints.First().X(), 0 ).Distance( gp_Pnt2d( aPrevPoint.X(), 0 ) );
75
76       double aFindDist = theDistance - aComPntDist;
77       double aRatio = aFindDist / ( aPntDist - aFindDist );
78
79       aResDepth = ( aPrevPoint.Y() + aRatio * aCurPoint.Y() ) / ( 1 + aRatio );
80       break;
81     }
82
83     aPrevPoint = aCurPoint;
84   }
85
86   return aResDepth;
87 }
88
89 int HYDROData_ProfileUZ::NbSections() const
90 {
91   return 1;
92 }
93
94 void HYDROData_ProfileUZ::AddSection( const TCollection_AsciiString& /*theSectName*/,
95                                       const SectionType              /*theSectionType*/,
96                                       const bool                     /*theIsClosed*/ )
97 {
98 }
99
100 TCollection_AsciiString HYDROData_ProfileUZ::GetSectionName( const int /*theSectionIndex*/ ) const
101 {
102   return "Section_1";
103 }
104
105 void HYDROData_ProfileUZ::SetSectionName( const int                      /*theSectionIndex*/, 
106                                           const TCollection_AsciiString& /*theSectionName*/ )
107 {
108 }
109
110 HYDROData_ProfileUZ::SectionType HYDROData_ProfileUZ::GetSectionType( const int /*theSectionIndex*/ ) const
111 {
112   Handle(TDataStd_ExtStringList) aNamesList;
113   Handle(TDataStd_IntegerList) aTypesList;
114   Handle(TDataStd_BooleanList) aClosuresList;
115   getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
116   if ( aTypesList.IsNull() || aTypesList->IsEmpty() )
117     return SECTION_POLYLINE;
118
119   return (SectionType)aTypesList->First();
120 }
121
122 void HYDROData_ProfileUZ::SetSectionType( const int         /*theSectionIndex*/, 
123                                           const SectionType theSectionType )
124 {
125   Handle(TDataStd_ExtStringList) aNamesList;
126   Handle(TDataStd_IntegerList) aTypesList;
127   Handle(TDataStd_BooleanList) aClosuresList;
128   getSectionsLists( aNamesList, aTypesList, aClosuresList );
129   if ( aTypesList.IsNull()  )
130     return;
131
132   // Refill the existing list
133   aTypesList->Clear();
134   aTypesList->Append( theSectionType );
135 }
136
137 bool HYDROData_ProfileUZ::IsClosedSection( const int /*theSectionIndex*/ ) const
138 {
139   return false;
140 }
141
142 void HYDROData_ProfileUZ::SetSectionClosed( const int  /*theSectionIndex*/, 
143                                             const bool /*theIsClosed*/ )
144 {
145 }
146
147 void HYDROData_ProfileUZ::RemoveSection( const int /*theSectionIndex*/ )
148 {
149   RemoveSections();
150 }
151
152 void HYDROData_ProfileUZ::RemoveSections()
153 {
154   removePointsLists( 0 );
155 }
156
157 void HYDROData_ProfileUZ::AddPoint( const int    /*theSectionIndex*/,
158                                     const Point& thePoint,
159                                     const int    thePointIndex )
160 {
161   double aNewCoordU = thePoint.X();
162   double aNewCoordZ = thePoint.Y();
163
164   Handle(TDataStd_RealList) aListU, aListZ;
165   getPointsLists( 0, aListU, aListZ );
166
167   if ( aListU->IsEmpty() || aNewCoordU > aListU->Last() )
168   {
169     aListU->Append( aNewCoordU );
170     aListZ->Append( aNewCoordZ );
171     return;
172   }
173   else if ( aNewCoordU < aListU->First() )
174   {
175     aListU->Prepend( aNewCoordU );
176     aListZ->Prepend( aNewCoordZ );
177     return;
178   }
179
180   TColStd_ListOfReal anOldListU;
181   anOldListU = aListU->List();
182
183   TColStd_ListOfReal anOldListZ;
184   anOldListZ = aListZ->List();
185
186   // Crsat new lists
187   removePointsLists( 0 );
188   getPointsLists( 0, aListU, aListZ );
189
190   bool anIsInserted = false;
191   TColStd_ListIteratorOfListOfReal anIterU( anOldListU );
192   TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ );
193   for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() )
194   {
195     double aCoordU = anIterU.Value();
196     double aCoordZ = anIterZ.Value();
197
198     if ( !anIsInserted )
199     {
200       if ( ValuesEquals( aNewCoordU, aCoordU ) )
201       {
202         // Just update Z value
203         aCoordZ = aNewCoordZ;
204         anIsInserted = true;
205       }
206       else if ( aNewCoordU < aCoordU )
207       {
208         // Insert new point
209         aListU->Append( aNewCoordU );
210         aListZ->Append( aNewCoordZ );
211         anIsInserted = true;
212       }
213     }
214
215     aListU->Append( aCoordU );
216     aListZ->Append( aCoordZ );
217   }
218 }
219
220 void HYDROData_ProfileUZ::SetPoint( const int    theSectionIndex,
221                                     const Point& thePoint,
222                                     const int    /*thePointIndex*/ )
223 {
224   AddPoint( theSectionIndex, thePoint );
225 }
226
227 void HYDROData_ProfileUZ::RemovePoint( const int /*theSectionIndex*/,
228                                        const int thePointIndex )
229 {
230   Handle(TDataStd_RealList) aListU, aListZ;
231   getPointsLists( 0, aListU, aListZ, false );
232   if ( aListU.IsNull() || aListZ.IsNull() || aListU->IsEmpty() )
233     return;
234
235   TColStd_ListOfReal anOldListU;
236   anOldListU = aListU->List();
237
238   TColStd_ListOfReal anOldListZ;
239   anOldListZ = aListZ->List();
240
241   // Creat new lists
242   removePointsLists( 0 );
243   getPointsLists( 0, aListU, aListZ );
244
245   bool anIsInserted = false;
246   TColStd_ListIteratorOfListOfReal anIterU( anOldListU );
247   TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ );
248   for ( int i = 0; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next(), ++i )
249   {
250     if ( i == thePointIndex )
251       continue; // skip index to remove
252
253     aListU->Append( anIterU.Value() );
254     aListZ->Append( anIterZ.Value() );
255   }
256 }
257
258 HYDROData_ProfileUZ::PointsList HYDROData_ProfileUZ::GetPoints( const int /*theSectionIndex*/ ) const
259 {
260   PointsList aResList;
261
262   Handle(TDataStd_RealList) aListU, aListZ;
263   getPointsLists( 0, aListU, aListZ, false );
264   if ( aListU.IsNull() || aListZ.IsNull() )
265     return aResList;
266
267   TColStd_ListIteratorOfListOfReal anIterU( aListU->List() );
268   TColStd_ListIteratorOfListOfReal anIterZ( aListZ->List() );
269   for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() )
270   {
271     Point aPoint( anIterU.Value(), anIterZ.Value() );
272     aResList.Append( aPoint );
273   }
274
275   return aResList;
276 }
277
278