import numpy as np
from CellData import CellData
from DataParserFactory import DatParser
import matplotlib.pyplot as plt

test_cell = "data/final/2010-11-08-al-invivo-1/"


def main():
    parser = DatParser(test_cell)
    fi_trans_amplitudes, fi_intensities, fi_spiketimes = parser.get_fi_curve_spiketimes()
    for time, v1, eod, local_eod, stim in parser.__iget_traces__("FICurve"):
        del eod
        del local_eod
        del stim

        for i in range(len(fi_spiketimes)):
            for j, spikes in enumerate(fi_spiketimes[i]):
                offset = max(v1)+j

                plt.eventplot(np.array(spikes)+2*j, colors="black", lineoffsets=offset)  # , linelengths=0.5, lineoffsets=1+i*1+max(v1))
            plt.plot(np.array(time), v1)
            plt.show()

    pass




if __name__ == '__main__':
    main()