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