src/SMESHDS/SMESHDS_GroupBase.cxx

   1 //  SMESH SMESHDS : idl implementation based on 'SMESH' unit's classes
   2 //
   3 //  Copyright (C) 2004  CEA
   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.salome-platform.org/ or email : webmaster.salome@opencascade.com
  20 //
  21 //
  22 //
  23 //  File   : SMESHDS_Group.cxx
  24 //  Module : SMESH
  25 //  $Header$
  26
  27 #include "SMESHDS_GroupBase.hxx"
  28 #include "SMESHDS_Mesh.hxx"
  29
  30 #include "utilities.h"
  31
  32 using namespace std;
  33
  34 //=============================================================================
  35 /*!
  36  *
  37  */
  38 //=============================================================================
  39
  40 SMESHDS_GroupBase::SMESHDS_GroupBase (const int                 theID,
  41                                       const SMESHDS_Mesh*       theMesh,
  42                                       const SMDSAbs_ElementType theType):
  43        myID(theID), myMesh(theMesh), myType(theType), myStoreName(""),
  44        myCurIndex(0), myCurID(-1)
  45 {
  46   myColor = Quantity_Color( 0.0, 0.0, 0.0, Quantity_TOC_RGB );
  47 }
  48
  49 //=============================================================================
  50 /*!
  51  *
  52  */
  53 //=============================================================================
  54
  55 int SMESHDS_GroupBase::GetID (const int theIndex)
  56 {
  57   if (myCurIndex < 1 || myCurIndex > theIndex) {
  58     myIterator = GetElements();
  59     myCurIndex = 0;
  60     myCurID = -1;
  61   }
  62   while (myCurIndex < theIndex && myIterator->more()) {
  63     myCurIndex++;
  64     myCurID = myIterator->next()->GetID();
  65   }
  66   return myCurIndex == theIndex ? myCurID : -1;
  67 }
  68
  69 //=============================================================================
  70 /*!
  71  *
  72  */
  73 //=============================================================================
  74
  75 const SMDS_MeshElement* SMESHDS_GroupBase::findInMesh (const int theID) const
  76 {
  77   SMDSAbs_ElementType aType = GetType();
  78   const SMDS_MeshElement* aElem = NULL;
  79   if (aType == SMDSAbs_Node) {
  80     aElem = GetMesh()->FindNode(theID);
  81   }
  82   else if (aType != SMDSAbs_All) {
  83     aElem = GetMesh()->FindElement(theID);
  84     if (aElem && aType != aElem->GetType())
  85       aElem = NULL;
  86   }
  87   return aElem;
  88 }
  89
  90 //=============================================================================
  91 /*!
  92  *  Internal method: resets cached iterator, should be called by ancestors
  93  *  when they are modified (ex: Add() or Remove() )
  94  */
  95 //=============================================================================
  96 void SMESHDS_GroupBase::resetIterator()
  97 {
  98   myCurIndex = 0;
  99   myCurID = -1;
 100 }
 101
 102 //=======================================================================
 103 //function : Extent
 104 //purpose  :
 105 //=======================================================================
 106
 107 int SMESHDS_GroupBase::Extent()
 108 {
 109   SMDS_ElemIteratorPtr it = GetElements();
 110   int nb = 0;
 111   if ( it )
 112     for ( ; it->more(); it->next() )
 113       nb++;
 114   return nb;
 115 }
 116
 117 //=======================================================================
 118 //function : IsEmpty
 119 //purpose  :
 120 //=======================================================================
 121
 122 bool SMESHDS_GroupBase::IsEmpty()
 123 {
 124   SMDS_ElemIteratorPtr it = GetElements();
 125   return ( !it || !it->more() );
 126 }
 127
 128 //=======================================================================
 129 //function : Contains
 130 //purpose  :
 131 //=======================================================================
 132
 133 bool SMESHDS_GroupBase::Contains (const int theID)
 134 {
 135   if ( SMDS_ElemIteratorPtr it = GetElements() ) {
 136     while ( it->more() )
 137       if ( it->next()->GetID() == theID )
 138         return true;
 139   }
 140   return false;
 141 }
 142
 143 //=======================================================================
 144 //function : Contains
 145 //purpose  :
 146 //=======================================================================
 147
 148 bool SMESHDS_GroupBase::Contains (const SMDS_MeshElement* elem)
 149 {
 150   if ( elem )
 151     return Contains( elem->GetID() );
 152   return false;
 153 }
 154
 155 //=======================================================================
 156 //function : SetType
 157 //purpose  :
 158 //=======================================================================
 159
 160 void SMESHDS_GroupBase::SetType(SMDSAbs_ElementType theType)
 161 {
 162   myType = theType;
 163 }
 164
 165 //=======================================================================
 166 //function : SetType
 167 //purpose  :
 168 //=======================================================================
 169
 170 void SMESHDS_GroupBase::SetColorGroup(int theColorGroup)
 171 {
 172   int aRed = ( theColorGroup/1000000 );
 173   int aGreen = ( theColorGroup -aRed*1000000)/1000;
 174   int aBlue = ( theColorGroup - aRed*1000000 - aGreen*1000 );
 175   double aR = aRed/255.0;
 176   double aG = aGreen/255.0;
 177   double aB = aBlue/255.0;
 178   Quantity_Color aColor( aR, aG, aB, Quantity_TOC_RGB );
 179   SetColor( aColor );
 180 }
 181
 182 //=======================================================================
 183 //function : SetType
 184 //purpose  :
 185 //=======================================================================
 186
 187 int SMESHDS_GroupBase::GetColorGroup() const
 188 {
 189   Quantity_Color aColor = GetColor();
 190   double aRed = aColor.Red();
 191   double aGreen = aColor.Green();
 192   double aBlue = aColor.Blue();
 193   int aR = aRed*255;
 194   int aG = aGreen*255;
 195   int aB = aBlue*255;
 196   int aRet = (int)(aR*1000000 + aG*1000 + aB);
 197
 198   return aRet;
 199 }
 200