diff --git a/fishbook/frontend/util.py b/fishbook/frontend/util.py index 8e33e52..e9a9f88 100644 --- a/fishbook/frontend/util.py +++ b/fishbook/frontend/util.py @@ -200,24 +200,27 @@ class StimSpikesFile: index_map = {} trial_data = [] + trial_duration = 0.0 index = 0 trial = 0 for l in lines: l = l.strip() + if "duration:" in l: + trial_duration = float(l[1:].strip().split(":")[-3]) if "index:" in l: if len(trial_data) > 0: - index_map[(index, trial)] = trial_data + index_map[(index, trial)] = (trial_duration, trial_data) trial_data = [] index = int(l[1:].strip().split(":")[-1]) if "trial:" in l: if len(trial_data) > 0: - index_map[(index, trial)] = trial_data + index_map[(index, trial)] = (trial_duration, trial_data) trial_data = [] trial = int(l[1:].strip().split(":")[-1]) if len(l) > 0 and "#" not in l: trial_data.append(float(l)/1000) - + index_map[(index, trial)] = (trial_duration, trial_data) return index_map def get(self, run_index, trial_index):