IndexSetUtils.hpp

Go to the documentation of this file.

 
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~//
// Copyright (c) Lawrence Livermore National Security, LLC and other
// RAJA Project Developers. See top-level LICENSE and COPYRIGHT
// files for dates and other details. No copyright assignment is required
// to contribute to RAJA.
//
// SPDX-License-Identifier: (BSD-3-Clause)
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~//
 
#ifndef RAJA_IndexSetUtils_HPP
#define RAJA_IndexSetUtils_HPP
 
#include "RAJA/config.hpp"
 
#include "RAJA/pattern/forall.hpp"
 
#include "RAJA/policy/sequential.hpp"
 
namespace RAJA
{
 
 
template<typename CONTAINER_T, typename... SEG_TYPES>
RAJA_INLINE void getIndices(CONTAINER_T& con,
                            const TypedIndexSet<SEG_TYPES...>& iset)
{
  CONTAINER_T tcon;
  forall<ExecPolicy<seq_segit, seq_exec>>(
      iset, [&](typename CONTAINER_T::value_type idx) {
        tcon.push_back(idx);
      });
  con = tcon;
}
 
template<typename CONTAINER_T, typename SEGMENT_T>
RAJA_INLINE void getIndices(CONTAINER_T& con, const SEGMENT_T& seg)
{
  CONTAINER_T tcon;
  forall<seq_exec>(seg, [&](typename CONTAINER_T::value_type idx) {
    tcon.push_back(idx);
  });
  con = tcon;
}
 
template<typename CONTAINER_T, typename... SEG_TYPES, typename CONDITIONAL>
RAJA_INLINE void getIndicesConditional(CONTAINER_T& con,
                                       const TypedIndexSet<SEG_TYPES...>& iset,
                                       CONDITIONAL conditional)
{
  CONTAINER_T tcon;
  forall<ExecPolicy<seq_segit, seq_exec>>(
      iset, [&](typename CONTAINER_T::value_type idx) {
        if (conditional(idx)) tcon.push_back(idx);
      });
  con = tcon;
}
 
template<typename CONTAINER_T, typename SEGMENT_T, typename CONDITIONAL>
RAJA_INLINE void getIndicesConditional(CONTAINER_T& con,
                                       const SEGMENT_T& seg,
                                       CONDITIONAL conditional)
{
  CONTAINER_T tcon;
  forall<seq_exec>(seg, [&](typename CONTAINER_T::value_type idx) {
    if (conditional(idx)) tcon.push_back(idx);
  });
  con = tcon;
}
 
 
}  // namespace RAJA
 
#endif  // closing endif for header file include guard