2 #include "SMDS_UnstructuredGrid.hxx"
3 #include "SMDS_Mesh.hxx"
4 #include "SMDS_MeshInfo.hxx"
5 #include "SMDS_Downward.hxx"
6 #include "SMDS_MeshVolume.hxx"
10 #include <vtkCellArray.h>
11 #include <vtkCellLinks.h>
12 #include <vtkIdTypeArray.h>
13 #include <vtkUnsignedCharArray.h>
20 SMDS_CellLinks* SMDS_CellLinks::New()
22 MESSAGE("SMDS_CellLinks::New");
23 return new SMDS_CellLinks();
26 vtkCellLinks::Link* SMDS_CellLinks::ResizeL(vtkIdType sz)
28 return vtkCellLinks::Resize(sz);
31 vtkIdType SMDS_CellLinks::GetLinksSize()
36 SMDS_CellLinks::SMDS_CellLinks() :
41 SMDS_CellLinks::~SMDS_CellLinks()
45 SMDS_UnstructuredGrid* SMDS_UnstructuredGrid::New()
47 MESSAGE("SMDS_UnstructuredGrid::New");
48 return new SMDS_UnstructuredGrid();
51 SMDS_UnstructuredGrid::SMDS_UnstructuredGrid() :
54 _cellIdToDownId.clear();
60 SMDS_UnstructuredGrid::~SMDS_UnstructuredGrid()
64 unsigned long SMDS_UnstructuredGrid::GetMTime()
66 unsigned long mtime = vtkUnstructuredGrid::GetMTime();
67 MESSAGE("vtkUnstructuredGrid::GetMTime: " << mtime);
71 void SMDS_UnstructuredGrid::Update()
73 MESSAGE("SMDS_UnstructuredGrid::Update");
74 return vtkUnstructuredGrid::Update();
77 void SMDS_UnstructuredGrid::UpdateInformation()
79 MESSAGE("SMDS_UnstructuredGrid::UpdateInformation");
80 return vtkUnstructuredGrid::UpdateInformation();
83 vtkPoints* SMDS_UnstructuredGrid::GetPoints()
85 // TODO erreur incomprehensible de la macro vtk GetPoints apparue avec la version paraview de fin aout 2010
86 //MESSAGE("*********************** SMDS_UnstructuredGrid::GetPoints " << this->Points << " " << vtkUnstructuredGrid::GetPoints());
90 //#ifdef VTK_HAVE_POLYHEDRON
91 int SMDS_UnstructuredGrid::InsertNextLinkedCell(int type, int npts, vtkIdType *pts)
93 if (type != VTK_POLYHEDRON)
94 return vtkUnstructuredGrid::InsertNextLinkedCell(type, npts, pts);
96 // --- type = VTK_POLYHEDRON
97 //MESSAGE("InsertNextLinkedCell VTK_POLYHEDRON");
98 int cellid = this->InsertNextCell(type, npts, pts);
100 set<vtkIdType> setOfNodes;
104 for (int nf = 0; nf < nbfaces; nf++)
106 int nbnodes = pts[i];
108 for (int k = 0; k < nbnodes; k++)
110 //MESSAGE(" cell " << cellid << " face " << nf << " node " << pts[i]);
111 setOfNodes.insert(pts[i]);
116 set<vtkIdType>::iterator it = setOfNodes.begin();
117 for (; it != setOfNodes.end(); ++it)
119 //MESSAGE("reverse link for node " << *it << " cell " << cellid);
120 this->Links->ResizeCellList(*it, 1);
121 this->Links->AddCellReference(cellid, *it);
128 void SMDS_UnstructuredGrid::setSMDS_mesh(SMDS_Mesh *mesh)
133 void SMDS_UnstructuredGrid::compactGrid(std::vector<int>& idNodesOldToNew, int newNodeSize,
134 std::vector<int>& idCellsOldToNew, int newCellSize)
136 MESSAGE("------------------------- SMDS_UnstructuredGrid::compactGrid " << newNodeSize << " " << newCellSize);CHRONO(1);
137 int alreadyCopied = 0;
139 // --- if newNodeSize, create a new compacted vtkPoints
141 vtkPoints *newPoints = vtkPoints::New();
142 newPoints->SetDataType(VTK_DOUBLE);
143 newPoints->SetNumberOfPoints(newNodeSize);
146 MESSAGE("-------------- compactGrid, newNodeSize " << newNodeSize);
147 // rnv: to fix bug "21125: EDF 1233 SMESH: Degradation of precision in a test case for quadratic conversion"
148 // using double type for storing coordinates of nodes instead float.
149 int oldNodeSize = idNodesOldToNew.size();
152 while ( i < oldNodeSize )
154 // skip a hole if any
155 while ( i < oldNodeSize && idNodesOldToNew[i] < 0 )
158 // look for a block end
159 while ( i < oldNodeSize && idNodesOldToNew[i] >= 0 )
162 copyNodes(newPoints, idNodesOldToNew, alreadyCopied, startBloc, endBloc);
164 newPoints->Squeeze();
167 // --- create new compacted Connectivity, Locations and Types
169 int oldCellSize = this->Types->GetNumberOfTuples();
171 vtkCellArray *newConnectivity = vtkCellArray::New();
172 newConnectivity->Initialize();
173 int oldCellDataSize = this->Connectivity->GetData()->GetSize();
174 newConnectivity->Allocate(oldCellDataSize);
175 MESSAGE("oldCellSize="<< oldCellSize << " oldCellDataSize=" << oldCellDataSize);
177 vtkUnsignedCharArray *newTypes = vtkUnsignedCharArray::New();
178 newTypes->Initialize();
179 newTypes->SetNumberOfValues(newCellSize);
181 vtkIdTypeArray *newLocations = vtkIdTypeArray::New();
182 newLocations->Initialize();
183 newLocations->SetNumberOfValues(newCellSize);
185 // TODO some polyhedron may be huge (only in some tests)
186 vtkIdType tmpid[NBMAXNODESINCELL];
187 vtkIdType *pointsCell = &tmpid[0]; // --- points id to fill a new cell
191 while ( i < oldCellSize )
193 // skip a hole if any
194 while ( i < oldCellSize && this->Types->GetValue(i) == VTK_EMPTY_CELL )
197 // look for a block end
198 while ( i < oldCellSize && this->Types->GetValue(i) != VTK_EMPTY_CELL )
201 if ( endBloc > startBloc )
202 copyBloc(newTypes, idCellsOldToNew, idNodesOldToNew, newConnectivity, newLocations, pointsCell,
203 alreadyCopied, startBloc, endBloc);
206 newConnectivity->Squeeze();
208 if (1/*newNodeSize*/)
210 MESSAGE("------- newNodeSize, setPoints");
211 this->SetPoints(newPoints);
212 MESSAGE("NumberOfPoints: " << this->GetNumberOfPoints());
215 if (this->FaceLocations)
217 vtkIdTypeArray *newFaceLocations = vtkIdTypeArray::New();
218 newFaceLocations->Initialize();
219 newFaceLocations->Allocate(newTypes->GetSize());
220 vtkIdTypeArray *newFaces = vtkIdTypeArray::New();
221 newFaces->Initialize();
222 newFaces->Allocate(this->Faces->GetSize());
223 for (int i = 0; i < oldCellSize; i++)
225 if (this->Types->GetValue(i) == VTK_EMPTY_CELL)
227 int newCellId = idCellsOldToNew[i];
228 if (newTypes->GetValue(newCellId) == VTK_POLYHEDRON)
230 newFaceLocations->InsertNextValue(newFaces->GetMaxId()+1);
231 int oldFaceLoc = this->FaceLocations->GetValue(i);
232 int nCellFaces = this->Faces->GetValue(oldFaceLoc++);
233 newFaces->InsertNextValue(nCellFaces);
234 for (int n=0; n<nCellFaces; n++)
236 int nptsInFace = this->Faces->GetValue(oldFaceLoc++);
237 newFaces->InsertNextValue(nptsInFace);
238 for (int k=0; k<nptsInFace; k++)
240 int oldpt = this->Faces->GetValue(oldFaceLoc++);
241 newFaces->InsertNextValue(idNodesOldToNew[oldpt]);
247 newFaceLocations->InsertNextValue(-1);
250 newFaceLocations->Squeeze();
252 newFaceLocations->Register(this);
253 newFaces->Register(this);
254 this->SetCells(newTypes, newLocations, newConnectivity, newFaceLocations, newFaces);
255 newFaceLocations->Delete();
259 this->SetCells(newTypes, newLocations, newConnectivity, FaceLocations, Faces);
263 newLocations->Delete();
264 newConnectivity->Delete();
268 void SMDS_UnstructuredGrid::copyNodes(vtkPoints *newPoints, std::vector<int>& idNodesOldToNew, int& alreadyCopied,
271 MESSAGE("copyNodes " << alreadyCopied << " " << start << " " << end << " size: " << end - start << " total: " << alreadyCopied + end - start);
272 void *target = newPoints->GetVoidPointer(3 * alreadyCopied);
273 void *source = this->Points->GetVoidPointer(3 * start);
274 int nbPoints = end - start;
277 memcpy(target, source, 3 * sizeof(double) * nbPoints);
278 for (int j = start; j < end; j++)
279 idNodesOldToNew[j] = alreadyCopied++; // old vtkId --> new vtkId
283 void SMDS_UnstructuredGrid::copyBloc(vtkUnsignedCharArray *newTypes, std::vector<int>& idCellsOldToNew,
284 std::vector<int>& idNodesOldToNew, vtkCellArray* newConnectivity,
285 vtkIdTypeArray* newLocations, vtkIdType* pointsCell, int& alreadyCopied,
288 MESSAGE("copyBloc " << alreadyCopied << " " << start << " " << end << " size: " << end - start << " total: " << alreadyCopied + end - start);
289 for (int j = start; j < end; j++)
291 newTypes->SetValue(alreadyCopied, this->Types->GetValue(j));
292 idCellsOldToNew[j] = alreadyCopied; // old vtkId --> new vtkId
293 vtkIdType oldLoc = this->Locations->GetValue(j);
295 vtkIdType *oldPtsCell = 0;
296 this->Connectivity->GetCell(oldLoc, nbpts, oldPtsCell);
297 assert(nbpts < NBMAXNODESINCELL);
298 //MESSAGE(j << " " << alreadyCopied << " " << (int)this->Types->GetValue(j) << " " << oldLoc << " " << nbpts );
299 for (int l = 0; l < nbpts; l++)
301 int oldval = oldPtsCell[l];
302 pointsCell[l] = idNodesOldToNew[oldval];
303 //MESSAGE(" " << oldval << " " << pointsCell[l]);
305 /*int newcnt = */newConnectivity->InsertNextCell(nbpts, pointsCell);
306 int newLoc = newConnectivity->GetInsertLocation(nbpts);
307 //MESSAGE(newcnt << " " << newLoc);
308 newLocations->SetValue(alreadyCopied, newLoc);
313 int SMDS_UnstructuredGrid::CellIdToDownId(int vtkCellId)
315 if((vtkCellId < 0) || (vtkCellId >= _cellIdToDownId.size()))
317 std::cerr << "SMDS_UnstructuredGrid::CellIdToDownId problem vtkCellId="
318 << vtkCellId << " max="<< _cellIdToDownId.size() << std::endl;
321 return _cellIdToDownId[vtkCellId];
324 void SMDS_UnstructuredGrid::setCellIdToDownId(int vtkCellId, int downId)
326 // ASSERT((vtkCellId >= 0) && (vtkCellId < _cellIdToDownId.size()));
327 _cellIdToDownId[vtkCellId] = downId;
330 void SMDS_UnstructuredGrid::CleanDownwardConnectivity()
332 for (int i = 0; i < _downArray.size(); i++)
335 delete _downArray[i];
338 _cellIdToDownId.clear();
341 /*! Build downward connectivity: to do only when needed because heavy memory load.
342 * Downward connectivity is no more valid if vtkUnstructuredGrid is modified.
345 void SMDS_UnstructuredGrid::BuildDownwardConnectivity(bool withEdges)
347 MESSAGE("SMDS_UnstructuredGrid::BuildDownwardConnectivity");CHRONO(2);
348 // TODO calcul partiel sans edges
350 // --- erase previous data if any
352 this->CleanDownwardConnectivity();
354 // --- create SMDS_Downward structures (in _downArray vector[vtkCellType])
356 _downArray.resize(VTK_MAXTYPE + 1, 0); // --- max. type value = VTK_QUADRATIC_PYRAMID
358 _downArray[VTK_LINE] = new SMDS_DownEdge(this);
359 _downArray[VTK_QUADRATIC_EDGE] = new SMDS_DownQuadEdge(this);
360 _downArray[VTK_TRIANGLE] = new SMDS_DownTriangle(this);
361 _downArray[VTK_QUADRATIC_TRIANGLE] = new SMDS_DownQuadTriangle(this);
362 _downArray[VTK_QUAD] = new SMDS_DownQuadrangle(this);
363 _downArray[VTK_QUADRATIC_QUAD] = new SMDS_DownQuadQuadrangle(this);
364 _downArray[VTK_TETRA] = new SMDS_DownTetra(this);
365 _downArray[VTK_QUADRATIC_TETRA] = new SMDS_DownQuadTetra(this);
366 _downArray[VTK_PYRAMID] = new SMDS_DownPyramid(this);
367 _downArray[VTK_QUADRATIC_PYRAMID] = new SMDS_DownQuadPyramid(this);
368 _downArray[VTK_WEDGE] = new SMDS_DownPenta(this);
369 _downArray[VTK_QUADRATIC_WEDGE] = new SMDS_DownQuadPenta(this);
370 _downArray[VTK_HEXAHEDRON] = new SMDS_DownHexa(this);
371 _downArray[VTK_QUADRATIC_HEXAHEDRON] = new SMDS_DownQuadHexa(this);
373 // --- get detailed info of number of cells of each type, allocate SMDS_downward structures
375 const SMDS_MeshInfo &meshInfo = _mesh->GetMeshInfo();
377 int nbLinTetra = meshInfo.NbTetras(ORDER_LINEAR);
378 int nbQuadTetra = meshInfo.NbTetras(ORDER_QUADRATIC);
379 int nbLinPyra = meshInfo.NbPyramids(ORDER_LINEAR);
380 int nbQuadPyra = meshInfo.NbPyramids(ORDER_QUADRATIC);
381 int nbLinPrism = meshInfo.NbPrisms(ORDER_LINEAR);
382 int nbQuadPrism = meshInfo.NbPrisms(ORDER_QUADRATIC);
383 int nbLinHexa = meshInfo.NbHexas(ORDER_LINEAR);
384 int nbQuadHexa = meshInfo.NbHexas(ORDER_QUADRATIC);
386 int nbLineGuess = int((4.0 / 3.0) * nbLinTetra + 2 * nbLinPrism + 2.5 * nbLinPyra + 3 * nbLinHexa);
387 int nbQuadEdgeGuess = int((4.0 / 3.0) * nbQuadTetra + 2 * nbQuadPrism + 2.5 * nbQuadPyra + 3 * nbQuadHexa);
388 int nbLinTriaGuess = 2 * nbLinTetra + nbLinPrism + 2 * nbLinPyra;
389 int nbQuadTriaGuess = 2 * nbQuadTetra + nbQuadPrism + 2 * nbQuadPyra;
390 int nbLinQuadGuess = int((2.0 / 3.0) * nbLinPrism + (1.0 / 2.0) * nbLinPyra + 3 * nbLinHexa);
391 int nbQuadQuadGuess = int((2.0 / 3.0) * nbQuadPrism + (1.0 / 2.0) * nbQuadPyra + 3 * nbQuadHexa);
393 int GuessSize[VTK_QUADRATIC_TETRA];
394 GuessSize[VTK_LINE] = nbLineGuess;
395 GuessSize[VTK_QUADRATIC_EDGE] = nbQuadEdgeGuess;
396 GuessSize[VTK_TRIANGLE] = nbLinTriaGuess;
397 GuessSize[VTK_QUADRATIC_TRIANGLE] = nbQuadTriaGuess;
398 GuessSize[VTK_QUAD] = nbLinQuadGuess;
399 GuessSize[VTK_QUADRATIC_QUAD] = nbQuadQuadGuess;
400 GuessSize[VTK_TETRA] = nbLinTetra;
401 GuessSize[VTK_QUADRATIC_TETRA] = nbQuadTetra;
402 GuessSize[VTK_PYRAMID] = nbLinPyra;
403 GuessSize[VTK_QUADRATIC_PYRAMID] = nbQuadPyra;
404 GuessSize[VTK_WEDGE] = nbLinPrism;
405 GuessSize[VTK_QUADRATIC_WEDGE] = nbQuadPrism;
406 GuessSize[VTK_HEXAHEDRON] = nbLinHexa;
407 GuessSize[VTK_QUADRATIC_HEXAHEDRON] = nbQuadHexa;
409 _downArray[VTK_LINE]->allocate(nbLineGuess);
410 _downArray[VTK_QUADRATIC_EDGE]->allocate(nbQuadEdgeGuess);
411 _downArray[VTK_TRIANGLE]->allocate(nbLinTriaGuess);
412 _downArray[VTK_QUADRATIC_TRIANGLE]->allocate(nbQuadTriaGuess);
413 _downArray[VTK_QUAD]->allocate(nbLinQuadGuess);
414 _downArray[VTK_QUADRATIC_QUAD]->allocate(nbQuadQuadGuess);
415 _downArray[VTK_TETRA]->allocate(nbLinTetra);
416 _downArray[VTK_QUADRATIC_TETRA]->allocate(nbQuadTetra);
417 _downArray[VTK_PYRAMID]->allocate(nbLinPyra);
418 _downArray[VTK_QUADRATIC_PYRAMID]->allocate(nbQuadPyra);
419 _downArray[VTK_WEDGE]->allocate(nbLinPrism);
420 _downArray[VTK_QUADRATIC_WEDGE]->allocate(nbQuadPrism);
421 _downArray[VTK_HEXAHEDRON]->allocate(nbLinHexa);
422 _downArray[VTK_QUADRATIC_HEXAHEDRON]->allocate(nbQuadHexa);
424 // --- iteration on vtkUnstructuredGrid cells, only faces
425 // for each vtk face:
426 // create a downward face entry with its downward id.
427 // compute vtk volumes, create downward volumes entry.
428 // mark face in downward volumes
429 // mark volumes in downward face
431 MESSAGE("--- iteration on vtkUnstructuredGrid cells, only faces");CHRONO(20);
432 int cellSize = this->Types->GetNumberOfTuples();
433 _cellIdToDownId.resize(cellSize, -1);
435 for (int i = 0; i < cellSize; i++)
437 int vtkFaceType = this->GetCellType(i);
438 if (SMDS_Downward::getCellDimension(vtkFaceType) == 2)
441 //ASSERT(_downArray[vtkFaceType]);
442 int connFaceId = _downArray[vtkFaceType]->addCell(vtkFaceId);
443 SMDS_Down2D* downFace = static_cast<SMDS_Down2D*> (_downArray[vtkFaceType]);
444 downFace->setTempNodes(connFaceId, vtkFaceId);
445 int vols[2] = { -1, -1 };
446 int nbVolumes = downFace->computeVolumeIds(vtkFaceId, vols);
447 //MESSAGE("nbVolumes="<< nbVolumes);
448 for (int ivol = 0; ivol < nbVolumes; ivol++)
450 int vtkVolId = vols[ivol];
451 int vtkVolType = this->GetCellType(vtkVolId);
452 //ASSERT(_downArray[vtkVolType]);
453 int connVolId = _downArray[vtkVolType]->addCell(vtkVolId);
454 _downArray[vtkVolType]->addDownCell(connVolId, connFaceId, vtkFaceType);
455 _downArray[vtkFaceType]->addUpCell(connFaceId, connVolId, vtkVolType);
456 // MESSAGE("Face " << vtkFaceId << " belongs to volume " << vtkVolId);
461 // --- iteration on vtkUnstructuredGrid cells, only volumes
462 // for each vtk volume:
463 // create downward volumes entry if not already done
464 // build a temporary list of faces described with their nodes
466 // compute the vtk volumes containing this face
467 // check if the face is already listed in the volumes (comparison of ordered list of nodes)
468 // if not, create a downward face entry (resizing of structure required)
469 // (the downward faces store a temporary list of nodes to ease the comparison)
470 // create downward volumes entry if not already done
471 // mark volumes in downward face
472 // mark face in downward volumes
475 MESSAGE("--- iteration on vtkUnstructuredGrid cells, only volumes");CHRONO(21);
477 for (int i = 0; i < cellSize; i++)
479 int vtkType = this->GetCellType(i);
480 if (SMDS_Downward::getCellDimension(vtkType) == 3)
484 // MESSAGE("vtk volume " << vtkVolId);
485 //ASSERT(_downArray[vtkType]);
486 /*int connVolId = */_downArray[vtkType]->addCell(vtkVolId);
488 // --- find all the faces of the volume, describe the faces by their nodes
490 SMDS_Down3D* downVol = static_cast<SMDS_Down3D*> (_downArray[vtkType]);
491 ListElemByNodesType facesWithNodes;
492 downVol->computeFacesWithNodes(vtkVolId, facesWithNodes);
493 // MESSAGE("vtk volume " << vtkVolId << " contains " << facesWithNodes.nbElems << " faces");
495 for (int iface = 0; iface < facesWithNodes.nbElems; iface++)
497 // --- find the volumes containing the face
500 int vtkFaceType = facesWithNodes.elems[iface].vtkType;
501 SMDS_Down2D* downFace = static_cast<SMDS_Down2D*> (_downArray[vtkFaceType]);
502 int vols[2] = { -1, -1 };
503 int *nodes = &facesWithNodes.elems[iface].nodeIds[0];
504 int lg = facesWithNodes.elems[iface].nbNodes;
505 int nbVolumes = downFace->computeVolumeIdsFromNodesFace(nodes, lg, vols);
506 // MESSAGE("vtk volume " << vtkVolId << " face " << iface << " belongs to " << nbVolumes << " volumes");
508 // --- check if face is registered in the volumes
513 for (int ivol = 0; ivol < nbVolumes; ivol++)
515 int vtkVolId2 = vols[ivol];
516 int vtkVolType = this->GetCellType(vtkVolId2);
517 //ASSERT(_downArray[vtkVolType]);
518 int connVolId2 = _downArray[vtkVolType]->addCell(vtkVolId2);
519 SMDS_Down3D* downVol2 = static_cast<SMDS_Down3D*> (_downArray[vtkVolType]);
520 connFaceId = downVol2->FindFaceByNodes(connVolId2, facesWithNodes.elems[iface]);
522 break; // --- face already created
525 // --- if face is not registered in the volumes, create face
530 connFaceId = _downArray[vtkFaceType]->addCell();
531 downFace->setTempNodes(connFaceId, facesWithNodes.elems[iface]);
534 // --- mark volumes in downward face and mark face in downward volumes
537 for (int ivol = 0; ivol < nbVolumes; ivol++)
539 int vtkVolId2 = vols[ivol];
540 int vtkVolType = this->GetCellType(vtkVolId2);
541 //ASSERT(_downArray[vtkVolType]);
542 int connVolId2 = _downArray[vtkVolType]->addCell(vtkVolId2);
543 _downArray[vtkVolType]->addDownCell(connVolId2, connFaceId, vtkFaceType);
544 _downArray[vtkFaceType]->addUpCell(connFaceId, connVolId2, vtkVolType);
545 // MESSAGE(" From volume " << vtkVolId << " face " << connFaceId << " belongs to volume " << vtkVolId2);
551 // --- iteration on vtkUnstructuredGrid cells, only edges
552 // for each vtk edge:
553 // create downward edge entry
554 // store the nodes id's in downward edge (redundant with vtkUnstructuredGrid)
555 // find downward faces
556 // (from vtk faces or volumes, get downward faces, they have a temporary list of nodes)
557 // mark edge in downward faces
558 // mark faces in downward edge
561 MESSAGE("--- iteration on vtkUnstructuredGrid cells, only edges");CHRONO(22);
563 for (int i = 0; i < cellSize; i++)
565 int vtkEdgeType = this->GetCellType(i);
566 if (SMDS_Downward::getCellDimension(vtkEdgeType) == 1)
569 //ASSERT(_downArray[vtkEdgeType]);
570 int connEdgeId = _downArray[vtkEdgeType]->addCell(vtkEdgeId);
571 SMDS_Down1D* downEdge = static_cast<SMDS_Down1D*> (_downArray[vtkEdgeType]);
572 downEdge->setNodes(connEdgeId, vtkEdgeId);
574 int nbVtkCells = downEdge->computeVtkCells(connEdgeId, vtkIds);
576 unsigned char downTypes[1000];
577 int nbDownFaces = downEdge->computeFaces(connEdgeId, &vtkIds[0], nbVtkCells, downFaces, downTypes);
578 for (int n = 0; n < nbDownFaces; n++)
580 _downArray[downTypes[n]]->addDownCell(downFaces[n], connEdgeId, vtkEdgeType);
581 _downArray[vtkEdgeType]->addUpCell(connEdgeId, downFaces[n], downTypes[n]);
586 // --- iteration on downward faces (they are all listed now)
587 // for each downward face:
588 // build a temporary list of edges with their ordered list of nodes
590 // find all the vtk cells containing this edge
591 // then identify all the downward faces containing the edge, from the vtk cells
592 // check if the edge is already listed in the faces (comparison of ordered list of nodes)
593 // if not, create a downward edge entry with the node id's
594 // mark edge in downward faces
595 // mark downward faces in edge (size of list unknown, to be allocated)
597 CHRONOSTOP(22);CHRONO(23);
599 for (int vtkFaceType = 0; vtkFaceType < VTK_QUADRATIC_PYRAMID; vtkFaceType++)
601 if (SMDS_Downward::getCellDimension(vtkFaceType) != 2)
604 // --- find all the edges of the face, describe the edges by their nodes
606 SMDS_Down2D* downFace = static_cast<SMDS_Down2D*> (_downArray[vtkFaceType]);
607 int maxId = downFace->getMaxId();
608 for (int faceId = 0; faceId < maxId; faceId++)
611 ListElemByNodesType edgesWithNodes;
612 downFace->computeEdgesWithNodes(faceId, edgesWithNodes);
613 // MESSAGE("downward face type " << vtkFaceType << " num " << faceId << " contains " << edgesWithNodes.nbElems << " edges");
616 for (int iedge = 0; iedge < edgesWithNodes.nbElems; iedge++)
619 // --- check if the edge is already registered by exploration of the faces
623 unsigned char vtkEdgeType = edgesWithNodes.elems[iedge].vtkType;
624 int *pts = &edgesWithNodes.elems[iedge].nodeIds[0];
625 SMDS_Down1D* downEdge = static_cast<SMDS_Down1D*> (_downArray[vtkEdgeType]);
626 int nbVtkCells = downEdge->computeVtkCells(pts, vtkIds);
627 //CHRONOSTOP(41);CHRONO(42);
629 unsigned char downTypes[1000];
630 int nbDownFaces = downEdge->computeFaces(pts, &vtkIds[0], nbVtkCells, downFaces, downTypes);
635 for (int idf = 0; idf < nbDownFaces; idf++)
637 int faceId2 = downFaces[idf];
638 int faceType = downTypes[idf];
639 //ASSERT(_downArray[faceType]);
640 SMDS_Down2D* downFace2 = static_cast<SMDS_Down2D*> (_downArray[faceType]);
641 connEdgeId = downFace2->FindEdgeByNodes(faceId2, edgesWithNodes.elems[iedge]);
643 break; // --- edge already created
646 // --- if edge is not registered in the faces, create edge
651 connEdgeId = _downArray[vtkEdgeType]->addCell();
652 downEdge->setNodes(connEdgeId, edgesWithNodes.elems[iedge].nodeIds);
656 // --- mark faces in downward edge and mark edge in downward faces
659 for (int idf = 0; idf < nbDownFaces; idf++)
661 int faceId2 = downFaces[idf];
662 int faceType = downTypes[idf];
663 //ASSERT(_downArray[faceType]);
664 //SMDS_Down2D* downFace2 = static_cast<SMDS_Down2D*> (_downArray[faceType]);
665 _downArray[vtkEdgeType]->addUpCell(connEdgeId, faceId2, faceType);
666 _downArray[faceType]->addDownCell(faceId2, connEdgeId, vtkEdgeType);
667 // MESSAGE(" From face t:" << vtkFaceType << " " << faceId <<
668 // " edge " << connEdgeId << " belongs to face t:" << faceType << " " << faceId2);
674 CHRONOSTOP(23);CHRONO(24);
676 // compact downward connectivity structure: adjust downward arrays size, replace vector<vector int>> by a single vector<int>
677 // 3D first then 2D and last 1D to release memory before edge upCells reorganization, (temporary memory use)
679 for (int vtkType = VTK_QUADRATIC_PYRAMID; vtkType >= 0; vtkType--)
681 if (SMDS_Downward *down = _downArray[vtkType])
683 down->compactStorage();
689 for (int vtkType = 0; vtkType <= VTK_QUADRATIC_PYRAMID; vtkType++)
691 if (SMDS_Downward *down = _downArray[vtkType])
693 if (down->getMaxId())
695 MESSAGE("Cells of Type " << vtkType << " : number of entities, est: "
696 << GuessSize[vtkType] << " real: " << down->getMaxId());
699 }CHRONOSTOP(24);CHRONOSTOP(2);
703 /*! Get the neighbors of a cell.
704 * Only the neighbors having the dimension of the cell are taken into account
705 * (neighbors of a volume are the volumes sharing a face with this volume,
706 * neighbors of a face are the faces sharing an edge with this face...).
707 * @param neighborsVtkIds vector of neighbors vtk id's to fill (reserve enough space).
708 * @param downIds downward id's of cells of dimension n-1, to fill (reserve enough space).
709 * @param downTypes vtk types of cells of dimension n-1, to fill (reserve enough space).
710 * @param vtkId the vtk id of the cell
711 * @return number of neighbors
713 int SMDS_UnstructuredGrid::GetNeighbors(int* neighborsVtkIds, int* downIds, unsigned char* downTypes, int vtkId)
715 int vtkType = this->GetCellType(vtkId);
716 int cellDim = SMDS_Downward::getCellDimension(vtkType);
718 return 0; // TODO voisins des edges = edges connectees
719 int cellId = this->_cellIdToDownId[vtkId];
721 int nbCells = _downArray[vtkType]->getNumberOfDownCells(cellId);
722 const int *downCells = _downArray[vtkType]->getDownCells(cellId);
723 const unsigned char* downTyp = _downArray[vtkType]->getDownTypes(cellId);
725 // --- iteration on faces of the 3D cell (or edges on the 2D cell).
728 for (int i = 0; i < nbCells; i++)
730 int downId = downCells[i];
731 int cellType = downTyp[i];
732 int nbUp = _downArray[cellType]->getNumberOfUpCells(downId);
733 const int *upCells = _downArray[cellType]->getUpCells(downId);
734 const unsigned char* upTypes = _downArray[cellType]->getUpTypes(downId);
736 // ---for a volume, max 2 upCells, one is this cell, the other is a neighbor
737 // for a face, number of neighbors (connected faces) not known
739 for (int j = 0; j < nbUp; j++)
741 if ((upCells[j] == cellId) && (upTypes[j] == vtkType))
743 int vtkNeighbor = _downArray[upTypes[j]]->getVtkCellId(upCells[j]);
744 neighborsVtkIds[nb] = vtkNeighbor;
745 downIds[nb] = downId;
746 downTypes[nb] = cellType;
749 if (nb >= NBMAXNEIGHBORS)
755 /*! get the volumes containing a face or an edge of the grid
756 * The edge or face belongs to the vtkUnstructuredGrid
757 * @param volVtkIds vector of parent volume ids to fill (reserve enough space!)
758 * @param vtkId vtk id of the face or edge
760 int SMDS_UnstructuredGrid::GetParentVolumes(int* volVtkIds, int vtkId)
762 int vtkType = this->GetCellType(vtkId);
763 int dim = SMDS_Downward::getCellDimension(vtkType);
766 unsigned char cellTypes[1000];
767 int downCellId[1000];
770 int downId = this->CellIdToDownId(vtkId);
771 nbFaces = _downArray[vtkType]->getNumberOfUpCells(downId);
772 const int *upCells = _downArray[vtkType]->getUpCells(downId);
773 const unsigned char* upTypes = _downArray[vtkType]->getUpTypes(downId);
774 for (int i=0; i< nbFaces; i++)
776 faces[i] = _downArray[upTypes[i]]->getVtkCellId(upCells[i]);
777 cellTypes[i] = upTypes[i];
778 downCellId[i] = upCells[i];
785 cellTypes[0] = this->GetCellType(vtkId);
786 downCellId[0] = this->CellIdToDownId(vtkId);
790 for (int i=0; i<nbFaces; i++)
792 int vtkTypeFace = cellTypes[i];
793 int downId = downCellId[i];
794 int nv = _downArray[vtkTypeFace]->getNumberOfUpCells(downId);
795 const int *upCells = _downArray[vtkTypeFace]->getUpCells(downId);
796 const unsigned char* upTypes = _downArray[vtkTypeFace]->getUpTypes(downId);
797 for (int j=0; j<nv; j++)
799 int vtkVolId = _downArray[upTypes[j]]->getVtkCellId(upCells[j]);
801 volVtkIds[nbvol++] = vtkVolId;
807 /*! get the volumes containing a face or an edge of the downward structure
808 * The edge or face does not necessary belong to the vtkUnstructuredGrid
809 * @param volVtkIds vector of parent volume ids to fill (reserve enough space!)
810 * @param downId id in the downward structure
811 * @param downType type of cell
813 int SMDS_UnstructuredGrid::GetParentVolumes(int* volVtkIds, int downId, unsigned char downType)
815 int vtkType = downType;
816 int dim = SMDS_Downward::getCellDimension(vtkType);
819 unsigned char cellTypes[1000];
820 int downCellId[1000];
823 nbFaces = _downArray[vtkType]->getNumberOfUpCells(downId);
824 const int *upCells = _downArray[vtkType]->getUpCells(downId);
825 const unsigned char* upTypes = _downArray[vtkType]->getUpTypes(downId);
826 for (int i=0; i< nbFaces; i++)
828 faces[i] = _downArray[upTypes[i]]->getVtkCellId(upCells[i]);
829 cellTypes[i] = upTypes[i];
830 downCellId[i] = upCells[i];
836 cellTypes[0] = vtkType;
837 downCellId[0] = downId;
841 for (int i=0; i<nbFaces; i++)
843 int vtkTypeFace = cellTypes[i];
844 int downId = downCellId[i];
845 int nv = _downArray[vtkTypeFace]->getNumberOfUpCells(downId);
846 const int *upCells = _downArray[vtkTypeFace]->getUpCells(downId);
847 const unsigned char* upTypes = _downArray[vtkTypeFace]->getUpTypes(downId);
848 for (int j=0; j<nv; j++)
850 int vtkVolId = _downArray[upTypes[j]]->getVtkCellId(upCells[j]);
852 volVtkIds[nbvol++] = vtkVolId;
858 /*! get the node id's of a cell.
859 * The cell is defined by it's downward connectivity id and type.
860 * @param nodeSet set of of vtk node id's to fill.
861 * @param downId downward connectivity id of the cell.
862 * @param downType type of cell.
864 void SMDS_UnstructuredGrid::GetNodeIds(std::set<int>& nodeSet, int downId, unsigned char downType)
866 _downArray[downType]->getNodeIds(downId, nodeSet);
869 /*! change some nodes in cell without modifying type or internal connectivity.
870 * Nodes inverse connectivity is maintained up to date.
871 * @param vtkVolId vtk id of the cell
872 * @param localClonedNodeIds map old node id to new node id.
874 void SMDS_UnstructuredGrid::ModifyCellNodes(int vtkVolId, std::map<int, int> localClonedNodeIds)
877 vtkIdType *pts; // will refer to the point id's of the face
878 this->GetCellPoints(vtkVolId, npts, pts);
879 for (int i = 0; i < npts; i++)
881 if (localClonedNodeIds.count(pts[i]))
883 vtkIdType oldpt = pts[i];
884 pts[i] = localClonedNodeIds[oldpt];
885 //MESSAGE(oldpt << " --> " << pts[i]);
886 //this->RemoveReferenceToCell(oldpt, vtkVolId);
887 //this->AddReferenceToCell(pts[i], vtkVolId);
892 /*! reorder the nodes of a face
893 * @param vtkVolId vtk id of a volume containing the face, to get an orientation for the face.
894 * @param orderedNodes list of nodes to reorder (in out)
895 * @return size of the list
897 int SMDS_UnstructuredGrid::getOrderedNodesOfFace(int vtkVolId, std::vector<vtkIdType>& orderedNodes)
899 int vtkType = this->GetCellType(vtkVolId);
900 int cellDim = SMDS_Downward::getCellDimension(vtkType);
903 SMDS_Down3D *downvol = static_cast<SMDS_Down3D*> (_downArray[vtkType]);
904 int downVolId = this->_cellIdToDownId[vtkVolId];
905 downvol->getOrderedNodesOfFace(downVolId, orderedNodes);
906 return orderedNodes.size();
909 void SMDS_UnstructuredGrid::BuildLinks()
911 // Remove the old links if they are already built
914 this->Links->UnRegister(this);
917 this->Links = SMDS_CellLinks::New();
918 this->Links->Allocate(this->GetNumberOfPoints());
919 this->Links->Register(this);
920 this->Links->BuildLinks(this, this->Connectivity);
921 this->Links->Delete();
924 /*! Create a volume (prism or hexahedron) by duplication of a face.
925 * Designed for use in creation of flat elements separating volume domains.
926 * A face separating two domains is shared by two volume cells.
927 * All the nodes are already created (for the two faces).
928 * Each original Node is associated to corresponding nodes in the domains.
929 * Some nodes may be duplicated for more than two domains, when domain separations intersect.
930 * In that case, even some of the nodes to use for the original face may be changed.
931 * @param vtkVolId: vtk id of a volume containing the face, to get an orientation for the face.
932 * @param domain1: domain of the original face
933 * @param domain2: domain of the duplicated face
934 * @param originalNodes: the vtk node ids of the original face
935 * @param nodeDomains: map(original id --> map(domain --> duplicated node id))
936 * @return ok if success.
938 SMDS_MeshVolume* SMDS_UnstructuredGrid::extrudeVolumeFromFace(int vtkVolId,
941 std::set<int>& originalNodes,
942 std::map<int, std::map<int, int> >& nodeDomains,
943 std::map<int, std::map<long, int> >& nodeQuadDomains)
945 //MESSAGE("extrudeVolumeFromFace " << vtkVolId);
946 vector<vtkIdType> orderedOriginals;
947 orderedOriginals.clear();
948 set<int>::const_iterator it = originalNodes.begin();
949 for (; it != originalNodes.end(); ++it)
950 orderedOriginals.push_back(*it);
952 int nbNodes = this->getOrderedNodesOfFace(vtkVolId, orderedOriginals);
953 vector<vtkIdType> orderedNodes;
955 switch (orderedOriginals.size())
959 for (int i = 0; i < nbNodes; i++)
960 orderedNodes.push_back(nodeDomains[orderedOriginals[i]][domain1]);
961 for (int i = 0; i < nbNodes; i++)
962 orderedNodes.push_back(nodeDomains[orderedOriginals[i]][domain2]);
969 long dom1_2; // for nodeQuadDomains
970 if (domain1 < domain2)
971 dom1_2 = dom1 + INT_MAX * dom2;
973 dom1_2 = dom2 + INT_MAX * dom1;
974 //cerr << "dom1=" << dom1 << " dom2=" << dom2 << " dom1_2=" << dom1_2 << endl;
975 int ima = orderedOriginals.size();
976 int mid = orderedOriginals.size() / 2;
977 //cerr << "ima=" << ima << " mid=" << mid << endl;
978 for (int i = 0; i < mid; i++)
979 orderedNodes.push_back(nodeDomains[orderedOriginals[i]][domain1]);
980 for (int i = 0; i < mid; i++)
981 orderedNodes.push_back(nodeDomains[orderedOriginals[i]][domain2]);
982 for (int i = mid; i < ima; i++)
983 orderedNodes.push_back(nodeDomains[orderedOriginals[i]][domain1]);
984 for (int i = mid; i < ima; i++)
985 orderedNodes.push_back(nodeDomains[orderedOriginals[i]][domain2]);
986 for (int i = 0; i < mid; i++)
988 int oldId = orderedOriginals[i];
990 if (nodeQuadDomains.count(oldId) && nodeQuadDomains[oldId].count(dom1_2))
991 newId = nodeQuadDomains[oldId][dom1_2];
994 double *coords = this->GetPoint(oldId);
995 SMDS_MeshNode *newNode = _mesh->AddNode(coords[0], coords[1], coords[2]);
996 newId = newNode->getVtkId();
997 std::map<long, int> emptyMap;
998 nodeQuadDomains[oldId] = emptyMap;
999 nodeQuadDomains[oldId][dom1_2] = newId;
1001 orderedNodes.push_back(newId);
1009 SMDS_MeshVolume *vol = _mesh->AddVolumeFromVtkIds(orderedNodes);
1011 // TODO update subshape list of elements and nodes