-#ifndef _Standard_HeaderFile
-#include <Standard.hxx>
-#endif
-#ifndef _Handle_SMESHDS_SubMesh_HeaderFile
-#include "Handle_SMESHDS_SubMesh.hxx"
-#endif
-
-#ifndef _Handle_SMDS_Mesh_HeaderFile
-#include "Handle_SMDS_Mesh.hxx"
-#endif
-#ifndef _SMDS_MapOfMeshElement_HeaderFile
-#include "SMDS_MapOfMeshElement.hxx"
-#endif
-#ifndef _Standard_Boolean_HeaderFile
-#include <Standard_Boolean.hxx>
-#endif
-#ifndef _TColStd_ListOfInteger_HeaderFile
-#include <TColStd_ListOfInteger.hxx>
-#endif
-#ifndef _MMgt_TShared_HeaderFile
-#include <MMgt_TShared.hxx>
-#endif
-#ifndef _Handle_SMDS_MeshElement_HeaderFile
-#include "Handle_SMDS_MeshElement.hxx"
-#endif
-#ifndef _Handle_SMDS_MeshNode_HeaderFile
-#include "Handle_SMDS_MeshNode.hxx"
-#endif
-#ifndef _Standard_Integer_HeaderFile
-#include <Standard_Integer.hxx>
-#endif
-class SMDS_Mesh;
-class SMDS_MeshElement;
-class SMDS_MeshNode;
-class SMDS_MapOfMeshElement;
-class TColStd_ListOfInteger;
-
-
-class SMESHDS_SubMesh : public MMgt_TShared {
-
-public:
-
- inline void* operator new(size_t,void* anAddress)
- {
- return anAddress;
- }
- inline void* operator new(size_t size)
- {
- return Standard::Allocate(size);
- }
- inline void operator delete(void *anAddress)
- {
- if (anAddress) Standard::Free((Standard_Address&)anAddress);
- }
-// inline void operator delete(void *anAddress, size_t size)
-// {
-// if (anAddress) Standard::Free((Standard_Address&)anAddress,size);
-// }
- // Methods PUBLIC
- //
-Standard_EXPORT SMESHDS_SubMesh(const Handle(SMDS_Mesh)& M);
-Standard_EXPORT void AddElement(const Handle(SMDS_MeshElement)& ME) ;
-Standard_EXPORT void RemoveElement(const Handle(SMDS_MeshElement)& ME) ;
-Standard_EXPORT void AddNode(const Handle(SMDS_MeshNode)& ME) ;
-Standard_EXPORT void RemoveNode(const Handle(SMDS_MeshNode)& ME) ;
-Standard_EXPORT Standard_Integer NbElements() ;
-Standard_EXPORT const SMDS_MapOfMeshElement& GetElements() ;
-Standard_EXPORT const TColStd_ListOfInteger& GetIDElements() ;
-Standard_EXPORT Standard_Integer NbNodes() ;
-Standard_EXPORT const SMDS_MapOfMeshElement& GetNodes() ;
-Standard_EXPORT const TColStd_ListOfInteger& GetIDNodes() ;
-Standard_EXPORT ~SMESHDS_SubMesh();
-
-
-
-
- // Type management
- //
- Standard_EXPORT friend Handle_Standard_Type& SMESHDS_SubMesh_Type_();
- Standard_EXPORT const Handle(Standard_Type)& DynamicType() const;
- Standard_EXPORT Standard_Boolean IsKind(const Handle(Standard_Type)&) const;
-
-protected:
-
- // Methods PROTECTED
- //
-
-
- // Fields PROTECTED
- //
-
-
-private:
-
- // Methods PRIVATE
- //
-
-
- // Fields PRIVATE
- //
-Handle_SMDS_Mesh myMesh;
-SMDS_MapOfMeshElement myElements;
-SMDS_MapOfMeshElement myNodes;
-Standard_Boolean myListOfEltIDIsUpdate;
-TColStd_ListOfInteger myListOfEltID;
-Standard_Boolean myListOfNodeIDIsUpdate;
-TColStd_ListOfInteger myListOfNodeID;
-
-
+#include "SMESH_SMESHDS.hxx"
+
+#include "SMDS_Mesh.hxx"
+#include <set>
+#include <vector>
+
+class SMESHDS_SubMesh;
+typedef SMDS_Iterator<const SMESHDS_SubMesh*> SMESHDS_SubMeshIterator;
+typedef boost::shared_ptr< SMESHDS_SubMeshIterator > SMESHDS_SubMeshIteratorPtr;
+
+class SMESHDS_Mesh;
+
+class SMESHDS_EXPORT SMESHDS_SubMesh
+{
+ public:
+ SMESHDS_SubMesh(SMESHDS_Mesh *parent, int index);
+ virtual ~SMESHDS_SubMesh();
+
+ virtual bool IsComplexSubmesh() const { return !mySubMeshes.empty(); }
+
+ // if !IsComplexSubmesh()
+ virtual void AddElement(const SMDS_MeshElement * ME);
+ virtual bool RemoveElement(const SMDS_MeshElement * ME, bool isElemDeleted); // ret true if ME was in
+ virtual void AddNode(const SMDS_MeshNode * ME);
+ virtual bool RemoveNode(const SMDS_MeshNode * ME, bool isNodeDeleted); // ret true if ME was in
+
+ // if IsComplexSubmesh()
+ void AddSubMesh( const SMESHDS_SubMesh* theSubMesh );
+ bool RemoveSubMesh( const SMESHDS_SubMesh* theSubMesh );
+ void RemoveAllSubmeshes();
+ bool ContainsSubMesh( const SMESHDS_SubMesh* theSubMesh ) const;
+ int NbSubMeshes() const { return mySubMeshes.size(); }
+ SMESHDS_SubMeshIteratorPtr GetSubMeshIterator() const;
+
+ // for both types
+ virtual int NbElements() const;
+ virtual SMDS_ElemIteratorPtr GetElements() const;
+ virtual int NbNodes() const;
+ virtual SMDS_NodeIteratorPtr GetNodes() const;
+ virtual bool Contains(const SMDS_MeshElement * ME) const; // check if elem or node is in
+
+ // clear the contents
+ virtual void Clear();
+ int getSize();
+ void compactList();
+
+ SMESHDS_Mesh *GetParent() { return myParent; }
+ int GetID() const { return myIndex; }
+
+ private:
+ SMESHDS_Mesh * myParent;
+ std::vector<const SMDS_MeshElement*> myElements;
+ std::vector<const SMDS_MeshNode*> myNodes;
+
+ int myUnusedIdNodes;
+ int myUnusedIdElements;
+ int myIndex;
+
+ std::set<const SMESHDS_SubMesh*> mySubMeshes;