// $Header:
using namespace std;
-#include "SMESHDS_SubMesh.ixx"
-#include "SMDS_MapIteratorOfExtendedMap.hxx"
-
-//=======================================================================
-//function : SMESHDS_SubMesh
-//purpose :
-//=======================================================================
-SMESHDS_SubMesh::SMESHDS_SubMesh(const Handle(SMDS_Mesh)& M) : myMesh(M)
-{
- myListOfEltIDIsUpdate = Standard_False;
- myListOfNodeIDIsUpdate = Standard_False;
-}
+#include "SMESHDS_SubMesh.hxx"
//=======================================================================
//function : AddElement
//purpose :
//=======================================================================
-void SMESHDS_SubMesh::AddElement (const Handle(SMDS_MeshElement)& ME)
+void SMESHDS_SubMesh::AddElement(const SMDS_MeshElement * ME)
{
- myElements.Add(ME);
- myListOfEltIDIsUpdate = Standard_False;
+ myElements.insert(ME);
}
//=======================================================================
//function : RemoveElement
//purpose :
//=======================================================================
-void SMESHDS_SubMesh::RemoveElement(const Handle(SMDS_MeshElement)& ME)
+void SMESHDS_SubMesh::RemoveElement(const SMDS_MeshElement * ME)
{
- myElements.Remove(ME);
- myListOfEltIDIsUpdate = Standard_False;
+ myElements.erase(ME);
}
+
//=======================================================================
//function : AddNode
//purpose :
//=======================================================================
-void SMESHDS_SubMesh::AddNode (const Handle(SMDS_MeshNode)& N)
+void SMESHDS_SubMesh::AddNode(const SMDS_MeshNode * N)
{
- myNodes.Add(N);
- myListOfNodeIDIsUpdate = Standard_False;
+ myNodes.insert(N);
}
//=======================================================================
//function : RemoveNode
//purpose :
//=======================================================================
-void SMESHDS_SubMesh::RemoveNode (const Handle(SMDS_MeshNode)& N)
+void SMESHDS_SubMesh::RemoveNode(const SMDS_MeshNode * N)
{
- myNodes.Remove(N);
- myListOfNodeIDIsUpdate = Standard_False;
+ myNodes.erase(N);
}
//=======================================================================
//function : NbElements
//purpose :
//=======================================================================
-Standard_Integer SMESHDS_SubMesh::NbElements()
+int SMESHDS_SubMesh::NbElements() const
{
- return myElements.Extent();
+ return myElements.size();
}
-//=======================================================================
-//function : GetElements
-//purpose :
-//=======================================================================
-const SMDS_MapOfMeshElement& SMESHDS_SubMesh::GetElements()
-{
- return myElements;
-}
//=======================================================================
//function : NbNodes
//purpose :
//=======================================================================
-Standard_Integer SMESHDS_SubMesh::NbNodes()
+int SMESHDS_SubMesh::NbNodes() const
{
- return myNodes.Extent();
+ return myNodes.size();
}
-//=======================================================================
-//function : GetNodes
-//purpose :
-//=======================================================================
-const SMDS_MapOfMeshElement& SMESHDS_SubMesh::GetNodes()
+template<typename T> class MySetIterator:public SMDS_Iterator<const T*>
{
- return myNodes;
-}
+ const set<const T*>& mySet;
+ set<const T*>::const_iterator myIt;
-//=======================================================================
-//function : GetIDElements
-//purpose :
-//=======================================================================
-const TColStd_ListOfInteger& SMESHDS_SubMesh::GetIDElements()
+ public:
+ MySetIterator(const set<const T*>& s):mySet(s), myIt(s.begin())
+ {
+ }
+
+ bool more()
+ {
+ return myIt!=mySet.end();
+ }
+ const T* next()
+ {
+ const T* t=*myIt;
+ myIt++;
+ return t;
+ }
+};
+///////////////////////////////////////////////////////////////////////////////
+///Return an iterator on the elements of submesh
+///The created iterator must be free by the caller
+///////////////////////////////////////////////////////////////////////////////
+SMDS_Iterator<const SMDS_MeshElement*> * SMESHDS_SubMesh::GetElements() const
{
- if (!myListOfEltIDIsUpdate) {
- myListOfEltID.Clear();
- for (SMDS_MapIteratorOfExtendedMap it(myElements); it.More(); it.Next()) {
- myListOfEltID.Append(it.Key()->GetID());
- }
- myListOfEltIDIsUpdate = Standard_True;
- }
- return myListOfEltID;
+ return new MySetIterator<SMDS_MeshElement>(myElements);
}
-//=======================================================================
-//function : GetIDNodes
-//purpose :
-//=======================================================================
-const TColStd_ListOfInteger& SMESHDS_SubMesh::GetIDNodes()
+///////////////////////////////////////////////////////////////////////////////
+///Return an iterator on the nodes of submesh
+///The created iterator must be free by the caller
+///////////////////////////////////////////////////////////////////////////////
+SMDS_Iterator<const SMDS_MeshNode*> * SMESHDS_SubMesh::GetNodes() const
{
- if (!myListOfNodeIDIsUpdate) {
- myListOfNodeID.Clear();
- for (SMDS_MapIteratorOfExtendedMap it(myNodes); it.More(); it.Next()) {
- myListOfNodeID.Append(it.Key()->GetID());
- }
- myListOfNodeIDIsUpdate = Standard_True;
- }
- return myListOfNodeID;
+ return new MySetIterator<SMDS_MeshNode>(myNodes);
}
+
