]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_GroupOnShapeDlg.cxx
Salome HOME
0020145: EDF 666 SMESH: Modifications of GEOM groups are not taken into account
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_GroupOnShapeDlg.cxx
1 //  Copyright (C) 2007-2008  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 //  SMESH SMESHGUI : GUI for SMESH component
23 // File      : SMESHGUI_GroupOnShapeDlg.cxx
24 // Created   : Wed Sep 17 18:36:51 2008
25 // Author    : Edward AGAPOV (eap)
26 // Module    : SMESH
27 // IDL Headers
28 //
29 #include "SMESHGUI_GroupOnShapeDlg.h"
30
31 #include "SMESH_TypeFilter.hxx"
32 #include "SMESHGUI_Utils.h"
33 #include "SMESHGUI_GEOMGenUtils.h"
34
35 #include <GeometryGUI.h>
36 #include <GEOM_SelectionFilter.h>
37
38 #include <SUIT_Session.h>
39 #include <SUIT_OverrideCursor.h>
40 #include <LightApp_UpdateFlags.h>
41 #include <SUIT_ResourceMgr.h>
42
43 #include <SALOMEconfig.h>
44 #include CORBA_SERVER_HEADER(SMESH_Mesh)
45 #include CORBA_SERVER_HEADER(SMESH_Group)
46
47 #include <QGroupBox>
48 #include <QLayout>
49 #include <QFrame>
50 #include <QLabel>
51 #include <QPushButton>
52 #include <QListWidget>
53 #include <QLineEdit>
54
55 #define SPACING 6
56 #define MARGIN  11
57
58 enum { _MESH, _ELEM_GEOM, _NODE_GEOM };
59
60 SMESHGUI_GroupOnShapeDlg::SMESHGUI_GroupOnShapeDlg()
61   : SMESHGUI_Dialog( 0, false, true )
62 {
63   QPixmap image (resourceMgr()->loadPixmap("SMESH", tr("ICON_SELECT")));
64
65   // Name
66   //QLabel* nameLabel = new QLabel( tr( "SMESH_NAME" ), mainFrame() );
67   //myGrpNameLine = new QLineEdit( mainFrame() );
68
69   // Mesh
70   QLabel* meshLabel = new QLabel( tr( "SMESH_OBJECT_MESH" ), mainFrame() );
71   myMeshBtn = new QPushButton( mainFrame() );
72   myMeshBtn->setCheckable(true);
73   myMeshBtn->setIcon(image);
74   myMeshLine = new QLineEdit( mainFrame() );
75   myMeshLine->setReadOnly(true);
76
77   // Elem geom
78
79   QGroupBox* elemsGrp = new QGroupBox( tr( "SMESH_ELEMENTS" ), mainFrame() );
80   QLabel* label = new QLabel( tr( "SMESH_GEOM" ), elemsGrp );
81   myElemGeomBtn = new QPushButton( elemsGrp );
82   myElemGeomBtn->setCheckable(true);
83   myElemGeomBtn->setIcon(image);
84   myElemGeomList = new QListWidget( elemsGrp );
85   myElemGeomList->setSelectionMode(QListWidget::NoSelection);
86
87   QGridLayout* elemLay = new QGridLayout(elemsGrp);
88   elemLay->setSpacing( SPACING );
89   elemLay->setMargin( MARGIN );
90   elemLay->setRowStretch( 1, 1 );
91   elemLay->addWidget( label, 0, 0 );
92   elemLay->addWidget( myElemGeomBtn, 0, 1 );
93   elemLay->addWidget( myElemGeomList, 0, 2, 2, 1);
94
95   // Node geom
96
97   QGroupBox* nodesGrp = new QGroupBox( tr( "SMESH_NODES" ), mainFrame() );
98
99   label = new QLabel( tr( "SMESH_GEOM" ), nodesGrp );
100   myNodeGeomBtn = new QPushButton( nodesGrp );
101   myNodeGeomBtn->setCheckable(true);
102   myNodeGeomBtn->setIcon(image);
103   myNodeGeomList = new QListWidget( nodesGrp );
104   myNodeGeomList->setSelectionMode(QListWidget::NoSelection);
105
106   QGridLayout* nodeLay = new QGridLayout(nodesGrp);
107   nodeLay->setSpacing( SPACING );
108   nodeLay->setMargin( MARGIN );
109   nodeLay->setRowStretch( 1, 1 );
110   nodeLay->addWidget( label, 0, 0 );
111   nodeLay->addWidget( myNodeGeomBtn, 0, 1 );
112   nodeLay->addWidget(myNodeGeomList, 0, 2, 2, 1);
113
114   // Fill layout
115   QGridLayout* aLay = new QGridLayout( mainFrame());
116   aLay->setSpacing( SPACING );
117   aLay->setMargin( MARGIN );
118   //aLay->addWidget( nameLabel, 0, 0 );
119   //aLay->addWidget( myGrpNameLine, 0, 2 );
120   aLay->addWidget( meshLabel, 1, 0 );
121   aLay->addWidget( myMeshBtn, 1, 1 );
122   aLay->addWidget( myMeshLine,1, 2 );
123   aLay->addWidget( elemsGrp, 2, 1, 1, 3 );
124   aLay->addWidget( nodesGrp, 3, 1, 1, 3 );
125
126   setWindowTitle( tr( "SMESH_CREATE_GROUP_FROM_GEOM" ) );
127 }
128
129 SMESHGUI_GroupOnShapeDlg::~SMESHGUI_GroupOnShapeDlg()
130 {
131 }
132
133 //================================================================================
134 /*!
135  * \brief slot to enable/diable [Apply]
136  */
137 //================================================================================
138
139 void SMESHGUI_GroupOnShapeDlg::updateButtons()
140 {
141   bool enable =
142     /*!myGrpNameLine->text().isEmpty() &&*/ myElemGeomList->count() + myNodeGeomList->count();
143
144   button(Apply)->setEnabled( enable );
145   button(OK)->setEnabled( enable );
146 }
147
148 //================================================================================
149 /*!
150  * \brief initialization
151  */
152 //================================================================================
153
154 void SMESHGUI_GroupOnShapeDlg::init()
155 {
156   //myGrpNameLine->setText("");
157
158   myMeshBtn->setChecked( true );
159   myMeshLine->setText("");
160
161   myElemGeomBtn->setChecked(false);
162   myElemGeomBtn->setEnabled(false);
163   myElemGeomList->clear();
164   myNodeGeomBtn->setChecked(false);
165   myNodeGeomBtn->setEnabled(false);
166   myNodeGeomList->clear();
167
168   updateButtons();
169 }
170
171 //================================================================================
172 /*!
173  * \brief operation constructor
174  */
175 //================================================================================
176
177 SMESHGUI_GroupOnShapeOp::SMESHGUI_GroupOnShapeOp()
178   : SMESHGUI_SelectionOp(ActorSelection),
179     myDlg( 0 )
180 {
181   myHelpFileName = "creating_groups_page.html";
182 }
183
184 SMESHGUI_GroupOnShapeOp::~SMESHGUI_GroupOnShapeOp()
185 {
186   if ( myDlg )
187     delete myDlg;
188 }
189 //================================================================================
190 /*!
191  * \brief Gets dialog of this operation
192   * \retval LightApp_Dialog* - pointer to dialog of this operation
193 */
194 //================================================================================
195
196 LightApp_Dialog* SMESHGUI_GroupOnShapeOp::dlg() const
197 {
198   return myDlg;
199 }
200
201 //================================================================================
202 /*!
203  * \brief return type of mesh group by geom object
204  */
205 //================================================================================
206
207 static SMESH::ElementType elementType(GEOM::GEOM_Object_var& geom)
208 {
209   if ( !geom->_is_nil() ) {
210     switch ( geom->GetShapeType() ) {
211     case GEOM::VERTEX:   return SMESH::NODE;
212     case GEOM::EDGE:     return SMESH::EDGE;
213     case GEOM::WIRE:     return SMESH::EDGE;
214     case GEOM::FACE:     return SMESH::FACE;
215     case GEOM::SHELL:    return SMESH::FACE;
216     case GEOM::SOLID:    return SMESH::VOLUME;
217     case GEOM::COMPSOLID:return SMESH::VOLUME;
218     case GEOM::COMPOUND: break;
219     default:             return SMESH::ALL;
220     }
221     _PTR(Study) aStudy = SMESH::GetActiveStudyDocument();
222     GEOM::GEOM_IShapesOperations_var aShapeOp =
223       SMESH::GetGEOMGen()->GetIShapesOperations(aStudy->StudyId());
224
225     if ( geom->GetType() == 37 ) { // geom group
226       GEOM::GEOM_IGroupOperations_var  aGroupOp =
227         SMESH::GetGEOMGen()->GetIGroupOperations(aStudy->StudyId());
228       if ( !aGroupOp->_is_nil() ) {
229         GEOM::GEOM_Object_var mainShape = aGroupOp->GetMainShape( geom );
230         GEOM::ListOfLong_var        ids = aGroupOp->GetObjects( geom );
231         if ( ids->length() && !mainShape->_is_nil() && !aShapeOp->_is_nil() ) {
232           GEOM::GEOM_Object_var member = aShapeOp->GetSubShape( mainShape, ids[0] );
233           return elementType( member );
234         }
235       }
236     }
237     else if ( !aShapeOp->_is_nil() ) { // just a compoud shape
238       GEOM::ListOfLong_var ids = aShapeOp->SubShapeAllIDs( geom, GEOM::SHAPE, false );
239       if ( ids->length() ) {
240         GEOM::GEOM_Object_var member = aShapeOp->GetSubShape( geom, ids[0] );
241         return elementType( member );
242       }
243     }
244   }
245   return SMESH::ALL;
246 }
247
248 //================================================================================
249 /*!
250  * \brief initialization
251  */
252 //================================================================================
253
254 void SMESHGUI_GroupOnShapeOp::init()
255 {
256   myMeshID="";
257   myElemGeoIDs.clear();
258   myNodeGeoIDs.clear();
259
260   myDlg->init();
261   removeCustomFilters();
262   onActivateObject( _MESH ); // install filter
263 }
264
265 //================================================================================
266 /*!
267  * \brief start operation
268  */
269 //================================================================================
270
271 void SMESHGUI_GroupOnShapeOp::startOperation()
272 {
273   if (!myDlg)
274   {
275     myDlg = new SMESHGUI_GroupOnShapeDlg();
276     connect(myDlg->myMeshBtn,     SIGNAL(clicked()), this, SLOT(onButtonClick()));
277     connect(myDlg->myElemGeomBtn, SIGNAL(clicked()), this, SLOT(onButtonClick()));
278     connect(myDlg->myNodeGeomBtn, SIGNAL(clicked()), this, SLOT(onButtonClick()));
279     //connect(myDlg->myGrpNameLine, SIGNAL(textChanged(const QString&)),myDlg,SLOT(updateButtons()));
280   }
281   SMESHGUI_SelectionOp::startOperation();
282
283   init();
284   myDlg->show();
285 }
286
287 //================================================================================
288 /*!
289  * \brief create groups
290  */
291 //================================================================================
292
293 bool SMESHGUI_GroupOnShapeOp::onApply()
294 {
295   SUIT_OverrideCursor aWaitCursor;
296
297   if (isStudyLocked())
298     return false;
299
300   // study
301   _PTR(Study) aStudy = SMESH::GetActiveStudyDocument();
302   if ( !aStudy ) return false;
303
304   // mesh
305   _PTR(SObject)       meshSO = aStudy->FindObjectID( myMeshID.toLatin1().data() );
306   SMESH::SMESH_Mesh_var mesh = SMESH::SObjectToInterface<SMESH::SMESH_Mesh>( meshSO );
307   if ( mesh->_is_nil() ) return false;
308
309   // names of all existing groups
310 //   QStringList groupNames;
311 //   QString givenName = myDlg->myGrpNameLine->text();
312 //   if ( !givenName.isEmpty() ) {
313 //     SMESH::ListOfGroups_var groups = mesh->GetGroups();
314 //     for ( int i = 0; i < groups->length(); ++i ) {
315 //       CORBA::String_var name = groups[i]->GetName();
316 //       groupNames.append( name.in() );
317 //     }
318 //   }
319
320   // create groups
321   SMESH::SMESH_GroupOnGeom_var group;
322   for ( int isNode = 0; isNode < 2; ++isNode ) // elems and then nodes
323   {
324     QStringList::iterator geomID = isNode ? myNodeGeoIDs.begin() : myElemGeoIDs.begin();
325     QStringList::iterator geomEnd = isNode ? myNodeGeoIDs.end() : myElemGeoIDs.end();
326
327     for ( int i = 0; geomID != geomEnd; ++geomID, ++i )
328     {
329       // selected geom
330       _PTR(SObject) geomSO = aStudy->FindObjectID( geomID->toLatin1().data() );
331       GEOM::GEOM_Object_var geom = SMESH::SObjectToInterface<GEOM::GEOM_Object>(geomSO);
332       if ( geom->_is_nil() ) continue;
333
334       // group type
335       SMESH::ElementType elemType = isNode ? SMESH::NODE : elementType( geom );
336       if ( elemType == SMESH::ALL )
337         continue;
338
339       // make a unique name
340       QString name =
341         isNode ? myDlg->myNodeGeomList->item(i)->text() : myDlg->myElemGeomList->item(i)->text();
342 //       int nb = 1;
343 //       QString name = myDlg->myGrpNameLine->text() + "_" + QString::number(nb);
344 //       while ( groupNames.contains( name ))
345 //         name = myDlg->myGrpNameLine->text() + "_" + QString::number(++nb);
346 //       groupNames.append( name );
347
348       //printf( "apply() %s %s\n", (*geomID).latin1(), name.latin1() );
349       group = mesh->CreateGroupFromGEOM( elemType, name.toLatin1().data(), geom );
350     }
351   }
352   update( UF_ObjBrowser | UF_Model );
353
354   init();
355
356   return !group->_is_nil();
357 }
358
359 //================================================================================
360 /*!
361  * \brief slot connected to selection buttons
362  */
363 //================================================================================
364
365 void SMESHGUI_GroupOnShapeOp::onButtonClick()
366 {
367   removeCustomFilters();
368
369   if ( sender() == myDlg->myMeshBtn && myDlg->myMeshBtn->isChecked() )
370   {
371     myDlg->myElemGeomBtn->setChecked(false);
372     myDlg->myNodeGeomBtn->setChecked(false);
373     onActivateObject( _MESH ); // install filter
374   }
375   else if ( sender() == myDlg->myElemGeomBtn && myDlg->myElemGeomBtn->isChecked() )
376   {
377     myDlg->myMeshBtn->setChecked(false);
378     myDlg->myNodeGeomBtn->setChecked(false);
379     onActivateObject( _ELEM_GEOM ); // install filter
380   }
381   else if ( sender() == myDlg->myNodeGeomBtn && myDlg->myNodeGeomBtn->isChecked() )
382   {
383     myDlg->myMeshBtn->setChecked(false);
384     myDlg->myElemGeomBtn->setChecked(false);
385     onActivateObject( _NODE_GEOM ); // install filter
386   }
387   //selectionDone();
388 }
389
390 //================================================================================
391 /*!
392  * \brief Creates selection filter
393   * \param theId - identifier of current selection widget
394   * \retval SUIT_SelectionFilter* - pointer to the created filter or null
395  *
396  * Creates selection filter in accordance with identifier of current selection widget
397  */
398 //================================================================================
399 SUIT_SelectionFilter* SMESHGUI_GroupOnShapeOp::createFilter( const int theId ) const
400 {
401   if ( theId == _ELEM_GEOM || theId == _NODE_GEOM )
402     return new GEOM_SelectionFilter( (SalomeApp_Study*)study(), true );
403   else if ( theId == _MESH )
404     return new SMESH_TypeFilter( MESH );
405   else
406     return ( SUIT_SelectionFilter*) 0;
407 }
408 //================================================================================
409 /*!
410  * \brief Updates dialog's look and feel
411  *
412  * Virtual method redefined from the base class updates dialog's look and feel
413  */
414 //================================================================================
415 void SMESHGUI_GroupOnShapeOp::selectionDone()
416 {
417   QStringList names, ids;
418   LightApp_Dialog::TypesList types;
419   selected( names, types, ids );
420   int nbSelected = names.size();
421
422   if ( myDlg->myMeshBtn->isChecked() ) // mesh selected
423   {
424     myDlg->myMeshLine->setText("");
425     myMeshID = "";
426     if ( nbSelected == 1 ) {
427       myDlg->myMeshLine->setText( names.front() );
428       myMeshID = ids.front();
429     }
430     myDlg->myElemGeomList->clear();
431     myDlg->myElemGeomBtn->setEnabled( nbSelected == 1 );
432     myDlg->myNodeGeomList->clear();
433     myDlg->myNodeGeomBtn->setEnabled( nbSelected == 1 );
434     myDlg->myElemGeomBtn->click();
435     return;
436   }
437
438   // Filter off inappropriate shapes
439
440   QStringList goodNames, goodIds;
441   if (nbSelected > 0) {
442     // study
443     if (_PTR(Study) aStudy = SMESH::GetActiveStudyDocument()) {
444       // mesh
445       if (_PTR(SObject)  meshSO = aStudy->FindObjectID( myMeshID.toLatin1().data() )) {
446         // shape to mesh
447         _PTR(SObject) anObj, shapeToMesh;
448         if (meshSO->FindSubObject(1, anObj) && anObj->ReferencedObject(shapeToMesh)) {
449           // loop on selected
450           QStringList::iterator name = names.begin(), id = ids.begin(), idEnd = ids.end();
451           for (; id != idEnd; ++id, ++name ) {
452             // shape SO
453             if (_PTR(SObject) shapeSO = aStudy->FindObjectID( id->toLatin1().data() )) {
454             // check if shape SO is a child of shape to mesh 
455               while ( shapeSO && shapeSO->GetID() != shapeToMesh->GetID() )
456                 if  ( shapeSO->Depth() < 2 )
457                   shapeSO.reset();
458                 else
459                   shapeSO = shapeSO->GetFather();
460               if ( shapeSO ) {
461                 //printf( "selectionDone() %s %s\n", (*id).latin1(), (*name).latin1() );
462                 if ( !goodIds.contains( *id )) {
463                   goodIds.append( *id );
464                   goodNames.append( *name );
465                 }
466               }
467             }
468           }
469         }
470       }
471     }
472   }
473
474   if ( myDlg->myElemGeomBtn->isChecked() ) // elem geomerty selection
475   {
476     myDlg->myElemGeomList->clear();
477     myDlg->myElemGeomList->addItems( goodNames );
478     myElemGeoIDs = goodIds;
479   }
480   else if ( myDlg->myNodeGeomBtn->isChecked() ) // Node geomerty selection
481   {
482     myDlg->myNodeGeomList->clear();
483     myDlg->myNodeGeomList->addItems( goodNames );
484     myNodeGeoIDs = goodIds;
485   }
486
487   // enable/diable Apply, which can change at selection
488   myDlg->updateButtons();
489 }