types.hpp

Go to the documentation of this file.

// core — shared types and conventions (see suite/docs/CONVENTIONS.md).
//
// Pillars: axis order is x-fastest (linear index I = x + y*nx + z*nx*ny), and SDF sign is
// negative-inside-solid (enforced in peclet::core::geom, not here). This header is C++17-clean so
// it can be pulled into CUDA translation units; C++20-only constructs live in host-only headers.
#ifndef PECLET_CORE_COMMON_TYPES_HPP
#define PECLET_CORE_COMMON_TYPES_HPP
 
#include <array>
#include <cstdint>
 
namespace peclet::core {
 
using Index = std::int64_t;
 
using Real = double;
 
template <int Dim>
using IVec = std::array<Index, Dim>;
 
template <int Dim>
using Vec = std::array<Real, Dim>;
 
template <typename Label = Index>
struct Range {
  std::size_t begin = 0;
  std::size_t end = 0;
  Label label{};
};
 
inline Index wrap(Index x, Index n) {
  return ((x % n) + n) % n;
}
 
template <int Dim, int Axis>
struct NestedLoop {
  template <typename Func>
  static void run(IVec<Dim>& idx, const IVec<Dim>& bgn, const IVec<Dim>& end, Func&& func) {
    for (idx[Axis] = bgn[Axis]; idx[Axis] < end[Axis]; ++idx[Axis]) {
      NestedLoop<Dim, Axis - 1>::run(idx, bgn, end, func);
    }
  }
};
 
template <int Dim>
struct NestedLoop<Dim, 0> {
  template <typename Func>
  static void run(IVec<Dim>& idx, const IVec<Dim>& bgn, const IVec<Dim>& end, Func&& func) {
    for (idx[0] = bgn[0]; idx[0] < end[0]; ++idx[0]) {
      func(const_cast<const IVec<Dim>&>(idx));
    }
  }
};
 
template <int Dim, typename Func>
inline void forEachInBox(const IVec<Dim>& bgn, const IVec<Dim>& end, Func&& func) {
  IVec<Dim> idx{};
  NestedLoop<Dim, Dim - 1>::run(idx, bgn, end, func);
}
 
}  // namespace peclet::core
 
#endif  // PECLET_CORE_COMMON_TYPES_HPP