CellBlockUtilities.hpp

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 TotalEnergies
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_MESH_GENERATORS_CELLBLOCKUTILITIES_HPP_
#define GEOS_MESH_GENERATORS_CELLBLOCKUTILITIES_HPP_
 
#include "mesh/ElementType.hpp"
#include "common/DataTypes.hpp"
#include "common/Span.hpp"
#include "common/GEOS_RAJA_Interface.hpp"
 
namespace geos
{
 
template< typename T >
struct ToCellRelation
{
  T toBlockIndex; 
  T toCellIndex;  
 
  ToCellRelation( T const & toBlockIndex_,
                  T const & toCellIndex_ )
    : toBlockIndex( toBlockIndex_ ),
    toCellIndex( toCellIndex_ )
  { }
 
  ToCellRelation( T && toBlockIndex_,
                  T && toCellIndex_ )
    : toBlockIndex( toBlockIndex_ ),
    toCellIndex( toCellIndex_ )
  { }
 
  ToCellRelation() = default;
};
 
localIndex buildEdgeMaps( localIndex numNodes,
                          ArrayOfArraysView< localIndex const > const & faceToNodeMap,
                          ArrayOfArrays< localIndex > & faceToEdgeMap,
                          ArrayOfArrays< localIndex > & edgeToFaceMap,
                          array2d< localIndex > & edgeToNodeMap );
 
localIndex getFaceNodes( ElementType const elementType,
                         localIndex const elemIdx,
                         localIndex const faceNumber,
                         arrayView2d< localIndex const, cells::NODE_MAP_USD > const & elementToNodes,
                         Span< localIndex > const faceNodes );
 
template< typename POLICY, typename T, typename COMP = std::equal_to<> >
array1d< localIndex >
computeUniqueValueOffsets( ArrayOfArraysView< T const > const & sortedLists, COMP && comp = {} )
{
  localIndex const numNodes = sortedLists.size();
  array1d< localIndex > uniqueValueOffsets( numNodes + 1 );
 
  // For each node, count number of unique edges that have the node as its lowest
  arrayView1d< localIndex > const numUniqueValuesView = uniqueValueOffsets.toView();
  forAll< POLICY >( numNodes, [sortedLists, numUniqueValuesView, comp=std::forward< COMP >( comp )]( localIndex const i )
  {
    arraySlice1d< T const > const list = sortedLists[ i ];
    forEqualRanges( list.begin(), list.end(), [&]( auto, auto )
    {
      ++numUniqueValuesView[ i + 1 ];
    }, comp );
  } );
 
  // Perform an inplace prefix-sum to get the unique edge offset.
  RAJA::inclusive_scan_inplace< POLICY >( RAJA::make_span( uniqueValueOffsets.data(), uniqueValueOffsets.size() ) );
  return uniqueValueOffsets;
}
 
}
 
#endif // GEOS_MESH_GENERATORS_CELLBLOCKUTILITIES_HPP_