From d0d42406c8450d4c9722bbddb546f52b9519ac99 Mon Sep 17 00:00:00 2001 From: "a.ott" Date: Fri, 14 Feb 2020 14:30:37 +0100 Subject: [PATCH] add getters for vs and sc --- CellData.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/CellData.py b/CellData.py index 45a6d60..40f0725 100644 --- a/CellData.py +++ b/CellData.py @@ -141,6 +141,33 @@ class CellData: def get_after_stimulus_duration(self) -> float: return self.recording_times[3] + def get_vector_strength(self): + times = self.get_base_traces(self.TIME) + eods = self.get_base_traces(self.EOD) + v1_traces = self.get_base_traces(self.V1) + return hf.calculate_vector_strength(times, eods, v1_traces) + + def get_serial_correlation(self, max_lag): + serial_cors = [] + for spiketimes in self.get_base_spikes(): + sc = hf.calculate_serial_correlation(spiketimes, max_lag) + print(sc) + serial_cors.append(sc) + serial_cors = np.array(serial_cors) + mean_sc = np.mean(serial_cors, axis=0) + + return mean_sc + + def get_eod_frequency(self): + eods = self.get_base_traces(self.EOD) + sampling_interval = self.get_sampling_interval() + frequencies = [] + for eod in eods: + time = np.arange(0, len(eod)*sampling_interval, sampling_interval) + frequencies.append(hf.calculate_eod_frequency(time, eod)) + + return np.mean(frequencies) + def __read_fi_spiketimes_info__(self): if self.fi_spiketimes is None: trans_amplitudes, intensities, spiketimes = self.parser.get_fi_curve_spiketimes()