Salome HOME
Updated for PAL14857.
[modules/geom.git] / src / GroupGUI / GroupGUI_GroupDlg.cxx
1 //  GEOM GEOMGUI : GUI for Geometry component
2 //
3 //  Copyright (C) 2004  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 //
24 //  File   : GroupGUI_GroupDlg.cxx
25 //  Author : Sergey ANIKIN
26 //  Module : GEOM
27 //  $Header$
28
29 #include "GroupGUI_GroupDlg.h"
30
31 #include "SUIT_Desktop.h"
32 #include "SUIT_Session.h"
33 #include "SalomeApp_Application.h"
34 #include "SalomeApp_Study.h"
35 #include "LightApp_SelectionMgr.h"
36
37 #include "GEOMBase.h"
38
39 #include "GEOMImpl_Types.hxx"
40
41 #include <qlabel.h>
42 #include <qlistbox.h>
43 #include <qlineedit.h>
44 #include <qmap.h>
45
46 #include <TColStd_IndexedMapOfInteger.hxx>
47 #include <TColStd_MapOfInteger.hxx>
48
49
50 GroupGUI_GroupDlg::GroupGUI_GroupDlg(Mode mode, GeometryGUI* theGeometryGUI, QWidget* parent)
51   :GEOMBase_Skeleton( theGeometryGUI, parent, "GroupGUI_GroupDlg", false,
52                       WStyle_Customize | WStyle_NormalBorder | WStyle_Title | WStyle_SysMenu),
53    myMode( mode ),
54    myBusy( false )
55 {
56   SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
57
58   QPixmap image0     (resMgr->loadPixmap("GEOM", tr("ICON_OBJBROWSER_VERTEX")));
59   QPixmap image1     (resMgr->loadPixmap("GEOM", tr("ICON_OBJBROWSER_EDGE")));
60   QPixmap image2     (resMgr->loadPixmap("GEOM", tr("ICON_OBJBROWSER_FACE")));
61   QPixmap image3     (resMgr->loadPixmap("GEOM", tr("ICON_OBJBROWSER_SOLID")));
62   QPixmap iconSelect (resMgr->loadPixmap("GEOM", tr("ICON_SELECT")));
63
64   setCaption( myMode == CreateGroup ? tr( "CREATE_GROUP_TITLE" ) : tr( "EDIT_GROUP_TITLE" ) );
65
66   // Shape type button group
67   GroupConstructors->setEnabled( myMode == CreateGroup );
68   GroupConstructors->setTitle( tr( "SHAPE_TYPE" ) );
69   RadioButton1->setPixmap( image0 );
70   RadioButton2->setPixmap( image1 );
71   RadioButton3->setPixmap( image2 );
72   RadioButton4->setPixmap( image3 );
73   RadioButton4->show();
74
75   // Group name
76   GroupBoxName->setTitle( tr( "GROUP_NAME" ) );
77
78   // Main shape and sub-shapes
79   GroupMedium = new QGroupBox( 1, Qt::Vertical, tr( "MAIN_SUB_SHAPES" ), this );
80   GroupMedium->setInsideMargin( 10 );
81   Layout1->addWidget( GroupMedium, 2, 0 );
82
83   QWidget* aFrame = new QWidget( GroupMedium );
84   QGridLayout* aMedLayout = new QGridLayout( aFrame, 4, 4, 0, 6 );
85
86   QLabel* aMainLabel = new QLabel( tr( "MAIN_SHAPE" ), aFrame );
87
88   mySelBtn = new QPushButton( aFrame );
89   mySelBtn->setPixmap( iconSelect );
90   mySelBtn->setEnabled( myMode == CreateGroup );
91
92   myMainName = new QLineEdit( aFrame );
93   myMainName->setReadOnly( true );
94   myMainName->setEnabled( myMode == CreateGroup );
95
96   mySelSubBtn = new QPushButton( tr( "SELECT_SUB_SHAPES" ), aFrame );
97   mySelAllBtn = new QPushButton( tr( "SELECT_ALL" ), aFrame );
98   myAddBtn    = new QPushButton( tr( "ADD" ), aFrame );
99   myRemBtn    = new QPushButton( tr( "REMOVE" ), aFrame );
100   myIdList    = new QListBox( aFrame );
101
102   myIdList->setSelectionMode( QListBox::Extended );
103   myIdList->setRowMode( QListBox::FitToWidth );
104
105   aMedLayout->addWidget( aMainLabel, 0, 0 );
106   aMedLayout->addWidget( mySelBtn, 0, 1 );
107   aMedLayout->addMultiCellWidget( myMainName, 0, 0, 2, 3 );
108   aMedLayout->addMultiCellWidget( mySelSubBtn, 1, 1, 0, 2 );
109   aMedLayout->addWidget( mySelAllBtn, 1, 3 );
110   aMedLayout->addMultiCellWidget( myIdList, 2, 3, 0, 2 );
111   aMedLayout->addWidget( myAddBtn, 2, 3 );
112   aMedLayout->addWidget( myRemBtn, 3, 3 );
113
114   setHelpFileName("working_with_groups.htm");
115
116   Init();
117 }
118
119 GroupGUI_GroupDlg::~GroupGUI_GroupDlg()
120 {
121 }
122
123
124 //=================================================================================
125 // function : Init()
126 // purpose  :
127 //=================================================================================
128 void GroupGUI_GroupDlg::Init()
129 {
130   // san -- TODO: clear selected sub-shapes...
131
132   if ( myMode == CreateGroup ) {
133     initName( tr( "GROUP_PREFIX" ) );
134
135     // Get ready for main shape selection
136     myEditCurrentArgument = myMainName;
137
138     connect( GroupConstructors, SIGNAL( clicked( int ) ), this, SLOT( ConstructorsClicked( int ) ) );
139     connect( mySelBtn,          SIGNAL( clicked() ),      this, SLOT( SetEditCurrentArgument() ) );
140   }
141   else if ( myMode == EditGroup && IObjectCount() ) {
142     Standard_Boolean aResult = Standard_False;
143     GEOM::GEOM_Object_var anObj =
144       GEOMBase::ConvertIOinGEOMObject( firstIObject(), aResult );
145
146     if ( aResult && !CORBA::is_nil( anObj ) && anObj->GetType() == GEOM_GROUP ) {
147       myGroup = anObj;
148
149       ResultName->setText( GEOMBase::GetName( myGroup ) );
150
151       GEOM::GEOM_IGroupOperations_var anOp = GEOM::GEOM_IGroupOperations::_narrow( getOperation() );
152       myMainObj = anOp->GetMainShape( myGroup );
153       if ( !CORBA::is_nil( myMainObj ) )
154         myMainName->setText( GEOMBase::GetName( myMainObj ) );
155
156       setShapeType( (TopAbs_ShapeEnum)anOp->GetType( myGroup ) );
157
158       GEOM::ListOfLong_var aCurrList = anOp->GetObjects( myGroup );
159       QListBoxItem* anItem;
160       for ( int i = 0, n = aCurrList->length(); i < n; i++ ) {
161         anItem = new QListBoxText( QString( "%1" ).arg(aCurrList[i] ) );
162         myIdList->insertItem( anItem );
163       }
164
165       myEditCurrentArgument = 0;
166     }
167   }
168
169   LightApp_SelectionMgr* aSelMgr =
170     ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->selectionMgr();
171
172   connect( aSelMgr,     SIGNAL( currentSelectionChanged() ), this, SLOT( SelectionIntoArgument() ) );
173
174   connect( buttonOk   , SIGNAL( clicked() ), this, SLOT( ClickOnOk()    ) );
175   connect( buttonApply, SIGNAL( clicked() ), this, SLOT( ClickOnApply() ) );
176
177   connect( mySelSubBtn, SIGNAL( clicked() ), this, SLOT( SetEditCurrentArgument() ) );
178   connect( mySelAllBtn, SIGNAL( clicked() ), this, SLOT( SetEditCurrentArgument() ) );
179   connect( myAddBtn,    SIGNAL( clicked() ), this, SLOT( add() ) );
180   connect( myRemBtn,    SIGNAL( clicked() ), this, SLOT( remove() ) );
181   connect( myIdList,    SIGNAL( selectionChanged() ), this, SLOT( selectionChanged() ) );
182
183   activateSelection();
184 }
185
186 //=================================================================================
187 // function : enterEvent()
188 // purpose  :
189 //=================================================================================
190 void GroupGUI_GroupDlg::enterEvent( QEvent* e )
191 {
192   if ( !buttonCancel->isEnabled() )
193     this->ActivateThisDialog();
194 }
195
196 //=================================================================================
197 // function : ClickOnOk()
198 // purpose  :
199 //=================================================================================
200 void GroupGUI_GroupDlg::ClickOnOk()
201 {
202   if ( ClickOnApply() )
203     ClickOnCancel();
204 }
205
206 //=================================================================================
207 // function : ClickOnApply()
208 // purpose  :
209 //=================================================================================
210 bool GroupGUI_GroupDlg::ClickOnApply()
211 {
212   if ( !onAccept( myMode == CreateGroup, true ) )
213     return false;
214
215   if ( myMode == CreateGroup )
216     {
217       initName();
218       myIdList->clear();
219     }
220   else
221     activateSelection();
222   return true;
223 }
224
225
226 //=================================================================================
227 // function : ActivateThisDialog()
228 // purpose  :
229 //=================================================================================
230 void GroupGUI_GroupDlg::ActivateThisDialog()
231 {
232   GEOMBase_Skeleton::ActivateThisDialog();
233
234   connect( ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->selectionMgr(),
235            SIGNAL( currentSelectionChanged() ), this, SLOT( SelectionIntoArgument() ) );
236
237   activateSelection();
238 }
239
240 //=================================================================================
241 // function : LineEditReturnPressed()
242 // purpose  :
243 //=================================================================================
244 void GroupGUI_GroupDlg::LineEditReturnPressed()
245 {
246   QLineEdit* send = ( QLineEdit* )sender();
247
248   if ( send == myMainName && !myEditCurrentArgument ) {
249     myEditCurrentArgument = myMainName;
250     activateSelection();
251   }
252   else
253     GEOMBase_Skeleton::LineEditReturnPressed();
254
255   updateState();
256 }
257
258
259 //=================================================================================
260 // function : SetEditCurrentArgument()
261 // purpose  :
262 //=================================================================================
263 void GroupGUI_GroupDlg::SetEditCurrentArgument()
264 {
265   QPushButton* send = (QPushButton*)sender();
266
267   if ( send == mySelBtn )
268     myEditCurrentArgument = myMainName;
269   else if ( send == mySelSubBtn || send == mySelAllBtn )
270     myEditCurrentArgument = 0;
271
272   activateSelection();
273
274   if ( send == mySelAllBtn )
275     selectAllSubShapes();
276   else
277     updateState();
278 }
279
280
281 //=================================================================================
282 // function : SelectionIntoArgument()
283 // purpose  : Called when selection has changed
284 //=================================================================================
285 void GroupGUI_GroupDlg::SelectionIntoArgument()
286 {
287   if ( myEditCurrentArgument ) {  // Selection of a main shape is active
288     myEditCurrentArgument->setText( "" );
289     myIdList->clear();
290
291     if ( IObjectCount() == 1 ) {
292       Standard_Boolean aResult = Standard_False;
293       GEOM::GEOM_Object_var anObj =
294         GEOMBase::ConvertIOinGEOMObject( firstIObject(), aResult );
295
296       if ( aResult && !anObj->_is_nil() && GEOMBase::IsShape( anObj ) ) {
297         myMainObj = anObj;
298         myEditCurrentArgument->setText( GEOMBase::GetName( anObj ) );
299         updateState();
300         return;
301       }
302     }
303
304     myMainObj = GEOM::GEOM_Object::_nil();
305   }
306   else { // an attempt to synchronize list box selection with 3d viewer
307     if ( myBusy )
308       return;
309
310     bool isBlocked = myIdList->signalsBlocked();
311     myIdList->blockSignals( true );
312     myIdList->clearSelection();
313
314     TColStd_IndexedMapOfInteger aMapIndex;
315
316     if ( IObjectCount() == 1 ) {
317       Standard_Boolean aResult = Standard_False;
318       GEOM::GEOM_Object_var anObj =
319         GEOMBase::ConvertIOinGEOMObject( firstIObject(), aResult );
320
321       if ( aResult && !anObj->_is_nil() )
322         ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->
323           selectionMgr()->GetIndexes( firstIObject(), aMapIndex );
324     }
325
326     // try to find out and process the object browser selection
327     if ( !aMapIndex.Extent() ) {
328       globalSelection( GEOM_ALLSHAPES );
329       
330       GEOM::ListOfGO anObjects;
331       GEOMBase::ConvertListOfIOInListOfGO(selectedIO(), anObjects);
332
333       GEOM::GEOM_ILocalOperations_var aLocOp = getGeomEngine()->GetILocalOperations( getStudyId() );
334       GEOM::GEOM_IShapesOperations_var aShapesOp = getGeomEngine()->GetIShapesOperations( getStudyId() );
335
336       for (int i = 0; i < anObjects.length(); i++) 
337         {
338           GEOM::GEOM_Object_var aGeomObj = anObjects[i];
339           GEOM::ListOfGO_var aSubObjects = new GEOM::ListOfGO();
340           TopoDS_Shape aShape;
341           if ( GEOMBase::GetShape(aGeomObj, aShape, getShapeType()) ) 
342             {
343               aSubObjects->length(1);
344               aSubObjects[0] = aGeomObj;
345             }
346           else if (aGeomObj->GetType() == GEOM_GROUP)
347             aSubObjects = aShapesOp->MakeExplode( aGeomObj, getShapeType(), false);
348           else
349             continue;
350
351           for (int i = 0; i < aSubObjects->length(); i++) 
352             {
353               TopoDS_Shape aShape;
354               if ( GEOMBase::GetShape(aSubObjects[i], aShape, getShapeType()) ) 
355                 {
356                   CORBA::Long anIndex = aLocOp->GetSubShapeIndex( myMainObj, aSubObjects[i] );
357                   if ( anIndex >= 0 )
358                     aMapIndex.Add( anIndex );
359                 }
360             }
361         }
362       
363       if ( !myMainObj->_is_nil() )
364         localSelection( myMainObj, getShapeType() );
365     }
366
367     if (aMapIndex.Extent() >= 1) {
368       QMap<int, int> aMap;
369       for ( int i = 0, n = myIdList->count(); i < n; i++ )
370         aMap.insert( myIdList->item( i )->text().toInt(), i );
371
372       for ( int ii = 1, nn = aMapIndex.Extent(); ii <= nn; ii++ ) {
373         if ( aMap.contains( aMapIndex( ii ) ) )
374           myIdList->setSelected( aMap[aMapIndex( ii )], true );
375       }
376     }
377     myIdList->blockSignals( isBlocked );
378   }
379
380   updateState();
381 }
382
383 //=================================================================================
384 // function : ConstructorsClicked()
385 // purpose  : Radio button management
386 //=================================================================================
387 void GroupGUI_GroupDlg::ConstructorsClicked( int constructorId )
388 {
389   myIdList->clear();
390   activateSelection();
391   updateState();
392 }
393
394 //=================================================================================
395 // function : selectAllSubShapes
396 // purpose  : 
397 //=================================================================================
398 void GroupGUI_GroupDlg::selectAllSubShapes()
399 {
400   if ( CORBA::is_nil( myMainObj ) )
401     return;
402
403   GEOM::GEOM_IShapesOperations_var aShOp = getGeomEngine()->GetIShapesOperations( getStudyId() );
404   GEOM::GEOM_ILocalOperations_var aLocOp = getGeomEngine()->GetILocalOperations( getStudyId() );
405
406   GEOM::ListOfGO_var aSubShapes = aShOp->MakeExplode( myMainObj, getShapeType(), false );
407   if ( !aShOp->IsDone() )
408     return;
409
410   bool isBlocked = myIdList->signalsBlocked();
411   myIdList->blockSignals( true );
412   myIdList->clear();
413
414   QListBoxItem* anItem;
415   for ( int i = 0, n = aSubShapes->length(); i < n; i++ ) {
416     CORBA::Long anIndex = aLocOp->GetSubShapeIndex( myMainObj, aSubShapes[i] );
417     if ( anIndex < 0 )
418       continue;
419
420     anItem = new QListBoxText( QString( "%1" ).arg( anIndex ) );
421     myIdList->insertItem( anItem );
422     myIdList->setSelected( anItem, true );
423   }
424
425   myIdList->blockSignals( isBlocked );
426   highlightSubShapes();
427   updateState();
428 }
429
430 //=================================================================================
431 // function : add
432 // purpose  : 
433 //=================================================================================
434 void GroupGUI_GroupDlg::add()
435 {
436   TColStd_MapOfInteger aMap;
437   for ( int i = 0, n = myIdList->count(); i < n; i++ )
438     aMap.Add( myIdList->item( i )->text().toInt() );
439
440   TColStd_IndexedMapOfInteger aMapIndex;
441
442   if ( IObjectCount() == 1 ) {
443     Standard_Boolean aResult = Standard_False;
444     GEOM::GEOM_Object_var anObj =
445       GEOMBase::ConvertIOinGEOMObject( firstIObject(), aResult );
446
447     if ( aResult && !anObj->_is_nil() )
448       ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->
449         selectionMgr()->GetIndexes( firstIObject(), aMapIndex );
450   }
451
452   // try to find out and process the object browser selection
453   if ( !aMapIndex.Extent() ) {
454     GEOM::ListOfGO anObjects;
455     GEOMBase::ConvertListOfIOInListOfGO(selectedIO(), anObjects);
456     
457     GEOM::GEOM_ILocalOperations_var aLocOp = getGeomEngine()->GetILocalOperations( getStudyId() );
458     GEOM::GEOM_IShapesOperations_var aShapesOp = getGeomEngine()->GetIShapesOperations( getStudyId() );
459     
460     for (int i = 0; i < anObjects.length(); i++) 
461       {
462         GEOM::GEOM_Object_var aGeomObj = anObjects[i];
463         GEOM::ListOfGO_var aSubObjects  = new GEOM::ListOfGO();
464         TopoDS_Shape aShape;
465         if ( GEOMBase::GetShape(aGeomObj, aShape, getShapeType()) ) 
466           {
467             aSubObjects->length(1);
468             aSubObjects[0] = aGeomObj;
469           }
470         else if (aGeomObj->GetType() == GEOM_GROUP)
471           aSubObjects = aShapesOp->MakeExplode( aGeomObj, getShapeType(), false);
472         else
473           break;
474         
475         for (int i = 0; i < aSubObjects->length(); i++) 
476           {
477           TopoDS_Shape aShape;
478           if ( GEOMBase::GetShape(aSubObjects[i], aShape, getShapeType()) ) 
479             {
480               CORBA::Long anIndex = aLocOp->GetSubShapeIndex( myMainObj, aSubObjects[i] );
481               if ( anIndex >= 0 )
482                 aMapIndex.Add( anIndex );
483             }
484         }
485       }
486   }
487
488   if ( aMapIndex.Extent() >= 1 ) {
489     QListBoxItem* anItem;
490     bool isBlocked = myIdList->signalsBlocked();
491     myIdList->blockSignals( true );
492
493     for ( int i = 1, n = aMapIndex.Extent(); i <= n; i++ ) {
494       if ( aMap.Contains( aMapIndex( i ) ) )
495         continue;
496
497       anItem = new QListBoxText( QString( "%1" ).arg( aMapIndex( i ) ) );
498       myIdList->insertItem( anItem );
499       myIdList->setSelected( anItem, true );
500     }
501
502     myIdList->blockSignals( isBlocked );
503   }
504
505   updateState();
506 }
507
508 //=================================================================================
509 // function : remove
510 // purpose  : 
511 //=================================================================================
512 void GroupGUI_GroupDlg::remove()
513 {
514   for ( int i = myIdList->count() - 1; i >= 0; i-- ) {
515     if ( myIdList->isSelected( i ) )
516       myIdList->removeItem( i );
517   }
518   updateState();
519 }
520
521
522 //=================================================================================
523 // function : getConstructorId()
524 // purpose  :
525 //=================================================================================
526 int GroupGUI_GroupDlg::getConstructorId() const
527 {
528   return GroupConstructors->id( GroupConstructors->selected() );
529 }
530
531 //=================================================================================
532 // function : getShapeType()
533 // purpose  :
534 //=================================================================================
535 TopAbs_ShapeEnum GroupGUI_GroupDlg::getShapeType() const
536 {
537   switch ( getConstructorId() ) {
538   case 0:  return TopAbs_VERTEX;
539   case 1:  return TopAbs_EDGE;
540   case 2:  return TopAbs_FACE;
541   case 3:  return TopAbs_SOLID;
542   default: return TopAbs_SHAPE;
543   }
544 }
545
546 //=================================================================================
547 // function : setShapeType()
548 // purpose  :
549 //=================================================================================
550 void GroupGUI_GroupDlg::setShapeType( const TopAbs_ShapeEnum theType )
551 {
552   int anId = 0;
553   switch ( theType ) {
554   case TopAbs_VERTEX: anId = 0; break;
555   case TopAbs_EDGE:   anId = 1; break;
556   case TopAbs_FACE:   anId = 2; break;
557   case TopAbs_SOLID:  anId = 3; break;
558   }
559   GroupConstructors->setButton( anId );
560 }
561
562
563 //=================================================================================
564 // function : activateSelection
565 // purpose  : Activate selection in accordance with myEditCurrentArgument
566 //=================================================================================
567 void GroupGUI_GroupDlg::activateSelection()
568 {
569   globalSelection( GEOM_ALLSHAPES );
570
571   if ( !myMainObj->_is_nil() && !myEditCurrentArgument ) {
572     localSelection( myMainObj, getShapeType() );
573   }
574
575   SelectionIntoArgument();
576 }
577
578 //=================================================================================
579 // function : updateState
580 // purpose  : 
581 //=================================================================================
582 void GroupGUI_GroupDlg::updateState()
583 {
584   bool isAdd = false;
585
586   TColStd_IndexedMapOfInteger aMapIndex;
587
588   if ( IObjectCount() == 1 ) {
589     Standard_Boolean aResult = Standard_False;
590     GEOM::GEOM_Object_var anObj =
591       GEOMBase::ConvertIOinGEOMObject( firstIObject(), aResult );
592
593     if ( aResult && !anObj->_is_nil() )
594       ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->
595         selectionMgr()->GetIndexes( firstIObject(), aMapIndex );
596   }
597
598   // try to find out and process the object browser selection
599   if ( !aMapIndex.Extent() && !CORBA::is_nil( myMainObj ) ) {
600     GEOM::ListOfGO anObjects;
601     GEOMBase::ConvertListOfIOInListOfGO(selectedIO(), anObjects);
602     
603     GEOM::GEOM_ILocalOperations_var aLocOp = getGeomEngine()->GetILocalOperations( getStudyId() );
604     GEOM::GEOM_IShapesOperations_var aShapesOp = getGeomEngine()->GetIShapesOperations( getStudyId() );
605
606      isAdd = true;
607      
608      for (int i = 0; i < anObjects.length(); i++) 
609        {
610          GEOM::GEOM_Object_var aGeomObj = anObjects[i];
611          GEOM::ListOfGO_var aSubObjects = new GEOM::ListOfGO();
612          TopoDS_Shape aShape;
613          if ( GEOMBase::GetShape(aGeomObj, aShape, getShapeType()) ) 
614            {
615              aSubObjects->length(1);
616              aSubObjects[0] = aGeomObj;
617            }
618          else if (aGeomObj->GetType() == GEOM_GROUP)
619            aSubObjects = aShapesOp->MakeExplode( aGeomObj, getShapeType(), false);
620          else
621            {
622              aMapIndex.Clear();
623              break;
624            }
625          
626          for (int i = 0; i < aSubObjects->length(); i++) 
627            {
628              TopoDS_Shape aShape;
629              aSubObjects[i];
630              if ( GEOMBase::GetShape(aSubObjects[i], aShape, getShapeType()) ) 
631                {
632                  CORBA::Long anIndex = aLocOp->GetSubShapeIndex( myMainObj, aSubObjects[i] );
633                  if ( anIndex >= 0 )
634                    aMapIndex.Add( anIndex );
635                  else
636                    isAdd = false;
637                }
638              else
639                isAdd = false;
640              
641              if ( !isAdd ) {
642                aMapIndex.Clear();
643                break;
644              }
645            }
646          
647          if ( !isAdd ) {
648            aMapIndex.Clear();
649            break;
650          }
651        }
652   }
653   
654   isAdd = aMapIndex.Extent() > 0;
655
656   myAddBtn->setEnabled( !myEditCurrentArgument && !CORBA::is_nil( myMainObj ) && isAdd );
657   bool hasSel = false;
658   for ( int ii = 0, nn = myIdList->count(); !hasSel && ii < nn; ii++ )
659     hasSel =  myIdList->isSelected( ii );
660   myRemBtn->setEnabled( hasSel );
661   mySelSubBtn->setEnabled( !CORBA::is_nil( myMainObj ) );
662   mySelAllBtn->setEnabled( !CORBA::is_nil( myMainObj ) );
663 }
664
665 //=================================================================================
666 // function : selectionChanged
667 // purpose  :
668 //=================================================================================
669 void GroupGUI_GroupDlg::selectionChanged()
670 {
671   highlightSubShapes();
672 }
673
674 //=================================================================================
675 // function : highlightSubShapes
676 // purpose  :
677 //=================================================================================
678 void GroupGUI_GroupDlg::highlightSubShapes()
679 {
680   if ( CORBA::is_nil( myMainObj ) )
681     return;
682
683   Standard_Boolean isOk;
684   Handle(GEOM_AISShape) aSh =
685     GEOMBase::ConvertIORinGEOMAISShape( GEOMBase::GetIORFromObject( myMainObj ), isOk, true );
686   if ( !isOk || aSh.IsNull() )
687     return;
688
689   TColStd_MapOfInteger anIds;
690
691   myBusy = true;
692
693   for ( int ii = 0, nn = myIdList->count(); ii < nn; ii++ )
694     if ( myIdList->isSelected( ii ) )
695       anIds.Add( myIdList->item( ii )->text().toInt() );
696
697   LightApp_SelectionMgr* aSelMgr =
698     ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->selectionMgr();
699   aSelMgr->clearSelected();
700   aSelMgr->AddOrRemoveIndex( aSh->getIO(), anIds, false );
701
702   myBusy = false;
703
704   updateState();
705 }
706
707 //=================================================================================
708 // function : createOperation
709 // purpose  :
710 //=================================================================================
711 GEOM::GEOM_IOperations_ptr GroupGUI_GroupDlg::createOperation()
712 {
713   return getGeomEngine()->GetIGroupOperations( getStudyId() );
714 }
715
716 #define RETURN_WITH_MSG(a, b) \
717   if ( !(a) ) { \
718     theMessage += (b); \
719     return false; \
720   }
721
722 //=================================================================================
723 // function : isValid()
724 // purpose  : Verify validity of input data
725 //=================================================================================
726 bool GroupGUI_GroupDlg::isValid( QString& theMessage )
727 {
728   SalomeApp_Study* study = getStudy();
729   ASSERT(study);
730   RETURN_WITH_MSG  ( !study->studyDS()->GetProperties()->IsLocked(), tr( "GEOM_STUDY_LOCKED" ) )
731
732   if ( myMode == CreateGroup ) {
733     RETURN_WITH_MSG( !CORBA::is_nil( myMainObj ), tr( "NO_MAIN_OBJ" ) )
734   }
735   else {
736     RETURN_WITH_MSG( !CORBA::is_nil( myMainObj ), tr( "NO_GROUP" ) )
737   }
738
739   QString aName (getNewObjectName());
740   RETURN_WITH_MSG  ( !aName.stripWhiteSpace().isEmpty(), tr( "EMPTY_NAME" ) )
741
742   RETURN_WITH_MSG  ( myIdList->count(), tr( "EMPTY_LIST" ) )
743   return true;
744 }
745
746 //=================================================================================
747 // function : execute
748 // purpose  :
749 //=================================================================================
750 bool GroupGUI_GroupDlg::execute( ObjectList& objects )
751 {
752   GEOM::GEOM_IGroupOperations_var anOp = GEOM::GEOM_IGroupOperations::_narrow( getOperation() );
753
754   GEOM::GEOM_Object_var aGroup;
755   if ( myMode == CreateGroup )
756     aGroup = anOp->CreateGroup( myMainObj, getShapeType() );
757   else if ( myMode == EditGroup )
758     aGroup = myGroup;
759
760   if ( CORBA::is_nil( aGroup ) || ( myMode == CreateGroup && !anOp->IsDone() ) )
761     return false;
762
763   GEOM::ListOfLong_var aCurrList = anOp->GetObjects( aGroup );
764   if ( !anOp->IsDone()  )
765     return false;
766
767   for ( int i = 0, n = aCurrList->length(); i < n; i++ ) {
768     anOp->RemoveObject( aGroup, aCurrList[i] );
769     if ( !anOp->IsDone()  )
770       return false;
771   }
772
773   for ( int ii = 0, nn = myIdList->count(); ii < nn; ii++ ) {
774     anOp->AddObject( aGroup, myIdList->item( ii )->text().toInt() );
775     if ( !anOp->IsDone()  )
776       return false;
777   }
778
779   SalomeApp_Study* study = getStudy();
780   if ( study ) {
781     string IOR = GEOMBase::GetIORFromObject( aGroup );
782     if ( IOR != "" ) {
783       _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( IOR ) );
784       if ( SO ) {
785         _PTR(StudyBuilder) aBuilder (study->studyDS()->NewBuilder());
786         aBuilder->SetName( SO, getNewObjectName() );
787       }
788     }
789   }
790
791   objects.push_back( aGroup._retn() );
792
793   return true;
794 }
795
796 //================================================================
797 // Function : getFather
798 // Purpose  : Get father object for object to be added in study
799 //            ( called with addInStudy method )
800 //================================================================
801 GEOM::GEOM_Object_ptr GroupGUI_GroupDlg::getFather( GEOM::GEOM_Object_ptr theObj )
802 {
803   GEOM::GEOM_Object_var aFatherObj;
804   if ( theObj->GetType() == GEOM_GROUP ) {
805     GEOM::GEOM_IGroupOperations_var anOp = GEOM::GEOM_IGroupOperations::_narrow( getOperation() );
806     aFatherObj = anOp->GetMainShape( theObj );
807   }
808   return aFatherObj._retn();
809 }
810