src/SMESHGUI/SMESHGUI_SelectionOp.cxx

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