aabb.h

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <span>
#include <type_traits>
 
#include "clustering/always_assert.h"
#ifdef CLUSTERING_USE_AVX2
#include "clustering/math/detail/aabb_gap_avx2.h"
#endif
 
namespace clustering::math {

template <class T>
[[nodiscard]] inline T pointAabbGapSq(const T *point, std::span<const T> boxMin,
                                      std::span<const T> boxMax) noexcept {
  static_assert(std::is_same_v<T, float>,
                "pointAabbGapSq: T must be float; a double specialization is out of scope.");
  CLUSTERING_ALWAYS_ASSERT(boxMin.size() == boxMax.size());
  const std::size_t d = boxMin.size();
#ifdef CLUSTERING_USE_AVX2
  return detail::pointAabbGapSqAvx2F32(point, boxMin.data(), boxMax.data(), d);
#else
  T sum = T{0};
  for (std::size_t j = 0; j < d; ++j) {
    T gap = T{0};
    if (point[j] < boxMin[j]) {
      gap = boxMin[j] - point[j];
    } else if (point[j] > boxMax[j]) {
      gap = point[j] - boxMax[j];
    }
    sum += gap * gap;
  }
  return sum;
#endif
}
 
} // namespace clustering::math