jgrewe/fishbook
2
1
Fork 1
Code Issues 3 Pull Requests Releases Wiki Activity

[util] Stimspikes stored trial duration and data

Browse Source
This commit is contained in:
Jan Grewe 2020-08-13 15:53:56 +02:00
parent 8844825c21
commit 0881947fd2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
fishbook/frontend/util.py
View File

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