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