From e951d2df039e5f186e64260ec0e5d28ed839cd09 Mon Sep 17 00:00:00 2001 From: Jan Grewe Date: Mon, 12 Nov 2018 13:05:03 +0100 Subject: [PATCH] fixes and new read_baseline_spikes, which does not work on broken files... --- code/read_baseline_data.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/code/read_baseline_data.py b/code/read_baseline_data.py index eca0157..71934e2 100644 --- a/code/read_baseline_data.py +++ b/code/read_baseline_data.py @@ -4,20 +4,31 @@ from IPython import embed def read_baseline_eod(dataset): base = dataset.split(os.path.sep)[-1] + ".nix" - nix_file = nix.File.open(os.path.join(dataset, base)) + nix_file = nix.File.open(os.path.join(dataset, base), nix.FileMode.ReadOnly) b = nix_file.blocks[0] - for t in b.tags(): - if "baseline" in t.name: - break; + t = b.tags["BaselineActivity_1"] eod_da = b.data_arrays["LocalEOD-1"] eod = t.retrieve_data("LocalEOD-1") time = eod_da.dimensions[0].axis(len(eod)) nix_file.close() return time, eod + + +def read_baseline_spikes(dataset): + base = dataset.split(os.path.sep)[-1] + ".nix" + nix_file = nix.File.open(os.path.join(dataset, base), nix.FileMode.ReadOnly) + b = nix_file.blocks[0] + t = b.tags["BaselineActivity_1"] + embed() + spike_times = t.retrieve_data("Spikes-1") + + nix_file.close() + return spike_times + if __name__ == "__main__": data_dir = "../data" dataset = "2018-11-09-ad-invivo-1" time, eod = read_baseline_eod(os.path.join(data_dir, dataset)) - + spike_times = read_baseline_spikes(os.path.join(data_dir, dataset))