Wave_analysis.py

import matplotlib.pyplot as plt
import neo
import numpy as np


class WaveAnalysis:

    def __init__(self, file):
        reader = neo.Spike2IO(filename=file).read()[0]  # 1 block, 1 segment
        segment = reader.segments[0]
        self.signals = segment.analogsignals
        self.event_times = segment.events[0].times

    def signal_spikes(self, signal_index):
        x = []
        y = []
        print(f"{self.signals[signal_index].name}")
        for i in range(len(self.event_times)):
            tup = self._spike_at(signal_index, i)
            x.append(tup[0])
            y.append(tup[1])

        plt.plot(x, y)
        plt.xlabel('Trigger Times', fontsize=14)
        plt.ylabel('Response', fontsize=14)
        plt.title(self.signals[signal_index].name, fontsize=18)
        plt.grid(True)
        plt.show()
        # plt.hold(True)
        return x, y


    def _spike_at(self, signal_index, event_idx, delta=2e-03) -> tuple:
        """
        :param signal_index: What signal?
        :param event_idx: Index into the event time
        :param delta:
        :return:
        """
        x = self.signals[signal_index].times    # same value for all the channels
        y = self.signals[signal_index]

        time = self.event_times[event_idx].magnitude

        left = time  # lower bound from event.times
        right = time + delta  # upper bound from event.times

        # print(time)
        arr = [(float(x[idx].magnitude), float(y[idx].magnitude)) for idx in
               np.where(np.logical_and(x >= left, x <= right))[0]]
        print(arr)
        # print([abs(time - x) for x, y in arr])

        coordinate = None
        diff = float("inf")
        for x, y in arr:
            if abs(time - x) < diff:
                diff = abs(time - x)
                coordinate = (x, y)
        return coordinate


if __name__ == '__main__':
    analysis = WaveAnalysis("File location")
    for i in range(1, 5):       # 4-channels
        analysis.signal_spikes(i)