src/SMESHGUI/SMESHGUI_SingleEditDlg.cxx

   1 //  SMESH SMESHGUI : GUI for SMESH component
   2 //
   3 //  Copyright (C) 2003  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
  21 //
  22 //
  23 //
  24 //  File   : SMESHGUI_SingleEditDlg.cxx
  25 //  Author : Sergey LITONIN
  26 //  Module : SMESH
  27
  28 #include "SMESHGUI_SingleEditDlg.h"
  29
  30 #include "SMESHGUI.h"
  31 #include "SMESHGUI_Utils.h"
  32 #include "SMESHGUI_VTKUtils.h"
  33 #include "SMESHGUI_MeshUtils.h"
  34 #include "SMESHGUI_SpinBox.h"
  35
  36 #include "SMESH_Actor.h"
  37 #include "SMDS_Mesh.hxx"
  38
  39 #include "LightApp_SelectionMgr.h"
  40 #include "SUIT_ResourceMgr.h"
  41 #include "SUIT_MessageBox.h"
  42 #include "SUIT_Desktop.h"
  43
  44 #include "SVTK_Selector.h"
  45 #include "SVTK_ViewWindow.h"
  46 #include "SALOME_ListIO.hxx"
  47
  48 #include "utilities.h"
  49
  50 // OCCT Includes
  51 #include <TColStd_MapOfInteger.hxx>
  52 #include <TColStd_IndexedMapOfInteger.hxx>
  53
  54 // QT Includes
  55 #include <qframe.h>
  56 #include <qlayout.h>
  57 #include <qlineedit.h>
  58 #include <qpushbutton.h>
  59 #include <qgroupbox.h>
  60 #include <qlabel.h>
  61 #include <qmessagebox.h>
  62 #include <qvalidator.h>
  63
  64
  65 #define SPACING 5
  66 #define MARGIN  10
  67
  68
  69 /*!
  70  *  Class       : SMESHGUI_DiagValidator
  71  *  Description : validate munual input of edge like "id1-id2"
  72  */
  73 class SMESHGUI_DiagValidator: public QValidator
  74 {
  75  public:
  76   SMESHGUI_DiagValidator (QWidget * parent, const char * name = 0):
  77     QValidator(parent,name) {}
  78
  79   State validate (QString & text, int & pos) const
  80   {
  81     text.stripWhiteSpace();
  82     text.replace(QRegExp("[^0-9]+"), "-");
  83     if (text == "-")
  84       text = "";
  85     int ind = text.find(QRegExp("-[0-9]+-"));
  86     if (ind > 0) { // leave only two ids
  87       ind = text.find('-', ind + 1);
  88       if (ind > 0)
  89         text.truncate(ind);
  90     }
  91     if (pos > text.length())
  92       pos = text.length();
  93     return Acceptable;
  94   }
  95 };
  96
  97 /*!
  98  *  Class       : SMESHGUI_SingleEditDlg
  99  *  Description : Inversion of the diagonal of a pseudo-quadrangle formed by
 100  *                2 neighboring triangles with 1 common edge
 101  */
 102
 103 //=======================================================================
 104 // name    : SMESHGUI_SingleEditDlg()
 105 // Purpose : Constructor
 106 //=======================================================================
 107 SMESHGUI_SingleEditDlg
 108 ::SMESHGUI_SingleEditDlg(SMESHGUI* theModule,
 109                          const char* theName):
 110   QDialog(SMESH::GetDesktop(theModule),
 111           theName,
 112           false,
 113           WStyle_Customize | WStyle_NormalBorder | WStyle_Title | WStyle_SysMenu),
 114     mySelector(SMESH::GetViewWindow(theModule)->GetSelector()),
 115     mySelectionMgr(SMESH::GetSelectionMgr(theModule)),
 116     mySMESHGUI(theModule)
 117 {
 118   QVBoxLayout* aDlgLay = new QVBoxLayout(this, MARGIN, SPACING);
 119
 120   QFrame* aMainFrame = createMainFrame  (this);
 121   QFrame* aBtnFrame  = createButtonFrame(this);
 122
 123   aDlgLay->addWidget(aMainFrame);
 124   aDlgLay->addWidget(aBtnFrame);
 125
 126   aDlgLay->setStretchFactor(aMainFrame, 1);
 127
 128   Init();
 129 }
 130
 131 //=======================================================================
 132 // name    : createMainFrame()
 133 // Purpose : Create frame containing dialog's input fields
 134 //=======================================================================
 135 QFrame* SMESHGUI_SingleEditDlg::createMainFrame (QWidget* theParent)
 136 {
 137   QGroupBox* aMainGrp = new QGroupBox(1, Qt::Vertical, tr("EDGE_BETWEEN"), theParent);
 138
 139   QPixmap aPix (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
 140
 141   new QLabel(tr("SMESH_EDGE"), aMainGrp);
 142   (new QPushButton(aMainGrp))->setPixmap(aPix);
 143   myEdge = new QLineEdit(aMainGrp);
 144   myEdge->setValidator(new SMESHGUI_DiagValidator(this, "validator"));
 145
 146   return aMainGrp;
 147 }
 148
 149 //=======================================================================
 150 // name    : createButtonFrame()
 151 // Purpose : Create frame containing buttons
 152 //=======================================================================
 153 QFrame* SMESHGUI_SingleEditDlg::createButtonFrame (QWidget* theParent)
 154 {
 155   QFrame* aFrame = new QFrame(theParent);
 156   aFrame->setFrameStyle(QFrame::Box | QFrame::Sunken);
 157
 158   myOkBtn     = new QPushButton(tr("SMESH_BUT_OK"   ), aFrame);
 159   myApplyBtn  = new QPushButton(tr("SMESH_BUT_APPLY"), aFrame);
 160   myCloseBtn  = new QPushButton(tr("SMESH_BUT_CLOSE"), aFrame);
 161
 162   QSpacerItem* aSpacer = new QSpacerItem(0, 0, QSizePolicy::Expanding, QSizePolicy::Minimum);
 163
 164   QHBoxLayout* aLay = new QHBoxLayout(aFrame, MARGIN, SPACING);
 165
 166   aLay->addWidget(myOkBtn);
 167   aLay->addWidget(myApplyBtn);
 168   aLay->addItem(aSpacer);
 169   aLay->addWidget(myCloseBtn);
 170
 171   return aFrame;
 172 }
 173
 174 //=======================================================================
 175 // name    : isValid()
 176 // Purpose : Verify validity of input data
 177 //=======================================================================
 178 bool SMESHGUI_SingleEditDlg::isValid (const bool theMess) const
 179 {
 180   int id1, id2;
 181   return getNodeIds(myEdge->text(), id1, id2);
 182 }
 183
 184 //=======================================================================
 185 // name    : getNodeIds()
 186 // Purpose : Retrieve node ids from string
 187 //=======================================================================
 188 bool SMESHGUI_SingleEditDlg::getNodeIds (const QString& theStr,
 189                                          int& theId1, int&  theId2) const
 190 {
 191   if (!theStr.contains('-'))
 192     return false;
 193
 194   bool ok1, ok2;
 195   QString str1 = theStr.section('-', 0, 0, QString::SectionSkipEmpty);
 196   QString str2 = theStr.section('-', 1, 1, QString::SectionSkipEmpty);
 197   theId1 = str1.toInt(&ok1);
 198   theId2 = str2.toInt(&ok2);
 199
 200   return ok1 & ok2;
 201 }
 202
 203 //=======================================================================
 204 // name    : ~SMESHGUI_SingleEditDlg()
 205 // Purpose : Destructor
 206 //=======================================================================
 207 SMESHGUI_SingleEditDlg::~SMESHGUI_SingleEditDlg()
 208 {
 209 }
 210
 211 //=======================================================================
 212 // name    : Init()
 213 // Purpose : Init dialog fields, connect signals and slots, show dialog
 214 //=======================================================================
 215 void SMESHGUI_SingleEditDlg::Init()
 216 {
 217   mySMESHGUI->SetActiveDialogBox((QDialog*)this);
 218   myBusy = false;
 219   myActor = 0;
 220
 221   // main buttons
 222   connect(myOkBtn,    SIGNAL(clicked()), SLOT(onOk()));
 223   connect(myCloseBtn, SIGNAL(clicked()), SLOT(onClose()));
 224   connect(myApplyBtn, SIGNAL(clicked()), SLOT(onApply()));
 225
 226   // selection and SMESHGUI
 227   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));
 228   connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), SLOT(onDeactivate()));
 229   connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()), SLOT(onClose()));
 230   connect(myEdge, SIGNAL(textChanged(const QString&)), SLOT(onTextChange(const QString&)));
 231
 232   myOkBtn->setEnabled(false);
 233   myApplyBtn->setEnabled(false);
 234   setEnabled(true);
 235
 236   this->show();
 237
 238   // set selection mode
 239   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 240     aViewWindow->SetSelectionMode(EdgeOfCellSelection);
 241
 242   onSelectionDone();
 243 }
 244
 245 //=======================================================================
 246 // name    : onOk()
 247 // Purpose : SLOT called when "Ok" button pressed.
 248 //           Assign filters VTK viewer and close dialog
 249 //=======================================================================
 250 void SMESHGUI_SingleEditDlg::onOk()
 251 {
 252   if (onApply())
 253     onClose();
 254 }
 255
 256 //=======================================================================
 257 // name    : onClose()
 258 // Purpose : SLOT called when "Close" button pressed. Close dialog
 259 //=======================================================================
 260 void SMESHGUI_SingleEditDlg::onClose()
 261 {
 262   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 263     aViewWindow->SetSelectionMode(ActorSelection);
 264   //mySelectionMgr->clearSelected();
 265   disconnect(mySelectionMgr, 0, this, 0);
 266   disconnect(mySMESHGUI, 0, this, 0);
 267   mySMESHGUI->ResetState();
 268   reject();
 269 }
 270
 271 //=======================================================================
 272 //function : findTriangles()
 273 //purpose  : find triangles sharing theNode1-theNode2 link
 274 //           THIS IS A PIECE OF SMESH_MeshEditor.cxx
 275 //           TO DO: make it available in SMDS for ex.
 276 //=======================================================================
 277 static bool findTriangles (const SMDS_MeshNode *    theNode1,
 278                            const SMDS_MeshNode *    theNode2,
 279                            const SMDS_MeshElement*& theTria1,
 280                            const SMDS_MeshElement*& theTria2)
 281 {
 282   if (!theNode1 || !theNode2) return false;
 283
 284   theTria1 = theTria2 = 0;
 285
 286   set< const SMDS_MeshElement* > emap;
 287   SMDS_ElemIteratorPtr it = theNode1->GetInverseElementIterator();
 288   while (it->more()) {
 289     const SMDS_MeshElement* elem = it->next();
 290     if (elem->GetType() == SMDSAbs_Face && elem->NbNodes() == 3)
 291       emap.insert(elem);
 292   }
 293   it = theNode2->GetInverseElementIterator();
 294   while (it->more()) {
 295     const SMDS_MeshElement* elem = it->next();
 296     if (elem->GetType() == SMDSAbs_Face &&
 297          emap.find(elem) != emap.end())
 298       if (theTria1) {
 299         theTria2 = elem;
 300         break;
 301       } else {
 302         theTria1 = elem;
 303       }
 304   }
 305   return (theTria1 && theTria2);
 306 }
 307
 308 //=======================================================================
 309 //function : onTextChange()
 310 //purpose  :
 311 //=======================================================================
 312 void SMESHGUI_SingleEditDlg::onTextChange (const QString& theNewText)
 313 {
 314   if (myBusy) return;
 315
 316   myOkBtn->setEnabled(false);
 317   myApplyBtn->setEnabled(false);
 318
 319   // hilight entered edge
 320   if(myActor){
 321     if(SMDS_Mesh* aMesh = myActor->GetObject()->GetMesh()){
 322       myBusy = true; // block onSelectionDone()
 323       Handle(SALOME_InteractiveObject) anIO = myActor->getIO();
 324       SALOME_ListIO aList;
 325       aList.Append(anIO);
 326       mySelectionMgr->setSelectedObjects(aList,false);
 327
 328       TColStd_IndexedMapOfInteger selectedIndices;
 329       TColStd_MapOfInteger newIndices;
 330       mySelector->GetIndex(anIO,selectedIndices);
 331       myBusy = false;
 332
 333       QStringList aListId = QStringList::split("-", theNewText, false);
 334       if (aListId.count() != 2)
 335         return;
 336
 337       int i;
 338       bool allOk = true;
 339       const SMDS_MeshNode* a2Nodes[2];
 340       for (i = 0; i < aListId.count(); i++) {
 341         if(const SMDS_MeshNode *aNode = aMesh->FindNode(aListId[ i ].toInt()))
 342           a2Nodes[ i ] = aNode;
 343         else
 344           allOk = false;
 345       }
 346
 347       // find a triangle and an edge nb
 348       const SMDS_MeshElement* tria[2];
 349       allOk &= a2Nodes[0] != a2Nodes[1] && findTriangles(a2Nodes[0],a2Nodes[1],tria[0],tria[1]);
 350       myBusy = true; // block onSelectionDone()
 351       if(allOk)
 352       {
 353         newIndices.Add(tria[0]->GetID());
 354
 355         const SMDS_MeshNode* a3Nodes [3];
 356         SMDS_ElemIteratorPtr it;
 357         int edgeInd = 2;
 358         for (i = 0, it = tria[0]->nodesIterator(); it->more(); i++) {
 359           a3Nodes[ i ] = static_cast<const SMDS_MeshNode*>(it->next());
 360           if (i > 0) {
 361             allOk = (a3Nodes[ i ] == a2Nodes[ 0 ] && a3Nodes[ i - 1] == a2Nodes[ 1 ]) ||
 362               (a3Nodes[ i ] == a2Nodes[ 1 ] && a3Nodes[ i - 1] == a2Nodes[ 0 ]);
 363             if (allOk) {
 364               edgeInd = i - 1;
 365               break;
 366             }
 367           }
 368         }
 369         newIndices.Add(-edgeInd-1);
 370
 371         myOkBtn->setEnabled(true);
 372         myApplyBtn->setEnabled(true);
 373       }
 374       mySelector->AddOrRemoveIndex(anIO,newIndices, false);
 375       SMESH::GetViewWindow(mySMESHGUI)->highlight( anIO, true, true );
 376
 377       myBusy = false;
 378     }
 379   }
 380 }
 381
 382 //=======================================================================
 383 // name    : onSelectionDone()
 384 // Purpose : SLOT called when selection changed
 385 //=======================================================================
 386 void SMESHGUI_SingleEditDlg::onSelectionDone()
 387 {
 388   if (myBusy) return;
 389
 390   int anId1 = 0, anId2 = 0;
 391
 392   myOkBtn->setEnabled(false);
 393   myApplyBtn->setEnabled(false);
 394
 395   SALOME_ListIO aList;
 396   mySelectionMgr->selectedObjects(aList);
 397
 398   if (aList.Extent() != 1) {
 399     myEdge->clear();
 400     return;
 401   }
 402
 403   Handle(SALOME_InteractiveObject) anIO = aList.First();
 404   myActor = SMESH::FindActorByEntry(anIO->getEntry());
 405   if(myActor){
 406     TVisualObjPtr aVisualObj = myActor->GetObject();
 407     if(SMDS_Mesh* aMesh = aVisualObj->GetMesh())
 408     {
 409       const SMDS_MeshElement* tria[2];
 410       if( SMESH::GetEdgeNodes( mySelector, aVisualObj, anId1, anId2 ) >= 1 &&
 411           findTriangles( aMesh->FindNode( anId1 ), aMesh->FindNode( anId2 ), tria[0],tria[1] ) )
 412       {
 413         QString aText = QString("%1-%2").arg(anId1).arg(anId2);
 414         myBusy = true;
 415         myEdge->setText(aText);
 416         myBusy = false;
 417
 418         myOkBtn->setEnabled(true);
 419         myApplyBtn->setEnabled(true);
 420       }
 421       else
 422       {
 423         myEdge->clear();
 424       }
 425     }
 426   }
 427 }
 428
 429 //=======================================================================
 430 // name    : onDeactivate()
 431 // Purpose : SLOT called when dialog must be deativated
 432 //=======================================================================
 433 void SMESHGUI_SingleEditDlg::onDeactivate()
 434 {
 435   setEnabled(false);
 436 }
 437
 438 //=======================================================================
 439 // name    : enterEvent()
 440 // Purpose : Event filter
 441 //=======================================================================
 442 void SMESHGUI_SingleEditDlg::enterEvent (QEvent*)
 443 {
 444   if (!isEnabled()) {
 445     mySMESHGUI->EmitSignalDeactivateDialog();
 446     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
 447       aViewWindow->SetSelectionMode(EdgeOfCellSelection);
 448     setEnabled(true);
 449   }
 450 }
 451
 452 //=================================================================================
 453 // function : closeEvent()
 454 // purpose  :
 455 //=================================================================================
 456 void SMESHGUI_SingleEditDlg::closeEvent (QCloseEvent*)
 457 {
 458   onClose();
 459 }
 460
 461 //=======================================================================
 462 //function : hideEvent()
 463 //purpose  : caused by ESC key
 464 //=======================================================================
 465 void SMESHGUI_SingleEditDlg::hideEvent (QHideEvent*)
 466 {
 467   if (!isMinimized())
 468     onClose();
 469 }
 470
 471 //=================================================================================
 472 // function : onApply()
 473 // purpose  : SLOT. Called when apply button is pressed
 474 //=================================================================================
 475 bool SMESHGUI_SingleEditDlg::onApply()
 476 {
 477   if (mySMESHGUI->isActiveStudyLocked())
 478     return false;
 479   // verify validity of input data
 480   if (!isValid(true))
 481     return false;
 482
 483   // get mesh, actor and nodes
 484   SALOME_ListIO aList;
 485   mySelectionMgr->selectedObjects(aList);
 486
 487   SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(aList.First());
 488
 489   if (aMesh->_is_nil()) {
 490     SUIT_MessageBox::info1(SMESH::GetDesktop(mySMESHGUI),
 491                            tr("SMESH_ERROR"),
 492                            tr("SMESHG_NO_MESH"),
 493                            tr("SMESH_BUT_OK"));
 494     return false;
 495   }
 496
 497   SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
 498   int anId1= 0, anId2 = 0;
 499   if (aMeshEditor->_is_nil() || !getNodeIds(myEdge->text(), anId1, anId2))
 500     return false;
 501
 502   // perform operation
 503   bool aResult = process(aMeshEditor.in(), anId1, anId2);
 504
 505   // update actor
 506   if (aResult) {
 507     mySelector->ClearIndex();
 508     mySelectionMgr->setSelectedObjects(aList, false);
 509     onSelectionDone();
 510     SMESH::UpdateView();
 511   }
 512
 513   return aResult;
 514 }
 515
 516 /*!
 517  *  Class       : SMESHGUI_TrianglesInversionDlg
 518  *  Description : Inversion of the diagonal of a pseudo-quadrangle formed by
 519  *                2 neighboring triangles with 1 common edge
 520  */
 521
 522 SMESHGUI_TrianglesInversionDlg
 523 ::SMESHGUI_TrianglesInversionDlg(SMESHGUI* theModule,
 524                                  const char* theName)
 525 : SMESHGUI_SingleEditDlg(theModule,theName)
 526 {
 527   setCaption(tr("CAPTION"));
 528 }
 529
 530 SMESHGUI_TrianglesInversionDlg::~SMESHGUI_TrianglesInversionDlg()
 531 {
 532 }
 533
 534 bool SMESHGUI_TrianglesInversionDlg::process (SMESH::SMESH_MeshEditor_ptr theMeshEditor,
 535                                               const int theId1, const int theId2)
 536 {
 537   return theMeshEditor->InverseDiag(theId1, theId2);
 538 }
 539
 540 /*!
 541  *  Class       : SMESHGUI_UnionOfTwoTrianglesDlg
 542  *  Description : Construction of a quadrangle by deletion of the
 543  *                common border of 2 neighboring triangles
 544  */
 545
 546 SMESHGUI_UnionOfTwoTrianglesDlg
 547 ::SMESHGUI_UnionOfTwoTrianglesDlg(SMESHGUI* theModule,
 548                                   const char* theName)
 549 : SMESHGUI_SingleEditDlg(theModule,theName)
 550 {
 551   setCaption(tr("CAPTION"));
 552 }
 553
 554 SMESHGUI_UnionOfTwoTrianglesDlg::~SMESHGUI_UnionOfTwoTrianglesDlg()
 555 {
 556 }
 557
 558 bool SMESHGUI_UnionOfTwoTrianglesDlg::process (SMESH::SMESH_MeshEditor_ptr theMeshEditor,
 559                                                const int theId1, const int theId2)
 560 {
 561   return theMeshEditor->DeleteDiag(theId1, theId2);
 562 }