Fft2

include/nbl/asset/utils/IShaderCompiler.h

@@ -261,6 +261,9 @@ class NBL_API2 IShaderCompiler : public core::IReferenceCounted
 			SPreprocessorOptions preprocessorOptions = {};
 			CCache* readCache = nullptr;
 			CCache* writeCache = nullptr;
+			bool optimizerIsExtraPasses = false; // Instead of disabling the default opt passes, run the provided optimization passes at the end
+			std::string preprocessedOutputPath = "";
+			std::string spvOutputPath = "";
 		};
 
 		class CCache final : public IReferenceCounted

include/nbl/builtin/hlsl/complex.hlsl

@@ -7,6 +7,7 @@
 
 #include <nbl/builtin/hlsl/cpp_compat.hlsl>
 #include <nbl/builtin/hlsl/functional.hlsl>
+#include <nbl/builtin/hlsl/tgmath.hlsl>
 
 using namespace nbl::hlsl;
 
@@ -32,22 +33,6 @@ struct complex_t : public std::complex<Scalar>
     }
 };
 
-// Fast mul by i
-template<typename Scalar>
-complex_t<Scalar> rotateLeft(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
-{
-    complex_t<Scalar> retVal = { -value.imag(), value.real() };
-    return retVal;
-}
-
-// Fast mul by -i
-template<typename Scalar>
-complex_t<Scalar> rotateRight(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
-{
-    complex_t<Scalar> retVal = { value.imag(), -value.real() };
-    return retVal;
-}
-
 }
 }
 
@@ -414,28 +399,11 @@ complex_t<Scalar> proj(const complex_t<Scalar> c)
 template<typename Scalar>
 complex_t<Scalar> polar(const Scalar r, const Scalar theta) 
 {
-    complex_t<Scalar> retVal = {r * cos(theta), r * sin(theta)};
+    complex_t<Scalar> retVal = {r * nbl::hlsl::cos<Scalar>(theta), r * nbl::hlsl::sin<Scalar>(theta)};
     return retVal;
 }
 
 
-// --------------------------------------------- Some more functions that come in handy --------------------------------------
-// Fast mul by i
-template<typename Scalar>
-complex_t<Scalar> rotateLeft(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
-{
-    complex_t<Scalar> retVal = { -value.imag(), value.real() };
-    return retVal;
-}
-
-// Fast mul by -i
-template<typename Scalar>
-complex_t<Scalar> rotateRight(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
-{
-    complex_t<Scalar> retVal = { value.imag(), -value.real() };
-    return retVal;
-}
-
 }
 }
 
@@ -456,4 +424,40 @@ NBL_REGISTER_OBJ_TYPE(complex_t<float64_t4>,::nbl::hlsl::alignment_of_v<float64_
 // -------------------------------------- END HLSL VERSION ---------------------------------------
 #endif
 
+// ---------------------------------------- COMMON ---------------------------------------
+
+namespace nbl
+{
+namespace hlsl
+{
+
+// Fast mul by i
+template<typename Scalar>
+complex_t<Scalar> rotateLeft(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
+{
+    complex_t<Scalar> retVal = { -value.imag(), value.real() };
+    return retVal;
+}
+
+// Fast mul by -i
+template<typename Scalar>
+complex_t<Scalar> rotateRight(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
+{
+    complex_t<Scalar> retVal = { value.imag(), -value.real() };
+    return retVal;
+}
+
+// Fast square
+template<typename Scalar>
+complex_t<Scalar> square(NBL_CONST_REF_ARG(complex_t<Scalar>) value)
+{
+    Scalar real = value.real() * value.real() - value.imag() * value.imag();
+    Scalar imag = 2 * value.real() * value.imag();
+    complex_t<Scalar> retVal = { real, imag };
+    return retVal;
+}
+
+}
+}
+
 #endif