#define _SMESHDS_SubMesh_HeaderFile
#include "SMDS_Mesh.hxx"
-#include <vector>
#include <set>
-using namespace std;
-
class SMESHDS_SubMesh
{
- public:
- SMESHDS_SubMesh(const SMDS_Mesh * M);
- void AddElement(const SMDS_MeshElement * ME);
- void RemoveElement(const SMDS_MeshElement * ME);
- void AddNode(const SMDS_MeshNode * ME);
- void RemoveNode(const SMDS_MeshNode * ME);
- int NbElements() const;
- const set<const SMDS_MeshElement*> & GetElements();
- const vector<int> & GetIDElements();
- int NbNodes() const;
- const set<const SMDS_MeshNode*> & GetNodes() const;
- const vector<int> & GetIDNodes();
- ~SMESHDS_SubMesh();
- private:
- const SMDS_Mesh * myMesh;
- set<const SMDS_MeshElement*> myElements;
- set<const SMDS_MeshNode*> myNodes;
- bool myListOfEltIDIsUpdate;
- vector<int> myListOfEltID;
- bool myListOfNodeIDIsUpdate;
- vector<int> myListOfNodeID;
+ public:
+
+ bool IsComplexSubmesh() const { return !mySubMeshes.empty(); }
+
+ // if !IsComplexSubmesh()
+ void AddElement(const SMDS_MeshElement * ME);
+ bool RemoveElement(const SMDS_MeshElement * ME); // ret true if ME was in
+ void AddNode(const SMDS_MeshNode * ME);
+ bool RemoveNode(const SMDS_MeshNode * ME); // ret true if ME was in
+
+ // if IsComplexSubmesh()
+ void AddSubMesh( const SMESHDS_SubMesh* theSubMesh );
+ bool RemoveSubMesh( const SMESHDS_SubMesh* theSubMesh );
+ bool ContainsSubMesh( const SMESHDS_SubMesh* theSubMesh ) const;
+
+ // for both types
+ int NbElements() const;
+ SMDS_ElemIteratorPtr GetElements() const;
+ int NbNodes() const;
+ SMDS_NodeIteratorPtr GetNodes() const;
+ bool Contains(const SMDS_MeshElement * ME) const; // check if elem or node is in
+
+ private:
+ //const SMDS_Mesh * myMesh;
+ std::set<const SMDS_MeshElement*> myElements;
+ std::set<const SMDS_MeshNode*> myNodes;
+ std::set<const SMESHDS_SubMesh*> mySubMeshes;
};
#endif
