src/SMDS/SMDS_VtkCellIterator.cxx

   1
   2 #include "SMDS_VtkCellIterator.hxx"
   3
   4 SMDS_VtkCellIterator::SMDS_VtkCellIterator(SMDS_Mesh* mesh, int vtkCellId):
   5   _mesh(mesh), _cellId(vtkCellId), _index(0)
   6 {
   7   vtkUnstructuredGrid* grid = _mesh->getGrid();
   8   _vtkIdList = grid->GetCell(_cellId)->GetPointIds();
   9   _nbNodes = _vtkIdList->GetNumberOfIds();
  10 }
  11
  12 bool SMDS_VtkCellIterator::more()
  13 {
  14   return (_index < _nbNodes);
  15 }
  16
  17 const SMDS_MeshElement* SMDS_VtkCellIterator::next()
  18 {
  19   vtkIdType id = _vtkIdList->GetId(_index++);
  20   return _mesh->FindNode(id);
  21 }