CD_ParticleLoops.H

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2021-2026 SINTEF Energy Research
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 */
 
#ifndef CD_PARTICLELOOPS_H
#define CD_PARTICLELOOPS_H
 
// Std includes
#include <cstddef>
 
// Our includes
#include <CD_Decorations.H>
#include <CD_ParticleSoA.H>
#include <CD_NamespaceHeader.H>
 
namespace ParticleLoops {
 
template <typename P, typename Traits, typename Functor>
ALWAYS_INLINE void
loop(const ParticleSoA<P, Traits>& a_soa, Functor&& a_kernel)
{
  const std::size_t numParticles = a_soa.size();
 
  CD_PRAGMA_SIMD
  for (std::size_t i = 0; i < numParticles; i++) {
    a_kernel(i);
  }
}
 
template <typename P, typename Traits, typename T, typename Functor>
ALWAYS_INLINE T
reduce(const ParticleSoA<P, Traits>& a_soa, T a_initial, Functor&& a_kernel)
{
  const std::size_t numParticles = a_soa.size();
 
  for (std::size_t i = 0; i < numParticles; i++) {
    a_initial = a_kernel(a_initial, i);
  }
 
  return a_initial;
}
 
} // namespace ParticleLoops
 
#include <CD_NamespaceFooter.H>
 
#endif