src/HYDROGUI/HYDROGUI_StreamOp.cxx

   1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "HYDROGUI_StreamOp.h"
  24
  25 #include "HYDROGUI_Module.h"
  26 #include "HYDROGUI_Shape.h"
  27 #include "HYDROGUI_StreamDlg.h"
  28 #include "HYDROGUI_Tool.h"
  29 #include "HYDROGUI_UpdateFlags.h"
  30
  31 #include <HYDROData_Document.h>
  32 #include <HYDROData_PolylineXY.h>
  33 #include <HYDROData_Profile.h>
  34
  35 #include <LightApp_Application.h>
  36 #include <LightApp_SelectionMgr.h>
  37 #include <LightApp_UpdateFlags.h>
  38
  39 #include <SUIT_MessageBox.h>
  40 #include <SUIT_Desktop.h>
  41
  42 #include <OCCViewer_ViewManager.h>
  43 #include <OCCViewer_ViewModel.h>
  44 #include <OCCViewer_ViewWindow.h>
  45 #include <gp_Ax1.hxx>
  46 #include <gp_Ax2.hxx>
  47 #include <gp_Ax3.hxx>
  48 #include <gp_Vec.hxx>
  49 #include <gp_Pnt.hxx>
  50 #include <gp_Pln.hxx>
  51 #include <gp.hxx>
  52 #include <TopoDS_Face.hxx>
  53 #include <TopoDS.hxx>
  54 #include <BRepBuilderAPI_MakeFace.hxx>
  55 HYDROGUI_StreamOp::HYDROGUI_StreamOp( HYDROGUI_Module* theModule, bool theIsEdit )
  56 : HYDROGUI_Operation( theModule ),
  57   myIsEdit( theIsEdit ),
  58   myViewManager( NULL ),
  59   myPreviewPrs( NULL )
  60 {
  61   setName( theIsEdit ? tr( "EDIT_STREAM" ) : tr( "CREATE_STREAM" ) );
  62 }
  63
  64 HYDROGUI_StreamOp::~HYDROGUI_StreamOp()
  65 {
  66   erasePreview();
  67 }
  68
  69 void HYDROGUI_StreamOp::startOperation()
  70 {
  71   HYDROGUI_Operation::startOperation();
  72
  73   // We start operation in the document
  74   startDocOperation();
  75
  76   // Get panel and reset its state
  77   HYDROGUI_StreamDlg* aPanel = (HYDROGUI_StreamDlg*)inputPanel();
  78   aPanel->reset();
  79
  80   // Get/create the edited object
  81   if( myIsEdit ) {
  82     myEditedObject = Handle(HYDROData_Stream)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  83   } else {
  84     myEditedObject = Handle(HYDROData_Stream)::DownCast( doc()->CreateObject( KIND_STREAM ) );
  85   }
  86
  87   // Get the edited object name
  88   QString anObjectName;
  89   if ( !myIsEdit ) {
  90     anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_STREAM_NAME" ) );
  91   } else if ( !myEditedObject.IsNull() ) {
  92     anObjectName = myEditedObject->GetName();
  93   }
  94
  95   // Update the panel
  96   // set the edited object name
  97   aPanel->setObjectName( anObjectName );
  98   // set the existing 2D polylines names to the panel
  99   aPanel->setAxisNames( HYDROGUI_Tool::FindExistingObjectsNames( doc(), KIND_POLYLINEXY ) );
 100   // synchronize the panel state with the edited object state
 101   updatePanelData();
 102
 103   // Create preview
 104   createPreview();
 105 }
 106
 107 void HYDROGUI_StreamOp::abortOperation()
 108 {
 109   erasePreview();
 110   abortDocOperation();
 111
 112   HYDROGUI_Operation::abortOperation();
 113 }
 114
 115 void HYDROGUI_StreamOp::commitOperation()
 116 {
 117   erasePreview();
 118
 119   HYDROGUI_Operation::commitOperation();
 120 }
 121
 122 HYDROGUI_InputPanel* HYDROGUI_StreamOp::createInputPanel() const
 123 {
 124   HYDROGUI_StreamDlg* aPanel = new HYDROGUI_StreamDlg( module(), getName() );
 125
 126   connect( aPanel, SIGNAL( AddProfiles() ), this, SLOT( onAddProfiles() ) );
 127   connect( aPanel, SIGNAL( RemoveProfiles( const QStringList& ) ),
 128            this, SLOT( onRemoveProfiles( const QStringList& ) ) );
 129   connect( aPanel, SIGNAL( AxisChanged( const QString& ) ),
 130            this, SLOT( onAxisChanged( const QString& ) ) );
 131
 132   return aPanel;
 133 }
 134
 135 bool HYDROGUI_StreamOp::processApply( int& theUpdateFlags,
 136                                       QString& theErrorMsg )
 137 {
 138   HYDROGUI_StreamDlg* aPanel = ::qobject_cast<HYDROGUI_StreamDlg*>( inputPanel() );
 139   if ( !aPanel || myEditedObject.IsNull() ) {
 140     return false;
 141   }
 142
 143   // Check whether the object name is not empty
 144   QString anObjectName = aPanel->getObjectName().simplified();
 145   if ( anObjectName.isEmpty() ) {
 146     theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
 147     return false;
 148   }
 149
 150   // Check that there are no other objects with the same name in the document
 151   if( myEditedObject->GetName() != anObjectName )
 152   {
 153     Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anObjectName );
 154     if( !anObject.IsNull() ) {
 155       theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anObjectName );
 156       return false;
 157     }
 158   }
 159
 160   // Check if the axis is set
 161   Handle(HYDROData_PolylineXY) aHydraulicAxis = myEditedObject->GetHydraulicAxis();
 162   if ( aHydraulicAxis.IsNull() ) {
 163     theErrorMsg = tr( "AXIS_NOT_DEFINED" );
 164     return false;
 165   }
 166
 167   // Check if at least 2 profiles is set
 168   HYDROData_SequenceOfObjects aProfiles = myEditedObject->GetProfiles();
 169   if ( aProfiles.Length() < 2 ) {
 170     theErrorMsg = tr( "PROFILES_NOT_DEFINED" );
 171     return false;
 172   }
 173
 174   // Set the object name
 175   myEditedObject->SetName( anObjectName );
 176
 177   if ( !myIsEdit )
 178   {
 179     myEditedObject->SetFillingColor( HYDROData_Stream::DefaultFillingColor() );
 180     myEditedObject->SetBorderColor( HYDROData_Stream::DefaultBorderColor() );
 181   }
 182
 183   // Erase preview
 184   erasePreview();
 185
 186   // Show the object in case of creation mode of the operation
 187   if( !myIsEdit ) {
 188     module()->setObjectVisible( HYDROGUI_Tool::GetActiveOCCViewId( module() ), myEditedObject, true );
 189   }
 190
 191   // Set update flags
 192   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced;
 193
 194   return true;
 195 }
 196
 197 void HYDROGUI_StreamOp::createPreview()
 198 {
 199   if ( myEditedObject.IsNull() ) {
 200     return;
 201   }
 202
 203   LightApp_Application* anApp = module()->getApp();
 204   if ( !myViewManager ) {
 205     myViewManager = ::qobject_cast<OCCViewer_ViewManager*>(
 206       anApp->getViewManager( OCCViewer_Viewer::Type(), true ) );
 207   }
 208
 209   if ( myViewManager && !myPreviewPrs ) {
 210     if ( OCCViewer_Viewer* aViewer = myViewManager->getOCCViewer() ) {
 211       Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
 212       if ( !aCtx.IsNull() ) {
 213         myPreviewPrs = new HYDROGUI_Shape( aCtx, myEditedObject );
 214       }
 215     }
 216   }
 217
 218   if ( myPreviewPrs ) {
 219     myPreviewPrs->update();
 220   }
 221 }
 222
 223 void HYDROGUI_StreamOp::erasePreview()
 224 {
 225   if( myPreviewPrs ) {
 226     delete myPreviewPrs;
 227     myPreviewPrs = 0;
 228   }
 229 }
 230
 231 void HYDROGUI_StreamOp::onAddProfiles()
 232 {
 233   if ( myEditedObject.IsNull() ) {
 234     return;
 235   }
 236
 237   // Get the current profiles list
 238   HYDROData_SequenceOfObjects aProfiles = myEditedObject->GetProfiles();
 239
 240   // TODO: to be optimized
 241   QStringList aCurrentProfiles;
 242   for( int i = 1, n = aProfiles.Length(); i <= n; i++ ) {
 243     Handle(HYDROData_Profile) aProfile =
 244       Handle(HYDROData_Profile)::DownCast( aProfiles.Value( i ) );
 245     if ( !aProfile.IsNull() ) {
 246       aCurrentProfiles << aProfile->GetName();
 247     }
 248   }
 249
 250   // Get the selected profiles ( in the Object Browser )
 251   QStringList anInvalidProfiles, anExistingProfiles, aHasNoIntersectionProfiles;
 252
 253   HYDROData_SequenceOfObjects aVerifiedProfiles;
 254   HYDROData_SequenceOfObjects aSelectedProfiles = HYDROGUI_Tool::GetSelectedObjects( module() );
 255   Handle(HYDROData_PolylineXY) aHydAxis = myEditedObject->GetHydraulicAxis();
 256   if ( aHydAxis.IsNull() )
 257     return;
 258   TopoDS_Face aPlane;
 259   if(!myEditedObject->BuildFace(aHydAxis, aPlane))
 260     return;
 261   Standard_Real aPar(.0);
 262   for( int i = 1, n = aSelectedProfiles.Length(); i <= n; i++ ) {
 263     Handle(HYDROData_Profile) aProfile =
 264       Handle(HYDROData_Profile)::DownCast( aSelectedProfiles.Value( i ) );
 265     if ( !aProfile.IsNull() ) {
 266       QString aProfileName = aProfile->GetName();
 267
 268       // Check the profile, if all is ok - add it to the list
 269       if ( !aProfile->IsValid() ) { // check whether the profile is valid
 270         anInvalidProfiles << aProfileName;
 271       } else if ( aCurrentProfiles.contains( aProfileName ) ) { // check whether the profile is already added
 272         anExistingProfiles << aProfileName;
 273       } else if ( !myEditedObject->HasIntersection( aProfile, aPlane, aPar ) ) {  // check whether the profile has intersection
 274         aHasNoIntersectionProfiles << aProfileName;
 275       } else {
 276         aVerifiedProfiles.Append( aProfile );
 277       }
 278     }
 279   }
 280
 281   // Show message box with the ignored profiles
 282   if ( !anInvalidProfiles.isEmpty() || !anExistingProfiles.isEmpty() ||
 283        !aHasNoIntersectionProfiles.isEmpty() ) {
 284     QString aMessage = tr( "IGNORED_PROFILES" );
 285     if ( !anInvalidProfiles.isEmpty() ) {
 286       aMessage.append( "\n\n" );
 287       aMessage.append( tr("INVALID_PROFILES").arg( anInvalidProfiles.join( "\n" ) ) );
 288     }
 289     if ( !anExistingProfiles.isEmpty() ) {
 290       aMessage.append( "\n\n" );
 291       aMessage.append( tr("EXISTING_PROFILES").arg( anExistingProfiles.join( "\n" ) ) );
 292     }
 293     if ( !aHasNoIntersectionProfiles.isEmpty() ) {
 294       aMessage.append( "\n\n" );
 295       aMessage.append( tr("NOT_INTERSECTED_PROFILES").arg( aHasNoIntersectionProfiles.join( "\n" ) ) );
 296     }
 297
 298     SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "WARNING" ), aMessage );
 299   }
 300
 301   // Update the stream object
 302   for( int i = 1, n = aVerifiedProfiles.Length(); i <= n; i++ ) {
 303     Handle(HYDROData_Profile) aProfile =
 304       Handle(HYDROData_Profile)::DownCast( aVerifiedProfiles.Value( i ) );
 305     myEditedObject->AddProfile( aProfile );
 306   }
 307   myEditedObject->Update();
 308
 309   // Update the panel
 310   updatePanelData();
 311
 312   // Update preview
 313   createPreview();
 314 }
 315
 316 void HYDROGUI_StreamOp::onRemoveProfiles( const QStringList& theProfilesToRemove )
 317 {
 318   if ( myEditedObject.IsNull() ) {
 319     return;
 320   }
 321
 322   bool isRemoved = false;
 323
 324   // Take the Object Browser selection into account
 325   HYDROData_SequenceOfObjects aSelectedObjects = HYDROGUI_Tool::GetSelectedObjects( module() );
 326   for( int i = 1, n = aSelectedObjects.Length(); i <= n; i++ ) {
 327     Handle(HYDROData_Profile) aProfile =
 328       Handle(HYDROData_Profile)::DownCast( aSelectedObjects.Value( i ) );
 329     if ( !aProfile.IsNull() && !theProfilesToRemove.contains(aProfile->GetName()) ) {
 330       if ( myEditedObject->RemoveProfile( aProfile ) ) {
 331         isRemoved = true;
 332       }
 333     }
 334   }
 335
 336   // Remove profiles
 337   foreach( const QString& aProfileName, theProfilesToRemove ) {
 338     Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast(
 339       HYDROGUI_Tool::FindObjectByName( module(), aProfileName, KIND_PROFILE ) );
 340     if ( myEditedObject->RemoveProfile( aProfile ) ) {
 341       isRemoved = true;
 342     }
 343   }
 344
 345   if ( isRemoved ) {
 346     // Update the edited stream object
 347     myEditedObject->Update();
 348
 349     // Update the panel
 350     updatePanelData();
 351
 352     // Update preview
 353     createPreview();
 354   }
 355 }
 356
 357 void HYDROGUI_StreamOp::onAxisChanged( const QString& theNewAxis )
 358 {
 359   if ( myEditedObject.IsNull() ) {
 360     return;
 361   }
 362
 363   // Get axis object
 364   Handle(HYDROData_PolylineXY) anAxis = Handle(HYDROData_PolylineXY)::DownCast(
 365     HYDROGUI_Tool::FindObjectByName( module(), theNewAxis, KIND_POLYLINEXY ) );
 366
 367   // Prepare data for intersection check
 368   TopoDS_Face aPlane;
 369   if ( anAxis.IsNull() || !myEditedObject->BuildFace(anAxis, aPlane) ) {
 370     // To restore the old axis
 371     updatePanelData();
 372     return;
 373   }
 374   Standard_Real aPar(.0);
 375
 376   // Get list of profiles which do not intersect the axis
 377   QStringList aHasNoIntersectionProfiles;
 378   HYDROData_SequenceOfObjects aCurrentProfiles = myEditedObject->GetProfiles();
 379   for( int anIndex = 1, aLength = aCurrentProfiles.Length(); anIndex <= aLength; anIndex++ ) {
 380     Handle(HYDROData_Profile) aProfile =
 381       Handle(HYDROData_Profile)::DownCast( aCurrentProfiles.Value( anIndex ) );
 382     if ( !aProfile.IsNull() ) {
 383       if ( !HYDROData_Stream::HasIntersection( anAxis, aProfile, aPlane, aPar ) ) {
 384         aHasNoIntersectionProfiles << aProfile->GetName();
 385       }
 386     }
 387   }
 388
 389   // If there are profiles which don't intersect the new axis - show confirmation message box
 390   bool isConfirmed = true;
 391   if ( !aHasNoIntersectionProfiles.isEmpty() ) {
 392     SUIT_MessageBox::StandardButtons aButtons =
 393       SUIT_MessageBox::Yes | SUIT_MessageBox::No;
 394
 395     QString aMsg = aHasNoIntersectionProfiles.join( "\n" );
 396     isConfirmed = SUIT_MessageBox::question( module()->getApp()->desktop(),
 397                                              tr( "CONFIRMATION" ),
 398                                              tr( "CONFIRM_AXIS_CHANGE" ).arg( aMsg ),
 399                                              aButtons,
 400                                              SUIT_MessageBox::Yes) == SUIT_MessageBox::Yes;
 401   }
 402
 403   // Check if the user has confirmed axis change
 404   if ( !isConfirmed ) {
 405     // To restore the old axis
 406     updatePanelData();
 407   } else {
 408     // Set axis
 409     myEditedObject->SetHydraulicAxis( anAxis );
 410     myEditedObject->Update();
 411
 412     // Update the panel
 413     updatePanelData();
 414
 415     // Update preview
 416     createPreview();
 417   }
 418 }
 419
 420 void HYDROGUI_StreamOp::updatePanelData()
 421 {
 422   HYDROGUI_StreamDlg* aPanel = ::qobject_cast<HYDROGUI_StreamDlg*>( inputPanel() );
 423   if ( !aPanel ) {
 424     return;
 425   }
 426
 427   // Hydraulic axis
 428   Handle(HYDROData_PolylineXY) aHydraulicAxis = myEditedObject->GetHydraulicAxis();
 429   if ( !aHydraulicAxis.IsNull() ) {
 430     aPanel->setAxisName( aHydraulicAxis->GetName() );
 431   } else {
 432     aPanel->setAxisName( "" );
 433   }
 434
 435   // Stream profiles
 436   QStringList aProfiles;
 437
 438   HYDROData_SequenceOfObjects aStreamProfiles = myEditedObject->GetProfiles();
 439   for ( int i = 1, n = aStreamProfiles.Length(); i <= n; ++i ) {
 440     Handle(HYDROData_Profile) aProfile =
 441       Handle(HYDROData_Profile)::DownCast( aStreamProfiles.Value( i ) );
 442     if ( !aProfile.IsNull() ) {
 443       QString aProfileName = aProfile->GetName();
 444       aProfiles << aProfileName;
 445     }
 446   }
 447
 448   aPanel->setProfiles( aProfiles );
 449 }