1 // Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // SMESH SMDS : implementaion of Salome mesh data structure
23 // File : SMDS_MeshElementIDFactory.cxx
24 // Author : Jean-Michel BOULCOURT
28 #pragma warning(disable:4786)
31 #include "SMDS_MeshElementIDFactory.hxx"
32 #include "SMDS_MeshElement.hxx"
33 #include "SMDS_Mesh.hxx"
35 #include "utilities.h"
37 #include <vtkUnstructuredGrid.h>
38 #include <vtkCellType.h>
42 //=======================================================================
43 //function : SMDS_MeshElementIDFactory
45 //=======================================================================
46 SMDS_MeshElementIDFactory::SMDS_MeshElementIDFactory():
47 SMDS_MeshNodeIDFactory()
51 myVtkCellTypes.clear();
52 myVtkCellTypes.reserve(SMDSEntity_Last);
53 myVtkCellTypes[SMDSEntity_Node] = VTK_VERTEX;
54 myVtkCellTypes[SMDSEntity_0D] = VTK_VERTEX;
55 myVtkCellTypes[SMDSEntity_Edge] = VTK_LINE;
56 myVtkCellTypes[SMDSEntity_Quad_Edge] = VTK_QUADRATIC_EDGE;
57 myVtkCellTypes[SMDSEntity_Triangle] = VTK_TRIANGLE;
58 myVtkCellTypes[SMDSEntity_Quad_Triangle] = VTK_QUADRATIC_TRIANGLE;
59 myVtkCellTypes[SMDSEntity_Quadrangle] = VTK_QUAD;
60 myVtkCellTypes[SMDSEntity_Quad_Quadrangle] = VTK_QUADRATIC_TRIANGLE;
61 myVtkCellTypes[SMDSEntity_Polygon] = VTK_POLYGON;
62 myVtkCellTypes[SMDSEntity_Quad_Polygon] = VTK_POLYGON; // -PR- verifer
63 myVtkCellTypes[SMDSEntity_Tetra] = VTK_TETRA;
64 myVtkCellTypes[SMDSEntity_Quad_Tetra] = VTK_QUADRATIC_TETRA;
65 myVtkCellTypes[SMDSEntity_Pyramid] = VTK_PYRAMID;
66 myVtkCellTypes[SMDSEntity_Quad_Pyramid] = VTK_CONVEX_POINT_SET;
67 myVtkCellTypes[SMDSEntity_Hexa] = VTK_HEXAHEDRON;
68 myVtkCellTypes[SMDSEntity_Quad_Hexa] = VTK_QUADRATIC_HEXAHEDRON;
69 myVtkCellTypes[SMDSEntity_Penta] = VTK_WEDGE;
70 myVtkCellTypes[SMDSEntity_Quad_Penta] = VTK_QUADRATIC_WEDGE;
71 myVtkCellTypes[SMDSEntity_Polyhedra] = VTK_CONVEX_POINT_SET;
72 myVtkCellTypes[SMDSEntity_Quad_Polyhedra] = VTK_CONVEX_POINT_SET;
75 //=======================================================================
78 //=======================================================================
79 bool SMDS_MeshElementIDFactory::BindID(int ID, SMDS_MeshElement * elem)
81 if ((ID < myIDElements.size()) && myIDElements[ID] >= 0) // --- already bound
83 MESSAGE(" --------------------------------- already bound "<< ID << " " << myIDElements[ID]);
87 if (ID >= myIDElements.size()) // --- resize local vector
89 //MESSAGE(" ------------------- resize myIDElements " << ID << " --> " << ID+SMDS_Mesh::chunkSize);
90 myIDElements.resize(ID+SMDS_Mesh::chunkSize,-1); // fill new elements with -1
93 // --- retreive nodes ID
96 SMDS_ElemIteratorPtr it = elem->nodesIterator();
99 int nodeId = it->next()->GetID();
100 //MESSAGE(" node in cell " << ID << " : " <<nodeId)
101 nodeIds.push_back(nodeId);
104 // --- insert cell in vtkUnstructuredGrid
106 vtkUnstructuredGrid * grid = myMesh->getGrid();
107 int typ = GetVtkCellType(elem->GetType());
108 int cellId = grid->InsertNextLinkedCell(typ, nodeIds.size(), &nodeIds[0]);
110 // --- fill local vector
112 myIDElements[ID] = cellId;
113 //MESSAGE("smds:" << ID << " vtk:" << cellId );
115 if (cellId >= myVtkIndex.size()) // --- resize local vector
117 //MESSAGE(" --------------------- resize myVtkIndex " << cellId << " --> " << cellId+SMDS_Mesh::chunkSize);
118 myVtkIndex.resize(cellId+SMDS_Mesh::chunkSize, -1);
120 myVtkIndex[cellId] = ID;
123 SMDS_MeshCell *cell = dynamic_cast<SMDS_MeshCell*>(elem);
125 cell->setVtkId(cellId);
130 //=======================================================================
131 //function : MeshElement
133 //=======================================================================
134 SMDS_MeshElement* SMDS_MeshElementIDFactory::MeshElement(int ID)
136 if ((ID<0) || (ID>myMax) || (myIDElements[ID]<0))
138 const SMDS_MeshElement* elem = GetMesh()->FindElement(ID);
139 return (SMDS_MeshElement*)(elem);
142 //=======================================================================
143 //function : ReleaseID
145 //=======================================================================
146 void SMDS_MeshElementIDFactory::ReleaseID(const int ID)
148 myIDElements[ID] = -1;
149 SMDS_MeshIDFactory::ReleaseID(ID);
156 //=======================================================================
157 //function : updateMinMax
159 //=======================================================================
161 void SMDS_MeshElementIDFactory::updateMinMax() const
163 myMin = IntegerLast();
165 for (int i=0; i<myIDElements.size(); i++)
166 if (int id=myIDElements[i] >=0)
168 if (id > myMax) myMax = id;
169 if (id < myMin) myMin = id;
171 if (myMin == IntegerLast())
175 //=======================================================================
176 //function : elementsIterator
177 //purpose : Return an iterator on elements of the factory
178 //=======================================================================
180 SMDS_ElemIteratorPtr SMDS_MeshElementIDFactory::elementsIterator() const
182 return myMesh->elementsIterator(SMDSAbs_All);
185 void SMDS_MeshElementIDFactory::Clear()
187 myIDElements.clear();
189 SMDS_MeshIDFactory::Clear();
192 int SMDS_MeshElementIDFactory::GetVtkCellType(int SMDSType)
194 assert((SMDSType >=0) && (SMDSType< SMDSEntity_Last));
195 return myVtkCellTypes[SMDSType];