-// Copyright (C) 2010-2015 CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2010-2022 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
#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 <vtkCell.h>
+#include <vtkIdList.h>
+
+_GetVtkNodes::_GetVtkNodes( TVtkIdList& vtkIds,
+ SMDS_Mesh* mesh,
+ vtkIdType vtkCellId,
+ SMDSAbs_EntityType type )
{
- vtkUnstructuredGrid* grid = _mesh->getGrid();
- _vtkIdList = vtkIdList::New();
- const std::vector<int>& interlace = SMDS_MeshCell::fromVtkOrder( aType );
+ vtkUnstructuredGrid* grid = mesh->GetGrid();
+ const std::vector<int>& interlace = SMDS_MeshCell::fromVtkOrder( type );
+ vtkIdType npts;
+ vtkIdType const *pts(nullptr);
+ grid->GetCellPoints( vtkCellId, npts, pts );
+ vtkIds.resize( npts );
if ( interlace.empty() )
{
- grid->GetCellPoints(_cellId, _vtkIdList);
- _nbNodes = _vtkIdList->GetNumberOfIds();
+ vtkIds.assign( pts, pts + npts );
}
else
{
- vtkIdType npts, *pts;
- grid->GetCellPoints( _cellId, npts, pts );
- _vtkIdList->SetNumberOfIds( _nbNodes = npts );
- for (int i = 0; i < _nbNodes; i++)
- _vtkIdList->SetId(i, pts[interlace[i]]);
+ for (vtkIdType i = 0; i < npts; i++)
+ vtkIds[ 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( TVtkIdList& vtkIds,
+ SMDS_Mesh* mesh,
+ vtkIdType vtkCellId,
+ SMDSAbs_EntityType type )
{
- _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)
+ vtkUnstructuredGrid* grid = mesh->GetGrid();
+ vtkIdType npts;
+ vtkIdType const *pts(nullptr);
+ grid->GetCellPoints( vtkCellId, npts, pts );
+ const int *ids = 0;
+ switch ( type )
+ {
+ case SMDSEntity_Quad_Edge:
{
- 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, 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:
+ {
+ 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<int>& i = SMDS_MeshCell::interlacedSmdsOrder( type, npts );
+ if ( !i.empty() )
+ ids = & i[0];
}
- if ( ids )
- for (int i = 0; i < _nbNodes; i++)
- _vtkIdList->SetId(i, pts[ids[i]]);
- else
- for (int i = 0; i < _nbNodes; 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() );
-}
+ vtkIds.resize( npts );
-SMDS_VtkCellIteratorToUNV::~SMDS_VtkCellIteratorToUNV()
-{
+ if ( ids )
+ for (int i = 0; i < npts; i++)
+ vtkIds[ i ] = pts[ ids[i] ];
+ else
+ vtkIds.assign( pts, pts + npts );
}
-SMDS_VtkCellIteratorPolyH::SMDS_VtkCellIteratorPolyH(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType aType) :
- SMDS_VtkCellIterator()
+_GetVtkNodesPolyh::_GetVtkNodesPolyh( TVtkIdList& vtkIds,
+ SMDS_Mesh* mesh,
+ vtkIdType vtkCellId,
+ SMDSAbs_EntityType type )
{
- _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 ( type )
{
- 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 const *ptIds(nullptr);
+ 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);
+ }
+ vtkIds.resize( 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++ )
+ vtkIds[ n++ ] = ptIds[ id + k ];
+ id += (nodesInFace + 1);
+ }
+ break;
+ }
+ default:
+ assert(0);
}
-}
-
-SMDS_VtkCellIteratorPolyH::~SMDS_VtkCellIteratorPolyH()
-{
-}
-
-bool SMDS_VtkCellIteratorPolyH::more()
-{
- return (_index < _nbNodesInFaces);
}