Merge pull request #1789 from HEXRD/export-polar-nans

Export masked lineout intensities as nans
HEXRD · Feb 15, 2025 · fb8cd18 · fb8cd18
2 parents 385e321 + 60d91ea
commit fb8cd18
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/hexrdgui/calibration/polar_plot.py b/hexrdgui/calibration/polar_plot.py
@@ -92,8 +92,15 @@ def update_detectors(self, detectors):
     def write_image(self, filename='polar_image.npz'):
         filename = Path(filename)
 
-        azimuthal_integration = np.array(
-            HexrdConfig().last_unscaled_azimuthal_integral_data)
+        tth, intensities = (
+            HexrdConfig().last_unscaled_azimuthal_integral_data
+        )
+
+        # Remove any nan values
+        mask = intensities.mask
+        tth = tth[~mask]
+        intensities = intensities.data[~mask]
+        azimuthal_integration = np.array((tth, intensities))
 
         if HexrdConfig().polar_x_axis_type == PolarXAxisType.q:
             # Convert to Q