P-unit_model/test.py
2021-01-30 15:53:18 +01:00

38 lines
1.2 KiB
Python

import os
from parser.CellData import CellData
# from plottools.axes import labelaxes_params
import matplotlib.pyplot as plt
data_folder = "./data/final/"
for cell in sorted(os.listdir(data_folder)):
print(cell)
cell_folder = os.path.join(data_folder, cell)
if not os.path.exists(os.path.join(cell_folder, "samspikes1.dat")):
continue
cell_data = CellData(cell_folder)
sam_spikes = cell_data.get_sam_spiketimes()
delta_freqs = cell_data.get_sam_delta_frequencies()
# [time_traces, v1_traces, eod_traces, local_eod_traces, stimulus_traces]
[time_traces, v1_traces, eod_traces, local_eod_traces, stimulus_traces] = cell_data.get_sam_traces()
print(len(time_traces))
# for i in range(len(delta_freqs)):
#
# fig, axes = plt.subplots(2, 1, sharex="all")
#
# axes[0].plot(time_traces[i], local_eod_traces[i])
# axes[0].set_title("Local EOD - dF {}".format(delta_freqs[i]))
# axes[1].plot(time_traces[i], v1_traces[i])
# axes[1].set_title("v1 trace")
# axes[1].eventplot(sam_spikes[i], lineoffsets=max(v1_traces[i]))
# plt.show()
# plt.close()
# break