Salome HOME
c7b308e919cbcf73fe5d9cb0c20d344c5d6a47c9
[modules/smesh.git] / src / SMDS / SMDS_VolumeOfNodes.cxx
1 //  SMESH SMDS : implementaion of Salome mesh data structure
2 //
3 //  Copyright (C) 2003  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
21
22 #include "utilities.h"
23 #include "SMDS_VolumeOfNodes.hxx"
24 #include "SMDS_MeshNode.hxx"
25
26 ///////////////////////////////////////////////////////////////////////////////
27 /// Create an hexahedron. node 1,2,3,4 and 5,6,7,8 are quadrangle and
28 /// 5,1 and 7,3 are an edges.
29 ///////////////////////////////////////////////////////////////////////////////
30 SMDS_VolumeOfNodes::SMDS_VolumeOfNodes(
31                 SMDS_MeshNode * node1,
32                 SMDS_MeshNode * node2,
33                 SMDS_MeshNode * node3,
34                 SMDS_MeshNode * node4,
35                 SMDS_MeshNode * node5,
36                 SMDS_MeshNode * node6,
37                 SMDS_MeshNode * node7,
38                 SMDS_MeshNode * node8)
39 {
40         myNodes.resize(8);
41         myNodes[0]=node1;
42         myNodes[1]=node2;
43         myNodes[2]=node3;
44         myNodes[3]=node4;
45         myNodes[4]=node5;
46         myNodes[5]=node6;
47         myNodes[6]=node7;
48         myNodes[7]=node8;
49 }
50
51 SMDS_VolumeOfNodes::SMDS_VolumeOfNodes(
52                 SMDS_MeshNode * node1,
53                 SMDS_MeshNode * node2,
54                 SMDS_MeshNode * node3,
55                 SMDS_MeshNode * node4)
56 {
57         myNodes.resize(4);
58         myNodes[0]=node1;
59         myNodes[1]=node2;
60         myNodes[2]=node3;
61         myNodes[3]=node4;
62 }
63
64 SMDS_VolumeOfNodes::SMDS_VolumeOfNodes(
65                 SMDS_MeshNode * node1,
66                 SMDS_MeshNode * node2,
67                 SMDS_MeshNode * node3,
68                 SMDS_MeshNode * node4,
69                 SMDS_MeshNode * node5)
70 {
71         myNodes.resize(5);
72         myNodes[0]=node1;
73         myNodes[1]=node2;
74         myNodes[2]=node3;
75         myNodes[3]=node4;
76         myNodes[4]=node5;
77 }
78
79 SMDS_VolumeOfNodes::SMDS_VolumeOfNodes(
80                 SMDS_MeshNode * node1,
81                 SMDS_MeshNode * node2,
82                 SMDS_MeshNode * node3,
83                 SMDS_MeshNode * node4,
84                 SMDS_MeshNode * node5,
85                 SMDS_MeshNode * node6)
86 {
87         myNodes.resize(6);
88         myNodes[0]=node1;
89         myNodes[1]=node2;
90         myNodes[2]=node3;
91         myNodes[3]=node4;
92         myNodes[4]=node5;
93         myNodes[5]=node6;
94 }
95 //=======================================================================
96 //function : Print
97 //purpose  : 
98 //=======================================================================
99
100 void SMDS_VolumeOfNodes::Print(ostream & OS) const
101 {
102         OS << "volume <" << GetID() << "> : ";
103         int i;
104         for (i = 0; i < 7; ++i) OS << myNodes[i] << ",";
105         OS << myNodes[7]<< ") " << endl;
106 }
107
108 int SMDS_VolumeOfNodes::NbFaces() const
109 {
110         switch(NbNodes())
111         {
112         case 4: return 4;
113         case 5: return 5;
114         case 6: return 5;
115         case 8: return 6;
116         default: MESSAGE("invalid number of nodes");
117         }
118 }
119
120 int SMDS_VolumeOfNodes::NbNodes() const
121 {
122         return myNodes.size();
123 }
124
125 int SMDS_VolumeOfNodes::NbEdges() const
126 {
127         switch(NbNodes())
128         {
129         case 4: return 6;
130         case 5: return 8;
131         case 6: return 9;
132         case 8: return 12;
133         default: MESSAGE("invalid number of nodes");
134         }
135 }
136
137 SMDS_Iterator<const SMDS_MeshElement *> * SMDS_VolumeOfNodes::
138         elementsIterator(SMDSAbs_ElementType type) const
139 {
140         class MyIterator:public SMDS_Iterator<const SMDS_MeshElement*>
141         {
142                 const vector<const SMDS_MeshNode*>& mySet;
143                 int index;
144           public:
145                 MyIterator(const vector<const SMDS_MeshNode*>& s):mySet(s),index(0)
146                 {}
147
148                 bool more()
149                 {
150                         return index<mySet.size();
151                 }
152
153                 const SMDS_MeshElement* next()
154                 {
155                         index++;
156                         return mySet[index-1];
157                 }       
158         };
159         switch(type)
160         {
161         case SMDSAbs_Volume:return SMDS_MeshElement::elementsIterator(SMDSAbs_Volume);
162         case SMDSAbs_Node:return new MyIterator(myNodes);
163         default: MESSAGE("ERROR : Iterator not implemented");
164         }
165 }
166
167 SMDSAbs_ElementType SMDS_VolumeOfNodes::GetType() const
168 {
169         return SMDSAbs_Volume;
170 }