Compare commits

..

6 Commits

Author SHA1 Message Date
dd421e329e added files 2026-04-08 18:23:07 +03:00
8f65183882 fixed tests 2026-04-08 15:34:10 +03:00
327db8d441 updated contributing 2026-04-03 20:59:34 +03:00
d8188de736 keeping 1 AABB type 2026-03-28 14:22:36 +03:00
33cd3f64e4 Merge pull request #180 from orange-cpp/feature/aabb-linetrace
added aabb line trace
2026-03-25 03:37:35 +03:00
67a07eed45 added aabb line trace 2026-03-25 03:14:22 +03:00
10 changed files with 1512 additions and 39 deletions

View File

@@ -1,7 +0,0 @@
{
"permissions": {
"allow": [
"Bash(ls:*)"
]
}
}

10
.idea/editor.xml generated
View File

@@ -17,7 +17,7 @@
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppBoostFormatTooManyArgs/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppCStyleCast/@EntryIndexedValue" value="SUGGESTION" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppCVQualifierCanNotBeAppliedToReference/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppClassCanBeFinal/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppClassCanBeFinal/@EntryIndexedValue" value="DO_NOT_SHOW" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppClassIsIncomplete/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppClassNeedsConstructorBecauseOfUninitializedMember/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppClassNeverUsed/@EntryIndexedValue" value="WARNING" type="string" />
@@ -103,14 +103,14 @@
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppImplicitDefaultConstructorNotAvailable/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppIncompatiblePointerConversion/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppIncompleteSwitchStatement/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppInconsistentNaming/@EntryIndexedValue" value="HINT" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppInconsistentNaming/@EntryIndexedValue" value="DO_NOT_SHOW" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppIntegralToPointerConversion/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppInvalidLineContinuation/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppJoinDeclarationAndAssignment/@EntryIndexedValue" value="SUGGESTION" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppLambdaCaptureNeverUsed/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppLocalVariableMayBeConst/@EntryIndexedValue" value="HINT" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppLocalVariableMightNotBeInitialized/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppLocalVariableWithNonTrivialDtorIsNeverUsed/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppLocalVariableWithNonTrivialDtorIsNeverUsed/@EntryIndexedValue" value="DO_NOT_SHOW" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppLongFloat/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppMemberFunctionMayBeConst/@EntryIndexedValue" value="SUGGESTION" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppMemberFunctionMayBeStatic/@EntryIndexedValue" value="SUGGESTION" type="string" />
@@ -202,7 +202,7 @@
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppStaticDataMemberInUnnamedStruct/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppStaticSpecifierOnAnonymousNamespaceMember/@EntryIndexedValue" value="SUGGESTION" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppStringLiteralToCharPointerConversion/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppTabsAreDisallowed/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppTabsAreDisallowed/@EntryIndexedValue" value="DO_NOT_SHOW" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppTemplateArgumentsCanBeDeduced/@EntryIndexedValue" value="HINT" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppTemplateParameterNeverUsed/@EntryIndexedValue" value="HINT" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppTemplateParameterShadowing/@EntryIndexedValue" value="WARNING" type="string" />
@@ -216,7 +216,7 @@
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnmatchedPragmaEndRegionDirective/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnmatchedPragmaRegionDirective/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnnamedNamespaceInHeaderFile/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnnecessaryWhitespace/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnnecessaryWhitespace/@EntryIndexedValue" value="DO_NOT_SHOW" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnsignedZeroComparison/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUnusedIncludeDirective/@EntryIndexedValue" value="WARNING" type="string" />
<option name="/Default/CodeInspection/Highlighting/InspectionSeverities/=CppUseAlgorithmWithCount/@EntryIndexedValue" value="SUGGESTION" type="string" />

View File

