aselimov/Vec3
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add CUDA support

Browse source
This commit is contained in:
Alex Selimov 2025-04-16 18:00:55 -04:00
parent fbc34a0bdd
commit 80d3b6276e
7 changed files with 223 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

25
include/vec3.h
View file

@ -3,35 +3,44 @@
#include <cmath>
#include <limits>
#ifdef __CUDACC__
#define CUDA_CALLABLE __host__ __device__
#else
#define CUDA_CALLABLE
#endif
template <typename T> struct Vec3 {
T x;
T y;
T z;
inline Vec3<T> operator+(Vec3<T> other) const {
CUDA_CALLABLE inline Vec3<T> operator+(Vec3<T> other) const {
return {x + other.x, y + other.y, z + other.z};
};
inline Vec3<T> operator-(Vec3<T> other) const {
CUDA_CALLABLE inline Vec3<T> operator-(Vec3<T> other) const {
return {x - other.x, y - other.y, z - other.z};
};
inline Vec3 scale(T scalar) { return {x * scalar, y * scalar, z * scalar}; };
CUDA_CALLABLE inline Vec3 scale(T scalar) {
return {x * scalar, y * scalar, z * scalar};
};
inline T dot(Vec3<T> other) const {
CUDA_CALLABLE inline T dot(Vec3<T> other) const {
return x * other.x + y * other.y + z * other.z;
}
inline Vec3<T> cross(Vec3<T> other) const {
CUDA_CALLABLE inline Vec3<T> cross(Vec3<T> other) const {
return {y * other.z - z * other.y, z * other.x - x * other.z,
x * other.y - y * other.x};
}
inline T squared_norm2() const { return x * x + y * y + z * z; }
CUDA_CALLABLE inline T squared_norm2() const { return x * x + y * y + z * z; }
inline T norm2() const { return std::sqrt(squared_norm2()); }
CUDA_CALLABLE inline T norm2() const { return std::sqrt(squared_norm2()); }
inline Vec3<T> normalized() {
CUDA_CALLABLE inline Vec3<T> normalized() {
// Add epsilon to the norm for stability when the norm is 0
T norm = std::max(norm2(), std::numeric_limits<T>::epsilon());
return {x / norm, y / norm, z / norm};