]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESH_I/SMESH_subMesh_i.cxx
Salome HOME
0022221: [CEA 819] Error "'NoneType' object is not callable" at Salome closure
[modules/smesh.git] / src / SMESH_I / SMESH_subMesh_i.cxx
1 // Copyright (C) 2007-2013  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 SMESH_I : idl implementation based on 'SMESH' unit's calsses
24 //  File   : SMESH_subMesh_i.cxx
25 //  Author : Paul RASCLE, EDF
26 //  Module : SMESH
27 //
28 #include "SMESH_subMesh_i.hxx"
29 #include "SMESH_Gen_i.hxx"
30 #include "SMESH_Mesh_i.hxx"
31 #include "SMESH_PreMeshInfo.hxx"
32
33 #include "Utils_CorbaException.hxx"
34 #include "utilities.h"
35 #include "OpUtil.hxx"
36 #include "Utils_ExceptHandlers.hxx"
37
38 #include <TopoDS_Iterator.hxx>
39 #include <TopExp_Explorer.hxx>
40
41 using namespace std;
42
43 //=============================================================================
44 /*!
45  *  
46  */
47 //=============================================================================
48
49 SMESH_subMesh_i::SMESH_subMesh_i()
50      : SALOME::GenericObj_i( PortableServer::POA::_nil() )
51 {
52   MESSAGE("SMESH_subMesh_i::SMESH_subMesh_i default, not for use");
53   ASSERT(0);
54 }
55
56 //=============================================================================
57 /*!
58  *  
59  */
60 //=============================================================================
61
62 SMESH_subMesh_i::SMESH_subMesh_i( PortableServer::POA_ptr thePOA,
63                                   SMESH_Gen_i*            gen_i,
64                                   SMESH_Mesh_i*           mesh_i,
65                                   int                     localId )
66      : SALOME::GenericObj_i( thePOA )
67 {
68   _gen_i = gen_i;
69   _mesh_i = mesh_i;
70   _localId = localId;
71   _preMeshInfo = NULL;
72 }
73 //=============================================================================
74 /*!
75  *  
76  */
77 //=============================================================================
78
79 SMESH_subMesh_i::~SMESH_subMesh_i()
80 {
81   MESSAGE("SMESH_subMesh_i::~SMESH_subMesh_i");
82   if ( _preMeshInfo ) delete _preMeshInfo;
83   _preMeshInfo = NULL;
84 }
85
86 //=======================================================================
87 //function : getSubMeshes
88 //purpose  : for a submesh on shape to which elements are not bound directly,
89 //           return submeshes containing elements
90 //=======================================================================
91
92 typedef list<SMESHDS_SubMesh*> TListOfSubMeshes;
93
94 bool getSubMeshes(::SMESH_subMesh*  theSubMesh,
95                   TListOfSubMeshes& theSubMeshList)
96 {
97   int size = theSubMeshList.size();
98
99   SMESH_Mesh*      aMesh      = theSubMesh->GetFather();
100   SMESHDS_Mesh*    aMeshDS    = aMesh->GetMeshDS();
101   SMESHDS_SubMesh* aSubMeshDS = theSubMesh->GetSubMeshDS();
102
103   // nodes can be bound to either vertex, edge, face or solid_or_shell
104   TopoDS_Shape aShape = theSubMesh->GetSubShape();
105   switch ( aShape.ShapeType() )
106   {
107   case TopAbs_SOLID: {
108     // add submesh of solid itself
109     aSubMeshDS = aMeshDS->MeshElements( aShape );
110     if ( aSubMeshDS )
111       theSubMeshList.push_back( aSubMeshDS );
112     // and of the first shell
113     TopExp_Explorer exp( aShape, TopAbs_SHELL );
114     if ( exp.More() ) {
115       aSubMeshDS = aMeshDS->MeshElements( exp.Current() );
116       if ( aSubMeshDS )
117         theSubMeshList.push_back( aSubMeshDS );
118     }
119     break;
120   }
121   case TopAbs_WIRE:
122   case TopAbs_COMPOUND:
123   case TopAbs_COMPSOLID: {
124     // call getSubMeshes() for sub-shapes
125     list<TopoDS_Shape> shapeList;
126     shapeList.push_back( aShape );
127     list<TopoDS_Shape>::iterator sh = shapeList.begin();
128     for ( ; sh != shapeList.end(); ++sh ) {
129       for ( TopoDS_Iterator it( *sh ); it.More(); it.Next() ) {
130         if ( ::SMESH_subMesh* aSubMesh = aMesh->GetSubMeshContaining( it.Value() ))
131           getSubMeshes( aSubMesh, theSubMeshList ); // add found submesh or explore deeper
132         else
133           // no submesh for a compound inside compound
134           shapeList.push_back( it.Value() );
135       }
136     }
137     // return only unique submeshes
138     set<SMESHDS_SubMesh*> smSet;
139     TListOfSubMeshes::iterator sm = theSubMeshList.begin();
140     while ( sm != theSubMeshList.end() ) {
141       if ( !smSet.insert( *sm ).second )
142         sm = theSubMeshList.erase( sm );
143       else
144         ++sm;
145     }
146     break;
147   }
148   default:
149     if ( aSubMeshDS )
150       theSubMeshList.push_back( aSubMeshDS );
151   }
152   return size < theSubMeshList.size();
153 }
154
155 //=============================================================================
156 /*!
157  *  
158  */
159 //=============================================================================
160
161 CORBA::Long SMESH_subMesh_i::GetNumberOfElements()
162   throw (SALOME::SALOME_Exception)
163 {
164   Unexpect aCatch(SALOME_SalomeException);
165
166   if ( _preMeshInfo )
167     return _preMeshInfo->NbElements();
168
169   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
170     return 0;
171
172   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
173   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
174
175   int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
176
177   // volumes are bound to shell
178   TListOfSubMeshes smList;
179   if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
180   {
181     TListOfSubMeshes::iterator sm = smList.begin();
182     for ( ; sm != smList.end(); ++sm )
183       nbElems += (*sm)->NbElements();
184   }
185   return nbElems;
186 }
187
188 //=============================================================================
189 /*!
190  *  
191  */
192 //=============================================================================
193
194 CORBA::Long SMESH_subMesh_i::GetNumberOfNodes(CORBA::Boolean all)
195   throw (SALOME::SALOME_Exception)
196 {
197   Unexpect aCatch(SALOME_SalomeException);
198
199   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
200     return 0;
201
202   if ( _preMeshInfo )
203   {
204     if ( all ) return _preMeshInfo->NbNodes();
205     else _preMeshInfo->FullLoadFromFile();
206   }
207   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
208   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
209
210   if ( aSubMeshDS && aSubMeshDS->IsComplexSubmesh() )
211   {
212     // sub-mesh on a geom group, always return all nodes
213     return aSubMeshDS->NbNodes();
214   }
215   if ( aSubMeshDS && !all )
216   {
217     // return anything we have
218     return aSubMeshDS->NbNodes();
219   }
220   if ( all ) // get nodes from aSubMesh and all child sub-meshes
221   {
222     int nbNodes = 0;
223     SMESH_subMeshIteratorPtr smIt = aSubMesh->getDependsOnIterator( /*includeSelf=*/true );
224     while ( smIt->more() )
225     {
226       aSubMesh = smIt->next();
227       if (( aSubMeshDS = aSubMesh->GetSubMeshDS() ))
228         nbNodes += aSubMeshDS->NbNodes();
229     }
230     return nbNodes;
231   }
232
233   return aSubMeshDS ? aSubMeshDS->NbNodes() : 0;
234 }
235
236 //=============================================================================
237 /*!
238  *  
239  */
240 //=============================================================================
241
242 SMESH::long_array* SMESH_subMesh_i::GetElementsId()
243   throw (SALOME::SALOME_Exception)
244 {
245   Unexpect aCatch(SALOME_SalomeException);
246
247   SMESH::long_array_var aResult = new SMESH::long_array();
248
249   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
250     return aResult._retn();
251
252   if ( _preMeshInfo )
253     _preMeshInfo->FullLoadFromFile();
254
255   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
256   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
257
258   int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
259   TListOfSubMeshes smList;
260   if ( nbElems )
261     smList.push_back( aSubMeshDS );
262
263   // volumes are bound to shell
264   if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
265   {
266     TListOfSubMeshes::iterator sm = smList.begin();
267     for ( ; sm != smList.end(); ++sm )
268       nbElems += (*sm)->NbElements();
269   }
270
271   aResult->length( nbElems );
272   if ( nbElems )
273   {
274     TListOfSubMeshes::iterator sm = smList.begin();
275     for ( int i = 0; sm != smList.end(); sm++ )
276     {
277       SMDS_ElemIteratorPtr anIt = (*sm)->GetElements();
278       for ( ; i < nbElems && anIt->more(); i++ )
279         aResult[i] = anIt->next()->GetID();
280     }
281   }
282   return aResult._retn();
283 }
284
285
286 //=============================================================================
287 /*!
288  *  
289  */
290 //=============================================================================
291
292 SMESH::long_array* SMESH_subMesh_i::GetElementsByType( SMESH::ElementType theElemType )
293     throw (SALOME::SALOME_Exception)
294 {
295   Unexpect aCatch(SALOME_SalomeException);
296
297   SMESH::long_array_var aResult = new SMESH::long_array();
298
299   if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
300     return aResult._retn();
301
302   if ( _preMeshInfo )
303     _preMeshInfo->FullLoadFromFile();
304
305   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
306   SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
307
308   // PAL5440, return all nodes belonging to elements of submesh
309   set<int> nodeIds;
310   int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
311
312   // volumes may be bound to shell instead of solid
313   TListOfSubMeshes smList;
314   if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
315   {
316     TListOfSubMeshes::iterator sm = smList.begin();
317     for ( ; sm != smList.end(); ++sm )
318     {
319       if ( theElemType == SMESH::NODE )
320       {
321         SMDS_ElemIteratorPtr eIt = (*sm)->GetElements();
322         if ( eIt->more() ) {
323           while ( eIt->more() ) {
324             const SMDS_MeshElement* anElem = eIt->next();
325             SMDS_ElemIteratorPtr nIt = anElem->nodesIterator();
326             while ( nIt->more() )
327               nodeIds.insert( nIt->next()->GetID() );
328           }
329         } else {
330           SMDS_NodeIteratorPtr nIt = (*sm)->GetNodes();
331           while ( nIt->more() )
332             nodeIds.insert( nIt->next()->GetID() );
333         }
334       }
335       else
336       {
337         nbElems += (*sm)->NbElements();
338       }
339     }
340     aSubMeshDS = 0;
341   }
342   else
343   {
344     if ( nbElems )
345       smList.push_back( aSubMeshDS );
346   }
347
348   if ( theElemType == SMESH::NODE && aSubMeshDS )
349   {
350     SMDS_ElemIteratorPtr eIt = aSubMeshDS->GetElements();
351     if ( eIt->more() ) {
352       while ( eIt->more() ) {
353         const SMDS_MeshElement* anElem = eIt->next();
354         SMDS_ElemIteratorPtr nIt = anElem->nodesIterator();
355         while ( nIt->more() )
356           nodeIds.insert( nIt->next()->GetID() );
357       }
358     } else {
359       SMDS_NodeIteratorPtr nIt = aSubMeshDS->GetNodes();
360       while ( nIt->more() )
361         nodeIds.insert( nIt->next()->GetID() );
362     }
363   }
364
365   if ( theElemType == SMESH::NODE )
366     aResult->length( nodeIds.size() );
367   else
368     aResult->length( nbElems );
369
370   int i = 0, n = aResult->length();
371
372   if ( theElemType == SMESH::NODE && !nodeIds.empty() ) {
373     set<int>::iterator idIt = nodeIds.begin();
374     for ( ; i < n && idIt != nodeIds.end() ; i++, idIt++ )
375       aResult[i] = *idIt;
376   }
377
378   if ( theElemType != SMESH::NODE ) {
379     TListOfSubMeshes::iterator sm = smList.begin();
380     for ( i = 0; sm != smList.end(); sm++ )
381     {
382       aSubMeshDS = *sm;
383       SMDS_ElemIteratorPtr anIt = aSubMeshDS->GetElements();
384       while ( i < n && anIt->more() ) {
385         const SMDS_MeshElement* anElem = anIt->next();
386         if ( theElemType == SMESH::ALL || anElem->GetType() == (SMDSAbs_ElementType)theElemType )
387           aResult[i++] = anElem->GetID();
388       }
389     }
390   }
391
392   aResult->length( i );
393
394   return aResult._retn();
395 }
396
397 //=============================================================================
398 /*!
399  *  
400  */
401 //=============================================================================
402   
403 SMESH::long_array* SMESH_subMesh_i::GetNodesId()
404   throw (SALOME::SALOME_Exception)
405 {
406   Unexpect aCatch(SALOME_SalomeException);
407
408   SMESH::long_array_var aResult = GetElementsByType( SMESH::NODE );
409   return aResult._retn();
410 }
411
412 //=============================================================================
413 /*!
414  *  
415  */
416 //=============================================================================
417   
418 SMESH::SMESH_Mesh_ptr SMESH_subMesh_i::GetFather()
419   throw (SALOME::SALOME_Exception)
420 {
421   Unexpect aCatch(SALOME_SalomeException);
422   return _mesh_i->_this();
423 }
424
425 //=============================================================================
426 /*!
427  *  
428  */
429 //=============================================================================
430   
431 CORBA::Long SMESH_subMesh_i::GetId()
432 {
433   return _localId;
434 }
435
436 //=======================================================================
437 //function : GetSubShape
438 //purpose  : 
439 //=======================================================================
440
441 GEOM::GEOM_Object_ptr SMESH_subMesh_i::GetSubShape()
442      throw (SALOME::SALOME_Exception)
443 {
444   Unexpect aCatch(SALOME_SalomeException);
445   GEOM::GEOM_Object_var aShapeObj;
446   try {
447     if ( _mesh_i->_mapSubMesh.find( _localId ) != _mesh_i->_mapSubMesh.end()) {
448       TopoDS_Shape S = _mesh_i->_mapSubMesh[ _localId ]->GetSubShape();
449       if ( !S.IsNull() ) {
450         aShapeObj = _gen_i->ShapeToGeomObject( S );
451         //mzn: N7PAL16232, N7PAL16233
452         //In some cases it's possible that GEOM_Client contains the shape same to S, but
453         //with another orientation.
454         if (aShapeObj->_is_nil())
455           aShapeObj = _gen_i->ShapeToGeomObject( S.Reversed() );
456       }
457     }
458   }
459   catch(SALOME_Exception & S_ex) {
460     THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
461   }
462   return aShapeObj._retn();
463 }
464
465 //=============================================================================
466 /*!
467  *  
468  */
469 //=============================================================================
470 SMESH::long_array* SMESH_subMesh_i::GetIDs()
471 {
472   return GetElementsId();
473 }
474
475 //=============================================================================
476 /*!
477  *
478  */
479 //=============================================================================
480 SMESH::ElementType SMESH_subMesh_i::GetElementType( const CORBA::Long id, const bool iselem )
481   throw (SALOME::SALOME_Exception)
482 {
483   if ( _preMeshInfo )
484     _preMeshInfo->FullLoadFromFile();
485   return GetFather()->GetElementType( id, iselem );
486 }
487
488 //=============================================================================
489 /*
490  * Returns number of mesh elements of each \a EntityType
491  * @return array of number of elements per \a EntityType
492  */
493 //=============================================================================
494
495 SMESH::long_array* SMESH_subMesh_i::GetMeshInfo()
496 {
497   if ( _preMeshInfo )
498     return _preMeshInfo->GetMeshInfo();
499
500   SMESH::long_array_var aRes = new SMESH::long_array();
501   aRes->length(SMESH::Entity_Last);
502   for (int i = SMESH::Entity_Node; i < SMESH::Entity_Last; i++)
503     aRes[i] = 0;
504   
505   // get number of nodes
506   aRes[ SMESH::Entity_Node ] = GetNumberOfNodes(true);
507  
508   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
509
510   // get statistic from child sub-meshes
511   TListOfSubMeshes smList;
512   if ( getSubMeshes( aSubMesh, smList ) )
513     for ( TListOfSubMeshes::iterator sm = smList.begin(); sm != smList.end(); ++sm )
514       SMESH_Mesh_i::CollectMeshInfo( (*sm)->GetElements(), aRes );
515
516   return aRes._retn();
517 }
518
519 //=======================================================================
520 /*
521  * Returns number of mesh elements of each \a ElementType
522  */
523 //=======================================================================
524
525 SMESH::long_array* SMESH_subMesh_i::GetNbElementsByType()
526 {
527   SMESH::long_array_var aRes = new SMESH::long_array();
528   aRes->length(SMESH::NB_ELEMENT_TYPES);
529   for (int i = 0; i < SMESH::NB_ELEMENT_TYPES; i++)
530     if ( _preMeshInfo )
531       aRes[ i ] = _preMeshInfo->NbElements( SMDSAbs_ElementType( i ));
532     else
533       aRes[ i ] = 0;
534
535   if ( !_preMeshInfo )
536   {
537     aRes[ SMESH::NODE ] = GetNumberOfNodes(true);
538
539     ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
540     if ( SMESHDS_SubMesh* smDS = aSubMesh->GetSubMeshDS() )
541     {
542       SMDS_ElemIteratorPtr eIt = smDS->GetElements();
543       if ( eIt->more() )
544         aRes[ eIt->next()->GetType() ] = smDS->NbElements();
545     }
546   }
547   return aRes._retn();  
548 }
549
550
551 //=======================================================================
552 //function : GetTypes
553 //purpose  : Returns types of elements it contains
554 //=======================================================================
555
556 SMESH::array_of_ElementType* SMESH_subMesh_i::GetTypes()
557 {
558   if ( _preMeshInfo )
559     return _preMeshInfo->GetTypes();
560
561   SMESH::array_of_ElementType_var types = new SMESH::array_of_ElementType;
562
563   ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
564   if ( SMESHDS_SubMesh* smDS = aSubMesh->GetSubMeshDS() )
565   {
566     SMDS_ElemIteratorPtr eIt = smDS->GetElements();
567     if ( eIt->more() )
568     {
569       types->length( 1 );
570       types[0] = SMESH::ElementType( eIt->next()->GetType());
571     }
572     else if ( smDS->GetNodes()->more() )
573     {
574       TopoDS_Shape shape = aSubMesh->GetSubShape();
575       while ( !shape.IsNull() && shape.ShapeType() == TopAbs_COMPOUND )
576       {
577         TopoDS_Iterator it( shape );
578         shape = it.More() ? it.Value() : TopoDS_Shape();
579       }
580       if ( !shape.IsNull() && shape.ShapeType() == TopAbs_VERTEX )
581       {
582         types->length( 1 );
583         types[0] = SMESH::NODE;
584       }
585     }
586   }
587   return types._retn();
588 }
589
590 //=======================================================================
591 //function : GetMesh
592 //purpose  : interface SMESH_IDSource
593 //=======================================================================
594
595 SMESH::SMESH_Mesh_ptr SMESH_subMesh_i::GetMesh()
596 {
597   return GetFather();
598 }
599
600 //=======================================================================
601 //function : IsMeshInfoCorrect
602 //purpose  : * Returns false if GetMeshInfo() returns incorrect information that may
603 //           * happen if mesh data is not yet fully loaded from the file of study.
604 //=======================================================================
605
606 bool SMESH_subMesh_i::IsMeshInfoCorrect()
607 {
608   return _preMeshInfo ? _preMeshInfo->IsMeshInfoCorrect() : true;
609 }