]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROData/HYDROData_Profile.cxx
Salome HOME
Methods renamed.
[modules/hydro.git] / src / HYDROData / HYDROData_Profile.cxx
1
2 #include "HYDROData_Profile.h"
3
4 #include "HYDROData_Document.h"
5 #include "HYDROData_Iterator.h"
6 #include "HYDROData_Tool.h"
7
8 #include <BRepBuilderAPI_MakeEdge.hxx>
9 #include <BRepBuilderAPI_MakeWire.hxx>
10
11 #include <gp_XY.hxx>
12 #include <gp_XYZ.hxx>
13 #include <gp_Pnt2d.hxx>
14
15 #include <TDataStd_AsciiString.hxx>
16 #include <TDataStd_RealArray.hxx>
17
18 #include <TopoDS_Edge.hxx>
19 #include <TopoDS_Wire.hxx>
20
21 #include <OSD_File.hxx>
22 #include <OSD_Protection.hxx>
23
24 #include <QColor>
25 #include <QStringList>
26
27 #define PYTHON_PROFILE_ID "KIND_PROFILE"
28
29 IMPLEMENT_STANDARD_HANDLE(HYDROData_Profile, HYDROData_Object)
30 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Profile, HYDROData_Object)
31
32 HYDROData_Profile::HYDROData_Profile()
33 : HYDROData_Object()
34 {
35 }
36
37 HYDROData_Profile::~HYDROData_Profile()
38 {
39 }
40
41 QStringList HYDROData_Profile::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
42 {
43   QStringList aResList;
44
45   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
46   if ( aDocument.IsNull() )
47     return aResList;
48                              
49   QString aDocName = aDocument->GetDocPyName();
50   QString aProfileName = GetName();
51
52   aResList << QString( "%1 = %2.CreateObject( %3 );" )
53               .arg( aProfileName ).arg( aDocName ).arg( PYTHON_PROFILE_ID );
54   aResList << QString( "%1.SetName( \"%1\" );" ).arg( aProfileName );
55
56   return aResList;
57 }
58
59 TopoDS_Shape HYDROData_Profile::GetTopShape() const
60 {
61   TopoDS_Wire aWire;
62
63   gp_XY aFirstPoint, aLastPoint;
64   if ( !GetLeftPoint( aFirstPoint ) || !GetRightPoint( aLastPoint ) )
65     return aWire;
66
67   gp_Pnt aPnt1( aFirstPoint.X(), aFirstPoint.Y(), 0 );
68   gp_Pnt aPnt2( aLastPoint.X(),  aLastPoint.Y(),  0 );
69
70   BRepBuilderAPI_MakeEdge aMakeEdge( aPnt1, aPnt2 );
71   TopoDS_Edge anEdge = aMakeEdge;
72
73   BRepBuilderAPI_MakeWire aMakeWire( anEdge );
74   aWire = aMakeWire;
75
76   return aWire;
77 }
78
79 TopoDS_Shape HYDROData_Profile::GetShape3D() const
80 {
81   return getShape3D();
82 }
83
84 void HYDROData_Profile::Update()
85 {
86   HYDROData_Object::Update();
87
88   BRepBuilderAPI_MakeWire aMakeWire;
89
90   ProfilePoints aProfilePoints = GetProfilePoints();
91   for ( int i = 1, n = aProfilePoints.Length(); i < n ; ++i )
92   {
93     ProfilePoint aFirstPoint = aProfilePoints.Value( i );
94     ProfilePoint aSecPoint = aProfilePoints.Value( i + 1 );
95
96     gp_Pnt aPnt1( aFirstPoint.X(), aFirstPoint.Y(), aFirstPoint.Z() );
97     gp_Pnt aPnt2( aSecPoint.X(),   aSecPoint.Y(),   aSecPoint.Z()   );
98
99     BRepBuilderAPI_MakeEdge aMakeEdge( aPnt1, aPnt2 );
100     TopoDS_Edge anEdge = aMakeEdge;
101
102     aMakeWire.Add( anEdge );
103   }
104
105   TopoDS_Wire aWire;
106   if ( aMakeWire.IsDone() )
107     aWire = aMakeWire;
108
109   SetShape3D( aWire );
110 }
111
112 QColor HYDROData_Profile::DefaultFillingColor()
113 {
114   return QColor( Qt::transparent );
115 }
116
117 QColor HYDROData_Profile::DefaultBorderColor()
118 {
119   return QColor( Qt::black );
120 }
121
122 QColor HYDROData_Profile::getDefaultFillingColor() const
123 {
124   return DefaultFillingColor();
125 }
126
127 QColor HYDROData_Profile::getDefaultBorderColor() const
128 {
129   return DefaultBorderColor();
130 }
131
132 bool HYDROData_Profile::IsValid() const
133 {
134   gp_XY aFirstPoint, aLastPoint;
135   if ( !GetLeftPoint( aFirstPoint ) || !GetRightPoint( aLastPoint ) )
136     return false;
137
138   int aNbPoints = NbPoints();
139   return aNbPoints > 1;
140 }
141
142 void HYDROData_Profile::SetLeftPoint( const gp_XY& thePoint )
143 {
144   TDF_Label aLabel = myLab.FindChild( DataTag_FirstPoint );
145
146   Handle(TDataStd_RealArray) anArray;
147   if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
148     anArray = TDataStd_RealArray::Set( aLabel, 0, 1 );
149
150   anArray->SetValue( 0, thePoint.X() );
151   anArray->SetValue( 1, thePoint.Y() );
152
153   SetToUpdate( true );
154 }
155
156 bool HYDROData_Profile::GetLeftPoint( gp_XY& thePoint ) const
157 {
158   TDF_Label aLabel = myLab.FindChild( DataTag_FirstPoint, false );
159   if ( aLabel.IsNull() )
160     return false;
161
162   Handle(TDataStd_RealArray) anArray;
163   if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
164     return false;
165
166   thePoint.SetX( anArray->Value( 0 ) );
167   thePoint.SetY( anArray->Value( 1 ) );
168
169   return true;
170 }
171
172 void HYDROData_Profile::SetRightPoint( const gp_XY& thePoint )
173 {
174   TDF_Label aLabel = myLab.FindChild( DataTag_LastPoint );
175
176   Handle(TDataStd_RealArray) anArray;
177   if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
178     anArray = TDataStd_RealArray::Set( aLabel, 0, 1 );
179
180   anArray->SetValue( 0, thePoint.X() );
181   anArray->SetValue( 1, thePoint.Y() );
182
183   SetToUpdate( true );
184 }
185
186 bool HYDROData_Profile::GetRightPoint( gp_XY& thePoint ) const
187 {
188   TDF_Label aLabel = myLab.FindChild( DataTag_LastPoint, false );
189   if ( aLabel.IsNull() )
190     return false;
191
192   Handle(TDataStd_RealArray) anArray;
193   if ( !aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
194     return false;
195
196   thePoint.SetX( anArray->Value( 0 ) );
197   thePoint.SetY( anArray->Value( 1 ) );
198
199   return true;
200 }
201
202 void HYDROData_Profile::Invalidate()
203 {
204   TDF_Label aFirstLabel = myLab.FindChild( DataTag_FirstPoint, false );
205   if ( !aFirstLabel.IsNull() )
206     aFirstLabel.ForgetAllAttributes();
207
208   TDF_Label aLastLabel = myLab.FindChild( DataTag_LastPoint, false );
209   if ( !aLastLabel.IsNull() )
210     aLastLabel.ForgetAllAttributes();
211
212   SetToUpdate( true );
213 }
214
215 Handle(HYDROData_ProfileUZ) HYDROData_Profile::GetProfileUZ( const bool theIsCreate ) const
216 {
217   Handle(HYDROData_ProfileUZ) aProfileUZ;
218
219   TDF_Label aLabel = myLab.FindChild( DataTag_ChildProfileUZ, theIsCreate );
220   if ( aLabel.IsNull() )
221     return aProfileUZ;
222
223   aProfileUZ = Handle(HYDROData_ProfileUZ)::DownCast( HYDROData_Iterator::Object( aLabel ) );
224   if ( aProfileUZ.IsNull() && theIsCreate )
225   {
226     aProfileUZ = Handle(HYDROData_ProfileUZ)::DownCast(
227       HYDROData_Iterator::CreateObject( aLabel, KIND_PROFILEUZ ) );
228   }
229
230   return aProfileUZ;
231 }
232
233 int HYDROData_Profile::NbPoints() const
234 {
235   Handle(HYDROData_ProfileUZ) aProfileUZ = GetProfileUZ( false );
236   return aProfileUZ.IsNull() ? 0 : aProfileUZ->NbPoints();
237 }
238
239 void HYDROData_Profile::RemovePoints()
240 {
241   Handle(HYDROData_ProfileUZ) aProfileUZ = GetProfileUZ( false );
242   if ( !aProfileUZ.IsNull() )
243   {
244     aProfileUZ->RemoveSections();
245     SetToUpdate( true );
246   }
247 }
248
249 void HYDROData_Profile::SetParametricPoints( const HYDROData_ProfileUZ::PointsList& thePoints )
250 {
251   RemovePoints();
252
253   Handle(HYDROData_ProfileUZ) aProfileUZ = GetProfileUZ();
254   for ( int i = 1, n = thePoints.Length(); i <= n ; ++i )
255   {
256     const HYDROData_ProfileUZ::Point& aPoint = thePoints.Value( i );
257     aProfileUZ->AddPoint( 0, aPoint );
258   }
259
260   SetToUpdate( true );
261 }
262
263 HYDROData_ProfileUZ::PointsList HYDROData_Profile::GetParametricPoints() const
264 {
265   Handle(HYDROData_ProfileUZ) aProfileUZ = GetProfileUZ( false );
266   return aProfileUZ.IsNull() ? HYDROData_ProfileUZ::PointsList() : aProfileUZ->GetPoints();
267 }
268
269 void HYDROData_Profile::SetProfilePoints( const ProfilePoints& thePoints )
270 {
271   RemovePoints();
272   if ( thePoints.Length() < 2 )
273     return;
274
275   gp_XY aFirstPoint, aLastPoint;
276
277   Handle(HYDROData_ProfileUZ) aProfileUZ = GetProfileUZ();
278   for ( int i = 1, n = thePoints.Length(); i <= n ; ++i )
279   {
280     const ProfilePoint& aPoint = thePoints.Value( i );
281     gp_XY aPointXY( aPoint.X(), aPoint.Y() );
282
283     if ( i == 1 )
284       aFirstPoint = aPointXY;
285     else if ( i == n )
286       aLastPoint = aPointXY;
287
288     double aDistance = gp_Pnt2d( aFirstPoint ).Distance( aPointXY );
289     
290     HYDROData_ProfileUZ::Point aParPoint( aDistance, aPoint.Z() );
291     aProfileUZ->AddPoint( 0, aParPoint );
292   }
293
294   SetLeftPoint( aFirstPoint );
295   SetRightPoint( aLastPoint );
296 }
297
298 HYDROData_Profile::ProfilePoints HYDROData_Profile::GetProfilePoints() const
299 {
300   ProfilePoints aResPoints;
301
302   gp_XY aFirstPoint, aLastPoint;
303   if ( !GetLeftPoint( aFirstPoint ) || !GetRightPoint( aLastPoint ) )
304     return aResPoints;
305
306   HYDROData_ProfileUZ::PointsList aParametricPoints = GetParametricPoints();
307   if ( aParametricPoints.Length() < 2 )
308     return aResPoints;
309
310   const HYDROData_ProfileUZ::Point& aFirstParPoint = aParametricPoints.First();
311   const HYDROData_ProfileUZ::Point& aLastParPoint = aParametricPoints.Last();
312
313   double aGeoDistance = gp_Pnt2d( aFirstPoint ).Distance( aLastPoint );
314   double aParCommonDist = gp_Pnt2d( aFirstParPoint.X(), 0 ).Distance( gp_Pnt2d( aLastParPoint.X(), 0 ) );
315
316   // Add first point as is
317   aResPoints.Append( ProfilePoint( aFirstPoint.X(), aFirstPoint.Y(), aFirstParPoint.Y() ) );
318
319   // Compute all other points
320   for ( int i = 2, n = aParametricPoints.Length(); i < n ; ++i )
321   {
322     const HYDROData_ProfileUZ::Point& aParPoint = aParametricPoints.Value( i );
323
324     double aParPointDist = gp_Pnt2d( aFirstParPoint.X(), 0 ).Distance( gp_Pnt2d( aParPoint.X(), 0 ) );
325     
326     double aParLen = ( aParPointDist / aParCommonDist ) * aGeoDistance;
327
328     double aRatio = aParLen / ( aGeoDistance - aParLen );
329
330     double aParX = ( aFirstPoint.X() + aRatio * aLastPoint.X() ) / ( 1 + aRatio );
331     double aParY = ( aFirstPoint.Y() + aRatio * aLastPoint.Y() ) / ( 1 + aRatio );
332
333     ProfilePoint aCompPoint( aParX, aParY, aParPoint.Y() );
334     aResPoints.Append( aCompPoint );
335   }
336
337   // Add last point as is
338   aResPoints.Append( ProfilePoint( aLastPoint.X(), aLastPoint.Y(), aLastParPoint.Y() ) );
339
340   return aResPoints;
341 }
342
343 void HYDROData_Profile::SetFilePath( const TCollection_AsciiString& theFilePath )
344 {
345   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_FilePath ), theFilePath );
346 }
347
348 TCollection_AsciiString HYDROData_Profile::GetFilePath() const
349 {
350   TCollection_AsciiString aRes;
351
352   Handle(TDataStd_AsciiString) anAsciiStr;
353   if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
354     aRes = anAsciiStr->Get();
355
356   return aRes;
357 }
358
359 bool HYDROData_Profile::ImportFromFile( const Handle(HYDROData_Document)& theDoc,
360                                         const TCollection_AsciiString&    theFileName )
361 {
362   if ( theDoc.IsNull() || theFileName.IsEmpty() )
363     return false;
364
365   OSD_File aFile( theFileName );
366   if ( !aFile.IsReadable() )
367     return false;
368
369   aFile.Open( OSD_ReadOnly, OSD_Protection() );
370   if ( !aFile.IsOpen() )
371     return false;
372
373   NCollection_Sequence<Handle(HYDROData_Profile)> aCreatedProfiles;
374
375   Handle(HYDROData_Profile) aNewProfile;
376   while ( !aFile.IsAtEnd() )
377   {
378     if ( aNewProfile.IsNull() )
379       aNewProfile = Handle(HYDROData_Profile)::DownCast( theDoc->CreateObject( KIND_PROFILE ) );
380     
381     if ( aNewProfile->ImportFromFile( aFile ) )
382     {
383       aCreatedProfiles.Append( aNewProfile );
384       aNewProfile.Nullify();
385     }
386   }
387
388   if ( !aNewProfile.IsNull() )
389     aNewProfile->Remove();
390
391   // Close the file
392   aFile.Close();
393
394   for ( int i = 1, n = aCreatedProfiles.Length(); i <= n ; ++i )
395   {
396     Handle(HYDROData_Profile) aProfile = aCreatedProfiles.Value( i );
397
398     QString aProfileName = HYDROData_Tool::GenerateObjectName( theDoc, "Profile" );
399     aProfile->SetName( aProfileName );
400
401     aProfile->SetFilePath( theFileName );
402
403     aProfile->SetBorderColor( HYDROData_Profile::DefaultBorderColor() );
404   }
405
406   return !aCreatedProfiles.IsEmpty();
407 }
408
409 bool HYDROData_Profile::ImportFromFile( const TCollection_AsciiString& theFileName )
410 {
411   // Try to open the file
412   OSD_File aFile( theFileName );
413   if ( !aFile.IsReadable() )
414     return false;
415
416   aFile.Open( OSD_ReadOnly, OSD_Protection() );
417   if ( !aFile.IsOpen() )
418     return false;
419
420   bool aRes = ImportFromFile( aFile );
421
422   // Close the file
423   aFile.Close();
424
425   if ( aRes )
426   {
427     // Update file path
428     SetFilePath( theFileName );
429   }
430
431   return aRes;
432 }
433
434 bool HYDROData_Profile::ImportFromFile( OSD_File& theFile )
435 {
436   if ( !theFile.IsOpen() )
437     return false;
438
439   bool aRes = true;
440
441   bool anIsParametric = false;
442   bool anIsGeoref     = false;
443
444   HYDROData_ProfileUZ::PointsList aPointsUZ;
445   ProfilePoints                   aPointsXYZ;
446
447   double aPrevVal = -DBL_MAX;
448   while ( !theFile.IsAtEnd() )
449   {
450     Standard_Integer aNbRead = 0;
451     TCollection_AsciiString aLine;
452     theFile.ReadLine( aLine, 1024, aNbRead );
453
454     aLine.LeftAdjust(); aLine.RightAdjust();
455     if ( aLine.IsEmpty() )
456     {
457       if ( !anIsParametric && !anIsGeoref )
458         continue; // Definition is not started yet
459
460       break; // Next profile started
461     }
462
463     TCollection_AsciiString aValX = aLine.Token( " \t", 1 );
464     TCollection_AsciiString aValY = aLine.Token( " \t", 2 );
465     TCollection_AsciiString aValZ = aLine.Token( " \t", 3 );
466
467     if ( aValX.IsEmpty() || !aValX.IsRealValue() ||
468          aValY.IsEmpty() || !aValY.IsRealValue() )
469     {
470       aRes = false;
471       break;
472     }
473
474     if ( !anIsParametric && !anIsGeoref )
475     {
476       anIsParametric = aValZ.IsEmpty();
477       anIsGeoref = !aValZ.IsEmpty();
478     }
479
480     double aCoordX = aValX.RealValue();
481     double aCoordY = aValY.RealValue();
482
483     if ( anIsParametric )
484     {
485       if ( aCoordX < aPrevVal )
486       {
487         // Move back readed line
488         theFile.Seek( -( aNbRead + 1 ), OSD_FromHere );
489         break;
490       }
491
492       HYDROData_ProfileUZ::Point aPoint( aCoordX, aCoordY );
493       aPointsUZ.Append( aPoint );
494
495       aPrevVal = aCoordX;
496     }
497     else
498     {
499       if ( aValZ.IsEmpty() || !aValZ.IsRealValue() )
500       {
501         aRes = false;
502         break;
503       }
504
505       double aCoordZ = aValZ.RealValue();
506
507       ProfilePoint aPoint( aCoordX, aCoordY, aCoordZ );
508       aPointsXYZ.Append( aPoint );
509     }
510   }
511   
512   aRes = aRes && ( anIsParametric && !aPointsUZ.IsEmpty() || 
513                    anIsGeoref && !aPointsXYZ.IsEmpty() );
514   if ( aRes )
515   {
516     // Update profile points
517     if ( anIsParametric )
518     {
519       SetParametricPoints( aPointsUZ );
520     }
521     else if ( anIsGeoref )
522     {
523       SetProfilePoints( aPointsXYZ );
524     }
525
526     Update();
527   }
528
529   return aRes;
530 }
531
532
533