EBGeometry_PointSoA.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Robert Marskar <robert.marskar@sintef.no>
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef EBGEOMETRY_POINTSOA_HPP
#define EBGEOMETRY_POINTSOA_HPP
 
// Std includes
#include <cstddef>
#include <cstdint>
#include <type_traits>
 
#if defined(__SSE4_1__) || defined(__AVX__) || defined(__AVX512F__)
#include <immintrin.h>
#endif
 
// Our includes
#include "EBGeometry_Array.hpp"
#include "EBGeometry_Vec.hpp"
 
namespace EBGeometry {
 
namespace PointSoA {
 
template <typename T>
[[nodiscard]] constexpr size_t
DefaultWidth() noexcept
{
  static_assert(std::is_floating_point_v<T>, "EBGeometry::PointSoA::DefaultWidth requires a floating-point T");
 
  return 4;
}
 
template <typename T>
[[nodiscard]] constexpr size_t
HostWidth() noexcept
{
  static_assert(std::is_floating_point_v<T>, "EBGeometry::PointSoA::HostWidth requires a floating-point T");
#if defined(__AVX512F__)
  if constexpr (std::is_same_v<T, double>) {
    return 8;
  }
  else {
    return 16;
  }
#elif defined(__AVX__)
  if constexpr (std::is_same_v<T, double>) {
    return 4;
  }
  else {
    return 8;
  }
#else
  return 4;
#endif
}
 
} // namespace PointSoA
 
template <class T, size_t W = PointSoA::DefaultWidth<T>()>
struct PointSoAT
{
  static_assert(W > 0, "W must be positive");
  static_assert(std::is_floating_point_v<T>, "PointSoAT requires a floating-point type T");
 
public:
  EBGEOMETRY_HOST
  void
  pack(const Vec3T<T>* a_positions, uint32_t a_count) noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  Array<T, W>
  getDistances2(const Vec3T<T>& a_point) const noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  Array<T, W>
  getDistances(const Vec3T<T>& a_point) const noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  T
  getMinimumDistance2(const Vec3T<T>& a_point) const noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  T
  getMinimumDistance(const Vec3T<T>& a_point) const noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  T
  getMaximumDistance2(const Vec3T<T>& a_point) const noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  T
  getMaximumDistance(const Vec3T<T>& a_point) const noexcept;
 
  template <class BV>
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  BV
  computeBoundingVolume() const noexcept;
 
  [[nodiscard]] EBGEOMETRY_HOST_DEVICE
  uint32_t
  numValid() const noexcept
  {
    return m_validCount;
  }
 
protected:
  alignas(W * sizeof(T)) T m_x[W];
 
  T m_y[W];
 
  T m_z[W];
 
  uint32_t m_validCount = 0;
};
 
static_assert(std::is_trivially_copyable_v<PointSoAT<float>>, "PointSoAT<float> must be trivially copyable");
static_assert(std::is_trivially_copyable_v<PointSoAT<double>>, "PointSoAT<double> must be trivially copyable");
 
} // namespace EBGeometry
 
#include "EBGeometry_PointSoAImplem.hpp"
 
#endif