Skip to content

Commit de17419

Browse files
committed
Fixed Application.experiment data structures to match Spectrum_Storage
1 parent fe6f562 commit de17419

File tree

1 file changed

+17
-15
lines changed

1 file changed

+17
-15
lines changed

application.py

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -314,18 +314,17 @@ def get_selected_absorbance(self, scopes):
314314
# Naming convention for absorbance spectrum is as follows :
315315
# "ABSORBANCE-{channel_id}"
316316

317-
abs_spectras = []
317+
abs_spectras = dict([])
318318

319319
for key in scopes:
320320

321321
# This is not useful to compute the absorbanceSpectrum of the
322322
# referenceChannel, thus we don't do it.
323323

324324
if key != chosen:
325-
abs_spectras.append(
326-
("ABSORBANCE-{}".format(key),
327-
spectro.Spectrum.absorbanceSpectrum(
328-
ref_spectrum, scopes[key])))
325+
abs_spectras["ABSORBANCE-{}".format(key)] = \
326+
spectro.Spectrum.absorbanceSpectrum(
327+
ref_spectrum, scopes[key])
329328

330329
return abs_spectras
331330

@@ -724,23 +723,26 @@ def experiment(self):
724723
self.liveDisplay.putSpectrasAndUpdate(
725724
3, self.spectra_storage[exp_timestamp, n_d, :])
726725

727-
black_corrected_scopes = [
728-
(tp_scopes[i][0],
729-
tp_scopes[i][1]-self.spectra_storage.latest_black)
730-
for i in range(len(tp_scopes))
731-
]
726+
black_corrected_scopes = dict([])
727+
728+
for id, spectrum in tp_scopes.items():
729+
black_corrected_scopes[id] = \
730+
spectrum - self.spectra_storage.latest_black[id]
732731

733732
tp_absorbance = self.get_selected_absorbance(
734733
black_corrected_scopes
735734
)
736735

737-
first_absorbance_spectrum_name = black_corrected_scopes[0][0]
736+
first_absorbance_spectrum_name = tp_absorbance.keys()[0]
737+
738+
corrected_absorbance = dict([])
739+
740+
for key in tp_absorbance:
741+
corrected_absorbance[key] = \
742+
tp_absorbance[key]-correction_spectrum[key]
738743

739744
self.spectra_storage.putSpectra(
740-
exp_timestamp, n_d,
741-
[(tp_absorbance[i][0],
742-
tp_absorbance[i][1]-correction_spectrum[i][1])
743-
for i in range(len(tp_absorbance))])
745+
exp_timestamp, n_d, tp_absorbance)
744746

745747
self.liveDisplay.putSpectrasAndUpdate(
746748
4,

0 commit comments

Comments
 (0)