fix scaling when dealing with masked arrays

HEXRD · Dec 19, 2024 · 674e2d6 · 674e2d6
1 parent 95e7586
commit 674e2d6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/hexrdgui/scaling.py b/hexrdgui/scaling.py
@@ -12,14 +12,23 @@ def none(x):
 
 
 def sqrt(x):
+    if isinstance(x, np.ma.masked_array):
+        return np.sqrt(x.filled(np.nan) - 
+               np.nanmin(x.filled(np.nan)))
     return np.sqrt(x - np.nanmin(x))
 
 
 def log(x):
+    if isinstance(x, np.ma.masked_array):
+        return np.log(x.filled(np.nan) - 
+               np.nanmin(x.filled(np.nan)) + 1)
     return np.log(x - np.nanmin(x) + 1)
 
 
 def log_log_sqrt(x):
+    if isinstance(x, np.ma.masked_array):
+        return np.log(np.log(np.sqrt(x.filled(np.nan) - 
+               np.nanmin(x.filled(np.nan))) + 1) + 1)
     return np.log(np.log(np.sqrt(x - np.nanmin(x)) + 1) + 1)