1 // Copyright (C) 2010-2016 CEA/DEN, EDF R&D, OPEN CASCADE
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "SMDS_VtkVolume.hxx"
21 #include "SMDS_MeshNode.hxx"
22 #include "SMDS_Mesh.hxx"
23 #include "SMDS_VtkCellIterator.hxx"
25 #include "utilities.h"
29 SMDS_VtkVolume::SMDS_VtkVolume()
33 SMDS_VtkVolume::SMDS_VtkVolume(const std::vector<vtkIdType>& nodeIds, SMDS_Mesh* mesh)
38 * typed used are vtk types (@see vtkCellType.h)
39 * see GetEntityType() for conversion in SMDS type (@see SMDSAbs_ElementType.hxx)
41 void SMDS_VtkVolume::init(const std::vector<vtkIdType>& nodeIds, SMDS_Mesh* mesh)
43 SMDS_MeshVolume::init();
44 myMeshId = mesh->getMeshId();
45 vtkIdType aType = VTK_TETRA;
46 switch (nodeIds.size()) // cases are in order of usage frequency
48 case 4: aType = VTK_TETRA; break;
49 case 8: aType = VTK_HEXAHEDRON; break;
50 case 5: aType = VTK_PYRAMID; break;
51 case 6: aType = VTK_WEDGE; break;
52 case 10: aType = VTK_QUADRATIC_TETRA; break;
53 case 20: aType = VTK_QUADRATIC_HEXAHEDRON; break;
54 case 13: aType = VTK_QUADRATIC_PYRAMID; break;
55 case 15: aType = VTK_QUADRATIC_WEDGE; break;
56 case 18: aType = VTK_BIQUADRATIC_QUADRATIC_WEDGE; break;
57 case 12: aType = VTK_HEXAGONAL_PRISM; break;
58 case 27: aType = VTK_TRIQUADRATIC_HEXAHEDRON; break;
59 default: aType = VTK_HEXAHEDRON;
61 myVtkID = mesh->getGrid()->InsertNextLinkedCell(aType, nodeIds.size(), (vtkIdType *) &nodeIds[0]);
62 mesh->setMyModified();
65 void SMDS_VtkVolume::initPoly(const std::vector<vtkIdType>& nodeIds,
66 const std::vector<int>& nbNodesPerFace,
69 SMDS_MeshVolume::init();
70 SMDS_UnstructuredGrid* grid = mesh->getGrid();
72 //this->gravityCenter(grid, &nodeIds[0], nodeIds.size(), ¢er[0]);
73 std::vector<vtkIdType> ptIds;
74 vtkIdType nbFaces = nbNodesPerFace.size();
76 for (int i = 0; i < nbFaces; i++)
78 int nf = nbNodesPerFace[i];
80 // EAP: a right approach is:
81 // - either the user should care of order of nodes or
82 // - the user should use a service method arranging nodes if he
83 // don't want or can't to do it by him-self
84 // The method below works OK only with planar faces and convex polyhedrones
89 // grid->GetPoints()->GetPoint(nodeIds[k], a);
90 // grid->GetPoints()->GetPoint(nodeIds[k + 1], b);
91 // grid->GetPoints()->GetPoint(nodeIds[k + 2], c);
92 // bool isFaceForward = this->isForward(a, b, c, center);
93 const vtkIdType *facePts = &nodeIds[k];
95 for (int n = 0; n < nf; n++)
96 ptIds.push_back(facePts[n]);
98 // for (int n = nf - 1; n >= 0; n--)
99 // ptIds.push_back(facePts[n]);
102 myVtkID = grid->InsertNextLinkedCell(VTK_POLYHEDRON, nbFaces, &ptIds[0]);
103 mesh->setMyModified();
106 bool SMDS_VtkVolume::ChangeNodes(const SMDS_MeshNode* nodes[], const int nbNodes)
108 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
111 grid->GetCellPoints(myVtkID, npts, pts);
114 MESSAGE("ChangeNodes problem: not the same number of nodes " << npts << " -> " << nbNodes);
117 for (int i = 0; i < nbNodes; i++)
119 pts[i] = nodes[i]->getVtkId();
121 SMDS_Mesh::_meshList[myMeshId]->setMyModified();
126 * Reorder in VTK order a list of nodes given in SMDS order.
127 * To be used before ChangeNodes: lists are given or computed in SMDS order.
129 bool SMDS_VtkVolume::vtkOrder(const SMDS_MeshNode* nodes[], const int nbNodes)
131 if (nbNodes != this->NbNodes())
133 MESSAGE("vtkOrder, wrong number of nodes " << nbNodes << " instead of "<< this->NbNodes());
136 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
137 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
138 const std::vector<int>& interlace = SMDS_MeshCell::toVtkOrder( VTKCellType( aVtkType ));
139 if ( !interlace.empty() )
141 ASSERT( (int)interlace.size() == nbNodes );
142 std::vector<const SMDS_MeshNode*> initNodes( nodes, nodes+nbNodes );
143 for ( size_t i = 0; i < interlace.size(); ++i )
144 nodes[i] = initNodes[ interlace[i] ];
149 SMDS_VtkVolume::~SMDS_VtkVolume()
153 void SMDS_VtkVolume::Print(ostream & OS) const
155 OS << "volume <" << GetID() << "> : ";
158 int SMDS_VtkVolume::NbFaces() const
160 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
161 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
166 case VTK_QUADRATIC_TETRA:
171 case VTK_QUADRATIC_PYRAMID:
172 case VTK_QUADRATIC_WEDGE:
173 case VTK_BIQUADRATIC_QUADRATIC_WEDGE:
177 case VTK_QUADRATIC_HEXAHEDRON:
178 case VTK_TRIQUADRATIC_HEXAHEDRON:
183 vtkIdType nFaces = 0;
184 vtkIdType* ptIds = 0;
185 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
189 case VTK_HEXAGONAL_PRISM:
193 MESSAGE("invalid volume type")
201 int SMDS_VtkVolume::NbNodes() const
203 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
204 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
205 vtkIdType nbPoints = 0;
206 if (aVtkType != VTK_POLYHEDRON)
209 grid->GetCellPoints( myVtkID, nbPoints, pts );
213 vtkIdType nFaces = 0;
214 vtkIdType* ptIds = 0;
215 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
217 for (int i = 0; i < nFaces; i++)
219 int nodesInFace = ptIds[id];
220 nbPoints += nodesInFace;
221 id += (nodesInFace + 1);
227 int SMDS_VtkVolume::NbEdges() const
229 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
230 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
235 case VTK_QUADRATIC_TETRA:
239 case VTK_QUADRATIC_PYRAMID:
243 case VTK_QUADRATIC_WEDGE:
244 case VTK_BIQUADRATIC_QUADRATIC_WEDGE:
248 case VTK_QUADRATIC_HEXAHEDRON:
249 case VTK_TRIQUADRATIC_HEXAHEDRON:
254 vtkIdType nFaces = 0;
255 vtkIdType* ptIds = 0;
256 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
259 for (int i = 0; i < nFaces; i++)
261 int edgesInFace = ptIds[id];
262 id += (edgesInFace + 1);
263 nbEdges += edgesInFace;
265 nbEdges = nbEdges / 2;
268 case VTK_HEXAGONAL_PRISM:
272 MESSAGE("invalid volume type")
281 * 1 <= face_ind <= NbFaces()
283 int SMDS_VtkVolume::NbFaceNodes(const int face_ind) const
285 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
286 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
288 if (aVtkType == VTK_POLYHEDRON)
290 vtkIdType nFaces = 0;
291 vtkIdType* ptIds = 0;
292 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
294 for (int i = 0; i < nFaces; i++)
296 int nodesInFace = ptIds[id];
297 id += (nodesInFace + 1);
298 if (i == face_ind - 1)
300 nbNodes = nodesInFace;
309 * 1 <= face_ind <= NbFaces()
310 * 1 <= node_ind <= NbFaceNodes()
312 const SMDS_MeshNode* SMDS_VtkVolume::GetFaceNode(const int face_ind, const int node_ind) const
314 SMDS_Mesh *mesh = SMDS_Mesh::_meshList[myMeshId];
315 vtkUnstructuredGrid* grid = mesh->getGrid();
316 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
317 const SMDS_MeshNode* node = 0;
318 if (aVtkType == VTK_POLYHEDRON)
320 vtkIdType nFaces = 0;
321 vtkIdType* ptIds = 0;
322 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
324 for (int i = 0; i < nFaces; i++)
326 int nodesInFace = ptIds[id]; // nodeIds in ptIds[id+1 .. id+nodesInFace]
327 if (i == face_ind - 1) // first face is number 1
329 if ((node_ind > 0) && (node_ind <= nodesInFace))
330 node = mesh->FindNodeVtk(ptIds[id + node_ind]); // ptIds[id+1] : first node
333 id += (nodesInFace + 1);
340 * return number of nodes for each face
342 std::vector<int> SMDS_VtkVolume::GetQuantities() const
344 std::vector<int> quantities;
345 SMDS_Mesh *mesh = SMDS_Mesh::_meshList[myMeshId];
346 vtkUnstructuredGrid* grid = mesh->getGrid();
347 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
348 if (aVtkType == VTK_POLYHEDRON)
350 vtkIdType nFaces = 0;
351 vtkIdType* ptIds = 0;
352 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
354 for (int i = 0; i < nFaces; i++)
356 int nodesInFace = ptIds[id]; // nodeIds in ptIds[id+1 .. id+nodesInFace]
357 quantities.push_back(nodesInFace);
358 id += (nodesInFace + 1);
364 SMDS_ElemIteratorPtr SMDS_VtkVolume::elementsIterator(SMDSAbs_ElementType type) const
370 SMDSAbs_EntityType aType = this->GetEntityType();
371 if (aType == SMDSEntity_Polyhedra)
372 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIteratorPolyH(SMDS_Mesh::_meshList[myMeshId], myVtkID, aType));
374 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(SMDS_Mesh::_meshList[myMeshId], myVtkID, aType));
377 MESSAGE("ERROR : Iterator not implemented");
378 return SMDS_ElemIteratorPtr((SMDS_ElemIterator*) NULL);
382 SMDS_NodeIteratorPtr SMDS_VtkVolume::nodesIteratorToUNV() const
384 return SMDS_NodeIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
387 SMDS_NodeIteratorPtr SMDS_VtkVolume::interlacedNodesIterator() const
389 return SMDS_NodeIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
392 SMDSAbs_ElementType SMDS_VtkVolume::GetType() const
394 return SMDSAbs_Volume;
398 * \brief Return node by its index
399 * \param ind - node index
400 * \retval const SMDS_MeshNode* - the node
402 const SMDS_MeshNode* SMDS_VtkVolume::GetNode(const int ind) const
404 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
405 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
406 if ( aVtkType == VTK_POLYHEDRON)
408 vtkIdType nFaces = 0;
409 vtkIdType* ptIds = 0;
410 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
411 int id = 0, nbPoints = 0;
412 for (int i = 0; i < nFaces; i++)
414 int nodesInFace = ptIds[id];
415 if ( ind < nbPoints + nodesInFace )
416 return SMDS_Mesh::_meshList[myMeshId]->FindNodeVtk( ptIds[ ind + i ]);
417 nbPoints += nodesInFace;
418 id += (nodesInFace + 1);
422 vtkIdType npts, *pts;
423 grid->GetCellPoints( this->myVtkID, npts, pts );
424 const std::vector<int>& interlace = SMDS_MeshCell::fromVtkOrder( VTKCellType( aVtkType ));
425 return SMDS_Mesh::_meshList[myMeshId]->FindNodeVtk( pts[ interlace.empty() ? ind : interlace[ind]] );
428 * \brief Check if a node belongs to the element
429 * \param node - the node to check
430 * \retval int - node index within the element, -1 if not found
432 int SMDS_VtkVolume::GetNodeIndex( const SMDS_MeshNode* node ) const
434 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
435 const vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
436 if ( aVtkType == VTK_POLYHEDRON)
438 vtkIdType nFaces = 0;
439 vtkIdType* ptIds = 0;
440 grid->GetFaceStream(this->myVtkID, nFaces, ptIds);
442 for (int iF = 0; iF < nFaces; iF++)
444 int nodesInFace = ptIds[id];
445 for ( vtkIdType i = 0; i < nodesInFace; ++i )
446 if ( ptIds[id+i+1] == node->getVtkId() )
448 id += (nodesInFace + 1);
452 vtkIdType npts, *pts;
453 grid->GetCellPoints( this->myVtkID, npts, pts );
454 for ( vtkIdType i = 0; i < npts; ++i )
455 if ( pts[i] == node->getVtkId() )
457 const std::vector<int>& interlace = SMDS_MeshCell::toVtkOrder( VTKCellType( aVtkType ));
458 return interlace.empty() ? i : interlace[i];
463 bool SMDS_VtkVolume::IsQuadratic() const
465 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
466 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
467 // TODO quadratic polyhedrons ?
470 case VTK_QUADRATIC_TETRA:
471 case VTK_QUADRATIC_PYRAMID:
472 case VTK_QUADRATIC_WEDGE:
473 case VTK_BIQUADRATIC_QUADRATIC_WEDGE:
474 case VTK_QUADRATIC_HEXAHEDRON:
475 case VTK_TRIQUADRATIC_HEXAHEDRON:
483 bool SMDS_VtkVolume::IsPoly() const
485 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
486 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
487 return (aVtkType == VTK_POLYHEDRON);
490 bool SMDS_VtkVolume::IsMediumNode(const SMDS_MeshNode* node) const
492 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
493 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
494 int rankFirstMedium = 0;
497 case VTK_QUADRATIC_TETRA:
498 rankFirstMedium = 4; // medium nodes are of rank 4 to 9
500 case VTK_QUADRATIC_PYRAMID:
501 rankFirstMedium = 5; // medium nodes are of rank 5 to 12
503 case VTK_QUADRATIC_WEDGE:
504 case VTK_BIQUADRATIC_QUADRATIC_WEDGE:
505 rankFirstMedium = 6; // medium nodes are of rank 6 to 14
507 case VTK_QUADRATIC_HEXAHEDRON:
508 case VTK_TRIQUADRATIC_HEXAHEDRON:
509 rankFirstMedium = 8; // medium nodes are of rank 8 to 19
516 grid->GetCellPoints(myVtkID, npts, pts);
517 vtkIdType nodeId = node->getVtkId();
518 for (int rank = 0; rank < npts; rank++)
520 if (pts[rank] == nodeId)
522 if (rank < rankFirstMedium)
528 MESSAGE("======================================================");
529 MESSAGE("= IsMediumNode: node does not belong to this element =");
530 MESSAGE("======================================================");
534 int SMDS_VtkVolume::NbCornerNodes() const
536 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
537 vtkIdType aVtkType = grid->GetCellType(myVtkID);
540 case VTK_QUADRATIC_TETRA: return 4;
541 case VTK_QUADRATIC_PYRAMID: return 5;
542 case VTK_QUADRATIC_WEDGE:
543 case VTK_BIQUADRATIC_QUADRATIC_WEDGE: return 6;
544 case VTK_QUADRATIC_HEXAHEDRON:
545 case VTK_TRIQUADRATIC_HEXAHEDRON: return 8;
551 SMDSAbs_EntityType SMDS_VtkVolume::GetEntityType() const
553 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
554 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
556 SMDSAbs_EntityType aType = SMDSEntity_Tetra;
560 aType = SMDSEntity_Tetra;
563 aType = SMDSEntity_Pyramid;
566 aType = SMDSEntity_Penta;
569 aType = SMDSEntity_Hexa;
571 case VTK_QUADRATIC_TETRA:
572 aType = SMDSEntity_Quad_Tetra;
574 case VTK_QUADRATIC_PYRAMID:
575 aType = SMDSEntity_Quad_Pyramid;
577 case VTK_QUADRATIC_WEDGE:
578 aType = SMDSEntity_Quad_Penta;
580 case VTK_BIQUADRATIC_QUADRATIC_WEDGE:
581 aType = SMDSEntity_BiQuad_Penta;
583 case VTK_QUADRATIC_HEXAHEDRON:
584 aType = SMDSEntity_Quad_Hexa;
586 case VTK_TRIQUADRATIC_HEXAHEDRON:
587 aType = SMDSEntity_TriQuad_Hexa;
589 case VTK_HEXAGONAL_PRISM:
590 aType = SMDSEntity_Hexagonal_Prism;
593 aType = SMDSEntity_Polyhedra;
596 aType = SMDSEntity_Polyhedra;
602 SMDSAbs_GeometryType SMDS_VtkVolume::GetGeomType() const
604 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
605 vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
607 SMDSAbs_GeometryType aType = SMDSGeom_NONE;
611 case VTK_QUADRATIC_TETRA:
612 aType = SMDSGeom_TETRA;
615 case VTK_QUADRATIC_PYRAMID:
616 aType = SMDSGeom_PYRAMID;
619 case VTK_QUADRATIC_WEDGE:
620 case VTK_BIQUADRATIC_QUADRATIC_WEDGE:
621 aType = SMDSGeom_PENTA;
624 case VTK_QUADRATIC_HEXAHEDRON:
625 case VTK_TRIQUADRATIC_HEXAHEDRON:
626 aType = SMDSGeom_HEXA;
628 case VTK_HEXAGONAL_PRISM:
629 aType = SMDSGeom_HEXAGONAL_PRISM;
632 aType = SMDSGeom_POLYHEDRA;
635 aType = SMDSGeom_POLYHEDRA;
641 vtkIdType SMDS_VtkVolume::GetVtkType() const
643 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
644 vtkIdType aType = grid->GetCellType(myVtkID);
648 void SMDS_VtkVolume::gravityCenter(SMDS_UnstructuredGrid* grid,
649 const vtkIdType * nodeIds,
653 for (int j = 0; j < 3; j++)
657 for (int i = 0; i < nbNodes; i++)
659 double *coords = grid->GetPoint(nodeIds[i]);
660 for (int j = 0; j < 3; j++)
661 result[j] += coords[j];
663 for (int j = 0; j < 3; j++)
664 result[j] = result[j] / nbNodes;
668 bool SMDS_VtkVolume::isForward(double* a, double* b, double* c, double* d)
670 double u[3], v[3], w[3];
671 for (int j = 0; j < 3; j++)
677 double prodmixte = ((u[1]*v[2] - u[2]*v[1]) * w[0]
678 + (u[2]*v[0] - u[0]*v[2]) * w[1]
679 + (u[0]*v[1] - u[1]*v[0]) * w[2] );
680 return (prodmixte < 0);
683 /*! For polyhedron only
684 * @return actual number of nodes (not the sum of nodes of all faces)
686 int SMDS_VtkVolume::NbUniqueNodes() const
688 vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
689 return grid->GetCell(myVtkID)->GetNumberOfPoints();
692 /*! For polyhedron use only
693 * @return iterator on actual nodes (not through the faces)
695 SMDS_ElemIteratorPtr SMDS_VtkVolume::uniqueNodesIterator() const
697 return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));