import ArenaBehaviorAnalysis as aba
import pylab
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as pyplot
import pylab
import scipy
import os

e_safe5_first = aba.loadDataFromFile(
    os.path.expanduser("~/Dropbox/ConchisData/2013-05-16/f00438/f00438_2013-05-16-14-13-03.json")
)
e_shock5 = aba.loadDataFromFile(
    os.path.expanduser("~/Dropbox/ConchisData/2013-05-16/f00438/f00438_2013-05-16-14-44-45.json")
)
e_safe5_sec = aba.loadDataFromFile(
    os.path.expanduser("~/Dropbox/ConchisData/2013-05-16/f00438/f00438_2013-05-16-15-39-01.json")
)

e_safe_same = aba.loadDataFromFile(
    os.path.expanduser("~/Dropbox/ConchisData/2013-05-07/f00411/f00411_2013-05-07-14-15-17.json")
)  # ,bAcqArenaPoly=True)
e_shock_same = aba.loadDataFromFile(
    os.path.expanduser("~/Dropbox/ConchisData/2013-05-07/f00411/f00411_2013-05-07-14-46-43.json")
)
e_safe_same_sec = aba.loadDataFromFile(
    os.path.expanduser("~/Dropbox/ConchisData/2013-05-07/f00411/f00411_2013-05-07-15-40-02.json")
)

mpl.rcParams.update({"font.size": 18})
import ArenaBehaviorAnalysis as aba
import pylab
import numpy as np

e_fish_youngNR = [
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00134/f00134_2013-02-12-17-57-45.json",
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00135/f00135_2013-02-12-17-57-30.json",
]
e_fish_youngNR = aba.loadMultipleDataFiles(e_fish_youngNR)

e_fish_youngR = "/home/vburns/Dropbox/ConchisData/2013-02-12/f00136/f00136_2013-02-12-17-57-33.json"
e_fish_youngR = aba.loadDataFromFile(e_fish_youngR)
e_fish_youngR = [e_fish_youngR]

e_fish_oldNR = [
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00138/f00138_2013-02-12-17-57-37.json",
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00139/f00139_2013-02-12-17-57-40.json",
]
e_fish_oldNR = aba.loadMultipleDataFiles(e_fish_oldNR)

e_fish_oldR = [
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00140/f00140_2013-02-12-17-57-48.json",
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00141/f00141_2013-02-12-17-57-51.json",
    "/home/vburns/Dropbox/ConchisData/2013-02-12/f00142/f00142_2013-02-12-17-57-53.json",
]
e_fish_oldR = aba.loadMultipleDataFiles(e_fish_oldR)

# assumes shock is on red
(e_frac_yNR, e_dist_yNR) = aba.getSidePreference_Multi(e_fish_youngNR, cond=[8, 8], refState="Red")
(e_frac_yR, e_dist_yR) = aba.getSidePreference_Multi(e_fish_youngR, cond=[8, 8], refState="Red")
(e_frac_oNR, e_dist_oNR) = aba.getSidePreference_Multi(e_fish_oldNR, cond=[8, 8], refState="Red")
import ArenaBehaviorAnalysis as aba
import pylab
import numpy as np

#f is five day old fish, t is thirteen day old fish
f_fish = ['/home/vburns/Dropbox/ConchisData/2013-01-09/f00056/f00056_2013-01-09-15-15-30.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-09/f00057/f00057_2013-01-09-15-15-28.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-09/f00058/f00058_2013-01-09-15-15-26.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-30/f00113/f00113_2013-01-30-10-58-49.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-30/f00014/f00014_2013-01-30-10-58-47.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-30/f00115/f00115_2013-01-30-10-58-45.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-30/f00116/f00116_2013-01-30-10-58-42.json']
f_fish = aba.loadMultipleDataFiles(f_fish)

fb = '/home/vburns/Dropbox/ConchisData/2013-01-09/f00059/f00059_2013-01-09-15-15-24.json'
fb = aba.loadDataFromFile(fb)
fb['warpedTracking'][:,1] = 48-fb['warpedTracking'][:,1]

f_fish = f_fish + [fb]

t_fish = ['/home/vburns/Dropbox/ConchisData/2013-01-09/f00061/f00061_2013-01-09-14-30-06.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-09/f00062/f00062_2013-01-09-14-30-08.json',
 '/home/vburns/Dropbox/ConchisData/2013-01-09/f00063/f00063_2013-01-09-14-30-10.json']
t_fish = aba.loadMultipleDataFiles(t_fish)

tb = '/home/vburns/Dropbox/ConchisData/2013-01-09/f00060/f00060_2013-01-09-14-30-04.json'
tb = aba.loadDataFromFile(tb)
tb['warpedTracking'][:,1] = 48-tb['warpedTracking'][:,1]

t_fish = t_fish + [tb]