bugfix when stripe cannot be found (trx in between stripes)

2026-06-25 07:17:48 +02:00
parent d9e8c42d3d
commit 6e1cdf4fa5
1 changed files with 4 additions and 0 deletions
@@ -167,6 +167,8 @@ def cm_reflectivity(cm_stripe: str, cm_pitch: float, energy: float) -> float:
    Returns:
        float: Reflectivity [0-1]
    """
+    if cm_stripe is None:
+        return np.nan
    index = bl.cm.surface.index(cm_stripe)
    rs, _ = bl.cm.material[index].get_amplitude(energy, np.sin(cm_pitch))[0:2]
    refl = abs(rs) ** 2
@@ -186,6 +188,8 @@ def fm_reflectivity(fm_stripe: str, fm_pitch: float, energy: float) -> float:
    Returns:
        float: Reflectivity [0-1]
    """
+    if fm_stripe is None:
+        return np.nan
    if fm_stripe in ("Rh (toroid)", "Pt (toroid)"):
        surface = bl.fm.surfaceToroid
        material = bl.fm.materialToroid