Skip to content
Snippets Groups Projects
Commit 92cb6a1f authored by Hampus Wikmark Kreuger's avatar Hampus Wikmark Kreuger
Browse files

Allow load_dark to load an array of dark images

parent bd45362a
No related branches found
No related tags found
2 merge requests!195Commented centroid,!188P2866 updates
...@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt ...@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
from scipy.optimize import leastsq from scipy.optimize import leastsq
from scipy.optimize import curve_fit from scipy.optimize import curve_fit
from scipy.signal import fftconvolve from scipy.signal import fftconvolve
import xarray as xr
import toolbox_scs as tb import toolbox_scs as tb
...@@ -323,7 +323,15 @@ class hRIXS: ...@@ -323,7 +323,15 @@ class hRIXS:
return data return data
def load_dark(self, runNB, proposal=None): def load_dark(self, runNB, proposal=None):
data = self.from_run(runNB, proposal) try:
# Checks to see if runNB can be iterated over (is list-like)
data_list = []
for run in runNB:
data_list.append(self.from_run(run, proposal))
data = xr.concat(data_list, dim='trainId')
except TypeError:
# If runNB cannot be iterated over, we assume it's a single run
data = self.from_run(runNB, proposal)
self.dark_image = data['hRIXS_det'].mean(dim='trainId') self.dark_image = data['hRIXS_det'].mean(dim='trainId')
self.USE_DARK = True self.USE_DARK = True
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment