]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHDS/SMESHDS_GroupBase.cxx
Salome HOME
d1b0d6ecdf94e904038c545cf04e8b8d02e7fa2c
[modules/smesh.git] / src / SMESHDS / SMESHDS_GroupBase.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_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