320 lines
9.6 KiB
C++
320 lines
9.6 KiB
C++
/**
|
|
* Copyright (c) 2014-2018 Enzien Audio Ltd.
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
* AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
* OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
* PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
#ifndef _HEAVY_UTILS_H_
|
|
#define _HEAVY_UTILS_H_
|
|
|
|
// platform definitions
|
|
#if _WIN32 || _WIN64 || _MSC_VER
|
|
#define HV_WIN 1
|
|
#elif __APPLE__
|
|
#define HV_APPLE 1
|
|
#elif __ANDROID__
|
|
#define HV_ANDROID 1
|
|
#elif __unix__ || __unix
|
|
#define HV_UNIX 1
|
|
#else
|
|
#warning Could not detect platform. Assuming Unix-like.
|
|
#endif
|
|
|
|
#ifdef EMSCRIPTEN
|
|
#define HV_EMSCRIPTEN 1
|
|
#endif
|
|
|
|
// basic includes
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
// type definitions
|
|
#include <stdint.h>
|
|
#include <stdbool.h>
|
|
#define hv_uint8_t uint8_t
|
|
#define hv_int16_t int16_t
|
|
#define hv_uint16_t uint16_t
|
|
#define hv_int32_t int32_t
|
|
#define hv_uint32_t uint32_t
|
|
#define hv_uint64_t uint64_t
|
|
#define hv_size_t size_t
|
|
#define hv_uintptr_t uintptr_t
|
|
|
|
// SIMD-specific includes
|
|
#if !(HV_SIMD_NONE || HV_SIMD_NEON || HV_SIMD_SSE || HV_SIMD_AVX)
|
|
#define HV_SIMD_NEON __ARM_NEON__
|
|
#define HV_SIMD_SSE (__SSE__ && __SSE2__ && __SSE3__ && __SSSE3__ && __SSE4_1__)
|
|
#define HV_SIMD_AVX (__AVX__ && HV_SIMD_SSE)
|
|
#endif
|
|
#ifndef HV_SIMD_FMA
|
|
#define HV_SIMD_FMA __FMA__
|
|
#endif
|
|
|
|
#if HV_SIMD_AVX || HV_SIMD_SSE
|
|
#include <immintrin.h>
|
|
#elif HV_SIMD_NEON
|
|
#include <arm_neon.h>
|
|
#endif
|
|
|
|
#if HV_SIMD_NEON // NEON
|
|
#define HV_N_SIMD 4
|
|
#define hv_bufferf_t float32x4_t
|
|
#define hv_bufferi_t int32x4_t
|
|
#define hv_bInf_t float32x4_t
|
|
#define hv_bOutf_t float32x4_t*
|
|
#define hv_bIni_t int32x4_t
|
|
#define hv_bOuti_t int32x4_t*
|
|
#define VIf(_x) (_x)
|
|
#define VOf(_x) (&_x)
|
|
#define VIi(_x) (_x)
|
|
#define VOi(_x) (&_x)
|
|
#elif HV_SIMD_AVX // AVX
|
|
#define HV_N_SIMD 8
|
|
#define hv_bufferf_t __m256
|
|
#define hv_bufferi_t __m256i
|
|
#define hv_bInf_t __m256
|
|
#define hv_bOutf_t __m256*
|
|
#define hv_bIni_t __m256i
|
|
#define hv_bOuti_t __m256i*
|
|
#define VIf(_x) (_x)
|
|
#define VOf(_x) (&_x)
|
|
#define VIi(_x) (_x)
|
|
#define VOi(_x) (&_x)
|
|
#elif HV_SIMD_SSE // SSE
|
|
#define HV_N_SIMD 4
|
|
#define hv_bufferf_t __m128
|
|
#define hv_bufferi_t __m128i
|
|
#define hv_bInf_t __m128
|
|
#define hv_bOutf_t __m128*
|
|
#define hv_bIni_t __m128i
|
|
#define hv_bOuti_t __m128i*
|
|
#define VIf(_x) (_x)
|
|
#define VOf(_x) (&_x)
|
|
#define VIi(_x) (_x)
|
|
#define VOi(_x) (&_x)
|
|
#else // DEFAULT
|
|
#define HV_N_SIMD 1
|
|
#undef HV_SIMD_NONE
|
|
#define HV_SIMD_NONE 1
|
|
#define hv_bufferf_t float
|
|
#define hv_bufferi_t int
|
|
#define hv_bInf_t float
|
|
#define hv_bOutf_t float*
|
|
#define hv_bIni_t int
|
|
#define hv_bOuti_t int*
|
|
#define VIf(_x) (_x)
|
|
#define VOf(_x) (&_x)
|
|
#define VIi(_x) (_x)
|
|
#define VOi(_x) (&_x)
|
|
#endif
|
|
|
|
#define HV_N_SIMD_MASK (HV_N_SIMD-1)
|
|
|
|
// Strings
|
|
#include <string.h>
|
|
#define hv_strlen(a) strlen(a)
|
|
#define hv_strcmp(a, b) strcmp(a, b)
|
|
#define hv_snprintf(a, b, c, ...) snprintf(a, b, c, __VA_ARGS__)
|
|
#if HV_WIN
|
|
#define hv_strncpy(_dst, _src, _len) strncpy_s(_dst, _len, _src, _TRUNCATE)
|
|
#else
|
|
#define hv_strncpy(_dst, _src, _len) strncpy(_dst, _src, _len)
|
|
#endif
|
|
|
|
// Memory management
|
|
#define hv_memcpy(a, b, c) memcpy(a, b, c)
|
|
#define hv_memclear(a, b) memset(a, 0, b)
|
|
#if HV_WIN
|
|
#include <malloc.h>
|
|
#define hv_alloca(_n) _alloca(_n)
|
|
#if HV_SIMD_AVX
|
|
#define hv_malloc(_n) _aligned_malloc(_n, 32)
|
|
#define hv_realloc(a, b) _aligned_realloc(a, b, 32)
|
|
#define hv_free(x) _aligned_free(x)
|
|
#elif HV_SIMD_SSE || HV_SIMD_NEON
|
|
#define hv_malloc(_n) _aligned_malloc(_n, 16)
|
|
#define hv_realloc(a, b) _aligned_realloc(a, b, 16)
|
|
#define hv_free(x) _aligned_free(x)
|
|
#else // HV_SIMD_NONE
|
|
#define hv_malloc(_n) malloc(_n)
|
|
#define hv_realloc(a, b) realloc(a, b)
|
|
#define hv_free(_n) free(_n)
|
|
#endif
|
|
#elif HV_APPLE
|
|
#define hv_alloca(_n) alloca(_n)
|
|
#define hv_realloc(a, b) realloc(a, b)
|
|
#if HV_SIMD_AVX
|
|
#include <mm_malloc.h>
|
|
#define hv_malloc(_n) _mm_malloc(_n, 32)
|
|
#define hv_free(x) _mm_free(x)
|
|
#elif HV_SIMD_SSE
|
|
#include <mm_malloc.h>
|
|
#define hv_malloc(_n) _mm_malloc(_n, 16)
|
|
#define hv_free(x) _mm_free(x)
|
|
#elif HV_SIMD_NEON
|
|
// malloc on ios always has 16-byte alignment
|
|
#define hv_malloc(_n) malloc(_n)
|
|
#define hv_free(x) free(x)
|
|
#else // HV_SIMD_NONE
|
|
#define hv_malloc(_n) malloc(_n)
|
|
#define hv_free(x) free(x)
|
|
#endif
|
|
#else
|
|
#include <alloca.h>
|
|
#define hv_alloca(_n) alloca(_n)
|
|
#define hv_realloc(a, b) realloc(a, b)
|
|
#if HV_SIMD_AVX
|
|
#define hv_malloc(_n) aligned_alloc(32, _n)
|
|
#define hv_free(x) free(x)
|
|
#elif HV_SIMD_SSE
|
|
#define hv_malloc(_n) aligned_alloc(16, _n)
|
|
#define hv_free(x) free(x)
|
|
#elif HV_SIMD_NEON
|
|
#if HV_ANDROID
|
|
#define hv_malloc(_n) memalign(16, _n)
|
|
#define hv_free(x) free(x)
|
|
#else
|
|
#define hv_malloc(_n) aligned_alloc(16, _n)
|
|
#define hv_free(x) free(x)
|
|
#endif
|
|
#else // HV_SIMD_NONE
|
|
#define hv_malloc(_n) malloc(_n)
|
|
#define hv_free(_n) free(_n)
|
|
#endif
|
|
#endif
|
|
|
|
// Assert
|
|
#include <assert.h>
|
|
#define hv_assert(e) assert(e)
|
|
|
|
// Export and Inline
|
|
#if HV_WIN
|
|
#define HV_EXPORT __declspec(dllexport)
|
|
#ifndef __cplusplus // MSVC doesn't like redefining "inline" keyword
|
|
#define inline __inline
|
|
#endif
|
|
#define HV_FORCE_INLINE __forceinline
|
|
#else
|
|
#define HV_EXPORT
|
|
#define HV_FORCE_INLINE inline __attribute__((always_inline))
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
// Returns a 32-bit hash of any string. Returns 0 if string is NULL.
|
|
hv_uint32_t hv_string_to_hash(const char *str);
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
// Math
|
|
#include <math.h>
|
|
static inline hv_size_t __hv_utils_max_ui(hv_size_t x, hv_size_t y) { return (x > y) ? x : y; }
|
|
static inline hv_size_t __hv_utils_min_ui(hv_size_t x, hv_size_t y) { return (x < y) ? x : y; }
|
|
static inline hv_int32_t __hv_utils_max_i(hv_int32_t x, hv_int32_t y) { return (x > y) ? x : y; }
|
|
static inline hv_int32_t __hv_utils_min_i(hv_int32_t x, hv_int32_t y) { return (x < y) ? x : y; }
|
|
#define hv_max_ui(a, b) __hv_utils_max_ui(a, b)
|
|
#define hv_min_ui(a, b) __hv_utils_min_ui(a, b)
|
|
#define hv_max_i(a, b) __hv_utils_max_i(a, b)
|
|
#define hv_min_i(a, b) __hv_utils_min_i(a, b)
|
|
#define hv_max_f(a, b) fmaxf(a, b)
|
|
#define hv_min_f(a, b) fminf(a, b)
|
|
#define hv_max_d(a, b) fmax(a, b)
|
|
#define hv_min_d(a, b) fmin(a, b)
|
|
#define hv_sin_f(a) sinf(a)
|
|
#define hv_sinh_f(a) sinhf(a)
|
|
#define hv_cos_f(a) cosf(a)
|
|
#define hv_cosh_f(a) coshf(a)
|
|
#define hv_tan_f(a) tanf(a)
|
|
#define hv_tanh_f(a) tanhf(a)
|
|
#define hv_asin_f(a) asinf(a)
|
|
#define hv_asinh_f(a) asinhf(a)
|
|
#define hv_acos_f(a) acosf(a)
|
|
#define hv_acosh_f(a) acoshf(a)
|
|
#define hv_atan_f(a) atanf(a)
|
|
#define hv_atanh_f(a) atanhf(a)
|
|
#define hv_atan2_f(a, b) atan2f(a, b)
|
|
#define hv_exp_f(a) expf(a)
|
|
#define hv_abs_f(a) fabsf(a)
|
|
#define hv_sqrt_f(a) sqrtf(a)
|
|
#define hv_log_f(a) logf(a)
|
|
#define hv_ceil_f(a) ceilf(a)
|
|
#define hv_floor_f(a) floorf(a)
|
|
#define hv_round_f(a) roundf(a)
|
|
#define hv_pow_f(a, b) powf(a, b)
|
|
#if HV_EMSCRIPTEN
|
|
#define hv_fma_f(a, b, c) ((a*b)+c) // emscripten does not support fmaf (yet?)
|
|
#else
|
|
#define hv_fma_f(a, b, c) fmaf(a, b, c)
|
|
#endif
|
|
#if HV_WIN
|
|
// finds ceil(log2(x))
|
|
#include <intrin.h>
|
|
static inline hv_uint32_t __hv_utils_min_max_log2(hv_uint32_t x) {
|
|
unsigned long z = 0;
|
|
_BitScanReverse(&z, x);
|
|
return (hv_uint32_t) (z+1);
|
|
}
|
|
#else
|
|
static inline hv_uint32_t __hv_utils_min_max_log2(hv_uint32_t x) {
|
|
return (hv_uint32_t) (32 - __builtin_clz(x-1));
|
|
}
|
|
#endif
|
|
#define hv_min_max_log2(a) __hv_utils_min_max_log2(a)
|
|
|
|
// Atomics
|
|
#if HV_WIN
|
|
#include <windows.h>
|
|
#define hv_atomic_bool volatile LONG
|
|
#define HV_SPINLOCK_ACQUIRE(_x) while (InterlockedCompareExchange(&_x, true, false)) { }
|
|
#define HV_SPINLOCK_TRY(_x) return !InterlockedCompareExchange(&_x, true, false)
|
|
#define HV_SPINLOCK_RELEASE(_x) (_x = false)
|
|
#elif HV_ANDROID
|
|
// Android support for atomics isn't that great, we'll do it manually
|
|
// https://gcc.gnu.org/onlinedocs/gcc-4.1.2/gcc/Atomic-Builtins.html
|
|
#define hv_atomic_bool hv_uint8_t
|
|
#define HV_SPINLOCK_ACQUIRE(_x) while (__sync_lock_test_and_set(&_x, 1))
|
|
#define HV_SPINLOCK_TRY(_x) return !__sync_lock_test_and_set(&_x, 1)
|
|
#define HV_SPINLOCK_RELEASE(_x) __sync_lock_release(&_x)
|
|
#elif __cplusplus
|
|
#include <atomic>
|
|
#define hv_atomic_bool std::atomic_flag
|
|
#define HV_SPINLOCK_ACQUIRE(_x) while (_x.test_and_set(std::memory_order_acquire))
|
|
#define HV_SPINLOCK_TRY(_x) return !_x.test_and_set(std::memory_order_acquire)
|
|
#define HV_SPINLOCK_RELEASE(_x) _x.clear(std::memory_order_release)
|
|
#elif defined(__has_include)
|
|
#if __has_include(<stdatomic.h>)
|
|
#include <stdatomic.h>
|
|
#define hv_atomic_bool atomic_flag
|
|
#define HV_SPINLOCK_ACQUIRE(_x) while (atomic_flag_test_and_set_explicit(&_x, memory_order_acquire))
|
|
#define HV_SPINLOCK_TRY(_x) return !atomic_flag_test_and_set_explicit(&_x, memory_order_acquire)
|
|
#define HV_SPINLOCK_RELEASE(_x) atomic_flag_clear_explicit(memory_order_release)
|
|
#endif
|
|
#endif
|
|
#ifndef hv_atomic_bool
|
|
#define hv_atomic_bool volatile bool
|
|
#define HV_SPINLOCK_ACQUIRE(_x) \
|
|
while (_x) {} \
|
|
_x = true;
|
|
#define HV_SPINLOCK_TRY(_x) \
|
|
if (!_x) { \
|
|
_x = true; \
|
|
return true; \
|
|
} else return false;
|
|
#define HV_SPINLOCK_RELEASE(_x) (_x = false)
|
|
#endif
|
|
|
|
#endif // _HEAVY_UTILS_H_
|