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