1 // Copyright (C) 2007-2010 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // SMESH SMESHGUI_Selection
24 // File : SMESHGUI_Selection.cxx
25 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S.
28 #include "SMESHGUI_Selection.h"
30 #include "SMESHGUI_Utils.h"
31 #include "SMESHGUI_VTKUtils.h"
32 #include "SMESHGUI_GEOMGenUtils.h"
33 #include "SMESHGUI_ComputeDlg.h"
35 #include <SMESH_Type.h>
36 #include <SMESH_Actor.h>
38 // SALOME GUI includes
39 #include <SalomeApp_Study.h>
40 #include <LightApp_VTKSelector.h>
41 #include <SVTK_ViewWindow.h>
44 #include <SALOMEconfig.h>
45 #include CORBA_CLIENT_HEADER(SMESH_Gen)
46 #include CORBA_CLIENT_HEADER(SMESH_Mesh)
47 #include CORBA_CLIENT_HEADER(SMESH_Group)
49 //=======================================================================
50 //function : SMESHGUI_Selection
52 //=======================================================================
53 SMESHGUI_Selection::SMESHGUI_Selection()
54 : LightApp_Selection()
58 //=======================================================================
59 //function : ~SMESHGUI_Selection
61 //=======================================================================
62 SMESHGUI_Selection::~SMESHGUI_Selection()
66 //=======================================================================
69 //=======================================================================
70 void SMESHGUI_Selection::init( const QString& client, LightApp_SelectionMgr* mgr )
72 LightApp_Selection::init( client, mgr );
76 SalomeApp_Study* aSStudy = dynamic_cast<SalomeApp_Study*>(study());
79 _PTR(Study) aStudy = aSStudy->studyDS();
81 for( int i=0, n=count(); i<n; i++ )
82 myTypes.append( typeName( type( entry( i ), aStudy ) ) );
86 //=======================================================================
87 //function : processOwner
89 //=======================================================================
90 bool SMESHGUI_Selection::processOwner( const LightApp_DataOwner* ow )
92 const LightApp_SVTKDataOwner* owner =
93 dynamic_cast<const LightApp_SVTKDataOwner*> ( ow );
95 myActors.append( dynamic_cast<SMESH_Actor*>( owner->GetActor() ) );
101 //=======================================================================
102 //function : parameter
104 //=======================================================================
105 QVariant SMESHGUI_Selection::parameter( const int ind, const QString& p ) const
108 if ( p=="client" ) val = QVariant( LightApp_Selection::parameter( p ) );
109 else if ( p=="type" ) val = QVariant( myTypes[ind] );
110 else if ( p=="elemTypes" ) val = QVariant( elemTypes( ind ) );
111 else if ( p=="isAutoColor" ) val = QVariant( isAutoColor( ind ) );
112 else if ( p=="numberOfNodes" ) val = QVariant( numberOfNodes( ind ) );
113 else if ( p=="labeledTypes" ) val = QVariant( labeledTypes( ind ) );
114 else if ( p=="shrinkMode" ) val = QVariant( shrinkMode( ind ) );
115 else if ( p=="entityMode" ) val = QVariant( entityMode( ind ) );
116 else if ( p=="controlMode" ) val = QVariant( controlMode( ind ) );
117 else if ( p=="isNumFunctor" ) val = QVariant( isNumFunctor( ind ) );
118 else if ( p=="displayMode" ) val = QVariant( displayMode( ind ) );
119 else if ( p=="isComputable" ) val = QVariant( isComputable( ind ) );
120 else if ( p=="isPreComputable" ) val = QVariant( isPreComputable( ind ) );
121 else if ( p=="hasReference" ) val = QVariant( hasReference( ind ) );
122 else if ( p=="isImported" ) val = QVariant( isImported( ind ) );
123 else if ( p=="facesOrientationMode" ) val = QVariant( facesOrientationMode( ind ) );
124 else if ( p=="groupType" ) val = QVariant( groupType( ind ) );
125 else if ( p=="quadratic2DMode") val = QVariant(quadratic2DMode(ind));
130 return LightApp_Selection::parameter( ind, p );
133 //=======================================================================
134 //function : getVtkOwner
136 //=======================================================================
138 SMESH_Actor* SMESHGUI_Selection::getActor( int ind ) const
140 if( ind >= 0 && ind < count() )
141 return myActors.isEmpty() ? 0 : myActors.at( ind );
146 //=======================================================================
147 //function : elemTypes
148 //purpose : may return {'Elem0d' 'Edge' 'Face' 'Volume'} at most
149 //=======================================================================
151 QList<QVariant> SMESHGUI_Selection::elemTypes( int ind ) const
153 QList<QVariant> types;
154 SMESH_Actor* actor = getActor( ind );
156 TVisualObjPtr object = actor->GetObject();
158 if ( object->GetNbEntities( SMDSAbs_0DElement )) types.append( "Elem0d" );
159 if ( object->GetNbEntities( SMDSAbs_Edge )) types.append( "Edge" );
160 if ( object->GetNbEntities( SMDSAbs_Face )) types.append( "Face" );
161 if ( object->GetNbEntities( SMDSAbs_Volume )) types.append( "Volume" );
167 //=======================================================================
168 //function : labeledTypes
169 //purpose : may return {'Point' 'Cell'} at most
170 //=======================================================================
172 QList<QVariant> SMESHGUI_Selection::labeledTypes( int ind ) const
174 QList<QVariant> types;
175 SMESH_Actor* actor = getActor( ind );
177 if ( actor->GetPointsLabeled()) types.append( "Point" );
178 if ( actor->GetCellsLabeled()) types.append( "Cell" );
183 //=======================================================================
184 //function : displayMode
185 //purpose : return SMESH_Actor::EReperesent
186 //=======================================================================
188 QString SMESHGUI_Selection::displayMode( int ind ) const
190 SMESH_Actor* actor = getActor( ind );
192 switch( actor->GetRepresentation() ) {
193 case SMESH_Actor::eEdge: return "eEdge";
194 case SMESH_Actor::eSurface: return "eSurface";
195 case SMESH_Actor::ePoint: return "ePoint";
203 //=======================================================================
204 //function : quadratic2DMode
205 //purpose : return SMESH_Actor::EQuadratic2DRepresentation
206 //=======================================================================
207 QString SMESHGUI_Selection::quadratic2DMode( int ind ) const
209 SMESH_Actor* actor = getActor( ind );
211 switch( actor->GetQuadratic2DRepresentation() ) {
212 case SMESH_Actor::eLines: return "eLines";
213 case SMESH_Actor::eArcs: return "eArcs";
220 //=======================================================================
221 //function : shrinkMode
222 //purpose : return either 'IsSrunk', 'IsNotShrunk' or 'IsNotShrinkable'
223 //=======================================================================
225 QString SMESHGUI_Selection::shrinkMode( int ind ) const
227 SMESH_Actor* actor = getActor( ind );
228 if ( actor && actor->IsShrunkable() ) {
229 if ( actor->IsShrunk() )
231 return "IsNotShrunk";
233 return "IsNotShrinkable";
236 //=======================================================================
237 //function : entityMode
238 //purpose : may return {'Elem0d' 'Edge' 'Face' 'Volume'} at most
239 //=======================================================================
241 QList<QVariant> SMESHGUI_Selection::entityMode( int ind ) const
243 QList<QVariant> types;
244 SMESH_Actor* actor = getActor( ind );
246 unsigned int aMode = actor->GetEntityMode();
247 if ( aMode & SMESH_Actor::eVolumes ) types.append( "Volume" );
248 if ( aMode & SMESH_Actor::eFaces ) types.append( "Face" );
249 if ( aMode & SMESH_Actor::eEdges ) types.append( "Edge" );
250 if ( aMode & SMESH_Actor::e0DElements ) types.append( "Elem0d" );
255 //=======================================================================
256 //function : controlMode
257 //purpose : return SMESH_Actor::eControl
258 //=======================================================================
260 QString SMESHGUI_Selection::controlMode( int ind ) const
262 SMESH_Actor* actor = getActor( ind );
264 switch( actor->GetControlMode() ) {
265 case SMESH_Actor::eLength: return "eLength";
266 case SMESH_Actor::eLength2D: return "eLength2D";
267 case SMESH_Actor::eFreeEdges: return "eFreeEdges";
268 case SMESH_Actor::eFreeNodes: return "eFreeNodes";
269 case SMESH_Actor::eFreeBorders: return "eFreeBorders";
270 case SMESH_Actor::eFreeFaces: return "eFreeFaces";
271 case SMESH_Actor::eMultiConnection: return "eMultiConnection";
272 case SMESH_Actor::eMultiConnection2D: return "eMultiConnection2D";
273 case SMESH_Actor::eArea: return "eArea";
274 case SMESH_Actor::eVolume3D: return "eVolume3D";
275 case SMESH_Actor::eMaxElementLength2D: return "eMaxElementLength2D";
276 case SMESH_Actor::eMaxElementLength3D: return "eMaxElementLength3D";
277 case SMESH_Actor::eTaper: return "eTaper";
278 case SMESH_Actor::eAspectRatio: return "eAspectRatio";
279 case SMESH_Actor::eAspectRatio3D: return "eAspectRatio3D";
280 case SMESH_Actor::eMinimumAngle: return "eMinimumAngle";
281 case SMESH_Actor::eWarping: return "eWarping";
282 case SMESH_Actor::eSkew: return "eSkew";
289 bool SMESHGUI_Selection::isNumFunctor( int ind ) const
292 SMESH_Actor* actor = getActor( ind );
294 switch( actor->GetControlMode() ) {
295 case SMESH_Actor::eLength:
296 case SMESH_Actor::eLength2D:
297 case SMESH_Actor::eMultiConnection:
298 case SMESH_Actor::eMultiConnection2D:
299 case SMESH_Actor::eArea:
300 case SMESH_Actor::eVolume3D:
301 case SMESH_Actor::eMaxElementLength2D:
302 case SMESH_Actor::eMaxElementLength3D:
303 case SMESH_Actor::eTaper:
304 case SMESH_Actor::eAspectRatio:
305 case SMESH_Actor::eAspectRatio3D:
306 case SMESH_Actor::eMinimumAngle:
307 case SMESH_Actor::eWarping:
308 case SMESH_Actor::eSkew:
318 //=======================================================================
319 //function : facesOrientationMode
321 //=======================================================================
323 QString SMESHGUI_Selection::facesOrientationMode( int ind ) const
325 SMESH_Actor* actor = getActor( ind );
327 if ( actor->GetFacesOriented() )
329 return "IsNotOriented";
334 //=======================================================================
335 //function : isAutoColor
337 //=======================================================================
339 bool SMESHGUI_Selection::isAutoColor( int ind ) const
341 if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
343 _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
344 CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
346 if ( ! CORBA::is_nil( obj )) {
347 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
348 if ( ! mesh->_is_nil() )
349 return mesh->GetAutoColor();
355 //=======================================================================
356 //function : numberOfNodes
358 //=======================================================================
360 int SMESHGUI_Selection::numberOfNodes( int ind ) const
362 if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
364 _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
365 CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
367 if ( ! CORBA::is_nil( obj )) {
368 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
369 if ( ! mesh->_is_nil() )
370 return mesh->NbNodes();
371 SMESH::SMESH_subMesh_var aSubMeshObj = SMESH::SMESH_subMesh::_narrow( obj );
372 if ( !aSubMeshObj->_is_nil() )
373 return aSubMeshObj->GetNumberOfNodes(true);
374 SMESH::SMESH_GroupBase_var aGroupObj = SMESH::SMESH_GroupBase::_narrow( obj );
375 if ( !aGroupObj->_is_nil() )
376 return aGroupObj->Size();
382 //=======================================================================
383 //function : isComputable
385 //=======================================================================
387 QVariant SMESHGUI_Selection::isComputable( int ind ) const
389 if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
391 /* Handle(SALOME_InteractiveObject) io =
392 static_cast<LightApp_DataOwner*>( myDataOwners[ ind ].get() )->IO();
393 if ( !io.IsNull() ) {
394 SMESH::SMESH_Mesh_var mesh = SMESH::GetMeshByIO(io); // m,sm,gr->m
395 if ( !mesh->_is_nil() ) {*/
396 _PTR(SObject) so = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
397 //FindSObject( mesh );
399 CORBA::Object_var obj = SMESH::SObjectToObject(so, SMESH::GetActiveStudyDocument());
400 if(!CORBA::is_nil(obj)){
401 SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
402 if (!mesh->_is_nil()){
403 if(mesh->HasShapeToMesh()) {
404 GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
405 return QVariant( !shape->_is_nil() );
409 return QVariant(!mesh->NbFaces()==0);
414 GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
415 return QVariant( !shape->_is_nil() );
422 return QVariant( false );
425 //=======================================================================
426 //function : isPreComputable
428 //=======================================================================
430 QVariant SMESHGUI_Selection::isPreComputable( int ind ) const
432 if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
434 QMap<int,int> modeMap;
435 _PTR(SObject) pMesh = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).toLatin1().data() );
436 SMESHGUI_PrecomputeOp::getAssignedAlgos( pMesh, modeMap );
437 return QVariant( modeMap.size() > 1 );
439 return QVariant( false );
442 //=======================================================================
443 //function : hasReference
445 //=======================================================================
447 QVariant SMESHGUI_Selection::hasReference( int ind ) const
449 return QVariant( isReference( ind ) );
452 //=======================================================================
453 //function : isVisible
455 //=======================================================================
457 QVariant SMESHGUI_Selection::isVisible( int ind ) const
459 if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
461 QString ent = entry( ind );
462 SMESH_Actor* actor = SMESH::FindActorByEntry( ent.toLatin1().data() );
463 if ( actor && actor->hasIO() ) {
464 if(SVTK_ViewWindow* aViewWindow = SMESH::GetCurrentVtkView())
465 return QVariant( aViewWindow->isVisible( actor->getIO() ) );
468 return QVariant( false );
471 //=======================================================================
474 //=======================================================================
476 int SMESHGUI_Selection::type( const QString& entry, _PTR(Study) study )
478 _PTR(SObject) obj (study->FindObjectID(entry.toLatin1().data()));
483 if( obj->ReferencedObject( ref ) )
486 _PTR(SObject) objFather = obj->GetFather();
487 _PTR(SComponent) objComponent = obj->GetFatherComponent();
489 if( objComponent->ComponentDataType()!="SMESH" )
492 if( objComponent->GetIOR()==obj->GetIOR() )
495 int aLevel = obj->Depth() - objComponent->Depth(),
496 aFTag = objFather->Tag(),
503 if (anOTag >= SMESH::Tag_FirstMeshRoot)
509 case SMESH::Tag_HypothesisRoot:
512 case SMESH::Tag_AlgorithmsRoot:
520 case SMESH::Tag_SubMeshOnVertex:
521 res = SUBMESH_VERTEX;
523 case SMESH::Tag_SubMeshOnEdge:
526 case SMESH::Tag_SubMeshOnFace:
529 case SMESH::Tag_SubMeshOnSolid:
532 case SMESH::Tag_SubMeshOnCompound:
533 res = SUBMESH_COMPOUND;
536 if (aFTag >= SMESH::Tag_FirstGroup)
547 //=======================================================================
548 //function : typeName
550 //=======================================================================
552 QString SMESHGUI_Selection::typeName( const int t )
565 return "Mesh or submesh";
567 return "Mesh vertex";
574 case SUBMESH_COMPOUND:
575 return "Mesh compound";
585 bool SMESHGUI_Selection::isImported( const int ind ) const
587 QString e = entry( ind );
588 _PTR(SObject) SO = SMESH::GetActiveStudyDocument()->FindObjectID( e.toLatin1().constData() );
592 SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow( SMESH::SObjectToObject( SO ) );
593 if( !aMesh->_is_nil() )
595 SALOME_MED::MedFileInfo* inf = aMesh->GetMEDFileInfo();
596 res = strlen( (char*)inf->fileName ) > 0;
602 //=======================================================================
603 //function : groupType
605 //=======================================================================
607 QString SMESHGUI_Selection::groupType( int ind ) const
609 QString e = entry( ind );
610 _PTR(SObject) SO = SMESH::GetActiveStudyDocument()->FindObjectID( e.toLatin1().constData() );
614 CORBA::Object_var obj = SMESH::SObjectToObject( SO );
616 SMESH::SMESH_Group_var aGroup = SMESH::SMESH_Group::_narrow( obj );
617 SMESH::SMESH_GroupOnGeom_var aGroupOnGeom = SMESH::SMESH_GroupOnGeom::_narrow( obj );
618 if( !aGroup->_is_nil() )
619 type = QString( "Group" );
620 else if ( !aGroupOnGeom->_is_nil() )
621 type = QString( "GroupOnGeom" );