From 0881947fd27d025b0beddc59f8daa45fe6b92031 Mon Sep 17 00:00:00 2001 From: Jan Grewe Date: Thu, 13 Aug 2020 15:53:56 +0200 Subject: [PATCH] [util] Stimspikes stored trial duration and data --- fishbook/frontend/util.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) 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):