import csv

# read offline
print "reading offline"

# exp = 'm_25_1_tracer'
# label = 'm_25_1_tracer'
# filename2D = 'traj_m_25_1_tracer_0_640_2D.csv'
# filename3D = 'traj_m_25_1_tracer_0_640_3D.csv'
# tt = 640 # IC + 24-48 included

label = "m_25_1b_particles"
dim = "2D"
filename2D_B = "./csv/RD_" + dim + "_" + label + ".csv"

time2D_B, RD_2D_B = lagrangian_stats.read_dispersion(filename2D_B)

time = time2D_B[:] - time2D_B[0]

depths = [5, 10, 15]
depthid = [1, 2, 3]

nl = len(depths)

# relative D
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

fig, ax = plt.subplots()

RD_2D = np.log10(RD_2D_B)
# read offline
print 'reading offline'

#exp = 'm_25_1_tracer'
#label = 'm_25_1_tracer'
#filename2D = 'traj_m_25_1_tracer_0_640_2D.csv'
#filename3D = 'traj_m_25_1_tracer_0_640_3D.csv'
#tt = 640 # IC + 24-48 included

label_B = 'm_25_1b_particles'
label_BW = 'm_25_2b_particles'
label = 'm_25_2b_particles'
filename_B = './csv/CDv_'+label_B+'.csv'
filename_BW = './csv/CDv_'+label_BW+'.csv'

time_B, RD_B = lagrangian_stats.read_dispersion(filename_B)
time_BW, RD_BW = lagrangian_stats.read_dispersion(filename_BW)

time = time_B[:]  - time_B[0]
time = time_BW[:]  - time_BW[0]

depths = [5, 10, 15] 
depthid = [1, 2, 3] 

nl = len(depths)

# relative D
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

fig, ax = plt.subplots()
days = 1

z0 = [0, 5, 10, 15]

depths = [1, 2, 3]

nl = len(depths)

CD_2D = np.zeros((tt,nl))
CD_3D = np.zeros((tt,nl))

vi = 10
vf = -1
vfp = 500

time2D, RD_2D =lagrangian_stats.read_dispersion(filename2D)
time3D, RD_3D =lagrangian_stats.read_dispersion(filename3D)

CD_2D[:,:] = 100**2*RD_2D
CD_3D[:,:] = 100**2*RD_3D

time0 = time2D - time2D[0] #+ 5*3600
time = time0[vi:vf]
timep = time0[vi:vfp]

xm = 2*10**3
xM = 5*10**6
ym = 10**1
yM = 10**6

OKx = np.linspace(xm,xM)
#exp = 'm_25_1_tracer'
#label = 'm_25_1_tracer'
#filename2D = 'traj_m_25_1_tracer_0_640_2D.csv'
#filename3D = 'traj_m_25_1_tracer_0_640_3D.csv'
#tt = 640 # IC + 24-48 included

label = 'm_25_2b_particles'
label_B = 'm_25_1b_particles'
label_BW = 'm_25_2b_particles'
filename2D_BW = './csv/CDv_'+label_BW+'.csv'
filename2D_B = './csv/CDv_'+label_B+'.csv'

tt = 500-61

time2D_B, CD_2D_B =lagrangian_stats.read_dispersion(filename2D_B)
time2D_BW, CD_2D_BW =lagrangian_stats.read_dispersion(filename2D_BW)

time = time2D_B[:]
time = np.asarray(range(tt))*1440

depths = [5, 10, 15] 
depthid = [1, 2, 3] 

nl = len(depths)


# relative D

#CD_2D_B = np.log10(CD_2D_B)
#CD_2D_B = CD_2D_B[1:]