Salome HOME
Adjust to CVS HEAD modifications
[modules/smesh.git] / src / SMDS / SMDS_MeshGroup.hxx
index f73c1dd1f4a79bc51ab36f6b070e0c755940ec38..37893fab42bb46d6e4405b38d9e7f6f659f21165 100644 (file)
 
 #include "SMDS_Mesh.hxx"
 #include <set>
-using namespace std;
 
-class SMDS_MeshGroup:public SMDS_MeshObject
+//#ifdef WNT
+//#include <SALOME_WNT.hxx>
+//#else
+//#define SALOME_WNT_EXPORT
+//#endif
+
+#if defined WNT && defined WIN32 && defined SMDS_EXPORTS
+#define SMDS_WNT_EXPORT __declspec( dllexport )
+#else
+#define SMDS_WNT_EXPORT
+#endif
+
+class SMDS_WNT_EXPORT 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<TIterator&>(myIterator) = myElements.begin(); }
+
+        bool More() const { return myIterator != myElements.end(); }
+
+        const SMDS_MeshElement* Next() const
+        { return *(const_cast<TIterator&>(myIterator))++; }
+
   private:
-       SMDS_MeshGroup(SMDS_MeshGroup* parent);
-       const SMDS_Mesh *myMesh;
-       SMDSAbs_ElementType myType;
-       set<const SMDS_MeshElement *> myElements;
-       SMDS_MeshGroup * myParent;
-       list<const SMDS_MeshGroup*> myChildren;
+       SMDS_MeshGroup(SMDS_MeshGroup* theParent,
+                       const SMDSAbs_ElementType theType = SMDSAbs_All);
+
+        typedef std::set<const SMDS_MeshElement *>::const_iterator TIterator;
+       const SMDS_Mesh *                       myMesh;
+       SMDSAbs_ElementType                     myType;
+       std::set<const SMDS_MeshElement *>      myElements;
+       SMDS_MeshGroup *                        myParent;
+       std::list<const SMDS_MeshGroup*>        myChildren;
+        TIterator                               myIterator;
 };
 #endif