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 typ = GetVtkCellType(elem->GetType());
99 int cellId = grid->InsertNextLinkedCell(typ, nodeIds.size(), &nodeIds[0]);
100 cell->setVtkId(cellId);
101 //MESSAGE("SMDS_MeshElementIDFactory::SetInVtkGrid " << cellId);
105 //=======================================================================
108 //=======================================================================
110 bool SMDS_MeshElementIDFactory::BindID(int ID, SMDS_MeshElement * elem)
112 MESSAGE("SMDS_MeshElementIDFactory::BindID " << ID);
114 return myMesh->registerElement(ID, elem);
117 //=======================================================================
118 //function : MeshElement
120 //=======================================================================
121 SMDS_MeshElement* SMDS_MeshElementIDFactory::MeshElement(int ID)
123 if ((ID<1) || (ID>=myMesh->myCells.size()))
125 const SMDS_MeshElement* elem = GetMesh()->FindElement(ID);
126 return (SMDS_MeshElement*)(elem);
129 //=======================================================================
130 //function : ReleaseID
132 //=======================================================================
133 void SMDS_MeshElementIDFactory::ReleaseID(int ID, int vtkId)
135 if (ID < 1) // TODO check case ID == O
137 MESSAGE("~~~~~~~~~~~~~~ SMDS_MeshElementIDFactory::ReleaseID ID = " << ID);
140 //MESSAGE("~~~~~~~~~~~~~~ SMDS_MeshElementIDFactory::ReleaseID smdsId vtkId " << ID << " " << vtkId);
143 assert(vtkId < myMesh->myCellIdVtkToSmds.size());
144 myMesh->myCellIdVtkToSmds[vtkId] = -1;
145 myMesh->setMyModified();
147 SMDS_MeshIDFactory::ReleaseID(ID);
154 //=======================================================================
155 //function : updateMinMax
157 //=======================================================================
159 void SMDS_MeshElementIDFactory::updateMinMax() const
161 myMin = IntegerLast();
163 for (int i = 0; i < myMesh->myCells.size(); i++)
165 if (myMesh->myCells[i])
167 int id = myMesh->myCells[i]->GetID();
174 if (myMin == IntegerLast())
178 //=======================================================================
179 //function : elementsIterator
180 //purpose : Return an iterator on elements of the factory
181 //=======================================================================
183 SMDS_ElemIteratorPtr SMDS_MeshElementIDFactory::elementsIterator() const
185 return myMesh->elementsIterator(SMDSAbs_All);
188 void SMDS_MeshElementIDFactory::Clear()
190 //myMesh->myCellIdSmdsToVtk.clear();
191 myMesh->myCellIdVtkToSmds.clear();
193 SMDS_MeshIDFactory::Clear();
196 int SMDS_MeshElementIDFactory::GetVtkCellType(int SMDSType)
198 assert((SMDSType >=0) && (SMDSType< SMDSEntity_Last));
199 return myVtkCellTypes[SMDSType];