1 // Copyright (C) 2007-2010 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
23 // SMESH SMDS : implementaion of Salome mesh data structure
24 // File : SMDS_MeshElementIDFactory.cxx
25 // Author : Jean-Michel BOULCOURT
29 #pragma warning(disable:4786)
32 #include "SMDS_MeshElementIDFactory.hxx"
33 #include "SMDS_MeshElement.hxx"
34 #include "SMDS_Mesh.hxx"
36 #include "utilities.h"
38 #include "SMDS_UnstructuredGrid.hxx"
39 #include <vtkCellType.h>
43 //=======================================================================
44 //function : SMDS_MeshElementIDFactory
46 //=======================================================================
47 SMDS_MeshElementIDFactory::SMDS_MeshElementIDFactory():
48 SMDS_MeshNodeIDFactory()
50 // myIDElements.clear();
51 // myVtkIndex.clear();
52 myVtkCellTypes.clear();
53 myVtkCellTypes.reserve(SMDSEntity_Last);
54 myVtkCellTypes[SMDSEntity_Node] = VTK_VERTEX;
55 myVtkCellTypes[SMDSEntity_0D] = VTK_VERTEX;
56 myVtkCellTypes[SMDSEntity_Edge] = VTK_LINE;
57 myVtkCellTypes[SMDSEntity_Quad_Edge] = VTK_QUADRATIC_EDGE;
58 myVtkCellTypes[SMDSEntity_Triangle] = VTK_TRIANGLE;
59 myVtkCellTypes[SMDSEntity_Quad_Triangle] = VTK_QUADRATIC_TRIANGLE;
60 myVtkCellTypes[SMDSEntity_Quadrangle] = VTK_QUAD;
61 myVtkCellTypes[SMDSEntity_Quad_Quadrangle] = VTK_QUADRATIC_TRIANGLE;
62 myVtkCellTypes[SMDSEntity_Polygon] = VTK_POLYGON;
63 myVtkCellTypes[SMDSEntity_Quad_Polygon] = VTK_POLYGON; // -PR- verifer
64 myVtkCellTypes[SMDSEntity_Tetra] = VTK_TETRA;
65 myVtkCellTypes[SMDSEntity_Quad_Tetra] = VTK_QUADRATIC_TETRA;
66 myVtkCellTypes[SMDSEntity_Pyramid] = VTK_PYRAMID;
67 myVtkCellTypes[SMDSEntity_Quad_Pyramid] = VTK_CONVEX_POINT_SET;
68 myVtkCellTypes[SMDSEntity_Hexa] = VTK_HEXAHEDRON;
69 myVtkCellTypes[SMDSEntity_Quad_Hexa] = VTK_QUADRATIC_HEXAHEDRON;
70 myVtkCellTypes[SMDSEntity_Penta] = VTK_WEDGE;
71 myVtkCellTypes[SMDSEntity_Quad_Penta] = VTK_QUADRATIC_WEDGE;
72 //#ifdef VTK_HAVE_POLYHEDRON
73 myVtkCellTypes[SMDSEntity_Polyhedra] = VTK_POLYHEDRON;
75 // myVtkCellTypes[SMDSEntity_Polyhedra] = VTK_CONVEX_POINT_SET;
77 myVtkCellTypes[SMDSEntity_Quad_Polyhedra] = VTK_CONVEX_POINT_SET;
80 int SMDS_MeshElementIDFactory::SetInVtkGrid(SMDS_MeshElement * elem)
82 // --- retrieve nodes ID
84 SMDS_MeshCell *cell = dynamic_cast<SMDS_MeshCell*>(elem);
86 vector<vtkIdType> nodeIds;
87 SMDS_ElemIteratorPtr it = elem->nodesIterator();
90 int nodeId = (static_cast<const SMDS_MeshNode*>(it->next()))->getVtkId();
91 MESSAGE(" node in cell " << cell->getVtkId() << " : " << nodeId)
92 nodeIds.push_back(nodeId);
95 // --- insert cell in vtkUnstructuredGrid
97 vtkUnstructuredGrid * grid = myMesh->getGrid();
98 //int locType = elem->GetType();
99 int typ = VTK_VERTEX;//GetVtkCellType(locType);
100 int cellId = grid->InsertNextLinkedCell(typ, nodeIds.size(), &nodeIds[0]);
101 cell->setVtkId(cellId);
102 //MESSAGE("SMDS_MeshElementIDFactory::SetInVtkGrid " << cellId);
106 //=======================================================================
109 //=======================================================================
111 bool SMDS_MeshElementIDFactory::BindID(int ID, SMDS_MeshElement * elem)
113 MESSAGE("SMDS_MeshElementIDFactory::BindID " << ID);
115 return myMesh->registerElement(ID, elem);
118 //=======================================================================
119 //function : MeshElement
121 //=======================================================================
122 SMDS_MeshElement* SMDS_MeshElementIDFactory::MeshElement(int ID)
124 if ((ID<1) || (ID>=myMesh->myCells.size()))
126 const SMDS_MeshElement* elem = GetMesh()->FindElement(ID);
127 return (SMDS_MeshElement*)(elem);
130 //=======================================================================
131 //function : GetFreeID
133 //=======================================================================
135 int SMDS_MeshElementIDFactory::GetFreeID()
139 ID = SMDS_MeshIDFactory::GetFreeID();
140 } while ( MeshElement( ID ));
144 //=======================================================================
145 //function : ReleaseID
147 //=======================================================================
148 void SMDS_MeshElementIDFactory::ReleaseID(int ID, int vtkId)
150 if (ID < 1) // TODO check case ID == O
152 MESSAGE("~~~~~~~~~~~~~~ SMDS_MeshElementIDFactory::ReleaseID ID = " << ID);
155 //MESSAGE("~~~~~~~~~~~~~~ SMDS_MeshElementIDFactory::ReleaseID smdsId vtkId " << ID << " " << vtkId);
158 assert(vtkId < myMesh->myCellIdVtkToSmds.size());
159 myMesh->myCellIdVtkToSmds[vtkId] = -1;
160 myMesh->setMyModified();
162 SMDS_MeshIDFactory::ReleaseID(ID);
169 //=======================================================================
170 //function : updateMinMax
172 //=======================================================================
174 void SMDS_MeshElementIDFactory::updateMinMax() const
176 myMin = IntegerLast();
178 for (int i = 0; i < myMesh->myCells.size(); i++)
180 if (myMesh->myCells[i])
182 int id = myMesh->myCells[i]->GetID();
189 if (myMin == IntegerLast())
193 //=======================================================================
194 //function : elementsIterator
195 //purpose : Return an iterator on elements of the factory
196 //=======================================================================
198 SMDS_ElemIteratorPtr SMDS_MeshElementIDFactory::elementsIterator() const
200 return myMesh->elementsIterator(SMDSAbs_All);
203 void SMDS_MeshElementIDFactory::Clear()
205 //myMesh->myCellIdSmdsToVtk.clear();
206 myMesh->myCellIdVtkToSmds.clear();
208 SMDS_MeshIDFactory::Clear();
211 int SMDS_MeshElementIDFactory::GetVtkCellType(int SMDSType)
213 assert((SMDSType >=0) && (SMDSType< SMDSEntity_Last));
214 return myVtkCellTypes[SMDSType];