Exemplo n.º 1
0
import os
#import princax
from matplotlib.patches import Ellipse
import lagrangian_stats

exp = 'm_25_2_23D_particles'

try: os.stat('./plot/'+exp)
except OSError: os.mkdir('./plot/'+exp)

exp2D = 'm_25_2_2D_particles'

filename0 = './mli_checkpoint.detectors'
filename0 = '/tamay2/mensa/fluidity/'+exp2D+'/'+filename0

time2D, par2D = lagrangian_stats.read_particles(filename0)

exp3D = 'm_25_2_3D_particle'

filename0 = './mli_checkpoint.detectors'
filename0 = '/tamay2/mensa/fluidity/'+exp3D+'/'+filename0

time3D, par3D = lagrangian_stats.read_particles(filename0)

tt = min(len(time2D),len(time3D))

par3D = par3D[:,:,:tt]
par2D = par2D[:,:,:tt]

time = time2D[:tt]
par0 = lagrangian_stats.periodicCoords(par0,10000,4000)

time0 = (timet)*1200 + 48*3600 - 1200

# read online
print 'reading online'
exp = 'm_25_2_512'

try: os.stat('./plot/'+exp)
except OSError: os.mkdir('./plot/'+exp)

filename0 = './mli_tracer.detectors'
filename0 = '/tamay2/mensa/fluidity/'+exp+'/'+filename0

time1, parT = lagrangian_stats.read_particles(filename0)

dt = 1200
pt = len(parT)
timei = np.asarray(range(0,tt,1))*dt + 86400*2.0 + dt
par1 = np.zeros((pt,3,len(timei)))

for p in range(len(par1)):
 f0 = interp1d(time1,parT[p,0,:])
 f1 = interp1d(time1,parT[p,1,:])
 f2 = interp1d(time1,parT[p,2,:])
 par1[p,0,:] = f0(timei)
 par1[p,1,:] = f1(timei)
 par1[p,2,:] = f2(timei)

time = timei
import lagrangian_stats

exp = 'r_1k_B_1F1_particles'

filename0 = './ring_checkpoint_checkpoint.detectors'
filename1 = './ring_checkpoint_checkpoint_checkpoint.detectors'

filename0 = '/tamay2/mensa/fluidity/'+exp+'/'+filename0
filename1 = '/tamay2/mensa/fluidity/'+exp+'/'+filename1

print 'Reading ', filename1 , filename0 

#try: os.stat('./output/'+exp3D)
#except OSError: os.mkdir('./output/'+exp3D)

time0, par0 = lagrangian_stats.read_particles(filename0)
time1, par1 = lagrangian_stats.read_particles(filename1)

time = np.hstack((time0[time0<time1[0]],time1))
par = np.concatenate((par0[:,:,time0<time1[0]], par1), axis=2)

tt = len(time)

# horizontal

Zlist = [-10,-50,-100,-200,-500,-800]

for z in range(0,len(Zlist),2): 
 print 'depth', z
 par2Dz = par[z::6,:,:] 
 #
import myfun
import numpy as np
import os
#import princax
from matplotlib.patches import Ellipse
import lagrangian_stats

exp = 'm_25_1_particles'

try: os.stat('./plot/'+exp)
except OSError: os.mkdir('./plot/'+exp)

filename0 = './mli_checkpoint.detectors'
filename0 = '/tamay2/mensa/fluidity/'+exp+'/'+filename0

time, par3D = lagrangian_stats.read_particles(filename0)

tt = len(time)

# horizontal
depths = [1, 5, 11, 17, 26]
depths = [1, 5, 17]
depths = [1] #, 17, 1]

# periodicity

#par3DP = lagrangian_stats.periodicCoords(par3D,2000,2000)

# interpolate to same timeseries as offline advection
from scipy.interpolate import interp1d
dt = 600