Salome HOME
IPAL15344 (SMESH_MeshEditor.idl: signature of MergeEqualElements() was changed):
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_MoveNodesDlg.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.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //
23 //
24 //  File   : SMESHGUI_MoveNodesDlg.cxx
25 //  Author : Nicolas REJNERI
26 //  Module : SMESH
27 //  $Header$
28
29 #include "SMESHGUI_MoveNodesDlg.h"
30
31 #include "SMESHGUI.h"
32 #include "SMESHGUI_SpinBox.h"
33 #include "SMESHGUI_IdValidator.h"
34 #include "SMESHGUI_Utils.h"
35 #include "SMESHGUI_VTKUtils.h"
36 #include "SMESHGUI_MeshUtils.h"
37
38 #include "SMESH_Actor.h"
39 #include "SMDS_Mesh.hxx"
40 #include "SMDS_MeshNode.hxx"
41
42 #include "LightApp_SelectionMgr.h"
43 #include "LightApp_Application.h"
44 #include "SUIT_ResourceMgr.h"
45 #include "SUIT_Desktop.h"
46 #include "SUIT_Session.h"
47 #include "SUIT_MessageBox.h"
48
49 #include "SVTK_Selector.h"
50 #include "SVTK_ViewModel.h"
51 #include "SVTK_ViewWindow.h"
52 #include "SALOME_ListIO.hxx"
53
54 #include "SVTK_ViewWindow.h"
55 #include "VTKViewer_CellLocationsArray.h"
56
57 #include "utilities.h"
58
59 // OCCT includes
60 #include <TColStd_MapOfInteger.hxx>
61
62 // VTK includes
63 #include <vtkCell.h>
64 #include <vtkIdList.h>
65 #include <vtkCellArray.h>
66 #include <vtkUnsignedCharArray.h>
67 #include <vtkUnstructuredGrid.h>
68 #include <vtkDataSetMapper.h>
69 #include <vtkProperty.h>
70
71 // QT includes
72 #include <qgroupbox.h>
73 #include <qlabel.h>
74 #include <qlineedit.h>
75 #include <qpushbutton.h>
76 #include <qradiobutton.h>
77 #include <qlayout.h>
78 #include <qpixmap.h>
79 #include <qmessagebox.h>
80 #include <qbuttongroup.h>
81
82 // IDL Headers
83 #include <SALOMEconfig.h>
84 #include CORBA_SERVER_HEADER(SMESH_Mesh)
85 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
86
87 #define MARGIN  10
88 #define SPACING 5
89
90
91 //=================================================================================
92 // name    : SMESHGUI_MoveNodesDlg::SMESHGUI_MoveNodesDlg
93 // Purpose :
94 //=================================================================================
95 SMESHGUI_MoveNodesDlg::SMESHGUI_MoveNodesDlg (SMESHGUI* theModule, 
96                                               const char* theName):
97   QDialog(SMESH::GetDesktop(theModule), 
98           theName, 
99           false,
100           WStyle_Customize | WStyle_NormalBorder | WStyle_Title | WStyle_SysMenu),
101   mySelectionMgr(SMESH::GetSelectionMgr(theModule)),
102   mySMESHGUI(theModule)
103 {
104   myPreviewActor = 0;
105   myBusy = false;
106
107   setCaption(tr("CAPTION"));
108
109   QVBoxLayout* aDlgLay = new QVBoxLayout (this, MARGIN, SPACING);
110
111   QFrame* aMainFrame = createMainFrame  (this);
112   QFrame* aBtnFrame  = createButtonFrame(this);
113
114   aDlgLay->addWidget(aMainFrame);
115   aDlgLay->addWidget(aBtnFrame);
116
117   aDlgLay->setStretchFactor(aMainFrame, 1);
118
119   mySelector = (SMESH::GetViewWindow( mySMESHGUI ))->GetSelector();
120
121   myHelpFileName = "/files/displacing_nodes.htm";
122
123   Init();
124 }
125
126 //=======================================================================
127 // name    : SMESHGUI_MoveNodesDlg::createButtonFrame
128 // Purpose : Create frame containing buttons
129 //=======================================================================
130 QFrame* SMESHGUI_MoveNodesDlg::createButtonFrame (QWidget* theParent)
131 {
132   QFrame* aFrame = new QFrame(theParent);
133   aFrame->setFrameStyle(QFrame::Box | QFrame::Sunken);
134
135   myOkBtn     = new QPushButton(tr("SMESH_BUT_OK"   ), aFrame);
136   myApplyBtn  = new QPushButton(tr("SMESH_BUT_APPLY"), aFrame);
137   myCloseBtn  = new QPushButton(tr("SMESH_BUT_CLOSE"), aFrame);
138   myHelpBtn   = new QPushButton(tr("SMESH_BUT_HELP"), aFrame);
139
140   QSpacerItem* aSpacer = new QSpacerItem(0, 0, QSizePolicy::Expanding, QSizePolicy::Minimum);
141
142   QHBoxLayout* aLay = new QHBoxLayout(aFrame, MARGIN, SPACING);
143
144   aLay->addWidget(myOkBtn);
145   aLay->addWidget(myApplyBtn);
146   aLay->addItem(aSpacer);
147   aLay->addWidget(myCloseBtn);
148   aLay->addWidget(myHelpBtn);
149
150   connect(myOkBtn,    SIGNAL(clicked()), SLOT(onOk()));
151   connect(myCloseBtn, SIGNAL(clicked()), SLOT(onClose()));
152   connect(myApplyBtn, SIGNAL(clicked()), SLOT(onApply()));
153   connect(myHelpBtn,  SIGNAL(clicked()), SLOT(onHelp()));
154
155   return aFrame;
156 }
157
158 //=======================================================================
159 // name    : SMESHGUI_MoveNodesDlg::createMainFrame
160 // Purpose : Create frame containing dialog's input fields
161 //=======================================================================
162 QFrame* SMESHGUI_MoveNodesDlg::createMainFrame (QWidget* theParent)
163 {
164   QFrame* aFrame = new QFrame(theParent);
165
166   QPixmap iconMoveNode (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_DLG_MOVE_NODE")));
167   QPixmap iconSelect   (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
168
169   QButtonGroup* aPixGrp = new QButtonGroup(1, Qt::Vertical, tr("MESH_NODE"), aFrame);
170   aPixGrp->setExclusive(TRUE);
171   QRadioButton* aRBut = new QRadioButton(aPixGrp);
172   aRBut->setPixmap(iconMoveNode);
173   aRBut->setChecked(TRUE);
174
175   QGroupBox* anIdGrp = new QGroupBox(1, Qt::Vertical, tr("SMESH_MOVE"), aFrame);
176   new QLabel(tr("NODE_ID"), anIdGrp);
177   (new QPushButton(anIdGrp))->setPixmap(iconSelect);
178   myId = new QLineEdit(anIdGrp);
179   myId->setValidator(new SMESHGUI_IdValidator(this, "validator", 1));
180
181   QGroupBox* aCoordGrp = new QGroupBox(1, Qt::Vertical, tr("SMESH_COORDINATES"), aFrame);
182   QLabel* aXLabel = new QLabel(tr("SMESH_X"), aCoordGrp);
183   aXLabel->setSizePolicy( QSizePolicy( QSizePolicy::Fixed, QSizePolicy::Fixed ) );
184   myX = new SMESHGUI_SpinBox(aCoordGrp);
185
186   QLabel* aYLabel = new QLabel(tr("SMESH_Y"), aCoordGrp);
187   aYLabel->setAlignment( Qt::AlignRight | Qt::AlignVCenter | Qt::ExpandTabs );
188   myY = new SMESHGUI_SpinBox(aCoordGrp);
189
190   QLabel* aZLabel = new QLabel(tr("SMESH_Z"), aCoordGrp);
191   aZLabel->setAlignment( Qt::AlignRight | Qt::AlignVCenter | Qt::ExpandTabs );
192   myZ = new SMESHGUI_SpinBox(aCoordGrp);
193
194   myX->RangeStepAndValidator(COORD_MIN, COORD_MAX, 25.0, 3);
195   myY->RangeStepAndValidator(COORD_MIN, COORD_MAX, 25.0, 3);
196   myZ->RangeStepAndValidator(COORD_MIN, COORD_MAX, 25.0, 3);
197
198   QVBoxLayout* aLay = new QVBoxLayout(aFrame);
199   aLay->addWidget(aPixGrp);
200   aLay->addWidget(anIdGrp);
201   aLay->addWidget(aCoordGrp);
202
203   // connect signale and slots
204   connect(myX, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
205   connect(myY, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
206   connect(myZ, SIGNAL (valueChanged(double)), this, SLOT(redisplayPreview()));
207   connect(myId, SIGNAL(textChanged(const QString&)), SLOT(onTextChange(const QString&)));
208
209   return aFrame;
210 }
211
212 //=======================================================================
213 // name    : SMESHGUI_MoveNodesDlg::~SMESHGUI_MoveNodesDlg
214 // Purpose :
215 //=======================================================================
216 SMESHGUI_MoveNodesDlg::~SMESHGUI_MoveNodesDlg()
217 {
218   erasePreview();
219 }
220
221 //=======================================================================
222 // name    : SMESHGUI_MoveNodesDlg::Init
223 // Purpose : Init dialog fields
224 //=======================================================================
225 void SMESHGUI_MoveNodesDlg::Init()
226 {
227   myPreviewActor = 0;
228   myMeshActor = 0;
229   myBusy = false;
230
231   mySMESHGUI->SetActiveDialogBox((QDialog*)this);
232
233   // selection and SMESHGUI
234   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(onSelectionDone()));
235   connect(mySMESHGUI, SIGNAL(SignalDeactivateActiveDialog()), SLOT(onDeactivate()));
236   connect(mySMESHGUI, SIGNAL(SignalCloseAllDialogs()), SLOT(onClose()));
237
238   reset();
239   setEnabled(true);
240
241   this->show();
242
243   // set selection mode
244   SMESH::SetPointRepresentation(true);
245   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
246     aViewWindow->SetSelectionMode(NodeSelection);
247
248   onSelectionDone();
249 }
250
251 //=======================================================================
252 // name    : SMESHGUI_MoveNodesDlg::isValid
253 // Purpose : Verify validity of entry information
254 //=======================================================================
255 bool SMESHGUI_MoveNodesDlg::isValid (const bool theMess) const
256 {
257   if (myId->text().isEmpty()) {
258     if (theMess)
259       QMessageBox::information(SMESHGUI::desktop(), tr("SMESH_WARNING"),
260                                tr("NODE_ID_IS_NOT_DEFINED"), QMessageBox::Ok);
261     return false;
262   }
263   return true;
264 }
265
266 //=======================================================================
267 // name    : SMESHGUI_MoveNodesDlg::reset
268 // Purpose : Reset the dialog state
269 //=======================================================================
270 void SMESHGUI_MoveNodesDlg::reset()
271 {
272   myId->clear();
273   myX->SetValue(0);
274   myY->SetValue(0);
275   myZ->SetValue(0);
276   redisplayPreview();
277   updateButtons();
278 }
279
280 //=======================================================================
281 // name    : SMESHGUI_MoveNodesDlg::onApply
282 // Purpose : SLOT called when "Apply" button pressed.
283 //=======================================================================
284 bool SMESHGUI_MoveNodesDlg::onApply()
285 {
286   if (mySMESHGUI->isActiveStudyLocked())
287     return false;
288
289   if (!isValid(true))
290     return false;
291
292   SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(myMeshActor->getIO());
293   if (aMesh->_is_nil()) {
294     QMessageBox::information(SMESHGUI::desktop(), tr("SMESH_ERROR"),
295                              tr("SMESHG_NO_MESH"), QMessageBox::Ok);
296     return false;
297   }
298
299   SMESH::SMESH_MeshEditor_var aMeshEditor = aMesh->GetMeshEditor();
300   if (aMeshEditor->_is_nil())
301     return false;
302
303   int anId = myId->text().toInt();
304   bool aResult = false;
305   try {
306     aResult = aMeshEditor->MoveNode(anId, myX->GetValue(), myY->GetValue(), myZ->GetValue());
307   } catch (...) {
308   }
309
310   if (aResult) {
311     SALOME_ListIO aList;
312     aList.Append(myMeshActor->getIO());
313     mySelectionMgr->setSelectedObjects(aList,false);
314     SMESH::UpdateView();
315     reset();
316   }
317
318   return aResult;
319 }
320
321 //=======================================================================
322 // name    : SMESHGUI_MoveNodesDlg::onOk
323 // Purpose : SLOT called when "Ok" button pressed.
324 //=======================================================================
325 void SMESHGUI_MoveNodesDlg::onOk()
326 {
327   if (onApply())
328     onClose();
329 }
330
331 //=======================================================================
332 // name    : SMESHGUI_MoveNodesDlg::onClose
333 // Purpose : SLOT called when "Close" button pressed. Close dialog
334 //=======================================================================
335 void SMESHGUI_MoveNodesDlg::onClose()
336 {
337   //mySelectionMgr->clearSelected();
338   SMESH::SetPointRepresentation(false);
339   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
340     aViewWindow->SetSelectionMode(ActorSelection);
341   disconnect(mySelectionMgr, 0, this, 0);
342   disconnect(mySMESHGUI, 0, this, 0);
343   erasePreview();
344   mySMESHGUI->ResetState();
345   reject();
346 }
347
348 //=================================================================================
349 // function : onHelp()
350 // purpose  :
351 //=================================================================================
352 void SMESHGUI_MoveNodesDlg::onHelp()
353 {
354   LightApp_Application* app = (LightApp_Application*)(SUIT_Session::session()->activeApplication());
355   if (app) 
356     app->onHelpContextModule(mySMESHGUI ? app->moduleName(mySMESHGUI->moduleName()) : QString(""), myHelpFileName);
357   else {
358     SUIT_MessageBox::warn1(0, QObject::tr("WRN_WARNING"),
359                            QObject::tr("EXTERNAL_BROWSER_CANNOT_SHOW_PAGE").
360                            arg(app->resourceMgr()->stringValue("ExternalBrowser", "application")).arg(myHelpFileName),
361                            QObject::tr("BUT_OK"));
362   }
363 }
364
365 //=======================================================================
366 // name    : SMESHGUI_MoveNodesDlg::onTextChange
367 // Purpose :
368 //=======================================================================
369 void SMESHGUI_MoveNodesDlg::onTextChange (const QString& theNewText)
370 {
371   if (myBusy) return;
372
373   myOkBtn->setEnabled(false);
374   myApplyBtn->setEnabled(false);
375   erasePreview();
376
377   // select entered node
378   if(myMeshActor){
379     if(SMDS_Mesh* aMesh = myMeshActor->GetObject()->GetMesh()){
380       myBusy = true;
381       Handle(SALOME_InteractiveObject) anIO = myMeshActor->getIO();
382       SALOME_ListIO aList;
383       aList.Append(anIO);
384       mySelectionMgr->setSelectedObjects(aList,false);
385       myBusy = false;
386
387       if(const SMDS_MeshElement *anElem = aMesh->FindElement(theNewText.toInt())) {
388         TColStd_MapOfInteger aListInd;
389         aListInd.Add(anElem->GetID());
390         mySelector->AddOrRemoveIndex(anIO,aListInd, false);
391         if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
392           aViewWindow->highlight(anIO,true,true);
393         
394         onSelectionDone();
395       }
396     }
397   }
398 }
399
400 //=======================================================================
401 // name    : SMESHGUI_MoveNodesDlg::onSelectionDone
402 // Purpose : SLOT called when selection changed
403 //=======================================================================
404 void SMESHGUI_MoveNodesDlg::onSelectionDone()
405 {
406   if (myBusy) return;
407   myMeshActor = 0;
408
409   SALOME_ListIO aList;
410   mySelectionMgr->selectedObjects(aList,SVTK_Viewer::Type());
411
412   if (aList.Extent() == 1) {
413     Handle(SALOME_InteractiveObject) anIO = aList.First();
414     myMeshActor = SMESH::FindActorByEntry(anIO->getEntry());
415     if(myMeshActor){
416       QString aText;
417       if (SMESH::GetNameOfSelectedNodes(mySelector,anIO,aText) == 1) {
418         if(SMDS_Mesh* aMesh = myMeshActor->GetObject()->GetMesh()) {
419           if(const SMDS_MeshNode* aNode = aMesh->FindNode(aText.toInt())) {
420             myBusy = true;
421             myId->setText(aText);
422             myX->SetValue(aNode->X());
423             myY->SetValue(aNode->Y());
424             myZ->SetValue(aNode->Z());
425             myBusy = false;
426             erasePreview(); // avoid overlapping of a selection and a preview
427             updateButtons();
428             return;
429           }
430         }
431       }
432     }
433   }
434
435   reset();
436 }
437
438 //=======================================================================
439 // name    : SMESHGUI_MoveNodesDlg::onDeactivate
440 // Purpose : SLOT called when dialog must be deativated
441 //=======================================================================
442 void SMESHGUI_MoveNodesDlg::onDeactivate()
443 {
444   setEnabled(false);
445   erasePreview();
446 }
447
448 //=======================================================================
449 // name    : SMESHGUI_MoveNodesDlg::enterEvent
450 // Purpose : Event filter
451 //=======================================================================
452 void SMESHGUI_MoveNodesDlg::enterEvent (QEvent*)
453 {
454   if (!isEnabled()) {
455     mySMESHGUI->EmitSignalDeactivateDialog();
456
457     // set selection mode
458     SMESH::SetPointRepresentation(true);
459     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
460       aViewWindow->SetSelectionMode(NodeSelection);
461
462     redisplayPreview();
463
464     setEnabled(true);
465   }
466 }
467
468 //=======================================================================
469 // name    : SMESHGUI_MoveNodesDlg::closeEvent
470 // Purpose :
471 //=======================================================================
472 void SMESHGUI_MoveNodesDlg::closeEvent (QCloseEvent*)
473 {
474   onClose();
475   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
476     aViewWindow->Repaint();
477 }
478
479 //=======================================================================
480 // name    : SMESHGUI_MoveNodesDlg::hideEvent
481 // Purpose : may be caused by ESC key
482 //=======================================================================
483 void SMESHGUI_MoveNodesDlg::hideEvent (QHideEvent*)
484 {
485   if (!isMinimized())
486     onClose();
487 }
488
489 //=======================================================================
490 // name    : SMESHGUI_MoveNodesDlg::updateButtons
491 // Purpose : Update buttons state
492 //=======================================================================
493 void SMESHGUI_MoveNodesDlg::updateButtons()
494 {
495   bool isEnabled = isValid(false);
496   myOkBtn->setEnabled(isEnabled);
497   myApplyBtn->setEnabled(isEnabled);
498 }
499
500 //=======================================================================
501 // name    : SMESHGUI_MoveNodesDlg::erasePreview
502 // Purpose : Erase preview
503 //=======================================================================
504 void  SMESHGUI_MoveNodesDlg::erasePreview()
505 {
506   if (myPreviewActor == 0)
507     return;
508
509   SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI );
510   if (aViewWindow)
511     aViewWindow->RemoveActor(myPreviewActor);
512   myPreviewActor->Delete();
513   myPreviewActor = 0;
514   if (aViewWindow)
515     aViewWindow->Repaint();
516 }
517
518 //=======================================================================
519 // name    : SMESHGUI_MoveNodesDlg::redisplayPreview
520 // Purpose : Redisplay preview
521 //=======================================================================
522 void SMESHGUI_MoveNodesDlg::redisplayPreview()
523 {
524   if (myBusy)
525     return;
526
527   if (myPreviewActor != 0)
528     erasePreview();
529
530   if (!isValid(false))
531     return;
532
533   vtkUnstructuredGrid* aGrid = vtkUnstructuredGrid::New();
534
535   vtkPoints* aPoints = vtkPoints::New();
536   aPoints->SetNumberOfPoints(1);
537   aPoints->SetPoint(0, myX->GetValue(), myY->GetValue(), myZ->GetValue());
538
539   // Create cells
540
541   vtkIdList *anIdList = vtkIdList::New();
542   anIdList->SetNumberOfIds(1);
543
544   vtkCellArray *aCells = vtkCellArray::New();
545   aCells->Allocate(2, 0);
546
547   vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
548   aCellTypesArray->SetNumberOfComponents(1);
549   aCellTypesArray->Allocate(1);
550
551   anIdList->SetId(0, 0);
552   aCells->InsertNextCell(anIdList);
553   aCellTypesArray->InsertNextValue(VTK_VERTEX);
554   anIdList->Delete();
555
556   VTKViewer_CellLocationsArray* aCellLocationsArray = VTKViewer_CellLocationsArray::New();
557   aCellLocationsArray->SetNumberOfComponents(1);
558   aCellLocationsArray->SetNumberOfTuples(1);
559
560   aCells->InitTraversal();
561   vtkIdType npts;
562   aCellLocationsArray->SetValue(0, aCells->GetTraversalLocation(npts));
563
564   aGrid->SetPoints(aPoints);
565   aPoints->Delete();
566
567   aGrid->SetCells(aCellTypesArray,aCellLocationsArray,aCells);
568   aCellLocationsArray->Delete();
569   aCellTypesArray->Delete();
570   aCells->Delete();
571
572   // Create and display actor
573   vtkDataSetMapper* aMapper = vtkDataSetMapper::New();
574   aMapper->SetInput(aGrid);
575   aGrid->Delete();
576
577   myPreviewActor = SALOME_Actor::New();
578   myPreviewActor->PickableOff();
579   myPreviewActor->SetMapper(aMapper);
580   aMapper->Delete();
581
582   vtkProperty* aProp = vtkProperty::New();
583   aProp->SetRepresentationToWireframe();
584   aProp->SetColor(250, 0, 250);
585   aProp->SetPointSize(5);
586   myPreviewActor->SetProperty(aProp);
587   aProp->Delete();
588
589   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
590     {
591       aViewWindow->AddActor(myPreviewActor);
592       aViewWindow->Repaint();
593     }
594 }
595
596 //=================================================================================
597 // function : keyPressEvent()
598 // purpose  :
599 //=================================================================================
600 void SMESHGUI_MoveNodesDlg::keyPressEvent( QKeyEvent* e )
601 {
602   QDialog::keyPressEvent( e );
603   if ( e->isAccepted() )
604     return;
605
606   if ( e->key() == Key_F1 )
607     {
608       e->accept();
609       onHelp();
610     }
611 }