]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_AddMeshElementDlg.cxx
Salome HOME
3108f8d65a8a6b2c41bbc9b339a05a1889c6d92e
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_AddMeshElementDlg.cxx
1 //  Copyright (C) 2007-2010  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 // SMESH SMESHGUI : GUI for SMESH component
24 // File   : SMESHGUI_AddMeshElementDlg.cxx
25 // Author : Nicolas REJNERI, Open CASCADE S.A.S.
26 //  SMESH includes
27 //
28 #include "SMESHGUI_AddMeshElementDlg.h"
29
30 #include "SMESHGUI.h"
31 #include "SMESHGUI_Utils.h"
32 #include "SMESHGUI_VTKUtils.h"
33 #include "SMESHGUI_MeshUtils.h"
34 #include "SMESHGUI_GroupUtils.h"
35 #include "SMESHGUI_IdValidator.h"
36
37 #include <SMESH_Actor.h>
38 #include <SMESH_ActorUtils.h>
39 #include <SMESH_FaceOrientationFilter.h>
40 #include <SMDS_Mesh.hxx>
41
42 // SALOME GUI inclues
43 #include <SUIT_Desktop.h>
44 #include <SUIT_Session.h>
45 #include <SUIT_ResourceMgr.h>
46 #include <SUIT_MessageBox.h>
47 #include <SUIT_ViewManager.h>
48 #include <LightApp_SelectionMgr.h>
49 #include <SALOME_ListIO.hxx>
50 #include <SalomeApp_Application.h>
51 #include <SVTK_ViewModel.h>
52 #include <SVTK_ViewWindow.h>
53
54 // IDL incldues
55 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
56
57 // OCCT includes
58 #include <TColStd_MapOfInteger.hxx>
59
60 // VTK includes
61 #include <vtkCell.h>
62 #include <vtkIdList.h>
63 #include <vtkUnstructuredGrid.h>
64 #include <vtkDataSetMapper.h>
65 #include <vtkPolyDataMapper.h>
66 #include <vtkProperty.h>
67
68 // Qt includes
69 #include <QComboBox>
70 #include <QGroupBox>
71 #include <QLabel>
72 #include <QLineEdit>
73 #include <QPushButton>
74 #include <QRadioButton>
75 #include <QHBoxLayout>
76 #include <QVBoxLayout>
77 #include <QGridLayout>
78 #include <QVariant>
79 #include <QCheckBox>
80 #include <QKeyEvent>
81 #include <QButtonGroup>
82
83 #define SPACING 6
84 #define MARGIN  11
85
86 namespace SMESH
87 {
88   class TElementSimulation
89   {
90     SalomeApp_Application* myApplication;
91     SUIT_ViewWindow* myViewWindow;
92     SVTK_ViewWindow* myVTKViewWindow;
93
94     SALOME_Actor* myPreviewActor;
95     vtkDataSetMapper* myMapper;
96     vtkUnstructuredGrid* myGrid;
97
98     SALOME_Actor* myFaceOrientation;
99     vtkPolyDataMapper* myFaceOrientationDataMapper;
100     SMESH_FaceOrientationFilter* myFaceOrientationFilter;
101
102   public:
103     TElementSimulation (SalomeApp_Application* theApplication)
104     {
105       myApplication = theApplication;
106       SUIT_ViewManager* mgr = theApplication->activeViewManager();
107       if (!mgr) return;
108       myViewWindow = mgr->getActiveView();
109       myVTKViewWindow = GetVtkViewWindow(myViewWindow);
110
111       myGrid = vtkUnstructuredGrid::New();
112
113       // Create and display actor
114       myMapper = vtkDataSetMapper::New();
115       myMapper->SetInput(myGrid);
116
117       myPreviewActor = SALOME_Actor::New();
118       myPreviewActor->PickableOff();
119       myPreviewActor->VisibilityOff();
120       myPreviewActor->SetMapper(myMapper);
121
122       vtkFloatingPointType anRGB[3];
123       vtkProperty* aProp = vtkProperty::New();
124       GetColor( "SMESH", "fill_color", anRGB[0], anRGB[1], anRGB[2], QColor( 0, 170, 255 ) );
125       aProp->SetColor( anRGB[0], anRGB[1], anRGB[2] );
126       myPreviewActor->SetProperty( aProp );
127       aProp->Delete();
128
129       vtkProperty* aBackProp = vtkProperty::New();
130       GetColor( "SMESH", "backface_color", anRGB[0], anRGB[1], anRGB[2], QColor( 0, 0, 255 ) );
131       aBackProp->SetColor( anRGB[0], anRGB[1], anRGB[2] );
132       myPreviewActor->SetBackfaceProperty( aBackProp );
133       aBackProp->Delete();
134
135       myVTKViewWindow->AddActor(myPreviewActor);
136
137       // Orientation of faces
138       myFaceOrientationFilter = SMESH_FaceOrientationFilter::New();
139       myFaceOrientationFilter->SetInput(myGrid);
140
141       myFaceOrientationDataMapper = vtkPolyDataMapper::New();
142       myFaceOrientationDataMapper->SetInput(myFaceOrientationFilter->GetOutput());
143
144       myFaceOrientation = SALOME_Actor::New();
145       myFaceOrientation->PickableOff();
146       myFaceOrientation->VisibilityOff();
147       myFaceOrientation->SetMapper(myFaceOrientationDataMapper);
148
149       vtkProperty* anOrientationProp = vtkProperty::New();
150       GetColor( "SMESH", "orientation_color", anRGB[0], anRGB[1], anRGB[2], QColor( 255, 255, 255 ) );
151       anOrientationProp->SetColor( anRGB[0], anRGB[1], anRGB[2] );
152       myFaceOrientation->SetProperty( anOrientationProp );
153       anOrientationProp->Delete();
154
155       myVTKViewWindow->AddActor(myFaceOrientation);
156     }
157
158     typedef std::vector<vtkIdType> TVTKIds;
159     void SetPosition (SMESH_Actor* theActor,
160                       vtkIdType theType,
161                       const TVTKIds& theIds)
162     {
163       vtkUnstructuredGrid *aGrid = theActor->GetUnstructuredGrid();
164       myGrid->SetPoints(aGrid->GetPoints());
165
166       const int* aConn = NULL;
167       switch (theType) {
168       case VTK_TETRA:
169         {
170           static int anIds[] = {0,2,1,3};
171           aConn = anIds;
172           break;
173         }
174       case VTK_PYRAMID:
175         {
176           static int anIds[] = {0,3,2,1,4};
177           aConn = anIds;
178           break;
179         }
180       case VTK_HEXAHEDRON:
181         {
182           static int anIds[] = {0,3,2,1,4,7,6,5};
183           aConn = anIds;
184           break;
185         }
186       }
187
188       myGrid->Reset();
189       vtkIdList *anIds = vtkIdList::New();
190
191       if(aConn)
192         for (int i = 0, iEnd = theIds.size(); i < iEnd; i++)
193           anIds->InsertId(i,theIds[aConn[i]]);
194       else
195         for (int i = 0, iEnd = theIds.size(); i < iEnd; i++)
196           anIds->InsertId(i,theIds[i]);
197
198       myGrid->InsertNextCell(theType,anIds);
199       anIds->Delete();
200
201       myGrid->Modified();
202
203       SetVisibility(true, theActor->GetFacesOriented());
204     }
205
206
207     void SetVisibility (bool theVisibility, bool theShowOrientation = false)
208     {
209       myPreviewActor->SetVisibility(theVisibility);
210       myFaceOrientation->SetVisibility(theShowOrientation);
211       RepaintCurrentView();
212     }
213
214
215     ~TElementSimulation()
216     {
217       if (FindVtkViewWindow(myApplication->activeViewManager(), myViewWindow)) {
218         myVTKViewWindow->RemoveActor(myPreviewActor);
219         myVTKViewWindow->RemoveActor(myFaceOrientation);
220       }
221       myPreviewActor->Delete();
222       myFaceOrientation->Delete();
223
224       myMapper->RemoveAllInputs();
225       myMapper->Delete();
226
227       myFaceOrientationFilter->Delete();
228
229       myFaceOrientationDataMapper->RemoveAllInputs();
230       myFaceOrientationDataMapper->Delete();
231
232       myGrid->Delete();
233     }
234   };
235 }
236
237 //=================================================================================
238 // function : SMESHGUI_AddMeshElementDlg()
239 // purpose  : constructor
240 //=================================================================================
241 SMESHGUI_AddMeshElementDlg::SMESHGUI_AddMeshElementDlg( SMESHGUI* theModule,
242                                                         SMDSAbs_ElementType ElementType,
243                                                         int nbNodes )
244   : QDialog( SMESH::GetDesktop( theModule ) ),
245     mySMESHGUI( theModule ),
246     mySelectionMgr( SMESH::GetSelectionMgr( theModule ) )
247 {
248   setModal( false );
249   setAttribute( Qt::WA_DeleteOnClose, true );
250
251   SalomeApp_Application* anApp = dynamic_cast<SalomeApp_Application*>
252     (SUIT_Session::session()->activeApplication());
253   myIsPoly = false;
254   mySimulation = new SMESH::TElementSimulation (anApp);
255   mySelector = (SMESH::GetViewWindow( mySMESHGUI ))->GetSelector();
256
257   // verify nb nodes and type
258   myNbNodes = nbNodes;
259   myElementType = ElementType;
260   switch (ElementType) {
261   case SMDSAbs_0DElement:
262     if (myNbNodes != 1)
263       myNbNodes = 1;
264     break;
265   case SMDSAbs_Face:
266     //     if (myNbNodes != 3 && myNbNodes != 4)
267     //       myNbNodes = 3;
268     //     break;
269   case SMDSAbs_Volume:
270     //     if (myNbNodes != 4 && myNbNodes != 8) //(nbNodes < 4 || nbNodes > 8 || nbNodes == 7)
271     //       myNbNodes = 4;
272     break;
273   default:
274     myElementType = SMDSAbs_Edge;
275     myNbNodes = 2;
276   }
277
278   QString elemName;
279   if (myNbNodes == 1) {
280     elemName = "ELEM0D";
281     myHelpFileName = "adding_nodes_and_elements_page.html#adding_0delems_anchor";
282   }
283   else if (myNbNodes == 2) {
284     elemName = "EDGE";
285     myHelpFileName = "adding_nodes_and_elements_page.html#adding_edges_anchor";
286   }
287   else if (myNbNodes == 3) {
288     elemName = "TRIANGLE";
289     myHelpFileName = "adding_nodes_and_elements_page.html#adding_triangles_anchor";
290   }
291   else if (myNbNodes == 4) {
292     if (myElementType == SMDSAbs_Face) {
293       elemName = "QUADRANGLE";
294       myHelpFileName = "adding_nodes_and_elements_page.html#adding_quadrangles_anchor";
295     }
296     else {
297       elemName = "TETRAS";
298       myHelpFileName = "adding_nodes_and_elements_page.html#adding_tetrahedrons_anchor";
299     }
300   }
301   else if (myNbNodes == 8) {
302     elemName = "HEXAS";
303     myHelpFileName = "adding_nodes_and_elements_page.html#adding_hexahedrons_anchor";
304   }
305   else if (myElementType == SMDSAbs_Face) {
306     elemName = "POLYGON";
307     myIsPoly = true;
308     myHelpFileName = "adding_nodes_and_elements_page.html#adding_polygons_anchor";
309   }
310   else if (myElementType == SMDSAbs_Volume) {
311     myHelpFileName = "adding_nodes_and_elements_page.html#adding_polyhedrons_anchor";
312   }
313
314   QString iconName      = tr(QString("ICON_DLG_%1").arg(elemName).toLatin1().data());
315   QString buttonGrTitle = tr(QString("SMESH_%1").arg(elemName).toLatin1().data());
316   QString caption       = tr(QString("SMESH_ADD_%1_TITLE").arg(elemName).toLatin1().data());
317   QString grBoxTitle    = tr(QString("SMESH_ADD_%1").arg(elemName).toLatin1().data());
318
319   QPixmap image0 (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", iconName));
320   QPixmap image1 (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
321
322   setWindowTitle(caption);
323   setSizeGripEnabled(true);
324
325   QVBoxLayout* aTopLayout = new QVBoxLayout(this);
326   aTopLayout->setSpacing(SPACING);
327   aTopLayout->setMargin(MARGIN);
328
329   /***************************************************************/
330   GroupConstructors = new QGroupBox(buttonGrTitle, this);
331   QButtonGroup* ButtonGroup = new QButtonGroup(this);
332   QHBoxLayout* GroupConstructorsLayout = new QHBoxLayout(GroupConstructors);
333   GroupConstructorsLayout->setSpacing(SPACING);
334   GroupConstructorsLayout->setMargin(MARGIN);
335
336   Constructor1 = new QRadioButton(GroupConstructors);
337   Constructor1->setIcon(image0);
338   Constructor1->setChecked(true);
339
340   GroupConstructorsLayout->addWidget(Constructor1);
341   ButtonGroup->addButton( Constructor1, 0 );
342
343   /***************************************************************/
344   GroupC1 = new QGroupBox(grBoxTitle, this);
345   QGridLayout* GroupC1Layout = new QGridLayout(GroupC1);
346   GroupC1Layout->setSpacing(SPACING);
347   GroupC1Layout->setMargin(MARGIN);
348
349   TextLabelC1A1 = new QLabel(tr("SMESH_ID_NODES"), GroupC1);
350   SelectButtonC1A1 = new QPushButton(GroupC1);
351   SelectButtonC1A1->setIcon(image1);
352   LineEditC1A1 = new QLineEdit(GroupC1);
353   //  LineEditC1A1->setReadOnly(true);
354   if (!myIsPoly)
355     LineEditC1A1->setValidator(new SMESHGUI_IdValidator(this, myNbNodes));
356
357   Reverse = myElementType == SMDSAbs_Face ? new QCheckBox(tr("SMESH_REVERSE"), GroupC1) : 0;
358
359   GroupC1Layout->addWidget(TextLabelC1A1,    0, 0);
360   GroupC1Layout->addWidget(SelectButtonC1A1, 0, 1);
361   GroupC1Layout->addWidget(LineEditC1A1,     0, 2);
362   if ( Reverse ) GroupC1Layout->addWidget(Reverse, 1, 0, 1, 3);
363
364   /***************************************************************/
365   GroupGroups = new QGroupBox( tr( "SMESH_ADD_TO_GROUP" ), this );
366   GroupGroups->setCheckable( true );
367   QHBoxLayout* GroupGroupsLayout = new QHBoxLayout(GroupGroups);
368   GroupGroupsLayout->setSpacing(SPACING);
369   GroupGroupsLayout->setMargin(MARGIN);
370
371   TextLabel_GroupName = new QLabel( tr( "SMESH_GROUP" ), GroupGroups );
372   ComboBox_GroupName = new QComboBox( GroupGroups );
373   ComboBox_GroupName->setEditable( true );
374
375   GroupGroupsLayout->addWidget( TextLabel_GroupName );
376   GroupGroupsLayout->addWidget( ComboBox_GroupName, 1 );
377
378   /***************************************************************/
379   GroupButtons = new QGroupBox(this);
380   QHBoxLayout* GroupButtonsLayout = new QHBoxLayout(GroupButtons);
381   GroupButtonsLayout->setSpacing(SPACING);
382   GroupButtonsLayout->setMargin(MARGIN);
383
384   buttonOk = new QPushButton(tr("SMESH_BUT_APPLY_AND_CLOSE"), GroupButtons);
385   buttonOk->setAutoDefault(true);
386   buttonOk->setDefault(true);
387   buttonApply = new QPushButton(tr("SMESH_BUT_APPLY"), GroupButtons);
388   buttonApply->setAutoDefault(true);
389   buttonCancel = new QPushButton(tr("SMESH_BUT_CLOSE"), GroupButtons);
390   buttonCancel->setAutoDefault(true);
391   buttonHelp = new QPushButton(tr("SMESH_BUT_HELP"), GroupButtons);
392   buttonHelp->setAutoDefault(true);
393
394   GroupButtonsLayout->addWidget(buttonOk);
395   GroupButtonsLayout->addSpacing(10);
396   GroupButtonsLayout->addWidget(buttonApply);
397   GroupButtonsLayout->addSpacing(10);
398   GroupButtonsLayout->addStretch();
399   GroupButtonsLayout->addWidget(buttonCancel);
400   GroupButtonsLayout->addWidget(buttonHelp);
401
402   /***************************************************************/
403   aTopLayout->addWidget(GroupConstructors);
404   aTopLayout->addWidget(GroupC1);
405   aTopLayout->addWidget(GroupGroups);
406   aTopLayout->addWidget(GroupButtons);
407
408   Init(); /* Initialisations */
409 }
410
411 //=================================================================================
412 // function : ~SMESHGUI_AddMeshElementDlg()
413 // purpose  : Destroys the object and frees any allocated resources
414 //=================================================================================
415 SMESHGUI_AddMeshElementDlg::~SMESHGUI_AddMeshElementDlg()
416 {
417   delete mySimulation;
418 }
419
420 //=================================================================================
421 // function : Init()
422 // purpose  :
423 //=================================================================================
424 void SMESHGUI_AddMeshElementDlg::Init()
425 {
426   GroupC1->show();
427   Constructor1->setChecked(true);
428   myEditCurrentArgument = LineEditC1A1;
429   mySMESHGUI->SetActiveDialogBox((QDialog*)this);
430
431   /* reset "Add to group" control */
432   GroupGroups->setChecked( false );
433   GroupGroups->setVisible( myElementType != SMDSAbs_0DElement );
434
435   myNbOkNodes = 0;
436   myActor = 0;
437
438   /* signals and slots connections */
439   connect(buttonOk, SIGNAL(clicked()),     SLOT(ClickOnOk()));
440   connect(buttonCancel, SIGNAL(clicked()), SLOT(ClickOnCancel()));
441   connect(buttonApply, SIGNAL(clicked()),  SLOT(ClickOnApply()));
442   connect(buttonHelp, SIGNAL(clicked()),   SLOT(ClickOnHelp()));
443
444   connect(SelectButtonC1A1, SIGNAL(clicked()), SLOT(SetEditCurrentArgument()));
445   connect(LineEditC1A1, SIGNAL(textChanged(const QString&)), SLOT(onTextChange(const QString&)));
446   connect(mySMESHGUI, SIGNAL (SignalDeactivateActiveDialog()), SLOT(DeactivateActiveDialog()));
447   connect(mySelectionMgr, SIGNAL(currentSelectionChanged()), SLOT(SelectionIntoArgument()));
448   /* to close dialog if study frame change */
449   connect(mySMESHGUI, SIGNAL (SignalStudyFrameChanged()), SLOT(ClickOnCancel()));
450
451   if (Reverse)
452     connect(Reverse, SIGNAL(stateChanged(int)), SLOT(CheckBox(int)));
453
454   // set selection mode
455   SMESH::SetPointRepresentation(true);
456
457   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
458     aViewWindow->SetSelectionMode( NodeSelection );
459
460   myBusy = false;
461
462   SelectionIntoArgument();
463 }
464
465 //=================================================================================
466 // function : ClickOnApply()
467 // purpose  :
468 //=================================================================================
469 void SMESHGUI_AddMeshElementDlg::ClickOnApply()
470 {
471   if( !isValid() )
472     return;
473
474   if (myNbOkNodes && !mySMESHGUI->isActiveStudyLocked()) {
475     myBusy = true;
476     SMESH::long_array_var anArrayOfIndices = new SMESH::long_array;
477     anArrayOfIndices->length(myNbNodes);
478     bool reverse = (Reverse && Reverse->isChecked());
479     QStringList aListId = myEditCurrentArgument->text().split(" ", QString::SkipEmptyParts);
480     for (int i = 0; i < aListId.count(); i++)
481       if (reverse)
482         anArrayOfIndices[i] = aListId[ myNbNodes - i - 1 ].toInt();
483       else
484         anArrayOfIndices[i] = aListId[ i ].toInt();
485
486     bool addToGroup = GroupGroups->isChecked();
487     QString aGroupName;
488
489     SMESH::SMESH_GroupBase_var aGroup;
490     int idx = 0;
491     if( addToGroup ) {
492       aGroupName = ComboBox_GroupName->currentText();
493       for ( int i = 1; i < ComboBox_GroupName->count(); i++ ) {
494         QString aName = ComboBox_GroupName->itemText( i );
495         if ( aGroupName == aName && ( i == ComboBox_GroupName->currentIndex() || idx == 0 ) )
496           idx = i;
497       }
498       if ( idx > 0 ) {
499         SMESH::SMESH_GroupOnGeom_var aGeomGroup = SMESH::SMESH_GroupOnGeom::_narrow( myGroups[idx-1] );
500         if ( !aGeomGroup->_is_nil() ) {
501           int res = SUIT_MessageBox::question( this, tr( "SMESH_WRN_WARNING" ),
502                                                tr( "MESH_STANDALONE_GRP_CHOSEN" ).arg( aGroupName ),
503                                                tr( "SMESH_BUT_YES" ), tr( "SMESH_BUT_NO" ), 0, 1 );
504           if ( res == 1 ) return;
505         }
506         aGroup = myGroups[idx-1];
507       }
508     }
509
510     long anElemId = -1;
511     SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditor();
512     switch (myElementType) {
513     case SMDSAbs_0DElement:
514       anElemId = aMeshEditor->Add0DElement(anArrayOfIndices[0]); break;
515     case SMDSAbs_Edge:
516       anElemId = aMeshEditor->AddEdge(anArrayOfIndices.inout()); break;
517     case SMDSAbs_Face: {
518       if(myIsPoly)
519         anElemId = aMeshEditor->AddPolygonalFace(anArrayOfIndices.inout());
520       else
521         anElemId = aMeshEditor->AddFace(anArrayOfIndices.inout());
522       break;
523     }
524     case SMDSAbs_Volume:
525       anElemId = aMeshEditor->AddVolume(anArrayOfIndices.inout()); break;
526     default: break;
527     }
528
529     if ( anElemId != -1 && addToGroup && !aGroupName.isEmpty() ) {
530       SMESH::SMESH_Group_var aGroupUsed;
531       if ( aGroup->_is_nil() ) {
532         // create new group 
533         aGroupUsed = SMESH::AddGroup( myMesh, (SMESH::ElementType)myElementType, aGroupName );
534         if ( !aGroupUsed->_is_nil() ) {
535           myGroups.append(SMESH::SMESH_GroupBase::_duplicate(aGroupUsed));
536           ComboBox_GroupName->addItem( aGroupName );
537         }
538       }
539       else {
540         SMESH::SMESH_GroupOnGeom_var aGeomGroup = SMESH::SMESH_GroupOnGeom::_narrow( aGroup );
541         if ( !aGeomGroup->_is_nil() ) {
542           aGroupUsed = myMesh->ConvertToStandalone( aGeomGroup );
543           if ( !aGroupUsed->_is_nil() && idx > 0 ) {
544             myGroups[idx-1] = SMESH::SMESH_GroupBase::_duplicate(aGroupUsed);
545             SMESHGUI::GetSMESHGUI()->getApp()->updateObjectBrowser();
546           }
547         }
548         else
549           aGroupUsed = SMESH::SMESH_Group::_narrow( aGroup );
550       }
551
552       if ( !aGroupUsed->_is_nil() ) {
553         SMESH::long_array_var anIdList = new SMESH::long_array;
554         anIdList->length( 1 );
555         anIdList[0] = anElemId;
556         aGroupUsed->Add( anIdList.inout() );
557       }
558     }
559
560     SALOME_ListIO aList; aList.Append( myActor->getIO() );
561     mySelector->ClearIndex();
562     mySelectionMgr->setSelectedObjects( aList, false );
563
564     SMESH::UpdateView();
565     mySimulation->SetVisibility(false);
566
567     buttonOk->setEnabled(false);
568     buttonApply->setEnabled(false);
569
570     myEditCurrentArgument->setText("");
571
572     myBusy = false;
573
574     SMESHGUI::Modified();
575   }
576 }
577
578 //=================================================================================
579 // function : ClickOnOk()
580 // purpose  :
581 //=================================================================================
582 void SMESHGUI_AddMeshElementDlg::ClickOnOk()
583 {
584   ClickOnApply();
585   ClickOnCancel();
586 }
587
588 //=================================================================================
589 // function : ClickOnCancel()
590 // purpose  :
591 //=================================================================================
592 void SMESHGUI_AddMeshElementDlg::ClickOnCancel()
593 {
594   //mySelectionMgr->clearSelected();
595   mySimulation->SetVisibility(false);
596   SMESH::SetPointRepresentation(false);
597   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
598     aViewWindow->SetSelectionMode( ActorSelection );
599   disconnect(mySelectionMgr, 0, this, 0);
600   mySMESHGUI->ResetState();
601   reject();
602 }
603
604 //=================================================================================
605 // function : ClickOnHelp()
606 // purpose  :
607 //=================================================================================
608 void SMESHGUI_AddMeshElementDlg::ClickOnHelp()
609 {
610   LightApp_Application* app = (LightApp_Application*)(SUIT_Session::session()->activeApplication());
611   if (app)
612     app->onHelpContextModule(mySMESHGUI ? app->moduleName(mySMESHGUI->moduleName()) : QString(""),
613                              myHelpFileName);
614   else {
615     QString platform;
616 #ifdef WIN32
617     platform = "winapplication";
618 #else
619     platform = "application";
620 #endif
621     SUIT_MessageBox::warning(this, tr("WRN_WARNING"),
622                              tr("EXTERNAL_BROWSER_CANNOT_SHOW_PAGE").
623                              arg(app->resourceMgr()->stringValue("ExternalBrowser",
624                                                                  platform)).
625                              arg(myHelpFileName));
626   }
627 }
628
629 //=================================================================================
630 // function : onTextChange()
631 // purpose  :
632 //=================================================================================
633 void SMESHGUI_AddMeshElementDlg::onTextChange (const QString& theNewText)
634 {
635   if (myBusy) return;
636   myBusy = true;
637
638   myNbOkNodes = 0;
639
640   buttonOk->setEnabled(false);
641   buttonApply->setEnabled(false);
642
643   mySimulation->SetVisibility(false);
644
645   // hilight entered nodes
646   SMDS_Mesh* aMesh = 0;
647   if (myActor)
648     aMesh = myActor->GetObject()->GetMesh();
649
650   if (aMesh) {
651     TColStd_MapOfInteger newIndices;
652
653     QStringList aListId = theNewText.split(" ", QString::SkipEmptyParts);
654     bool allOk = true;
655     for (int i = 0; i < aListId.count(); i++) {
656       if( const SMDS_MeshNode * n = aMesh->FindNode( aListId[ i ].toInt() ) )
657         {
658           newIndices.Add( n->GetID() );
659           myNbOkNodes++;
660         }
661       else
662         allOk = false;  
663     }
664
665     mySelector->AddOrRemoveIndex( myActor->getIO(), newIndices, false );
666     if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
667       aViewWindow->highlight( myActor->getIO(), true, true );
668
669     myNbOkNodes = ( allOk && myNbNodes == aListId.count() );
670
671     if (myIsPoly)
672       {
673         if ( !allOk || myElementType != SMDSAbs_Face || aListId.count() < 3 )
674           myNbOkNodes = 0;
675         else
676           myNbOkNodes = aListId.count();
677       }
678   }
679
680   if(myNbOkNodes) {
681     buttonOk->setEnabled(true);
682     buttonApply->setEnabled(true);
683     displaySimulation();
684   }
685
686   myBusy = false;
687 }
688
689 //=================================================================================
690 // function : SelectionIntoArgument()
691 // purpose  : Called when selection has changed
692 //=================================================================================
693 void SMESHGUI_AddMeshElementDlg::SelectionIntoArgument()
694 {
695   if (myBusy) return;
696
697   // clear
698   myNbOkNodes = 0;
699   myActor = 0;
700
701   myBusy = true;
702   myEditCurrentArgument->setText("");
703   myBusy = false;
704
705   if (!GroupButtons->isEnabled()) // inactive
706     return;
707
708   buttonOk->setEnabled(false);
709   buttonApply->setEnabled(false);
710
711   mySimulation->SetVisibility(false);
712   //  SMESH::SetPointRepresentation(true);
713
714   QString aCurrentEntry = myEntry;
715
716   // get selected mesh
717   SALOME_ListIO aList;
718   mySelectionMgr->selectedObjects(aList,SVTK_Viewer::Type());
719
720   if (aList.Extent() != 1)
721     return;
722
723   Handle(SALOME_InteractiveObject) anIO = aList.First();
724   myEntry = anIO->getEntry();
725   myMesh = SMESH::GetMeshByIO(anIO);
726   if (myMesh->_is_nil())
727     return;
728
729   // process groups
730   if ( !myMesh->_is_nil() && myEntry != aCurrentEntry ) {
731     myGroups.clear();
732     ComboBox_GroupName->clear();
733     ComboBox_GroupName->addItem( QString() );
734     SMESH::ListOfGroups aListOfGroups = *myMesh->GetGroups();
735     for ( int i = 0, n = aListOfGroups.length(); i < n; i++ ) {
736       SMESH::SMESH_GroupBase_var aGroup = aListOfGroups[i];
737       if ( !aGroup->_is_nil() && aGroup->GetType() == (SMESH::ElementType)myElementType ) {
738         QString aGroupName( aGroup->GetName() );
739         if ( !aGroupName.isEmpty() ) {
740           myGroups.append(SMESH::SMESH_GroupBase::_duplicate(aGroup));
741           ComboBox_GroupName->addItem( aGroupName );
742         }
743       }
744     }
745   }
746
747   myActor = SMESH::FindActorByEntry(anIO->getEntry());
748   if (!myActor)
749     return;
750
751   // get selected nodes
752   QString aString = "";
753   int nbNodes = SMESH::GetNameOfSelectedNodes(mySelector,myActor->getIO(),aString);
754   myBusy = true;
755   myEditCurrentArgument->setText(aString);
756   myBusy = false;
757   if (myIsPoly && myElementType == SMDSAbs_Face && nbNodes >= 3 ) {
758     myNbNodes = nbNodes;
759   } else if (myNbNodes != nbNodes) {
760     return;
761   }
762
763   // OK
764   myNbOkNodes = nbNodes;
765
766   buttonOk->setEnabled(true);
767   buttonApply->setEnabled(true);
768
769   displaySimulation();
770 }
771
772 //=================================================================================
773 // function : displaySimulation()
774 // purpose  :
775 //=================================================================================
776 void SMESHGUI_AddMeshElementDlg::displaySimulation()
777 {
778   if (myNbOkNodes && GroupButtons->isEnabled()) {
779     SMESH::TElementSimulation::TVTKIds anIds;
780     QStringList aListId = myEditCurrentArgument->text().split(" ", QString::SkipEmptyParts);
781     for (int i = 0; i < aListId.count(); i++)
782       anIds.push_back(myActor->GetObject()->GetNodeVTKId(aListId[ i ].toInt()));
783
784     if (Reverse && Reverse->isChecked())
785       reverse(anIds.begin(),anIds.end());
786
787     vtkIdType aType = 0;
788     if (myIsPoly)
789       switch ( myElementType ) {
790       case SMDSAbs_Face  : aType = VTK_POLYGON; break;
791       default: return;
792       }
793     else {
794       switch (myNbNodes) {
795       case 2: aType = VTK_LINE; break;
796       case 3: aType = VTK_TRIANGLE; break;
797       case 4: aType = myElementType == SMDSAbs_Face ? VTK_QUAD : VTK_TETRA; break;
798       case 8: aType = VTK_HEXAHEDRON; break;
799       default: return;
800       }
801     }
802
803     mySimulation->SetPosition(myActor,aType,anIds);
804     SMESH::UpdateView();
805   }
806 }
807
808 //=================================================================================
809 // function : SetEditCurrentArgument()
810 // purpose  :
811 //=================================================================================
812 void SMESHGUI_AddMeshElementDlg::SetEditCurrentArgument()
813 {
814   QPushButton* send = (QPushButton*)sender();
815   if (send == SelectButtonC1A1) {
816     LineEditC1A1->setFocus();
817     myEditCurrentArgument = LineEditC1A1;
818   }
819   SelectionIntoArgument();
820 }
821
822 //=================================================================================
823 // function : DeactivateActiveDialog()
824 // purpose  :
825 //=================================================================================
826 void SMESHGUI_AddMeshElementDlg::DeactivateActiveDialog()
827 {
828   if (GroupConstructors->isEnabled()) {
829     GroupConstructors->setEnabled(false);
830     GroupC1->setEnabled(false);
831     GroupButtons->setEnabled(false);
832     mySimulation->SetVisibility(false);
833     mySMESHGUI->ResetState();
834     mySMESHGUI->SetActiveDialogBox(0);
835   }
836 }
837
838 //=================================================================================
839 // function : ActivateThisDialog()
840 // purpose  :
841 //=================================================================================
842 void SMESHGUI_AddMeshElementDlg::ActivateThisDialog()
843 {
844   /* Emit a signal to deactivate the active dialog */
845   mySMESHGUI->EmitSignalDeactivateDialog();
846
847   GroupConstructors->setEnabled(true);
848   GroupC1->setEnabled(true);
849   GroupButtons->setEnabled(true);
850
851   SMESH::SetPointRepresentation(true);
852
853   if ( SVTK_ViewWindow* aViewWindow = SMESH::GetViewWindow( mySMESHGUI ))
854     aViewWindow->SetSelectionMode( NodeSelection );
855   SelectionIntoArgument();
856 }
857
858 //=================================================================================
859 // function : enterEvent()
860 // purpose  :
861 //=================================================================================
862 void SMESHGUI_AddMeshElementDlg::enterEvent (QEvent*)
863 {
864   if (GroupConstructors->isEnabled())
865     return;
866   ActivateThisDialog();
867 }
868
869 //=================================================================================
870 // function : closeEvent()
871 // purpose  :
872 //=================================================================================
873 void SMESHGUI_AddMeshElementDlg::closeEvent (QCloseEvent*)
874 {
875   /* same than click on cancel button */
876   ClickOnCancel();
877 }
878
879 //=================================================================================
880 // function : hideEvent()
881 // purpose  : caused by ESC key
882 //=================================================================================
883 void SMESHGUI_AddMeshElementDlg::hideEvent (QHideEvent*)
884 {
885   if (!isMinimized())
886     ClickOnCancel();
887 }
888
889 //=================================================================================
890 // function : CheckBox()
891 // purpose  :
892 //=================================================================================
893 void SMESHGUI_AddMeshElementDlg::CheckBox (int state)
894 {
895   if (!myNbOkNodes)
896     return;
897
898   if (state >= 0) {
899     mySimulation->SetVisibility(false);
900     displaySimulation();
901   }
902 }
903
904 //=================================================================================
905 // function : keyPressEvent()
906 // purpose  :
907 //=================================================================================
908 void SMESHGUI_AddMeshElementDlg::keyPressEvent( QKeyEvent* e )
909 {
910   QDialog::keyPressEvent( e );
911   if ( e->isAccepted() )
912     return;
913
914   if ( e->key() == Qt::Key_F1 ) {
915     e->accept();
916     ClickOnHelp();
917   }
918 }
919
920 //=================================================================================
921 // function : isValid
922 // purpose  :
923 //=================================================================================
924 bool SMESHGUI_AddMeshElementDlg::isValid()
925 {
926   if( GroupGroups->isChecked() && ComboBox_GroupName->currentText().isEmpty() ) {
927     SUIT_MessageBox::warning( this, tr( "SMESH_WRN_WARNING" ), tr( "GROUP_NAME_IS_EMPTY" ) );
928     return false;
929   }
930   return true;
931 }