CollocatedNodes.hpp

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2020 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2020 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2020 TotalEnergies
 * Copyright (c) 2019-     GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_COLLOCATEDNODES_HPP
#define GEOS_COLLOCATEDNODES_HPP
 
#include "common/DataTypes.hpp"
 
#include <vtkDataSet.h>
#include <vtkIdTypeArray.h>
#include <vtkSmartPointer.h>
 
namespace geos::vtk
{
 
class CollocatedNodes
{
public:
  CollocatedNodes( string const & faceBlockName,
                   vtkSmartPointer< vtkDataSet > faceMesh,
                   bool isParallel = true );
 
  std::vector< vtkIdType > const & operator[]( std::size_t i ) const
  {
    return m_collocatedNodes[i];
  }
 
  [[nodiscard]] std::size_t size() const
  {
    return m_collocatedNodes.size();
  }
 
private:
 
  void init( vtkIdTypeArray const * collocatedNodes );
 
  std::vector< std::vector< vtkIdType > > m_collocatedNodes;
};
 
} // geos::vtk
 
#endif //GEOS_COLLOCATEDNODES_HPP