add getters for vs and sc

This commit is contained in:
a.ott 2020-02-14 14:30:37 +01:00
parent 9ccf46f3c3
commit d0d42406c8

View File

@ -141,6 +141,33 @@ class CellData:
def get_after_stimulus_duration(self) -> float: def get_after_stimulus_duration(self) -> float:
return self.recording_times[3] 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): def __read_fi_spiketimes_info__(self):
if self.fi_spiketimes is None: if self.fi_spiketimes is None:
trans_amplitudes, intensities, spiketimes = self.parser.get_fi_curve_spiketimes() trans_amplitudes, intensities, spiketimes = self.parser.get_fi_curve_spiketimes()