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