1 // Copyright (C) 2007-2013 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 SMESH_I : idl implementation based on 'SMESH' unit's calsses
24 // File : SMESH_subMesh_i.cxx
25 // Author : Paul RASCLE, EDF
28 #include "SMESH_subMesh_i.hxx"
29 #include "SMESH_Gen_i.hxx"
30 #include "SMESH_Mesh_i.hxx"
31 #include "SMESH_PreMeshInfo.hxx"
33 #include "Utils_CorbaException.hxx"
34 #include "utilities.h"
36 #include "Utils_ExceptHandlers.hxx"
38 #include <TopoDS_Iterator.hxx>
39 #include <TopExp_Explorer.hxx>
43 //=============================================================================
47 //=============================================================================
49 SMESH_subMesh_i::SMESH_subMesh_i()
50 : SALOME::GenericObj_i( PortableServer::POA::_nil() )
52 MESSAGE("SMESH_subMesh_i::SMESH_subMesh_i default, not for use");
56 //=============================================================================
60 //=============================================================================
62 SMESH_subMesh_i::SMESH_subMesh_i( PortableServer::POA_ptr thePOA,
66 : SALOME::GenericObj_i( thePOA )
73 //=============================================================================
77 //=============================================================================
79 SMESH_subMesh_i::~SMESH_subMesh_i()
81 MESSAGE("SMESH_subMesh_i::~SMESH_subMesh_i");
82 if ( _preMeshInfo ) delete _preMeshInfo;
86 //=======================================================================
87 //function : getSubMeshes
88 //purpose : for a submesh on shape to which elements are not bound directly,
89 // return submeshes containing elements
90 //=======================================================================
92 typedef list<SMESHDS_SubMesh*> TListOfSubMeshes;
94 bool getSubMeshes(::SMESH_subMesh* theSubMesh,
95 TListOfSubMeshes& theSubMeshList)
97 int size = theSubMeshList.size();
99 SMESH_Mesh* aMesh = theSubMesh->GetFather();
100 SMESHDS_Mesh* aMeshDS = aMesh->GetMeshDS();
101 SMESHDS_SubMesh* aSubMeshDS = theSubMesh->GetSubMeshDS();
103 // nodes can be bound to either vertex, edge, face or solid_or_shell
104 TopoDS_Shape aShape = theSubMesh->GetSubShape();
105 switch ( aShape.ShapeType() )
108 // add submesh of solid itself
109 aSubMeshDS = aMeshDS->MeshElements( aShape );
111 theSubMeshList.push_back( aSubMeshDS );
112 // and of the first shell
113 TopExp_Explorer exp( aShape, TopAbs_SHELL );
115 aSubMeshDS = aMeshDS->MeshElements( exp.Current() );
117 theSubMeshList.push_back( aSubMeshDS );
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
133 // no submesh for a compound inside compound
134 shapeList.push_back( it.Value() );
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 );
150 theSubMeshList.push_back( aSubMeshDS );
152 return size < theSubMeshList.size();
155 //=============================================================================
159 //=============================================================================
161 CORBA::Long SMESH_subMesh_i::GetNumberOfElements()
162 throw (SALOME::SALOME_Exception)
164 Unexpect aCatch(SALOME_SalomeException);
167 return _preMeshInfo->NbElements();
169 if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
172 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
173 SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
175 int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
177 // volumes are bound to shell
178 TListOfSubMeshes smList;
179 if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
181 TListOfSubMeshes::iterator sm = smList.begin();
182 for ( ; sm != smList.end(); ++sm )
183 nbElems += (*sm)->NbElements();
188 //=============================================================================
192 //=============================================================================
194 CORBA::Long SMESH_subMesh_i::GetNumberOfNodes(CORBA::Boolean all)
195 throw (SALOME::SALOME_Exception)
197 Unexpect aCatch(SALOME_SalomeException);
199 if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
204 if ( all ) return _preMeshInfo->NbNodes();
205 else _preMeshInfo->FullLoadFromFile();
207 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
208 SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
210 if ( aSubMeshDS && aSubMeshDS->IsComplexSubmesh() )
212 // sub-mesh on a geom group, always return all nodes
213 return aSubMeshDS->NbNodes();
215 if ( aSubMeshDS && !all )
217 // return anything we have
218 return aSubMeshDS->NbNodes();
220 if ( all ) // get nodes from aSubMesh and all child sub-meshes
223 SMESH_subMeshIteratorPtr smIt = aSubMesh->getDependsOnIterator( /*includeSelf=*/true );
224 while ( smIt->more() )
226 aSubMesh = smIt->next();
227 if (( aSubMeshDS = aSubMesh->GetSubMeshDS() ))
228 nbNodes += aSubMeshDS->NbNodes();
232 return aSubMeshDS ? aSubMeshDS->NbNodes() : 0;
235 //=============================================================================
239 //=============================================================================
241 SMESH::long_array* SMESH_subMesh_i::GetElementsId()
242 throw (SALOME::SALOME_Exception)
244 Unexpect aCatch(SALOME_SalomeException);
246 SMESH::long_array_var aResult = new SMESH::long_array();
248 if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
249 return aResult._retn();
252 _preMeshInfo->FullLoadFromFile();
254 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
255 SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
257 int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
258 TListOfSubMeshes smList;
260 smList.push_back( aSubMeshDS );
262 // volumes are bound to shell
263 if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
265 TListOfSubMeshes::iterator sm = smList.begin();
266 for ( ; sm != smList.end(); ++sm )
267 nbElems += (*sm)->NbElements();
270 aResult->length( nbElems );
273 TListOfSubMeshes::iterator sm = smList.begin();
274 for ( int i = 0; sm != smList.end(); sm++ )
276 SMDS_ElemIteratorPtr anIt = (*sm)->GetElements();
277 for ( ; i < nbElems && anIt->more(); i++ )
278 aResult[i] = anIt->next()->GetID();
281 return aResult._retn();
285 //=============================================================================
289 //=============================================================================
291 SMESH::long_array* SMESH_subMesh_i::GetElementsByType( SMESH::ElementType theElemType )
292 throw (SALOME::SALOME_Exception)
294 Unexpect aCatch(SALOME_SalomeException);
296 SMESH::long_array_var aResult = new SMESH::long_array();
298 if ( _mesh_i->_mapSubMesh.find( _localId ) == _mesh_i->_mapSubMesh.end() )
299 return aResult._retn();
302 _preMeshInfo->FullLoadFromFile();
304 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
305 SMESHDS_SubMesh* aSubMeshDS = aSubMesh->GetSubMeshDS();
307 // PAL5440, return all nodes belonging to elements of submesh
309 int nbElems = aSubMeshDS ? aSubMeshDS->NbElements() : 0;
311 // volumes may be bound to shell instead of solid
312 TListOfSubMeshes smList;
313 if ( nbElems == 0 && getSubMeshes( aSubMesh, smList ))
315 TListOfSubMeshes::iterator sm = smList.begin();
316 for ( ; sm != smList.end(); ++sm )
318 if ( theElemType == SMESH::NODE )
320 SMDS_ElemIteratorPtr eIt = (*sm)->GetElements();
322 while ( eIt->more() ) {
323 const SMDS_MeshElement* anElem = eIt->next();
324 SMDS_ElemIteratorPtr nIt = anElem->nodesIterator();
325 while ( nIt->more() )
326 nodeIds.insert( nIt->next()->GetID() );
329 SMDS_NodeIteratorPtr nIt = (*sm)->GetNodes();
330 while ( nIt->more() )
331 nodeIds.insert( nIt->next()->GetID() );
336 nbElems += (*sm)->NbElements();
344 smList.push_back( aSubMeshDS );
347 if ( theElemType == SMESH::NODE && aSubMeshDS )
349 SMDS_ElemIteratorPtr eIt = aSubMeshDS->GetElements();
351 while ( eIt->more() ) {
352 const SMDS_MeshElement* anElem = eIt->next();
353 SMDS_ElemIteratorPtr nIt = anElem->nodesIterator();
354 while ( nIt->more() )
355 nodeIds.insert( nIt->next()->GetID() );
358 SMDS_NodeIteratorPtr nIt = aSubMeshDS->GetNodes();
359 while ( nIt->more() )
360 nodeIds.insert( nIt->next()->GetID() );
364 if ( theElemType == SMESH::NODE )
365 aResult->length( nodeIds.size() );
367 aResult->length( nbElems );
369 int i = 0, n = aResult->length();
371 if ( theElemType == SMESH::NODE && !nodeIds.empty() ) {
372 set<int>::iterator idIt = nodeIds.begin();
373 for ( ; i < n && idIt != nodeIds.end() ; i++, idIt++ )
377 if ( theElemType != SMESH::NODE ) {
378 TListOfSubMeshes::iterator sm = smList.begin();
379 for ( i = 0; sm != smList.end(); sm++ )
382 SMDS_ElemIteratorPtr anIt = aSubMeshDS->GetElements();
383 while ( i < n && anIt->more() ) {
384 const SMDS_MeshElement* anElem = anIt->next();
385 if ( theElemType == SMESH::ALL || anElem->GetType() == (SMDSAbs_ElementType)theElemType )
386 aResult[i++] = anElem->GetID();
391 aResult->length( i );
393 return aResult._retn();
396 //=============================================================================
400 //=============================================================================
402 SMESH::long_array* SMESH_subMesh_i::GetNodesId()
403 throw (SALOME::SALOME_Exception)
405 Unexpect aCatch(SALOME_SalomeException);
407 SMESH::long_array_var aResult = GetElementsByType( SMESH::NODE );
408 return aResult._retn();
411 //=============================================================================
415 //=============================================================================
417 SMESH::SMESH_Mesh_ptr SMESH_subMesh_i::GetFather()
418 throw (SALOME::SALOME_Exception)
420 Unexpect aCatch(SALOME_SalomeException);
421 return _mesh_i->_this();
424 //=============================================================================
428 //=============================================================================
430 CORBA::Long SMESH_subMesh_i::GetId()
435 //=======================================================================
436 //function : GetSubShape
438 //=======================================================================
440 GEOM::GEOM_Object_ptr SMESH_subMesh_i::GetSubShape()
441 throw (SALOME::SALOME_Exception)
443 Unexpect aCatch(SALOME_SalomeException);
444 GEOM::GEOM_Object_var aShapeObj;
446 if ( _mesh_i->_mapSubMesh.find( _localId ) != _mesh_i->_mapSubMesh.end()) {
447 TopoDS_Shape S = _mesh_i->_mapSubMesh[ _localId ]->GetSubShape();
449 aShapeObj = _gen_i->ShapeToGeomObject( S );
450 //mzn: N7PAL16232, N7PAL16233
451 //In some cases it's possible that GEOM_Client contains the shape same to S, but
452 //with another orientation.
453 if (aShapeObj->_is_nil())
454 aShapeObj = _gen_i->ShapeToGeomObject( S.Reversed() );
458 catch(SALOME_Exception & S_ex) {
459 THROW_SALOME_CORBA_EXCEPTION(S_ex.what(), SALOME::BAD_PARAM);
461 return aShapeObj._retn();
464 //=============================================================================
468 //=============================================================================
469 SMESH::long_array* SMESH_subMesh_i::GetIDs()
471 return GetElementsId();
474 //=============================================================================
478 //=============================================================================
479 SMESH::ElementType SMESH_subMesh_i::GetElementType( const CORBA::Long id, const bool iselem )
480 throw (SALOME::SALOME_Exception)
483 _preMeshInfo->FullLoadFromFile();
484 return GetFather()->GetElementType( id, iselem );
487 //=============================================================================
489 * Returns number of mesh elements of each \a EntityType
490 * @return array of number of elements per \a EntityType
492 //=============================================================================
494 SMESH::long_array* SMESH_subMesh_i::GetMeshInfo()
497 return _preMeshInfo->GetMeshInfo();
499 SMESH::long_array_var aRes = new SMESH::long_array();
500 aRes->length(SMESH::Entity_Last);
501 for (int i = SMESH::Entity_Node; i < SMESH::Entity_Last; i++)
504 // get number of nodes
505 aRes[ SMESH::Entity_Node ] = GetNumberOfNodes(true);
507 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
509 // get statistic from child sub-meshes
510 TListOfSubMeshes smList;
511 if ( getSubMeshes( aSubMesh, smList ) )
512 for ( TListOfSubMeshes::iterator sm = smList.begin(); sm != smList.end(); ++sm )
513 SMESH_Mesh_i::CollectMeshInfo( (*sm)->GetElements(), aRes );
518 //=======================================================================
520 * Returns number of mesh elements of each \a ElementType
522 //=======================================================================
524 SMESH::long_array* SMESH_subMesh_i::GetNbElementsByType()
526 SMESH::long_array_var aRes = new SMESH::long_array();
527 aRes->length(SMESH::NB_ELEMENT_TYPES);
528 for (int i = 0; i < SMESH::NB_ELEMENT_TYPES; i++)
530 aRes[ i ] = _preMeshInfo->NbElements( SMDSAbs_ElementType( i ));
536 aRes[ SMESH::NODE ] = GetNumberOfNodes(true);
538 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
539 if ( SMESHDS_SubMesh* smDS = aSubMesh->GetSubMeshDS() )
541 SMDS_ElemIteratorPtr eIt = smDS->GetElements();
543 aRes[ eIt->next()->GetType() ] = smDS->NbElements();
550 //=======================================================================
551 //function : GetTypes
552 //purpose : Returns types of elements it contains
553 //=======================================================================
555 SMESH::array_of_ElementType* SMESH_subMesh_i::GetTypes()
558 return _preMeshInfo->GetTypes();
560 SMESH::array_of_ElementType_var types = new SMESH::array_of_ElementType;
562 ::SMESH_subMesh* aSubMesh = _mesh_i->_mapSubMesh[_localId];
563 if ( SMESHDS_SubMesh* smDS = aSubMesh->GetSubMeshDS() )
565 SMDS_ElemIteratorPtr eIt = smDS->GetElements();
569 types[0] = SMESH::ElementType( eIt->next()->GetType());
571 else if ( smDS->GetNodes()->more() )
573 TopoDS_Shape shape = aSubMesh->GetSubShape();
574 while ( !shape.IsNull() && shape.ShapeType() == TopAbs_COMPOUND )
576 TopoDS_Iterator it( shape );
577 shape = it.More() ? it.Value() : TopoDS_Shape();
579 if ( !shape.IsNull() && shape.ShapeType() == TopAbs_VERTEX )
582 types[0] = SMESH::NODE;
586 return types._retn();
589 //=======================================================================
591 //purpose : interface SMESH_IDSource
592 //=======================================================================
594 SMESH::SMESH_Mesh_ptr SMESH_subMesh_i::GetMesh()
599 //=======================================================================
600 //function : IsMeshInfoCorrect
601 //purpose : * Returns false if GetMeshInfo() returns incorrect information that may
602 // * happen if mesh data is not yet fully loaded from the file of study.
603 //=======================================================================
605 bool SMESH_subMesh_i::IsMeshInfoCorrect()
607 return _preMeshInfo ? _preMeshInfo->IsMeshInfoCorrect() : true;