]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_SelectionOp.cxx
Salome HOME
0020145: EDF 666 SMESH: Modifications of GEOM groups are not taken into account
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_SelectionOp.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 // File   : SMESHGUI_SelectionOp.cxx
23 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S.
24 // SMESH includes
25 //
26 #include "SMESHGUI_SelectionOp.h"
27
28 #include "SMESHGUI.h"
29 #include "SMESHGUI_VTKUtils.h"
30 #include "SMESHGUI_MeshUtils.h"
31 #include "SMESHGUI_Selection.h"
32
33 #include <SMESH_Actor.h>
34 #include <SMDS_Mesh.hxx>
35 #include <SMDS_MeshNode.hxx>
36
37 // SALOME GUI includes
38 #include <SUIT_SelectionFilter.h>
39 #include <LightApp_SelectionMgr.h>
40 #include <SalomeApp_Study.h>
41 #include <SVTK_ViewWindow.h>
42 #include <SVTK_ViewModel.h>
43
44 #include <SALOME_ListIO.hxx>
45 #include <SALOME_ListIteratorOfListIO.hxx>
46
47 // SALOME KERNEL includes 
48 #include <SALOMEDS_SObject.hxx>
49
50 /*
51   Class       : SMESHGUI_SelectionOp
52   Description : Base operation for all operations using object selection in viewer or objectbrowser
53                 through common widgets created by LightApp_Dialog::createObject
54 */
55
56 //=================================================================================
57 // name     : SMESHGUI_SelectionOp
58 // purpose  : 
59 //=================================================================================
60 SMESHGUI_SelectionOp::SMESHGUI_SelectionOp( const Selection_Mode mode )
61 : SMESHGUI_Operation(),
62   myDefSelectionMode( mode )
63 {
64 }
65
66 //=================================================================================
67 // name     : ~SMESHGUI_SelectionOp
68 // purpose  :
69 //=================================================================================
70 SMESHGUI_SelectionOp::~SMESHGUI_SelectionOp()
71 {
72   removeCustomFilters();
73 }
74
75 //=================================================================================
76 // name     : startOperation
77 // purpose  :
78 //=================================================================================
79 void SMESHGUI_SelectionOp::startOperation()
80 {
81   myOldSelectionMode = selectionMode();
82   setSelectionMode( myDefSelectionMode );
83
84   SMESHGUI_Operation::startOperation();
85   if( dlg() )
86   {
87     disconnect( dlg(), SIGNAL( objectActivated( int ) ), this, SLOT( onActivateObject( int ) ) );
88     disconnect( dlg(), SIGNAL( objectDeactivated( int ) ), this, SLOT( onDeactivateObject( int ) ) );
89     disconnect( dlg(), SIGNAL( selectionChanged( int ) ), this, SLOT( onSelectionChanged( int ) ) );
90     connect( dlg(), SIGNAL( objectActivated( int ) ), this, SLOT( onActivateObject( int ) ) );
91     connect( dlg(), SIGNAL( objectDeactivated( int ) ), this, SLOT( onDeactivateObject( int ) ) );
92     connect( dlg(), SIGNAL( selectionChanged( int ) ), this, SLOT( onSelectionChanged( int ) ) );
93   }
94 }
95
96 //=================================================================================
97 // name     : removeCustomFilters
98 // purpose  :
99 //=================================================================================
100 void SMESHGUI_SelectionOp::removeCustomFilters()
101 {
102   if (myFilters.count() > 0) {
103     LightApp_SelectionMgr* mgr = selectionMgr();
104     Filters::const_iterator anIt = myFilters.begin(),
105                             aLast = myFilters.end();
106     for ( ; anIt != aLast; anIt++) {
107       if (anIt.value()) {
108         if (mgr) mgr->removeFilter(anIt.value());
109         delete anIt.value();
110       }
111     }
112
113     myFilters.clear();
114   }
115 }
116
117 //=================================================================================
118 // name     : commitOperation
119 // purpose  :
120 //=================================================================================
121 void SMESHGUI_SelectionOp::commitOperation()
122 {
123   SMESHGUI_Operation::commitOperation();  
124   removeCustomFilters();
125   setSelectionMode( myOldSelectionMode );
126 }
127
128 //=================================================================================
129 // name     : abortOperation
130 // purpose  :
131 //=================================================================================
132 void SMESHGUI_SelectionOp::abortOperation()
133 {
134   SMESHGUI_Operation::abortOperation();
135   removeCustomFilters();
136   setSelectionMode( myOldSelectionMode );  
137 }
138
139 //=================================================================================
140 // name     : selectionDone
141 // purpose  :
142 //=================================================================================
143 void SMESHGUI_SelectionOp::selectionDone()
144 {
145   if( !dlg() )
146     return;
147
148   if( selectionMode()!=ActorSelection )
149   {
150     SALOME_ListIO aList;
151     selectionMgr()->selectedObjects( aList, SVTK_Viewer::Type() );
152
153     if( aList.Extent() != 1 ) //we can select nodes or elements only within one mesh
154     {
155       dlg()->clearSelection();
156       return;
157     }    
158   }
159     
160   QStringList names, ids;
161   LightApp_Dialog::TypesList types;
162   selected( names, types, ids );
163   dlg()->selectObject( names, types, ids );
164 }
165
166 //=================================================================================
167 // name     : createFilter
168 // purpose  :
169 //=================================================================================
170 SUIT_SelectionFilter* SMESHGUI_SelectionOp::createFilter( const int ) const
171 {
172   return 0;
173 }
174
175 //=================================================================================
176 // name     : onActivateObject
177 // purpose  :
178 //=================================================================================
179 void SMESHGUI_SelectionOp::onActivateObject( int id )
180 {
181   LightApp_SelectionMgr* mgr = selectionMgr();
182   if( !mgr )
183     return;
184     
185   if( !myFilters.contains( id ) )
186     myFilters[ id ] = createFilter( id );
187
188   if( myFilters[ id ] )
189     mgr->installFilter( myFilters[ id ] );
190
191   selectionDone();
192 }
193
194 //=================================================================================
195 // name     : onDeactivateObject
196 // purpose  :
197 //=================================================================================
198 void SMESHGUI_SelectionOp::onDeactivateObject( int id )
199 {
200   removeCustomFilters();
201 }
202
203 //=================================================================================
204 // name     : initDialog
205 // purpose  :
206 //=================================================================================
207 void SMESHGUI_SelectionOp::initDialog()
208 {
209   if( dlg() )
210   {
211     dlg()->clearSelection();
212     dlg()->deactivateAll();
213   }
214 }
215
216 //=================================================================================
217 // name     : initDialog
218 // purpose  :
219 //=================================================================================
220 void SMESHGUI_SelectionOp::onSelectionChanged( int )
221 {
222 }
223
224 //=======================================================================
225 // name    : selectionMode
226 // Purpose : Returns selection mode
227 //=======================================================================
228 Selection_Mode SMESHGUI_SelectionOp::selectionMode() const
229 {
230   SVTK_ViewWindow* wnd = viewWindow();
231   if( wnd )
232     return wnd->SelectionMode();
233   else
234     return ActorSelection;
235 }
236
237 //=======================================================================
238 // name    : setSelectionMode
239 // Purpose : Set selection mode
240 //=======================================================================
241 void SMESHGUI_SelectionOp::setSelectionMode( const Selection_Mode mode )
242 {
243   SVTK_ViewWindow* wnd = viewWindow();
244   if( wnd )
245     wnd->SetSelectionMode( mode );
246 }
247
248 //=======================================================================
249 // name    : highlight
250 // Purpose : Highlight object in 3d viewer
251 //=======================================================================
252 void SMESHGUI_SelectionOp::highlight( const Handle( SALOME_InteractiveObject )& obj,
253                                       const bool hilight, const bool immediately )
254 {
255   SVTK_ViewWindow* wnd = viewWindow();
256   if( wnd )
257     wnd->highlight( obj, hilight, immediately );
258 }
259
260 //=======================================================================
261 // name    : addOrRemoveIndex
262 // Purpose : Select/deselect cells of mesh
263 //=======================================================================
264 void SMESHGUI_SelectionOp::addOrRemoveIndex( const Handle( SALOME_InteractiveObject )& obj,
265                                              const TColStd_MapOfInteger& indices,
266                                              const bool isModeShift )
267 {
268   SVTK_Selector* sel = selector();
269   if( sel )
270     sel->AddOrRemoveIndex( obj, indices, isModeShift );
271 }
272
273 //=======================================================================
274 // name    : viewWindow
275 // Purpose : Get active view window
276 //=======================================================================
277 SVTK_ViewWindow* SMESHGUI_SelectionOp::viewWindow() const
278 {
279   return SMESH::GetViewWindow( getSMESHGUI() );
280 }
281
282 //=======================================================================
283 // name    : selector
284 // Purpose : Get selector
285 //=======================================================================
286 SVTK_Selector* SMESHGUI_SelectionOp::selector() const
287 {
288   SVTK_ViewWindow* wnd = viewWindow();
289   return wnd ? wnd->GetSelector() : 0;
290 }
291
292 //=======================================================================
293 // name    : typeById
294 // Purpose : Find type by id
295 //=======================================================================
296 int SMESHGUI_SelectionOp::typeById( const QString& str, const EntityType objtype ) const
297 {
298   SalomeApp_Study* _study = dynamic_cast<SalomeApp_Study*>( study() );
299   if( !_study )
300     return -1;
301
302   _PTR( Study ) st = _study->studyDS();
303
304   int res = -1;
305   if( objtype == Object )
306   {
307     SalomeApp_Study* _study = dynamic_cast<SalomeApp_Study*>( study() );
308     if( _study )
309     {
310       int t = SMESHGUI_Selection::type( str, _study->studyDS() );
311       if( t<0 )
312       {
313         //try to get GEOM type
314         _PTR( SObject ) sobj = st->FindObjectID( str.toLatin1().data() );
315         if( sobj )
316         {
317           GEOM::GEOM_Object_var obj = GEOM::GEOM_Object::_narrow(
318             dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
319           if( !CORBA::is_nil( obj ) )
320             res = SMESHGUI_Dialog::prefix( "GEOM" ) + obj->GetType();
321         }
322       }
323       else
324         res = SMESHGUI_Dialog::prefix( "SMESH" ) + t;
325     }
326   }
327   else
328   {
329     int pos = str.indexOf( idChar() );
330     QString entry = str.left( pos ),
331             _id = str.mid( pos+1 );
332     bool ok;
333     int id = _id.toInt( &ok );
334     if( ok )
335     {
336       _PTR( SObject ) sobj = st->FindObjectID( entry.toLatin1().data() );
337       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( 
338         dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
339       SMESH::SMESH_subMesh_var submesh = SMESH::SMESH_subMesh::_narrow( 
340         dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
341       if( !CORBA::is_nil( mesh ) )
342         res = SMESHGUI_Dialog::prefix( "SMESH element" ) + 
343           mesh->GetElementType( id, objtype==MeshElement );
344
345       else if( !CORBA::is_nil( submesh ) )
346         res = SMESHGUI_Dialog::prefix( "SMESH element" ) + 
347           submesh->GetElementType( id, objtype==MeshElement );
348     }
349   }
350
351   return res;
352 }
353
354 //=======================================================================
355 // name    : selected
356 // Purpose : Get names, types and ids of selected objects
357 //=======================================================================
358 void SMESHGUI_SelectionOp::selected( QStringList& names,
359                                      LightApp_Dialog::TypesList& types,
360                                      QStringList& ids ) const
361 {
362 /*  SUIT_DataOwnerPtrList list; selectionMgr()->selected( list );
363   SUIT_DataOwnerPtrList::const_iterator anIt = list.begin(),
364                                         aLast = list.end();
365   for( ; anIt!=aLast; anIt++ )
366   {
367     LightApp_DataOwner* owner = dynamic_cast<LightApp_DataOwner*>( (*anIt).operator->() );
368     LightApp_SVTKDataOwner* vtkowner = dynamic_cast<LightApp_SVTKDataOwner*>( (*anIt).operator->() );
369
370     if( vtkowner )
371     {
372       QString id_str = QString( "%1%2%3" ).arg( vtkowner->entry() ).arg( idChar() ), current_id_str;
373       Selection_Mode mode = vtkowner->GetMode();
374       EntityType objtype = mode == NodeSelection ? MeshNode : MeshElement;
375       const TColStd_IndexedMapOfInteger& ownerids = vtkowner->GetIds();
376
377       for( int i=1, n=ownerids.Extent(); i<=n; i++ )
378       {
379         int curid = ownerids( i );
380         current_id_str = id_str.arg( curid );
381         ids.append( current_id_str );
382         types.append( typeById( current_id_str, objtype ) );
383         names.append( QString( "%1" ).arg( curid ) );
384       }
385     }
386
387     else if( owner )
388     {
389       QString id = owner->entry();
390       ids.append( id );
391       types.append( typeById( id, Object ) );
392       names.append( owner->IO()->getName() );
393     }
394   }*/
395
396   SALOME_ListIO selObjs;
397   TColStd_IndexedMapOfInteger selIndices;
398   selectionMgr()->selectedObjects( selObjs );
399   Selection_Mode mode = selectionMode();
400   EntityType objtype = mode == NodeSelection ? MeshNode : MeshElement;
401
402   for( SALOME_ListIteratorOfListIO anIt( selObjs ); anIt.More(); anIt.Next() )
403   {
404     selIndices.Clear();
405     selectionMgr()->GetIndexes( anIt.Value(), selIndices );
406     if( selIndices.Extent() > 0 )
407     {
408       QString id_str = QString( "%1%2%3" ).arg( anIt.Value()->getEntry() ).arg( idChar() ), current_id_str;
409       for( int i=1, n=selIndices.Extent(); i<=n; i++ )
410       {
411         int curid = selIndices( i );
412         current_id_str = id_str.arg( curid );
413         ids.append( current_id_str );
414         types.append( typeById( current_id_str, objtype ) );
415         names.append( QString( "%1" ).arg( curid ) );
416       }
417     }
418     else
419     {
420       QString id = anIt.Value()->getEntry();
421       ids.append( id );
422       types.append( typeById( id, Object ) );
423       SalomeApp_Study* _study = dynamic_cast<SalomeApp_Study*>( study() );
424       if( _study )
425       {
426         _PTR(SObject) obj = _study->studyDS()->FindObjectID( anIt.Value()->getEntry() );
427         if( obj )
428           names.append( obj->GetName().c_str() );
429       }
430     }
431   }
432 }
433
434 //=======================================================================
435 // name    : idChar
436 // Purpose : Char using to divide <entry> and <id> in string id representation. By default, '#'
437 //=======================================================================
438 QChar SMESHGUI_SelectionOp::idChar() const
439 {
440   return '#';
441 }
442
443 //=================================================================================
444 // name     : mesh
445 // purpose  :
446 //=================================================================================
447 SMESH::SMESH_Mesh_var SMESHGUI_SelectionOp::mesh() const
448 {
449   if( selectionMode()==ActorSelection )
450     return SMESH::SMESH_Mesh::_nil();
451     
452   SALOME_ListIO sel; selectionMgr()->selectedObjects( sel, SVTK_Viewer::Type() );
453   if( sel.Extent()==1 )
454     return SMESH::GetMeshByIO( sel.First() );
455   else
456     return SMESH::SMESH_Mesh::_nil();
457 }
458
459 //=================================================================================
460 // name     : actor
461 // purpose  :
462 //=================================================================================
463 SMESH_Actor* SMESHGUI_SelectionOp::actor() const
464 {
465   SMESH::SMESH_Mesh_var m = mesh();
466   if( !m->_is_nil() )
467     return SMESH::FindActorByObject( m.in() );
468   else
469     return 0;
470 }
471
472 //=================================================================================
473 // name     : onTextChanged
474 // purpose  :
475 //=================================================================================
476 void SMESHGUI_SelectionOp::onTextChanged( int, const QStringList& list )
477 {
478     if( !dlg() )
479       return;
480
481     TColStd_MapOfInteger newIndices;
482
483     SALOME_ListIO sel; selectionMgr()->selectedObjects( sel );
484     SMESH_Actor* anActor = actor();
485     if( sel.Extent()==0 || !anActor )
486       return;
487
488     SMDS_Mesh* aMesh = anActor->GetObject()->GetMesh();
489
490     IdList ids; extractIds( list, ids, '\0' );
491     IdList::const_iterator anIt = ids.begin(),
492                            aLast = ids.end();
493     for( ; anIt!=aLast; anIt++ )
494       if( const SMDS_MeshNode * n = aMesh->FindNode( *anIt ) )
495         newIndices.Add( n->GetID() );
496
497     selector()->AddOrRemoveIndex( sel.First(), newIndices, false );
498     highlight( sel.First(), true, true );
499
500     QStringList names, _ids; LightApp_Dialog::TypesList types;
501     selected( names, types, _ids );
502     dlg()->selectObject( names, types, _ids, false );
503 }
504
505 //=================================================================================
506 // name     : selectedIds
507 // purpose  :
508 //=================================================================================
509 void SMESHGUI_SelectionOp::selectedIds( const int id, IdList& list ) const
510 {
511   if( !dlg() )
512     return;
513
514   QStringList ids; dlg()->selectedObject( id, ids );
515   extractIds( ids, list );
516 }
517
518 //=================================================================================
519 // name     : extractIds
520 // purpose  :
521 //=================================================================================
522 void SMESHGUI_SelectionOp::extractIds( const QStringList& ids, IdList& list, const QChar idchar )
523 {
524   QStringList::const_iterator anIt = ids.begin(),
525                               aLast = ids.end();
526   QString id_str;
527   for( ; anIt!=aLast; anIt++ )
528   {
529     id_str = *anIt;
530     int pos = idchar=='\0' ? -1 : id_str.indexOf( idchar );
531     int id = -1;
532     if( idchar=='\0' || pos>=0 )
533     {
534       id = id_str.mid( pos+1 ).toInt();
535       list.append( id );
536     }
537   }
538 }
539
540 //=================================================================================
541 // name     : extractIds
542 // purpose  :
543 //=================================================================================
544 void SMESHGUI_SelectionOp::extractIds( const QStringList& ids, IdList& list ) const
545 {
546   extractIds( ids, list, idChar() );
547 }