Salome HOME
Prefernces.
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_NodesDlg.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_NodesDlg.cxx
25 //  Author : Nicolas REJNERI
26 //  Module : SMESH
27 //  $Header$
28
29 #include "SMESHGUI_NodesDlg.h"
30
31 #include "SMESHGUI.h"
32 #include "SMESHGUI_SpinBox.h"
33 #include "SMESHGUI_Utils.h"
34 #include "SMESHGUI_VTKUtils.h"
35 #include "SMESHGUI_MeshUtils.h"
36
37 #include "SMESH_Actor.h"
38 #include "SMESH_ActorUtils.h"
39 #include "SMESH_ObjectDef.h"
40
41 #include "SMDS_Mesh.hxx"
42 #include "SMDS_MeshNode.hxx"
43
44 #include "SUIT_Session.h"
45 #include "SUIT_OverrideCursor.h"
46 #include "SUIT_ViewWindow.h"
47 #include "SUIT_ViewManager.h"
48 #include "SUIT_MessageBox.h"
49 #include "SUIT_Desktop.h"
50
51 #include "SalomeApp_Study.h"
52 #include "SalomeApp_SelectionMgr.h"
53
54 #include "SVTK_Selector.h"
55 #include "SVTK_ViewWindow.h"
56
57 #include "SALOME_Actor.h"
58 #include "SALOME_ListIO.hxx"
59
60 #include "utilities.h"
61
62 // VTK Includes
63 #include <vtkCell.h>
64 #include <vtkIdList.h>
65 #include <vtkIntArray.h>
66 #include <vtkCellArray.h>
67 #include <vtkUnsignedCharArray.h>
68 #include <vtkUnstructuredGrid.h>
69 #include <vtkDataSetMapper.h>
70 #include <vtkActorCollection.h>
71 #include <vtkRenderer.h>
72
73 // QT Includes
74 #include <qbuttongroup.h>
75 #include <qframe.h>
76 #include <qgroupbox.h>
77 #include <qlabel.h>
78 #include <qlineedit.h>
79 #include <qpushbutton.h>
80 #include <qradiobutton.h>
81 #include <qlayout.h>
82 #include <qvariant.h>
83 #include <qtooltip.h>
84 #include <qwhatsthis.h>
85 #include <qimage.h>
86 #include <qpixmap.h>
87 #include <qvalidator.h>
88 #include <qevent.h>
89
90 using namespace std;
91
92
93 namespace SMESH {
94
95   void AddNode (SMESH::SMESH_Mesh_ptr theMesh, float x, float y, float z)
96   {
97     SUIT_OverrideCursor wc;
98     try {
99       _PTR(SObject) aSobj = SMESH::FindSObject(theMesh);
100       SMESH::SMESH_MeshEditor_var aMeshEditor = theMesh->GetMeshEditor();
101       aMeshEditor->AddNode(x,y,z);
102       _PTR(Study) aStudy = GetActiveStudyDocument();
103       CORBA::Long anId = aStudy->StudyId();
104       if (TVisualObjPtr aVisualObj = SMESH::GetVisualObj(anId, aSobj->GetID().c_str())) {
105         aVisualObj->Update(true);
106       }
107     } catch (SALOME::SALOME_Exception& exc) {
108       INFOS("Follow exception was cought:\n\t" << exc.details.text);
109     } catch (const std::exception& exc) {
110       INFOS("Follow exception was cought:\n\t" << exc.what());
111     } catch (...) {
112       INFOS("Unknown exception was cought !!!");
113     }
114   }
115
116   class TNodeSimulation {
117     SVTK_ViewWindow* myViewWindow;
118
119     SALOME_Actor *myPreviewActor;
120     vtkDataSetMapper* myMapper;
121     vtkPoints* myPoints;
122
123   public:
124     TNodeSimulation(SVTK_ViewWindow* theViewWindow):
125       myViewWindow(theViewWindow)
126     {
127       vtkUnstructuredGrid* aGrid = vtkUnstructuredGrid::New();
128
129       // Create points
130       myPoints = vtkPoints::New();
131       myPoints->SetNumberOfPoints(1);
132       myPoints->SetPoint(0,0.0,0.0,0.0);
133
134       // Create cells
135       vtkIdList *anIdList = vtkIdList::New();
136       anIdList->SetNumberOfIds(1);
137
138       vtkCellArray *aCells = vtkCellArray::New();
139       aCells->Allocate(2, 0);
140
141       vtkUnsignedCharArray* aCellTypesArray = vtkUnsignedCharArray::New();
142       aCellTypesArray->SetNumberOfComponents(1);
143       aCellTypesArray->Allocate(1);
144
145       anIdList->SetId(0, 0);
146       aCells->InsertNextCell(anIdList);
147       aCellTypesArray->InsertNextValue(VTK_VERTEX);
148
149       vtkIntArray* aCellLocationsArray = vtkIntArray::New();
150       aCellLocationsArray->SetNumberOfComponents(1);
151       aCellLocationsArray->SetNumberOfTuples(1);
152
153       aCells->InitTraversal();
154       vtkIdType npts;
155       aCellLocationsArray->SetValue(0, aCells->GetTraversalLocation(npts));
156
157       aGrid->SetCells(aCellTypesArray, aCellLocationsArray, aCells);
158
159       aGrid->SetPoints(myPoints);
160       aGrid->SetCells(aCellTypesArray, aCellLocationsArray,aCells);
161       aCellLocationsArray->Delete();
162       aCellTypesArray->Delete();
163       aCells->Delete();
164       anIdList->Delete();
165
166       // Create and display actor
167       myMapper = vtkDataSetMapper::New();
168       myMapper->SetInput(aGrid);
169       aGrid->Delete();
170
171       myPreviewActor = SALOME_Actor::New();
172       myPreviewActor->SetInfinitive(true);
173       myPreviewActor->VisibilityOff();
174       myPreviewActor->PickableOff();
175       myPreviewActor->SetMapper(myMapper);
176
177       vtkProperty* aProp = vtkProperty::New();
178       aProp->SetRepresentationToPoints();
179
180       float anRGB[3];
181       GetColor( "SMESH", "node_color", anRGB[0], anRGB[1], anRGB[2], QColor( 0, 255, 0 ) );
182       aProp->SetColor( anRGB[0], anRGB[1], anRGB[2] );
183
184       float aPointSize = GetFloat( "SMESH:node_size", 3 );
185       aProp->SetPointSize( aPointSize );
186
187       myPreviewActor->SetProperty( aProp );
188       aProp->Delete();
189
190       myViewWindow->AddActor(myPreviewActor);
191     }
192
193     void SetPosition (float x, float y, float z)
194     {
195       myPoints->SetPoint(0,x,y,z);
196       myPoints->Modified();
197       SetVisibility(true);
198     }
199
200     void SetVisibility (bool theVisibility)
201     {
202       myPreviewActor->SetVisibility(theVisibility);
203       RepaintCurrentView();
204     }
205
206     ~TNodeSimulation()
207     {
208       myViewWindow->RemoveActor(myPreviewActor);
209       myPreviewActor->Delete();
210
211       myMapper->RemoveAllInputs();
212       myMapper->Delete();
213
214       myPoints->Delete();
215     }
216   };
217 }
218
219 //=================================================================================
220 // class    : SMESHGUI_NodesDlg()
221 // purpose  :
222 //=================================================================================
223 SMESHGUI_NodesDlg::SMESHGUI_NodesDlg (SMESHGUI* theModule,
224                                       const char* name,
225                                       bool modal,
226                                       WFlags fl): 
227   QDialog(SMESH::GetDesktop(theModule), 
228           name, 
229           modal, 
230           WStyle_Customize | WStyle_NormalBorder | WStyle_Title | WStyle_SysMenu | Qt::WDestructiveClose),
231   mySelector(SMESH::GetViewWindow(theModule)->GetSelector()),
232   mySelectionMgr(SMESH::GetSelectionMgr(theModule)),
233   myViewWindow(SMESH::GetViewWindow(theModule)),
234   mySMESHGUI(theModule)
235 {
236   mySimulation = new SMESH::TNodeSimulation(myViewWindow);
237
238   QPixmap image0 (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_DLG_NODE")));
239   if (!name)
240     setName("SMESHGUI_NodesDlg");
241   resize(303, 185);
242   setCaption(tr("MESH_NODE_TITLE"));
243   setSizeGripEnabled(TRUE);
244   SMESHGUI_NodesDlgLayout = new QGridLayout(this);
245   SMESHGUI_NodesDlgLayout->setSpacing(6);
246   SMESHGUI_NodesDlgLayout->setMargin(11);
247
248   /***************************************************************/
249   GroupButtons = new QGroupBox(this, "GroupButtons");
250   GroupButtons->setGeometry(QRect(10, 10, 281, 48));
251   GroupButtons->setTitle(tr("" ));
252   GroupButtons->setColumnLayout(0, Qt::Vertical);
253   GroupButtons->layout()->setSpacing(0);
254   GroupButtons->layout()->setMargin(0);
255   GroupButtonsLayout = new QGridLayout(GroupButtons->layout());
256   GroupButtonsLayout->setAlignment(Qt::AlignTop);
257   GroupButtonsLayout->setSpacing(6);
258   GroupButtonsLayout->setMargin(11);
259   buttonCancel = new QPushButton(GroupButtons, "buttonCancel");
260   buttonCancel->setText(tr("SMESH_BUT_CLOSE" ));
261   buttonCancel->setAutoDefault(TRUE);
262   GroupButtonsLayout->addWidget(buttonCancel, 0, 3);
263   buttonApply = new QPushButton(GroupButtons, "buttonApply");
264   buttonApply->setText(tr("SMESH_BUT_APPLY" ));
265   buttonApply->setAutoDefault(TRUE);
266   GroupButtonsLayout->addWidget(buttonApply, 0, 1);
267   QSpacerItem* spacer_9 = new QSpacerItem(20, 20, QSizePolicy::Expanding, QSizePolicy::Minimum);
268   GroupButtonsLayout->addItem(spacer_9, 0, 2);
269   buttonOk = new QPushButton(GroupButtons, "buttonOk");
270   buttonOk->setText(tr("SMESH_BUT_OK" ));
271   buttonOk->setAutoDefault(TRUE);
272   buttonOk->setDefault(TRUE);
273   GroupButtonsLayout->addWidget(buttonOk, 0, 0);
274   SMESHGUI_NodesDlgLayout->addWidget(GroupButtons, 2, 0);
275
276   /***************************************************************/
277   GroupConstructors = new QButtonGroup(this, "GroupConstructors");
278   GroupConstructors->setTitle(tr("MESH_NODE" ));
279   GroupConstructors->setExclusive(TRUE);
280   GroupConstructors->setColumnLayout(0, Qt::Vertical);
281   GroupConstructors->layout()->setSpacing(0);
282   GroupConstructors->layout()->setMargin(0);
283   GroupConstructorsLayout = new QGridLayout(GroupConstructors->layout());
284   GroupConstructorsLayout->setAlignment(Qt::AlignTop);
285   GroupConstructorsLayout->setSpacing(6);
286   GroupConstructorsLayout->setMargin(11);
287   Constructor1 = new QRadioButton(GroupConstructors, "Constructor1");
288   Constructor1->setText(tr("" ));
289   Constructor1->setPixmap(image0);
290   Constructor1->setChecked(TRUE);
291   GroupConstructorsLayout->addWidget(Constructor1, 0, 0);
292   QSpacerItem* spacer_2 = new QSpacerItem(20, 20, QSizePolicy::Expanding, QSizePolicy::Minimum);
293   GroupConstructorsLayout->addItem(spacer_2, 0, 1);
294   SMESHGUI_NodesDlgLayout->addWidget(GroupConstructors, 0, 0);
295
296   /***************************************************************/
297   GroupCoordinates = new QGroupBox(this, "GroupCoordinates");
298   GroupCoordinates->setTitle(tr("SMESH_COORDINATES" ));
299   GroupCoordinates->setColumnLayout(0, Qt::Vertical);
300   GroupCoordinates->layout()->setSpacing(0);
301   GroupCoordinates->layout()->setMargin(0);
302   GroupCoordinatesLayout = new QGridLayout(GroupCoordinates->layout());
303   GroupCoordinatesLayout->setAlignment(Qt::AlignTop);
304   GroupCoordinatesLayout->setSpacing(6);
305   GroupCoordinatesLayout->setMargin(11);
306   TextLabel_X = new QLabel(GroupCoordinates, "TextLabel_X");
307   TextLabel_X->setText(tr("SMESH_X" ));
308   GroupCoordinatesLayout->addWidget(TextLabel_X, 0, 0);
309   TextLabel_Y = new QLabel(GroupCoordinates, "TextLabel_Y");
310   TextLabel_Y->setText(tr("SMESH_Y" ));
311   GroupCoordinatesLayout->addWidget(TextLabel_Y, 0, 2);
312
313   TextLabel_Z = new QLabel(GroupCoordinates, "TextLabel_Z");
314   TextLabel_Z->setText(tr("SMESH_Z" ));
315   GroupCoordinatesLayout->addWidget(TextLabel_Z, 0, 4);
316
317   SpinBox_X = new SMESHGUI_SpinBox(GroupCoordinates, "SpinBox_X");
318   GroupCoordinatesLayout->addWidget(SpinBox_X, 0, 1);
319
320   SpinBox_Y = new SMESHGUI_SpinBox(GroupCoordinates, "SpinBox_Y");
321   GroupCoordinatesLayout->addWidget(SpinBox_Y, 0, 3);
322
323   SpinBox_Z = new SMESHGUI_SpinBox(GroupCoordinates, "SpinBox_Z");
324   GroupCoordinatesLayout->addWidget(SpinBox_Z, 0, 5);
325
326   SMESHGUI_NodesDlgLayout->addWidget(GroupCoordinates, 1, 0);
327
328   /* Initialisation and display */
329   Init();
330 }
331
332 //=======================================================================
333 // function : ~SMESHGUI_NodesDlg()
334 // purpose  : Destructor
335 //=======================================================================
336 SMESHGUI_NodesDlg::~SMESHGUI_NodesDlg()
337 {
338   delete mySimulation;
339 }
340
341 //=================================================================================
342 // function : Init()
343 // purpose  :
344 //=================================================================================
345 void SMESHGUI_NodesDlg::Init ()
346 {
347   /* Get setting of step value from file configuration */
348   double step;
349   // QString St = SUIT_CONFIG->getSetting("xxxxxxxxxxxxx");  TODO
350   // step = St.toDouble();                                    TODO
351   step = 25.0;
352
353   /* min, max, step and decimals for spin boxes */
354   SpinBox_X->RangeStepAndValidator(-999.999, +999.999, step, 3);
355   SpinBox_Y->RangeStepAndValidator(-999.999, +999.999, step, 3);
356   SpinBox_Z->RangeStepAndValidator(-999.999, +999.999, step, 3);
357   SpinBox_X->SetValue(0.0);
358   SpinBox_Y->SetValue(0.0);
359   SpinBox_Z->SetValue(0.0);
360
361   mySMESHGUI->SetActiveDialogBox((QDialog*)this);
362
363   /* signals and slots connections */
364   connect(buttonOk, SIGNAL(clicked()), this, SLOT(ClickOnOk()));
365   connect(buttonCancel, SIGNAL(clicked()), this, SLOT(ClickOnCancel()));
366   connect(buttonApply, SIGNAL(clicked()), this, SLOT(ClickOnApply()));
367
368   connect(SpinBox_X, SIGNAL (valueChanged(double)), SLOT(ValueChangedInSpinBox(double)));
369   connect(SpinBox_Y, SIGNAL (valueChanged(double)), SLOT(ValueChangedInSpinBox(double)));
370   connect(SpinBox_Z, SIGNAL (valueChanged(double)), SLOT(ValueChangedInSpinBox(double)));
371
372   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(SelectionIntoArgument()));
373   connect(mySMESHGUI, SIGNAL (SignalDeactivateActiveDialog()), SLOT(DeactivateActiveDialog()));
374   /* to close dialog if study frame change */
375   connect(mySMESHGUI, SIGNAL (SignalStudyFrameChanged()), SLOT(ClickOnCancel()));
376
377   /* Move widget on the botton right corner of main widget */
378   int x, y;
379   mySMESHGUI->DefineDlgPosition(this, x, y);
380   this->move(x, y);
381   this->show();
382
383   // set selection mode
384   SMESH::SetPointRepresentation(true);
385   myViewWindow->SetSelectionMode(NodeSelection);
386
387   SelectionIntoArgument();
388 }
389
390 //=================================================================================
391 // function : ValueChangedInSpinBox()
392 // purpose  :
393 //=================================================================================
394 void SMESHGUI_NodesDlg::ValueChangedInSpinBox (double newValue)
395 {
396   if (!myMesh->_is_nil()) {
397     double vx = SpinBox_X->GetValue();
398     double vy = SpinBox_Y->GetValue();
399     double vz = SpinBox_Z->GetValue();
400
401     mySimulation->SetPosition(vx,vy,vz);
402   }
403 }
404
405 //=================================================================================
406 // function : ClickOnOk()
407 // purpose  :
408 //=================================================================================
409 void SMESHGUI_NodesDlg::ClickOnOk()
410 {
411   if (ClickOnApply())
412     ClickOnCancel();
413 }
414
415 //=================================================================================
416 // function : ClickOnApply()
417 // purpose  :
418 //=================================================================================
419 bool SMESHGUI_NodesDlg::ClickOnApply()
420 {
421   if (mySMESHGUI->isActiveStudyLocked())
422     return false;
423
424   if (myMesh->_is_nil()) {
425     SUIT_MessageBox::warn1(SMESHGUI::desktop(), tr("SMESH_WRN_WARNING"),
426                            tr("MESH_IS_NOT_SELECTED"), tr("SMESH_BUT_OK"));
427     return false;
428   }
429
430   /* Recup args and call method */
431   double x = SpinBox_X->GetValue();
432   double y = SpinBox_Y->GetValue();
433   double z = SpinBox_Z->GetValue();
434   mySimulation->SetVisibility(false);
435   SMESH::AddNode(myMesh,x,y,z);
436   SMESH::SetPointRepresentation(true);
437
438   // select myMesh
439   SALOME_ListIO aList;
440   mySelectionMgr->selectedObjects(aList);
441   if (aList.Extent() != 1) {
442     if (SVTK_ViewWindow* aViewWindow = SMESH::GetCurrentVtkView()) {
443       vtkActorCollection *aCollection = aViewWindow->getRenderer()->GetActors();
444       aCollection->InitTraversal();
445       while (vtkActor *anAct = aCollection->GetNextActor()) {
446         if (SMESH_Actor *anActor = dynamic_cast<SMESH_Actor*>(anAct)) {
447           if (anActor->hasIO()) {
448             if (SMESH_MeshObj *aMeshObj = dynamic_cast<SMESH_MeshObj*>(anActor->GetObject().get())) {
449               if (myMesh->_is_equivalent(aMeshObj->GetMeshServer())) {
450                 aList.Clear();
451                 aList.Append(anActor->getIO());
452                 mySelectionMgr->setSelectedObjects(aList, false);
453                 break;
454               }
455             }
456           }
457         }
458       }
459     }
460   }
461   return true;
462 }
463
464 //=================================================================================
465 // function : ClickOnCancel()
466 // purpose  :
467 //=================================================================================
468 void SMESHGUI_NodesDlg::ClickOnCancel()
469 {
470   disconnect(mySelectionMgr, 0, this, 0);
471   myViewWindow->SetSelectionMode(ActorSelection);
472
473   mySimulation->SetVisibility(false);
474   SMESH::SetPointRepresentation(false);
475   mySMESHGUI->ResetState();
476
477   reject();
478 }
479
480 //=================================================================================
481 // function : SelectionIntoArgument()
482 // purpose  : Called when selection as changed or other case
483 //=================================================================================
484 void SMESHGUI_NodesDlg::SelectionIntoArgument()
485 {
486   if (!GroupConstructors->isEnabled())
487     return;
488
489   mySimulation->SetVisibility(false);
490   SMESH::SetPointRepresentation(true);
491
492   const SALOME_ListIO& aList = mySelector->StoredIObjects();
493   if (aList.Extent() == 1) {
494     Handle(SALOME_InteractiveObject) anIO = aList.First();
495     if (anIO->hasEntry()) {
496       myMesh = SMESH::GetMeshByIO(anIO);
497       if (myMesh->_is_nil()) return;
498       QString aText;
499       if (SMESH::GetNameOfSelectedNodes(mySelector,anIO,aText) == 1) {
500         if (SMESH_Actor* anActor = SMESH::FindActorByObject(myMesh.in())) {
501           if (SMDS_Mesh* aMesh = anActor->GetObject()->GetMesh()) {
502             if (const SMDS_MeshNode* aNode = aMesh->FindNode(aText.toInt())) {
503               SpinBox_X->SetValue(aNode->X());
504               SpinBox_Y->SetValue(aNode->Y());
505               SpinBox_Z->SetValue(aNode->Z());
506             }
507           }
508         }
509       }
510       mySimulation->SetPosition(SpinBox_X->GetValue(),
511                                 SpinBox_Y->GetValue(),
512                                 SpinBox_Z->GetValue());
513     }
514   }
515 }
516
517 //=================================================================================
518 // function : closeEvent()
519 // purpose  :
520 //=================================================================================
521 void SMESHGUI_NodesDlg::closeEvent (QCloseEvent*)
522 {
523   this->ClickOnCancel(); /* same than click on cancel button */
524 }
525
526 //=================================================================================
527 // function : hideEvent()
528 // purpose  : caused by ESC key
529 //=================================================================================
530 void SMESHGUI_NodesDlg::hideEvent (QHideEvent*)
531 {
532   if (!isMinimized())
533     ClickOnCancel();
534 }
535
536 //=================================================================================
537 // function : enterEvent()
538 // purpose  : to reactivate this dialog box when mouse enter onto the window
539 //=================================================================================
540 void SMESHGUI_NodesDlg::enterEvent(QEvent*)
541 {
542   if (!GroupConstructors->isEnabled())
543     ActivateThisDialog();
544 }
545
546 //=================================================================================
547 // function : DeactivateActiveDialog()
548 // purpose  : public slot to deactivate if active
549 //=================================================================================
550 void SMESHGUI_NodesDlg::DeactivateActiveDialog()
551 {
552   if (GroupConstructors->isEnabled()) {
553     GroupConstructors->setEnabled(false);
554     GroupCoordinates->setEnabled(false);
555     GroupButtons->setEnabled(false);
556     mySimulation->SetVisibility(false);
557     mySMESHGUI->ResetState();
558     mySMESHGUI->SetActiveDialogBox(0);
559   }
560 }
561
562 //=================================================================================
563 // function : ActivateThisDialog()
564 // purpose  :
565 //=================================================================================
566 void SMESHGUI_NodesDlg::ActivateThisDialog()
567 {
568   mySMESHGUI->EmitSignalDeactivateDialog();
569   GroupConstructors->setEnabled(true);
570   GroupCoordinates->setEnabled(true);
571   GroupButtons->setEnabled(true);
572
573   SMESH::SetPointRepresentation(true);
574   myViewWindow->SetSelectionMode(NodeSelection);
575
576   SelectionIntoArgument();
577 }