Salome HOME
52564: TC7.5.0: total quantity of 2D and 3D mesh elements is not calculated
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_SelectionOp.cxx
1 // Copyright (C) 2007-2014  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 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             // as decoding of type id is not realized in LightApp_Dialog,
321             //make all GEOM objects have same type id
322             res = SMESHGUI_Dialog::prefix( "GEOM" );// + obj->GetType();
323         }
324       }
325       else
326         res = SMESHGUI_Dialog::prefix( "SMESH" ) + t;
327     }
328   }
329   else
330   {
331     int pos = str.indexOf( idChar() );
332     QString entry = str.left( pos ),
333             _id = str.mid( pos+1 );
334     bool ok;
335     int id = _id.toInt( &ok );
336     if( ok )
337     {
338       _PTR( SObject ) sobj = st->FindObjectID( entry.toLatin1().data() );
339       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( 
340         dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
341       SMESH::SMESH_subMesh_var submesh = SMESH::SMESH_subMesh::_narrow( 
342         dynamic_cast<SALOMEDS_SObject*>( sobj.get() )->GetObject() );
343       if( !CORBA::is_nil( mesh ) )
344         res = SMESHGUI_Dialog::prefix( "SMESH element" ) + 
345           mesh->GetElementType( id, objtype==MeshElement );
346
347       else if( !CORBA::is_nil( submesh ) )
348         res = SMESHGUI_Dialog::prefix( "SMESH element" ) + 
349           submesh->GetElementType( id, objtype==MeshElement );
350     }
351   }
352
353   return res;
354 }
355
356 //=======================================================================
357 // name    : selected
358 // Purpose : Get names, types and ids of selected objects
359 //=======================================================================
360 void SMESHGUI_SelectionOp::selected( QStringList& names,
361                                      LightApp_Dialog::TypesList& types,
362                                      QStringList& ids ) const
363 {
364 /*  SUIT_DataOwnerPtrList list; selectionMgr()->selected( list );
365   SUIT_DataOwnerPtrList::const_iterator anIt = list.begin(),
366                                         aLast = list.end();
367   for( ; anIt!=aLast; anIt++ )
368   {
369     LightApp_DataOwner* owner = dynamic_cast<LightApp_DataOwner*>( (*anIt).operator->() );
370     LightApp_SVTKDataOwner* vtkowner = dynamic_cast<LightApp_SVTKDataOwner*>( (*anIt).operator->() );
371
372     if( vtkowner )
373     {
374       QString id_str = QString( "%1%2%3" ).arg( vtkowner->entry() ).arg( idChar() ), current_id_str;
375       Selection_Mode mode = vtkowner->GetMode();
376       EntityType objtype = mode == NodeSelection ? MeshNode : MeshElement;
377       const TColStd_IndexedMapOfInteger& ownerids = vtkowner->GetIds();
378
379       for( int i=1, n=ownerids.Extent(); i<=n; i++ )
380       {
381         int curid = ownerids( i );
382         current_id_str = id_str.arg( curid );
383         ids.append( current_id_str );
384         types.append( typeById( current_id_str, objtype ) );
385         names.append( QString( "%1" ).arg( curid ) );
386       }
387     }
388
389     else if( owner )
390     {
391       QString id = owner->entry();
392       ids.append( id );
393       types.append( typeById( id, Object ) );
394       names.append( owner->IO()->getName() );
395     }
396   }*/
397
398   SALOME_ListIO selObjs;
399   TColStd_IndexedMapOfInteger selIndices;
400   selectionMgr()->selectedObjects( selObjs );
401   Selection_Mode mode = selectionMode();
402   EntityType objtype = mode == NodeSelection ? MeshNode : MeshElement;
403
404   for( SALOME_ListIteratorOfListIO anIt( selObjs ); anIt.More(); anIt.Next() )
405   {
406     selIndices.Clear();
407     selectionMgr()->GetIndexes( anIt.Value(), selIndices );
408     if( selIndices.Extent() > 0 )
409     {
410       QString id_str = QString( "%1%2%3" ).arg( anIt.Value()->getEntry() ).arg( idChar() ), current_id_str;
411       for( int i=1, n=selIndices.Extent(); i<=n; i++ )
412       {
413         int curid = selIndices( i );
414         current_id_str = id_str.arg( curid );
415         ids.append( current_id_str );
416         types.append( typeById( current_id_str, objtype ) );
417         names.append( QString( "%1" ).arg( curid ) );
418       }
419     }
420     else
421     {
422       QString id = anIt.Value()->getEntry();
423       ids.append( id );
424       types.append( typeById( id, Object ) );
425       SalomeApp_Study* _study = dynamic_cast<SalomeApp_Study*>( study() );
426       if( _study )
427       {
428         _PTR(SObject) obj = _study->studyDS()->FindObjectID( anIt.Value()->getEntry() );
429         if( obj )
430           names.append( obj->GetName().c_str() );
431       }
432     }
433   }
434 }
435
436 //=======================================================================
437 // name    : idChar
438 // Purpose : Char using to divide <entry> and <id> in string id representation. By default, '#'
439 //=======================================================================
440 QChar SMESHGUI_SelectionOp::idChar() const
441 {
442   return '#';
443 }
444
445 //=================================================================================
446 // name     : mesh
447 // purpose  :
448 //=================================================================================
449 SMESH::SMESH_Mesh_var SMESHGUI_SelectionOp::mesh() const
450 {
451   if( selectionMode()==ActorSelection )
452     return SMESH::SMESH_Mesh::_nil();
453     
454   SALOME_ListIO sel; selectionMgr()->selectedObjects( sel, SVTK_Viewer::Type() );
455   if( sel.Extent()==1 )
456     return SMESH::GetMeshByIO( sel.First() );
457   else
458     return SMESH::SMESH_Mesh::_nil();
459 }
460
461 //=================================================================================
462 // name     : actor
463 // purpose  :
464 //=================================================================================
465 SMESH_Actor* SMESHGUI_SelectionOp::actor() const
466 {
467   SMESH::SMESH_Mesh_var m = mesh();
468   if( !m->_is_nil() )
469     return SMESH::FindActorByObject( m.in() );
470   else
471     return 0;
472 }
473
474 //=================================================================================
475 // name     : onTextChanged
476 // purpose  :
477 //=================================================================================
478 void SMESHGUI_SelectionOp::onTextChanged( int, const QStringList& list )
479 {
480     if( !dlg() )
481       return;
482
483     TColStd_MapOfInteger newIndices;
484
485     SALOME_ListIO sel; selectionMgr()->selectedObjects( sel );
486     SMESH_Actor* anActor = actor();
487     if( sel.Extent()==0 || !anActor )
488       return;
489
490     SMDS_Mesh* aMesh = anActor->GetObject()->GetMesh();
491
492     IdList ids; extractIds( list, ids, '\0' );
493     IdList::const_iterator anIt = ids.begin(),
494                            aLast = ids.end();
495     if ( selectionMode() == NodeSelection )
496       for( ; anIt!=aLast; anIt++ ) {
497         if( const SMDS_MeshNode * n = aMesh->FindNode( *anIt ) )
498           newIndices.Add( n->GetID() );
499       }
500     else 
501       for( ; anIt!=aLast; anIt++ ) {
502         if( const SMDS_MeshElement* e = aMesh->FindElement( *anIt ) )
503           newIndices.Add( e->GetID() );
504       }
505
506     selector()->AddOrRemoveIndex( sel.First(), newIndices, false );
507     highlight( sel.First(), true, true );
508
509     QStringList names, _ids; LightApp_Dialog::TypesList types;
510     selected( names, types, _ids );
511     dlg()->selectObject( names, types, _ids, false );
512 }
513
514 //=================================================================================
515 // name     : selectedIds
516 // purpose  :
517 //=================================================================================
518 void SMESHGUI_SelectionOp::selectedIds( const int id, IdList& list ) const
519 {
520   if( !dlg() )
521     return;
522
523   QStringList ids; dlg()->selectedObject( id, ids );
524   extractIds( ids, list );
525 }
526
527 //=================================================================================
528 // name     : extractIds
529 // purpose  :
530 //=================================================================================
531 void SMESHGUI_SelectionOp::extractIds( const QStringList& ids, IdList& list, const QChar idchar )
532 {
533   QStringList::const_iterator anIt = ids.begin(),
534                               aLast = ids.end();
535   QString id_str;
536   for( ; anIt!=aLast; anIt++ )
537   {
538     id_str = *anIt;
539     int pos = idchar=='\0' ? -1 : id_str.indexOf( idchar );
540     int id = -1;
541     if( idchar=='\0' || pos>=0 )
542     {
543       id = id_str.mid( pos+1 ).toInt();
544       list.append( id );
545     }
546   }
547 }
548
549 //=================================================================================
550 // name     : extractIds
551 // purpose  :
552 //=================================================================================
553 void SMESHGUI_SelectionOp::extractIds( const QStringList& ids, IdList& list ) const
554 {
555   extractIds( ids, list, idChar() );
556 }