src/SMDS/SMDS_MeshGroup.hxx

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