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 SMESHDS : management of mesh data and SMESH document
24 // File : SMESH_SubMesh.cxx
25 // Author : Yves FRICAUD, OCC
29 #include "SMESHDS_SubMesh.hxx"
30 #include "SMESHDS_Mesh.hxx"
32 #include "utilities.h"
33 #include "SMDS_SetIterator.hxx"
40 //================================================================================
44 //================================================================================
46 SMESHDS_SubMesh::SMESHDS_SubMesh(SMESHDS_Mesh *parent, int index)
53 myUnusedIdElements = 0;
56 //================================================================================
60 //================================================================================
62 SMESHDS_SubMesh::~SMESHDS_SubMesh()
66 //=======================================================================
67 //function : AddElement
69 //=======================================================================
70 void SMESHDS_SubMesh::AddElement(const SMDS_MeshElement * ME)
72 if (!IsComplexSubmesh())
74 //MESSAGE("in " << myIndex << " AddElement "<< ME->GetID());
75 int oldShapeId = ME->getshapeId();
78 if (oldShapeId != myIndex)
80 MESSAGE("add element in subshape already belonging to another subshape "
81 << ME->GetID() << " " << oldShapeId << " " << myIndex);
82 throw SALOME_Exception(LOCALIZED("add element in subshape already belonging to a subshape"));
86 int idInSubShape = ME->getIdInShape();
87 if (idInSubShape >= 0)
89 MESSAGE("add element in subshape already belonging to that subshape "
90 << ME->GetID() << " " << oldShapeId << " " << idInSubShape);
91 // check if ok: do nothing if ok
92 if (idInSubShape >= myElements.size())
94 MESSAGE("out of bounds " << idInSubShape << " " << myElements.size());
95 throw SALOME_Exception(LOCALIZED("out of bounds"));
97 if (ME != myElements[idInSubShape])
99 MESSAGE("not the same element");
100 throw SALOME_Exception(LOCALIZED("not the same element"));
102 MESSAGE("already done, OK, nothing to do");
108 SMDS_MeshElement* elem = (SMDS_MeshElement*) (ME);
109 elem->setShapeId(myIndex);
110 elem->setIdInShape(myElements.size());
111 myElements.push_back(ME);
115 //=======================================================================
116 //function : RemoveElement
118 //=======================================================================
119 bool SMESHDS_SubMesh::RemoveElement(const SMDS_MeshElement * ME, bool isElemDeleted)
123 MESSAGE("-----------------> Remove Null Element " << isElemDeleted);
126 //MESSAGE("-----------------> RemoveElement "<< ME->GetID() << " " << isElemDeleted);
127 if (!IsComplexSubmesh())
129 // if (!isElemDeleted) // alive element has valid ID and can be found
131 int idInSubShape = ME->getIdInShape();
132 //MESSAGE("in "<< myIndex << " RemoveElement " << ME->GetID() << " " << idInSubShape << " " << myUnusedIdElements);
133 SMDS_MeshElement* elem = (SMDS_MeshElement*) (ME);
135 elem->setIdInShape(-1);
136 if ((idInSubShape >= 0) && (idInSubShape < myElements.size()))
138 myElements[idInSubShape] = 0; // this vector entry is no more used
139 myUnusedIdElements++;
145 MESSAGE("Try to remove an element from a complex submesh ");
149 //=======================================================================
152 //=======================================================================
153 void SMESHDS_SubMesh::AddNode(const SMDS_MeshNode * N)
155 if ( !IsComplexSubmesh() )
157 int idInSubShape = N->getIdInShape();
158 int shapeId = N->getshapeId();
159 if ((shapeId > 0) && (idInSubShape >= 0))
161 MESSAGE("========== AddNode already belonging to other subShape " << N->GetID());
162 // OK for vertex nodes
163 //this->getParent()->UnSetNodeOnShape(N);
165 SMDS_MeshNode* node = (SMDS_MeshNode*)(N);
166 node->setShapeId(myIndex);
167 node->setIdInShape(myNodes.size());
168 myNodes.push_back(N);
169 //MESSAGE("in "<< myIndex << " AddNode " << node->GetID());
171 //MESSAGE("try to add node in a complex submesh " << N->GetID());
174 //=======================================================================
175 //function : RemoveNode
177 //=======================================================================
179 bool SMESHDS_SubMesh::RemoveNode(const SMDS_MeshNode * N, bool isNodeDeleted)
181 if (!IsComplexSubmesh())
183 // if (!isNodeDeleted) // alive node has valid ID and can be found
185 int idInSubShape = N->getIdInShape();
186 //int shapeId = N->getshapeId();
187 //MESSAGE("in "<< myIndex << " RemoveNode " << shapeId << " " << idInSubShape << " " << N->GetID());
188 SMDS_MeshNode* node = (SMDS_MeshNode*) (N);
190 node->setIdInShape(-1);
191 if ((idInSubShape >= 0) && (idInSubShape < myNodes.size()))
193 myNodes[idInSubShape] = 0; // this vector entry is no more used
200 MESSAGE("Try to remove a node from a complex submesh");
204 //=======================================================================
205 //function : NbElements
207 //=======================================================================
208 int SMESHDS_SubMesh::NbElements() const
210 //MESSAGE(this << " NbElements " << IsComplexSubmesh() << " " << myElements.size() - myUnusedIdElements);
211 if ( !IsComplexSubmesh() )
212 return myElements.size() - myUnusedIdElements;
215 set<const SMESHDS_SubMesh*>::const_iterator it = mySubMeshes.begin();
216 for ( ; it != mySubMeshes.end(); it++ )
217 nbElems += (*it)->NbElements();
222 //=======================================================================
225 //=======================================================================
227 int SMESHDS_SubMesh::NbNodes() const
229 //MESSAGE(this << " NbNodes " << IsComplexSubmesh() << " " << myNodes.size() - myUnusedIdNodes);
230 if ( !IsComplexSubmesh() )
231 return myNodes.size() - myUnusedIdNodes;
234 set<const SMESHDS_SubMesh*>::const_iterator it = mySubMeshes.begin();
235 for ( ; it != mySubMeshes.end(); it++ )
236 nbElems += (*it)->NbNodes();
242 * template class used for iteration on submesh elements. Interface of iterator remains
243 * unchanged after redesign of SMDS to avoid modification everywhere in SMESH.
244 * instances are stored in shared_ptr for automatic destruction.
245 * Container is copied for iteration, because original can be modified
246 * by addition of elements, for instance, and then reallocated (vector)
248 template <class ELEM, typename TSET> class MySetIterator : public SMDS_Iterator<ELEM>
251 typename TSET::const_iterator _it, _end;
254 MySetIterator(const TSET& table)
257 _it = _table.begin();
259 while ((_it != _end) && (*_it == 0))
265 while ((_it != _end) && (*_it == 0))
267 return (_it != _end);
278 // =====================
280 // =====================
282 template<typename VALUE> class MyIterator : public SMDS_Iterator<VALUE>
285 MyIterator (const set<const SMESHDS_SubMesh*>& theSubMeshes)
286 : mySubIt( theSubMeshes.begin() ), mySubEnd( theSubMeshes.end() ), myMore(false)
290 while (( !myElemIt.get() || !myElemIt->more() ) && mySubIt != mySubEnd)
292 myElemIt = getElements(*mySubIt);
295 myMore = myElemIt.get() && myElemIt->more();
302 elem = myElemIt->next();
306 virtual boost::shared_ptr< SMDS_Iterator<VALUE> >
307 getElements(const SMESHDS_SubMesh*) const = 0;
311 set<const SMESHDS_SubMesh*>::const_iterator mySubIt, mySubEnd;
312 boost::shared_ptr< SMDS_Iterator<VALUE> > myElemIt;
315 // =====================
316 // class MyElemIterator
317 // =====================
319 class MyElemIterator: public MyIterator<const SMDS_MeshElement*>
322 MyElemIterator (const set<const SMESHDS_SubMesh*>& theSubMeshes)
323 :MyIterator<const SMDS_MeshElement*>( theSubMeshes ) {}
324 SMDS_ElemIteratorPtr getElements(const SMESHDS_SubMesh* theSubMesh) const
325 { return theSubMesh->GetElements(); }
328 // =====================
329 // class MyNodeIterator
330 // =====================
332 class MyNodeIterator: public MyIterator<const SMDS_MeshNode*>
335 MyNodeIterator (const set<const SMESHDS_SubMesh*>& theSubMeshes)
336 :MyIterator<const SMDS_MeshNode*>( theSubMeshes ) {}
337 SMDS_NodeIteratorPtr getElements(const SMESHDS_SubMesh* theSubMesh) const
338 { return theSubMesh->GetNodes(); }
341 //=======================================================================
342 //function : GetElements
344 //=======================================================================
346 SMDS_ElemIteratorPtr SMESHDS_SubMesh::GetElements() const
348 if ( IsComplexSubmesh() )
349 return SMDS_ElemIteratorPtr( new MyElemIterator( mySubMeshes ));
350 return SMDS_ElemIteratorPtr(new MySetIterator<const SMDS_MeshElement*, std::vector<const SMDS_MeshElement*> >(myElements));
353 //=======================================================================
354 //function : GetNodes
356 //=======================================================================
358 SMDS_NodeIteratorPtr SMESHDS_SubMesh::GetNodes() const
360 if ( IsComplexSubmesh() )
361 return SMDS_NodeIteratorPtr( new MyNodeIterator( mySubMeshes ));
363 return SMDS_NodeIteratorPtr(new MySetIterator<const SMDS_MeshNode*, std::vector<const SMDS_MeshNode*> >(myNodes));
366 //=======================================================================
367 //function : Contains
368 //purpose : check if elem or node is in
369 //=======================================================================
371 bool SMESHDS_SubMesh::Contains(const SMDS_MeshElement * ME) const
373 // DO NOT TRY TO FIND A REMOVED ELEMENT !!
374 //if ( IsComplexSubmesh() || !ME )
378 if (IsComplexSubmesh())
380 set<const SMESHDS_SubMesh*>::const_iterator aSubIt = mySubMeshes.begin();
381 for (; aSubIt != mySubMeshes.end(); aSubIt++)
382 if ((*aSubIt)->Contains(ME))
387 if (ME->GetType() == SMDSAbs_Node)
389 int idInShape = ME->getIdInShape();
390 if ((idInShape >= 0) && (idInShape < myNodes.size()))
391 if (myNodes[idInShape] == ME)
396 int idInShape = ME->getIdInShape();
397 if ((idInShape >= 0) && (idInShape < myElements.size()))
398 if (myElements[idInShape] == ME)
404 //=======================================================================
405 //function : AddSubMesh
407 //=======================================================================
409 void SMESHDS_SubMesh::AddSubMesh( const SMESHDS_SubMesh* theSubMesh )
411 ASSERT( theSubMesh );
412 mySubMeshes.insert( theSubMesh );
415 //=======================================================================
416 //function : RemoveSubMesh
418 //=======================================================================
420 bool SMESHDS_SubMesh::RemoveSubMesh( const SMESHDS_SubMesh* theSubMesh )
422 return mySubMeshes.erase( theSubMesh );
425 //=======================================================================
426 //function : RemoveAllSubmeshes
428 //=======================================================================
430 void SMESHDS_SubMesh::RemoveAllSubmeshes()
435 //=======================================================================
436 //function : ContainsSubMesh
438 //=======================================================================
440 bool SMESHDS_SubMesh::ContainsSubMesh( const SMESHDS_SubMesh* theSubMesh ) const
442 return mySubMeshes.find( theSubMesh ) != mySubMeshes.end();
445 //=======================================================================
446 //function : GetSubMeshIterator
448 //=======================================================================
450 SMESHDS_SubMeshIteratorPtr SMESHDS_SubMesh::GetSubMeshIterator() const
452 typedef set<const SMESHDS_SubMesh*>::const_iterator TIterator;
453 return SMESHDS_SubMeshIteratorPtr
454 ( new SMDS_SetIterator< const SMESHDS_SubMesh*, TIterator >( mySubMeshes.begin(),
458 //=======================================================================
460 //purpose : remove the contents
461 //=======================================================================
463 void SMESHDS_SubMesh::Clear()
468 myUnusedIdElements = 0;
469 SMESHDS_SubMeshIteratorPtr sub = GetSubMeshIterator();
470 while ( sub->more() ) {
471 if ( SMESHDS_SubMesh* sm = (SMESHDS_SubMesh*) sub->next())
476 int SMESHDS_SubMesh::getSize()
479 int d = NbElements();
480 //cerr << "SMESHDS_SubMesh::NbNodes " << c << endl;
481 //cerr << "SMESHDS_SubMesh::NbElements " << d << endl;
485 void SMESHDS_SubMesh::compactList()
487 //MESSAGE("compactList old: nodes " << myNodes.size() << " elements " << myElements.size());
493 std::vector<const SMDS_MeshElement*> newElems;
495 for (int i = 0; i < myElements.size(); i++)
498 SMDS_MeshElement* elem = (SMDS_MeshElement*)myElements[i];
499 elem->setIdInShape(newElems.size());
500 newElems.push_back(elem);
501 //a << elem->GetID() << " ";
502 //b << elem->GetID() << " ";
506 myElements.swap(newElems);
507 myUnusedIdElements = 0;
508 //MESSAGE("in " << myIndex << " oldElems " << a.str());
509 //MESSAGE("in " << myIndex << " newElems " << b.str());
511 std::vector<const SMDS_MeshNode*> newNodes;
513 for (int i = 0; i < myNodes.size(); i++)
516 SMDS_MeshNode* node = (SMDS_MeshNode*)myNodes[i];
517 node->setIdInShape(newNodes.size());
518 newNodes.push_back(node);
519 //c << node->GetID() << " ";
520 //d << node->GetID() << " ";
524 myNodes.swap(newNodes);
526 //MESSAGE("in " << myIndex << " oldNodes " << c.str());
527 //MESSAGE("in " << myIndex << " newNodes " << d.str());
528 //MESSAGE("compactList new: nodes " << myNodes.size() << " elements " << myElements.size());