Salome HOME
710be521b4ad8962c9f26aa0afcfe71cef492832
[modules/smesh.git] / src / SMDS / SMDS_MeshElementIDFactory.cxx
1 // Copyright (C) 2007-2016  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 SMDS : implementaion of Salome mesh data structure
24 //  File   : SMDS_MeshElementIDFactory.cxx
25 //  Author : Jean-Michel BOULCOURT
26 //  Module : SMESH
27 //
28 #ifdef _MSC_VER
29 #pragma warning(disable:4786)
30 #endif
31
32 #include "SMDS_MeshElementIDFactory.hxx"
33 #include "SMDS_MeshElement.hxx"
34 #include "SMDS_Mesh.hxx"
35
36 #include "utilities.h"
37
38 #include "SMDS_UnstructuredGrid.hxx"
39 #include <vtkCellType.h>
40
41 #include <climits>
42
43 using namespace std;
44
45 //=======================================================================
46 //function : SMDS_MeshElementIDFactory
47 //purpose  : 
48 //=======================================================================
49 SMDS_MeshElementIDFactory::SMDS_MeshElementIDFactory():
50   SMDS_MeshNodeIDFactory()
51 {
52 }
53
54 int SMDS_MeshElementIDFactory::SetInVtkGrid(SMDS_MeshElement * elem)
55 {
56    // --- retrieve nodes ID
57
58   SMDS_MeshCell *cell = dynamic_cast<SMDS_MeshCell*>(elem);
59   assert(cell);
60   vector<vtkIdType> nodeIds;
61   SMDS_ElemIteratorPtr it = elem->nodesIterator();
62   while(it->more())
63   {
64       int nodeId = (static_cast<const SMDS_MeshNode*>(it->next()))->getVtkId();
65       MESSAGE("   node in cell " << cell->getVtkId() << " : " << nodeId)
66       nodeIds.push_back(nodeId);
67   }
68
69   // --- insert cell in vtkUnstructuredGrid
70
71   vtkUnstructuredGrid * grid = myMesh->getGrid();
72   //int locType = elem->GetType();
73   int typ = VTK_VERTEX;//GetVtkCellType(locType);
74   int cellId = grid->InsertNextLinkedCell(typ, nodeIds.size(), &nodeIds[0]);
75   cell->setVtkId(cellId); 
76   //MESSAGE("SMDS_MeshElementIDFactory::SetInVtkGrid " << cellId);
77   return cellId;
78 }
79
80 //=======================================================================
81 //function : BindID
82 //purpose  : 
83 //=======================================================================
84
85 bool SMDS_MeshElementIDFactory::BindID(int ID, SMDS_MeshElement * elem)
86 {
87   MESSAGE("SMDS_MeshElementIDFactory::BindID " << ID);
88   SetInVtkGrid(elem);
89   return myMesh->registerElement(ID, elem);
90 }
91
92 //=======================================================================
93 //function : MeshElement
94 //purpose  : 
95 //=======================================================================
96 SMDS_MeshElement* SMDS_MeshElementIDFactory::MeshElement(int ID)
97 {
98   if ( ID<1 || ID >= (int) myMesh->myCells.size() )
99     return NULL;
100   const SMDS_MeshElement* elem = GetMesh()->FindElement(ID);
101   return (SMDS_MeshElement*)(elem);
102 }
103
104 //=======================================================================
105 //function : GetFreeID
106 //purpose  : 
107 //=======================================================================
108
109 int SMDS_MeshElementIDFactory::GetFreeID()
110 {
111   int ID;
112   do {
113     ID = SMDS_MeshIDFactory::GetFreeID();
114   } while ( MeshElement( ID ));
115   return ID;
116 }
117
118 //=======================================================================
119 //function : ReleaseID
120 //purpose  : 
121 //=======================================================================
122 void SMDS_MeshElementIDFactory::ReleaseID(int ID, int vtkId)
123 {
124   if (ID < 1) // TODO check case ID == O
125     {
126       MESSAGE("~~~~~~~~~~~~~~ SMDS_MeshElementIDFactory::ReleaseID ID = " << ID);
127       return;
128     }
129   //MESSAGE("~~~~~~~~~~~~~~ SMDS_MeshElementIDFactory::ReleaseID smdsId vtkId " << ID << " " << vtkId);
130   if (vtkId >= 0)
131     {
132       assert(vtkId < (int)myMesh->myCellIdVtkToSmds.size());
133       myMesh->myCellIdVtkToSmds[vtkId] = -1;
134       myMesh->setMyModified();
135     }
136   SMDS_MeshIDFactory::ReleaseID(ID);
137   if (ID == myMax)
138     myMax = 0;
139   if (ID == myMin)
140     myMax = 0;
141 }
142
143 //=======================================================================
144 //function : updateMinMax
145 //purpose  : 
146 //=======================================================================
147
148 void SMDS_MeshElementIDFactory::updateMinMax() const
149 {
150   myMin = INT_MAX;
151   myMax = 0;
152   for (size_t i = 0; i < myMesh->myCells.size(); i++)
153     {
154       if (myMesh->myCells[i])
155         {
156           int id = myMesh->myCells[i]->GetID();
157           if (id > myMax)
158             myMax = id;
159           if (id < myMin)
160             myMin = id;
161         }
162     }
163   if (myMin == INT_MAX)
164     myMin = 0;
165 }
166
167 //=======================================================================
168 //function : elementsIterator
169 //purpose  : Return an iterator on elements of the factory
170 //=======================================================================
171
172 SMDS_ElemIteratorPtr SMDS_MeshElementIDFactory::elementsIterator() const
173 {
174     return myMesh->elementsIterator(SMDSAbs_All);
175 }
176
177 void SMDS_MeshElementIDFactory::Clear()
178 {
179   //myMesh->myCellIdSmdsToVtk.clear();
180   myMesh->myCellIdVtkToSmds.clear();
181   myMin = myMax = 0;
182   SMDS_MeshIDFactory::Clear();
183 }