src/SMESHDS/SMESHDS_Group.cxx

   1 // Copyright (C) 2007-2014  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, or (at your option) any later version.
  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 SMESHDS : idl implementation based on 'SMESH' unit's classes
  24 //  File   : SMESHDS_Group.cxx
  25 //  Module : SMESH
  26 //  $Header$
  27 //
  28 #include "SMESHDS_Group.hxx"
  29 #include "SMESHDS_Mesh.hxx"
  30
  31 using namespace std;
  32
  33 //=============================================================================
  34 /*!
  35  *
  36  */
  37 //=============================================================================
  38
  39 SMESHDS_Group::SMESHDS_Group (const int                 theID,
  40                               const SMESHDS_Mesh*       theMesh,
  41                               const SMDSAbs_ElementType theType)
  42      : SMESHDS_GroupBase(theID,theMesh,theType),
  43        myGroup(theMesh,theType)
  44 {
  45 }
  46
  47 //=======================================================================
  48 //function : Extent
  49 //purpose  :
  50 //=======================================================================
  51
  52 int SMESHDS_Group::Extent() const
  53 {
  54   return myGroup.Extent();
  55 }
  56
  57 //=======================================================================
  58 //function : IsEmpty
  59 //purpose  :
  60 //=======================================================================
  61
  62 bool SMESHDS_Group::IsEmpty()
  63 {
  64   return myGroup.IsEmpty();
  65 }
  66
  67 //=============================================================================
  68 /*!
  69  *
  70  */
  71 //=============================================================================
  72
  73 bool SMESHDS_Group::Contains (const int theID)
  74 {
  75   const SMDS_MeshElement* aElem = findInMesh (theID);
  76   if (aElem)
  77     return myGroup.Contains(aElem);
  78   return false;
  79 }
  80
  81 //=======================================================================
  82 //function : Contains
  83 //purpose  :
  84 //=======================================================================
  85
  86 bool SMESHDS_Group::Contains (const SMDS_MeshElement* elem)
  87 {
  88   if (elem)
  89     return myGroup.Contains(elem);
  90   return false;
  91 }
  92
  93 //=============================================================================
  94 /*!
  95  *
  96  */
  97 //=============================================================================
  98
  99 bool SMESHDS_Group::Add (const int theID)
 100 {
 101   const SMDS_MeshElement* aElem = findInMesh (theID);
 102   if (!aElem || myGroup.Contains(aElem))
 103     return false;
 104
 105   if (myGroup.IsEmpty())
 106     SetType( aElem->GetType() );
 107
 108   myGroup.Add (aElem);
 109   resetIterator();
 110   return true;
 111 }
 112
 113 //=============================================================================
 114 /*!
 115  *
 116  */
 117 //=============================================================================
 118
 119 bool SMESHDS_Group::Remove (const int theID)
 120 {
 121   const SMDS_MeshElement* aElem = findInMesh (theID);
 122   if (!aElem || !myGroup.Contains(aElem))
 123     return false;
 124   myGroup.Remove (aElem);
 125   resetIterator();
 126   return true;
 127 }
 128
 129
 130 //======================================================================
 131 //function : Clear
 132 //purpose  :
 133 //=======================================================================
 134
 135 void SMESHDS_Group::Clear()
 136 {
 137   myGroup.Clear();
 138   resetIterator();
 139 }
 140
 141 // =====================
 142 // class MyGroupIterator
 143 // =====================
 144
 145 class MyGroupIterator: public SMDS_ElemIterator
 146 {
 147   const SMDS_MeshGroup& myGroup;
 148  public:
 149   MyGroupIterator(const SMDS_MeshGroup& group): myGroup(group) { myGroup.InitIterator(); }
 150   bool more() { return myGroup.More(); }
 151   const SMDS_MeshElement* next() { return myGroup.Next(); }
 152 };
 153
 154 //=======================================================================
 155 //function : GetElements
 156 //purpose  :
 157 //=======================================================================
 158
 159 SMDS_ElemIteratorPtr SMESHDS_Group::GetElements() const
 160 {
 161   return SMDS_ElemIteratorPtr( new MyGroupIterator ( myGroup ));
 162 }
 163
 164 //================================================================================
 165 /*!
 166  * \brief Return a value allowing to find out if a group has changed or not
 167  */
 168 //================================================================================
 169
 170 int SMESHDS_Group::GetTic() const
 171 {
 172   return myGroup.Tic();
 173 }
 174
 175 //=======================================================================
 176 //function : SetType
 177 //purpose  :
 178 //=======================================================================
 179
 180 void SMESHDS_Group::SetType(SMDSAbs_ElementType theType)
 181 {
 182   if ( myGroup.IsEmpty() || GetType() == SMDSAbs_All ) {
 183     SMESHDS_GroupBase::SetType( theType );
 184     myGroup.SetType ( theType );
 185   }
 186   else
 187     SMESHDS_GroupBase::SetType( myGroup.GetType() );
 188 }
 189