def waveforms_1(self, channel_group):
action_1 = self._actions[self.action_id_1]
data_path_1 = get_data_path(action_1)
spike_trains_1 = load_spiketrains(data_path_1,
channel_group=channel_group,
load_waveforms=True)
return [np.array(sptr.waveforms) for sptr in spike_trains_1]
def load_waveforms(self, action_id, unit_id, channel_group):
action = self._actions[action_id]
data_path = get_data_path(action)
spike_trains = load_spiketrains(
data_path, channel_group=channel_group, load_waveforms=True)
wfs = [np.array(sptr.waveforms) for sptr in spike_trains if get_unit_id(sptr)==unit_id]
if len(wfs) != 1:
raise ValueError(
f'Unable to load waveforms from unit {unit_id}' +
f' in {action_id} channel group {channel_group}')
return wfs[0]