src/SMESHGUI/SMESHGUI_SingleEditDlg.cxx

   1 // Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   3 //
   4 // This library is free software; you can redistribute it and/or
   5 // modify it under the terms of the GNU Lesser General Public
   6 // License as published by the Free Software Foundation; either
   7 // version 2.1 of the License.
   8 //
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 // Lesser General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Lesser General Public
  15 // License along with this library; if not, write to the Free Software
  16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 //
  18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 //
  20 // File   : SMESHGUI_SingleEditDlg.cxx
  21 // Author : Sergey LITONIN, Open CASCADE S.A.S.
  22 //
  23
  24 // SMESH includes
  25 #include "SMESHGUI_SingleEditDlg.h"
  26
  27 #include "SMESHGUI.h"
  28 #include "SMESHGUI_Utils.h"
  29 #include "SMESHGUI_VTKUtils.h"
  30 #include "SMESHGUI_MeshUtils.h"
  31
  32 #include <SMESH_Actor.h>
  33 #include <SMDS_Mesh.hxx>
  34
  35 // SALOME GUI includes
  36 #include <LightApp_SelectionMgr.h>
  37 #include <LightApp_Application.h>
  38 #include <SUIT_ResourceMgr.h>
  39 #include <SUIT_MessageBox.h>
  40 #include <SUIT_Desktop.h>
  41 #include <SUIT_Session.h>
  42
  43 #include <SVTK_Selector.h>
  44 #include <SVTK_ViewWindow.h>
  45 #include <SALOME_ListIO.hxx>
  46
  47 // OCCT includes
  48 #include <TColStd_MapOfInteger.hxx>
  49 #include <TColStd_IndexedMapOfInteger.hxx>
  50
  51 // Qt includes
  52 #include <QVBoxLayout>
  53 #include <QHBoxLayout>
  54 #include <QLineEdit>
  55 #include <QPushButton>
  56 #include <QGroupBox>
  57 #include <QLabel>
  58 #include <QValidator>
  59 #include <QKeyEvent>
  60
  61 #define SPACING 6
  62 #define MARGIN  11
  63
  64 /*!
  65   \class BusyLocker
  66   \brief Simple 'busy state' flag locker.
  67   \internal
  68 */
  69
  70 class BusyLocker
  71 {
  72 public:
  73   //! Constructor. Sets passed boolean flag to \c true.
  74   BusyLocker( bool& busy ) : myBusy( busy ) { myBusy = true; }
  75   //! Destructor. Clear external boolean flag passed as parameter to the constructor to \c false.
  76   ~BusyLocker() { myBusy = false; }
  77 private:
  78   bool& myBusy; //! External 'busy state' boolean flag
  79 };
  80
  81 /*!
  82  *  Class       : SMESHGUI_SingleEditDlg
  83  *  Description : Inversion of the diagonal of a pseudo-quadrangle formed by
  84  *                2 neighboring triangles with 1 common edge
  85  */
  86
  87 //=======================================================================
  88 // name    : SMESHGUI_SingleEditDlg()
  89 // Purpose : Constructor
  90 //=======================================================================
  91 SMESHGUI_SingleEditDlg
  92 ::SMESHGUI_SingleEditDlg(SMESHGUI* theModule)
  93   : QDialog(SMESH::GetDesktop(theModule)),
  94     mySelector(SMESH::GetViewWindow(theModule)->GetSelector()),
  95     mySelectionMgr(SMESH::GetSelectionMgr(theModule)),
  96     mySMESHGUI(theModule)
  97 {
  98   setModal(false);
  99
 100   QVBoxLayout* aDlgLay = new QVBoxLayout(this);
 101   aDlgLay->setMargin(MARGIN);
 102   aDlgLay->setSpacing(SPACING);
 103
 104   QWidget* aMainFrame = createMainFrame  (this);
 105   QWidget* aBtnFrame  = createButtonFrame(this);
 106
 107   aDlgLay->addWidget(aMainFrame);
 108   aDlgLay->addWidget(aBtnFrame);
 109
 110   Init();
 111 }
 112
 113 //=======================================================================
 114 // name    : createMainFrame()
 115 // Purpose : Create frame containing dialog's input fields
 116 //=======================================================================
 117 QWidget* SMESHGUI_SingleEditDlg::createMainFrame (QWidget* theParent)
 118 {
 119   QGroupBox* aMainGrp = new QGroupBox(tr("EDGE_BETWEEN"), theParent);
 120   QHBoxLayout* aLay = new QHBoxLayout(aMainGrp);
 121   aLay->setMargin(MARGIN);
 122   aLay->setSpacing(SPACING);
 123
 124   QPixmap aPix (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
 125
 126   QLabel* aLab = new QLabel(tr("SMESH_EDGE"), aMainGrp);
 127   QPushButton* aBtn = new QPushButton(aMainGrp);
 128   aBtn->setIcon(aPix);
 129   myEdge = new QLineEdit(aMainGrp);
 130   myEdge->setValidator(new QRegExpValidator(QRegExp("[\\d]*-[\\d]*"), this));
 131
 132   aLay->addWidget(aLab);
 133   aLay->addWidget(aBtn);
 134   aLay->addWidget(myEdge);
 135
 136   return aMainGrp;
 137 }
 138
 139 //=======================================================================
 140 // name    : createButtonFrame()
 141 // Purpose : Create frame containing buttons
 142 //=======================================================================
 143 QWidget* SMESHGUI_SingleEditDlg::createButtonFrame (QWidget* theParent)
 144 {
 145   QGroupBox* aFrame = new QGroupBox(theParent);
 146
 147   myOkBtn     = new QPushButton(tr("SMESH_BUT_APPLY_AND_CLOSE"), aFrame);
 148   myApplyBtn  = new QPushButton(tr("SMESH_BUT_APPLY"), aFrame);
 149   myCloseBtn  = new QPushButton(tr("SMESH_BUT_CLOSE"), aFrame);
 150   myHelpBtn   = new QPushButton(tr("SMESH_BUT_HELP"),  aFrame);
 151
 152   QHBoxLayout* aLay = new QHBoxLayout(aFrame);
 153   aLay->setMargin(MARGIN);
 154   aLay->setSpacing(SPACING);
 155
 156   aLay->addWidget(myOkBtn);
 157   aLay->addSpacing(10);
 158   aLay->addWidget(myApplyBtn);
 159   aLay->addSpacing(10);
 160   aLay->addStretch();
 161   aLay->addWidget(myCloseBtn);
 162   aLay->addWidget(myHelpBtn);
 163
 164   return aFrame;
 165 }
 166
 167 //=======================================================================
 168 // name    : isValid()
 169 // Purpose : Verify validity of input data
 170 //=======================================================================
 171 bool SMESHGUI_SingleEditDlg::isValid (const bool theMess) const
 172 {
 173   int id1, id2;
 174   return getNodeIds(myEdge->text(), id1, id2);
 175 }
 176
 177 //=======================================================================
 178 // name    : getNodeIds()
 179 // Purpose : Retrieve node ids from string
 180 //=======================================================================
 181 bool SMESHGUI_SingleEditDlg::getNodeIds (const QString& theStr,
 182                                          int& theId1, int&  theId2) const
 183 {
 184   if (!theStr.contains('-'))
 185     return false;
 186
 187   bool ok1, ok2;
 188   QString str1 = theStr.section('-', 0, 0, QString::SectionSkipEmpty);
 189   QString str2 = theStr.section('-', 1, 1, QString::SectionSkipEmpty);
 190   theId1 = str1.toInt(&ok1);
 191   theId2 = str2.toInt(&ok2);
 192
 193   return ok1 & ok2;
 194 }
 195
 196 //=======================================================================
 197 // name    : ~SMESHGUI_SingleEditDlg()
 198 // Purpose : Destructor
 199 //=======================================================================
 200 SMESHGUI_SingleEditDlg::~SMESHGUI_SingleEditDlg()
 201 {
 202 }
 203
 204 //=======================================================================
 205 // name    : Init()
 206 // Purpose : Init dialog fields, connect signals and slots, show dialog
 207 //=======================================================================
 208 void SMESHGUI_SingleEditDlg::Init()
 209 {
 210   mySMESHGUI->SetActiveDialogBox((QDialog*)this);
 211   myBusy = false;
 212   myActor = 0;
 213
 214   // main buttons
 215   connect(myOkBtn,    SIGNAL(clicked()), SLOT(onOk()));
 216   connect(myCloseBtn, SIGNAL(clicked()), SLOT(onClose()));
 217   connect(myApplyBtn, SIGNAL(clicked()), SLOT(onApply()));
 218   connect(myHelpBtn,  SIGNAL(clicked()), SLOT(onHelp()));
 219
 220   // selection and SMESHGUI
 221   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));
 222   connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), SLOT(onDeactivate()));
 223   connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()), SLOT(onClose()));
 224   connect(myEdge, SIGNAL(textChanged(const QString&)), SLOT(onTextChange(const QString&)));
 225
 226   myOkBtn->setEnabled(false);
 227   myApplyBtn->setEnabled(false);
 228   setEnabled(true);
 229
 230   // set selection mode
 231   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 232     aViewWindow->SetSelectionMode(EdgeOfCellSelection);
 233
 234   onSelectionDone();
 235 }
 236
 237 //=======================================================================
 238 // name    : onOk()
 239 // Purpose : SLOT called when "Ok" button pressed.
 240 //           Assign filters VTK viewer and close dialog
 241 //=======================================================================
 242 void SMESHGUI_SingleEditDlg::onOk()
 243 {
 244   if (onApply())
 245     onClose();
 246 }
 247
 248 //=======================================================================
 249 // name    : onClose()
 250 // Purpose : SLOT called when "Close" button pressed. Close dialog
 251 //=======================================================================
 252 void SMESHGUI_SingleEditDlg::onClose()
 253 {
 254   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 255     aViewWindow->SetSelectionMode(ActorSelection);
 256   //mySelectionMgr->clearSelected();
 257   disconnect(mySelectionMgr, 0, this, 0);
 258   disconnect(mySMESHGUI, 0, this, 0);
 259   mySMESHGUI->ResetState();
 260   reject();
 261 }
 262
 263 //=================================================================================
 264 // function : onHelp()
 265 // purpose  :
 266 //=================================================================================
 267 void SMESHGUI_SingleEditDlg::onHelp()
 268 {
 269   LightApp_Application* app = (LightApp_Application*)(SUIT_Session::session()->activeApplication());
 270   if (app)
 271     app->onHelpContextModule(mySMESHGUI ? app->moduleName(mySMESHGUI->moduleName()) : QString(""), myHelpFileName);
 272   else {
 273     QString platform;
 274 #ifdef WIN32
 275     platform = "winapplication";
 276 #else
 277     platform = "application";
 278 #endif
 279     SUIT_MessageBox::warning(this, tr("WRN_WARNING"),
 280                              tr("EXTERNAL_BROWSER_CANNOT_SHOW_PAGE").
 281                              arg(app->resourceMgr()->stringValue("ExternalBrowser",
 282                                                                  platform)).
 283                              arg(myHelpFileName));
 284   }
 285 }
 286
 287 //=======================================================================
 288 //function : findTriangles()
 289 //purpose  : find triangles sharing theNode1-theNode2 link
 290 //           THIS IS A PIECE OF SMESH_MeshEditor.cxx
 291 //           TO DO: make it available in SMDS for ex.
 292 //=======================================================================
 293 static bool findTriangles (const SMDS_MeshNode *    theNode1,
 294                            const SMDS_MeshNode *    theNode2,
 295                            const SMDS_MeshElement*& theTria1,
 296                            const SMDS_MeshElement*& theTria2)
 297 {
 298   if (!theNode1 || !theNode2) return false;
 299
 300   theTria1 = theTria2 = 0;
 301
 302   std::set< const SMDS_MeshElement* > emap;
 303   SMDS_ElemIteratorPtr it = theNode1->GetInverseElementIterator();
 304   while (it->more()) {
 305     const SMDS_MeshElement* elem = it->next();
 306     if (elem->GetType() == SMDSAbs_Face && elem->NbNodes() == 3)
 307       emap.insert(elem);
 308   }
 309   it = theNode2->GetInverseElementIterator();
 310   while (it->more()) {
 311     const SMDS_MeshElement* elem = it->next();
 312     if (elem->GetType() == SMDSAbs_Face &&
 313          emap.find(elem) != emap.end())
 314       if (theTria1) {
 315         theTria2 = elem;
 316         break;
 317       } else {
 318         theTria1 = elem;
 319       }
 320   }
 321   return (theTria1 && theTria2);
 322 }
 323
 324 //=======================================================================
 325 //function : onTextChange()
 326 //purpose  :
 327 //=======================================================================
 328 void SMESHGUI_SingleEditDlg::onTextChange (const QString& theNewText)
 329 {
 330   if (myBusy) return;
 331   BusyLocker lock(myBusy);
 332
 333   myOkBtn->setEnabled(false);
 334   myApplyBtn->setEnabled(false);
 335
 336   // hilight entered edge
 337   if(myActor){
 338     if(SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh()){
 339       Handle(SALOME_InteractiveObject) anIO = myActor->getIO();
 340       SALOME_ListIO aList;
 341       aList.Append(anIO);
 342       mySelectionMgr->setSelectedObjects(aList,false);
 343
 344       TColStd_IndexedMapOfInteger selectedIndices;
 345       TColStd_MapOfInteger newIndices;
 346       mySelector->GetIndex(anIO,selectedIndices);
 347
 348       int id1, id2;
 349       if ( !getNodeIds(myEdge->text(), id1, id2) )
 350         return;
 351
 352       const SMDS_MeshNode* aNode1 = aMesh->FindNode( id1 );
 353       const SMDS_MeshNode* aNode2 = aMesh->FindNode( id2 );
 354
 355       if ( !aNode1 || !aNode2 || aNode1 == aNode2 )
 356         return;
 357
 358       // find a triangle and an edge index
 359       const SMDS_MeshElement* tria1;
 360       const SMDS_MeshElement* tria2;
 361
 362       if ( findTriangles(aNode1,aNode2,tria1,tria2) )
 363       {
 364         newIndices.Add(tria1->GetID());
 365
 366         const SMDS_MeshNode* a3Nodes[3];
 367         SMDS_ElemIteratorPtr it;
 368         int edgeInd = 2, i;
 369         for (i = 0, it = tria1->nodesIterator(); it->more(); i++) {
 370           a3Nodes[ i ] = static_cast<const SMDS_MeshNode*>(it->next());
 371           if (i > 0 && ( a3Nodes[ i ] == aNode1 && a3Nodes[ i - 1] == aNode2 ||
 372                          a3Nodes[ i ] == aNode2 && a3Nodes[ i - 1] == aNode1 ) ) {
 373             edgeInd = i - 1;
 374             break;
 375           }
 376         }
 377         newIndices.Add(-edgeInd-1);
 378
 379         myOkBtn->setEnabled(true);
 380         myApplyBtn->setEnabled(true);
 381       }
 382       mySelector->AddOrRemoveIndex(anIO,newIndices, false);
 383       SMESH::GetViewWindow(mySMESHGUI)->highlight( anIO, true, true );
 384     }
 385   }
 386 }
 387
 388 //=======================================================================
 389 // name    : onSelectionDone()
 390 // Purpose : SLOT called when selection changed
 391 //=======================================================================
 392 void SMESHGUI_SingleEditDlg::onSelectionDone()
 393 {
 394   if (myBusy) return;
 395   BusyLocker lock(myBusy);
 396
 397   int anId1 = 0, anId2 = 0;
 398
 399   myOkBtn->setEnabled(false);
 400   myApplyBtn->setEnabled(false);
 401
 402   SALOME_ListIO aList;
 403   mySelectionMgr->selectedObjects(aList);
 404
 405   if (aList.Extent() != 1) {
 406     myEdge->clear();
 407     return;
 408   }
 409
 410   Handle(SALOME_InteractiveObject) anIO = aList.First();
 411   myActor = SMESH::FindActorByEntry(anIO->getEntry());
 412   if(myActor){
 413     TVisualObjPtr aVisualObj = myActor->GetObject();
 414     if(SMDS_Mesh* aMesh = aVisualObj->GetMesh())
 415     {
 416       const SMDS_MeshElement* tria[2];
 417       if( SMESH::GetEdgeNodes( mySelector, aVisualObj, anId1, anId2 ) >= 1 &&
 418           findTriangles( aMesh->FindNode( anId1 ), aMesh->FindNode( anId2 ), tria[0],tria[1] ) )
 419       {
 420         QString aText = QString("%1-%2").arg(anId1).arg(anId2);
 421         myEdge->setText(aText);
 422
 423         myOkBtn->setEnabled(true);
 424         myApplyBtn->setEnabled(true);
 425       }
 426       else
 427       {
 428         myEdge->clear();
 429       }
 430     }
 431   }
 432 }
 433
 434 //=======================================================================
 435 // name    : onDeactivate()
 436 // Purpose : SLOT called when dialog must be deativated
 437 //=======================================================================
 438 void SMESHGUI_SingleEditDlg::onDeactivate()
 439 {
 440   setEnabled(false);
 441 }
 442
 443 //=======================================================================
 444 // name    : enterEvent()
 445 // Purpose : Event filter
 446 //=======================================================================
 447 void SMESHGUI_SingleEditDlg::enterEvent (QEvent*)
 448 {
 449   if (!isEnabled()) {
 450     mySMESHGUI->EmitSignalDeactivateDialog();
 451     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 452       aViewWindow->SetSelectionMode(EdgeOfCellSelection);
 453     setEnabled(true);
 454   }
 455 }
 456
 457 //=================================================================================
 458 // function : closeEvent()
 459 // purpose  :
 460 //=================================================================================
 461 void SMESHGUI_SingleEditDlg::closeEvent (QCloseEvent*)
 462 {
 463   onClose();
 464 }
 465
 466 //=======================================================================
 467 //function : hideEvent()
 468 //purpose  : caused by ESC key
 469 //=======================================================================
 470 void SMESHGUI_SingleEditDlg::hideEvent (QHideEvent*)
 471 {
 472   if (!isMinimized())
 473     onClose();
 474 }
 475
 476 //=================================================================================
 477 // function : onApply()
 478 // purpose  : SLOT. Called when apply button is pressed
 479 //=================================================================================
 480 bool SMESHGUI_SingleEditDlg::onApply()
 481 {
 482   if (mySMESHGUI->isActiveStudyLocked())
 483     return false;
 484   // verify validity of input data
 485   if (!isValid(true))
 486     return false;
 487
 488   // get mesh, actor and nodes
 489   SALOME_ListIO aList;
 490   mySelectionMgr->selectedObjects(aList);
 491
 492   SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(aList.First());
 493
 494   if (aMesh->_is_nil()) {
 495     SUIT_MessageBox::information(SMESH::GetDesktop(mySMESHGUI),
 496                                  tr("SMESH_ERROR"),
 497                                  tr("SMESHG_NO_MESH"));
 498     return false;
 499   }
 500
 501   SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
 502   int anId1= 0, anId2 = 0;
 503   if (aMeshEditor->_is_nil() || !getNodeIds(myEdge->text(), anId1, anId2))
 504     return false;
 505
 506   // perform operation
 507   bool aResult = process(aMeshEditor.in(), anId1, anId2);
 508
 509   // update actor
 510   if (aResult) {
 511     mySelector->ClearIndex();
 512     mySelectionMgr->setSelectedObjects(aList, false);
 513     onSelectionDone();
 514     SMESH::UpdateView();
 515   }
 516
 517   return aResult;
 518 }
 519
 520 //=================================================================================
 521 // function : keyPressEvent()
 522 // purpose  :
 523 //=================================================================================
 524 void SMESHGUI_SingleEditDlg::keyPressEvent( QKeyEvent* e )
 525 {
 526   QDialog::keyPressEvent( e );
 527   if ( e->isAccepted() )
 528     return;
 529
 530   if ( e->key() == Qt::Key_F1 ) {
 531     e->accept();
 532     onHelp();
 533   }
 534 }
 535
 536 /*!
 537  *  Class       : SMESHGUI_TrianglesInversionDlg
 538  *  Description : Inversion of the diagonal of a pseudo-quadrangle formed by
 539  *                2 neighboring triangles with 1 common edge
 540  */
 541
 542 SMESHGUI_TrianglesInversionDlg
 543 ::SMESHGUI_TrianglesInversionDlg(SMESHGUI* theModule)
 544 : SMESHGUI_SingleEditDlg(theModule)
 545 {
 546   setWindowTitle(tr("CAPTION"));
 547   myHelpFileName = "diagonal_inversion_of_elements_page.html";
 548 }
 549
 550 SMESHGUI_TrianglesInversionDlg::~SMESHGUI_TrianglesInversionDlg()
 551 {
 552 }
 553
 554 bool SMESHGUI_TrianglesInversionDlg::process (SMESH::SMESH_MeshEditor_ptr theMeshEditor,
 555                                               const int theId1, const int theId2)
 556 {
 557   return theMeshEditor->InverseDiag(theId1, theId2);
 558 }
 559
 560 /*!
 561  *  Class       : SMESHGUI_UnionOfTwoTrianglesDlg
 562  *  Description : Construction of a quadrangle by deletion of the
 563  *                common border of 2 neighboring triangles
 564  */
 565
 566 SMESHGUI_UnionOfTwoTrianglesDlg
 567 ::SMESHGUI_UnionOfTwoTrianglesDlg(SMESHGUI* theModule)
 568 : SMESHGUI_SingleEditDlg(theModule)
 569 {
 570   setWindowTitle(tr("CAPTION"));
 571   myHelpFileName = "uniting_two_triangles_page.html";
 572 }
 573
 574 SMESHGUI_UnionOfTwoTrianglesDlg::~SMESHGUI_UnionOfTwoTrianglesDlg()
 575 {
 576 }
 577
 578 bool SMESHGUI_UnionOfTwoTrianglesDlg::process (SMESH::SMESH_MeshEditor_ptr theMeshEditor,
 579                                                const int theId1, const int theId2)
 580 {
 581   return theMeshEditor->DeleteDiag(theId1, theId2);
 582 }