src/SMESHDS/SMESHDS_GroupBase.cxx

   1 //  Copyright (C) 2007-2008  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 //  SMESH SMESHDS : idl implementation based on 'SMESH' unit's classes
  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   if ( aR < 0. || aR > 1. || // PAL19395
 179        aG < 0. || aG > 1. ||
 180        aB < 0. || aB > 1. )
 181 // #ifdef _DEBUG_
 182 //     cout << "SMESHDS_GroupBase::SetColorGroup("<<theColorGroup<<"), invalid color ignored"<<endl;
 183 // #endif
 184     return;
 185   Quantity_Color aColor( aR, aG, aB, Quantity_TOC_RGB );
 186   SetColor( aColor );
 187 }
 188
 189 //=======================================================================
 190 //function : SetType
 191 //purpose  :
 192 //=======================================================================
 193
 194 int SMESHDS_GroupBase::GetColorGroup() const
 195 {
 196   Quantity_Color aColor = GetColor();
 197   double aRed = aColor.Red();
 198   double aGreen = aColor.Green();
 199   double aBlue = aColor.Blue();
 200   int aR = int( aRed  *255 );
 201   int aG = int( aGreen*255 );
 202   int aB = int( aBlue *255 );
 203   int aRet = (int)(aR*1000000 + aG*1000 + aB);
 204
 205   return aRet;
 206 }
 207