]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_Selection.cxx
Salome HOME
Bug NPAL12872 - EDF189 GEOM, SMESH : Visualization of groups with many colors
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_Selection.cxx
1 // Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
3 // 
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either 
7 // version 2.1 of the License.
8 // 
9 // This library is distributed in the hope that it will be useful 
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
12 // Lesser General Public License for more details.
13 //
14 // You should have received a copy of the GNU Lesser General Public  
15 // License along with this library; if not, write to the Free Software 
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 //
18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 //
20
21 #include "SMESHGUI_Selection.h"
22 #include "SMESHGUI_Utils.h"
23 #include "SMESHGUI_VTKUtils.h"
24 #include "SMESHGUI_MeshUtils.h"
25 #include "SMESHGUI_GEOMGenUtils.h"
26
27 #include "SMESH_Type.h"
28 #include "SMESH_Actor.h"
29
30 #include "LightApp_SelectionMgr.h"
31 #include "SalomeApp_Study.h"
32 #include "LightApp_VTKSelector.h"
33
34 #include "SUIT_Session.h"
35
36 #include "SVTK_RenderWindowInteractor.h"
37 #include "SVTK_ViewWindow.h"
38
39 #include CORBA_SERVER_HEADER(SMESH_Mesh)
40 #include CORBA_SERVER_HEADER(SMESH_Group)
41
42 //=======================================================================
43 //function : SMESHGUI_Selection
44 //purpose  : 
45 //=======================================================================
46 SMESHGUI_Selection::SMESHGUI_Selection()
47 : LightApp_Selection()
48 {
49 }
50
51 //=======================================================================
52 //function : ~SMESHGUI_Selection
53 //purpose  : 
54 //=======================================================================
55 SMESHGUI_Selection::~SMESHGUI_Selection()
56 {
57 }
58
59 //=======================================================================
60 //function : init
61 //purpose  : 
62 //=======================================================================
63 void SMESHGUI_Selection::init( const QString& client, LightApp_SelectionMgr* mgr )
64 {
65   LightApp_Selection::init( client, mgr );
66
67   if( mgr && study() )
68   {
69     SalomeApp_Study* aSStudy = dynamic_cast<SalomeApp_Study*>(study());
70     if (!aSStudy)
71       return;
72     _PTR(Study) aStudy = aSStudy->studyDS();
73
74     for( int i=0, n=count(); i<n; i++ )
75       myTypes.append( typeName( type( entry( i ), aStudy ) ) );
76   }
77 }
78
79 //=======================================================================
80 //function : processOwner
81 //purpose  : 
82 //=======================================================================
83 void SMESHGUI_Selection::processOwner( const LightApp_DataOwner* ow )
84 {
85   const LightApp_SVTKDataOwner* owner = 
86     dynamic_cast<const LightApp_SVTKDataOwner*> ( ow );
87   if( owner )
88     myActors.append( dynamic_cast<SMESH_Actor*>( owner->GetActor() ) );
89   else
90     myActors.append( 0 );
91 }
92
93 //=======================================================================
94 //function : param
95 //purpose  : 
96 //=======================================================================
97 QtxValue SMESHGUI_Selection::param( const int ind, const QString& p ) const
98 {
99   QtxValue val;
100        if ( p=="client" )        val = QtxValue( globalParam( p ) );
101   else if ( p=="type" )          val = QtxValue( myTypes[ind] );
102   else if ( p=="elemTypes" )     val = QtxValue( elemTypes( ind ) );
103   else if ( p=="isAutoColor" )   val = QtxValue( isAutoColor( ind ) );
104   else if ( p=="numberOfNodes" ) val = QtxValue( numberOfNodes( ind ) );
105   else if ( p=="labeledTypes" )  val = QtxValue( labeledTypes( ind ) );
106   else if ( p=="shrinkMode" )    val = QtxValue( shrinkMode( ind ) );
107   else if ( p=="entityMode" )    val = QtxValue( entityMode( ind ) );
108   else if ( p=="controlMode" )   val = QtxValue( controlMode( ind ) );
109   else if ( p=="displayMode" )   val = QtxValue( displayMode( ind ) );
110   else if ( p=="isComputable" )  val = QtxValue( isComputable( ind ) );
111   else if ( p=="hasReference" )  val = QtxValue( hasReference( ind ) );
112 //  else if ( p=="isVisible" )     val = QtxValue( isVisible( ind ) );
113
114        // printf( "--> param() : [%s] = %s (%s)\n", p.latin1(), val.toString().latin1(), val.typeName() );
115   //if ( val.type() == QVariant::List )
116   //cout << "size: " << val.toList().count() << endl;
117
118   if( val.isValid() )
119     return val;
120   else
121     return LightApp_Selection::param( ind, p );
122 }
123
124 //=======================================================================
125 //function : getVtkOwner
126 //purpose  : 
127 //=======================================================================
128
129 SMESH_Actor* SMESHGUI_Selection::getActor( int ind ) const
130 {
131   if( ind >= 0 && ind < count() )
132     return ((QPtrList<SMESH_Actor>&)myActors).at( ind );
133   else
134     return 0;
135 }
136
137 //=======================================================================
138 //function : elemTypes
139 //purpose  : may return {'Edge' 'Face' 'Volume'} at most
140 //=======================================================================
141
142 QValueList<QVariant> SMESHGUI_Selection::elemTypes( int ind ) const
143 {
144   QValueList<QVariant> types;
145   SMESH_Actor* actor = getActor( ind );
146   if ( actor ) {
147     TVisualObjPtr object = actor->GetObject();
148     if ( object ) {
149       if ( object->GetNbEntities( SMDSAbs_Edge )) types.append( "Edge" );
150       if ( object->GetNbEntities( SMDSAbs_Face )) types.append( "Face" );
151       if ( object->GetNbEntities( SMDSAbs_Volume )) types.append( "Volume" );
152     }
153   }
154   return types;
155 }
156
157 //=======================================================================
158 //function : labeledTypes
159 //purpose  : may return {'Point' 'Cell'} at most
160 //=======================================================================
161
162 QValueList<QVariant> SMESHGUI_Selection::labeledTypes( int ind ) const
163 {
164   QValueList<QVariant> types;
165   SMESH_Actor* actor = getActor( ind );
166   if ( actor ) {
167     if ( actor->GetPointsLabeled()) types.append( "Point" );
168     if ( actor->GetCellsLabeled()) types.append( "Cell" );
169   }
170   return types;
171 }
172
173 //=======================================================================
174 //function : displayMode
175 //purpose  : return SMESH_Actor::EReperesent
176 //=======================================================================
177
178 QString SMESHGUI_Selection::displayMode( int ind ) const
179 {
180   SMESH_Actor* actor = getActor( ind );
181   if ( actor ) {
182     switch( actor->GetRepresentation() ) {
183     case SMESH_Actor::eEdge:    return "eEdge";
184     case SMESH_Actor::eSurface: return "eSurface";
185     case SMESH_Actor::ePoint:   return "ePoint";
186     default:;
187     }
188   }
189   return "Unknown";
190 }
191
192 //=======================================================================
193 //function : shrinkMode
194 //purpose  : return either 'IsSrunk', 'IsNotShrunk' or 'IsNotShrinkable'
195 //=======================================================================
196
197 QString SMESHGUI_Selection::shrinkMode( int ind ) const
198 {
199   SMESH_Actor* actor = getActor( ind );
200   if ( actor && actor->IsShrunkable() ) {
201     if ( actor->IsShrunk() )
202       return "IsShrunk";
203     return "IsNotShrunk";
204   }
205   return "IsNotShrinkable";
206 }
207
208 //=======================================================================
209 //function : entityMode
210 //purpose  : may return {'Edge' 'Face' 'Volume'} at most
211 //=======================================================================
212
213 QValueList<QVariant> SMESHGUI_Selection::entityMode( int ind ) const
214 {
215   QValueList<QVariant> types;
216   SMESH_Actor* actor = getActor( ind );
217   if ( actor ) {
218     unsigned int aMode = actor->GetEntityMode();
219     if ( aMode & SMESH_Actor::eVolumes) types.append( "Volume");
220     if ( aMode & SMESH_Actor::eFaces  ) types.append( "Face"  );
221     if ( aMode & SMESH_Actor::eEdges  ) types.append( "Edge"  );
222   }
223   return types;
224 }
225
226 //=======================================================================
227 //function : controlMode
228 //purpose  : return SMESH_Actor::eControl
229 //=======================================================================
230
231 QString SMESHGUI_Selection::controlMode( int ind ) const
232 {
233   SMESH_Actor* actor = getActor( ind );
234   if ( actor ) {
235     switch( actor->GetControlMode() ) {
236     case SMESH_Actor::eLength:            return "eLength";
237     case SMESH_Actor::eLength2D:          return "eLength2D";
238     case SMESH_Actor::eFreeEdges:         return "eFreeEdges";
239     case SMESH_Actor::eFreeBorders:       return "eFreeBorders";
240     case SMESH_Actor::eMultiConnection:   return "eMultiConnection";
241     case SMESH_Actor::eMultiConnection2D: return "eMultiConnection2D";
242     case SMESH_Actor::eArea:              return "eArea";
243     case SMESH_Actor::eVolume3D:          return "eVolume3D";
244     case SMESH_Actor::eTaper:             return "eTaper";
245     case SMESH_Actor::eAspectRatio:       return "eAspectRatio";
246     case SMESH_Actor::eAspectRatio3D:     return "eAspectRatio3D";
247     case SMESH_Actor::eMinimumAngle:      return "eMinimumAngle";
248     case SMESH_Actor::eWarping:           return "eWarping";
249     case SMESH_Actor::eSkew:              return "eSkew";
250     default:;
251     }
252   }
253   return "eNone";
254 }
255
256 //=======================================================================
257 //function : isAutoColor
258 //purpose  : 
259 //=======================================================================
260
261 bool SMESHGUI_Selection::isAutoColor( int ind ) const
262 {
263   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
264   {
265     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).latin1() );
266     CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
267
268     if ( ! CORBA::is_nil( obj )) {
269       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
270       if ( ! mesh->_is_nil() )
271         return mesh->GetAutoColor();
272     }
273   }
274   return false;
275 }
276
277 //=======================================================================
278 //function : numberOfNodes
279 //purpose  : 
280 //=======================================================================
281
282 int SMESHGUI_Selection::numberOfNodes( int ind ) const
283 {
284   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
285   {
286     _PTR(SObject) sobj = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).latin1() );
287     CORBA::Object_var obj = SMESH::SObjectToObject( sobj, SMESH::GetActiveStudyDocument() );
288
289     if ( ! CORBA::is_nil( obj )) {
290       SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
291       if ( ! mesh->_is_nil() )
292         return mesh->NbNodes();
293       SMESH::SMESH_subMesh_var aSubMeshObj = SMESH::SMESH_subMesh::_narrow( obj );
294       if ( !aSubMeshObj->_is_nil() )
295         return aSubMeshObj->GetNumberOfNodes(true);
296       SMESH::SMESH_GroupBase_var aGroupObj = SMESH::SMESH_GroupBase::_narrow( obj );
297       if ( !aGroupObj->_is_nil() )
298         return aGroupObj->Size();
299     }
300   }
301   return 0;
302 }
303
304 //=======================================================================
305 //function : isComputable
306 //purpose  : 
307 //=======================================================================
308
309 QVariant SMESHGUI_Selection::isComputable( int ind ) const
310 {
311   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
312   {
313 /*    Handle(SALOME_InteractiveObject) io =
314       static_cast<LightApp_DataOwner*>( myDataOwners[ ind ].get() )->IO();
315     if ( !io.IsNull() ) {
316       SMESH::SMESH_Mesh_var mesh = SMESH::GetMeshByIO(io) ; // m,sm,gr->m
317       if ( !mesh->_is_nil() ) {*/
318         _PTR(SObject) so = SMESH::GetActiveStudyDocument()->FindObjectID( entry( ind ).latin1() );
319         //FindSObject( mesh );
320         if ( so ) {
321           CORBA::Object_var obj = SMESH::SObjectToObject(so, SMESH::GetActiveStudyDocument());
322           if(!CORBA::is_nil(obj)){
323             SMESH::SMESH_Mesh_var mesh = SMESH::SMESH_Mesh::_narrow( obj );
324             if (!mesh->_is_nil()){
325               if(mesh->HasShapeToMesh()) {
326                 GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
327                 return QVariant( !shape->_is_nil(), 0 );
328               }
329               else
330               {
331                 return QVariant(!mesh->NbFaces()==0, 0);
332               }
333             }
334             else
335             {
336               GEOM::GEOM_Object_var shape = SMESH::GetShapeOnMeshOrSubMesh( so );
337               return QVariant( !shape->_is_nil(), 0 );
338             }
339           }
340         }
341 //      }
342 //    }
343   }
344   return QVariant( false, 0 );
345 }
346
347 //=======================================================================
348 //function : hasReference
349 //purpose  : 
350 //=======================================================================
351
352 QVariant SMESHGUI_Selection::hasReference( int ind ) const
353 {
354   return QVariant( isReference( ind ), 0 );
355 }
356
357 //=======================================================================
358 //function : isVisible
359 //purpose  : 
360 //=======================================================================
361
362 QVariant SMESHGUI_Selection::isVisible( int ind ) const
363 {
364   if ( ind >= 0 && ind < myTypes.count() && myTypes[ind] != "Unknown" )
365   {
366     QString ent = entry( ind );
367     SMESH_Actor* actor = SMESH::FindActorByEntry( ent.latin1() );
368     if ( actor && actor->hasIO() ) {
369       if(SVTK_ViewWindow* aViewWindow = SMESH::GetCurrentVtkView())
370         return QVariant( aViewWindow->isVisible( actor->getIO() ), 0 );
371     }
372   }
373   return QVariant( false, 0 );
374 }
375
376 //=======================================================================
377 //function : type
378 //purpose  : 
379 //=======================================================================
380
381 int SMESHGUI_Selection::type( const QString& entry, _PTR(Study) study )
382 {
383   _PTR(SObject) obj (study->FindObjectID(entry.latin1()));
384   if( !obj )
385     return -1;
386
387   _PTR(SObject) ref;
388   if( obj->ReferencedObject( ref ) )
389     obj = ref;
390
391   _PTR(SObject) objFather = obj->GetFather();
392   _PTR(SComponent) objComponent = obj->GetFatherComponent();
393
394   if( objComponent->ComponentDataType()!="SMESH" )
395     return -1;
396
397   if( objComponent->GetIOR()==obj->GetIOR() )
398     return COMPONENT;
399
400   int aLevel = obj->Depth() - objComponent->Depth(),
401       aFTag = objFather->Tag(),
402       anOTag = obj->Tag(),
403       res = -1;
404
405   switch( aLevel )
406   {
407   case 1:
408     if( anOTag>=3 )
409       res = MESH;
410     break;
411   case 2:
412     switch( aFTag )
413     {
414     case 1:
415       res = HYPOTHESIS;
416       break;
417     case 2:
418       res = ALGORITHM;
419       break;
420     }
421     break;
422   case 3:
423     switch( aFTag )
424     {
425     case 4:
426       res = SUBMESH_VERTEX;
427       break;
428     case 5:
429       res = SUBMESH_EDGE;
430       break;
431     case 7:
432       res = SUBMESH_FACE;
433       break;
434     case 9:
435       res = SUBMESH_SOLID;
436       break;
437     case 10:
438       res = SUBMESH_COMPOUND;
439       break;
440     }
441     if( aFTag>10 )
442       res = GROUP;
443     else
444       res = SUBMESH;
445
446     break;
447   }
448
449   return res;
450 }
451
452 //=======================================================================
453 //function : typeName
454 //purpose  : 
455 //=======================================================================
456
457 QString SMESHGUI_Selection::typeName( const int t )
458 {
459   switch( t )
460   {
461   case HYPOTHESIS:
462     return "Hypothesis";
463   case ALGORITHM:
464     return "Algorithm";
465   case MESH:
466     return "Mesh";
467   case SUBMESH:
468     return "SubMesh";
469   case MESHorSUBMESH:
470     return "Mesh or submesh";
471   case SUBMESH_VERTEX:
472     return "Mesh vertex";
473   case SUBMESH_EDGE:
474     return "Mesh edge";
475   case SUBMESH_FACE:
476     return "Mesh face";
477   case SUBMESH_SOLID:
478     return "Mesh solid";
479   case SUBMESH_COMPOUND:
480     return "Mesh compound";
481   case GROUP:
482     return "Group";
483   case COMPONENT:
484     return "Component";
485   default:
486     return "Unknown";
487   }
488 }