Load a mask when loading image for Xeuss Eiger detector

gfreychet · Nov 6, 2024 · 87cab36 · 87cab36
1 parent caa7c94
commit 87cab36
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/smi_analysis/SMI_beamline.py b/smi_analysis/SMI_beamline.py
@@ -146,6 +146,8 @@ def open_data_db(self, lst_img, optional_mask=None):
         for img, bs in zip(lst_img, self.bs):
             if self.detector != 'rayonix' and self.detector != 'Eiger1M_xeuss' and self.detector != 'Eiger500k_xeuss':
                 self.masks.append(self.det.calc_mask(bs=bs, bs_kind=self.bs_kind, optional_mask=optional_mask))
+            if self.detector == 'Eiger1M_xeuss' or self.detector == 'Eiger500k_xeuss':
+                self.masks.append(self.det.calc_mask(img=img))
             if self.detector == 'Pilatus1m':
                 self.imgs.append(img)
             elif self.detector == 'Pilatus900kw':
@@ -164,6 +166,7 @@ def open_data_db(self, lst_img, optional_mask=None):
             elif self.detector == 'Eiger500k_xeuss':
                 self.imgs.append(img)
 
+
     def calculate_integrator_trans(self, det_rots):
         self.ai = []
         ai = azimuthalIntegrator.AzimuthalIntegrator(**{'detector': self.det,