X-Git-Url: http://git.salome-platform.org/gitweb/?p=modules%2Fsmesh.git;a=blobdiff_plain;f=src%2FSMDS%2FSMDS_MeshGroup.hxx;h=c5bfcb43216dcdf8dfbd8a5e10a11381224831a6;hp=f73c1dd1f4a79bc51ab36f6b070e0c755940ec38;hb=c3bf92bd87b770fd81631a3853f7f5bb1ac6a4e8;hpb=13b8c20dfc9f5bedf2dd4699ac1e05a8d4227791 diff --git a/src/SMDS/SMDS_MeshGroup.hxx b/src/SMDS/SMDS_MeshGroup.hxx index f73c1dd1f..c5bfcb432 100644 --- a/src/SMDS/SMDS_MeshGroup.hxx +++ b/src/SMDS/SMDS_MeshGroup.hxx @@ -33,25 +33,44 @@ using namespace std; class SMDS_MeshGroup:public SMDS_MeshObject { public: - SMDS_MeshGroup(const SMDS_Mesh * aMesh); - const SMDS_MeshGroup * AddSubGroup(); - virtual bool RemoveSubGroup(const SMDS_MeshGroup* aGroup); + SMDS_MeshGroup(const SMDS_Mesh * theMesh, + const SMDSAbs_ElementType theType = SMDSAbs_All); + const SMDS_MeshGroup * AddSubGroup + (const SMDSAbs_ElementType theType = SMDSAbs_All); + virtual bool RemoveSubGroup(const SMDS_MeshGroup* theGroup); virtual bool RemoveFromParent(); - void Clear(); - void Add(const SMDS_MeshElement * ME); - void Remove(const SMDS_MeshElement * ME); - bool IsEmpty() const; - int Extent() const; - SMDSAbs_ElementType Type() const; - bool Contains(const SMDS_MeshElement * ME) const; - - ~SMDS_MeshGroup(); + + const SMDS_Mesh* GetMesh() const { return myMesh; } + + void SetType (const SMDSAbs_ElementType theType); + void Clear(); + void Add(const SMDS_MeshElement * theElem); + void Remove(const SMDS_MeshElement * theElem); + bool IsEmpty() const { return myElements.empty(); } + int Extent() const { return myElements.size(); } + + SMDSAbs_ElementType GetType() const { return myType; } + + bool Contains(const SMDS_MeshElement * theElem) const; + + void InitIterator() const + { const_cast(myIterator) = myElements.begin(); } + + bool More() const { return myIterator != myElements.end(); } + + const SMDS_MeshElement* Next() const + { return *(const_cast(myIterator))++; } + private: - SMDS_MeshGroup(SMDS_MeshGroup* parent); - const SMDS_Mesh *myMesh; - SMDSAbs_ElementType myType; - set myElements; - SMDS_MeshGroup * myParent; - list myChildren; + SMDS_MeshGroup(SMDS_MeshGroup* theParent, + const SMDSAbs_ElementType theType = SMDSAbs_All); + + typedef set::iterator iterator; + const SMDS_Mesh * myMesh; + SMDSAbs_ElementType myType; + set myElements; + SMDS_MeshGroup * myParent; + list myChildren; + iterator myIterator; }; #endif