src/SMESHGUI/SMESHGUI_Selection.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 // SMESH SMESHGUI_Selection
  23 // File   : SMESHGUI_Selection.cxx
  24 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S.
  25 // SMESH includes
  26 //
  27 #include "SMESHGUI_Selection.h"
  28
  29 #include "SMESHGUI_Utils.h"
  30 #include "SMESHGUI_VTKUtils.h"
  31 #include "SMESHGUI_GEOMGenUtils.h"
  32 #include "SMESHGUI_ComputeDlg.h"
  33
  34 #include <SMESH_Type.h>
  35 #include <SMESH_Actor.h>
  36
  37 // SALOME GUI includes
  38 #include <SalomeApp_Study.h>
  39 #include <LightApp_VTKSelector.h>
  40 #include <SVTK_ViewWindow.h>
  41
  42 // IDL includes
  43 #include <SALOMEconfig.h>
  44 #include CORBA_CLIENT_HEADER(SMESH_Gen)
  45 #include CORBA_CLIENT_HEADER(SMESH_Mesh)
  46 #include CORBA_CLIENT_HEADER(SMESH_Group)
  47
  48 //=======================================================================
  49 //function : SMESHGUI_Selection
  50 //purpose  :
  51 //=======================================================================
  52 SMESHGUI_Selection::SMESHGUI_Selection()
  53 : LightApp_Selection()
  54 {
  55 }
  56
  57 //=======================================================================
  58 //function : ~SMESHGUI_Selection
  59 //purpose  :
  60 //=======================================================================
  61 SMESHGUI_Selection::~SMESHGUI_Selection()
  62 {
  63 }
  64
  65 //=======================================================================
  66 //function : init
  67 //purpose  :
  68 //=======================================================================
  69 void SMESHGUI_Selection::init( const QString& client, LightApp_SelectionMgr* mgr )
  70 {
  71   LightApp_Selection::init( client, mgr );
  72
  73   if( mgr && study() )
  74   {
  75     SalomeApp_Study* aSStudy = dynamic_cast<SalomeApp_Study*>(study());
  76     if (!aSStudy)
  77       return;
  78     _PTR(Study) aStudy = aSStudy->studyDS();
  79
  80     for( int i=0, n=count(); i<n; i++ )
  81       myTypes.append( typeName( type( entry( i ), aStudy ) ) );
  82   }
  83 }
  84
  85 //=======================================================================
  86 //function : processOwner
  87 //purpose  :
  88 //=======================================================================
  89 void SMESHGUI_Selection::processOwner( const LightApp_DataOwner* ow )
  90 {
  91   const LightApp_SVTKDataOwner* owner =
  92     dynamic_cast<const LightApp_SVTKDataOwner*> ( ow );
  93   if( owner )
  94     myActors.append( dynamic_cast<SMESH_Actor*>( owner->GetActor() ) );
  95   else
  96     myActors.append( 0 );
  97 }
  98
  99 //=======================================================================
 100 //function : parameter
 101 //purpose  :
 102 //=======================================================================
 103 QVariant SMESHGUI_Selection::parameter( const int ind, const QString& p ) const
 104 {
 105   QVariant val;
 106   if      ( p=="client" )        val = QVariant( LightApp_Selection::parameter( p ) );
 107   else if ( p=="type" )          val = QVariant( myTypes[ind] );
 108   else if ( p=="elemTypes" )     val = QVariant( elemTypes( ind ) );
 109   else if ( p=="isAutoColor" )   val = QVariant( isAutoColor( ind ) );
 110   else if ( p=="numberOfNodes" ) val = QVariant( numberOfNodes( ind ) );
 111   else if ( p=="labeledTypes" )  val = QVariant( labeledTypes( ind ) );
 112   else if ( p=="shrinkMode" )    val = QVariant( shrinkMode( ind ) );
 113   else if ( p=="entityMode" )    val = QVariant( entityMode( ind ) );
 114   else if ( p=="controlMode" )   val = QVariant( controlMode( ind ) );
 115   else if ( p=="displayMode" )   val = QVariant( displayMode( ind ) );
 116   else if ( p=="isComputable" )  val = QVariant( isComputable( ind ) );
 117   else if ( p=="isPreComputable" )  val = QVariant( isPreComputable( ind ) );
 118   else if ( p=="hasReference" )  val = QVariant( hasReference( ind ) );
 119   else if ( p=="isImported" )    val = QVariant( isImported( ind ) );
 120   else if ( p=="facesOrientationMode" ) val = QVariant( facesOrientationMode( ind ) );
 121   else if ( p=="groupType" )     val = QVariant( groupType( ind ) );
 122   else if ( p=="quadratic2DMode") val =  QVariant(quadratic2DMode(ind));
 123
 124   if( val.isValid() )
 125     return val;
 126   else
 127     return LightApp_Selection::parameter( ind, p );
 128 }
 129
 130 //=======================================================================
 131 //function : getVtkOwner
 132 //purpose  :
 133 //=======================================================================
 134
 135 SMESH_Actor* SMESHGUI_Selection::getActor( int ind ) const
 136 {
 137   if( ind >= 0 && ind < count() )
 138     return myActors.isEmpty() ? 0 : myActors.at( ind );
 139   else
 140     return 0;
 141 }
 142
 143 //=======================================================================
 144 //function : elemTypes
 145 //purpose  : may return {'Edge' 'Face' 'Volume'} at most
 146 //=======================================================================
 147
 148 QList<QVariant> SMESHGUI_Selection::elemTypes( int ind ) const
 149 {
 150   QList<QVariant> types;
 151   SMESH_Actor* actor = getActor( ind );
 152   if ( actor ) {
 153     TVisualObjPtr object = actor->GetObject();
 154     if ( object ) {
 155       if ( object->GetNbEntities( SMDSAbs_Edge )) types.append( "Edge" );
 156       if ( object->GetNbEntities( SMDSAbs_Face )) types.append( "Face" );
 157       if ( object->GetNbEntities( SMDSAbs_Volume )) types.append( "Volume" );
 158     }
 159   }
 160   return types;
 161 }
 162
 163 //=======================================================================
 164 //function : labeledTypes
 165 //purpose  : may return {'Point' 'Cell'} at most
 166 //=======================================================================
 167
 168 QList<QVariant> SMESHGUI_Selection::labeledTypes( int ind ) const
 169 {
 170   QList<QVariant> types;
 171   SMESH_Actor* actor = getActor( ind );
 172   if ( actor ) {
 173     if ( actor->GetPointsLabeled()) types.append( "Point" );
 174     if ( actor->GetCellsLabeled()) types.append( "Cell" );
 175   }
 176   return types;
 177 }
 178
 179 //=======================================================================
 180 //function : displayMode
 181 //purpose  : return SMESH_Actor::EReperesent
 182 //=======================================================================
 183
 184 QString SMESHGUI_Selection::displayMode( int ind ) const
 185 {
 186   SMESH_Actor* actor = getActor( ind );
 187   if ( actor ) {
 188     switch( actor->GetRepresentation() ) {
 189     case SMESH_Actor::eEdge:    return "eEdge";
 190     case SMESH_Actor::eSurface: return "eSurface";
 191     case SMESH_Actor::ePoint:   return "ePoint";
 192     default: break;
 193     }
 194   }
 195   return "Unknown";
 196 }
 197
 198
 199 //=======================================================================
 200 //function : quadratic2DMode
 201 //purpose  : return SMESH_Actor::EQuadratic2DRepresentation
 202 //=======================================================================
 203 QString SMESHGUI_Selection::quadratic2DMode( int ind ) const
 204 {
 205   SMESH_Actor* actor = getActor( ind );
 206   if ( actor ) {
 207     switch( actor->GetQuadratic2DRepresentation() ) {
 208     case SMESH_Actor::eLines:    return "eLines";
 209     case SMESH_Actor::eArcs: return "eArcs";
 210     default: break;
 211     }
 212   }
 213   return "Unknown";
 214 }
 215
 216 //=======================================================================
 217 //function : shrinkMode
 218 //purpose  : return either 'IsSrunk', 'IsNotShrunk' or 'IsNotShrinkable'
 219 //=======================================================================
 220
 221 QString SMESHGUI_Selection::shrinkMode( int ind ) const
 222 {
 223   SMESH_Actor* actor = getActor( ind );
 224   if ( actor && actor->IsShrunkable() ) {
 225     if ( actor->IsShrunk() )
 226       return "IsShrunk";
 227     return "IsNotShrunk";
 228   }
 229   return "IsNotShrinkable";
 230 }
 231
 232 //=======================================================================
 233 //function : entityMode
 234 //purpose  : may return {'Edge' 'Face' 'Volume'} at most
 235 //=======================================================================
 236
 237 QList<QVariant> SMESHGUI_Selection::entityMode( int ind ) const
 238 {
 239   QList<QVariant> types;
 240   SMESH_Actor* actor = getActor( ind );
 241   if ( actor ) {
 242     unsigned int aMode = actor->GetEntityMode();
 243     if ( aMode & SMESH_Actor::eVolumes) types.append( "Volume");
 244     if ( aMode & SMESH_Actor::eFaces  ) types.append( "Face"  );
 245     if ( aMode & SMESH_Actor::eEdges  ) types.append( "Edge"  );
 246   }
 247   return types;
 248 }
 249
 250 //=======================================================================
 251 //function : controlMode
 252 //purpose  : return SMESH_Actor::eControl
 253 //=======================================================================
 254
 255 QString SMESHGUI_Selection::controlMode( int ind ) const
 256 {
 257   SMESH_Actor* actor = getActor( ind );
 258   if ( actor ) {
 259     switch( actor->GetControlMode() ) {
 260     case SMESH_Actor::eLength:            return "eLength";
 261     case SMESH_Actor::eLength2D:          return "eLength2D";
 262     case SMESH_Actor::eFreeEdges:         return "eFreeEdges";
 263     case SMESH_Actor::eFreeNodes:         return "eFreeNodes";
 264     case SMESH_Actor::eFreeBorders:       return "eFreeBorders";
 265     case SMESH_Actor::eFreeFaces:         return "eFreeFaces";
 266     case SMESH_Actor::eMultiConnection:   return "eMultiConnection";
 267     case SMESH_Actor::eMultiConnection2D: return "eMultiConnection2D";
 268     case SMESH_Actor::eArea:              return "eArea";
 269     case SMESH_Actor::eVolume3D:          return "eVolume3D";
 270     case SMESH_Actor::eTaper:             return "eTaper";
 271     case SMESH_Actor::eAspectRatio:       return "eAspectRatio";
 272     case SMESH_Actor::eAspectRatio3D:     return "eAspectRatio3D";
 273     case SMESH_Actor::eMinimumAngle:      return "eMinimumAngle";
 274     case SMESH_Actor::eWarping:           return "eWarping";
 275     case SMESH_Actor::eSkew:              return "eSkew";
 276     default:;
 277     }
 278   }
 279   return "eNone";
 280 }
 281
 282 //=======================================================================
 283 //function : facesOrientationMode
 284 //purpose  :
 285 //=======================================================================
 286
 287 QString SMESHGUI_Selection::facesOrientationMode( int ind ) const
 288 {
 289   SMESH_Actor* actor = getActor( ind );
 290   if ( actor ) {
 291     if ( actor->GetFacesOriented() )
 292       return "IsOriented";
 293     return "IsNotOriented";
 294   }
 295   return "Unknown";
 296 }
 297
 298 //=======================================================================
 299 //function : isAutoColor
 300 //purpose  :
 301 //=======================================================================
 302
 303 bool SMESHGUI_Selection::isAutoColor( int ind ) const
 304 {
 305   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 306   {
 307     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 308     CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
 309
 310     if ( ! CORBA::is_nil( obj )) {
 311       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
 312       if ( ! mesh->_is_nil() )
 313         return mesh->GetAutoColor();
 314     }
 315   }
 316   return false;
 317 }
 318
 319 //=======================================================================
 320 //function : numberOfNodes
 321 //purpose  :
 322 //=======================================================================
 323
 324 int SMESHGUI_Selection::numberOfNodes( int ind ) const
 325 {
 326   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 327   {
 328     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 329     CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
 330
 331     if ( ! CORBA::is_nil( obj )) {
 332       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
 333       if ( ! mesh->_is_nil() )
 334         return mesh->NbNodes();
 335       SMESH::SMESH_subMesh_var aSubMeshObj = SMESH::SMESH_subMesh::_narrow( obj );
 336       if ( !aSubMeshObj->_is_nil() )
 337         return aSubMeshObj->GetNumberOfNodes(true);
 338       SMESH::SMESH_GroupBase_var aGroupObj = SMESH::SMESH_GroupBase::_narrow( obj );
 339       if ( !aGroupObj->_is_nil() )
 340         return aGroupObj->Size();
 341     }
 342   }
 343   return 0;
 344 }
 345
 346 //=======================================================================
 347 //function : isComputable
 348 //purpose  :
 349 //=======================================================================
 350
 351 QVariant SMESHGUI_Selection::isComputable( int ind ) const
 352 {
 353   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 354   {
 355 /*    Handle(SALOME_InteractiveObject) io =
 356       static_cast<LightApp_DataOwner*>( myDataOwners[ ind ].get() )->IO();
 357     if ( !io.IsNull() ) {
 358       SMESH::SMESH_Mesh_var mesh = SMESH::GetMeshByIO(io); // m,sm,gr->m
 359       if ( !mesh->_is_nil() ) {*/
 360         _PTR(SObject) so = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 361         //FindSObject( mesh );
 362         if ( so ) {
 363           CORBA::Object_var obj = SMESH::SObjectToObject(so, SMESH::GetActiveStudyDocument());
 364           if(!CORBA::is_nil(obj)){
 365             SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
 366             if (!mesh->_is_nil()){
 367               if(mesh->HasShapeToMesh()) {
 368                 GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
 369                 return QVariant( !shape->_is_nil() );
 370               }
 371               else
 372               {
 373                 return QVariant(!mesh->NbFaces()==0);
 374               }
 375             }
 376             else
 377             {
 378               GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
 379               return QVariant( !shape->_is_nil() );
 380             }
 381           }
 382         }
 383 //      }
 384 //    }
 385   }
 386   return QVariant( false );
 387 }
 388
 389 //=======================================================================
 390 //function : isPreComputable
 391 //purpose  :
 392 //=======================================================================
 393
 394 QVariant SMESHGUI_Selection::isPreComputable( int ind ) const
 395 {
 396   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 397   {
 398     QMap<int,int> modeMap;
 399     _PTR(SObject) pMesh = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 400     SMESHGUI_PrecomputeOp::getAssignedAlgos( pMesh, modeMap );
 401     return QVariant( modeMap.size() > 1 );
 402   }
 403   return QVariant( false );
 404 }
 405
 406 //=======================================================================
 407 //function : hasReference
 408 //purpose  :
 409 //=======================================================================
 410
 411 QVariant SMESHGUI_Selection::hasReference( int ind ) const
 412 {
 413   return QVariant( isReference( ind ) );
 414 }
 415
 416 //=======================================================================
 417 //function : isVisible
 418 //purpose  :
 419 //=======================================================================
 420
 421 QVariant SMESHGUI_Selection::isVisible( int ind ) const
 422 {
 423   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 424   {
 425     QString ent = entry( ind );
 426     SMESH_Actor* actor = SMESH::FindActorByEntry( ent.toLatin1().data() );
 427     if ( actor && actor->hasIO() ) {
 428       if(SVTK_ViewWindow* aViewWindow = SMESH::GetCurrentVtkView())
 429         return QVariant( aViewWindow->isVisible( actor->getIO() ) );
 430     }
 431   }
 432   return QVariant( false );
 433 }
 434
 435 //=======================================================================
 436 //function : type
 437 //purpose  :
 438 //=======================================================================
 439
 440 int SMESHGUI_Selection::type( const QString& entry, _PTR(Study) study )
 441 {
 442   _PTR(SObject) obj (study->FindObjectID(entry.toLatin1().data()));
 443   if( !obj )
 444     return -1;
 445
 446   _PTR(SObject) ref;
 447   if( obj->ReferencedObject( ref ) )
 448     obj = ref;
 449
 450   _PTR(SObject) objFather = obj->GetFather();
 451   _PTR(SComponent) objComponent = obj->GetFatherComponent();
 452
 453   if( objComponent->ComponentDataType()!="SMESH" )
 454     return -1;
 455
 456   if( objComponent->GetIOR()==obj->GetIOR() )
 457     return COMPONENT;
 458
 459   int aLevel = obj->Depth() - objComponent->Depth(),
 460       aFTag = objFather->Tag(),
 461       anOTag = obj->Tag(),
 462       res = -1;
 463
 464   switch (aLevel)
 465   {
 466   case 1:
 467     if (anOTag >= SMESH::Tag_FirstMeshRoot)
 468       res = MESH;
 469     break;
 470   case 2:
 471     switch (aFTag)
 472     {
 473     case SMESH::Tag_HypothesisRoot:
 474       res = HYPOTHESIS;
 475       break;
 476     case SMESH::Tag_AlgorithmsRoot:
 477       res = ALGORITHM;
 478       break;
 479     }
 480     break;
 481   case 3:
 482     switch (aFTag)
 483     {
 484     case SMESH::Tag_SubMeshOnVertex:
 485       res = SUBMESH_VERTEX;
 486       break;
 487     case SMESH::Tag_SubMeshOnEdge:
 488       res = SUBMESH_EDGE;
 489       break;
 490     case SMESH::Tag_SubMeshOnFace:
 491       res = SUBMESH_FACE;
 492       break;
 493     case SMESH::Tag_SubMeshOnSolid:
 494       res = SUBMESH_SOLID;
 495       break;
 496     case SMESH::Tag_SubMeshOnCompound:
 497       res = SUBMESH_COMPOUND;
 498       break;
 499     default:
 500       if (aFTag >= SMESH::Tag_FirstGroup)
 501         res = GROUP;
 502       else
 503         res = SUBMESH;
 504     }
 505     break;
 506   }
 507
 508   return res;
 509 }
 510
 511 //=======================================================================
 512 //function : typeName
 513 //purpose  :
 514 //=======================================================================
 515
 516 QString SMESHGUI_Selection::typeName( const int t )
 517 {
 518   switch( t )
 519   {
 520   case HYPOTHESIS:
 521     return "Hypothesis";
 522   case ALGORITHM:
 523     return "Algorithm";
 524   case MESH:
 525     return "Mesh";
 526   case SUBMESH:
 527     return "SubMesh";
 528   case MESHorSUBMESH:
 529     return "Mesh or submesh";
 530   case SUBMESH_VERTEX:
 531     return "Mesh vertex";
 532   case SUBMESH_EDGE:
 533     return "Mesh edge";
 534   case SUBMESH_FACE:
 535     return "Mesh face";
 536   case SUBMESH_SOLID:
 537     return "Mesh solid";
 538   case SUBMESH_COMPOUND:
 539     return "Mesh compound";
 540   case GROUP:
 541     return "Group";
 542   case COMPONENT:
 543     return "Component";
 544   default:
 545     return "Unknown";
 546   }
 547 }
 548
 549 bool SMESHGUI_Selection::isImported( const int ind ) const
 550 {
 551   QString e = entry( ind );
 552   _PTR(SObject) SO = SMESH::GetActiveStudyDocument()->FindObjectID( e.toLatin1().constData() );
 553   bool res = false;
 554   if( SO )
 555   {
 556     SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow( SMESH::SObjectToObject( SO ) );
 557     if( !aMesh->_is_nil() )
 558     {
 559       SALOME_MED::MedFileInfo* inf = aMesh->GetMEDFileInfo();
 560       res = strlen( (char*)inf->fileName ) > 0;
 561     }
 562   }
 563   return res;
 564 }
 565
 566 //=======================================================================
 567 //function : groupType
 568 //purpose  :
 569 //=======================================================================
 570
 571 QString SMESHGUI_Selection::groupType( int ind ) const
 572 {
 573   QString e = entry( ind );
 574   _PTR(SObject) SO = SMESH::GetActiveStudyDocument()->FindObjectID( e.toLatin1().constData() );
 575   QString type;
 576   if( SO )
 577   {
 578     CORBA::Object_var obj = SMESH::SObjectToObject( SO );
 579
 580     SMESH::SMESH_Group_var aGroup = SMESH::SMESH_Group::_narrow( obj );
 581     SMESH::SMESH_GroupOnGeom_var aGroupOnGeom = SMESH::SMESH_GroupOnGeom::_narrow( obj );
 582     if( !aGroup->_is_nil() )
 583       type = QString( "Group" );
 584     else if ( !aGroupOnGeom->_is_nil() )
 585       type = QString( "GroupOnGeom" );
 586   }
 587   return type;
 588 }
 589