linear/sRGB conversion functions

Author: Wenzel Jakob

drjit/__init__.py

@@ -2647,6 +2647,22 @@ def assert_equal(
         **kwargs,
     )
 
+def srgb_to_linear(x):
+    x = clip(x, 0, 1)
+    return select(
+        x < 0.04045,
+        x / 12.92,
+        ((x + 0.055) / 1.055) ** 2.4
+    )
+
+def linear_to_srgb(x):
+    x = clip(x, 0, 1)
+    return select(
+        x < 0.0031308,
+        x * 12.92,
+        1.055 * (x ** (1.0 / 2.4)) - 0.055
+    )
+
 
 newaxis = None