import neo
import numpy as np
import quantities as pq
import matplotlib.pyplot as plt
from elephant.spike_train_generation import homogeneous_poisson_process
from viziphant.rasterplot import rasterplot
from viziphant.events import add_event

np.random.seed(7)
spiketrains = [homogeneous_poisson_process(rate=r * pq.Hz,
               t_stop=10 * pq.s) for r in range(1, 21)]
event = neo.Event([0.5, 2.8] * pq.s, labels=['Trig ON', 'Trig OFF'])

axes = rasterplot(spiketrains, histogram_bins=50, title='Title', s=0.5)
add_event(axes, event=event)
plt.show()