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