enable computing mot number for multiple images

lib/imaging.py

@@ -156,18 +156,21 @@ class FluorescenceData(ImagingData):
         _dA = self.config.get("bin_size", 1)**2 \
                 * (DEF_DX**2 if dA is None else dA)
         if printflag: print("Compute results")
-        N = compute_mot_number(
-            self.arrays["image"],
-            0.76,
-            self.config["gain"], # dB
-            self.config["exposure_time"],
-            (0.0127**2 * np.pi) / (4 * np.pi * 0.125**2),
-            62e6 * 2 * np.pi,
-            0.1
-        )
-        sx, sy = fit_gaussian(self.arrays["image"], _dA)
-        self.results = dict() if self.results is None else self.results
-        self.results.update({"N": N, "sx": sx, "sy": sy})
+        for label, array in self.arrays.items():
+            N = compute_mot_number(
+                array,
+                0.76,
+                self.config["gain"], # dB
+                self.config["exposure_time"],
+                (0.0127**2 * np.pi) / (4 * np.pi * 0.125**2),
+                62e6 * 2 * np.pi,
+                0.1
+            )
+            sx, sy = fit_gaussian(array, _dA)
+            self.results = dict() if self.results is None else self.results
+            self.results.update(
+                {f"N ({label})": N, f"sx ({label})": sx, f"sy ({label})": sy}
+            )
         return self
 
 def compute_mot_number(image, QE, gain, exposure_time, solid_angle,