sliceHelpers.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2020, Lawrence Livermore National Security, LLC and LvArray contributors.
 * All rights reserved.
 * See the LICENSE file for details.
 * SPDX-License-Identifier: (BSD-3-Clause)
 */

#pragma once

// Source includes
#include "ArraySlice.hpp"

namespace LvArray
{

DISABLE_HD_WARNING
template< typename T, typename LAMBDA >
LVARRAY_HOST_DEVICE
void forValuesInSlice( T & value, LAMBDA && f )
{ f( value ); }

DISABLE_HD_WARNING
template< typename T, int NDIM, int USD, typename INDEX_TYPE, typename LAMBDA >
LVARRAY_HOST_DEVICE
void forValuesInSlice( ArraySlice< T, NDIM, USD, INDEX_TYPE > const slice, LAMBDA && f )
{
  INDEX_TYPE const bounds = slice.size( 0 );
  for( INDEX_TYPE i = 0; i < bounds; ++i )
  {
    forValuesInSlice( slice[ i ], f );
  }
}

DISABLE_HD_WARNING
template< typename T, typename LAMBDA, typename ... INDICES >
LVARRAY_HOST_DEVICE
void forValuesInSliceWithIndices( T & value, LAMBDA && f, INDICES const ... indices )
{ f( value, indices ... ); }

DISABLE_HD_WARNING
template< typename T, int NDIM, int USD, typename INDEX_TYPE, typename LAMBDA, typename ... INDICES >
LVARRAY_HOST_DEVICE
void forValuesInSliceWithIndices( ArraySlice< T, NDIM, USD, INDEX_TYPE > const slice,
                                  LAMBDA && f,
                                  INDICES const ... indices )
{
  INDEX_TYPE const bounds = slice.size( 0 );
  for( INDEX_TYPE i = 0; i < bounds; ++i )
  {
    forValuesInSliceWithIndices( slice[ i ], f, indices ..., i );
  }
}

template< typename T, int USD_SRC, typename INDEX_TYPE >
void sumOverFirstDimension( ArraySlice< T const, 1, USD_SRC, INDEX_TYPE > const src,
                            T & dst )
{
  INDEX_TYPE const bounds = src.size( 0 );
  for( INDEX_TYPE i = 0; i < bounds; ++i )
  {
    dst += src( i );
  }
}

template< typename T, int NDIM, int USD_SRC, int USD_DST, typename INDEX_TYPE >
void sumOverFirstDimension( ArraySlice< T const, NDIM, USD_SRC, INDEX_TYPE > const src,
                            ArraySlice< T, NDIM - 1, USD_DST, INDEX_TYPE > const dst )
{
#ifdef ARRAY_SLICE_CHECK_BOUNDS
  for( int i = 1; i < NDIM; ++i )
  {
    LVARRAY_ERROR_IF_NE( src.size( i ), dst.size( i - 1 ) );
  }
#endif

  forValuesInSliceWithIndices( src, [dst] ( T const & value, INDEX_TYPE const, auto const ... indices )
  {
    dst( indices ... ) += value;
  } );
}

} // namespace LvArray