[util] Stimspikes stored trial duration and data

This commit is contained in:
Jan Grewe 2020-08-13 15:53:56 +02:00
parent 8844825c21
commit 0881947fd2

View File

@ -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):