Merge pull request #164 from usnistgov/bugfix_energy_index

Handle non-energy scans that are saturated/underexposed
usnistgov · Dec 5, 2024 · 3adbdc3 · 3adbdc3
2 parents 5059025 + 5742d64
commit 3adbdc3
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/PyHyperScattering/SST1RSoXSDB.py b/src/PyHyperScattering/SST1RSoXSDB.py
@@ -1102,8 +1102,11 @@ def loadMd(self, run):
                     message += "Wide Angle CCD Detector is reported as underexposed at all energies."
                 else:
                     idx = np.where(md["Wide Angle CCD Detector_under_exposed"])
-                    warning_e = md["energy"][idx]
-                    message += f"Affected energies include: \n{warning_e}"
+                    try:
+                        warning_e = md["energy"][idx]
+                        message += f"Affected energies include: \n{warning_e}"
+                    except Exception:
+                        message += f"Affected frames were {idx}."
                 warnings.warn(message, stacklevel=2)
         else:
             warnings.warn(
@@ -1118,8 +1121,11 @@ def loadMd(self, run):
                     message += "\tWide Angle CCD Detector is reported as saturated at all energies."
                 else:
                     idx = np.where(md["Wide Angle CCD Detector_saturated"])
-                    warning_e = md["energy"][idx]
-                    message += f"Affected energies include: \n{warning_e}"
+                    try:
+                        warning_e = md["energy"][idx]
+                        message += f"Affected energies include: \n{warning_e}"
+                    except Exception:
+                        message += f"Affected frames were {idx}."
                 warnings.warn(message, stacklevel=2)
         else:
             warnings.warn(