mirror of
https://github.com/orange-cpp/omath.git
synced 2026-04-19 08:03:27 +00:00
Compare commits
36 Commits
624683aed6
...
v5.1.0.rc6
| Author | SHA1 | Date | |
|---|---|---|---|
| 308f7ed481 | |||
| 8802ad9af1 | |||
| 2ac508d6e8 | |||
| eb1ca6055b | |||
| b528e41de3 | |||
| 8615ab2b7c | |||
| 5a4c042fec | |||
| 8063c1697a | |||
| 7567501f00 | |||
| 46d999f846 | |||
| b54601132b | |||
| 5c8ce2d163 | |||
| 04a86739b4 | |||
| 575b411863 | |||
| 5a91151bc0 | |||
| 66d4df0524 | |||
| 54e14760ca | |||
| ee61c47d7d | |||
| d737aee1c5 | |||
| ef422f0a86 | |||
| e99ca0bc2b | |||
| 5f94e36965 | |||
| 29510cf9e7 | |||
| 927508a76b | |||
| f390b386d7 | |||
| 012d837e8b | |||
| 6236c8fd68 | |||
| 06dc36089f | |||
| 91136a61c4 | |||
| 9cdffcbdb1 | |||
| a3e93ac259 | |||
| 59f6d7a361 | |||
| dcf1ef1ea9 | |||
| 89bd879187 | |||
| aa08c7cb65 | |||
| a5c0ca0cbd |
7
.claude/settings.local.json
Normal file
7
.claude/settings.local.json
Normal file
@@ -0,0 +1,7 @@
|
||||
{
|
||||
"permissions": {
|
||||
"allow": [
|
||||
"Bash(ls:*)"
|
||||
]
|
||||
}
|
||||
}
|
||||
3
.github/workflows/cmake-multi-platform.yml
vendored
3
.github/workflows/cmake-multi-platform.yml
vendored
@@ -370,6 +370,8 @@ jobs:
|
||||
shell: bash
|
||||
run: |
|
||||
cmake --preset ${{ matrix.preset }} \
|
||||
-DCMAKE_C_COMPILER=$(xcrun --find clang) \
|
||||
-DCMAKE_CXX_COMPILER=$(xcrun --find clang++) \
|
||||
-DOMATH_BUILD_TESTS=ON \
|
||||
-DOMATH_BUILD_BENCHMARK=OFF \
|
||||
-DOMATH_ENABLE_COVERAGE=${{ matrix.coverage == true && 'ON' || 'OFF' }} \
|
||||
@@ -380,6 +382,7 @@ jobs:
|
||||
run: cmake --build cmake-build/build/${{ matrix.preset }} --target unit_tests omath
|
||||
|
||||
- name: Run unit_tests
|
||||
if: ${{ matrix.coverage != true }}
|
||||
shell: bash
|
||||
run: ./out/Release/unit_tests
|
||||
|
||||
|
||||
62
.github/workflows/docs.yml
vendored
Normal file
62
.github/workflows/docs.yml
vendored
Normal file
@@ -0,0 +1,62 @@
|
||||
name: Documentation
|
||||
|
||||
on:
|
||||
push:
|
||||
branches: [ main ]
|
||||
paths:
|
||||
- 'docs/**'
|
||||
- 'mkdocs.yml'
|
||||
- '.github/workflows/docs.yml'
|
||||
pull_request:
|
||||
branches: [ main ]
|
||||
paths:
|
||||
- 'docs/**'
|
||||
- 'mkdocs.yml'
|
||||
- '.github/workflows/docs.yml'
|
||||
|
||||
concurrency:
|
||||
group: docs-${{ github.ref }}
|
||||
cancel-in-progress: true
|
||||
|
||||
permissions:
|
||||
contents: read
|
||||
pages: write
|
||||
id-token: write
|
||||
|
||||
jobs:
|
||||
build:
|
||||
name: Build Documentation
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v5
|
||||
with:
|
||||
python-version: '3.x'
|
||||
|
||||
- name: Install mkdocs and dependencies
|
||||
run: pip install mkdocs mkdocs-bootswatch
|
||||
|
||||
- name: Build documentation
|
||||
run: mkdocs build --strict
|
||||
|
||||
- name: Upload artifact
|
||||
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
|
||||
uses: actions/upload-pages-artifact@v3
|
||||
with:
|
||||
path: site/
|
||||
|
||||
deploy:
|
||||
name: Deploy to GitHub Pages
|
||||
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
|
||||
needs: build
|
||||
runs-on: ubuntu-latest
|
||||
environment:
|
||||
name: github-pages
|
||||
url: ${{ steps.deployment.outputs.page_url }}
|
||||
steps:
|
||||
- name: Deploy to GitHub Pages
|
||||
id: deployment
|
||||
uses: actions/deploy-pages@v4
|
||||
29
.github/workflows/release.yml
vendored
29
.github/workflows/release.yml
vendored
@@ -12,6 +12,35 @@ permissions:
|
||||
contents: write
|
||||
|
||||
jobs:
|
||||
##############################################################################
|
||||
# 0) Documentation – MkDocs
|
||||
##############################################################################
|
||||
docs-release:
|
||||
name: Documentation
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v5
|
||||
with:
|
||||
python-version: '3.x'
|
||||
|
||||
- name: Install mkdocs and dependencies
|
||||
run: pip install mkdocs mkdocs-bootswatch
|
||||
|
||||
- name: Build documentation
|
||||
run: mkdocs build --strict
|
||||
|
||||
- name: Package
|
||||
run: tar -czf omath-docs.tar.gz -C site .
|
||||
|
||||
- name: Upload release asset
|
||||
env:
|
||||
GH_TOKEN: ${{ github.token }}
|
||||
run: gh release upload "${{ github.event.release.tag_name }}" omath-docs.tar.gz --clobber
|
||||
|
||||
##############################################################################
|
||||
# 1) Linux – Clang / Ninja
|
||||
##############################################################################
|
||||
|
||||
@@ -3,7 +3,6 @@
|
||||
Thanks to everyone who made this possible, including:
|
||||
|
||||
- Saikari aka luadebug for VCPKG port and awesome new initial logo design.
|
||||
- AmbushedRaccoon for telegram post about omath to boost repository activity.
|
||||
- Billy O'Neal aka BillyONeal for fixing compilation issues due to C math library compatibility.
|
||||
- Alex2772 for reference of AUI declarative interface design for omath::hud
|
||||
|
||||
|
||||
23
INSTALL.md
23
INSTALL.md
@@ -28,6 +28,29 @@ target("...")
|
||||
add_packages("omath")
|
||||
```
|
||||
|
||||
## <img width="28px" src="https://conan.io/favicon.png" /> Using Conan
|
||||
**Note**: Support Conan for package management
|
||||
1. Install [Conan](https://conan.io/downloads)
|
||||
2. Run the following command to install the omath package:
|
||||
```
|
||||
conan install --requires="omath/[*]" --build=missing
|
||||
```
|
||||
conanfile.txt
|
||||
```ini
|
||||
[requires]
|
||||
omath/[*]
|
||||
|
||||
[generators]
|
||||
CMakeDeps
|
||||
CMakeToolchain
|
||||
```
|
||||
CMakeLists.txt
|
||||
```cmake
|
||||
find_package(omath CONFIG REQUIRED)
|
||||
target_link_libraries(main PRIVATE omath::omath)
|
||||
```
|
||||
For more details, see the [Conan documentation](https://docs.conan.io/2/).
|
||||
|
||||
## <img width="28px" src="https://github.githubassets.com/favicons/favicon.svg" /> Using prebuilt binaries (GitHub Releases)
|
||||
|
||||
**Note**: This is the fastest option if you don’t want to build from source.
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
# Installation
|
||||
# Installation Guide
|
||||
|
||||
## <img width="28px" src="https://vcpkg.io/assets/mark/mark.svg" /> Using vcpkg
|
||||
## <img width="28px" src="https://vcpkg.io/assets/mark/mark.svg" /> Using vcpkg (recomended)
|
||||
**Note**: Support vcpkg for package management
|
||||
1. Install [vcpkg](https://github.com/microsoft/vcpkg)
|
||||
2. Run the following command to install the orange-math package:
|
||||
@@ -28,6 +28,69 @@ target("...")
|
||||
add_packages("omath")
|
||||
```
|
||||
|
||||
## <img width="28px" src="https://conan.io/favicon.png" /> Using Conan
|
||||
**Note**: Support Conan for package management
|
||||
1. Install [Conan](https://conan.io/downloads)
|
||||
2. Run the following command to install the omath package:
|
||||
```
|
||||
conan install --requires="omath/[*]" --build=missing
|
||||
```
|
||||
conanfile.txt
|
||||
```ini
|
||||
[requires]
|
||||
omath/[*]
|
||||
|
||||
[generators]
|
||||
CMakeDeps
|
||||
CMakeToolchain
|
||||
```
|
||||
CMakeLists.txt
|
||||
```cmake
|
||||
find_package(omath CONFIG REQUIRED)
|
||||
target_link_libraries(main PRIVATE omath::omath)
|
||||
```
|
||||
For more details, see the [Conan documentation](https://docs.conan.io/2/).
|
||||
|
||||
## <img width="28px" src="https://github.githubassets.com/favicons/favicon.svg" /> Using prebuilt binaries (GitHub Releases)
|
||||
|
||||
**Note**: This is the fastest option if you don’t want to build from source.
|
||||
|
||||
1. **Go to the Releases page**
|
||||
- Open the project’s GitHub **Releases** page and choose the latest version.
|
||||
|
||||
2. **Download the correct asset for your platform**
|
||||
- Pick the archive that matches your OS and architecture (for example: Windows x64 / Linux x64 / macOS arm64).
|
||||
|
||||
3. **Extract the archive**
|
||||
- You should end up with something like:
|
||||
- `include/` (headers)
|
||||
- `lib/` or `bin/` (library files / DLLs)
|
||||
- sometimes `cmake/` (CMake package config)
|
||||
|
||||
4. **Use it in your project**
|
||||
|
||||
### Option A: CMake package (recommended if the release includes CMake config files)
|
||||
If the extracted folder contains something like `lib/cmake/omath` or `cmake/omath`, you can point CMake to it:
|
||||
|
||||
```cmake
|
||||
# Example: set this to the extracted prebuilt folder
|
||||
list(APPEND CMAKE_PREFIX_PATH "path/to/omath-prebuilt")
|
||||
|
||||
find_package(omath CONFIG REQUIRED)
|
||||
target_link_libraries(main PRIVATE omath::omath)
|
||||
```
|
||||
### Option B: Manual include + link (works with any layout)
|
||||
If there’s no CMake package config, link it manually:
|
||||
```cmake
|
||||
target_include_directories(main PRIVATE "path/to/omath-prebuilt/include")
|
||||
|
||||
# Choose ONE depending on what you downloaded:
|
||||
# - Static library: .lib / .a
|
||||
# - Shared library: .dll + .lib import (Windows), .so (Linux), .dylib (macOS)
|
||||
|
||||
target_link_directories(main PRIVATE "path/to/omath-prebuilt/lib")
|
||||
target_link_libraries(main PRIVATE omath) # or the actual library filename
|
||||
```
|
||||
## <img width="28px" src="https://upload.wikimedia.org/wikipedia/commons/e/ef/CMake_logo.svg?" /> Build from source using CMake
|
||||
1. **Preparation**
|
||||
|
||||
@@ -62,7 +125,7 @@ target("...")
|
||||
Use **\<platform\>-\<build configuration\>** preset to build suitable version for yourself. Like **windows-release** or **linux-release**.
|
||||
|
||||
| Platform Name | Build Config |
|
||||
|---------------|---------------|
|
||||
|---------------|---------------|
|
||||
| windows | release/debug |
|
||||
| linux | release/debug |
|
||||
| darwin | release/debug |
|
||||
|
||||
103
include/omath/algorithm/targeting.hpp
Normal file
103
include/omath/algorithm/targeting.hpp
Normal file
@@ -0,0 +1,103 @@
|
||||
//
|
||||
// Created by Vladislav on 19.03.2026.
|
||||
//
|
||||
|
||||
#pragma once
|
||||
#include "omath/linear_algebra/vector3.hpp"
|
||||
#include <functional>
|
||||
#include <iterator>
|
||||
#include <optional>
|
||||
#include <ranges>
|
||||
|
||||
namespace omath::algorithm
|
||||
{
|
||||
template<class CameraType, std::input_or_output_iterator IteratorType, class FilterT>
|
||||
requires std::is_invocable_r_v<bool, std::function<FilterT>, std::iter_reference_t<IteratorType>>
|
||||
[[nodiscard]]
|
||||
IteratorType get_closest_target_by_fov(const IteratorType& begin, const IteratorType& end, const CameraType& camera,
|
||||
auto get_position,
|
||||
const std::optional<std::function<FilterT>>& filter_func = std::nullopt)
|
||||
{
|
||||
auto best_target = end;
|
||||
const auto& camera_angles = camera.get_view_angles();
|
||||
const Vector2<float> camera_angles_vec = {camera_angles.pitch.as_degrees(), camera_angles.yaw.as_degrees()};
|
||||
|
||||
for (auto current = begin; current != end; current = std::next(current))
|
||||
{
|
||||
if (filter_func && !filter_func.value()(*current))
|
||||
continue;
|
||||
|
||||
if (best_target == end)
|
||||
{
|
||||
best_target = current;
|
||||
continue;
|
||||
}
|
||||
const auto current_target_angles = camera.calc_look_at_angles(get_position(*current));
|
||||
const auto best_target_angles = camera.calc_look_at_angles(get_position(*best_target));
|
||||
|
||||
const Vector2<float> current_angles_vec = {current_target_angles.pitch.as_degrees(),
|
||||
current_target_angles.yaw.as_degrees()};
|
||||
const Vector2<float> best_angles_vec = {best_target_angles.pitch.as_degrees(),
|
||||
best_target_angles.yaw.as_degrees()};
|
||||
|
||||
const auto current_target_distance = camera_angles_vec.distance_to(current_angles_vec);
|
||||
const auto best_target_distance = camera_angles_vec.distance_to(best_angles_vec);
|
||||
if (current_target_distance < best_target_distance)
|
||||
best_target = current;
|
||||
}
|
||||
return best_target;
|
||||
}
|
||||
|
||||
template<class CameraType, std::ranges::range RangeType, class FilterT>
|
||||
requires std::is_invocable_r_v<bool, std::function<FilterT>,
|
||||
std::ranges::range_reference_t<const RangeType>>
|
||||
[[nodiscard]]
|
||||
auto get_closest_target_by_fov(const RangeType& range, const CameraType& camera,
|
||||
auto get_position,
|
||||
const std::optional<std::function<FilterT>>& filter_func = std::nullopt)
|
||||
{
|
||||
return get_closest_target_by_fov<CameraType, decltype(std::ranges::begin(range)), FilterT>(
|
||||
std::ranges::begin(range), std::ranges::end(range), camera, get_position, filter_func);
|
||||
}
|
||||
|
||||
// ── By world-space distance ───────────────────────────────────────────────
|
||||
|
||||
template<std::input_or_output_iterator IteratorType, class FilterT>
|
||||
requires std::is_invocable_r_v<bool, std::function<FilterT>, std::iter_reference_t<IteratorType>>
|
||||
[[nodiscard]]
|
||||
IteratorType get_closest_target_by_distance(const IteratorType& begin, const IteratorType& end,
|
||||
const Vector3<float>& origin, auto get_position,
|
||||
const std::optional<std::function<FilterT>>& filter_func = std::nullopt)
|
||||
{
|
||||
auto best_target = end;
|
||||
|
||||
for (auto current = begin; current != end; current = std::next(current))
|
||||
{
|
||||
if (filter_func && !filter_func.value()(*current))
|
||||
continue;
|
||||
|
||||
if (best_target == end)
|
||||
{
|
||||
best_target = current;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (origin.distance_to(get_position(*current)) < origin.distance_to(get_position(*best_target)))
|
||||
best_target = current;
|
||||
}
|
||||
return best_target;
|
||||
}
|
||||
|
||||
template<std::ranges::range RangeType, class FilterT>
|
||||
requires std::is_invocable_r_v<bool, std::function<FilterT>,
|
||||
std::ranges::range_reference_t<const RangeType>>
|
||||
[[nodiscard]]
|
||||
auto get_closest_target_by_distance(const RangeType& range, const Vector3<float>& origin,
|
||||
auto get_position,
|
||||
const std::optional<std::function<FilterT>>& filter_func = std::nullopt)
|
||||
{
|
||||
return get_closest_target_by_distance<decltype(std::ranges::begin(range)), FilterT>(
|
||||
std::ranges::begin(range), std::ranges::end(range), origin, get_position, filter_func);
|
||||
}
|
||||
|
||||
} // namespace omath::algorithm
|
||||
@@ -16,7 +16,8 @@ namespace omath::cry_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -16,7 +16,8 @@ namespace omath::frostbite_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -17,7 +17,8 @@ namespace omath::iw_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -16,7 +16,8 @@ namespace omath::opengl_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -17,7 +17,8 @@ namespace omath::source_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -16,7 +16,8 @@ namespace omath::unity_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -16,7 +16,8 @@ namespace omath::unreal_engine
|
||||
const float pitch, const float yaw,
|
||||
const float time, const float gravity) noexcept
|
||||
{
|
||||
auto current_pos = projectile.m_origin
|
||||
const auto launch_pos = projectile.m_origin + projectile.m_launch_offset;
|
||||
auto current_pos = launch_pos
|
||||
+ forward_vector({PitchAngle::from_degrees(-pitch), YawAngle::from_degrees(yaw),
|
||||
RollAngle::from_degrees(0)})
|
||||
* projectile.m_launch_speed * time;
|
||||
|
||||
@@ -8,12 +8,23 @@
|
||||
|
||||
namespace omath::projectile_prediction
|
||||
{
|
||||
struct AimAngles
|
||||
{
|
||||
float pitch{};
|
||||
float yaw{};
|
||||
};
|
||||
|
||||
class ProjPredEngineInterface
|
||||
{
|
||||
public:
|
||||
[[nodiscard]]
|
||||
virtual std::optional<Vector3<float>> maybe_calculate_aim_point(const Projectile& projectile,
|
||||
const Target& target) const = 0;
|
||||
|
||||
[[nodiscard]]
|
||||
virtual std::optional<AimAngles> maybe_calculate_aim_angles(const Projectile& projectile,
|
||||
const Target& target) const = 0;
|
||||
|
||||
virtual ~ProjPredEngineInterface() = default;
|
||||
};
|
||||
} // namespace omath::projectile_prediction
|
||||
|
||||
@@ -12,6 +12,9 @@ namespace omath::projectile_prediction
|
||||
[[nodiscard]] std::optional<Vector3<float>>
|
||||
maybe_calculate_aim_point(const Projectile& projectile, const Target& target) const override;
|
||||
|
||||
[[nodiscard]] std::optional<AimAngles>
|
||||
maybe_calculate_aim_angles(const Projectile& projectile, const Target& target) const override;
|
||||
|
||||
ProjPredEngineAvx2(float gravity_constant, float simulation_time_step, float maximum_simulation_time);
|
||||
~ProjPredEngineAvx2() override = default;
|
||||
|
||||
|
||||
@@ -54,6 +54,36 @@ namespace omath::projectile_prediction
|
||||
[[nodiscard]]
|
||||
std::optional<Vector3<float>> maybe_calculate_aim_point(const Projectile& projectile,
|
||||
const Target& target) const override
|
||||
{
|
||||
const auto solution = find_solution(projectile, target);
|
||||
if (!solution)
|
||||
return std::nullopt;
|
||||
|
||||
return EngineTrait::calc_viewpoint_from_angles(projectile, solution->predicted_target_position,
|
||||
solution->pitch);
|
||||
}
|
||||
|
||||
[[nodiscard]]
|
||||
std::optional<AimAngles> maybe_calculate_aim_angles(const Projectile& projectile,
|
||||
const Target& target) const override
|
||||
{
|
||||
const auto solution = find_solution(projectile, target);
|
||||
if (!solution)
|
||||
return std::nullopt;
|
||||
|
||||
const auto yaw = EngineTrait::calc_direct_yaw_angle(projectile.m_origin + projectile.m_launch_offset, solution->predicted_target_position);
|
||||
return AimAngles{solution->pitch, yaw};
|
||||
}
|
||||
|
||||
private:
|
||||
struct Solution
|
||||
{
|
||||
Vector3<float> predicted_target_position;
|
||||
float pitch;
|
||||
};
|
||||
|
||||
[[nodiscard]]
|
||||
std::optional<Solution> find_solution(const Projectile& projectile, const Target& target) const
|
||||
{
|
||||
for (float time = 0.f; time < m_maximum_simulation_time; time += m_simulation_time_step)
|
||||
{
|
||||
@@ -70,12 +100,11 @@ namespace omath::projectile_prediction
|
||||
time))
|
||||
continue;
|
||||
|
||||
return EngineTrait::calc_viewpoint_from_angles(projectile, predicted_target_position, projectile_pitch);
|
||||
return Solution{predicted_target_position, projectile_pitch.value()};
|
||||
}
|
||||
return std::nullopt;
|
||||
}
|
||||
|
||||
private:
|
||||
const float m_gravity_constant;
|
||||
const float m_simulation_time_step;
|
||||
const float m_maximum_simulation_time;
|
||||
@@ -100,10 +129,12 @@ namespace omath::projectile_prediction
|
||||
{
|
||||
const auto bullet_gravity = m_gravity_constant * projectile.m_gravity_scale;
|
||||
|
||||
if (bullet_gravity == 0.f)
|
||||
return EngineTrait::calc_direct_pitch_angle(projectile.m_origin, target_position);
|
||||
const auto launch_origin = projectile.m_origin + projectile.m_launch_offset;
|
||||
|
||||
const auto delta = target_position - projectile.m_origin;
|
||||
if (bullet_gravity == 0.f)
|
||||
return EngineTrait::calc_direct_pitch_angle(launch_origin, target_position);
|
||||
|
||||
const auto delta = target_position - launch_origin;
|
||||
|
||||
const auto distance2d = EngineTrait::calc_vector_2d_distance(delta);
|
||||
const auto distance2d_sqr = distance2d * distance2d;
|
||||
@@ -126,7 +157,7 @@ namespace omath::projectile_prediction
|
||||
bool is_projectile_reached_target(const Vector3<float>& target_position, const Projectile& projectile,
|
||||
const float pitch, const float time) const noexcept
|
||||
{
|
||||
const auto yaw = EngineTrait::calc_direct_yaw_angle(projectile.m_origin, target_position);
|
||||
const auto yaw = EngineTrait::calc_direct_yaw_angle(projectile.m_origin + projectile.m_launch_offset, target_position);
|
||||
const auto projectile_position =
|
||||
EngineTrait::predict_projectile_position(projectile, pitch, yaw, time, m_gravity_constant);
|
||||
|
||||
|
||||
@@ -11,6 +11,7 @@ namespace omath::projectile_prediction
|
||||
{
|
||||
public:
|
||||
Vector3<float> m_origin;
|
||||
Vector3<float> m_launch_offset{0.f, 0.f, 0.f};
|
||||
float m_launch_speed{};
|
||||
float m_gravity_scale{};
|
||||
};
|
||||
|
||||
@@ -36,7 +36,11 @@ namespace omath::projection
|
||||
}
|
||||
};
|
||||
using FieldOfView = Angle<float, 0.f, 180.f, AngleFlags::Clamped>;
|
||||
|
||||
enum class ViewPortClipping
|
||||
{
|
||||
AUTO,
|
||||
MANUAL,
|
||||
};
|
||||
template<class T, class MatType, class ViewAnglesType>
|
||||
concept CameraEngineConcept =
|
||||
requires(const Vector3<float>& cam_origin, const Vector3<float>& look_at, const ViewAnglesType& angles,
|
||||
@@ -82,6 +86,11 @@ namespace omath::projection
|
||||
m_view_projection_matrix = std::nullopt;
|
||||
m_view_matrix = std::nullopt;
|
||||
}
|
||||
[[nodiscard]]
|
||||
ViewAnglesType calc_look_at_angles(const Vector3<float>& look_to) const
|
||||
{
|
||||
return TraitClass::calc_look_at_angle(m_origin, look_to);
|
||||
}
|
||||
|
||||
[[nodiscard]]
|
||||
Vector3<float> get_forward() const noexcept
|
||||
@@ -138,16 +147,16 @@ namespace omath::projection
|
||||
m_projection_matrix = std::nullopt;
|
||||
}
|
||||
|
||||
void set_near_plane(const float near) noexcept
|
||||
void set_near_plane(const float near_plane) noexcept
|
||||
{
|
||||
m_near_plane_distance = near;
|
||||
m_near_plane_distance = near_plane;
|
||||
m_view_projection_matrix = std::nullopt;
|
||||
m_projection_matrix = std::nullopt;
|
||||
}
|
||||
|
||||
void set_far_plane(const float far) noexcept
|
||||
void set_far_plane(const float far_plane) noexcept
|
||||
{
|
||||
m_far_plane_distance = far;
|
||||
m_far_plane_distance = far_plane;
|
||||
m_view_projection_matrix = std::nullopt;
|
||||
m_projection_matrix = std::nullopt;
|
||||
}
|
||||
@@ -213,6 +222,22 @@ namespace omath::projection
|
||||
else
|
||||
std::unreachable();
|
||||
}
|
||||
template<ScreenStart screen_start = ScreenStart::TOP_LEFT_CORNER>
|
||||
[[nodiscard]] std::expected<Vector3<float>, Error>
|
||||
world_to_screen_unclipped(const Vector3<float>& world_position) const noexcept
|
||||
{
|
||||
const auto normalized_cords = world_to_view_port(world_position, ViewPortClipping::MANUAL);
|
||||
|
||||
if (!normalized_cords.has_value())
|
||||
return std::unexpected{normalized_cords.error()};
|
||||
|
||||
if constexpr (screen_start == ScreenStart::TOP_LEFT_CORNER)
|
||||
return ndc_to_screen_position_from_top_left_corner(*normalized_cords);
|
||||
else if constexpr (screen_start == ScreenStart::BOTTOM_LEFT_CORNER)
|
||||
return ndc_to_screen_position_from_bottom_left_corner(*normalized_cords);
|
||||
else
|
||||
std::unreachable();
|
||||
}
|
||||
|
||||
[[nodiscard]] bool is_culled_by_frustum(const Triangle<Vector3<float>>& triangle) const noexcept
|
||||
{
|
||||
@@ -262,24 +287,34 @@ namespace omath::projection
|
||||
}
|
||||
|
||||
[[nodiscard]] std::expected<Vector3<float>, Error>
|
||||
world_to_view_port(const Vector3<float>& world_position) const noexcept
|
||||
world_to_view_port(const Vector3<float>& world_position,
|
||||
const ViewPortClipping& clipping = ViewPortClipping::AUTO) const noexcept
|
||||
{
|
||||
auto projected = get_view_projection_matrix()
|
||||
* mat_column_from_vector<float, Mat4X4Type::get_store_ordering()>(world_position);
|
||||
|
||||
const auto& w = projected.at(3, 0);
|
||||
if (w <= std::numeric_limits<float>::epsilon())
|
||||
return std::unexpected(Error::WORLD_POSITION_IS_OUT_OF_SCREEN_BOUNDS);
|
||||
constexpr auto eps = std::numeric_limits<float>::epsilon();
|
||||
if (w <= eps)
|
||||
return std::unexpected(Error::PERSPECTIVE_DIVIDER_LESS_EQ_ZERO);
|
||||
|
||||
projected /= w;
|
||||
|
||||
if (is_ndc_out_of_bounds(projected))
|
||||
// ReSharper disable once CppTooWideScope
|
||||
const auto clipped_automatically = clipping == ViewPortClipping::AUTO && is_ndc_out_of_bounds(projected);
|
||||
if (clipped_automatically)
|
||||
return std::unexpected(Error::WORLD_POSITION_IS_OUT_OF_SCREEN_BOUNDS);
|
||||
|
||||
// ReSharper disable once CppTooWideScope
|
||||
const auto clipped_manually = clipping == ViewPortClipping::MANUAL && (projected.at(2, 0) < 0.0f - eps
|
||||
|| projected.at(2, 0) > 1.0f + eps);
|
||||
if (clipped_manually)
|
||||
return std::unexpected(Error::WORLD_POSITION_IS_OUT_OF_SCREEN_BOUNDS);
|
||||
|
||||
return Vector3<float>{projected.at(0, 0), projected.at(1, 0), projected.at(2, 0)};
|
||||
}
|
||||
[[nodiscard]]
|
||||
std::expected<Vector3<float>, Error> view_port_to_screen(const Vector3<float>& ndc) const noexcept
|
||||
std::expected<Vector3<float>, Error> view_port_to_world(const Vector3<float>& ndc) const noexcept
|
||||
{
|
||||
const auto inv_view_proj = get_view_projection_matrix().inverted();
|
||||
|
||||
@@ -304,7 +339,7 @@ namespace omath::projection
|
||||
[[nodiscard]]
|
||||
std::expected<Vector3<float>, Error> screen_to_world(const Vector3<float>& screen_pos) const noexcept
|
||||
{
|
||||
return view_port_to_screen(screen_to_ndc<screen_start>(screen_pos));
|
||||
return view_port_to_world(screen_to_ndc<screen_start>(screen_pos));
|
||||
}
|
||||
|
||||
template<ScreenStart screen_start = ScreenStart::TOP_LEFT_CORNER>
|
||||
|
||||
@@ -11,5 +11,6 @@ namespace omath::projection
|
||||
{
|
||||
WORLD_POSITION_IS_OUT_OF_SCREEN_BOUNDS,
|
||||
INV_VIEW_PROJ_MAT_DET_EQ_ZERO,
|
||||
PERSPECTIVE_DIVIDER_LESS_EQ_ZERO,
|
||||
};
|
||||
}
|
||||
@@ -3,11 +3,43 @@
|
||||
//
|
||||
|
||||
#pragma once
|
||||
#include <cassert>
|
||||
#include <cstddef>
|
||||
#include <cstdint>
|
||||
#include <string_view>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include "omath/utility/pe_pattern_scan.hpp"
|
||||
#include <windows.h>
|
||||
#elif defined(__APPLE__)
|
||||
#include "omath/utility/macho_pattern_scan.hpp"
|
||||
#include <mach-o/dyld.h>
|
||||
#else
|
||||
#include "omath/utility/elf_pattern_scan.hpp"
|
||||
#include <link.h>
|
||||
#endif
|
||||
|
||||
namespace omath::rev_eng
|
||||
{
|
||||
template<std::size_t N>
|
||||
struct FixedString final
|
||||
{
|
||||
char data[N]{};
|
||||
// ReSharper disable once CppNonExplicitConvertingConstructor
|
||||
constexpr FixedString(const char (&str)[N]) noexcept // NOLINT(*-explicit-constructor)
|
||||
{
|
||||
for (std::size_t i = 0; i < N; ++i)
|
||||
data[i] = str[i];
|
||||
}
|
||||
// ReSharper disable once CppNonExplicitConversionOperator
|
||||
constexpr operator std::string_view() const noexcept // NOLINT(*-explicit-constructor)
|
||||
{
|
||||
return {data, N - 1};
|
||||
}
|
||||
};
|
||||
template<std::size_t N>
|
||||
FixedString(const char (&)[N]) -> FixedString<N>;
|
||||
|
||||
class InternalReverseEngineeredObject
|
||||
{
|
||||
protected:
|
||||
@@ -23,26 +55,123 @@ namespace omath::rev_eng
|
||||
return *reinterpret_cast<Type*>(reinterpret_cast<std::uintptr_t>(this) + offset);
|
||||
}
|
||||
|
||||
template<std::size_t id, class ReturnType>
|
||||
template<class ReturnType>
|
||||
ReturnType call_method(const void* ptr, auto... arg_list)
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
using MethodType = ReturnType(__thiscall*)(void*, decltype(arg_list)...);
|
||||
#else
|
||||
using MethodType = ReturnType (*)(void*, decltype(arg_list)...);
|
||||
#endif
|
||||
return reinterpret_cast<MethodType>(const_cast<void*>(ptr))(this, arg_list...);
|
||||
}
|
||||
template<class ReturnType>
|
||||
ReturnType call_method(const void* ptr, auto... arg_list) const
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
using MethodType = ReturnType(__thiscall*)(const void*, decltype(arg_list)...);
|
||||
#else
|
||||
using MethodType = ReturnType (*)(const void*, decltype(arg_list)...);
|
||||
#endif
|
||||
return reinterpret_cast<MethodType>(const_cast<void*>(ptr))(this, arg_list...);
|
||||
}
|
||||
|
||||
template<FixedString ModuleName, FixedString Pattern, class ReturnType>
|
||||
ReturnType call_method(auto... arg_list)
|
||||
{
|
||||
static const auto* address = resolve_pattern(ModuleName, Pattern);
|
||||
return call_method<ReturnType>(address, arg_list...);
|
||||
}
|
||||
|
||||
template<FixedString ModuleName, FixedString Pattern, class ReturnType>
|
||||
ReturnType call_method(auto... arg_list) const
|
||||
{
|
||||
static const auto* address = resolve_pattern(ModuleName, Pattern);
|
||||
return call_method<ReturnType>(address, arg_list...);
|
||||
}
|
||||
|
||||
template<class ReturnType>
|
||||
ReturnType call_method(const std::string_view& module_name,const std::string_view& pattern, auto... arg_list)
|
||||
{
|
||||
static const auto* address = resolve_pattern(module_name, pattern);
|
||||
return call_method<ReturnType>(address, arg_list...);
|
||||
}
|
||||
|
||||
template<class ReturnType>
|
||||
ReturnType call_method(const std::string_view& module_name,const std::string_view& pattern, auto... arg_list) const
|
||||
{
|
||||
static const auto* address = resolve_pattern(module_name, pattern);
|
||||
return call_method<ReturnType>(address, arg_list...);
|
||||
}
|
||||
template<std::size_t Id, class ReturnType>
|
||||
ReturnType call_virtual_method(auto... arg_list)
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
using VirtualMethodType = ReturnType(__thiscall*)(void*, decltype(arg_list)...);
|
||||
#else
|
||||
using VirtualMethodType = ReturnType (*)(void*, decltype(arg_list)...);
|
||||
#endif
|
||||
return (*reinterpret_cast<VirtualMethodType**>(this))[id](this, arg_list...);
|
||||
const auto vtable = *reinterpret_cast<void***>(this);
|
||||
return call_method<ReturnType>(vtable[Id], arg_list...);
|
||||
}
|
||||
template<std::size_t id, class ReturnType>
|
||||
template<std::size_t Id, class ReturnType>
|
||||
ReturnType call_virtual_method(auto... arg_list) const
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
using VirtualMethodType = ReturnType(__thiscall*)(void*, decltype(arg_list)...);
|
||||
const auto vtable = *reinterpret_cast<void* const* const*>(this);
|
||||
return call_method<ReturnType>(vtable[Id], arg_list...);
|
||||
}
|
||||
|
||||
private:
|
||||
[[nodiscard]]
|
||||
static const void* resolve_pattern(const std::string_view module_name, const std::string_view pattern)
|
||||
{
|
||||
const auto* base = get_module_base(module_name);
|
||||
assert(base && "Failed to find module");
|
||||
|
||||
#ifdef _WIN32
|
||||
const auto result = PePatternScanner::scan_for_pattern_in_loaded_module(base, pattern);
|
||||
#elif defined(__APPLE__)
|
||||
const auto result = MachOPatternScanner::scan_for_pattern_in_loaded_module(base, pattern);
|
||||
#else
|
||||
using VirtualMethodType = ReturnType (*)(void*, decltype(arg_list)...);
|
||||
const auto result = ElfPatternScanner::scan_for_pattern_in_loaded_module(base, pattern);
|
||||
#endif
|
||||
assert(result.has_value() && "Pattern scan failed");
|
||||
return reinterpret_cast<const void*>(*result);
|
||||
}
|
||||
|
||||
[[nodiscard]]
|
||||
static const void* get_module_base(const std::string_view module_name)
|
||||
{
|
||||
#ifdef _WIN32
|
||||
return GetModuleHandleA(module_name.data());
|
||||
#elif defined(__APPLE__)
|
||||
// On macOS, iterate loaded images to find the module by name
|
||||
const auto count = _dyld_image_count();
|
||||
for (std::uint32_t i = 0; i < count; ++i)
|
||||
{
|
||||
const auto* name = _dyld_get_image_name(i);
|
||||
if (name && std::string_view{name}.find(module_name) != std::string_view::npos)
|
||||
return static_cast<const void*>(_dyld_get_image_header(i));
|
||||
}
|
||||
return nullptr;
|
||||
#else
|
||||
// On Linux, use dl_iterate_phdr to find loaded module by name
|
||||
struct CallbackData
|
||||
{
|
||||
std::string_view name;
|
||||
const void* base;
|
||||
} cb_data{module_name, nullptr};
|
||||
|
||||
dl_iterate_phdr(
|
||||
[](dl_phdr_info* info, std::size_t, void* data) -> int
|
||||
{
|
||||
auto* cb = static_cast<CallbackData*>(data);
|
||||
if (info->dlpi_name
|
||||
&& std::string_view{info->dlpi_name}.find(cb->name) != std::string_view::npos)
|
||||
{
|
||||
cb->base = reinterpret_cast<const void*>(info->dlpi_addr);
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
},
|
||||
&cb_data);
|
||||
return cb_data.base;
|
||||
#endif
|
||||
return (*static_cast<VirtualMethodType**>((void*)(this)))[id](
|
||||
const_cast<void*>(static_cast<const void*>(this)), arg_list...);
|
||||
}
|
||||
};
|
||||
} // namespace omath::rev_eng
|
||||
|
||||
@@ -16,15 +16,42 @@ echo "[*] Output dir: ${OUTPUT_DIR}"
|
||||
# Find llvm tools - handle versioned names (Linux) and xcrun (macOS)
|
||||
find_llvm_tool() {
|
||||
local tool_name="$1"
|
||||
|
||||
# macOS: use xcrun
|
||||
|
||||
# First priority: derive from the actual compiler used by cmake (CMakeCache.txt).
|
||||
# This guarantees the profraw format version matches the instrumented binary.
|
||||
local cache_file="${BINARY_DIR}/CMakeCache.txt"
|
||||
if [[ -f "$cache_file" ]]; then
|
||||
local cmake_cxx
|
||||
cmake_cxx=$(grep '^CMAKE_CXX_COMPILER:' "$cache_file" | cut -d= -f2)
|
||||
if [[ -n "$cmake_cxx" && -x "$cmake_cxx" ]]; then
|
||||
local tool_path
|
||||
tool_path="$(dirname "$cmake_cxx")/${tool_name}"
|
||||
if [[ -x "$tool_path" ]]; then
|
||||
echo "$tool_path"
|
||||
return 0
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
# macOS: derive from xcrun clang as fallback
|
||||
if [[ "$(uname)" == "Darwin" ]]; then
|
||||
local clang_path
|
||||
clang_path=$(xcrun --find clang 2>/dev/null)
|
||||
if [[ -n "$clang_path" ]]; then
|
||||
local tool_path
|
||||
tool_path="$(dirname "$clang_path")/${tool_name}"
|
||||
if [[ -x "$tool_path" ]]; then
|
||||
echo "$tool_path"
|
||||
return 0
|
||||
fi
|
||||
fi
|
||||
# Fallback: xcrun
|
||||
if xcrun --find "${tool_name}" &>/dev/null; then
|
||||
echo "xcrun ${tool_name}"
|
||||
return 0
|
||||
fi
|
||||
fi
|
||||
|
||||
|
||||
# Try versioned names (Linux with LLVM 21, 20, 19, etc.)
|
||||
for version in 21 20 19 18 17 ""; do
|
||||
local versioned_name="${tool_name}${version:+-$version}"
|
||||
@@ -33,7 +60,7 @@ find_llvm_tool() {
|
||||
return 0
|
||||
fi
|
||||
done
|
||||
|
||||
|
||||
echo ""
|
||||
return 1
|
||||
}
|
||||
@@ -51,6 +78,18 @@ fi
|
||||
echo "[*] Using: ${LLVM_PROFDATA}"
|
||||
echo "[*] Using: ${LLVM_COV}"
|
||||
|
||||
# Print version info for debugging version mismatches
|
||||
if [[ "$(uname)" == "Darwin" ]]; then
|
||||
echo "[*] Default clang: $(xcrun clang --version 2>&1 | head -1)"
|
||||
# Show actual compiler used by the build (from CMakeCache.txt if available)
|
||||
CACHE_FILE="${BINARY_DIR}/CMakeCache.txt"
|
||||
if [[ -f "$CACHE_FILE" ]]; then
|
||||
ACTUAL_CXX=$(grep '^CMAKE_CXX_COMPILER:' "$CACHE_FILE" | cut -d= -f2)
|
||||
echo "[*] Build compiler: ${ACTUAL_CXX} ($(${ACTUAL_CXX} --version 2>&1 | head -1))"
|
||||
fi
|
||||
echo "[*] profdata: $(${LLVM_PROFDATA} show --version 2>&1 | head -1 || true)"
|
||||
fi
|
||||
|
||||
# Find test binary
|
||||
if [[ -z "${TEST_BINARY}" ]]; then
|
||||
for path in \
|
||||
|
||||
@@ -3,6 +3,8 @@
|
||||
//
|
||||
#ifdef OMATH_ENABLE_LUA
|
||||
#include "omath/lua/lua.hpp"
|
||||
#include "omath/omath.hpp"
|
||||
#include "omath/projection/error_codes.hpp"
|
||||
#include <omath/engines/cry_engine/camera.hpp>
|
||||
#include <omath/engines/frostbite_engine/camera.hpp>
|
||||
#include <omath/engines/iw_engine/camera.hpp>
|
||||
@@ -33,6 +35,8 @@ namespace
|
||||
return "world position is out of screen bounds";
|
||||
case omath::projection::Error::INV_VIEW_PROJ_MAT_DET_EQ_ZERO:
|
||||
return "inverse view-projection matrix determinant is zero";
|
||||
case omath::projection::Error::PERSPECTIVE_DIVIDER_LESS_EQ_ZERO:
|
||||
return "perspective divider is less or equal to zero";
|
||||
}
|
||||
return "unknown error";
|
||||
}
|
||||
|
||||
@@ -21,7 +21,7 @@ namespace omath::projectile_prediction
|
||||
const float bullet_gravity = m_gravity_constant * projectile.m_gravity_scale;
|
||||
const float v0 = projectile.m_launch_speed;
|
||||
const float v0_sqr = v0 * v0;
|
||||
const Vector3 proj_origin = projectile.m_origin;
|
||||
const Vector3 proj_origin = projectile.m_origin + projectile.m_launch_offset;
|
||||
|
||||
constexpr int SIMD_FACTOR = 8;
|
||||
float current_time = m_simulation_time_step;
|
||||
@@ -124,6 +124,110 @@ namespace omath::projectile_prediction
|
||||
std::format("{} AVX2 feature is not enabled!", std::source_location::current().function_name()));
|
||||
#endif
|
||||
}
|
||||
std::optional<AimAngles>
|
||||
ProjPredEngineAvx2::maybe_calculate_aim_angles([[maybe_unused]] const Projectile& projectile,
|
||||
[[maybe_unused]] const Target& target) const
|
||||
{
|
||||
#if defined(OMATH_USE_AVX2) && defined(__i386__) && defined(__x86_64__)
|
||||
const float bullet_gravity = m_gravity_constant * projectile.m_gravity_scale;
|
||||
const float v0 = projectile.m_launch_speed;
|
||||
const Vector3 proj_origin = projectile.m_origin + projectile.m_launch_offset;
|
||||
|
||||
constexpr int SIMD_FACTOR = 8;
|
||||
float current_time = m_simulation_time_step;
|
||||
|
||||
for (; current_time <= m_maximum_simulation_time; current_time += m_simulation_time_step * SIMD_FACTOR)
|
||||
{
|
||||
const __m256 times
|
||||
= _mm256_setr_ps(current_time, current_time + m_simulation_time_step,
|
||||
current_time + m_simulation_time_step * 2, current_time + m_simulation_time_step * 3,
|
||||
current_time + m_simulation_time_step * 4, current_time + m_simulation_time_step * 5,
|
||||
current_time + m_simulation_time_step * 6, current_time + m_simulation_time_step * 7);
|
||||
|
||||
const __m256 target_x
|
||||
= _mm256_fmadd_ps(_mm256_set1_ps(target.m_velocity.x), times, _mm256_set1_ps(target.m_origin.x));
|
||||
const __m256 target_y
|
||||
= _mm256_fmadd_ps(_mm256_set1_ps(target.m_velocity.y), times, _mm256_set1_ps(target.m_origin.y));
|
||||
const __m256 times_sq = _mm256_mul_ps(times, times);
|
||||
const __m256 target_z = _mm256_fmadd_ps(_mm256_set1_ps(target.m_velocity.z), times,
|
||||
_mm256_fnmadd_ps(_mm256_set1_ps(0.5f * m_gravity_constant), times_sq,
|
||||
_mm256_set1_ps(target.m_origin.z)));
|
||||
|
||||
const __m256 delta_x = _mm256_sub_ps(target_x, _mm256_set1_ps(proj_origin.x));
|
||||
const __m256 delta_y = _mm256_sub_ps(target_y, _mm256_set1_ps(proj_origin.y));
|
||||
|
||||
const __m256 d_sqr = _mm256_add_ps(_mm256_mul_ps(delta_x, delta_x), _mm256_mul_ps(delta_y, delta_y));
|
||||
const __m256 delta_z = _mm256_sub_ps(target_z, _mm256_set1_ps(proj_origin.z));
|
||||
|
||||
const __m256 bg_times_sq = _mm256_mul_ps(_mm256_set1_ps(bullet_gravity), times_sq);
|
||||
const __m256 term = _mm256_add_ps(delta_z, _mm256_mul_ps(_mm256_set1_ps(0.5f), bg_times_sq));
|
||||
const __m256 term_sq = _mm256_mul_ps(term, term);
|
||||
const __m256 numerator = _mm256_add_ps(d_sqr, term_sq);
|
||||
const __m256 denominator = _mm256_add_ps(times_sq, _mm256_set1_ps(1e-8f));
|
||||
const __m256 required_v0_sqr = _mm256_div_ps(numerator, denominator);
|
||||
|
||||
const __m256 v0_sqr_vec = _mm256_set1_ps(v0 * v0 + 1e-3f);
|
||||
const __m256 mask = _mm256_cmp_ps(required_v0_sqr, v0_sqr_vec, _CMP_LE_OQ);
|
||||
|
||||
const unsigned valid_mask = _mm256_movemask_ps(mask);
|
||||
if (!valid_mask)
|
||||
continue;
|
||||
|
||||
alignas(32) float valid_times[SIMD_FACTOR];
|
||||
_mm256_store_ps(valid_times, times);
|
||||
|
||||
for (int i = 0; i < SIMD_FACTOR; ++i)
|
||||
{
|
||||
if (!(valid_mask & (1 << i)))
|
||||
continue;
|
||||
|
||||
const float candidate_time = valid_times[i];
|
||||
if (candidate_time > m_maximum_simulation_time)
|
||||
continue;
|
||||
|
||||
for (float fine_time = candidate_time - m_simulation_time_step * 2;
|
||||
fine_time <= candidate_time + m_simulation_time_step * 2; fine_time += m_simulation_time_step)
|
||||
{
|
||||
if (fine_time < 0)
|
||||
continue;
|
||||
|
||||
Vector3 target_pos = target.m_origin + target.m_velocity * fine_time;
|
||||
if (target.m_is_airborne)
|
||||
target_pos.z -= 0.5f * m_gravity_constant * fine_time * fine_time;
|
||||
|
||||
const auto pitch = calculate_pitch(proj_origin, target_pos, bullet_gravity, v0, fine_time);
|
||||
if (!pitch)
|
||||
continue;
|
||||
|
||||
const Vector3 delta = target_pos - projectile.m_origin;
|
||||
const float yaw = angles::radians_to_degrees(std::atan2(delta.y, delta.x));
|
||||
return AimAngles{*pitch, yaw};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (; current_time <= m_maximum_simulation_time; current_time += m_simulation_time_step)
|
||||
{
|
||||
Vector3 target_pos = target.m_origin + target.m_velocity * current_time;
|
||||
if (target.m_is_airborne)
|
||||
target_pos.z -= 0.5f * m_gravity_constant * current_time * current_time;
|
||||
|
||||
const auto pitch = calculate_pitch(proj_origin, target_pos, bullet_gravity, v0, current_time);
|
||||
if (!pitch)
|
||||
continue;
|
||||
|
||||
const Vector3 delta = target_pos - projectile.m_origin;
|
||||
const float yaw = angles::radians_to_degrees(std::atan2(delta.y, delta.x));
|
||||
return AimAngles{*pitch, yaw};
|
||||
}
|
||||
|
||||
return std::nullopt;
|
||||
#else
|
||||
throw std::runtime_error(
|
||||
std::format("{} AVX2 feature is not enabled!", std::source_location::current().function_name()));
|
||||
#endif
|
||||
}
|
||||
|
||||
ProjPredEngineAvx2::ProjPredEngineAvx2(const float gravity_constant, const float simulation_time_step,
|
||||
const float maximum_simulation_time)
|
||||
: m_gravity_constant(gravity_constant), m_simulation_time_step(simulation_time_step),
|
||||
|
||||
@@ -20,6 +20,8 @@
|
||||
#include <omath/engines/unreal_engine/traits/mesh_trait.hpp>
|
||||
#include <omath/engines/unreal_engine/traits/camera_trait.hpp>
|
||||
|
||||
#include <omath/engines/source_engine/traits/pred_engine_trait.hpp>
|
||||
|
||||
#include <omath/projectile_prediction/projectile.hpp>
|
||||
#include <omath/projectile_prediction/target.hpp>
|
||||
#include <optional>
|
||||
@@ -35,6 +37,132 @@ static void expect_matrix_near(const MatT& a, const MatT& b, float eps = 1e-5f)
|
||||
EXPECT_NEAR(a.at(r, c), b.at(r, c), eps);
|
||||
}
|
||||
|
||||
// ── Launch offset tests for all engines ──────────────────────────────────────
|
||||
#include <omath/engines/cry_engine/traits/pred_engine_trait.hpp>
|
||||
|
||||
// Helper: verify that zero offset matches default-initialized offset behavior
|
||||
template<typename Trait>
|
||||
static void verify_launch_offset_at_time_zero(const Vector3<float>& origin, const Vector3<float>& offset)
|
||||
{
|
||||
projectile_prediction::Projectile p;
|
||||
p.m_origin = origin;
|
||||
p.m_launch_offset = offset;
|
||||
p.m_launch_speed = 100.f;
|
||||
p.m_gravity_scale = 1.f;
|
||||
|
||||
const auto pos = Trait::predict_projectile_position(p, 0.f, 0.f, 0.f, 9.81f);
|
||||
const auto expected = origin + offset;
|
||||
EXPECT_NEAR(pos.x, expected.x, 1e-4f);
|
||||
EXPECT_NEAR(pos.y, expected.y, 1e-4f);
|
||||
EXPECT_NEAR(pos.z, expected.z, 1e-4f);
|
||||
}
|
||||
|
||||
template<typename Trait>
|
||||
static void verify_zero_offset_matches_default()
|
||||
{
|
||||
projectile_prediction::Projectile p;
|
||||
p.m_origin = {10.f, 20.f, 30.f};
|
||||
p.m_launch_offset = {0.f, 0.f, 0.f};
|
||||
p.m_launch_speed = 50.f;
|
||||
p.m_gravity_scale = 1.f;
|
||||
|
||||
projectile_prediction::Projectile p2;
|
||||
p2.m_origin = {10.f, 20.f, 30.f};
|
||||
p2.m_launch_speed = 50.f;
|
||||
p2.m_gravity_scale = 1.f;
|
||||
|
||||
const auto pos1 = Trait::predict_projectile_position(p, 15.f, 30.f, 1.f, 9.81f);
|
||||
const auto pos2 = Trait::predict_projectile_position(p2, 15.f, 30.f, 1.f, 9.81f);
|
||||
#if defined(__x86_64__) || defined(_M_X64) || defined(__aarch64__) || defined(_M_ARM64)
|
||||
constexpr float tol = 1e-6f;
|
||||
#else
|
||||
constexpr float tol = 1e-4f;
|
||||
#endif
|
||||
EXPECT_NEAR(pos1.x, pos2.x, tol);
|
||||
EXPECT_NEAR(pos1.y, pos2.y, tol);
|
||||
EXPECT_NEAR(pos1.z, pos2.z, tol);
|
||||
}
|
||||
|
||||
TEST(LaunchOffsetTests, Source_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<source_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, Source_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<source_engine::PredEngineTrait>();
|
||||
}
|
||||
TEST(LaunchOffsetTests, Frostbite_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<frostbite_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, Frostbite_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<frostbite_engine::PredEngineTrait>();
|
||||
}
|
||||
TEST(LaunchOffsetTests, IW_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<iw_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, IW_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<iw_engine::PredEngineTrait>();
|
||||
}
|
||||
TEST(LaunchOffsetTests, OpenGL_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<opengl_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, OpenGL_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<opengl_engine::PredEngineTrait>();
|
||||
}
|
||||
TEST(LaunchOffsetTests, Unity_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<unity_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, Unity_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<unity_engine::PredEngineTrait>();
|
||||
}
|
||||
TEST(LaunchOffsetTests, Unreal_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<unreal_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, Unreal_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<unreal_engine::PredEngineTrait>();
|
||||
}
|
||||
TEST(LaunchOffsetTests, CryEngine_OffsetAtTimeZero)
|
||||
{
|
||||
verify_launch_offset_at_time_zero<cry_engine::PredEngineTrait>({0, 0, 0}, {5, 3, -2});
|
||||
}
|
||||
TEST(LaunchOffsetTests, CryEngine_ZeroOffsetMatchesDefault)
|
||||
{
|
||||
verify_zero_offset_matches_default<cry_engine::PredEngineTrait>();
|
||||
}
|
||||
|
||||
// Test that offset shifts the projectile position at t>0 as well
|
||||
TEST(LaunchOffsetTests, OffsetShiftsTrajectory)
|
||||
{
|
||||
projectile_prediction::Projectile p_no_offset;
|
||||
p_no_offset.m_origin = {0.f, 0.f, 0.f};
|
||||
p_no_offset.m_launch_speed = 100.f;
|
||||
p_no_offset.m_gravity_scale = 1.f;
|
||||
|
||||
projectile_prediction::Projectile p_with_offset;
|
||||
p_with_offset.m_origin = {0.f, 0.f, 0.f};
|
||||
p_with_offset.m_launch_offset = {10.f, 5.f, -3.f};
|
||||
p_with_offset.m_launch_speed = 100.f;
|
||||
p_with_offset.m_gravity_scale = 1.f;
|
||||
|
||||
const auto pos1 = source_engine::PredEngineTrait::predict_projectile_position(p_no_offset, 20.f, 45.f, 2.f, 9.81f);
|
||||
const auto pos2 = source_engine::PredEngineTrait::predict_projectile_position(p_with_offset, 20.f, 45.f, 2.f, 9.81f);
|
||||
|
||||
// The difference should be exactly the launch offset
|
||||
EXPECT_NEAR(pos2.x - pos1.x, 10.f, 1e-4f);
|
||||
EXPECT_NEAR(pos2.y - pos1.y, 5.f, 1e-4f);
|
||||
EXPECT_NEAR(pos2.z - pos1.z, -3.f, 1e-4f);
|
||||
}
|
||||
|
||||
// Generic tests for PredEngineTrait behaviour across engines
|
||||
TEST(TraitTests, Frostbite_Pred_And_Mesh_And_Camera)
|
||||
{
|
||||
|
||||
@@ -53,6 +53,47 @@ TEST(PredEngineTrait, CalcViewpointFromAngles)
|
||||
EXPECT_NEAR(vp.z, 10.f, 1e-6f);
|
||||
}
|
||||
|
||||
TEST(PredEngineTrait, PredictProjectilePositionWithLaunchOffset)
|
||||
{
|
||||
projectile_prediction::Projectile p;
|
||||
p.m_origin = {0.f, 0.f, 0.f};
|
||||
p.m_launch_offset = {5.f, 3.f, -2.f};
|
||||
p.m_launch_speed = 10.f;
|
||||
p.m_gravity_scale = 1.f;
|
||||
|
||||
// At time=0, projectile should be at launch_pos = origin + offset
|
||||
const auto pos_t0 = PredEngineTrait::predict_projectile_position(p, 0.f, 0.f, 0.f, 9.81f);
|
||||
EXPECT_NEAR(pos_t0.x, 5.f, 1e-4f);
|
||||
EXPECT_NEAR(pos_t0.y, 3.f, 1e-4f);
|
||||
EXPECT_NEAR(pos_t0.z, -2.f, 1e-4f);
|
||||
|
||||
// At time=1 with zero pitch/yaw, should travel along X from the offset position
|
||||
const auto pos_t1 = PredEngineTrait::predict_projectile_position(p, 0.f, 0.f, 1.f, 9.81f);
|
||||
EXPECT_NEAR(pos_t1.x, 5.f + 10.f, 1e-3f);
|
||||
EXPECT_NEAR(pos_t1.y, 3.f, 1e-3f);
|
||||
EXPECT_NEAR(pos_t1.z, -2.f - 9.81f * 0.5f, 1e-3f);
|
||||
}
|
||||
|
||||
TEST(PredEngineTrait, ZeroLaunchOffsetMatchesOriginalBehavior)
|
||||
{
|
||||
projectile_prediction::Projectile p;
|
||||
p.m_origin = {10.f, 20.f, 30.f};
|
||||
p.m_launch_offset = {0.f, 0.f, 0.f};
|
||||
p.m_launch_speed = 15.f;
|
||||
p.m_gravity_scale = 0.5f;
|
||||
|
||||
projectile_prediction::Projectile p_no_offset;
|
||||
p_no_offset.m_origin = {10.f, 20.f, 30.f};
|
||||
p_no_offset.m_launch_speed = 15.f;
|
||||
p_no_offset.m_gravity_scale = 0.5f;
|
||||
|
||||
const auto pos1 = PredEngineTrait::predict_projectile_position(p, 30.f, 45.f, 2.f, 9.81f);
|
||||
const auto pos2 = PredEngineTrait::predict_projectile_position(p_no_offset, 30.f, 45.f, 2.f, 9.81f);
|
||||
EXPECT_NEAR(pos1.x, pos2.x, 1e-6f);
|
||||
EXPECT_NEAR(pos1.y, pos2.y, 1e-6f);
|
||||
EXPECT_NEAR(pos1.z, pos2.z, 1e-6f);
|
||||
}
|
||||
|
||||
TEST(PredEngineTrait, DirectAngles)
|
||||
{
|
||||
constexpr Vector3<float> origin{0.f, 0.f, 0.f};
|
||||
|
||||
@@ -16,3 +16,280 @@ TEST(UnitTestPrediction, PredictionTest)
|
||||
EXPECT_NEAR(-42.547142, pitch.as_degrees(), 0.01f);
|
||||
EXPECT_NEAR(-1.181189, yaw.as_degrees(), 0.01f);
|
||||
}
|
||||
|
||||
// Helper: verify aim_angles match angles derived from aim_point via CameraTrait
|
||||
static void expect_angles_match_aim_point(const omath::projectile_prediction::Projectile& proj,
|
||||
const omath::projectile_prediction::Target& target,
|
||||
float gravity, float step, float max_time, float tolerance,
|
||||
float angle_eps = 0.01f)
|
||||
{
|
||||
const omath::projectile_prediction::ProjPredEngineLegacy engine(gravity, step, max_time, tolerance);
|
||||
|
||||
const auto aim_point = engine.maybe_calculate_aim_point(proj, target);
|
||||
const auto aim_angles = engine.maybe_calculate_aim_angles(proj, target);
|
||||
|
||||
ASSERT_TRUE(aim_point.has_value()) << "aim_point should have a solution";
|
||||
ASSERT_TRUE(aim_angles.has_value()) << "aim_angles should have a solution";
|
||||
|
||||
// Source engine CameraTrait: pitch = -asin(dir.z), yaw = atan2(dir.y, dir.x)
|
||||
// PredEngineTrait: pitch = asin(delta.z / dist), yaw = atan2(delta.y, delta.x)
|
||||
// So aim_angles.pitch == -camera_pitch, aim_angles.yaw == camera_yaw
|
||||
const auto [cam_pitch, cam_yaw, cam_roll] =
|
||||
omath::source_engine::CameraTrait::calc_look_at_angle(proj.m_origin, aim_point.value());
|
||||
|
||||
EXPECT_NEAR(aim_angles->pitch, -cam_pitch.as_degrees(), angle_eps)
|
||||
<< "pitch from aim_angles must match pitch derived from aim_point";
|
||||
EXPECT_NEAR(aim_angles->yaw, cam_yaw.as_degrees(), angle_eps)
|
||||
<< "yaw from aim_angles must match yaw derived from aim_point";
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesMatchAimPoint_StaticTarget)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {100, 0, 90}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {3, 2, 1}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_angles_match_aim_point(proj, target, 400, 1.f / 1000.f, 50, 5.f);
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesMatchAimPoint_MovingTarget)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {500, 100, 0}, .m_velocity = {-50, 20, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 3000, .m_gravity_scale = 1.0};
|
||||
|
||||
expect_angles_match_aim_point(proj, target, 800, 1.f / 500.f, 30, 10.f);
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesMatchAimPoint_AirborneTarget)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {200, 50, 300}, .m_velocity = {10, -5, -20}, .m_is_airborne = true};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 4000, .m_gravity_scale = 0.5};
|
||||
|
||||
expect_angles_match_aim_point(proj, target, 400, 1.f / 1000.f, 50, 10.f);
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesMatchAimPoint_HighArc)
|
||||
{
|
||||
// Target nearly directly above — high pitch angle
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {10, 0, 500}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 5000, .m_gravity_scale = 0.3};
|
||||
|
||||
expect_angles_match_aim_point(proj, target, 400, 1.f / 1000.f, 50, 5.f);
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesMatchAimPoint_NegativeYaw)
|
||||
{
|
||||
// Target behind and to the left — negative yaw quadrant
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {-200, -150, 10}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_angles_match_aim_point(proj, target, 400, 1.f / 1000.f, 50, 5.f);
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesMatchAimPoint_WithLaunchOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {200, 0, 50}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {5, 0, -3}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_angles_match_aim_point(proj, target, 400, 1.f / 1000.f, 50, 5.f);
|
||||
}
|
||||
|
||||
// Helper: simulate projectile flight using aim_angles and verify it reaches the target.
|
||||
// Steps the projectile forward in small increments, simultaneously predicts target position,
|
||||
// and checks that the minimum distance is within hit_tolerance.
|
||||
static void expect_projectile_hits_target(const omath::projectile_prediction::Projectile& proj,
|
||||
const omath::projectile_prediction::Target& target,
|
||||
float gravity, float engine_step, float max_time, float engine_tolerance,
|
||||
float hit_tolerance, float sim_step = 1.f / 2000.f)
|
||||
{
|
||||
using Trait = omath::source_engine::PredEngineTrait;
|
||||
const omath::projectile_prediction::ProjPredEngineLegacy engine(gravity, engine_step, max_time, engine_tolerance);
|
||||
|
||||
const auto aim_angles = engine.maybe_calculate_aim_angles(proj, target);
|
||||
ASSERT_TRUE(aim_angles.has_value()) << "engine must find a solution";
|
||||
|
||||
float min_dist = std::numeric_limits<float>::max();
|
||||
float best_time = 0.f;
|
||||
|
||||
for (float t = 0.f; t <= max_time; t += sim_step)
|
||||
{
|
||||
const auto proj_pos = Trait::predict_projectile_position(proj, aim_angles->pitch, aim_angles->yaw, t, gravity);
|
||||
const auto tgt_pos = Trait::predict_target_position(target, t, gravity);
|
||||
const float dist = proj_pos.distance_to(tgt_pos);
|
||||
|
||||
if (dist < min_dist)
|
||||
{
|
||||
min_dist = dist;
|
||||
best_time = t;
|
||||
}
|
||||
|
||||
// Early exit once distance starts increasing significantly after approaching
|
||||
if (dist > min_dist + hit_tolerance * 10.f && min_dist < hit_tolerance * 100.f)
|
||||
break;
|
||||
}
|
||||
|
||||
EXPECT_LE(min_dist, hit_tolerance)
|
||||
<< "Projectile must reach target. Closest approach: " << min_dist
|
||||
<< " at t=" << best_time;
|
||||
}
|
||||
|
||||
// ── Simulation hit tests: no launch offset ─────────────────────────────────
|
||||
|
||||
TEST(ProjectileSimulation, HitsStaticTarget_NoOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {100, 0, 90}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {3, 2, 1}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsMovingTarget_NoOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {500, 100, 0}, .m_velocity = {-50, 20, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 3000, .m_gravity_scale = 1.0};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 800, 1.f / 500.f, 30, 10.f, 15.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsAirborneTarget_NoOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {200, 50, 300}, .m_velocity = {10, -5, -20}, .m_is_airborne = true};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 4000, .m_gravity_scale = 0.5};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 10.f, 15.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsHighTarget_NoOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {10, 0, 500}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 5000, .m_gravity_scale = 0.3};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsNegativeYawTarget_NoOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {-200, -150, 10}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
// ── Simulation hit tests: with launch offset ────────────────────────────────
|
||||
|
||||
TEST(ProjectileSimulation, HitsStaticTarget_SmallOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {200, 0, 50}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {5, 0, -3}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsStaticTarget_LargeXOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {300, 100, 0}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {20, 0, 0}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsStaticTarget_LargeYOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {150, -200, 30}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {0, 15, 0}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsStaticTarget_LargeZOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {100, 0, 200}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {0, 0, -10}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsStaticTarget_AllAxesOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {250, 80, 60}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {10, 5, 20}, .m_launch_offset = {8, -4, -6}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsMovingTarget_WithOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {400, 0, 50}, .m_velocity = {-30, 10, 5}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {10, -5, 2}, .m_launch_speed = 3000, .m_gravity_scale = 0.8};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 800, 1.f / 500.f, 30, 10.f, 15.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsAirborneTarget_WithOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {150, 80, 250}, .m_velocity = {5, -10, -30}, .m_is_airborne = true};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 50}, .m_launch_offset = {3, 7, -5}, .m_launch_speed = 4000, .m_gravity_scale = 0.5};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 10.f, 15.f);
|
||||
}
|
||||
|
||||
TEST(ProjectileSimulation, HitsNegativeYawTarget_WithOffset)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {-200, -150, 10}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
const omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_offset = {-5, 3, 2}, .m_launch_speed = 5000, .m_gravity_scale = 0.4};
|
||||
|
||||
expect_projectile_hits_target(proj, target, 400, 1.f / 1000.f, 50, 5.f, 10.f);
|
||||
}
|
||||
|
||||
TEST(UnitTestPrediction, AimAnglesReturnsNulloptWhenNoSolution)
|
||||
{
|
||||
constexpr omath::projectile_prediction::Target target{
|
||||
.m_origin = {100000, 0, 0}, .m_velocity = {0, 0, 0}, .m_is_airborne = false};
|
||||
constexpr omath::projectile_prediction::Projectile proj = {
|
||||
.m_origin = {0, 0, 0}, .m_launch_speed = 1, .m_gravity_scale = 1};
|
||||
|
||||
const omath::projectile_prediction::ProjPredEngineLegacy engine(9.81f, 0.1f, 2.f, 5.f);
|
||||
|
||||
const auto aim_point = engine.maybe_calculate_aim_point(proj, target);
|
||||
const auto aim_angles = engine.maybe_calculate_aim_angles(proj, target);
|
||||
|
||||
EXPECT_FALSE(aim_point.has_value());
|
||||
EXPECT_FALSE(aim_angles.has_value());
|
||||
}
|
||||
|
||||
@@ -46,6 +46,22 @@ TEST(ProjPredLegacyMore, ZeroGravityUsesDirectPitchAndReturnsViewpoint)
|
||||
EXPECT_NEAR(v.z, 3.f, 1e-6f);
|
||||
}
|
||||
|
||||
TEST(ProjPredLegacyMore, ZeroGravityAimAnglesReturnsPitchAndYaw)
|
||||
{
|
||||
constexpr Projectile proj{ .m_origin = {0.f, 0.f, 0.f}, .m_launch_speed = 10.f, .m_gravity_scale = 0.f };
|
||||
constexpr Target target{ .m_origin = {100.f, 0.f, 0.f}, .m_velocity = {0.f,0.f,0.f}, .m_is_airborne = false };
|
||||
|
||||
using Engine = omath::projectile_prediction::ProjPredEngineLegacy<FakeEngineZeroGravity>;
|
||||
const Engine engine(9.8f, 0.1f, 5.f, 1e-3f);
|
||||
|
||||
const auto res = engine.maybe_calculate_aim_angles(proj, target);
|
||||
ASSERT_TRUE(res.has_value());
|
||||
// FakeEngineZeroGravity::calc_direct_pitch_angle returns 12.5f
|
||||
EXPECT_NEAR(res->pitch, 12.5f, 1e-6f);
|
||||
// FakeEngineZeroGravity::calc_direct_yaw_angle returns 0.f
|
||||
EXPECT_NEAR(res->yaw, 0.f, 1e-6f);
|
||||
}
|
||||
|
||||
// Fake trait producing no valid launch angle (root < 0)
|
||||
struct FakeEngineNoSolution
|
||||
{
|
||||
@@ -69,6 +85,9 @@ TEST(ProjPredLegacyMore, NoSolutionRootReturnsNullopt)
|
||||
|
||||
const auto res = engine.maybe_calculate_aim_point(proj, target);
|
||||
EXPECT_FALSE(res.has_value());
|
||||
|
||||
const auto angles_res = engine.maybe_calculate_aim_angles(proj, target);
|
||||
EXPECT_FALSE(angles_res.has_value());
|
||||
}
|
||||
|
||||
// Fake trait where an angle exists but the projectile does not reach target (miss)
|
||||
|
||||
@@ -50,6 +50,126 @@ TEST(UnitTestProjection, ScreenToNdcBottomLeft)
|
||||
EXPECT_NEAR(ndc_bottom_left.y, 0.519615293f, 0.0001f);
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenInBounds)
|
||||
{
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
|
||||
const auto projected = cam.world_to_screen_unclipped({1000.f, 0, 50.f});
|
||||
ASSERT_TRUE(projected.has_value());
|
||||
EXPECT_NEAR(projected->x, 960.f, 0.001f);
|
||||
EXPECT_NEAR(projected->y, 504.f, 0.001f);
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenMatchesWorldToScreenWhenInBounds)
|
||||
{
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
|
||||
const auto w2s = cam.world_to_screen({1000.f, 0, 50.f});
|
||||
const auto no_clip = cam.world_to_screen_unclipped({1000.f, 0, 50.f});
|
||||
|
||||
ASSERT_TRUE(w2s.has_value());
|
||||
ASSERT_TRUE(no_clip.has_value());
|
||||
EXPECT_NEAR(w2s->x, no_clip->x, 0.001f);
|
||||
EXPECT_NEAR(w2s->y, no_clip->y, 0.001f);
|
||||
EXPECT_NEAR(w2s->z, no_clip->z, 0.001f);
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenRejectsBehindCamera)
|
||||
{
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
|
||||
const auto projected = cam.world_to_screen_unclipped({-1000.f, 0, 0});
|
||||
EXPECT_FALSE(projected.has_value());
|
||||
EXPECT_EQ(projected.error(), omath::projection::Error::PERSPECTIVE_DIVIDER_LESS_EQ_ZERO);
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenAllowsOutOfBoundsNdc)
|
||||
{
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
|
||||
// Point far to the side exceeds NDC [-1,1] bounds but unclipped returns it anyway
|
||||
const auto projected = cam.world_to_screen_unclipped({100.f, 5000.f, 0});
|
||||
EXPECT_TRUE(projected.has_value());
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, WorldToScreenRejectsOutOfBoundsNdc)
|
||||
{
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
|
||||
// Same point that unclipped allows — clipped world_to_screen rejects it
|
||||
const auto projected = cam.world_to_screen({100.f, 5000.f, 0});
|
||||
EXPECT_FALSE(projected.has_value());
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenBottomLeftCorner)
|
||||
{
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
using ScreenStart = omath::source_engine::Camera::ScreenStart;
|
||||
|
||||
const auto top_left = cam.world_to_screen_unclipped<ScreenStart::TOP_LEFT_CORNER>({1000.f, 0, 50.f});
|
||||
const auto bottom_left = cam.world_to_screen_unclipped<ScreenStart::BOTTOM_LEFT_CORNER>({1000.f, 0, 50.f});
|
||||
|
||||
ASSERT_TRUE(top_left.has_value());
|
||||
ASSERT_TRUE(bottom_left.has_value());
|
||||
// X should be identical, Y should differ (mirrored around center)
|
||||
EXPECT_NEAR(top_left->x, bottom_left->x, 0.001f);
|
||||
EXPECT_NEAR(top_left->y + bottom_left->y, 1080.f, 0.001f);
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenRoundTrip)
|
||||
{
|
||||
std::mt19937 gen(42);
|
||||
std::uniform_real_distribution dist_fwd(100.f, 900.f);
|
||||
std::uniform_real_distribution dist_side(-400.f, 400.f);
|
||||
std::uniform_real_distribution dist_up(-200.f, 200.f);
|
||||
|
||||
constexpr auto fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
const auto cam = omath::source_engine::Camera({0, 0, 0}, omath::source_engine::ViewAngles{}, {1920.f, 1080.f}, fov,
|
||||
0.01f, 1000.f);
|
||||
|
||||
for (int i = 0; i < 100; i++)
|
||||
{
|
||||
const omath::Vector3<float> world_pos{dist_fwd(gen), dist_side(gen), dist_up(gen)};
|
||||
const auto screen = cam.world_to_screen_unclipped(world_pos);
|
||||
if (!screen.has_value())
|
||||
continue;
|
||||
|
||||
const auto back_to_world = cam.screen_to_world(screen.value());
|
||||
ASSERT_TRUE(back_to_world.has_value());
|
||||
|
||||
const auto back_to_screen = cam.world_to_screen_unclipped(back_to_world.value());
|
||||
ASSERT_TRUE(back_to_screen.has_value());
|
||||
|
||||
EXPECT_NEAR(screen->x, back_to_screen->x, 0.01f);
|
||||
EXPECT_NEAR(screen->y, back_to_screen->y, 0.01f);
|
||||
}
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, UnclippedWorldToScreenUnityEngine)
|
||||
{
|
||||
constexpr auto fov = omath::projection::FieldOfView::from_degrees(60.f);
|
||||
const auto cam = omath::unity_engine::Camera({0, 0, 0}, {}, {1280.f, 720.f}, fov, 0.03f, 1000.f);
|
||||
using ScreenStart = omath::unity_engine::Camera::ScreenStart;
|
||||
|
||||
// Point directly in front
|
||||
const auto projected = cam.world_to_screen_unclipped<ScreenStart::BOTTOM_LEFT_CORNER>({0, 0, 500.f});
|
||||
ASSERT_TRUE(projected.has_value());
|
||||
EXPECT_NEAR(projected->x, 640.f, 0.5f);
|
||||
EXPECT_NEAR(projected->y, 360.f, 0.5f);
|
||||
}
|
||||
|
||||
TEST(UnitTestProjection, ScreenToWorldTopLeftCorner)
|
||||
{
|
||||
std::mt19937 gen(std::random_device{}()); // Seed with a non-deterministic source
|
||||
|
||||
@@ -20,6 +20,13 @@ public:
|
||||
int m_health{123};
|
||||
};
|
||||
|
||||
// Extract a raw function pointer from an object's vtable
|
||||
inline const void* get_vtable_entry(const void* obj, const std::size_t index)
|
||||
{
|
||||
const auto vtable = *static_cast<void* const* const*>(obj);
|
||||
return vtable[index];
|
||||
}
|
||||
|
||||
class RevPlayer final : omath::rev_eng::InternalReverseEngineeredObject
|
||||
{
|
||||
public:
|
||||
@@ -51,6 +58,17 @@ public:
|
||||
{
|
||||
return call_virtual_method<1, int>();
|
||||
}
|
||||
|
||||
// Wrappers exposing call_method for testing — use vtable entries as known-good function pointers
|
||||
int call_foo_via_ptr(const void* fn_ptr) const
|
||||
{
|
||||
return call_method<int>(fn_ptr);
|
||||
}
|
||||
|
||||
int call_bar_via_ptr(const void* fn_ptr) const
|
||||
{
|
||||
return call_method<int>(fn_ptr);
|
||||
}
|
||||
};
|
||||
|
||||
TEST(unit_test_reverse_enineering, read_test)
|
||||
@@ -64,4 +82,39 @@ TEST(unit_test_reverse_enineering, read_test)
|
||||
EXPECT_EQ(player_original.bar(), player_reversed->rev_bar());
|
||||
EXPECT_EQ(player_original.foo(), player_reversed->rev_foo());
|
||||
EXPECT_EQ(player_original.bar(), player_reversed->rev_bar_const());
|
||||
}
|
||||
|
||||
TEST(unit_test_reverse_enineering, call_method_with_vtable_ptr)
|
||||
{
|
||||
// Extract raw function pointers from Player's vtable, then call them via call_method
|
||||
Player player;
|
||||
const auto* rev = reinterpret_cast<const RevPlayer*>(&player);
|
||||
|
||||
const auto* foo_ptr = get_vtable_entry(&player, 0);
|
||||
const auto* bar_ptr = get_vtable_entry(&player, 1);
|
||||
|
||||
EXPECT_EQ(player.foo(), rev->call_foo_via_ptr(foo_ptr));
|
||||
EXPECT_EQ(player.bar(), rev->call_bar_via_ptr(bar_ptr));
|
||||
EXPECT_EQ(1, rev->call_foo_via_ptr(foo_ptr));
|
||||
EXPECT_EQ(2, rev->call_bar_via_ptr(bar_ptr));
|
||||
}
|
||||
|
||||
TEST(unit_test_reverse_enineering, call_method_same_result_as_virtual)
|
||||
{
|
||||
// call_virtual_method delegates to call_method — both paths must agree
|
||||
Player player;
|
||||
const auto* rev = reinterpret_cast<const RevPlayer*>(&player);
|
||||
|
||||
EXPECT_EQ(rev->rev_foo(), rev->call_foo_via_ptr(get_vtable_entry(&player, 0)));
|
||||
EXPECT_EQ(rev->rev_bar(), rev->call_bar_via_ptr(get_vtable_entry(&player, 1)));
|
||||
}
|
||||
|
||||
TEST(unit_test_reverse_enineering, call_virtual_method_delegates_to_call_method)
|
||||
{
|
||||
Player player;
|
||||
auto* rev = reinterpret_cast<RevPlayer*>(&player);
|
||||
|
||||
EXPECT_EQ(1, rev->rev_foo());
|
||||
EXPECT_EQ(2, rev->rev_bar());
|
||||
EXPECT_EQ(2, rev->rev_bar_const());
|
||||
}
|
||||
260
tests/general/unit_test_targeting.cpp
Normal file
260
tests/general/unit_test_targeting.cpp
Normal file
@@ -0,0 +1,260 @@
|
||||
//
|
||||
// Created by claude on 19.03.2026.
|
||||
//
|
||||
#include <gtest/gtest.h>
|
||||
#include <omath/algorithm/targeting.hpp>
|
||||
#include <omath/engines/source_engine/camera.hpp>
|
||||
#include <vector>
|
||||
|
||||
namespace
|
||||
{
|
||||
using Camera = omath::source_engine::Camera;
|
||||
using ViewAngles = omath::source_engine::ViewAngles;
|
||||
using Targets = std::vector<omath::Vector3<float>>;
|
||||
using Iter = Targets::const_iterator;
|
||||
using FilterSig = bool(const omath::Vector3<float>&);
|
||||
|
||||
constexpr auto k_fov = omath::Angle<float, 0.f, 180.f, omath::AngleFlags::Clamped>::from_degrees(90.f);
|
||||
|
||||
Camera make_camera(const omath::Vector3<float>& origin, float pitch_deg, float yaw_deg)
|
||||
{
|
||||
ViewAngles angles{
|
||||
omath::source_engine::PitchAngle::from_degrees(pitch_deg),
|
||||
omath::source_engine::YawAngle::from_degrees(yaw_deg),
|
||||
omath::source_engine::RollAngle::from_degrees(0.f),
|
||||
};
|
||||
return Camera{origin, angles, {1920.f, 1080.f}, k_fov, 0.01f, 1000.f};
|
||||
}
|
||||
|
||||
auto get_pos = [](const omath::Vector3<float>& v) -> const omath::Vector3<float>& { return v; };
|
||||
|
||||
Iter find_closest(const Iter begin, const Iter end, const Camera& camera)
|
||||
{
|
||||
return omath::algorithm::get_closest_target_by_fov<Camera, Iter, FilterSig>(
|
||||
begin, end, camera, get_pos);
|
||||
}
|
||||
|
||||
Iter find_nearest(const Iter begin, const Iter end, const omath::Vector3<float>& origin)
|
||||
{
|
||||
return omath::algorithm::get_closest_target_by_distance<Iter, FilterSig>(
|
||||
begin, end, origin, get_pos);
|
||||
}
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, returns_end_for_empty_range)
|
||||
{
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 0.f);
|
||||
Targets targets;
|
||||
|
||||
EXPECT_EQ(find_closest(targets.cbegin(), targets.cend(), camera), targets.cend());
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, single_target_returns_that_target)
|
||||
{
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 0.f);
|
||||
Targets targets = {{100.f, 0.f, 0.f}};
|
||||
|
||||
EXPECT_EQ(find_closest(targets.cbegin(), targets.cend(), camera), targets.cbegin());
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, picks_closest_to_crosshair)
|
||||
{
|
||||
// Camera looking forward along +X (yaw=0, pitch=0 in source engine)
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 0.f);
|
||||
|
||||
Targets targets = {
|
||||
{100.f, 50.f, 0.f}, // off to the side
|
||||
{100.f, 1.f, 0.f}, // nearly on crosshair
|
||||
{100.f, -30.f, 0.f}, // off to the other side
|
||||
};
|
||||
|
||||
const auto result = find_closest(targets.cbegin(), targets.cend(), camera);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 1);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, picks_closest_with_vertical_offset)
|
||||
{
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 0.f);
|
||||
|
||||
Targets targets = {
|
||||
{100.f, 0.f, 50.f}, // high above
|
||||
{100.f, 0.f, 2.f}, // slightly above
|
||||
{100.f, 0.f, 30.f}, // moderately above
|
||||
};
|
||||
|
||||
const auto result = find_closest(targets.cbegin(), targets.cend(), camera);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 1);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, respects_camera_direction)
|
||||
{
|
||||
// Camera looking along +Y (yaw=90)
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 90.f);
|
||||
|
||||
Targets targets = {
|
||||
{100.f, 0.f, 0.f}, // to the side relative to camera facing +Y
|
||||
{0.f, 100.f, 0.f}, // directly in front
|
||||
};
|
||||
|
||||
const auto result = find_closest(targets.cbegin(), targets.cend(), camera);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 1);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, equidistant_targets_returns_first)
|
||||
{
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 0.f);
|
||||
|
||||
// Two targets symmetric about the forward axis — same angular distance
|
||||
Targets targets = {
|
||||
{100.f, 10.f, 0.f},
|
||||
{100.f, -10.f, 0.f},
|
||||
};
|
||||
|
||||
const auto result = find_closest(targets.cbegin(), targets.cend(), camera);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
// First target should be selected (strict < means first wins on tie)
|
||||
EXPECT_EQ(result, targets.cbegin());
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, camera_pitch_affects_selection)
|
||||
{
|
||||
// Camera looking upward (pitch < 0)
|
||||
const auto camera = make_camera({0, 0, 0}, -40.f, 0.f);
|
||||
|
||||
Targets targets = {
|
||||
{100.f, 0.f, 0.f}, // on the horizon
|
||||
{100.f, 0.f, 40.f}, // above, closer to where camera is looking
|
||||
};
|
||||
|
||||
const auto result = find_closest(targets.cbegin(), targets.cend(), camera);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 1);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, many_targets_picks_best)
|
||||
{
|
||||
const auto camera = make_camera({0, 0, 0}, 0.f, 0.f);
|
||||
|
||||
Targets targets = {
|
||||
{100.f, 80.f, 80.f},
|
||||
{100.f, 60.f, 60.f},
|
||||
{100.f, 40.f, 40.f},
|
||||
{100.f, 20.f, 20.f},
|
||||
{100.f, 0.5f, 0.5f}, // closest to crosshair
|
||||
{100.f, 10.f, 10.f},
|
||||
{100.f, 30.f, 30.f},
|
||||
};
|
||||
|
||||
const auto result = find_closest(targets.cbegin(), targets.cend(), camera);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 4);
|
||||
}
|
||||
|
||||
// ── get_closest_target_by_distance tests ────────────────────────────────────
|
||||
|
||||
TEST(unit_test_targeting, distance_returns_end_for_empty_range)
|
||||
{
|
||||
Targets targets;
|
||||
|
||||
EXPECT_EQ(find_nearest(targets.cbegin(), targets.cend(), {0, 0, 0}), targets.cend());
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, distance_single_target)
|
||||
{
|
||||
Targets targets = {{50.f, 0.f, 0.f}};
|
||||
|
||||
EXPECT_EQ(find_nearest(targets.cbegin(), targets.cend(), {0, 0, 0}), targets.cbegin());
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, distance_picks_nearest)
|
||||
{
|
||||
const omath::Vector3<float> origin{0.f, 0.f, 0.f};
|
||||
|
||||
Targets targets = {
|
||||
{100.f, 0.f, 0.f}, // distance = 100
|
||||
{10.f, 0.f, 0.f}, // distance = 10 (closest)
|
||||
{50.f, 0.f, 0.f}, // distance = 50
|
||||
};
|
||||
|
||||
const auto result = find_nearest(targets.cbegin(), targets.cend(), origin);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 1);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, distance_considers_all_axes)
|
||||
{
|
||||
const omath::Vector3<float> origin{0.f, 0.f, 0.f};
|
||||
|
||||
Targets targets = {
|
||||
{30.f, 30.f, 30.f}, // distance = sqrt(2700) ~ 51.96
|
||||
{50.f, 0.f, 0.f}, // distance = 50
|
||||
{0.f, 0.f, 10.f}, // distance = 10 (closest)
|
||||
};
|
||||
|
||||
const auto result = find_nearest(targets.cbegin(), targets.cend(), origin);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 2);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, distance_from_nonzero_origin)
|
||||
{
|
||||
const omath::Vector3<float> origin{100.f, 100.f, 100.f};
|
||||
|
||||
Targets targets = {
|
||||
{0.f, 0.f, 0.f}, // distance = sqrt(30000) ~ 173
|
||||
{105.f, 100.f, 100.f}, // distance = 5 (closest)
|
||||
{200.f, 200.f, 200.f}, // distance = sqrt(30000) ~ 173
|
||||
};
|
||||
|
||||
const auto result = find_nearest(targets.cbegin(), targets.cend(), origin);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 1);
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, distance_equidistant_returns_first)
|
||||
{
|
||||
const omath::Vector3<float> origin{0.f, 0.f, 0.f};
|
||||
|
||||
// Both targets at distance 100, symmetric
|
||||
Targets targets = {
|
||||
{100.f, 0.f, 0.f},
|
||||
{-100.f, 0.f, 0.f},
|
||||
};
|
||||
|
||||
const auto result = find_nearest(targets.cbegin(), targets.cend(), origin);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin());
|
||||
}
|
||||
|
||||
TEST(unit_test_targeting, distance_many_targets)
|
||||
{
|
||||
const omath::Vector3<float> origin{0.f, 0.f, 0.f};
|
||||
|
||||
Targets targets = {
|
||||
{500.f, 0.f, 0.f},
|
||||
{200.f, 200.f, 0.f},
|
||||
{100.f, 100.f, 100.f},
|
||||
{50.f, 50.f, 50.f},
|
||||
{1.f, 1.f, 1.f}, // distance = sqrt(3) ~ 1.73 (closest)
|
||||
{10.f, 10.f, 10.f},
|
||||
{80.f, 0.f, 0.f},
|
||||
};
|
||||
|
||||
const auto result = find_nearest(targets.cbegin(), targets.cend(), origin);
|
||||
|
||||
ASSERT_NE(result, targets.cend());
|
||||
EXPECT_EQ(result, targets.cbegin() + 4);
|
||||
}
|
||||
Reference in New Issue
Block a user