from utils_suseptibility import cells_eigen
from utils_all import p_units_to_show
#from utils_all import load_folder_name
#from plt_RAM import plt_punit
from IPython import embed

if __name__ == '__main__':

    #'2010-06-21-ac-invivo-1'
    # Bei dem ampullary cells machen wir auch die Bursts weg
    ['2010-11-26-al-invivo-1', '2010-11-26-am-invivo-1',
     '2010-11-08-ab-invivo-1', '2010-11-08-ag-invivo-1', '2010-11-08-ah-invivo-1', '2010-11-08-ai-invivo-1',
     '2010-11-11-ak-invivo-1', '2010-11-11-ac-invovo-1',
     '2013-04-11-ab-invivo-1',
     ]
    cells = ['2010-11-26-al-invivo-1', '2010-11-26-am-invivo-1'
     ]

    # 10-08-11-ar
    # 5 + 10+20

    # 0.5, 1, 5, 10, 20

    # 2.5 5 10

    # 5 10 20 (2012-06-21-av, 2010-08-27-ag)
    # 0.5, 1, 5 (2021-08-03-ab)
    # 0.1, 0.5, 1, 5 (2021-08-03-ac)
    # 1 5 10 20, (2020-10-21-ac)
    # 0.1,1,10 (2021-12-17-ad)
    # 0.1, 0.5, 1, 5, 10 ,20 (2020-10-27-af)
    # 1, 2, 5, 10, 20 (2021-06-18-ae)

    unklar = ['2010-07-29-ae-invivo-1']
    # für mehrere

    # ampullary
    ## 2.5, 5, 10 (2011-10-25-ab, 2012-07-11-ag, 2012-12-13-ao)
    # amp_desired = [2.5, 10, 20]# '2010-08-31-ad-invivo-1',
    # cells_plot2 = ['2011-10-25-ac-invivo-1', '2012-05-15-ac-invivo-1','2012-06-08-ae-invivo-1']#['2010-06-21-av-invivo-1', '2010-08-27-ag-invivo-1','2020-10-27-ae-invivo-1']

    amp_desired = [1, 5, 10, 20]
    amp_desired = [0.1, 1, 10]
    amp_desired = [0.5, 1, 5]
    amp_desired = [0.5, 1, 5],

    # 0.1, 0.5, 1, 5 (2021-08-03-ac,2021-08-03-ae)
    amp_desired = [0.1, 0.5, 1, 5]
    cells_plot2 = []

    # 0.5, 1, 5 (2021-08-03-ab)
    amp_desired = [0.5, 1, 5]
    cells_plot2 = ['2021-08-03-ab-invivo-1','2021-08-03-ac-invivo-1','2021-08-03-ae-invivo-1']

    # 1 5 10 20, (2020-10-21-ac)
    amp_desired = [1,5,10,20]
    cells_plot2 = []
    # 0.1,1,10 (2021-12-17-ad)
    # 0.1, 0.5, 1, 5, 10 ,20 (2020-10-20-ab, 2020-10-27-af)
    #amp_desired = [1,5,10,20]
    #cells_plot2 = []

    # das hat uns glaube ich ganz gut gefallen
    # 5 10 20 (2010-06-21-av, 2010-08-27-ag, 2010-08-31-ad)
    amp_desired = [5,20]# '2010-08-31-ad-invivo-1', 10,
    cells_plot2 = p_units_to_show(type_here = 'eigen_small')
    # I like: '2020-10-21-ac' '2010-06-21-av-invivo-1', '2010-08-27-ag-invivo-1' '2021-08-03-ag-invivo-1, '20-10-21-aa-invivo-1
    #2021-12-17-ad
    #embed()
    #'2017-10-25-am-invivo-1', '2010-11-26-an-invivo-1'
    #save_names = ['noise_data10_nfft1sec_original__StimPreSaved4__mean5__CutatBeginning_0.05_s_NeurDelay_0.005_s_spikes_']
    #load_name = load_folder_name('calc_RAM') + '/' + save_names[0] + '_' +'2017-10-25-am-invivo-1'# '2010-11-26-an-invivo-1'#'2017-10-25-am-invivo-1'
    #stack = pd.read_pickle(load_name + '.pkl')
    #embed()#[0, 250]#'High CV P-unit', at Fr \n at 1/2 EODf'
    cells_eigen(titles = ['Low-CV P-unit,','Low-CV P-unit,'], cells_plot2 = cells_plot2, amp_desired = amp_desired, show = True,cell_class = ' P-unit')