caizhongang
59 lines
1.8 KiB
C++
59 lines
1.8 KiB
C++
/*
|
|
* Max-Planck-Gesellschaft zur Förderung der Wissenschaften e.V. (MPG) is
|
|
* holder of all proprietary rights on this computer program.
|
|
* You can only use this computer program if you have closed
|
|
* a license agreement with MPG or you get the right to use the computer
|
|
* program from someone who is authorized to grant you that right.
|
|
* Any use of the computer program without a valid license is prohibited and
|
|
* liable to prosecution.
|
|
*
|
|
* Copyright©2019 Max-Planck-Gesellschaft zur Förderung
|
|
* der Wissenschaften e.V. (MPG). acting on behalf of its Max Planck Institute
|
|
* for Intelligent Systems. All rights reserved.
|
|
*
|
|
* @author Vasileios Choutas
|
|
* Contact: vassilis.choutas@tuebingen.mpg.de
|
|
* Contact: ps-license@tuebingen.mpg.de
|
|
*
|
|
*/
|
|
|
|
#ifndef MATH_UTILS_H
|
|
#define MATH_UTILS_H
|
|
|
|
#include <cuda.h>
|
|
#include "device_launch_parameters.h"
|
|
#include <cuda_runtime.h>
|
|
|
|
#include "defs.hpp"
|
|
#include "double_vec_ops.hpp"
|
|
#include "helper_math.h"
|
|
|
|
|
|
template <typename T>
|
|
__host__ __device__ T sign(T x) {
|
|
return x > 0 ? 1 : (x < 0 ? -1 : 0);
|
|
}
|
|
|
|
|
|
template <typename T>
|
|
__host__ __device__ __forceinline__ float vec_abs_diff(const vec3<T> &vec1,
|
|
const vec3<T> &vec2) {
|
|
return fabs(vec1.x - vec2.x) + fabs(vec1.y - vec2.y) + fabs(vec1.z - vec2.z);
|
|
}
|
|
|
|
template <typename T>
|
|
__host__ __device__ __forceinline__ float vec_sq_diff(const vec3<T> &vec1,
|
|
const vec3<T> &vec2) {
|
|
return dot(vec1 - vec2, vec1 - vec2);
|
|
}
|
|
|
|
template <typename T>
|
|
__host__ __device__ __forceinline__ T clamp(T value, T min_value, T max_value) {
|
|
return min(max(value, min_value), max_value);
|
|
}
|
|
|
|
template <typename T> __host__ __device__ T dot2(vec3<T> v) {
|
|
return dot(v, v);
|
|
}
|
|
#endif // ifndef MATH_UTILS_H
|