AND WE HAVE REFERENCES IN HLSL!

Author: devsh

include/nbl/builtin/hlsl/functional.hlsl

@@ -5,14 +5,69 @@
 #define _NBL_BUILTIN_HLSL_FUNCTIONAL_INCLUDED_
 
 
-#include "nbl/builtin/hlsl/bit.hlsl"
+#include "nbl/builtin/hlsl/glsl_compat/core.hlsl"
 #include "nbl/builtin/hlsl/limits.hlsl"
 
 
 namespace nbl
 {
 namespace hlsl
 {
+template<uint32_t StorageClass, typename T>
+using __spv_ptr_t = spirv::pointer_t<StorageClass,T>;
+
+template<uint32_t StorageClass, typename T>
+[[vk::ext_instruction(spv::OpCopyObject)]]
+__spv_ptr_t<StorageClass,T> addrof([[vk::ext_reference]] T v); 
+
+template<uint32_t StorageClass, typename T>
+struct reference_wrapper_base
+{
+    using spv_ptr_t = __spv_ptr_t<StorageClass,T>;
+    spv_ptr_t ptr;
+
+    void __init(spv_ptr_t _ptr)
+    {
+        ptr = _ptr;
+    }
+
+    T load()
+    {
+        return spirv::load<T,spv_ptr_t>(ptr);
+    }
+
+    void store(const T val)
+    {
+        spirv::store<T,spv_ptr_t>(ptr,val);
+    }
+
+    // TODO: use the same defaults as `glsl::atomicAnd`
+    template<uint32_t memoryScope, uint32_t memorySemantics, typename S=T>
+    // TODO: instead of `is_scalar_v` we need a test on whether the `spirv::atomicAnd` expression is callable
+    enable_if_t<is_same_v<S,T>&&is_scalar_v<S>,T> atomicAnd(const T value)
+    {
+        return spirv::atomicAnd<T,spv_ptr_t>(ptr,memoryScope,memorySemantics,value);
+    }
+    // TODO: generate Or,Xor through a macro
+
+    // TODO: comp swap is special, has an extra parameter
+};
+template<typename T>
+struct reference_wrapper_base<spv::StorageClassPhysicalStorageBuffer,T>
+{
+    using spv_ptr_t = typename T::instead_use_nbl::hlsl::bda::__ref;
+
+    // normally would have specializations of load and store
+};
+
+template<uint32_t StorageClass, typename T>
+struct reference_wrapper : reference_wrapper_base<StorageClass,T>
+{
+};
+// TODO: generate atomic Add,Sub,Min,Max through partial template specializations on T
+// TODO: partial specializations for T being a special SPIR-V type for image ops, etc.
+
+
 #ifndef __HLSL_VERSION // CPP
 #define ALIAS_STD(NAME,OP) template<typename T> struct NAME : std::NAME<T> { \
     using type_t = T;