CD_BoxLoopsImplem.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_BOXLOOPSIMPLEM_H
#define CD_BOXLOOPSIMPLEM_H
 
// Our includes
#include <CD_NamespaceHeader.H>
 
#if CH_SPACEDIM == 2
template <int Si, int Sj, typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(const Box& computeBox, Functor&& func)
{
  static_assert(Si >= 1 && Sj >= 1, "BoxLoops::loop strides must be >= 1");
 
  const int* lo = computeBox.loVect();
  const int* hi = computeBox.hiVect();
 
  for (int j = lo[1]; j <= hi[1]; j += Sj) {
    CD_PRAGMA_SIMD
    for (int i = lo[0]; i <= hi[0]; i += Si) {
      func(IntVect(i, j));
    }
  }
}
#else
template <int Si, int Sj, int Sk, typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(const Box& computeBox, Functor&& func)
{
  static_assert(Si >= 1 && Sj >= 1 && Sk >= 1, "BoxLoops::loop strides must be >= 1");
 
  const int* lo = computeBox.loVect();
  const int* hi = computeBox.hiVect();
 
  for (int k = lo[2]; k <= hi[2]; k += Sk) {
    for (int j = lo[1]; j <= hi[1]; j += Sj) {
      CD_PRAGMA_SIMD
      for (int i = lo[0]; i <= hi[0]; i += Si) {
        func(IntVect(i, j, k));
      }
    }
  }
}
#endif
 
template <typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(const IntVectSet& a_ivs, Functor&& a_kernel)
{
  for (IVSIterator iter(a_ivs); iter.ok(); ++iter) {
    a_kernel(iter());
  }
}
 
template <typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(const DenseIntVectSet& a_ivs, Functor&& a_kernel)
{
  for (DenseIntVectSetIterator iter(a_ivs); iter.ok(); ++iter) {
    a_kernel(iter());
  }
}
 
template <typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(VoFIterator& iter, Functor&& a_kernel)
{
 
  // TLDR: This runs through all cells in the vof-iterator and calls the kernel.
  for (iter.reset(); iter.ok(); ++iter) {
    a_kernel(iter());
  }
}
 
template <typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(FaceIterator& iter, Functor&& a_kernel)
{
 
  // TLDR: This runs through all cells in the vof-iterator and calls the kernel.
  for (iter.reset(); iter.ok(); ++iter) {
    a_kernel(iter());
  }
}
 
template <typename T, typename Functor>
ALWAYS_INLINE void
BoxLoops::loop(const Vector<T>& a_subset, Functor&& a_kernel)
{
  const std::vector<T>& stdVec = ((Vector<T>&)a_subset).stdVector();
 
  for (const auto& v : stdVec) {
    a_kernel(v);
  }
}
 
#include <CD_NamespaceFooter.H>
 
#endif