@@ -1,32 +1,36 @@
## 🤝 Contributing to OMath or other Orange's Projects
# Contributing
### ❕ Prerequisites
## Prerequisites
- A working up-to-date OMath installation
- C++ knowledge
- Git knowledge
- Ability to ask for help (Feel free to create empty pull-request or PM a maintainer
in [Telegram](https://t.me/orange_cpp))
- C++ compiler with C++23 support (Clang 18+, GCC 14+, MSVC 19.38+)
- CMake 3.25+
- Git
- Familiarity with the codebase (see `INSTALL.md` for setup)
### ⏬ Setting up OMath
For questions, create a draft PR or reach out via [Telegram](https://t.me/orange_cpp).
Please read INSTALL.md file in repository
## Workflow
### 🔀 Pull requests and Branches
1. [Fork](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/fork-a-repo) the repository.
2. Create a feature branch from `main`.
3. Make your changes, ensuring tests pass.
4. Open a [pull request](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/proposing-changes-to-your-work-with-pull-requests/creating-a-pull-request-from-a-fork) against `main`.
In order to send code back to the official OMath repository, you must first create a copy of OMath on your github
account ([fork](https://help.github.com/articles/creating-a-pull-request-from-a-fork/)) and
then [create a pull request](https://help.github.com/articles/creating-a-pull-request-from-a-fork/) back to OMath.
## Code Style
OMath development is performed on multiple branches. Changes are then pull requested into master. By default, changes
merged into master will not roll out to stable build users unless the `stable` tag is updated.
Follow the project `.clang-format`. Run `clang-format` before committing.
### 📜 Code-Style
## Building
The orange code-style can be found in `.clang-format`.
Use one of the CMake presets defined in `CMakePresets.json`:
### 📦 Building
```bash
cmake --preset <preset-name> -DOMATH_BUILD_TESTS=ON
cmake --build --preset <preset-name>
```
OMath has already created the `cmake-build` and `out` directories where cmake/bin files are located. By default, you
can build OMath by running `cmake --build cmake-build/build/windows-release --target omath -j 6` in the source
directory.
Run `cmake --list-presets` to see available configurations.
## Tests
All new functionality must include unit tests. Run the test binary after building to verify nothing is broken.

View File

@@ -71,18 +71,18 @@ void drawChar(char c, float x, float y, float scale, const Color& color, std::ve
lines.push_back(x + x1 * w);
lines.push_back(y + y1 * h);
lines.push_back(0.0f);
lines.push_back(color.x);
lines.push_back(color.y);
lines.push_back(color.z);
lines.push_back(color.value().x);
lines.push_back(color.value().y);
lines.push_back(color.value().z);
lines.push_back(1.0f); // size
lines.push_back(1.0f); // isLine
lines.push_back(x + x2 * w);
lines.push_back(y + y2 * h);
lines.push_back(0.0f);
lines.push_back(color.x);
lines.push_back(color.y);
lines.push_back(color.z);
lines.push_back(color.value().x);
lines.push_back(color.value().y);
lines.push_back(color.value().z);
lines.push_back(1.0f); // size
lines.push_back(1.0f); // isLine
};

View File

@@ -12,5 +12,17 @@ namespace omath::primitives
{
Vector3<Type> min;
Vector3<Type> max;
[[nodiscard]]
constexpr Vector3<Type> center() const noexcept
{
return (min + max) / static_cast<Type>(2);
}
[[nodiscard]]
constexpr Vector3<Type> extents() const noexcept
{
return (max - min) / static_cast<Type>(2);
}
};
} // namespace omath::primitives

View File

@@ -0,0 +1,412 @@
//
// Created by Orange on 04/08/2026.
//
#pragma once
#include "omath/3d_primitives/aabb.hpp"
#include "omath/collision/line_tracer.hpp"
#include <algorithm>
#include <cstdint>
#include <numeric>
#include <span>
#include <vector>
namespace omath::collision
{
template<class Type = float>
class BvhTree final
{
public:
using AabbType = primitives::Aabb<Type>;
struct HitResult
{
std::size_t object_index;
Type distance_sqr;
};
BvhTree() = default;
explicit BvhTree(std::span<const AabbType> aabbs)
: m_aabbs(aabbs.begin(), aabbs.end())
{
if (aabbs.empty())
return;
m_indices.resize(aabbs.size());
std::iota(m_indices.begin(), m_indices.end(), std::size_t{0});
m_nodes.reserve(aabbs.size() * 2);
build(m_aabbs, 0, aabbs.size());
}
[[nodiscard]]
std::vector<std::size_t> query_overlaps(const AabbType& query_aabb) const
{
std::vector<std::size_t> results;
if (m_nodes.empty())
return results;
query_overlaps_impl(0, query_aabb, results);
return results;
}
template<class RayType = Ray<>>
[[nodiscard]]
std::vector<HitResult> query_ray(const RayType& ray) const
{
std::vector<HitResult> results;
if (m_nodes.empty())
return results;
query_ray_impl(0, ray, results);
std::ranges::sort(results, [](const HitResult& a, const HitResult& b)
{ return a.distance_sqr < b.distance_sqr; });
return results;
}
[[nodiscard]]
std::size_t node_count() const noexcept
{
return m_nodes.size();
}
[[nodiscard]]
bool empty() const noexcept
{
return m_nodes.empty();
}
private:
static constexpr std::size_t k_sah_bucket_count = 12;
static constexpr std::size_t k_leaf_threshold = 1;
static constexpr std::size_t k_null_index = std::numeric_limits<std::size_t>::max();
struct Node
{
AabbType bounds;
std::size_t left = k_null_index;
std::size_t right = k_null_index;
// For leaf nodes: index range into m_indices
std::size_t first_index = 0;
std::size_t index_count = 0;
[[nodiscard]]
bool is_leaf() const noexcept
{
return left == k_null_index;
}
};
struct SahBucket
{
AabbType bounds = {
{std::numeric_limits<Type>::max(), std::numeric_limits<Type>::max(),
std::numeric_limits<Type>::max()},
{std::numeric_limits<Type>::lowest(), std::numeric_limits<Type>::lowest(),
std::numeric_limits<Type>::lowest()}
};
std::size_t count = 0;
};
[[nodiscard]]
static constexpr Type surface_area(const AabbType& aabb) noexcept
{
const auto d = aabb.max - aabb.min;
return static_cast<Type>(2) * (d.x * d.y + d.y * d.z + d.z * d.x);
}
[[nodiscard]]
static constexpr AabbType merge(const AabbType& a, const AabbType& b) noexcept
{
return {
{std::min(a.min.x, b.min.x), std::min(a.min.y, b.min.y), std::min(a.min.z, b.min.z)},
{std::max(a.max.x, b.max.x), std::max(a.max.y, b.max.y), std::max(a.max.z, b.max.z)}
};
}
[[nodiscard]]
static constexpr bool overlaps(const AabbType& a, const AabbType& b) noexcept
{
return a.min.x <= b.max.x && a.max.x >= b.min.x
&& a.min.y <= b.max.y && a.max.y >= b.min.y
&& a.min.z <= b.max.z && a.max.z >= b.min.z;
}
std::size_t build(std::span<const AabbType> aabbs, std::size_t begin, std::size_t end)
{
const auto node_idx = m_nodes.size();
m_nodes.emplace_back();
auto& node = m_nodes[node_idx];
node.bounds = compute_bounds(aabbs, begin, end);
const auto count = end - begin;
if (count <= k_leaf_threshold)
{
node.first_index = begin;
node.index_count = count;
return node_idx;
}
// Find best SAH split
const auto centroid_bounds = compute_centroid_bounds(aabbs, begin, end);
const auto split = find_best_split(aabbs, begin, end, node.bounds, centroid_bounds);
// If SAH says don't split, make a leaf
if (!split.has_value())
{
node.first_index = begin;
node.index_count = count;
return node_idx;
}
const auto [axis, split_pos] = split.value();
// Partition indices around the split
const auto mid = partition_indices(aabbs, begin, end, axis, split_pos);
// Degenerate partition fallback: split in the middle
const auto actual_mid = (mid == begin || mid == end) ? begin + count / 2 : mid;
// Build children — careful: m_nodes may reallocate, so don't hold references across build calls
const auto left_idx = build(aabbs, begin, actual_mid);
const auto right_idx = build(aabbs, actual_mid, end);
m_nodes[node_idx].left = left_idx;
m_nodes[node_idx].right = right_idx;
m_nodes[node_idx].index_count = 0;
return node_idx;
}
[[nodiscard]]
AabbType compute_bounds(std::span<const AabbType> aabbs, std::size_t begin, std::size_t end) const
{
AabbType bounds = {
{std::numeric_limits<Type>::max(), std::numeric_limits<Type>::max(),
std::numeric_limits<Type>::max()},
{std::numeric_limits<Type>::lowest(), std::numeric_limits<Type>::lowest(),
std::numeric_limits<Type>::lowest()}
};
for (auto i = begin; i < end; ++i)
bounds = merge(bounds, aabbs[m_indices[i]]);
return bounds;
}
[[nodiscard]]
AabbType compute_centroid_bounds(std::span<const AabbType> aabbs, std::size_t begin, std::size_t end) const
{
AabbType bounds = {
{std::numeric_limits<Type>::max(), std::numeric_limits<Type>::max(),
std::numeric_limits<Type>::max()},
{std::numeric_limits<Type>::lowest(), std::numeric_limits<Type>::lowest(),
std::numeric_limits<Type>::lowest()}
};
for (auto i = begin; i < end; ++i)
{
const auto c = aabbs[m_indices[i]].center();
bounds.min.x = std::min(bounds.min.x, c.x);
bounds.min.y = std::min(bounds.min.y, c.y);
bounds.min.z = std::min(bounds.min.z, c.z);
bounds.max.x = std::max(bounds.max.x, c.x);
bounds.max.y = std::max(bounds.max.y, c.y);
bounds.max.z = std::max(bounds.max.z, c.z);
}
return bounds;
}
struct SplitResult
{
int axis;
Type position;
};
[[nodiscard]]
std::optional<SplitResult> find_best_split(std::span<const AabbType> aabbs, std::size_t begin,
std::size_t end, const AabbType& node_bounds,
const AabbType& centroid_bounds) const
{
const auto count = end - begin;
const auto leaf_cost = static_cast<Type>(count);
auto best_cost = leaf_cost;
std::optional<SplitResult> best_split;
for (int axis = 0; axis < 3; ++axis)
{
const auto axis_min = get_component(centroid_bounds.min, axis);
const auto axis_max = get_component(centroid_bounds.max, axis);
if (axis_max - axis_min < std::numeric_limits<Type>::epsilon())
continue;
SahBucket buckets[k_sah_bucket_count] = {};
const auto inv_extent = static_cast<Type>(k_sah_bucket_count) / (axis_max - axis_min);
// Fill buckets
for (auto i = begin; i < end; ++i)
{
const auto centroid = get_component(aabbs[m_indices[i]].center(), axis);
auto bucket_idx = static_cast<std::size_t>((centroid - axis_min) * inv_extent);
bucket_idx = std::min(bucket_idx, k_sah_bucket_count - 1);
buckets[bucket_idx].count++;
if (buckets[bucket_idx].count == 1)
buckets[bucket_idx].bounds = aabbs[m_indices[i]];
else
buckets[bucket_idx].bounds = merge(buckets[bucket_idx].bounds, aabbs[m_indices[i]]);
}
// Evaluate split costs using prefix/suffix sweeps
AabbType prefix_bounds[k_sah_bucket_count - 1];
std::size_t prefix_count[k_sah_bucket_count - 1];
prefix_bounds[0] = buckets[0].bounds;
prefix_count[0] = buckets[0].count;
for (std::size_t i = 1; i < k_sah_bucket_count - 1; ++i)
{
prefix_bounds[i] = (buckets[i].count > 0)
? merge(prefix_bounds[i - 1], buckets[i].bounds)
: prefix_bounds[i - 1];
prefix_count[i] = prefix_count[i - 1] + buckets[i].count;
}
AabbType suffix_bounds = buckets[k_sah_bucket_count - 1].bounds;
std::size_t suffix_count = buckets[k_sah_bucket_count - 1].count;
const auto parent_area = surface_area(node_bounds);
const auto inv_parent_area = static_cast<Type>(1) / parent_area;
for (auto i = static_cast<int>(k_sah_bucket_count) - 2; i >= 0; --i)
{
const auto left_count = prefix_count[i];
const auto right_count = suffix_count;
if (left_count == 0 || right_count == 0)
{
if (i > 0)
{
suffix_bounds = (buckets[i].count > 0)
? merge(suffix_bounds, buckets[i].bounds)
: suffix_bounds;
suffix_count += buckets[i].count;
}
continue;
}
const auto cost = static_cast<Type>(1)
+ (static_cast<Type>(left_count) * surface_area(prefix_bounds[i])
+ static_cast<Type>(right_count) * surface_area(suffix_bounds))
* inv_parent_area;
if (cost < best_cost)
{
best_cost = cost;
best_split = SplitResult{
axis,
axis_min + static_cast<Type>(i + 1) * (axis_max - axis_min)
/ static_cast<Type>(k_sah_bucket_count)
};
}
suffix_bounds = (buckets[i].count > 0)
? merge(suffix_bounds, buckets[i].bounds)
: suffix_bounds;
suffix_count += buckets[i].count;
}
}
return best_split;
}
std::size_t partition_indices(std::span<const AabbType> aabbs, std::size_t begin, std::size_t end,
int axis, Type split_pos)
{
auto it = std::partition(m_indices.begin() + static_cast<std::ptrdiff_t>(begin),
m_indices.begin() + static_cast<std::ptrdiff_t>(end),
[&](std::size_t idx)
{ return get_component(aabbs[idx].center(), axis) < split_pos; });
return static_cast<std::size_t>(std::distance(m_indices.begin(), it));
}
[[nodiscard]]
static constexpr Type get_component(const Vector3<Type>& v, int axis) noexcept
{
switch (axis)
{
case 0:
return v.x;
case 1:
return v.y;
default:
return v.z;
}
}
void query_overlaps_impl(std::size_t node_idx, const AabbType& query_aabb,
std::vector<std::size_t>& results) const
{
const auto& node = m_nodes[node_idx];
if (!overlaps(node.bounds, query_aabb))
return;
if (node.is_leaf())
{
for (auto i = node.first_index; i < node.first_index + node.index_count; ++i)
if (overlaps(query_aabb, m_aabbs[m_indices[i]]))
results.push_back(m_indices[i]);
return;
}
query_overlaps_impl(node.left, query_aabb, results);
query_overlaps_impl(node.right, query_aabb, results);
}
template<class RayType>
void query_ray_impl(std::size_t node_idx, const RayType& ray,
std::vector<HitResult>& results) const
{
const auto& node = m_nodes[node_idx];
// Quick AABB-ray rejection using the slab method
const auto hit = LineTracer<RayType>::get_ray_hit_point(ray, node.bounds);
if (hit == ray.end)
return;
if (node.is_leaf())
{
for (auto i = node.first_index; i < node.first_index + node.index_count; ++i)
{
const auto leaf_hit = LineTracer<RayType>::get_ray_hit_point(
ray, m_aabbs[m_indices[i]]);
if (leaf_hit != ray.end)
{
const auto diff = leaf_hit - ray.start;
results.push_back({m_indices[i], diff.dot(diff)});
}
}
return;
}
query_ray_impl(node.left, ray, results);
query_ray_impl(node.right, ray, results);
}
std::vector<Node> m_nodes;
std::vector<std::size_t> m_indices;
std::vector<AabbType> m_aabbs;
};
} // namespace omath::collision

View File

@@ -3,6 +3,7 @@
//
#pragma once
#include "omath/3d_primitives/aabb.hpp"
#include "omath/linear_algebra/triangle.hpp"
#include "omath/linear_algebra/vector3.hpp"
@@ -34,6 +35,7 @@ namespace omath::collision
class LineTracer final
{
using TriangleType = Triangle<typename RayType::VectorType>;
using AABBType = primitives::Aabb<typename RayType::VectorType::ContainedType>;
public:
LineTracer() = delete;
@@ -87,6 +89,54 @@ namespace omath::collision
return ray.start + ray_dir * t_hit;
}
// Slab method ray-AABB intersection
// Returns the hit point on the AABB surface, or ray.end if no intersection
[[nodiscard]]
constexpr static auto get_ray_hit_point(const RayType& ray, const AABBType& aabb) noexcept
{
using T = typename RayType::VectorType::ContainedType;
const auto dir = ray.direction_vector();
auto t_min = -std::numeric_limits<T>::infinity();
auto t_max = std::numeric_limits<T>::infinity();
const auto process_axis = [&](const T& d, const T& origin, const T& box_min,
const T& box_max) -> bool
{
constexpr T k_epsilon = std::numeric_limits<T>::epsilon();
if (std::abs(d) < k_epsilon)
return origin >= box_min && origin <= box_max;
const T inv = T(1) / d;
T t0 = (box_min - origin) * inv;
T t1 = (box_max - origin) * inv;
if (t0 > t1)
std::swap(t0, t1);
t_min = std::max(t_min, t0);
t_max = std::min(t_max, t1);
return t_min <= t_max;
};
if (!process_axis(dir.x, ray.start.x, aabb.min.x, aabb.max.x))
return ray.end;
if (!process_axis(dir.y, ray.start.y, aabb.min.y, aabb.max.y))
return ray.end;
if (!process_axis(dir.z, ray.start.z, aabb.min.z, aabb.max.z))
return ray.end;
// t_hit: use entry point if in front of origin, otherwise 0 (started inside)
const T t_hit = std::max(T(0), t_min);
if (t_max < T(0))
return ray.end; // box entirely behind origin
if (!ray.infinite_length && t_hit > T(1))
return ray.end; // box beyond ray endpoint
return ray.start + dir * t_hit;
}
template<class MeshType>
[[nodiscard]]
constexpr static auto get_ray_hit_point(const RayType& ray, const MeshType& mesh) noexcept

View File

@@ -35,6 +35,7 @@
#include "omath/collision/line_tracer.hpp"
#include "omath/collision/gjk_algorithm.hpp"
#include "omath/collision/epa_algorithm.hpp"
#include "omath/collision/bvh_tree.hpp"
// Pathfinding algorithms
#include "omath/pathfinding/a_star.hpp"
#include "omath/pathfinding/navigation_mesh.hpp"

View File

@@ -0,0 +1,876 @@
//
// Created by Orange on 04/08/2026.
//
#include <gtest/gtest.h>
#include <omath/collision/bvh_tree.hpp>
#include <algorithm>
#include <random>
#include <set>
using Aabb = omath::primitives::Aabb<float>;
using BvhTree = omath::collision::BvhTree<float>;
using Ray = omath::collision::Ray<>;
using HitResult = BvhTree::HitResult;
using AabbD = omath::primitives::Aabb<double>;
using BvhTreeD = omath::collision::BvhTree<double>;
// ============================================================================
// Helper: brute-force overlap query for verification
// ============================================================================
static std::set<std::size_t> brute_force_overlaps(const std::vector<Aabb>& aabbs, const Aabb& query)
{
std::set<std::size_t> result;
for (std::size_t i = 0; i < aabbs.size(); ++i)
{
if (query.min.x <= aabbs[i].max.x && query.max.x >= aabbs[i].min.x
&& query.min.y <= aabbs[i].max.y && query.max.y >= aabbs[i].min.y
&& query.min.z <= aabbs[i].max.z && query.max.z >= aabbs[i].min.z)
result.insert(i);
}
return result;
}
// ============================================================================
// Construction tests
// ============================================================================
TEST(UnitTestBvhTree, EmptyTree)
{
const BvhTree tree;
EXPECT_TRUE(tree.empty());
EXPECT_EQ(tree.node_count(), 0);
EXPECT_TRUE(tree.query_overlaps({}).empty());
}
TEST(UnitTestBvhTree, EmptySpan)
{
const std::vector<Aabb> empty;
const BvhTree tree(empty);
EXPECT_TRUE(tree.empty());
EXPECT_EQ(tree.node_count(), 0);
}
TEST(UnitTestBvhTree, SingleElement)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}}
};
const BvhTree tree(aabbs);
EXPECT_FALSE(tree.empty());
EXPECT_EQ(tree.node_count(), 1);
const auto results = tree.query_overlaps({{0.5f, 0.5f, 0.5f}, {1.5f, 1.5f, 1.5f}});
ASSERT_EQ(results.size(), 1);
EXPECT_EQ(results[0], 0);
const auto miss = tree.query_overlaps({{5.f, 5.f, 5.f}, {6.f, 6.f, 6.f}});
EXPECT_TRUE(miss.empty());
}
TEST(UnitTestBvhTree, TwoElements)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{5.f, 5.f, 5.f}, {6.f, 6.f, 6.f}},
};
const BvhTree tree(aabbs);
EXPECT_FALSE(tree.empty());
// Hit first only
auto r = tree.query_overlaps({{-0.5f, -0.5f, -0.5f}, {0.5f, 0.5f, 0.5f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
// Hit second only
r = tree.query_overlaps({{5.5f, 5.5f, 5.5f}, {7.f, 7.f, 7.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 1);
// Hit both
r = tree.query_overlaps({{-1.f, -1.f, -1.f}, {10.f, 10.f, 10.f}});
EXPECT_EQ(r.size(), 2);
}
TEST(UnitTestBvhTree, ThreeElements)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{2.f, 2.f, 2.f}, {3.f, 3.f, 3.f}},
{{10.f, 10.f, 10.f}, {11.f, 11.f, 11.f}},
};
const BvhTree tree(aabbs);
const auto results = tree.query_overlaps({{0.5f, 0.5f, 0.5f}, {2.5f, 2.5f, 2.5f}});
EXPECT_EQ(results.size(), 2);
const auto far = tree.query_overlaps({{9.5f, 9.5f, 9.5f}, {10.5f, 10.5f, 10.5f}});
ASSERT_EQ(far.size(), 1);
EXPECT_EQ(far[0], 2);
}
TEST(UnitTestBvhTree, NodeCountGrowsSublinearly)
{
// For N objects, node count should be at most 2N-1
std::vector<Aabb> aabbs;
for (int i = 0; i < 100; ++i)
{
const auto f = static_cast<float>(i) * 3.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
EXPECT_LE(tree.node_count(), 2 * aabbs.size());
}
// ============================================================================
// Overlap query tests
// ============================================================================
TEST(UnitTestBvhTree, OverlapExactTouch)
{
// Two boxes share exactly one face
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{1.f, 0.f, 0.f}, {2.f, 1.f, 1.f}},
};
const BvhTree tree(aabbs);
// Query exactly at the shared face — should overlap both
const auto r = tree.query_overlaps({{0.5f, 0.f, 0.f}, {1.5f, 1.f, 1.f}});
EXPECT_EQ(r.size(), 2);
}
TEST(UnitTestBvhTree, OverlapEdgeTouch)
{
// Query AABB edge-touches an object AABB
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
};
const BvhTree tree(aabbs);
// Touching at corner point (1,1,1)
const auto r = tree.query_overlaps({{1.f, 1.f, 1.f}, {2.f, 2.f, 2.f}});
EXPECT_EQ(r.size(), 1);
}
TEST(UnitTestBvhTree, OverlapQueryInsideObject)
{
// Query is fully inside an object
const std::vector<Aabb> aabbs = {
{{-10.f, -10.f, -10.f}, {10.f, 10.f, 10.f}},
};
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
}
TEST(UnitTestBvhTree, OverlapObjectInsideQuery)
{
// Object is fully inside the query
const std::vector<Aabb> aabbs = {
{{4.f, 4.f, 4.f}, {5.f, 5.f, 5.f}},
};
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{0.f, 0.f, 0.f}, {10.f, 10.f, 10.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
}
TEST(UnitTestBvhTree, OverlapMissOnSingleAxis)
{
// Overlap on X and Y but not Z
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
};
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{0.f, 0.f, 5.f}, {1.f, 1.f, 6.f}});
EXPECT_TRUE(r.empty());
}
TEST(UnitTestBvhTree, OverlapNegativeCoordinates)
{
const std::vector<Aabb> aabbs = {
{{-5.f, -5.f, -5.f}, {-3.f, -3.f, -3.f}},
{{-2.f, -2.f, -2.f}, {0.f, 0.f, 0.f}},
{{1.f, 1.f, 1.f}, {3.f, 3.f, 3.f}},
};
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{-6.f, -6.f, -6.f}, {-4.f, -4.f, -4.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
}
TEST(UnitTestBvhTree, OverlapMixedNegativePositive)
{
const std::vector<Aabb> aabbs = {
{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}},
};
const BvhTree tree(aabbs);
// Query spans negative and positive
const auto r = tree.query_overlaps({{-0.5f, -0.5f, -0.5f}, {0.5f, 0.5f, 0.5f}});
ASSERT_EQ(r.size(), 1);
}
TEST(UnitTestBvhTree, OverlapNoHitsAmongMany)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 50; ++i)
{
const auto f = static_cast<float>(i) * 5.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
// Query far from all objects
const auto r = tree.query_overlaps({{-100.f, -100.f, -100.f}, {-90.f, -90.f, -90.f}});
EXPECT_TRUE(r.empty());
}
TEST(UnitTestBvhTree, OverlapAllObjects)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 64; ++i)
{
const auto f = static_cast<float>(i);
aabbs.push_back({{f, f, f}, {f + 0.5f, f + 0.5f, f + 0.5f}});
}
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{-1.f, -1.f, -1.f}, {100.f, 100.f, 100.f}});
EXPECT_EQ(r.size(), 64);
}
TEST(UnitTestBvhTree, OverlapReturnsCorrectIndices)
{
// Specific index verification
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}}, // 0
{{10.f, 0.f, 0.f}, {11.f, 1.f, 1.f}}, // 1
{{20.f, 0.f, 0.f}, {21.f, 1.f, 1.f}}, // 2
{{30.f, 0.f, 0.f}, {31.f, 1.f, 1.f}}, // 3
{{40.f, 0.f, 0.f}, {41.f, 1.f, 1.f}}, // 4
};
const BvhTree tree(aabbs);
// Hit only index 2
auto r = tree.query_overlaps({{19.5f, -1.f, -1.f}, {20.5f, 2.f, 2.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 2);
// Hit only index 4
r = tree.query_overlaps({{39.5f, -1.f, -1.f}, {40.5f, 2.f, 2.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 4);
}
// ============================================================================
// Spatial distribution tests
// ============================================================================
TEST(UnitTestBvhTree, ObjectsAlongXAxis)
{
// All objects on a line along X
std::vector<Aabb> aabbs;
for (int i = 0; i < 20; ++i)
{
const auto f = static_cast<float>(i) * 4.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}});
EXPECT_EQ(r.size(), 1);
const auto mid = tree.query_overlaps({{7.5f, -1.f, -1.f}, {8.5f, 2.f, 2.f}});
EXPECT_EQ(mid.size(), 1);
}
TEST(UnitTestBvhTree, ObjectsAlongYAxis)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 20; ++i)
{
const auto f = static_cast<float>(i) * 4.f;
aabbs.push_back({{0.f, f, 0.f}, {1.f, f + 1.f, 1.f}});
}
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{-1.f, 38.f, -1.f}, {2.f, 40.f, 2.f}});
EXPECT_EQ(r.size(), 1);
}
TEST(UnitTestBvhTree, ObjectsAlongZAxis)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 20; ++i)
{
const auto f = static_cast<float>(i) * 4.f;
aabbs.push_back({{0.f, 0.f, f}, {1.f, 1.f, f + 1.f}});
}
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{-1.f, -1.f, 38.f}, {2.f, 2.f, 40.f}});
EXPECT_EQ(r.size(), 1);
}
TEST(UnitTestBvhTree, ObjectsInPlaneXY)
{
// Grid in the XY plane
std::vector<Aabb> aabbs;
for (int x = 0; x < 10; ++x)
for (int y = 0; y < 10; ++y)
{
const auto fx = static_cast<float>(x) * 3.f;
const auto fy = static_cast<float>(y) * 3.f;
aabbs.push_back({{fx, fy, 0.f}, {fx + 1.f, fy + 1.f, 1.f}});
}
const BvhTree tree(aabbs);
EXPECT_EQ(tree.query_overlaps({{-1.f, -1.f, -1.f}, {100.f, 100.f, 2.f}}).size(), 100);
// Single cell query
const auto r = tree.query_overlaps({{0.f, 0.f, 0.f}, {0.5f, 0.5f, 0.5f}});
EXPECT_EQ(r.size(), 1);
}
TEST(UnitTestBvhTree, ClusteredObjects)
{
// Two clusters far apart
std::vector<Aabb> aabbs;
for (int i = 0; i < 25; ++i)
{
const auto f = static_cast<float>(i) * 0.5f;
aabbs.push_back({{f, f, f}, {f + 0.4f, f + 0.4f, f + 0.4f}});
}
for (int i = 0; i < 25; ++i)
{
const auto f = 100.f + static_cast<float>(i) * 0.5f;
aabbs.push_back({{f, f, f}, {f + 0.4f, f + 0.4f, f + 0.4f}});
}
const BvhTree tree(aabbs);
// Query near first cluster
const auto r1 = tree.query_overlaps({{-1.f, -1.f, -1.f}, {15.f, 15.f, 15.f}});
EXPECT_EQ(r1.size(), 25);
// Query near second cluster
const auto r2 = tree.query_overlaps({{99.f, 99.f, 99.f}, {115.f, 115.f, 115.f}});
EXPECT_EQ(r2.size(), 25);
// Query between clusters — should find nothing
const auto gap = tree.query_overlaps({{50.f, 50.f, 50.f}, {60.f, 60.f, 60.f}});
EXPECT_TRUE(gap.empty());
}
TEST(UnitTestBvhTree, OverlappingObjects)
{
// Objects that overlap each other
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {2.f, 2.f, 2.f}},
{{1.f, 1.f, 1.f}, {3.f, 3.f, 3.f}},
{{1.5f, 1.5f, 1.5f}, {4.f, 4.f, 4.f}},
};
const BvhTree tree(aabbs);
// Query at the overlap region of all three
const auto r = tree.query_overlaps({{1.5f, 1.5f, 1.5f}, {2.f, 2.f, 2.f}});
EXPECT_EQ(r.size(), 3);
}
TEST(UnitTestBvhTree, IdenticalObjects)
{
// All objects at the same position
std::vector<Aabb> aabbs;
for (int i = 0; i < 10; ++i)
aabbs.push_back({{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}});
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}});
EXPECT_EQ(r.size(), 10);
}
TEST(UnitTestBvhTree, DegenerateThickPlanes)
{
// Very flat AABBs (thickness ~0 in one axis)
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {10.f, 10.f, 0.001f}},
{{0.f, 0.f, 5.f}, {10.f, 10.f, 5.001f}},
};
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{0.f, 0.f, -0.01f}, {10.f, 10.f, 0.01f}});
ASSERT_EQ(r.size(), 1);
}
TEST(UnitTestBvhTree, VaryingSizes)
{
// Objects of wildly different sizes
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {0.01f, 0.01f, 0.01f}}, // tiny
{{-500.f, -500.f, -500.f}, {500.f, 500.f, 500.f}}, // huge
{{10.f, 10.f, 10.f}, {11.f, 11.f, 11.f}}, // normal
};
const BvhTree tree(aabbs);
// The huge box should overlap almost any query
auto r = tree.query_overlaps({{200.f, 200.f, 200.f}, {201.f, 201.f, 201.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 1);
// Query at origin hits the tiny and the huge
r = tree.query_overlaps({{-0.1f, -0.1f, -0.1f}, {0.1f, 0.1f, 0.1f}});
EXPECT_EQ(r.size(), 2);
}
// ============================================================================
// Ray query tests
// ============================================================================
TEST(UnitTestBvhTree, RayQueryBasic)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{5.f, 0.f, 0.f}, {6.f, 1.f, 1.f}},
{{0.f, 5.f, 0.f}, {1.f, 6.f, 1.f}},
};
const BvhTree tree(aabbs);
Ray ray;
ray.start = {-1.f, 0.5f, 0.5f};
ray.end = {10.f, 0.5f, 0.5f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_GE(hits.size(), 2);
if (hits.size() >= 2)
EXPECT_LE(hits[0].distance_sqr, hits[1].distance_sqr);
}
TEST(UnitTestBvhTree, RayQueryMissesAll)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{5.f, 0.f, 0.f}, {6.f, 1.f, 1.f}},
};
const BvhTree tree(aabbs);
// Ray above everything
Ray ray;
ray.start = {-1.f, 100.f, 0.5f};
ray.end = {10.f, 100.f, 0.5f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_TRUE(hits.empty());
}
TEST(UnitTestBvhTree, RayQueryAlongY)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{0.f, 5.f, 0.f}, {1.f, 6.f, 1.f}},
{{0.f, 10.f, 0.f}, {1.f, 11.f, 1.f}},
};
const BvhTree tree(aabbs);
Ray ray;
ray.start = {0.5f, -1.f, 0.5f};
ray.end = {0.5f, 20.f, 0.5f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_EQ(hits.size(), 3);
// Verify sorted by distance
for (std::size_t i = 1; i < hits.size(); ++i)
EXPECT_LE(hits[i - 1].distance_sqr, hits[i].distance_sqr);
}
TEST(UnitTestBvhTree, RayQueryAlongZ)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{0.f, 0.f, 10.f}, {1.f, 1.f, 11.f}},
};
const BvhTree tree(aabbs);
Ray ray;
ray.start = {0.5f, 0.5f, -5.f};
ray.end = {0.5f, 0.5f, 20.f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_EQ(hits.size(), 2);
}
TEST(UnitTestBvhTree, RayQueryDiagonal)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{5.f, 5.f, 5.f}, {6.f, 6.f, 6.f}},
{{10.f, 10.f, 10.f}, {11.f, 11.f, 11.f}},
};
const BvhTree tree(aabbs);
// Diagonal ray through all three
Ray ray;
ray.start = {-1.f, -1.f, -1.f};
ray.end = {15.f, 15.f, 15.f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_EQ(hits.size(), 3);
}
TEST(UnitTestBvhTree, RayQueryOnEmptyTree)
{
const BvhTree tree;
Ray ray;
ray.start = {0.f, 0.f, 0.f};
ray.end = {10.f, 0.f, 0.f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_TRUE(hits.empty());
}
TEST(UnitTestBvhTree, RayQuerySortedByDistance)
{
// Many boxes along a line
std::vector<Aabb> aabbs;
for (int i = 0; i < 20; ++i)
{
const auto f = static_cast<float>(i) * 3.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
Ray ray;
ray.start = {-1.f, 0.5f, 0.5f};
ray.end = {100.f, 0.5f, 0.5f};
ray.infinite_length = true;
const auto hits = tree.query_ray(ray);
EXPECT_EQ(hits.size(), 20);
for (std::size_t i = 1; i < hits.size(); ++i)
EXPECT_LE(hits[i - 1].distance_sqr, hits[i].distance_sqr);
}
// ============================================================================
// Brute-force verification tests
// ============================================================================
TEST(UnitTestBvhTree, BruteForceVerificationGrid)
{
std::vector<Aabb> aabbs;
for (int x = 0; x < 10; ++x)
for (int y = 0; y < 10; ++y)
for (int z = 0; z < 10; ++z)
{
const auto fx = static_cast<float>(x) * 3.f;
const auto fy = static_cast<float>(y) * 3.f;
const auto fz = static_cast<float>(z) * 3.f;
aabbs.push_back({{fx, fy, fz}, {fx + 1.f, fy + 1.f, fz + 1.f}});
}
const BvhTree tree(aabbs);
// Test several queries and compare to brute force
const std::vector<Aabb> queries = {
{{0.f, 0.f, 0.f}, {1.5f, 1.5f, 1.5f}},
{{-1.f, -1.f, -1.f}, {100.f, 100.f, 100.f}},
{{13.f, 13.f, 13.f}, {14.f, 14.f, 14.f}},
{{-50.f, -50.f, -50.f}, {-40.f, -40.f, -40.f}},
{{5.5f, 5.5f, 5.5f}, {7.5f, 7.5f, 7.5f}},
};
for (const auto& q : queries)
{
const auto bvh_results = tree.query_overlaps(q);
const auto brute_results = brute_force_overlaps(aabbs, q);
const std::set<std::size_t> bvh_set(bvh_results.begin(), bvh_results.end());
EXPECT_EQ(bvh_set, brute_results)
<< "Mismatch for query [(" << q.min.x << "," << q.min.y << "," << q.min.z
<< ") -> (" << q.max.x << "," << q.max.y << "," << q.max.z << ")]";
}
}
TEST(UnitTestBvhTree, BruteForceVerificationRandom)
{
std::mt19937 rng(42);
std::uniform_real_distribution<float> pos_dist(-50.f, 50.f);
std::uniform_real_distribution<float> size_dist(0.5f, 3.f);
std::vector<Aabb> aabbs;
for (int i = 0; i < 200; ++i)
{
const auto x = pos_dist(rng);
const auto y = pos_dist(rng);
const auto z = pos_dist(rng);
const auto sx = size_dist(rng);
const auto sy = size_dist(rng);
const auto sz = size_dist(rng);
aabbs.push_back({{x, y, z}, {x + sx, y + sy, z + sz}});
}
const BvhTree tree(aabbs);
// Run 50 random queries
for (int i = 0; i < 50; ++i)
{
const auto qx = pos_dist(rng);
const auto qy = pos_dist(rng);
const auto qz = pos_dist(rng);
const auto qsx = size_dist(rng);
const auto qsy = size_dist(rng);
const auto qsz = size_dist(rng);
const Aabb query = {{qx, qy, qz}, {qx + qsx, qy + qsy, qz + qsz}};
const auto bvh_results = tree.query_overlaps(query);
const auto brute_results = brute_force_overlaps(aabbs, query);
const std::set<std::size_t> bvh_set(bvh_results.begin(), bvh_results.end());
EXPECT_EQ(bvh_set, brute_results) << "Mismatch on random query iteration " << i;
}
}
// ============================================================================
// Large dataset tests
// ============================================================================
TEST(UnitTestBvhTree, LargeGridDataset)
{
std::vector<Aabb> aabbs;
for (int x = 0; x < 10; ++x)
for (int y = 0; y < 10; ++y)
for (int z = 0; z < 10; ++z)
{
const auto fx = static_cast<float>(x) * 3.f;
const auto fy = static_cast<float>(y) * 3.f;
const auto fz = static_cast<float>(z) * 3.f;
aabbs.push_back({{fx, fy, fz}, {fx + 1.f, fy + 1.f, fz + 1.f}});
}
const BvhTree tree(aabbs);
EXPECT_FALSE(tree.empty());
const auto results = tree.query_overlaps({{0.f, 0.f, 0.f}, {1.5f, 1.5f, 1.5f}});
EXPECT_EQ(results.size(), 1);
const auto all_results = tree.query_overlaps({{-1.f, -1.f, -1.f}, {100.f, 100.f, 100.f}});
EXPECT_EQ(all_results.size(), 1000);
}
TEST(UnitTestBvhTree, FiveThousandObjects)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 5000; ++i)
{
const auto f = static_cast<float>(i) * 2.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
EXPECT_FALSE(tree.empty());
// Query that should hit exactly 1
const auto r = tree.query_overlaps({{0.f, 0.f, 0.f}, {0.5f, 0.5f, 0.5f}});
EXPECT_EQ(r.size(), 1);
// Query that misses
const auto miss = tree.query_overlaps({{-100.f, -100.f, -100.f}, {-90.f, -90.f, -90.f}});
EXPECT_TRUE(miss.empty());
}
// ============================================================================
// Double precision tests
// ============================================================================
TEST(UnitTestBvhTree, DoublePrecision)
{
const std::vector<AabbD> aabbs = {
{{0.0, 0.0, 0.0}, {1.0, 1.0, 1.0}},
{{5.0, 5.0, 5.0}, {6.0, 6.0, 6.0}},
{{10.0, 10.0, 10.0}, {11.0, 11.0, 11.0}},
};
const BvhTreeD tree(aabbs);
EXPECT_FALSE(tree.empty());
const auto r = tree.query_overlaps({{0.5, 0.5, 0.5}, {1.5, 1.5, 1.5}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
const auto r2 = tree.query_overlaps({{4.5, 4.5, 4.5}, {5.5, 5.5, 5.5}});
ASSERT_EQ(r2.size(), 1);
EXPECT_EQ(r2[0], 1);
}
TEST(UnitTestBvhTree, DoublePrecisionLargeCoordinates)
{
const std::vector<AabbD> aabbs = {
{{1e10, 1e10, 1e10}, {1e10 + 1.0, 1e10 + 1.0, 1e10 + 1.0}},
{{-1e10, -1e10, -1e10}, {-1e10 + 1.0, -1e10 + 1.0, -1e10 + 1.0}},
};
const BvhTreeD tree(aabbs);
const auto r = tree.query_overlaps({{1e10 - 0.5, 1e10 - 0.5, 1e10 - 0.5},
{1e10 + 0.5, 1e10 + 0.5, 1e10 + 0.5}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
}
// ============================================================================
// Edge case tests
// ============================================================================
TEST(UnitTestBvhTree, ZeroSizeQuery)
{
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
};
const BvhTree tree(aabbs);
// Point query inside the box
const auto r = tree.query_overlaps({{0.5f, 0.5f, 0.5f}, {0.5f, 0.5f, 0.5f}});
EXPECT_EQ(r.size(), 1);
// Point query outside the box
const auto miss = tree.query_overlaps({{5.f, 5.f, 5.f}, {5.f, 5.f, 5.f}});
EXPECT_TRUE(miss.empty());
}
TEST(UnitTestBvhTree, ZeroSizeObjects)
{
// Point-like AABBs
const std::vector<Aabb> aabbs = {
{{1.f, 1.f, 1.f}, {1.f, 1.f, 1.f}},
{{5.f, 5.f, 5.f}, {5.f, 5.f, 5.f}},
};
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{0.f, 0.f, 0.f}, {2.f, 2.f, 2.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 0);
}
TEST(UnitTestBvhTree, NoDuplicateResults)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 50; ++i)
{
const auto f = static_cast<float>(i) * 2.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
const auto r = tree.query_overlaps({{-1.f, -1.f, -1.f}, {200.f, 2.f, 2.f}});
// Check for duplicates
const std::set<std::size_t> unique_results(r.begin(), r.end());
EXPECT_EQ(unique_results.size(), r.size());
EXPECT_EQ(r.size(), 50);
}
TEST(UnitTestBvhTree, LargeSpread)
{
// Objects with huge gaps between them
const std::vector<Aabb> aabbs = {
{{0.f, 0.f, 0.f}, {1.f, 1.f, 1.f}},
{{1000.f, 0.f, 0.f}, {1001.f, 1.f, 1.f}},
{{-1000.f, 0.f, 0.f}, {-999.f, 1.f, 1.f}},
{{0.f, 1000.f, 0.f}, {1.f, 1001.f, 1.f}},
{{0.f, -1000.f, 0.f}, {1.f, -999.f, 1.f}},
};
const BvhTree tree(aabbs);
auto r = tree.query_overlaps({{999.f, -1.f, -1.f}, {1002.f, 2.f, 2.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 1);
r = tree.query_overlaps({{-1001.f, -1.f, -1.f}, {-998.f, 2.f, 2.f}});
ASSERT_EQ(r.size(), 1);
EXPECT_EQ(r[0], 2);
}
TEST(UnitTestBvhTree, AllObjectsSameCenter)
{
// All AABBs centered at origin but different sizes
std::vector<Aabb> aabbs;
for (int i = 1; i <= 10; ++i)
{
const auto s = static_cast<float>(i);
aabbs.push_back({{-s, -s, -s}, {s, s, s}});
}
const BvhTree tree(aabbs);
// Small query at origin should hit all
const auto r = tree.query_overlaps({{-0.1f, -0.1f, -0.1f}, {0.1f, 0.1f, 0.1f}});
EXPECT_EQ(r.size(), 10);
// Query touching only the largest box
const auto r2 = tree.query_overlaps({{9.5f, 9.5f, 9.5f}, {10.5f, 10.5f, 10.5f}});
ASSERT_EQ(r2.size(), 1);
EXPECT_EQ(r2[0], 9);
}
TEST(UnitTestBvhTree, MultipleQueriesSameTree)
{
std::vector<Aabb> aabbs;
for (int i = 0; i < 100; ++i)
{
const auto f = static_cast<float>(i) * 2.f;
aabbs.push_back({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
}
const BvhTree tree(aabbs);
// Run many queries on the same tree
for (int i = 0; i < 100; ++i)
{
const auto f = static_cast<float>(i) * 2.f;
const auto r = tree.query_overlaps({{f, 0.f, 0.f}, {f + 1.f, 1.f, 1.f}});
ASSERT_GE(r.size(), 1) << "Query for object " << i << " should find at least itself";
}
}

View File

@@ -0,0 +1,125 @@
//
// Created by Vlad on 3/25/2025.
//
#include "omath/collision/line_tracer.hpp"
#include "omath/3d_primitives/aabb.hpp"
#include <gtest/gtest.h>
using Vec3 = omath::Vector3<float>;
using Ray = omath::collision::Ray<>;
using LineTracer = omath::collision::LineTracer<>;
using AABB = omath::primitives::Aabb<float>;
static Ray make_ray(Vec3 start, Vec3 end, bool infinite = false)
{
Ray r;
r.start = start;
r.end = end;
r.infinite_length = infinite;
return r;
}
// Ray passing straight through the center along Z axis
TEST(LineTracerAABBTests, HitCenterAlongZ)
{
const AABB box{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}};
const auto ray = make_ray({0.f, 0.f, -5.f}, {0.f, 0.f, 5.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_NE(hit, ray.end);
EXPECT_NEAR(hit.z, -1.f, 1e-4f);
EXPECT_NEAR(hit.x, 0.f, 1e-4f);
EXPECT_NEAR(hit.y, 0.f, 1e-4f);
}
// Ray passing straight through the center along X axis
TEST(LineTracerAABBTests, HitCenterAlongX)
{
const AABB box{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}};
const auto ray = make_ray({-5.f, 0.f, 0.f}, {5.f, 0.f, 0.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_NE(hit, ray.end);
EXPECT_NEAR(hit.x, -1.f, 1e-4f);
}
// Ray that misses entirely (too far in Y)
TEST(LineTracerAABBTests, MissReturnsEnd)
{
const AABB box{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}};
const auto ray = make_ray({0.f, 5.f, -5.f}, {0.f, 5.f, 5.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_EQ(hit, ray.end);
}
// Ray that stops short before reaching the box
TEST(LineTracerAABBTests, RayTooShortReturnsEnd)
{
const AABB box{{3.f, -1.f, -1.f}, {5.f, 1.f, 1.f}};
const auto ray = make_ray({0.f, 0.f, 0.f}, {2.f, 0.f, 0.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_EQ(hit, ray.end);
}
// Infinite ray that starts before the box should hit
TEST(LineTracerAABBTests, InfiniteRayHits)
{
const AABB box{{3.f, -1.f, -1.f}, {5.f, 1.f, 1.f}};
const auto ray = make_ray({0.f, 0.f, 0.f}, {2.f, 0.f, 0.f}, true);
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_NE(hit, ray.end);
EXPECT_NEAR(hit.x, 3.f, 1e-4f);
}
// Ray starting inside the box — t_min=0, so hit point equals ray.start
TEST(LineTracerAABBTests, RayStartsInsideBox)
{
const AABB box{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}};
const auto ray = make_ray({0.f, 0.f, 0.f}, {0.f, 0.f, 5.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_NE(hit, ray.end);
// t_min is clamped to 0, so hit == start
EXPECT_NEAR(hit.x, 0.f, 1e-4f);
EXPECT_NEAR(hit.y, 0.f, 1e-4f);
EXPECT_NEAR(hit.z, 0.f, 1e-4f);
}
// Ray parallel to XY plane, pointing along X, at Z outside the box
TEST(LineTracerAABBTests, ParallelRayOutsideSlabMisses)
{
const AABB box{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}};
// Z component of ray is 3.0 — outside box's Z slab
const auto ray = make_ray({-5.f, 0.f, 3.f}, {5.f, 0.f, 3.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_EQ(hit, ray.end);
}
// Ray parallel to XY plane, pointing along X, at Z inside the box
TEST(LineTracerAABBTests, ParallelRayInsideSlabHits)
{
const AABB box{{-1.f, -1.f, -1.f}, {1.f, 1.f, 1.f}};
const auto ray = make_ray({-5.f, 0.f, 0.f}, {5.f, 0.f, 0.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_NE(hit, ray.end);
EXPECT_NEAR(hit.x, -1.f, 1e-4f);
}
// Diagonal ray hitting a corner region
TEST(LineTracerAABBTests, DiagonalRayHits)
{
const AABB box{{0.f, 0.f, 0.f}, {2.f, 2.f, 2.f}};
const auto ray = make_ray({-1.f, -1.f, -1.f}, {3.f, 3.f, 3.f});
const auto hit = LineTracer::get_ray_hit_point(ray, box);
EXPECT_NE(hit, ray.end);
// Entry point should be at (0,0,0)
EXPECT_NEAR(hit.x, 0.f, 1e-4f);
EXPECT_NEAR(hit.y, 0.f, 1e-4f);
EXPECT_NEAR(hit.z, 0.f, 1e-4f);
}