]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_Add0DElemsOnAllNodesDlg.cxx
Salome HOME
Unicode support: correct handling of unicode on GUI level
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_Add0DElemsOnAllNodesDlg.cxx
1 // Copyright (C) 2007-2016  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, or (at your option) any later version.
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_Add0DElemsOnAllNodesDlg.cxx
25 // Created   : Fri Oct 19 15:51:24 2012
26 // Author    : Edward AGAPOV (eap)
27
28 #include "SMESHGUI_Add0DElemsOnAllNodesDlg.h"
29
30 #include "SMESHGUI.h"
31 #include "SMESHGUI_FilterDlg.h"
32 #include "SMESHGUI_MeshUtils.h"
33 #include "SMESHGUI_Utils.h"
34 #include "SMESHGUI_VTKUtils.h"
35 #include "SMESH_LogicalFilter.hxx"
36 #include "SMESH_TypeFilter.hxx"
37
38 #include <LightApp_SelectionMgr.h>
39 #include <SALOME_ListIO.hxx>
40 #include <SUIT_MessageBox.h>
41 #include <SVTK_Selector.h>
42 #include <SalomeApp_Tools.h>
43
44 // Qt includes
45 #include <QButtonGroup>
46 #include <QCheckBox>
47 #include <QComboBox>
48 #include <QFrame>
49 #include <QGridLayout>
50 #include <QGroupBox>
51 #include <QHBoxLayout>
52 #include <QLabel>
53 #include <QLineEdit>
54 #include <QPushButton>
55 #include <QRadioButton>
56
57 #include <SALOMEconfig.h>
58 #include CORBA_SERVER_HEADER(SMESH_Mesh)
59 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
60
61
62 #define SPACING 6
63 #define MARGIN  11
64
65 enum { SEL_OBJECT, SEL_ELEMENTS, SEL_NODES };
66
67 //================================================================================
68 /*!
69  * \brief Dialog Constructor
70  */
71 //================================================================================
72
73 SMESHGUI_Add0DElemsOnAllNodesDlg::SMESHGUI_Add0DElemsOnAllNodesDlg()
74   : SMESHGUI_Dialog( 0, false, true ),
75     myIDValidator( this )
76 {
77   setWindowTitle( tr( "CAPTION" ) );
78
79   // Seletction type radio buttons
80
81   QGroupBox* selTypeGrBox = new QGroupBox( mainFrame() );
82   //
83   QRadioButton*    objBtn = new QRadioButton( tr( "SMESH_SUBMESH_GROUP"),selTypeGrBox );
84   QRadioButton*   elemBtn = new QRadioButton( tr( "SMESH_ELEMENTS" ),selTypeGrBox );
85   QRadioButton*   nodeBtn = new QRadioButton( tr( "SMESH_NODES" ),selTypeGrBox );
86   //
87   QHBoxLayout* selTypeLay = new QHBoxLayout( selTypeGrBox );
88   selTypeLay->setMargin(MARGIN);
89   selTypeLay->setSpacing(SPACING);
90   selTypeLay->addWidget( objBtn  );
91   selTypeLay->addWidget( elemBtn );
92   selTypeLay->addWidget( nodeBtn );
93   objBtn->setChecked(true);
94   //
95   mySelTypeBtnGrp         = new QButtonGroup( mainFrame() );
96   mySelTypeBtnGrp->addButton( objBtn , SEL_OBJECT );
97   mySelTypeBtnGrp->addButton( elemBtn, SEL_ELEMENTS );
98   mySelTypeBtnGrp->addButton( nodeBtn, SEL_NODES );
99
100   // Label, Select Btn, LineEdit, Filter Btn
101
102   setObjectPixmap( "SMESH", tr( "ICON_SELECT" ) );
103   createObject( tr( "SMESH_NAME" ), mainFrame(), 0 );
104
105   myFilterBtn = new QPushButton( tr( "SMESH_BUT_FILTER" ), mainFrame() );
106
107   myDuplicateElemsChkBox = new QCheckBox( tr( "SMESH_DUPLICATE_0D" ), mainFrame() );
108
109   // List of groups
110
111   myGroupBox = new QGroupBox( tr( "SMESH_ADD_TO_GROUP" ), mainFrame() );
112   myGroupBox->setCheckable( true );
113
114   myGroupLabel = new QLabel( tr( "SMESH_GROUP" ), myGroupBox );
115   myGroupListCmBox = new QComboBox( myGroupBox );
116   myGroupListCmBox->setEditable( true );
117   myGroupListCmBox->setInsertPolicy( QComboBox::NoInsert );
118
119   QHBoxLayout* groupsLayout = new QHBoxLayout( myGroupBox );
120   groupsLayout->setSpacing(SPACING);
121   groupsLayout->setMargin(MARGIN);
122   groupsLayout->addWidget( myGroupLabel );
123   groupsLayout->addWidget( myGroupListCmBox, 1 );
124
125   // Main layout
126
127   QGridLayout* aLay = new QGridLayout( mainFrame() );
128   aLay->setMargin(MARGIN);
129   aLay->setSpacing(SPACING);
130   //
131   aLay->addWidget( selTypeGrBox,          0, 0, 1, 5  );
132   //
133   aLay->addWidget( objectWg( 0, Label  ), 1, 0 );
134   aLay->addWidget( objectWg( 0, Btn    ), 1, 1 );
135   aLay->addWidget( objectWg( 0, Control), 1, 2, 1, 2 );
136   aLay->addWidget( myFilterBtn,           1, 4 );
137   aLay->addWidget( myDuplicateElemsChkBox,2, 0 );
138   //
139   aLay->addWidget( myGroupBox,            3, 0, 1, 5 );
140
141   // Signals
142
143   connect( myGroupBox,      SIGNAL( toggled( bool )),     SLOT( onGroupChecked() ));
144   connect( mySelTypeBtnGrp, SIGNAL( buttonClicked(int) ), SLOT( onSelTypeChange(int)));
145
146   onSelTypeChange( SEL_OBJECT );
147 }
148
149 //================================================================================
150 /*!
151  * \brief SLOT to enable/disable groups
152  */
153 //================================================================================
154
155 void SMESHGUI_Add0DElemsOnAllNodesDlg::onGroupChecked( bool on )
156 {
157   myGroupLabel->setEnabled( on );
158   myGroupListCmBox->setEnabled( on );
159 }
160
161 //================================================================================
162 /*!
163  * \brief SLOT to enable/disable groups
164  */
165 //================================================================================
166
167 void SMESHGUI_Add0DElemsOnAllNodesDlg::onSelTypeChange( int selType )
168 {
169   setNameIndication( 0, selType == SEL_OBJECT ? OneName : ListOfNames );
170   setReadOnly      ( 0, selType == SEL_OBJECT );
171
172   QLabel* label = qobject_cast< QLabel* >( objectWg(0, Label ));
173   switch( selType ) {
174   case SEL_OBJECT:   label->setText( tr("SMESH_NAME")); break;
175   case SEL_ELEMENTS: label->setText( tr("ELEMENT_IDS")); break;
176   case SEL_NODES:    label->setText( tr("NODE_IDS")); break;
177   default:;
178   }
179   QLineEdit* lineEdit = qobject_cast< QLineEdit* >( objectWg(0, Control ));
180   lineEdit->setText("");
181   lineEdit->setValidator( selType == SEL_OBJECT ? 0 : & myIDValidator );
182
183   myFilterBtn->setEnabled( selType != SEL_OBJECT );
184
185   emit selTypeChanged( selType );
186 }
187
188 //================================================================================
189 /*!
190  * \brief Return type of selected object: [SEL_OBJECT, SEL_ELEMENTS, SEL_NODES]
191  */
192 //================================================================================
193
194 int SMESHGUI_Add0DElemsOnAllNodesDlg::getSelectionType() const
195 {
196   return mySelTypeBtnGrp->checkedId();
197 }
198
199 //================================================================================
200 /*!
201  * \brief Checks consistency of data
202  */
203 //================================================================================
204
205 bool SMESHGUI_Add0DElemsOnAllNodesDlg::isValid()
206 {
207   if( myGroupBox->isChecked() && myGroupListCmBox->currentText().isEmpty() ) {
208     SUIT_MessageBox::warning( this, tr( "SMESH_WRN_WARNING" ), tr( "GROUP_NAME_IS_EMPTY" ) );
209     return false;
210   }
211   return true;
212 }
213
214 //================================================================================
215 /*!
216  * \brief Operation Constructor
217  */
218 //================================================================================
219
220 SMESHGUI_Add0DElemsOnAllNodesOp::SMESHGUI_Add0DElemsOnAllNodesOp()
221   :SMESHGUI_SelectionOp(),
222    myDlg( new SMESHGUI_Add0DElemsOnAllNodesDlg ),
223    myFilterDlg( 0 )
224 {
225   myHelpFileName = "adding_nodes_and_elements_page.html#adding_0delems_on_all_nodes_anchor";
226
227   connect( myDlg,              SIGNAL( selTypeChanged(int) ), SLOT( onSelTypeChange(int)));
228   connect( myDlg->myFilterBtn, SIGNAL( clicked()),            SLOT( onSetFilter() ));
229   connect( myDlg->myGroupBox,  SIGNAL( clicked(bool)),        SLOT( updateButtons() ));
230 }
231
232 //================================================================================
233 /*!
234  * \brief Destructor
235  */
236 //================================================================================
237
238 SMESHGUI_Add0DElemsOnAllNodesOp::~SMESHGUI_Add0DElemsOnAllNodesOp()
239 {
240   if ( myFilterDlg ) {
241     myFilterDlg->setParent( 0 );
242     delete myFilterDlg;
243     myFilterDlg = 0;
244   }
245 }
246
247 //================================================================================
248 /*!
249  * \brief Start
250  */
251 //================================================================================
252
253 void SMESHGUI_Add0DElemsOnAllNodesOp::startOperation()
254 {
255   SMESHGUI_SelectionOp::startOperation();
256
257   myDlg->myGroupBox->setChecked( false );
258   myDlg->activateObject( 0 );
259   myDlg->show();
260
261   selectionDone();
262 }
263
264 //================================================================================
265 /*!
266  * \brief Treat changed selection
267  */
268 //================================================================================
269
270 void SMESHGUI_Add0DElemsOnAllNodesOp::selectionDone()
271 {
272   if (myFilterDlg && myFilterDlg->isVisible()) return; // filter dgl active
273   if (!myDlg->myGroupBox->isEnabled())         return; // inactive
274
275   myIO.Nullify();
276   updateButtons();
277
278   SALOME_ListIO aList;
279   selectionMgr()->selectedObjects( aList );
280   if ( aList.Extent() == 1 ) {
281     myIO = aList.First();
282   }
283   else {
284     myDlg->setObjectText( 0, ""); // it clears the selection
285     return;
286   }
287   QString ids;
288   switch ( myDlg->getSelectionType() ) {
289   case SEL_OBJECT:
290     SMESHGUI_SelectionOp::selectionDone();
291     break;
292   case SEL_ELEMENTS:
293     SMESH::GetNameOfSelectedElements( selector(), myIO, ids );
294     myDlg->setObjectText( 0, ids );
295     break;
296   case SEL_NODES:
297     SMESH::GetNameOfSelectedNodes( selector(), myIO, ids );
298     myDlg->setObjectText( 0, ids );
299     break;
300   default:;
301   }
302
303   // fill the list of existing groups
304   myDlg->myGroupListCmBox->clear();
305   myDlg->myGroupListCmBox->addItem( QString() );
306   if ( !myIO.IsNull() && myIO->hasEntry())
307   {
308     SMESH::SMESH_Mesh_var mesh = SMESH::GetMeshByIO( myIO );
309     _PTR(SObject)       meshSO = SMESH::ObjectToSObject( mesh );
310     _PTR(SObject) group0DRoot;
311     if ( meshSO && meshSO->FindSubObject( SMESH::Tag_0DElementsGroups, group0DRoot ))
312     {
313       _PTR(ChildIterator) group0DIter = SMESH::getStudy()->NewChildIterator( group0DRoot );
314       for ( ; group0DIter->More(); group0DIter->Next() )
315       {
316         _PTR(SObject) groupSO = group0DIter->Value();
317         std::string groupName = groupSO->GetName();
318         if ( !groupName.empty() )
319           myDlg->myGroupListCmBox->addItem( groupName.c_str() );
320       }
321     }
322     // enable buttons
323     updateButtons();
324   }
325 }
326
327 //=======================================================================
328 //function : updateButtons
329 //purpose  : enable [Apply]
330 //=======================================================================
331
332 void SMESHGUI_Add0DElemsOnAllNodesOp::updateButtons()
333 {
334   bool ok = false;
335
336   if (( !myIO.IsNull() && myIO->hasEntry() && !myDlg->objectText( 0 ).isEmpty() ) &&
337       ( !myDlg->myGroupBox->isChecked() || !myDlg->myGroupListCmBox->currentText().isEmpty() ))
338   {
339     SMESH::SMESH_Mesh_var mesh = SMESH::GetMeshByIO( myIO );
340     if ( !mesh->_is_nil() )
341     {
342       if ( myDlg->getSelectionType() == SEL_OBJECT )
343         ok = true;
344       else
345       {
346         QString        ids = myDlg->objectText( 0 );
347         QStringList idList = ids.split( " ", QString::SkipEmptyParts );
348         const bool  isElem = ( myDlg->getSelectionType() == SEL_ELEMENTS );
349         QStringList::iterator idIt = idList.begin();
350         for ( ; idIt != idList.end() && !ok; ++idIt )
351           ok = ( mesh->GetElementType( idIt->toLong(), isElem ) != SMESH::ALL );
352       }
353     }
354   }
355
356   myDlg->button( QtxDialog::Apply )->setEnabled( ok );
357   myDlg->button( QtxDialog::OK    )->setEnabled( ok );
358 }
359
360 //================================================================================
361 /*!
362  * \brief Return a filter of objects
363  */
364 //================================================================================
365
366 SUIT_SelectionFilter* SMESHGUI_Add0DElemsOnAllNodesOp::createFilter( const int ) const
367 {
368   if ( myDlg->getSelectionType() == SEL_OBJECT )
369   {
370     // Create a filter of objects: any IDSource except the group of 0D elements
371
372     QList<SUIT_SelectionFilter*> filters;
373     filters.push_back( new SMESH_TypeFilter( SMESH::GROUP_0D ));
374     SMESH_LogicalFilter* not0DGroup = new SMESH_LogicalFilter( filters,
375                                                                SMESH_LogicalFilter::LO_NOT,
376                                                                /*takeOwnership=*/true);
377     filters[0] = not0DGroup;
378     filters.push_back( new SMESH_TypeFilter( SMESH::IDSOURCE ));
379     return new SMESH_LogicalFilter( filters,
380                                     SMESH_LogicalFilter::LO_AND,
381                                     /*takeOwnership=*/true);
382   }
383   return 0;
384 }
385
386 //================================================================================
387 /*!
388  * \brief Makes its main job
389  */
390 //================================================================================
391
392 bool SMESHGUI_Add0DElemsOnAllNodesOp::onApply()
393 {
394   if ( !myDlg->isValid() )
395     return false;
396
397   // get a mesh
398   SMESH::IDSource_wrap meshObject;
399   SMESH::SMESH_Mesh_var      mesh;
400   if ( !myIO.IsNull() )
401   {
402     CORBA::Object_var obj = SMESH::IObjectToObject( myIO );
403     meshObject = SMESH::SMESH_IDSource::_narrow( obj );
404     if ( !meshObject->_is_nil() )
405     {
406       mesh = meshObject->GetMesh();
407       meshObject->Register();
408     }
409   }
410   if ( mesh->_is_nil() )
411   {
412     SUIT_MessageBox::warning( myDlg, tr( "SMESH_WRN_WARNING" ), tr( "SMESH_BAD_SELECTION" ) );
413     return false;
414   }
415
416   try {
417     SMESH::SMESH_MeshEditor_var editor = mesh->GetMeshEditor();
418
419     // make SMESH_IDSource holding IDs of selected elements
420     if ( myDlg->getSelectionType() != SEL_OBJECT )
421     {
422       QString elemIDs = myDlg->objectText( 0 );
423       QStringList idList = elemIDs.split( " ", QString::SkipEmptyParts );
424       if ( idList.count() == 0 )
425       {
426         SUIT_MessageBox::warning( myDlg, tr( "SMESH_WRN_WARNING" ), tr( "SMESH_BAD_SELECTION" ) );
427         return false;
428       }
429       SMESH::long_array_var idArray = new SMESH::long_array;
430       idArray->length( idList.count() );
431       QStringList::iterator idIt = idList.begin();
432       for ( int i = 0; idIt != idList.end(); ++idIt, ++i )
433         idArray[i] = idIt->toLong();
434       SMESH::ElementType elemType =
435         myDlg->getSelectionType() == SEL_NODES ? SMESH::NODE : SMESH::ALL;
436       meshObject = editor->MakeIDSource( idArray, elemType );
437     }
438
439     // Create 0D elements
440
441     int prevNb0D = mesh->Nb0DElements();
442
443     QString groupName = myDlg->myGroupListCmBox->currentText();
444     SMESH::SMESH_IDSource_var newObj =
445       editor->Create0DElementsOnAllNodes( meshObject, groupName.toUtf8().data(),
446                                           myDlg->myDuplicateElemsChkBox->isChecked() );
447
448     int newNb0D = mesh->Nb0DElements() - prevNb0D;
449     SUIT_MessageBox::information( myDlg, tr( "SMESH_INFORMATION" ),
450                                   tr( "NB_NEW_0D" ).arg( newNb0D ),
451                                   SUIT_MessageBox::Ok, SUIT_MessageBox::Ok);
452   }
453   catch ( const SALOME::SALOME_Exception& S_ex ) {
454     SalomeApp_Tools::QtCatchCorbaException( S_ex );
455     return false;
456   }
457   catch (...) {
458     return false;
459   }
460
461   // clear selection
462
463   if ( myDlg->getSelectionType() == SEL_OBJECT )
464   {
465     SALOME_ListIO aList;
466     selectionMgr()->setSelectedObjects( aList, /*append=*/false );
467   }
468   else
469   {
470     selector()->ClearIndex();
471   }
472   selectionDone();
473
474   SMESH::UpdateView();
475   SMESHGUI::Modified();
476
477   if ( myDlg->myGroupBox->isChecked() )
478     SMESHGUI::GetSMESHGUI()->updateObjBrowser();
479
480   return true;
481 }
482
483 //================================================================================
484 /*!
485  * \brief Sets selection mode
486  */
487 //================================================================================
488
489 void SMESHGUI_Add0DElemsOnAllNodesOp::onSelTypeChange(int selType)
490 {
491   switch ( selType ) {
492   case SEL_OBJECT:   setSelectionMode( ActorSelection ); break;
493   case SEL_ELEMENTS: setSelectionMode( CellSelection );  break;
494   case SEL_NODES:    setSelectionMode( NodeSelection );  break;
495   }
496
497   if ( selType != SEL_OBJECT )
498     connect( myDlg, SIGNAL( objectChanged( int, const QStringList& )),
499              this,  SLOT  ( onTextChanged( int, const QStringList& )));
500   else
501     disconnect( myDlg, SIGNAL( objectChanged( int, const QStringList& )),
502                 this,  SLOT  ( onTextChanged( int, const QStringList& )));
503
504   connect( myDlg->myGroupListCmBox, SIGNAL( editTextChanged(const QString & )),
505            this,                    SLOT(   updateButtons() ));
506
507   selectionDone();
508 }
509
510 //================================================================================
511 /*!
512  * \brief Install filters
513  */
514 //================================================================================
515
516 void SMESHGUI_Add0DElemsOnAllNodesOp::onSetFilter()
517 {
518   SMESH::SMESH_Mesh_var mesh = SMESH::GetMeshByIO( myIO );
519   if ( mesh->_is_nil()) {
520     SUIT_MessageBox::critical( myDlg, tr("SMESH_ERROR"), tr("NO_MESH_SELECTED"));
521     return;
522   }
523   QList<int> types;
524   if ( myDlg->getSelectionType() == SEL_NODES ) {
525     types.push_back( SMESH::NODE );
526   }
527   else if ( myDlg->getSelectionType() == SEL_ELEMENTS ) {
528     types.push_back( SMESH::EDGE );
529     types.push_back( SMESH::FACE );
530     types.push_back( SMESH::VOLUME );
531   }
532   else
533     return;
534   if ( !myFilterDlg )
535     myFilterDlg = new SMESHGUI_FilterDlg( getSMESHGUI(), SMESH::ALL );
536
537   myFilterDlg->Init( types );
538   myFilterDlg->SetSelection();
539   myFilterDlg->SetMesh( mesh );
540   myFilterDlg->SetSourceWg( myDlg->objectWg( 0, LightApp_Dialog::Control ));
541
542   myFilterDlg->show();
543 }
544
545 //=======================================================================
546 //function : onTextChanged
547 //purpose  : SLOT called when the user types IDs
548 //=======================================================================
549
550 void SMESHGUI_Add0DElemsOnAllNodesOp::onTextChanged( int obj, const QStringList& text )
551 {
552   SMESHGUI_SelectionOp::onTextChanged( obj, text );
553   updateButtons();
554 }