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