src/SMESHDS/SMESHDS_GroupBase.cxx

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