VTKSuperCellPartitioning.hpp

 
#ifndef GEOS_MESH_GENERATORS_VTKSUPERCELLPARTITIONING_HPP
#define GEOS_MESH_GENERATORS_VTKSUPERCELLPARTITIONING_HPP
 
#include "common/DataTypes.hpp"
#include "common/MpiWrapper.hpp"
#include "common/TimingMacros.hpp"
#include "mesh/generators/VTKMeshGeneratorTools.hpp"
#include "mesh/generators/ParMETISInterface.hpp"
#include "LvArray/src/ArrayOfArrays.hpp"
 
#include <vtkSmartPointer.h>
 
class vtkUnstructuredGrid;
class vtkDataSet;
 
namespace geos
{
namespace vtk
{
 
enum class InitialDistributionStrategy
{
  MORTON,  
  BLOCK    
};
 
struct SuperCellInfo
{
  stdMap< vtkIdType, stdVector< vtkIdType > > superCellToOriginalCells;
 
  stdMap< vtkIdType, vtkIdType > vertexWeights;
 
  std::set< vtkIdType > atomicSuperCells;
};
 
SuperCellInfo tagCellsWithSuperCellIds(
  vtkSmartPointer< vtkUnstructuredGrid > cells3D,
  stdMap< string, ArrayOfArrays< localIndex, int64_t > > const & fractureNeighbors,
  integer fractureWeight );
 
SuperCellInfo reconstructSuperCellInfo( vtkSmartPointer< vtkUnstructuredGrid > mesh,
                                        integer fractureWeight );
 
vtkSmartPointer< vtkDataSet >
redistributeBySuperCellBlocks( vtkSmartPointer< vtkUnstructuredGrid > cells3D,
                               MPI_Comm comm,
                               InitialDistributionStrategy strategy = InitialDistributionStrategy::MORTON );
 
std::pair< ArrayOfArrays< pmet_idx_t, pmet_idx_t >, array1d< pmet_idx_t > >
buildSuperCellGraph(
  vtkSmartPointer< vtkUnstructuredGrid > cells3D,
  ArrayOfArrays< pmet_idx_t, pmet_idx_t > const & baseGraph,
  arrayView1d< pmet_idx_t const > const & baseElemDist,
  SuperCellInfo const & info,
  MPI_Comm comm );
 
void validateSuperCellGraph(
  ArrayOfArrays< pmet_idx_t, pmet_idx_t > const & superGraph,
  arrayView1d< pmet_idx_t const > const & superElemDist,
  arrayView1d< pmet_idx_t const > const & vertexWeights,
  MPI_Comm comm );
 
array1d< int64_t >
expandSuperCellPartitioningToCells(
  vtkSmartPointer< vtkUnstructuredGrid > cells3D,
  array1d< int64_t > const & superPartitioning,
  stdMap< vtkIdType, localIndex > const & superCellIdToLocalIdx,
  MPI_Comm comm );
 
} // namespace vtk
} // namespace geos
 
#endif /* GEOS_MESH_GENERATORS_VTKSUPERCELLPARTITIONING_HPP */