import matplotlib.pyplot as plt
import numpy as np
from read_chirp_data import *
from IPython import embed

data_dir = "../data"
dataset = "2018-11-09-ad-invivo-1"

spikes = read_chirp_spikes(os.path.join(data_dir, dataset))
eod = read_chirp_eod(os.path.join(data_dir, dataset))
times = read_chirp_times(os.path.join(data_dir, dataset))

df_map = {}
for k in spikes.keys():
    df = k[1]
    if df in df_map.keys():
        df_map[df].append(k)
    else:
        df_map[df] = [k]

#plt.scatter(spikes[0, '-50Hz', '20%', '100Hz'][0, 0.614])
#plt.show()