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