EBGeometry_BVHRefit.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_BVHREFIT_HPP
#define EBGEOMETRY_BVHREFIT_HPP
 
// Std includes
#include <cstddef>
#include <cstdint>
#include <type_traits>
#include <vector>
 
// Our includes
#include "EBGeometry_GPUTransform.hpp"
#include "EBGeometry_Macros.hpp"
#include "EBGeometry_PODVector.hpp"
#include "EBGeometry_PackedBVH.hpp"
#include "EBGeometry_Pool.hpp"
 
namespace EBGeometry {
 
namespace BVH {
 
class RefitSchedule
{
public:
  template <class T, class P, size_t K>
  EBGEOMETRY_HOST
  RefitSchedule(Pool& a_pool, const PackedBVH<T, P, K>& a_bvh);
 
  [[nodiscard]] EBGEOMETRY_HOST
  inline size_t
  numLevels() const noexcept
  {
    return m_levelBegin.empty() ? 0 : m_levelBegin.size() - 1;
  }
 
private:
  template <class T, class P, size_t K, class BVConstructor>
  friend void
  refit(Pool&                     a_pool,
        const PackedBVH<T, P, K>& a_bvh,
        const RefitSchedule&      a_schedule,
        const BVConstructor&      a_bvConstructor);
 
  PODVector<uint32_t> m_nodes;
 
  std::vector<uint32_t> m_levelBegin;
 
  uint64_t m_rootId = 0;
 
  uint64_t m_nodesOffset = 0;
};
 
template <class T, class P, size_t K, class BVConstructor>
EBGEOMETRY_HOST
inline void
refit(Pool&                     a_pool,
      const PackedBVH<T, P, K>& a_bvh,
      const RefitSchedule&      a_schedule,
      const BVConstructor&      a_bvConstructor);
 
namespace RefitDetail {
 
template <class Bvh, class BVConstructor>
struct Level
{
  mutable Bvh m_bvh;
 
  PODVector<uint32_t> m_nodes;
 
  uint32_t m_begin;
 
  BVConstructor m_bvConstructor;
 
  EBGEOMETRY_HOST_DEVICE
  inline void
  operator()(const std::size_t a_i) const
  {
    void* base = m_bvh.base();
 
    m_bvh.refitNode(base, m_nodes.at(base, m_begin + static_cast<uint32_t>(a_i)), m_bvConstructor);
  }
};
 
} // namespace RefitDetail
 
} // namespace BVH
 
} // namespace EBGeometry
 
#include "EBGeometry_BVHRefitImplem.hpp"
 
#endif