Common.h

Go to the documentation of this file.

 
#pragma once
 
#include "Vec.h"
 
namespace reg23 {
 
template <typename T> struct Linear2;
 
template <typename T> struct Linear {
    T intercept;
    T gradient;
 
    [[nodiscard]] __host__ __device__ T operator()(const T &x) const { return gradient * x + intercept; }
 
    [[nodiscard]] __host__ __device__ Linear operator()(const Linear &other) const {
        return {gradient * other.intercept + intercept, gradient * other.gradient};
    }
 
    [[nodiscard]] __host__ __device__ Linear2<T> operator()(const Linear2<T> &other) const;
};
 
template <typename T> struct Linear2 {
    T intercept;
    T gradient1;
    T gradient2;
 
    [[nodiscard]] __host__ __device__ T operator()(const T &x, const T &y) const {
        return gradient1 * x + gradient2 * y + intercept;
    }
 
    [[nodiscard]] __host__ __device__ Linear2 operator()(const Linear<T> &other) const {
        return {(gradient1 + gradient2) * other.intercept + intercept, gradient1 * other.gradient,
                gradient2 * other.gradient};
    }
 
    [[nodiscard]] __host__ __device__ Linear2 operator()(const Linear2 &other) const {
        const T gradientSum = gradient1 + gradient2;
        return {gradientSum * other.intercept + intercept, gradientSum * other.gradient1,
                gradientSum * other.gradient2};
    }
};
 
template <typename T>
[[nodiscard]] __host__ __device__ Linear2<T> Linear<T>::operator()(const Linear2<T> &other) const {
    return {gradient * other.intercept + intercept, gradient * other.gradient1, gradient * other.gradient2};
}
 
template <typename T> [[nodiscard]] __host__ __device__ Vec<T, 3> UnflipSphericalCoordinate(const Vec<T, 3> &coordSph) {
    constexpr T PI_T = static_cast<T>(M_PI);
    const T theta_div = std::floor((coordSph.Y() + .5f * PI_T) / PI_T);
    const bool theta_flip = static_cast<bool>(std::abs(static_cast<long>(theta_div)) % 2);
    const T phi_div = std::floor((coordSph.Z() + .5f * PI_T) / PI_T);
    const bool phi_flip = static_cast<bool>(std::abs(static_cast<long>(phi_div)) % 2);
 
    T r_ret = coordSph.X();
    T theta_ret = coordSph.Y() - PI_T * theta_div;
    T phi_ret = coordSph.Z() - PI_T * phi_div;
 
    if (phi_flip && !theta_flip) theta_ret *= -1;
    if (phi_flip != theta_flip) r_ret *= -1;
 
    return {r_ret, theta_ret, phi_ret};
}
 
#ifdef __CUDACC__
 
template <typename T>
__host__ cudaError_t CudaMemcpyToObjectSymbol(const T &symbol, T &src, cudaMemcpyKind kind = cudaMemcpyHostToDevice) {
    return cudaMemcpyToSymbol(symbol, &src, sizeof(T), 0, kind);
}
 
#endif
 
} // namespace reg23