X-Git-Url: http://git.salome-platform.org/gitweb/?p=modules%2Fsmesh.git;a=blobdiff_plain;f=src%2FSMDS%2FSMDS_VtkCellIterator.cxx;h=55570417b330d78381b627a807b41c431ef8a48f;hp=ce511047bf8fc355f08317a9cecc667da386e055;hb=6d32f944a0a115b6419184c50b57bf7c4eef5786;hpb=b0a908c0d20341651771d0249fb10882f54b2aad diff --git a/src/SMDS/SMDS_VtkCellIterator.cxx b/src/SMDS/SMDS_VtkCellIterator.cxx index ce511047b..55570417b 100644 --- a/src/SMDS/SMDS_VtkCellIterator.cxx +++ b/src/SMDS/SMDS_VtkCellIterator.cxx @@ -1,4 +1,4 @@ -// Copyright (C) 2010-2014 CEA/DEN, EDF R&D, OPEN CASCADE +// Copyright (C) 2010-2019 CEA/DEN, EDF R&D, OPEN CASCADE // // This library is free software; you can redistribute it and/or // modify it under the terms of the GNU Lesser General Public @@ -20,198 +20,149 @@ #include "SMDS_VtkCellIterator.hxx" #include "utilities.h" -SMDS_VtkCellIterator::SMDS_VtkCellIterator(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType aType) : - _mesh(mesh), _cellId(vtkCellId), _index(0), _type(aType) +#include +#include + +_GetVtkNodes::_GetVtkNodes( vtkIdList* _vtkIdList, + SMDS_Mesh* mesh, + int vtkCellId, + SMDSAbs_EntityType aType ) { - vtkUnstructuredGrid* grid = _mesh->getGrid(); - _vtkIdList = vtkIdList::New(); + vtkUnstructuredGrid* grid = mesh->GetGrid(); const std::vector& interlace = SMDS_MeshCell::fromVtkOrder( aType ); if ( interlace.empty() ) { - grid->GetCellPoints(_cellId, _vtkIdList); - _nbNodes = _vtkIdList->GetNumberOfIds(); + grid->GetCellPoints( vtkCellId, _vtkIdList ); } else { vtkIdType npts, *pts; - grid->GetCellPoints( _cellId, npts, pts ); - _vtkIdList->SetNumberOfIds( _nbNodes = npts ); - for (int i = 0; i < _nbNodes; i++) + grid->GetCellPoints( vtkCellId, npts, pts ); + _vtkIdList->SetNumberOfIds( npts ); + for (int i = 0; i < npts; i++) _vtkIdList->SetId(i, pts[interlace[i]]); } } -SMDS_VtkCellIterator::~SMDS_VtkCellIterator() -{ - _vtkIdList->Delete(); -} - -bool SMDS_VtkCellIterator::more() -{ - return (_index < _nbNodes); -} - -const SMDS_MeshElement* SMDS_VtkCellIterator::next() -{ - vtkIdType id = _vtkIdList->GetId(_index++); - return _mesh->FindNodeVtk(id); -} - -SMDS_VtkCellIteratorToUNV::SMDS_VtkCellIteratorToUNV(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType aType) : - SMDS_VtkCellIterator() +_GetVtkNodesToUNV::_GetVtkNodesToUNV( vtkIdList* _vtkIdList, + SMDS_Mesh* mesh, + int vtkCellId, + SMDSAbs_EntityType aType ) { - _mesh = mesh; - _cellId = vtkCellId; - _index = 0; - _type = aType; - //MESSAGE("SMDS_VtkCellInterlacedIterator (UNV)" << _type); - - _vtkIdList = vtkIdList::New(); - vtkIdType* pts; - vtkUnstructuredGrid* grid = _mesh->getGrid(); - grid->GetCellPoints((vtkIdType)_cellId, (vtkIdType&)_nbNodes, pts); - _vtkIdList->SetNumberOfIds(_nbNodes); - int *ids = 0; - switch (_type) + vtkIdType * pts, npts; + vtkUnstructuredGrid* grid = mesh->GetGrid(); + grid->GetCellPoints( (vtkIdType)vtkCellId, npts, pts ); + const int *ids = 0; + switch (aType) + { + case SMDSEntity_Quad_Edge: + { + static int id[] = { 0, 2, 1 }; + ids = id; + break; + } + case SMDSEntity_Quad_Triangle: + case SMDSEntity_BiQuad_Triangle: + { + static int id[] = { 0, 3, 1, 4, 2, 5 }; + ids = id; + npts = 6; + break; + } + case SMDSEntity_Quad_Quadrangle: + case SMDSEntity_BiQuad_Quadrangle: + { + static int id[] = { 0, 4, 1, 5, 2, 6, 3, 7 }; + ids = id; + npts = 8; + break; + } + case SMDSEntity_Quad_Tetra: + { + static int id[] = { 0, 4, 1, 5, 2, 6, 7, 8, 9, 3 }; + ids = id; + break; + } + case SMDSEntity_Quad_Pyramid: + { + static int id[] = { 0, 5, 1, 6, 2, 7, 3, 8, 9, 10, 11, 12, 4 }; + ids = id; + break; + } + case SMDSEntity_Penta: + { + static int id[] = { 0, 2, 1, 3, 5, 4 }; + ids = id; + break; + } + case SMDSEntity_Quad_Penta: + case SMDSEntity_BiQuad_Penta: //TODO: check + { + static int id[] = { 0, 8, 2, 7, 1, 6, 12, 14, 13, 3, 11, 5, 10, 4, 9 }; + ids = id; + break; + } + case SMDSEntity_Quad_Hexa: + case SMDSEntity_TriQuad_Hexa: { - case SMDSEntity_Quad_Edge: - { - static int id[] = { 0, 2, 1 }; - ids = id; - break; - } - case SMDSEntity_Quad_Triangle: - case SMDSEntity_BiQuad_Triangle: - { - static int id[] = { 0, 3, 1, 4, 2, 5 }; - ids = id; - _nbNodes = 6; - break; - } - case SMDSEntity_Quad_Quadrangle: - case SMDSEntity_BiQuad_Quadrangle: - { - static int id[] = { 0, 4, 1, 5, 2, 6, 3, 7 }; - ids = id; - _nbNodes = 8; - break; - } - case SMDSEntity_Quad_Tetra: - { - static int id[] = { 0, 4, 1, 5, 2, 6, 7, 8, 9, 3 }; - ids = id; - break; - } - case SMDSEntity_Quad_Pyramid: - { - static int id[] = { 0, 5, 1, 6, 2, 7, 3, 8, 9, 10, 11, 12, 4 }; - ids = id; - break; - } - case SMDSEntity_Penta: - { - static int id[] = { 0, 2, 1, 3, 5, 4 }; - ids = id; - break; - } - case SMDSEntity_Quad_Penta: - { - static int id[] = { 0, 8, 2, 7, 1, 6, 12, 14, 13, 3, 11, 5, 10, 4, 9 }; - ids = id; - break; - } - case SMDSEntity_Quad_Hexa: - case SMDSEntity_TriQuad_Hexa: - { - static int id[] = { 0, 8, 1, 9, 2, 10, 3, 11, 16, 17, 18, 19, 4, 12, 5, 13, 6, 14, 7, 15 }; - ids = id; - _nbNodes = 20; - break; - } - case SMDSEntity_Polygon: - case SMDSEntity_Quad_Polygon: - case SMDSEntity_Polyhedra: - case SMDSEntity_Quad_Polyhedra: - default: - { - // static int id[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, - // 25, 26, 27, 28, 29 }; - // ids = id; - // break; - } + static int id[] = { 0, 8, 1, 9, 2, 10, 3, 11, 16, 17, 18, 19, 4, 12, 5, 13, 6, 14, 7, 15 }; + ids = id; + npts = 20; + break; } + case SMDSEntity_Polygon: + case SMDSEntity_Quad_Polygon: + case SMDSEntity_Polyhedra: + case SMDSEntity_Quad_Polyhedra: + default: + const std::vector& i = SMDS_MeshCell::interlacedSmdsOrder( aType, npts ); + if ( !i.empty() ) + ids = & i[0]; + } + + _vtkIdList->SetNumberOfIds( npts ); + if ( ids ) - for (int i = 0; i < _nbNodes; i++) + for (int i = 0; i < npts; i++) _vtkIdList->SetId(i, pts[ids[i]]); else - for (int i = 0; i < _nbNodes; i++) + for (int i = 0; i < npts; i++) _vtkIdList->SetId(i, pts[i]); } -bool SMDS_VtkCellIteratorToUNV::more() -{ - return SMDS_VtkCellIterator::more(); -} - -const SMDS_MeshNode* SMDS_VtkCellIteratorToUNV::next() -{ - return static_cast< const SMDS_MeshNode* >( SMDS_VtkCellIterator::next() ); -} - -SMDS_VtkCellIteratorToUNV::~SMDS_VtkCellIteratorToUNV() +_GetVtkNodesPolyh::_GetVtkNodesPolyh( vtkIdList* _vtkIdList, + SMDS_Mesh* mesh, + int vtkCellId, + SMDSAbs_EntityType aType ) { -} - -SMDS_VtkCellIteratorPolyH::SMDS_VtkCellIteratorPolyH(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType aType) : - SMDS_VtkCellIterator() -{ - _mesh = mesh; - _cellId = vtkCellId; - _index = 0; - _type = aType; - //MESSAGE("SMDS_VtkCellIteratorPolyH " << _type); - _vtkIdList = vtkIdList::New(); - vtkUnstructuredGrid* grid = _mesh->getGrid(); - grid->GetCellPoints(_cellId, _vtkIdList); - _nbNodes = _vtkIdList->GetNumberOfIds(); - switch (_type) + vtkUnstructuredGrid* grid = mesh->GetGrid(); + switch (aType) { - case SMDSEntity_Polyhedra: - { - //MESSAGE("SMDS_VtkCellIterator Polyhedra"); - vtkIdType nFaces = 0; - vtkIdType* ptIds = 0; - grid->GetFaceStream(_cellId, nFaces, ptIds); - int id = 0; - _nbNodesInFaces = 0; - for (int i = 0; i < nFaces; i++) - { - int nodesInFace = ptIds[id]; // nodeIds in ptIds[id+1 .. id+nodesInFace] - _nbNodesInFaces += nodesInFace; - id += (nodesInFace + 1); - } - _vtkIdList->SetNumberOfIds(_nbNodesInFaces); - id = 0; - int n = 0; - for (int i = 0; i < nFaces; i++) - { - int nodesInFace = ptIds[id]; // nodeIds in ptIds[id+1 .. id+nodesInFace] - for (int k = 1; k <= nodesInFace; k++) - _vtkIdList->SetId(n++, ptIds[id + k]); - id += (nodesInFace + 1); - } - break; - } - default: - assert(0); + case SMDSEntity_Polyhedra: + { + vtkIdType nFaces = 0; + vtkIdType* ptIds = 0; + grid->GetFaceStream( vtkCellId, nFaces, ptIds ); + int id = 0, nbNodesInFaces = 0; + for (int i = 0; i < nFaces; i++) + { + int nodesInFace = ptIds[id]; // nodeIds in ptIds[id+1 .. id+nodesInFace] + nbNodesInFaces += nodesInFace; + id += (nodesInFace + 1); + } + _vtkIdList->SetNumberOfIds( nbNodesInFaces ); + id = 0; + int n = 0; + for (int i = 0; i < nFaces; i++) + { + int nodesInFace = ptIds[id]; // nodeIds in ptIds[id+1 .. id+nodesInFace] + for (int k = 1; k <= nodesInFace; k++) + _vtkIdList->SetId(n++, ptIds[id + k]); + id += (nodesInFace + 1); + } + break; + } + default: + assert(0); } -} - -SMDS_VtkCellIteratorPolyH::~SMDS_VtkCellIteratorPolyH() -{ -} - -bool SMDS_VtkCellIteratorPolyH::more() -{ - return (_index < _nbNodesInFaces); }