Add Safe Sqrt (#57)

marat569 · Oct 15, 2024 · b6f0e9e · b6f0e9e
1 parent 738f6d3
commit b6f0e9e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/shaders/math.hlsl b/src/shaders/math.hlsl
@@ -40,6 +40,17 @@ POWSAFE_FUNCTION_GENERATOR(float3);
 POWSAFE_FUNCTION_GENERATOR(float4);
 #undef POWSAFE_FUNCTION_GENERATOR
 
+#define SQRTSAFE_FUNCTION_GENERATOR(struct) \
+  struct SqrtSafe(struct x) {               \
+    return Sign(x) * sqrt(abs(x));          \
+  }
+
+SQRTSAFE_FUNCTION_GENERATOR(float);
+SQRTSAFE_FUNCTION_GENERATOR(float2);
+SQRTSAFE_FUNCTION_GENERATOR(float3);
+SQRTSAFE_FUNCTION_GENERATOR(float4);
+#undef SQRTSAFE_FUNCTION_GENERATOR
+
 float Average(float3 color) {
   return (color.x + color.y + color.z) / 3.f;
 }