src/SMESHGUI/SMESHGUI_Selection.cxx

   1 // Copyright (C) 2007-2016  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 // SMESH SMESHGUI_Selection
  24 // File   : SMESHGUI_Selection.cxx
  25 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S.
  26 //
  27
  28 // SMESH includes
  29 #include "SMESHGUI_Selection.h"
  30
  31 #include "SMESHGUI.h"
  32 #include "SMESHGUI_ComputeDlg.h"
  33 #include "SMESHGUI_ConvToQuadOp.h"
  34 #include "SMESHGUI_GEOMGenUtils.h"
  35 #include "SMESHGUI_Utils.h"
  36 #include "SMESHGUI_VTKUtils.h"
  37
  38 #include <SMESH_Type.h>
  39 #include <SMESH_Actor.h>
  40 #include <SMESH_ScalarBarActor.h>
  41
  42 // SALOME GUI includes
  43 #include <SalomeApp_Study.h>
  44 #include <LightApp_VTKSelector.h>
  45 #include <SVTK_ViewWindow.h>
  46
  47 // IDL includes
  48 #include <SALOMEconfig.h>
  49 #include CORBA_CLIENT_HEADER(SMESH_Gen)
  50 #include CORBA_CLIENT_HEADER(SMESH_Mesh)
  51 #include CORBA_CLIENT_HEADER(SMESH_Group)
  52
  53 //=======================================================================
  54 //function : SMESHGUI_Selection
  55 //purpose  :
  56 //=======================================================================
  57 SMESHGUI_Selection::SMESHGUI_Selection()
  58 : LightApp_Selection()
  59 {
  60 }
  61
  62 //=======================================================================
  63 //function : ~SMESHGUI_Selection
  64 //purpose  :
  65 //=======================================================================
  66 SMESHGUI_Selection::~SMESHGUI_Selection()
  67 {
  68 }
  69
  70 //=======================================================================
  71 //function : init
  72 //purpose  :
  73 //=======================================================================
  74 void SMESHGUI_Selection::init( const QString& client, LightApp_SelectionMgr* mgr )
  75 {
  76   LightApp_Selection::init( client, mgr );
  77
  78   if( mgr && study() )
  79   {
  80     SalomeApp_Study* aSStudy = dynamic_cast<SalomeApp_Study*>(study());
  81     if (!aSStudy)
  82       return;
  83     _PTR(Study) aStudy = aSStudy->studyDS();
  84
  85     for( int i=0, n=count(); i<n; i++ ) {
  86       myTypes.append( typeName( type( entry( i ), aStudy ) ) );
  87       myControls.append( controlMode( i ) );
  88     }
  89   }
  90 }
  91
  92 //=======================================================================
  93 //function : processOwner
  94 //purpose  :
  95 //=======================================================================
  96 bool SMESHGUI_Selection::processOwner( const LightApp_DataOwner* ow )
  97 {
  98   const LightApp_SVTKDataOwner* owner =
  99     dynamic_cast<const LightApp_SVTKDataOwner*> ( ow );
 100   if( owner ) {
 101     myActors.append( dynamic_cast<SMESH_Actor*>( owner->GetActor() ) );
 102   }
 103   else if ( ow ) { // SVTK selection disabled
 104     QString entry = ow->entry();
 105     myActors.append( SMESH::FindActorByEntry( entry.toStdString().c_str() ));
 106   }
 107   else {
 108     myActors.append( 0 );
 109   }
 110   return true;
 111 }
 112
 113 //=======================================================================
 114 //function : parameter
 115 //purpose  :
 116 //=======================================================================
 117 QVariant SMESHGUI_Selection::parameter( const int ind, const QString& p ) const
 118 {
 119   QVariant val;
 120   if      ( p=="client" )               val = QVariant( LightApp_Selection::parameter( p ) );
 121   else if ( p=="type" )                 val = QVariant( myTypes[ind] );
 122   else if ( p=="hasActor" )             val = QVariant( getActor( ind ) != 0 );
 123   else if ( p=="elemTypes" )            val = QVariant( elemTypes( ind, false ) );
 124   else if ( p=="objElemTypes" )         val = QVariant( elemTypes( ind, true ) );
 125   else if ( p=="isAutoColor" )          val = QVariant( isAutoColor( ind ) );
 126   else if ( p=="numberOfNodes" )        val = QVariant( numberOfNodes( ind ) );
 127   else if ( p=="dim" )                  val = QVariant( dim( ind ) );
 128   else if ( p=="labeledTypes" )         val = QVariant( labeledTypes( ind ) );
 129   else if ( p=="shrinkMode" )           val = QVariant( shrinkMode( ind ) );
 130   else if ( p=="entityMode" )           val = QVariant( entityMode( ind ) );
 131   else if ( p=="isNumFunctor" )         val = QVariant( isNumFunctor( ind ) );
 132   else if ( p=="displayMode" )          val = QVariant( displayMode( ind ) );
 133   else if ( p=="isComputable" )         val = QVariant( isComputable( ind ) );
 134   else if ( p=="isPreComputable" )      val = QVariant( isPreComputable( ind ) );
 135   else if ( p=="hasGeomReference" )     val = QVariant( hasGeomReference( ind ) );
 136   else if ( p=="isEditableHyp" )        val = QVariant( isEditableHyp( ind ) );
 137   else if ( p=="isImported" )           val = QVariant( isImported( ind ) );
 138   else if ( p=="facesOrientationMode" ) val = QVariant( facesOrientationMode( ind ) );
 139   else if ( p=="groupType" )            val = QVariant( groupType( ind ) );
 140   else if ( p=="isQuadratic" )          val = QVariant( isQuadratic( ind ) );
 141   else if ( p=="quadratic2DMode")       val = QVariant( quadratic2DMode( ind ) );
 142   else if ( p=="isDistributionVisible") val = QVariant( isDistributionVisible( ind ) );
 143   else if ( p=="isScalarBarVisible")    val = QVariant( isScalarBarVisible( ind ) );
 144   else if ( p=="hasChildren")           val = QVariant( hasChildren( ind ) );
 145   else if ( p=="nbChildren")            val = QVariant( nbChildren( ind ) );
 146   else if ( p=="isContainer")           val = QVariant( isContainer( ind ) );
 147   else if ( p=="guiState")              val = QVariant( guiState() );
 148
 149   if ( val.isValid() )
 150     return val;
 151   else
 152     return LightApp_Selection::parameter( ind, p );
 153 }
 154
 155 //=======================================================================
 156 //function : parameter
 157 //purpose  :
 158 //=======================================================================
 159 QVariant SMESHGUI_Selection::parameter( const QString& p ) const
 160 {
 161   QVariant val;
 162   if ( p=="controlMode" ) val = QVariant( controlMode() );
 163
 164   if ( val.isValid() )
 165     return val;
 166   else
 167     return LightApp_Selection::parameter( p );
 168 }
 169
 170 //=======================================================================
 171 //function : getVtkOwner
 172 //purpose  :
 173 //=======================================================================
 174
 175 SMESH_Actor* SMESHGUI_Selection::getActor( int ind ) const
 176 {
 177   if( ind >= 0 && ind < count() )
 178     return myActors.isEmpty() ? 0 : myActors.at( ind );
 179   else
 180     return 0;
 181 }
 182
 183 //=======================================================================
 184 //function : elemTypes
 185 //purpose  : may return {'Elem0d' 'Edge' 'Face' 'Volume' 'BallElem'} at most
 186 //=======================================================================
 187
 188 QList<QVariant> SMESHGUI_Selection::elemTypes( int ind, bool fromObj ) const
 189 {
 190   QList<QVariant> types;
 191   SMESH_Actor* actor = getActor( ind );
 192   if ( actor ) {
 193     TVisualObjPtr object = actor->GetObject();
 194     if ( object ) {
 195       if ( object->GetNbEntities( SMDSAbs_Edge ))      types.append( "Edge" );
 196       if ( object->GetNbEntities( SMDSAbs_Face ))      types.append( "Face" );
 197       if ( object->GetNbEntities( SMDSAbs_Volume ))    types.append( "Volume" );
 198       if ( object->GetNbEntities( SMDSAbs_0DElement )) types.append( "Elem0d" );
 199       if ( object->GetNbEntities( SMDSAbs_Ball ))      types.append( "BallElem" );
 200     }
 201   }
 202   else if ( fromObj )
 203   {
 204     if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 205     {
 206       _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 207       CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
 208       SMESH::SMESH_IDSource_var idSrc = SMESH::SMESH_IDSource::_narrow( obj );
 209       if ( !CORBA::is_nil( idSrc ) )
 210       {
 211         SMESH::array_of_ElementType_var typeVar = idSrc->GetTypes();
 212         for ( CORBA::ULong i = 0; i < typeVar->length(); ++i )
 213           switch ( typeVar[i] ) {
 214           case SMESH::EDGE:   types.append( "Edge" );     break;
 215           case SMESH::FACE:   types.append( "Face" );     break;
 216           case SMESH::VOLUME: types.append( "Volume" );   break;
 217           case SMESH::ELEM0D: types.append( "Elem0d" );   break;
 218           case SMESH::BALL:   types.append( "BallElem" ); break;
 219           case SMESH::ALL:
 220           case SMESH::NODE:
 221           case SMESH::NB_ELEMENT_TYPES: break;
 222           }
 223       }
 224     }
 225   }
 226   return types;
 227 }
 228
 229 //=======================================================================
 230 //function : labeledTypes
 231 //purpose  : may return {'Point' 'Cell'} at most
 232 //=======================================================================
 233
 234 QList<QVariant> SMESHGUI_Selection::labeledTypes( int ind ) const
 235 {
 236   QList<QVariant> types;
 237   SMESH_Actor* actor = getActor( ind );
 238   if ( actor ) {
 239     if ( actor->GetPointsLabeled()) types.append( "Point" );
 240     if ( actor->GetCellsLabeled())  types.append( "Cell" );
 241   }
 242   return types;
 243 }
 244
 245 //=======================================================================
 246 //function : displayMode
 247 //purpose  : return SMESH_Actor::EReperesent
 248 //=======================================================================
 249
 250 QString SMESHGUI_Selection::displayMode( int ind ) const
 251 {
 252   SMESH_Actor* actor = getActor( ind );
 253   if ( actor ) {
 254     switch( actor->GetRepresentation() ) {
 255     case SMESH_Actor::eEdge:    return "eEdge";
 256     case SMESH_Actor::eSurface: return "eSurface";
 257     case SMESH_Actor::ePoint:   return "ePoint";
 258     default: break;
 259     }
 260   }
 261   return "Unknown";
 262 }
 263
 264 //=======================================================================
 265 //function : isQuadratic
 266 //purpose  : return true if the mesh has quadratic/bi-quadratic type
 267 //=======================================================================
 268
 269 bool SMESHGUI_Selection::isQuadratic( int ind ) const
 270 {
 271   _PTR(SObject) so = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 272   if ( !so )
 273     return false;
 274   SMESH::SMESH_IDSource_var idSource =  SMESH::SObjectToInterface<SMESH::SMESH_IDSource>( so );
 275   if ( idSource->_is_nil() )
 276     return false;
 277   SMESHGUI_ConvToQuadOp::MeshDestinationType meshTgtType = SMESHGUI_ConvToQuadOp::DestinationMesh( idSource );
 278   if ( meshTgtType & SMESHGUI_ConvToQuadOp::Linear )
 279     return true;
 280   return false;
 281 }
 282
 283 //=======================================================================
 284 //function : quadratic2DMode
 285 //purpose  : return SMESH_Actor::EQuadratic2DRepresentation
 286 //=======================================================================
 287 QString SMESHGUI_Selection::quadratic2DMode( int ind ) const
 288 {
 289   SMESH_Actor* actor = getActor( ind );
 290   if ( actor ) {
 291     switch( actor->GetQuadratic2DRepresentation() ) {
 292     case SMESH_Actor::eLines: return "eLines";
 293     case SMESH_Actor::eArcs:  return "eArcs";
 294     default: break;
 295     }
 296   }
 297   return "Unknown";
 298 }
 299
 300 //=======================================================================
 301 //function : isDistributionVisible
 302 //purpose  : Visible/Invisible distribution of the ScalarBar Actor
 303 //=======================================================================
 304
 305 bool SMESHGUI_Selection::isDistributionVisible(int ind) const {
 306   SMESH_Actor* actor = getActor( ind );
 307   return (actor && actor->GetScalarBarActor() && actor->GetScalarBarActor()->GetDistributionVisibility());
 308 }
 309
 310 //=======================================================================
 311 //function : isScalarBarVisible
 312 //purpose  : Visible/Invisible Scalar Bar
 313 //=======================================================================
 314
 315 bool SMESHGUI_Selection::isScalarBarVisible(int ind) const {
 316   SMESH_Actor* actor = getActor( ind );
 317   return (actor && actor->GetScalarBarActor() && actor->GetScalarBarActor()->GetVisibility());
 318 }
 319
 320 //=======================================================================
 321 //function : shrinkMode
 322 //purpose  : return either 'IsSrunk', 'IsNotShrunk' or 'IsNotShrinkable'
 323 //=======================================================================
 324
 325 QString SMESHGUI_Selection::shrinkMode( int ind ) const
 326 {
 327   SMESH_Actor* actor = getActor( ind );
 328   if ( actor && actor->IsShrunkable() ) {
 329     return actor->IsShrunk() ? "IsShrunk" : "IsNotShrunk";
 330   }
 331   return "IsNotShrinkable";
 332 }
 333
 334 //=======================================================================
 335 //function : entityMode
 336 //purpose  : may return {'Elem0d' 'Edge' 'Face' 'Volume' 'BallElem' } at most
 337 //=======================================================================
 338
 339 QList<QVariant> SMESHGUI_Selection::entityMode( int ind ) const
 340 {
 341   QList<QVariant> types;
 342   SMESH_Actor* actor = getActor( ind );
 343   if ( actor ) {
 344     unsigned int aMode = actor->GetEntityMode();
 345     if ( aMode & SMESH_Actor::eVolumes    ) types.append( "Volume" );
 346     if ( aMode & SMESH_Actor::eFaces      ) types.append( "Face"   );
 347     if ( aMode & SMESH_Actor::eEdges      ) types.append( "Edge"   );
 348     if ( aMode & SMESH_Actor::e0DElements ) types.append( "Elem0d" );
 349     if ( aMode & SMESH_Actor::eBallElem )   types.append( "BallElem" );
 350   }
 351   return types;
 352 }
 353
 354 //=======================================================================
 355 //function : controlMode
 356 //purpose  : return SMESH_Actor::eControl
 357 //=======================================================================
 358
 359 QString SMESHGUI_Selection::controlMode( int ind ) const
 360 {
 361   SMESH_Actor* actor = getActor( ind );
 362   QString mode = "eNone";
 363   if ( actor ) {
 364     switch( actor->GetControlMode() ) {
 365     case SMESH_Actor::eLength:                mode = "eLength";                break;
 366     case SMESH_Actor::eLength2D:              mode = "eLength2D";              break;
 367     case SMESH_Actor::eFreeEdges:             mode = "eFreeEdges";             break;
 368     case SMESH_Actor::eFreeNodes:             mode = "eFreeNodes";             break;
 369     case SMESH_Actor::eFreeBorders:           mode = "eFreeBorders";           break;
 370     case SMESH_Actor::eFreeFaces:             mode = "eFreeFaces";             break;
 371     case SMESH_Actor::eMultiConnection:       mode = "eMultiConnection";       break;
 372     case SMESH_Actor::eMultiConnection2D:     mode = "eMultiConnection2D";     break;
 373     case SMESH_Actor::eArea:                  mode = "eArea";                  break;
 374     case SMESH_Actor::eVolume3D:              mode = "eVolume3D";              break;
 375     case SMESH_Actor::eMaxElementLength2D:    mode = "eMaxElementLength2D";    break;
 376     case SMESH_Actor::eMaxElementLength3D:    mode = "eMaxElementLength3D";    break;
 377     case SMESH_Actor::eTaper:                 mode = "eTaper";                 break;
 378     case SMESH_Actor::eAspectRatio:           mode = "eAspectRatio";           break;
 379     case SMESH_Actor::eAspectRatio3D:         mode = "eAspectRatio3D";         break;
 380     case SMESH_Actor::eMinimumAngle:          mode = "eMinimumAngle";          break;
 381     case SMESH_Actor::eWarping:               mode = "eWarping";               break;
 382     case SMESH_Actor::eSkew:                  mode = "eSkew";                  break;
 383     case SMESH_Actor::eBareBorderFace:        mode = "eBareBorderFace";        break;
 384     case SMESH_Actor::eBareBorderVolume:      mode = "eBareBorderVolume";      break;
 385     case SMESH_Actor::eOverConstrainedFace:   mode = "eOverConstrainedFace";   break;
 386     case SMESH_Actor::eOverConstrainedVolume: mode = "eOverConstrainedVolume"; break;
 387     case SMESH_Actor::eCoincidentNodes:       mode = "eCoincidentNodes";       break;
 388     case SMESH_Actor::eCoincidentElems1D:     mode = "eCoincidentElems1D";     break;
 389     case SMESH_Actor::eCoincidentElems2D:     mode = "eCoincidentElems2D";     break;
 390     case SMESH_Actor::eCoincidentElems3D:     mode = "eCoincidentElems3D";     break;
 391     default:break;
 392     }
 393   }
 394   return mode;
 395 }
 396
 397 //=======================================================================
 398 //function : controlMode
 399 //purpose  : gets global control mode; return SMESH_Actor::eControl
 400 //=======================================================================
 401 QString SMESHGUI_Selection::controlMode() const
 402 {
 403   if( myControls.count() > 0 ) {
 404     QString mode = myControls[0];
 405     for( int ind = 1; ind < myControls.count(); ind++ ) {
 406       if( mode != myControls[ind] )
 407         return "eMixed"; // different controls used for different actors
 408     }
 409     return mode;
 410   }
 411   return "eNone";
 412 }
 413
 414 bool SMESHGUI_Selection::isNumFunctor( int ind ) const
 415 {
 416   bool result = false;
 417   SMESH_Actor* actor = getActor( ind );
 418   if ( actor ) {
 419     switch( actor->GetControlMode() ) {
 420     case SMESH_Actor::eLength:
 421     case SMESH_Actor::eLength2D:
 422     case SMESH_Actor::eMultiConnection:
 423     case SMESH_Actor::eMultiConnection2D:
 424     case SMESH_Actor::eArea:
 425     case SMESH_Actor::eVolume3D:
 426     case SMESH_Actor::eMaxElementLength2D:
 427     case SMESH_Actor::eMaxElementLength3D:
 428     case SMESH_Actor::eTaper:
 429     case SMESH_Actor::eAspectRatio:
 430     case SMESH_Actor::eAspectRatio3D:
 431     case SMESH_Actor::eMinimumAngle:
 432     case SMESH_Actor::eWarping:
 433     case SMESH_Actor::eSkew:
 434       result = true;
 435       break;
 436     default:
 437       break;
 438     }
 439   }
 440   return result;
 441 }
 442
 443 //=======================================================================
 444 //function : facesOrientationMode
 445 //purpose  :
 446 //=======================================================================
 447
 448 QString SMESHGUI_Selection::facesOrientationMode( int ind ) const
 449 {
 450   SMESH_Actor* actor = getActor( ind );
 451   if ( actor ) {
 452     return actor->GetFacesOriented() ? "IsOriented" : "IsNotOriented";
 453   }
 454   return "Unknown";
 455 }
 456
 457 //=======================================================================
 458 //function : isAutoColor
 459 //purpose  :
 460 //=======================================================================
 461
 462 bool SMESHGUI_Selection::isAutoColor( int ind ) const
 463 {
 464   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 465   {
 466     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 467     CORBA::Object_var      obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
 468     SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
 469     if ( !CORBA::is_nil( mesh ) )
 470       return mesh->GetAutoColor();
 471   }
 472   return false;
 473 }
 474
 475 //=======================================================================
 476 //function : numberOfNodes
 477 //purpose  : this method is actually used to check if an object is empty or not
 478 //=======================================================================
 479
 480 int SMESHGUI_Selection::numberOfNodes( int ind ) const
 481 {
 482   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 483   {
 484     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 485     CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
 486
 487     if ( !CORBA::is_nil( obj ) ) {
 488       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
 489       if ( !CORBA::is_nil( mesh ) )
 490         return mesh->NbNodes();
 491       SMESH::SMESH_subMesh_var aSubMeshObj = SMESH::SMESH_subMesh::_narrow( obj );
 492       if ( !CORBA::is_nil( aSubMeshObj ) )
 493         return aSubMeshObj->GetNumberOfNodes(true);
 494       SMESH::SMESH_GroupBase_var aGroupObj = SMESH::SMESH_GroupBase::_narrow( obj );
 495       if ( !CORBA::is_nil( aGroupObj ) )
 496         return aGroupObj->IsEmpty() ? 0 : 1; // aGroupObj->Size();
 497     }
 498   }
 499   return 0;
 500 }
 501
 502 //================================================================================
 503 /*!
 504  * \brief return dimension of elements of the selected object
 505  *
 506  *  \retval int - 0 for 0D elements, -1 for an empty object (the rest as usual)
 507  */
 508 //================================================================================
 509
 510 int SMESHGUI_Selection::dim( int ind ) const
 511 {
 512   int dim = -1;
 513   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 514   {
 515     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 516     CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
 517
 518     if ( !CORBA::is_nil( obj ) ) {
 519       SMESH::SMESH_IDSource_var idSrc = SMESH::SMESH_IDSource::_narrow( obj );
 520       if ( !CORBA::is_nil( idSrc ) )
 521       {
 522         SMESH::array_of_ElementType_var types = idSrc->GetTypes();
 523         for ( size_t i = 0; i < types->length(); ++ i) {
 524           switch ( types[i] ) {
 525           case SMESH::EDGE  : dim = std::max( dim, 1 ); break;
 526           case SMESH::FACE  : dim = std::max( dim, 2 ); break;
 527           case SMESH::VOLUME: dim = std::max( dim, 3 ); break;
 528           case SMESH::ELEM0D: dim = std::max( dim, 0 ); break;
 529           case SMESH::BALL  : dim = std::max( dim, 0 ); break;
 530           default: break;
 531           }
 532         }
 533       }
 534     }
 535   }
 536   return dim;
 537 }
 538
 539 //=======================================================================
 540 //function : isComputable
 541 //purpose  : return true for a ready-to-compute mesh
 542 //=======================================================================
 543
 544 bool SMESHGUI_Selection::isComputable( int ind ) const
 545 {
 546   if ( ind >= 0 && ind < myTypes.count() && ( myTypes[ind] == "Mesh" ||
 547                                               myTypes[ind].startsWith("Mesh " )))
 548   {
 549     QMap<int,int> modeMap;
 550     _PTR(SObject) meshSO = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 551
 552     _PTR(SComponent) component = meshSO->GetFatherComponent();
 553     if ( meshSO->Depth() - component->Depth() > 1 ) // sub-mesh, get a mesh
 554       while ( meshSO->Depth() - component->Depth() > 1 )
 555         meshSO = meshSO->GetFather();
 556
 557     SMESHGUI_PrecomputeOp::getAssignedAlgos( meshSO, modeMap );
 558     return modeMap.size() > 0;
 559   }
 560   return false;
 561 }
 562
 563 //=======================================================================
 564 //function : isPreComputable
 565 //purpose  : returns true for a mesh with algorithms
 566 //=======================================================================
 567
 568 bool SMESHGUI_Selection::isPreComputable( int ind ) const
 569 {
 570   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] == "Mesh" )
 571   {
 572     int maxDim = dim( ind );
 573     if ( maxDim < 2 ) // we can preview 1D or 2D
 574     {
 575       QMap<int,int> modeMap;
 576       _PTR(SObject) pMesh = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 577       SMESHGUI_PrecomputeOp::getAssignedAlgos( pMesh, modeMap );
 578       if ( modeMap.size() > 1 )
 579         return (( modeMap.contains( SMESH::DIM_3D )) ||
 580                 ( modeMap.contains( SMESH::DIM_2D ) && maxDim < 1 ));
 581     }
 582   }
 583   return false;
 584 }
 585
 586 //=======================================================================
 587 //function : hasGeomReference
 588 //purpose  : returns true for a mesh or sub-mesh on geometry
 589 //=======================================================================
 590
 591 bool SMESHGUI_Selection::hasGeomReference( int ind ) const
 592 {
 593   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 594   {
 595     _PTR(SObject) so = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 596     GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
 597     return !shape->_is_nil();
 598   }
 599   return false;
 600 }
 601
 602 //=======================================================================
 603 //function : isEditableHyp
 604 //purpose  :
 605 //=======================================================================
 606
 607 bool SMESHGUI_Selection::isEditableHyp( int ind ) const
 608 {
 609   bool isEditable = true;
 610   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] == "Hypothesis" )
 611   {
 612     _PTR(SObject) so = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 613     SMESH::SMESH_Hypothesis_var hyp = SMESH::SObjectToInterface<SMESH::SMESH_Hypothesis>( so );
 614     if ( !hyp->_is_nil() )
 615     {
 616       isEditable = hyp->HasParameters();
 617     }
 618   }
 619   return isEditable;
 620 }
 621
 622 //=======================================================================
 623 //function : isVisible
 624 //purpose  :
 625 //=======================================================================
 626
 627 bool SMESHGUI_Selection::isVisible( int ind ) const
 628 {
 629   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
 630   {
 631     SMESH_Actor* actor = SMESH::FindActorByEntry( entry( ind ).toLatin1().data() );
 632     if ( actor && actor->hasIO() ) {
 633       if ( SVTK_ViewWindow* aViewWindow = SMESH::GetCurrentVtkView() )
 634         return aViewWindow->isVisible( actor->getIO() );
 635     }
 636   }
 637   return false;
 638 }
 639
 640 //=======================================================================
 641 //function : hasChildren
 642 //purpose  :
 643 //=======================================================================
 644
 645 bool SMESHGUI_Selection::hasChildren( int ind ) const
 646 {
 647   if ( ind >= 0 )
 648   {
 649     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 650     if ( sobj )
 651       return SMESH::GetActiveStudyDocument()->GetUseCaseBuilder()->HasChildren( sobj );
 652   }
 653   return false;
 654 }
 655
 656 //=======================================================================
 657 //function : hasChildren
 658 //purpose  :
 659 //=======================================================================
 660
 661 int SMESHGUI_Selection::nbChildren( int ind ) const
 662 {
 663   int nb = 0;
 664   if ( ind >= 0 )
 665   {
 666     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
 667     if ( sobj && sobj->GetStudy()->GetUseCaseBuilder()->IsUseCaseNode( sobj ) ) {
 668       _PTR(UseCaseIterator) it = sobj->GetStudy()->GetUseCaseBuilder()->GetUseCaseIterator( sobj );
 669       for ( it->Init( false ); it->More(); it->Next() ) nb++;
 670     }
 671   }
 672   return nb;
 673 }
 674
 675 //=======================================================================
 676 //function : isContainer
 677 //purpose  :
 678 //=======================================================================
 679
 680 bool SMESHGUI_Selection::isContainer( int ind ) const
 681 {
 682   return ind >= 0 && ind < myTypes.count() && myTypes[ind] == "Unknown";
 683 }
 684
 685 //=======================================================================
 686 //function : type
 687 //purpose  :
 688 //=======================================================================
 689
 690 int SMESHGUI_Selection::type( const QString& entry, _PTR(Study) study )
 691 {
 692   int res = -1;
 693   _PTR(SObject) obj = study->FindObjectID( entry.toLatin1().data() );
 694   if ( obj ) {
 695     _PTR(SObject) ref;
 696     if ( obj->ReferencedObject( ref ) )
 697       obj = ref;
 698
 699     _PTR(SObject) objFather = obj->GetFather();
 700     _PTR(SComponent) objComponent = obj->GetFatherComponent();
 701
 702     if ( objComponent->ComponentDataType() == "SMESH" ) {
 703       if ( objComponent->GetIOR() == obj->GetIOR() ) {
 704         res = SMESH::COMPONENT;
 705       }
 706       else {
 707         int aLevel = obj->Depth() - objComponent->Depth(),
 708           aFTag = objFather->Tag(),
 709           anOTag = obj->Tag();
 710
 711         switch ( aLevel )
 712         {
 713         case 1:
 714           if ( anOTag >= SMESH::Tag_FirstMeshRoot )
 715             res = SMESH::MESH;
 716           break;
 717         case 2:
 718           switch ( aFTag )
 719           {
 720           case SMESH::Tag_HypothesisRoot: res = SMESH::HYPOTHESIS; break;
 721           case SMESH::Tag_AlgorithmsRoot: res = SMESH::ALGORITHM;  break;
 722           default: break;
 723           }
 724           break;
 725         case 3:
 726           switch ( aFTag )
 727           {
 728           case SMESH::Tag_SubMeshOnVertex:   res = SMESH::SUBMESH_VERTEX;   break;
 729           case SMESH::Tag_SubMeshOnEdge:     res = SMESH::SUBMESH_EDGE;     break;
 730           case SMESH::Tag_SubMeshOnFace:     res = SMESH::SUBMESH_FACE;     break;
 731           case SMESH::Tag_SubMeshOnSolid:    res = SMESH::SUBMESH_SOLID;    break;
 732           case SMESH::Tag_SubMeshOnCompound: res = SMESH::SUBMESH_COMPOUND; break;
 733           default:
 734             if ( aFTag >= SMESH::Tag_FirstGroup) res = SMESH::GROUP;
 735             else                                 res = SMESH::SUBMESH;
 736             break;
 737           }
 738           break;
 739         }
 740       }
 741     }
 742   }
 743   return res;
 744 }
 745
 746 //=======================================================================
 747 //function : typeName
 748 //purpose  :
 749 //=======================================================================
 750
 751 QString SMESHGUI_Selection::typeName( const int t )
 752 {
 753   QString res = "Unknown";
 754   switch( t )
 755   {
 756   case SMESH::HYPOTHESIS:
 757     res = "Hypothesis"; break;
 758   case SMESH::ALGORITHM:
 759     res = "Algorithm"; break;
 760   case SMESH::MESH:
 761     res = "Mesh"; break;
 762   case SMESH::SUBMESH:
 763     res = "SubMesh"; break;
 764   case SMESH::MESHorSUBMESH:
 765     res = "Mesh or submesh"; break;
 766   case SMESH::SUBMESH_VERTEX:
 767     res = "Mesh vertex"; break;
 768   case SMESH::SUBMESH_EDGE:
 769     res = "Mesh edge"; break;
 770   case SMESH::SUBMESH_FACE:
 771     res = "Mesh face"; break;
 772   case SMESH::SUBMESH_SOLID:
 773     res = "Mesh solid"; break;
 774   case SMESH::SUBMESH_COMPOUND:
 775     res = "Mesh compound"; break;
 776   case SMESH::GROUP:
 777     res = "Group"; break;
 778   case SMESH::COMPONENT:
 779     res = "Component"; break;
 780   default:
 781      break;
 782   }
 783   return res;
 784 }
 785
 786 bool SMESHGUI_Selection::isImported( const int ind ) const
 787 {
 788   bool res = false;
 789   _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().constData() );
 790   if ( sobj )
 791   {
 792     SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow( SMESH::SObjectToObject( sobj ) );
 793     if( !aMesh->_is_nil() )
 794     {
 795       SMESH::MedFileInfo_var inf = aMesh->GetMEDFileInfo();
 796       res = strlen( (char*)inf->fileName ) > 0;
 797     }
 798   }
 799   return res;
 800 }
 801
 802 //=======================================================================
 803 //function : guiState
 804 //purpose  :
 805 //=======================================================================
 806
 807 int SMESHGUI_Selection::guiState()
 808 {
 809   return SMESHGUI::GetSMESHGUI() ? SMESHGUI::GetSMESHGUI()->GetState() : -1;
 810 }
 811
 812 //=======================================================================
 813 //function : groupType
 814 //purpose  :
 815 //=======================================================================
 816
 817 QString SMESHGUI_Selection::groupType( int ind ) const
 818 {
 819   _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().constData() );
 820   if ( sobj )
 821   {
 822     SMESH::SMESH_Group_var g = SMESH::SObjectToInterface<SMESH::SMESH_Group>( sobj );
 823     if ( !CORBA::is_nil( g ) )
 824       return "Group";
 825     SMESH::SMESH_GroupOnGeom_var gog = SMESH::SObjectToInterface<SMESH::SMESH_GroupOnGeom>( sobj );
 826     if( !CORBA::is_nil( gog ) )
 827       return "GroupOnGeom";
 828     SMESH::SMESH_GroupOnFilter_var gof = SMESH::SObjectToInterface<SMESH::SMESH_GroupOnFilter>( sobj );
 829     if ( !CORBA::is_nil( gof ) )
 830       return "GroupOnFilter";
 831   }
 832   return "";
 833 }