Merge pull request #561 from Devsh-Graphics-Programming/new_hlsl_xoroshiro

New hlsl xoroshiro

Author: devshgraphicsprogramming

include/nbl/builtin/hlsl/bit.hlsl

@@ -0,0 +1,63 @@
+#ifndef _NBL_BUILTIN_HLSL_BIT_INCLUDED_
+#define _NBL_BUILTIN_HLSL_BIT_INCLUDED_
+
+#include <nbl/builtin/hlsl/cpp_compat.hlsl>
+
+#ifndef __HLSL_VERSION
+#include <bit>
+
+namespace nbl::hlsl
+{
+
+NBL_ALIAS_TEMPLATE_FUNCTION(std::rotl, rotl);
+NBL_ALIAS_TEMPLATE_FUNCTION(std::rotr, rotr);
+
+}
+#else
+namespace nbl
+{
+namespace hlsl
+{
+
+template<typename T, typename S>
+T rotl(T x, S s);
+template<typename T, typename S>
+T rotr(T x, S s);
+
+template<typename T, typename S>
+T rotl(T x, S s)
+{
+    const T N = 32u;
+    const S r = s % N;
+    
+    if(r >= 0)
+    {
+        return (x << r) | (x >> (N - r));
+    }
+    else
+    {
+        return (x >> (-r)) | (x << (N - (-r)));
+    }
+}
+
+template<typename T, typename S>
+T rotr(T x, S s)
+{
+    const T N = 32u;
+    const S r = s % N;
+    
+    if(r >= 0)
+    {
+        return (x >> r) | (x << (N - r));
+    }
+    else
+    {
+        return (x << (-r)) | (x >> (N - (-r)));
+    }
+}
+
+}
+}
+#endif
+ 
+#endif

include/nbl/builtin/hlsl/cpp_compat.hlsl

@@ -9,6 +9,13 @@
 #define NBL_CONSTEXPR constexpr
 #define NBL_CONSTEXPR_STATIC_INLINE constexpr static inline
 
+#define NBL_ALIAS_TEMPLATE_FUNCTION(origFunctionName, functionAlias) \
+template<typename... Args> \
+inline auto functionAlias(Args&&... args) -> decltype(origFunctionName(std::forward<Args>(args)...)) \
+{ \
+    return origFunctionName(std::forward<Args>(args)...); \
+}
+
 namespace nbl::hlsl
 {
 

include/nbl/builtin/hlsl/random/xoroshiro.hlsl

@@ -0,0 +1,70 @@
+
+// Copyright (C) 2018-2020 - DevSH Graphics Programming Sp. z O.O.
+// This file is part of the "Nabla Engine".
+// For conditions of distribution and use, see copyright notice in nabla.h
+
+#ifndef _NBL_BUILTIN_GLSL_RANDOM_XOROSHIRO_HLSL_INCLUDED_
+#define _NBL_BUILTIN_GLSL_RANDOM_XOROSHIRO_HLSL_INCLUDED_
+
+#include <nbl/builtin/hlsl/cpp_compat/vector.hlsl>
+#include <nbl/builtin/hlsl/bit.hlsl>
+
+namespace nbl
+{
+namespace hlsl
+{
+
+struct Xoroshiro64StateHolder
+{
+	void xoroshiro64_state_advance()
+	{
+		state[1] ^= state[0];
+		state[0] = rotl(state[0], 26u) ^ state[1] ^ (state[1]<<9u); // a, b
+		state[1] = rotl(state[1], 13u); // c
+	}
+	
+	uint32_t2 state;
+};
+
+struct Xoroshiro64Star
+{
+	static Xoroshiro64Star construct(NBL_CONST_REF_ARG(uint32_t2) initialState)
+	{
+		Xoroshiro64StateHolder stateHolder = {initialState};
+		return Xoroshiro64Star(stateHolder);
+	}
+	
+	uint32_t operator()()
+	{
+		const uint32_t result = stateHolder.state[0]*0x9E3779BBu;
+		stateHolder.xoroshiro64_state_advance();
+
+		return result;
+	}
+	
+	Xoroshiro64StateHolder stateHolder;
+};
+
+struct Xoroshiro64StarStar
+{
+	static Xoroshiro64StarStar construct(NBL_CONST_REF_ARG(uint32_t2) initialState)
+	{
+		Xoroshiro64StateHolder stateHolder = {initialState};
+		return Xoroshiro64StarStar(stateHolder);
+	}
+	
+	uint32_t operator()()
+	{
+		const uint32_t result = rotl(stateHolder.state[0]*0x9E3779BBu,5u)*5u;
+	    stateHolder.xoroshiro64_state_advance();
+	
+		return result;
+	}
+
+	Xoroshiro64StateHolder stateHolder;
+};
+
+}
+}
+
+#endif

src/nbl/builtin/CMakeLists.txt

@@ -105,6 +105,7 @@ LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "glsl/property_pool/transfer.glsl")
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "glsl/property_pool/copy.comp")
 # random numbers
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "glsl/random/xoroshiro.glsl")
+LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/random/xoroshiro.hlsl")
 # sampling
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "glsl/sampling/bilinear.glsl")
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "glsl/sampling/box_muller_transform.glsl")
@@ -258,11 +259,11 @@ LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/cpp_compat/vector.hlsl")
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/cpp_compat/promote.hlsl")
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/cpp_compat/intrinsics.h")
 
+LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/bit.hlsl")
 
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/math/quadrature/gauss_legendre/gauss_legendre.hlsl")
 LIST_BUILTIN_RESOURCE(NBL_RESOURCES_TO_EMBED "hlsl/math/quadrature/gauss_legendre/impl.hlsl")
 
-
 macro(NBL_ADD_BUILTIN_RESOURCES _TARGET_) # internal & Nabla only, must be added with the macro to properly propagate scope
 	ADD_CUSTOM_BUILTIN_RESOURCES("${_TARGET_}" NBL_RESOURCES_TO_EMBED "${NBL_ROOT_PATH}/include" "nbl/builtin" "nbl::builtin" "${NBL_ROOT_PATH_BINARY}/include" "${NBL_ROOT_PATH_BINARY}/src" "STATIC" "INTERNAL")
 endmacro()