src/SMDS/SMDS_MeshGroup.hxx

   1 //  SMESH SMDS : implementaion of Salome mesh data structure
   2 //
   3 //  Copyright (C) 2003  OPEN CASCADE
   4 //
   5 //  This library is free software; you can redistribute it and/or
   6 //  modify it under the terms of the GNU Lesser General Public
   7 //  License as published by the Free Software Foundation; either
   8 //  version 2.1 of the License.
   9 //
  10 //  This library is distributed in the hope that it will be useful,
  11 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 //  Lesser General Public License for more details.
  14 //
  15 //  You should have received a copy of the GNU Lesser General Public
  16 //  License along with this library; if not, write to the Free Software
  17 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18 //
  19 //  See http://www.opencascade.org or email : webmaster@opencascade.org
  20 //
  21 //
  22 //
  23 //  File   : SMDS_MeshGroup.hxx
  24 //  Module : SMESH
  25
  26 #ifndef _SMDS_MeshGroup_HeaderFile
  27 #define _SMDS_MeshGroup_HeaderFile
  28
  29 #include "SMDS_Mesh.hxx"
  30 #include <set>
  31
  32 //#ifdef WNT
  33 //#include <SALOME_WNT.hxx>
  34 //#else
  35 //#define SALOME_WNT_EXPORT
  36 //#endif
  37
  38 #if defined WNT && defined WIN32 && defined SMDS_EXPORTS
  39 #define SMDS_WNT_EXPORT __declspec( dllexport )
  40 #else
  41 #define SMDS_WNT_EXPORT
  42 #endif
  43
  44 class SMDS_WNT_EXPORT SMDS_MeshGroup:public SMDS_MeshObject
  45 {
  46   public:
  47         SMDS_MeshGroup(const SMDS_Mesh * theMesh,
  48                        const SMDSAbs_ElementType theType = SMDSAbs_All);
  49         const SMDS_MeshGroup * AddSubGroup
  50                       (const SMDSAbs_ElementType theType = SMDSAbs_All);
  51         virtual bool RemoveSubGroup(const SMDS_MeshGroup* theGroup);
  52         virtual bool RemoveFromParent();
  53
  54         const SMDS_Mesh* GetMesh() const { return myMesh; }
  55
  56         void SetType (const SMDSAbs_ElementType theType);
  57         void Clear();
  58         void Add(const SMDS_MeshElement * theElem);
  59         void Remove(const SMDS_MeshElement * theElem);
  60         bool IsEmpty() const { return myElements.empty(); }
  61         int Extent() const { return myElements.size(); }
  62
  63         SMDSAbs_ElementType GetType() const { return myType; }
  64
  65         bool Contains(const SMDS_MeshElement * theElem) const;
  66
  67         void InitIterator() const
  68         { const_cast<TIterator&>(myIterator) = myElements.begin(); }
  69
  70         bool More() const { return myIterator != myElements.end(); }
  71
  72         const SMDS_MeshElement* Next() const
  73         { return *(const_cast<TIterator&>(myIterator))++; }
  74
  75   private:
  76         SMDS_MeshGroup(SMDS_MeshGroup* theParent,
  77                        const SMDSAbs_ElementType theType = SMDSAbs_All);
  78
  79         typedef std::set<const SMDS_MeshElement *>::const_iterator TIterator;
  80         const SMDS_Mesh *                       myMesh;
  81         SMDSAbs_ElementType                     myType;
  82         std::set<const SMDS_MeshElement *>      myElements;
  83         SMDS_MeshGroup *                        myParent;
  84         std::list<const SMDS_MeshGroup*>        myChildren;
  85         TIterator                               myIterator;
  86 };
  87 #endif