]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_SelectionOp.cxx
Salome HOME
précision
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_SelectionOp.cxx
1 // Copyright (C) 2007-2020  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 // File   : SMESHGUI_SelectionOp.cxx
24 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S.
25 // SMESH includes
26 //
27 #include "SMESHGUI_SelectionOp.h"
28
29 #include "SMESHGUI.h"
30 #include "SMESHGUI_VTKUtils.h"
31 #include "SMESHGUI_MeshUtils.h"
32 #include "SMESHGUI_Selection.h"
33
34 #include <SMESH_Actor.h>
35 #include <SMDS_Mesh.hxx>
36 #include <SMDS_MeshNode.hxx>
37
38 // SALOME GUI includes
39 #include <SUIT_SelectionFilter.h>
40 #include <LightApp_SelectionMgr.h>
41 #include <SalomeApp_Study.h>
42 #include <SVTK_ViewWindow.h>
43 #include <SVTK_ViewModel.h>
44
45 #include <SALOME_ListIO.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 highlight, const bool immediately )
254 {
255   SVTK_ViewWindow* wnd = viewWindow();
256   if( wnd )
257     wnd->highlight( obj, highlight, 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   _PTR(Study) aStudy = SMESH::getStudy();
299
300   int res = -1;
301   if( objtype == Object )
302   {
303     int t = SMESHGUI_Selection::type( str );
304     if( t<0 )
305     {
306       //try to get GEOM type
307       _PTR( SObject ) sobj = aStudy->FindObjectID( str.toUtf8().data() );
308       if( sobj )
309       {
310         GEOM::GEOM_Object_var obj = GEOM::GEOM_Object::_narrow(
311           dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
312         if( !CORBA::is_nil( obj ) )
313           // as decoding of type id is not realized in LightApp_Dialog,
314           //make all GEOM objects have same type id
315           res = SMESHGUI_Dialog::prefix( "GEOM" );// + obj->GetType();
316       }
317     }
318     else
319       res = SMESHGUI_Dialog::prefix( "SMESH" ) + t;
320   }
321   else
322   {
323     int pos = str.indexOf( idChar() );
324     QString entry = str.left( pos ),
325             _id = str.mid( pos+1 );
326     bool ok;
327     int id = _id.toInt( &ok );
328     if( ok )
329     {
330       _PTR( SObject ) sobj = aStudy->FindObjectID( entry.toUtf8().data() );
331       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( 
332         dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
333       SMESH::SMESH_subMesh_var submesh = SMESH::SMESH_subMesh::_narrow( 
334         dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
335       if( !CORBA::is_nil( mesh ) )
336         res = SMESHGUI_Dialog::prefix( "SMESH element" ) + 
337           mesh->GetElementType( id, objtype==MeshElement );
338
339       else if( !CORBA::is_nil( submesh ) )
340         res = SMESHGUI_Dialog::prefix( "SMESH element" ) + 
341           submesh->GetElementType( id, objtype==MeshElement );
342     }
343   }
344
345   return res;
346 }
347
348 //=======================================================================
349 // name    : selected
350 // Purpose : Get names, types and ids of selected objects
351 //=======================================================================
352 void SMESHGUI_SelectionOp::selected( QStringList& names,
353                                      LightApp_Dialog::TypesList& types,
354                                      QStringList& ids ) const
355 {
356 /*  SUIT_DataOwnerPtrList list; selectionMgr()->selected( list );
357   SUIT_DataOwnerPtrList::const_iterator anIt = list.begin(),
358                                         aLast = list.end();
359   for( ; anIt!=aLast; anIt++ )
360   {
361     LightApp_DataOwner* owner = dynamic_cast<LightApp_DataOwner*>( (*anIt).operator->() );
362     LightApp_SVTKDataOwner* vtkowner = dynamic_cast<LightApp_SVTKDataOwner*>( (*anIt).operator->() );
363
364     if( vtkowner )
365     {
366       QString id_str = QString( "%1%2%3" ).arg( vtkowner->entry() ).arg( idChar() ), current_id_str;
367       Selection_Mode mode = vtkowner->GetMode();
368       EntityType objtype = mode == NodeSelection ? MeshNode : MeshElement;
369       const TColStd_IndexedMapOfInteger& ownerids = vtkowner->GetIds();
370
371       for( int i=1, n=ownerids.Extent(); i<=n; i++ )
372       {
373         int curid = ownerids( i );
374         current_id_str = id_str.arg( curid );
375         ids.append( current_id_str );
376         types.append( typeById( current_id_str, objtype ) );
377         names.append( QString( "%1" ).arg( curid ) );
378       }
379     }
380
381     else if( owner )
382     {
383       QString id = owner->entry();
384       ids.append( id );
385       types.append( typeById( id, Object ) );
386       names.append( owner->IO()->getName() );
387     }
388   }*/
389
390   SALOME_ListIO selObjs;
391   TColStd_IndexedMapOfInteger selIndices;
392   selectionMgr()->selectedObjects( selObjs );
393   Selection_Mode mode = selectionMode();
394   EntityType objtype = mode == NodeSelection ? MeshNode : MeshElement;
395
396   for( SALOME_ListIteratorOfListIO anIt( selObjs ); anIt.More(); anIt.Next() )
397   {
398     selIndices.Clear();
399     selectionMgr()->GetIndexes( anIt.Value(), selIndices );
400     if( selIndices.Extent() > 0 )
401     {
402       QString id_str = QString( "%1%2%3" ).arg( anIt.Value()->getEntry() ).arg( idChar() ), current_id_str;
403       for( int i=1, n=selIndices.Extent(); i<=n; i++ )
404       {
405         int curid = selIndices( i );
406         current_id_str = id_str.arg( curid );
407         ids.append( current_id_str );
408         types.append( typeById( current_id_str, objtype ) );
409         names.append( QString( "%1" ).arg( curid ) );
410       }
411     }
412     else
413     {
414       QString id = anIt.Value()->getEntry();
415       ids.append( id );
416       types.append( typeById( id, Object ) );
417       _PTR(SObject) obj = SMESH::getStudy()->FindObjectID( anIt.Value()->getEntry() );
418       if( obj )
419         names.append( QString( obj->GetName().c_str() ).trimmed() );
420     }
421   }
422 }
423
424 //=======================================================================
425 // name    : idChar
426 // Purpose : Char using to divide <entry> and <id> in string id representation. By default, '#'
427 //=======================================================================
428 QChar SMESHGUI_SelectionOp::idChar() const
429 {
430   return '#';
431 }
432
433 //=================================================================================
434 // name     : mesh
435 // purpose  :
436 //=================================================================================
437 SMESH::SMESH_Mesh_var SMESHGUI_SelectionOp::mesh() const
438 {
439   if( selectionMode()==ActorSelection )
440     return SMESH::SMESH_Mesh::_nil();
441     
442   SALOME_ListIO sel; selectionMgr()->selectedObjects( sel, SVTK_Viewer::Type() );
443   if( sel.Extent()==1 )
444     return SMESH::GetMeshByIO( sel.First() );
445   else
446     return SMESH::SMESH_Mesh::_nil();
447 }
448
449 //=================================================================================
450 // name     : actor
451 // purpose  :
452 //=================================================================================
453 SMESH_Actor* SMESHGUI_SelectionOp::actor() const
454 {
455   SMESH::SMESH_Mesh_var m = mesh();
456   if( !m->_is_nil() )
457     return SMESH::FindActorByObject( m.in() );
458   else
459     return 0;
460 }
461
462 //=================================================================================
463 // name     : onTextChanged
464 // purpose  :
465 //=================================================================================
466 void SMESHGUI_SelectionOp::onTextChanged( int, const QStringList& list )
467 {
468     if( !dlg() )
469       return;
470
471     TColStd_MapOfInteger newIndices;
472
473     SALOME_ListIO sel; selectionMgr()->selectedObjects( sel );
474     SMESH_Actor* anActor = actor();
475     if( sel.Extent()==0 || !anActor )
476       return;
477
478     SMDS_Mesh* aMesh = anActor->GetObject()->GetMesh();
479
480     IdList ids; extractIds( list, ids, '\0' );
481     IdList::const_iterator anIt = ids.begin(),
482                            aLast = ids.end();
483     if ( selectionMode() == NodeSelection )
484       for( ; anIt!=aLast; anIt++ ) {
485         if( const SMDS_MeshNode * n = aMesh->FindNode( *anIt ) )
486           newIndices.Add( n->GetID() );
487       }
488     else 
489       for( ; anIt!=aLast; anIt++ ) {
490         if( const SMDS_MeshElement* e = aMesh->FindElement( *anIt ) )
491           newIndices.Add( e->GetID() );
492       }
493
494     selector()->AddOrRemoveIndex( sel.First(), newIndices, false );
495     highlight( sel.First(), true, true );
496
497     QStringList names, _ids; LightApp_Dialog::TypesList types;
498     selected( names, types, _ids );
499     dlg()->selectObject( names, types, _ids, false );
500 }
501
502 //=================================================================================
503 // name     : selectedIds
504 // purpose  :
505 //=================================================================================
506 void SMESHGUI_SelectionOp::selectedIds( const int id, IdList& list ) const
507 {
508   if( !dlg() )
509     return;
510
511   QStringList ids; dlg()->selectedObject( id, ids );
512   extractIds( ids, list );
513 }
514
515 //=================================================================================
516 // name     : extractIds
517 // purpose  :
518 //=================================================================================
519 void SMESHGUI_SelectionOp::extractIds( const QStringList& ids, IdList& list, const QChar idchar )
520 {
521   QStringList::const_iterator anIt = ids.begin(),
522                               aLast = ids.end();
523   QString id_str;
524   for( ; anIt!=aLast; anIt++ )
525   {
526     id_str = *anIt;
527     int pos = idchar=='\0' ? -1 : id_str.indexOf( idchar );
528     int id = -1;
529     if( idchar=='\0' || pos>=0 )
530     {
531       id = id_str.mid( pos+1 ).toInt();
532       list.append( id );
533     }
534   }
535 }
536
537 //=================================================================================
538 // name     : extractIds
539 // purpose  :
540 //=================================================================================
541 void SMESHGUI_SelectionOp::extractIds( const QStringList& ids, IdList& list ) const
542 {
543   extractIds( ids, list, idChar() );
544 }