src/SMESHGUI/SMESHGUI_FindElemByPointDlg.cxx

   1 // Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // File   : SMESHGUI_FindElemByPointDlg.cxx
  21 // Author : Edward AGAPOV, Open CASCADE S.A.S.
  22 //
  23
  24 // SMESH includes
  25 #include "SMESHGUI_FindElemByPointDlg.h"
  26
  27 #include "SMESHGUI.h"
  28 #include "SMESHGUI_MeshUtils.h"
  29 #include "SMESHGUI_VTKUtils.h"
  30 #include "SMESHGUI_SpinBox.h"
  31 #include "SMESHGUI_MeshEditPreview.h"
  32
  33 #include "SMESH_Actor.h"
  34 #include "SMESH_ActorUtils.h"
  35 #include "SMESH_TypeFilter.hxx"
  36 #include "SMESH_LogicalFilter.hxx"
  37
  38 // SALOME GUI includes
  39 #include <LightApp_SelectionMgr.h>
  40 #include <QtxComboBox.h>
  41 #include <SALOME_ListIO.hxx>
  42 #include <SUIT_Desktop.h>
  43 #include <SUIT_MessageBox.h>
  44 #include <SUIT_OverrideCursor.h>
  45 #include <SUIT_ResourceMgr.h>
  46 #include <SVTK_ViewModel.h>
  47 #include <SVTK_ViewWindow.h>
  48 #include <SalomeApp_Tools.h>
  49 #include <SalomeApp_TypeFilter.h>
  50
  51 // Qt includes
  52 #include <QAbstractButton>
  53 #include <QGridLayout>
  54 #include <QGroupBox>
  55 #include <QHBoxLayout>
  56 #include <QLabel>
  57 #include <QLineEdit>
  58 #include <QListWidget>
  59 #include <QPushButton>
  60 #include <QVBoxLayout>
  61
  62 // VTK includes
  63 #include <vtkProperty.h>
  64
  65 // IDL includes
  66 #include <SALOMEconfig.h>
  67 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
  68
  69 #define SPACING 6
  70 #define MARGIN  11
  71
  72 //=======================================================================
  73 /*!
  74  * \brief Dialog to find elements by a point coordinates
  75  */
  76 //=======================================================================
  77
  78 SMESHGUI_FindElemByPointDlg::SMESHGUI_FindElemByPointDlg()
  79   : SMESHGUI_Dialog( 0, false, true, OK | Help )
  80 {
  81   setWindowTitle(tr("CAPTION"));
  82
  83   QVBoxLayout* aDlgLay = new QVBoxLayout (mainFrame());
  84   aDlgLay->setMargin(0);
  85   aDlgLay->setSpacing(SPACING);
  86
  87   QWidget* aMainFrame = createMainFrame  (mainFrame());
  88
  89   aDlgLay->addWidget(aMainFrame);
  90   aDlgLay->setStretchFactor(aMainFrame, 1);
  91 }
  92
  93 //================================================================================
  94 /*!
  95  * \brief Set types of elements available for search
  96  */
  97 //================================================================================
  98
  99 void SMESHGUI_FindElemByPointDlg::setTypes(SMESH::array_of_ElementType_var & types)
 100 {
 101   myElemTypeCombo->blockSignals(true);
 102   myElemTypeCombo->clear();
 103   int nbTypes = 0, hasNodes = 0;
 104   for ( int i = 0; i < (int) types->length(); ++i )
 105   {
 106     switch ( types[i] ) {
 107     case SMESH::NODE:
 108       myElemTypeCombo->addItem( tr( "MEN_NODE" ));
 109       myElemTypeCombo->setId( nbTypes++, int( SMESH::NODE  ));
 110       hasNodes = 1;
 111       break;
 112     case SMESH::EDGE:
 113       myElemTypeCombo->addItem( tr( "MEN_EDGE" ));
 114       myElemTypeCombo->setId( nbTypes++, int( SMESH::EDGE  ));
 115       break;
 116     case SMESH::FACE:
 117       myElemTypeCombo->addItem( tr( "MEN_FACE" ));
 118       myElemTypeCombo->setId( nbTypes++, int( SMESH::FACE  ));
 119       break;
 120     case SMESH::VOLUME:
 121       myElemTypeCombo->addItem( tr( "MEN_VOLUME_3D" ));
 122       myElemTypeCombo->setId( nbTypes++, int( SMESH::VOLUME  ));
 123       break;
 124     case SMESH::ELEM0D:
 125       myElemTypeCombo->addItem( tr( "MEN_ELEM0D" ));
 126       myElemTypeCombo->setId( nbTypes++, int( SMESH::ELEM0D  ));
 127       break;
 128     case SMESH::BALL:
 129       myElemTypeCombo->addItem( tr( "MEN_BALL" ));
 130       myElemTypeCombo->setId( nbTypes++, int( SMESH::BALL  ));
 131       break;
 132     default:;
 133     }
 134   }
 135   if ( nbTypes - hasNodes > 1 )
 136   {
 137     myElemTypeCombo->addItem( tr( "MEN_ALL" ));
 138     myElemTypeCombo->setId( nbTypes++, int( SMESH::ALL   ));
 139   }
 140   if ( !hasNodes && nbTypes > 0 )
 141   {
 142     myElemTypeCombo->addItem( tr( "MEN_NODE" ));
 143     myElemTypeCombo->setId( nbTypes++, int( SMESH::NODE  ));
 144   }
 145   myElemTypeCombo->blockSignals(false);
 146 }
 147
 148 //=======================================================================
 149 // function : createMainFrame()
 150 // purpose  : Create frame containing dialog's input fields
 151 //=======================================================================
 152 QWidget* SMESHGUI_FindElemByPointDlg::createMainFrame (QWidget* theParent)
 153 {
 154   QWidget* aFrame = new QWidget(theParent);
 155
 156   //mesh name
 157
 158   QGroupBox* aMeshGrp = new QGroupBox(tr("MESH_GROUP"), aFrame);
 159   QHBoxLayout* aMeshGrpLayout = new QHBoxLayout(aMeshGrp);
 160   aMeshGrpLayout->setMargin(MARGIN);
 161   aMeshGrpLayout->setSpacing(SPACING);
 162
 163   myMeshName = new QLineEdit(aMeshGrp);
 164   aMeshGrpLayout->addWidget(myMeshName);
 165
 166   // coordinates
 167
 168   QGroupBox* aCoordGrp = new QGroupBox(tr("SMESH_COORDINATES"), aFrame);
 169   QHBoxLayout* aCoordGrpLayout = new QHBoxLayout(aCoordGrp);
 170   aCoordGrpLayout->setMargin(MARGIN);
 171   aCoordGrpLayout->setSpacing(SPACING);
 172
 173   QLabel* aXLabel = new QLabel(tr("SMESH_X"), aCoordGrp);
 174   myX = new SMESHGUI_SpinBox(aCoordGrp);
 175
 176   QLabel* aYLabel = new QLabel(tr("SMESH_Y"), aCoordGrp);
 177   myY = new SMESHGUI_SpinBox(aCoordGrp);
 178
 179   QLabel* aZLabel = new QLabel(tr("SMESH_Z"), aCoordGrp);
 180   myZ = new SMESHGUI_SpinBox(aCoordGrp);
 181
 182   myX->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, "length_precision");
 183   myY->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, "length_precision");
 184   myZ->RangeStepAndValidator(COORD_MIN, COORD_MAX, 10.0, "length_precision");
 185   myX->SetValue(0);
 186   myY->SetValue(0);
 187   myZ->SetValue(0);
 188
 189   aCoordGrpLayout->addWidget(aXLabel);
 190   aCoordGrpLayout->addWidget(myX);
 191   aCoordGrpLayout->addWidget(aYLabel);
 192   aCoordGrpLayout->addWidget(myY);
 193   aCoordGrpLayout->addWidget(aZLabel);
 194   aCoordGrpLayout->addWidget(myZ);
 195
 196   // Elements
 197
 198   QGroupBox* elementGrp = new QGroupBox(tr("Elements"), aFrame);
 199   QGridLayout* elementGrpLayout = new QGridLayout(elementGrp);
 200   elementGrpLayout->setSpacing(SPACING);
 201   elementGrpLayout->setMargin(MARGIN);
 202
 203
 204   myFindBtn = new QPushButton(elementGrp);
 205   myFindBtn->setText(tr("Find"));
 206   //myFindBtn->setCheckable(false);
 207
 208   myElemTypeCombo = new QtxComboBox(elementGrp);
 209
 210   myFoundList = new QListWidget(elementGrp);
 211
 212   elementGrpLayout->addWidget( myFindBtn,       0, 0 );
 213   elementGrpLayout->addWidget( myElemTypeCombo, 0, 1 );
 214   elementGrpLayout->addWidget( myFoundList,     1, 0, 2, 2 );
 215
 216
 217   QVBoxLayout* aLay = new QVBoxLayout(aFrame);
 218   aLay->setMargin( 0 );
 219   aLay->setSpacing( SPACING );
 220   aLay->addWidget(aMeshGrp);
 221   aLay->addWidget(aCoordGrp);
 222   aLay->addWidget(elementGrp);
 223
 224   // OK instead of "Apply and Close"
 225   if ( QAbstractButton* but = button(OK) )
 226     but->setText( tr("SMESH_BUT_OK"));
 227
 228   return aFrame;
 229 }
 230
 231 //================================================================================
 232 /*!
 233  * \brief Constructor
 234 */
 235 //================================================================================
 236
 237 SMESHGUI_FindElemByPointOp::SMESHGUI_FindElemByPointOp()
 238   :SMESHGUI_SelectionOp()
 239 {
 240   mySimulation = 0;
 241   mySMESHGUI = 0;
 242   myDlg = new SMESHGUI_FindElemByPointDlg;
 243   myHelpFileName = "find_element_by_point.html";
 244
 245   QList<SUIT_SelectionFilter*> filters;
 246   filters.append( new SMESH_TypeFilter( SMESH::MESH ) );
 247   filters.append( new SMESH_TypeFilter( SMESH::GROUP ) );
 248   myFilter = new SMESH_LogicalFilter( filters, SMESH_LogicalFilter::LO_OR );
 249
 250   myPreview = new SMESH::MeshPreviewStruct();
 251
 252   myPreview->nodesXYZ.length(1);
 253   myPreview->nodesXYZ[0].x = myDlg->myX->GetValue();
 254   myPreview->nodesXYZ[0].y = myDlg->myY->GetValue();
 255   myPreview->nodesXYZ[0].z = myDlg->myZ->GetValue();
 256
 257   myPreview->elementTypes.length(1);
 258   myPreview->elementTypes[0].SMDS_ElementType = SMESH::NODE;
 259   myPreview->elementTypes[0].isPoly = false;
 260   myPreview->elementTypes[0].nbNodesInElement = 1;
 261
 262   myPreview->elementConnectivities.length(1);
 263   myPreview->elementConnectivities[0] = 0;
 264
 265   // connect signals and slots
 266   connect(myDlg->myFindBtn,      SIGNAL(clicked()),               this, SLOT(onFind()));
 267   connect(myDlg->myX,            SIGNAL(valueChanged(double)),    this, SLOT(redisplayPreview()));
 268   connect(myDlg->myY,            SIGNAL(valueChanged(double)),    this, SLOT(redisplayPreview()));
 269   connect(myDlg->myZ,            SIGNAL(valueChanged(double)),    this, SLOT(redisplayPreview()));
 270   connect(myDlg->myFoundList,    SIGNAL(itemSelectionChanged()),  this, SLOT(onElemSelected()));
 271   connect(myDlg->myElemTypeCombo,SIGNAL(currentIndexChanged(int)),this, SLOT(onElemTypeChange(int)));
 272   connect(myDlg,                 SIGNAL(rejectedDlg()),           this, SLOT(onRejectedDlg()));
 273 }
 274
 275 //=======================================================================
 276 // function : startOperation()
 277 // purpose  : Init dialog fields, connect signals and slots, show dialog
 278 //=======================================================================
 279 void SMESHGUI_FindElemByPointOp::startOperation()
 280 {
 281   // init simulation with a current View
 282   if ( mySimulation ) delete mySimulation;
 283   mySMESHGUI = getSMESHGUI();
 284   mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( mySMESHGUI ) );
 285   connect(mySMESHGUI, SIGNAL (SignalActivatedViewManager()), this, SLOT(onOpenView()));
 286   connect(mySMESHGUI, SIGNAL (SignalCloseView()), this, SLOT(onCloseView()));
 287   vtkProperty* aProp = vtkProperty::New();
 288   aProp->SetRepresentationToWireframe();
 289   aProp->SetColor(250, 0, 250);
 290   aProp->SetPointSize(5);
 291   aProp->SetLineWidth( SMESH::GetFloat("SMESH:element_width",1) + 1);
 292   mySimulation->GetActor()->SetProperty(aProp);
 293   aProp->Delete();
 294
 295   SMESHGUI_SelectionOp::startOperation();
 296   myDlg->show();
 297   redisplayPreview();
 298
 299   onSelectionDone(); // init myMeshOrPart
 300 }
 301
 302 //================================================================================
 303 /*!
 304  * \brief Stops operation
 305  */
 306 //================================================================================
 307
 308 void SMESHGUI_FindElemByPointOp::stopOperation()
 309 {
 310   if ( mySimulation )
 311   {
 312     mySimulation->SetVisibility(false);
 313     delete mySimulation;
 314     mySimulation = 0;
 315   }
 316   disconnect(mySMESHGUI, SIGNAL (SignalActivatedViewManager()), this, SLOT(onOpenView()));
 317   disconnect(mySMESHGUI, SIGNAL (SignalCloseView()), this, SLOT(onCloseView()));
 318   selectionMgr()->removeFilter( myFilter );
 319   SMESHGUI_SelectionOp::stopOperation();
 320 }
 321
 322 //=================================================================================
 323 /*!
 324  * \brief SLOT called when the viewer opened
 325  */
 326 //=================================================================================
 327 void SMESHGUI_FindElemByPointOp::onOpenView()
 328 {
 329   if ( mySimulation ) {
 330     mySimulation->SetVisibility(false);
 331   }
 332   else {
 333     mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( mySMESHGUI ));
 334   }
 335 }
 336
 337 //=================================================================================
 338 /*!
 339  * \brief SLOT called when the viewer closed
 340  */
 341 //=================================================================================
 342 void SMESHGUI_FindElemByPointOp::onCloseView()
 343 {
 344   delete mySimulation;
 345   mySimulation = 0;
 346 }
 347 //================================================================================
 348 /*!
 349  * \brief highlight found selected elements
 350  */
 351 //================================================================================
 352
 353 void SMESHGUI_FindElemByPointOp::onElemSelected()
 354 {
 355   if ( !myMeshIO.IsNull() )
 356   {
 357     Selection_Mode selMode =
 358       myDlg->myElemTypeCombo->currentId().toInt() == int(SMESH::NODE) ? NodeSelection : CellSelection;
 359     if ( selectionMode() != selMode )
 360       setSelectionMode( selMode );
 361
 362     QList<QListWidgetItem *> ids = myDlg->myFoundList->selectedItems();
 363     QList<QListWidgetItem*>::iterator id = ids.begin();
 364     SVTK_TVtkIDsMap idMap;
 365     for ( ; id != ids.end(); ++id )
 366       idMap.Add( (*id)->text().toInt() );
 367
 368     addOrRemoveIndex( myMeshIO, idMap, false );
 369
 370     SALOME_ListIO aList;
 371     aList.Append(myMeshIO);
 372     selectionMgr()->setSelectedObjects(aList,false);
 373   }
 374 }
 375
 376 //================================================================================
 377 /*!
 378  * \brief Set selection mode according to element type
 379  */
 380 //================================================================================
 381
 382 void SMESHGUI_FindElemByPointOp::onElemTypeChange(int index)
 383 {
 384   Selection_Mode newMode = (index == 1) ? NodeSelection : CellSelection;
 385   if ( selectionMode() != newMode )
 386   {
 387     selectionMgr()->clearFilters();
 388     setSelectionMode( newMode );
 389   }
 390   myDlg->myFoundList->clear();
 391 }
 392
 393 //================================================================================
 394 /*!
 395  * \brief Method needed for internal cuisine
 396  */
 397 //================================================================================
 398
 399 void SMESHGUI_FindElemByPointDlg::reject()
 400 {
 401   emit rejectedDlg();
 402   QtxDialog::reject();
 403 }
 404
 405 //================================================================================
 406 /*!
 407  * \brief Method needed for internal cuisine
 408  */
 409 //================================================================================
 410
 411 void SMESHGUI_FindElemByPointOp::onRejectedDlg()
 412 {
 413   myMeshIO.Nullify();
 414 }
 415
 416 //================================================================================
 417 /*!
 418  * \brief perform it's intention action: find elements
 419  */
 420 //================================================================================
 421
 422 void SMESHGUI_FindElemByPointOp::onFind()
 423 {
 424   if ( myMeshOrPart->_is_nil() )
 425     return;
 426
 427   try {
 428     SUIT_OverrideCursor wc;
 429
 430     SMESH::SMESH_Mesh_var aMesh = myMeshOrPart->GetMesh();
 431     if ( aMesh->_is_nil() )
 432       return;
 433     SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
 434     if (aMeshEditor->_is_nil())
 435       return;
 436
 437     SMESH::smIdType_array_var foundIds;
 438     if ( aMesh->_is_equivalent( myMeshOrPart ) )
 439       foundIds =
 440         aMeshEditor->FindElementsByPoint( myDlg->myX->GetValue(),
 441                                           myDlg->myY->GetValue(),
 442                                           myDlg->myZ->GetValue(),
 443                                           SMESH::ElementType( myDlg->myElemTypeCombo->currentId().toInt()));
 444     else
 445       foundIds =
 446         aMeshEditor->FindAmongElementsByPoint( myMeshOrPart,
 447                                                myDlg->myX->GetValue(),
 448                                                myDlg->myY->GetValue(),
 449                                                myDlg->myZ->GetValue(),
 450                                                SMESH::ElementType( myDlg->myElemTypeCombo->currentId().toInt()));
 451     myDlg->myFoundList->clear();
 452     for ( int i = 0; i < (int) foundIds->length(); ++i )
 453       myDlg->myFoundList->addItem( QString::number( foundIds[i] ));
 454
 455     if ( foundIds->length() > 0 )
 456       myDlg->myFoundList->setCurrentRow(0);
 457   }
 458   catch (const SALOME::SALOME_Exception& S_ex) {
 459     SalomeApp_Tools::QtCatchCorbaException(S_ex);
 460   }
 461   catch (...) {
 462   }
 463 }
 464
 465 //================================================================================
 466 /*!
 467  * \brief Method needed for internal cuisine
 468  */
 469 //================================================================================
 470
 471 bool SMESHGUI_FindElemByPointOp::onApply()
 472 {
 473   onRejectedDlg();
 474   return true;
 475 }
 476
 477 //================================================================================
 478 /*!
 479  * \brief SLOT called when selection changed
 480  */
 481 //================================================================================
 482
 483 void SMESHGUI_FindElemByPointOp::onSelectionDone()
 484 {
 485   if ( !myDlg->isVisible() || !myDlg->isEnabled() )
 486     return;
 487
 488   QString oldMeshEntry, newMeshEntry;
 489   if ( !myMeshIO.IsNull() && myMeshIO->hasEntry() )
 490     oldMeshEntry = myMeshIO->getEntry();
 491
 492   try {
 493     SALOME_ListIO aList;
 494     selectionMgr()->selectedObjects(aList, SVTK_Viewer::Type());
 495     if (aList.Extent() == 1 && aList.First()->hasEntry())
 496     {
 497       Handle(SALOME_InteractiveObject) anIO = aList.First();
 498       _PTR(SObject) pObj = SMESH::getStudy()->FindObjectID(anIO->getEntry());
 499       CORBA::Object_var anObj = SMESH::IObjectToObject( anIO );
 500       newMeshEntry = anIO->getEntry();
 501       SMESH::SMESH_IDSource_var aMeshOrPart = SMESH::SMESH_IDSource::_narrow(anObj);
 502       if ( pObj && !aMeshOrPart->_is_nil() && oldMeshEntry != newMeshEntry )
 503       {
 504         myMeshOrPart = aMeshOrPart;
 505         myMeshIO.Nullify();
 506         myMeshIO = anIO;
 507         std::string name = pObj->GetName();
 508         myDlg->myMeshName->setText("");
 509         myDlg->myMeshName->setText( QString( name.c_str() ).trimmed() );
 510         SMESH::array_of_ElementType_var  types = myMeshOrPart->GetTypes();
 511         myDlg->setTypes( types );
 512       }
 513     }
 514   }
 515   catch (...) {
 516   }
 517
 518   if ( oldMeshEntry != newMeshEntry || newMeshEntry.isEmpty() )
 519     myDlg->myFoundList->clear();
 520
 521   myDlg->myFindBtn->setEnabled( !myMeshIO.IsNull() );
 522 }
 523
 524 //================================================================================
 525 /*!
 526  * \brief show point by coordinates
 527  */
 528 //================================================================================
 529
 530 void SMESHGUI_FindElemByPointOp::redisplayPreview()
 531 {
 532   myDlg->myFoundList->clear();
 533
 534   myPreview->nodesXYZ[0].x = myDlg->myX->GetValue();
 535   myPreview->nodesXYZ[0].y = myDlg->myY->GetValue();
 536   myPreview->nodesXYZ[0].z = myDlg->myZ->GetValue();
 537   if (!mySimulation)
 538     mySimulation = new SMESHGUI_MeshEditPreview(SMESH::GetViewWindow( mySMESHGUI ));
 539   mySimulation->SetData( myPreview.in());
 540 }
 541
 542 //================================================================================
 543 /*!
 544  * \brief install filter on meshes
 545  */
 546 //================================================================================
 547
 548 void SMESHGUI_FindElemByPointOp::activateSelection()
 549 {
 550   selectionMgr()->clearFilters();
 551   selectionMgr()->installFilter( myFilter );
 552 }
 553
 554 //================================================================================
 555 /*!
 556  * \brief Destructor
 557 */
 558 //================================================================================
 559
 560 SMESHGUI_FindElemByPointOp::~SMESHGUI_FindElemByPointOp()
 561 {
 562   if ( myDlg )         { delete myDlg;        myDlg = 0; }
 563   if ( mySimulation )  { delete mySimulation; mySimulation = 0; }
 564   if ( myFilter )
 565   {
 566     QList<SUIT_SelectionFilter*> sub =  ((SMESH_LogicalFilter*)myFilter)->getFilters();
 567     delete sub.front();
 568     delete sub.back();
 569     delete myFilter;
 570     myFilter = 0;
 571   }
 572 }
 573
 574 //================================================================================
 575 /*!
 576  * \brief Gets dialog of this operation
 577  * \retval LightApp_Dialog* - pointer to dialog of this operation
 578  */
 579 //================================================================================
 580
 581 LightApp_Dialog* SMESHGUI_FindElemByPointOp::dlg() const
 582 {
 583   return myDlg;
 584 }
 585