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