EBGeometry_TriangleSoA.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2024 Robert Marskar <robert.marskar@sintef.no>
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef EBGEOMETRY_TRIANGLESOA_HPP
#define EBGEOMETRY_TRIANGLESOA_HPP
 
// Std includes
#include <array>
#include <cstddef>
#include <cstdint>
#include <type_traits>
#include <vector>
 
#if defined(__SSE4_1__) || defined(__AVX__) || defined(__AVX512F__)
#include <immintrin.h>
#endif
 
// Our includes
#include "EBGeometry_Triangle.hpp"
#include "EBGeometry_Vec.hpp"
 
namespace EBGeometry {
 
namespace TriangleSoA {
 
template <typename T>
[[nodiscard]] constexpr size_t
DefaultWidth() noexcept
{
  static_assert(std::is_floating_point_v<T>, "EBGeometry::TriangleSoA::DefaultWidth 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 TriangleSoA
 
template <class T, size_t W>
struct TriangleSoAT
{
  static_assert(W > 0, "W must be positive");
  static_assert(std::is_floating_point_v<T>, "TriangleSoAT requires a floating-point type T");
 
public:
  template <class Meta>
  void
  pack(const Triangle<T, Meta>* tris, uint32_t count) noexcept;
 
  [[nodiscard]] T
  signedDistance(const Vec3T<T>& a_point) const noexcept;
 
  [[nodiscard]] std::array<T, W>
  signedDistances(const Vec3T<T>& a_point) const noexcept;
 
  template <class BV>
  [[nodiscard]] BV
  computeBoundingVolume() const noexcept;
 
protected:
  [[nodiscard]] T
  signedDistanceLane(uint32_t a_lane, const Vec3T<T>& a_point) const noexcept;
 
  alignas(W * sizeof(T)) T m_vx[3][W];
 
  T m_vy[3][W];
 
  T m_vz[3][W];
 
  alignas(W * sizeof(T)) T m_nx[W];
 
  T m_ny[W];
 
  T m_nz[W];
 
  alignas(W * sizeof(T)) T m_vnx[3][W];
 
  T m_vny[3][W];
 
  T m_vnz[3][W];
 
  alignas(W * sizeof(T)) T m_enx[3][W];
 
  T m_eny[3][W];
 
  T m_enz[3][W];
 
  uint32_t m_validCount = 0;
};
 
} // namespace EBGeometry
 
#include "EBGeometry_TriangleSoAImplem.hpp"
 
#endif