mld_10 = np.zeros(len(range(dayi,dayf,days))) for t in range(len(time)): print 'time:', time[t] tlabel = str(time[t]) while len(tlabel) < 3: tlabel = '0'+tlabel #Velocity_CG_m_50_6e_9.csv file0_50 = path+'Velocity_CG_0_'+label_50+'_'+str(time[t])+'.csv' file0_25 = path+'Velocity_CG_0_'+label_25+'_'+str(time[t])+'.csv' file0_10 = path+'Velocity_CG_0_'+label_10+'_'+str(time[t])+'.csv' file1 = 'drate_'+label+'_'+str(time[t]) file1_50 = 'drate_'+label_50 file1_25 = 'drate_'+label_25 file1_10 = 'drate_'+label_10 W_50 = lagrangian_stats.read_Scalar(file0_50,zn,yn_50,xn_50) W_25 = lagrangian_stats.read_Scalar(file0_25,zn,yn_25,xn_25) W_10 = lagrangian_stats.read_Scalar(file0_10,zn,yn_10,xn_10) for i in range(len(Xlist_50)): for j in range(len(Ylist_50)): FW_50[j,i,:,t] = (np.gradient(W_50[:,j,i])/dz_50)**2 for i in range(len(Xlist_25)): for j in range(len(Ylist_25)): FW_25[j,i,:,t] = (np.gradient(W_25[:,j,i])/dz_25)**2 for i in range(len(Xlist_10)): for j in range(len(Ylist_10)): FW_10[j,i,:,t] = (np.gradient(W_10[:,j,i])/dz_10)**2 plt.figure(figsize=(4,8))
v = np.zeros((len(Zlist),2)) #for time in range(dayi,dayf,days): for file in files: time = int(file) tlabel = str(time) while len(tlabel) < 3: tlabel = '0'+tlabel # file0 = basename + '_' + str(time) + '.pvtu' filepath = path+file0 file1 = label+'_' + tlabel fileout = path + file1 # print 'opening: ', filepath # Rho = lagrangian_stats.read_Scalar('/nethome/jmensa/scripts_fluidity/2D/RST/Density_CG/Density_CG_'+label+'_'+str(time),zn,xn,yn,[time]) # # for d in range(9,len(Zlist),100): if file == files[0]: v[d,:] = np.nanmin(Rho[d,:,:,0]), np.nanmax(Rho[d,:,:,0]) fig = plt.figure(figsize=(7, 6)) plt.contourf(Xlist/1000.0,Ylist/1000.0,Rho[d,:,:,0],np.linspace(v[d,0],v[d,1],50,endpoint=True),extend='both',cmap=plt.cm.PiYG) plt.colorbar(ticks=np.linspace(v[d,0],v[d,1],5,endpoint=True)) plt.axis('equal') plt.xlabel('X (Km)',fontsize=16) plt.ylabel('Y (Km)',fontsize=16) plt.title('day '+str(time/4.0),fontsize=16) plt.savefig('./plot/'+label+'/R_'+file1+'_'+str(Zlist[d])+'_h_snap.eps') print './plot/'+label+'/R_'+file1+'_'+str(Zlist[d])+'_h_snap.eps' plt.close()
zn = len(Zlist) dx = np.diff(Xlist) for time in range(dayi,dayf,days): print 'time:', time tlabel = str(time) while len(tlabel) < 3: tlabel = '0'+tlabel #Velocity_CG_m_6e_9.csv file0_B = path+'Velocity_CG_2_'+label_B+'_'+str(time)+'.csv' file0_BW = path+'Velocity_CG_2_'+label_BW+'_'+str(time)+'.csv' file1 = 'Velocity_CG_2_'+label+'_'+str(time) file1 = 'Velocity_CG_2_'+label+'_'+str(time) # T_B = lagrangian_stats.read_Scalar(file0_B,zn,xn,yn) T_BW = lagrangian_stats.read_Scalar(file0_BW,zn,xn,yn) FT_B = np.zeros((xn/1,yn)) FT_BW = np.zeros((xn/1,yn)) # for k in range(1): for j in range(0,len(Ylist),100): print j plt.plot(T_B[k,j,:]-np.mean(T_B[k,j,:])) plt.savefig('./plot/'+label+'/'+file1+'_'+str(Zlist[k])+'_'+str(j)+'_sec.eps',bbox_inches='tight') plt.close() print './plot/'+label+'/'+file1+'_'+str(Zlist[k])+'_'+str(j)+'_sec.eps'
for time in range(dayi,dayf,days): tlabel = str(time) while len(tlabel) < 3: tlabel = '0'+tlabel file0u = path+'Velocity_CG_0_'+label+'_'+str(time)+'.csv' file0v = path+'Velocity_CG_1_'+label+'_'+str(time)+'.csv' file1 = 'Velocity_CG_'+label+'_'+str(time) print file1 # # xn_50 = 101 # yn_50 = 101 # xn = 101 # yn = 101 u = lagrangian_stats.read_Scalar(file0u,zn,xn,yn) v = lagrangian_stats.read_Scalar(file0v,zn,xn,yn) # # u = np.squeeze(np.reshape(Vel[:,0],[len(Zlist),len(Xlist),len(Ylist)])) # v = np.squeeze(np.reshape(Vel[:,1],[len(Zlist),len(Xlist),len(Ylist)])) for k in range(len(Zlist)): # u = np.mean(u,0) # v = np.mean(v,0) uk = u[k,:,:] vk = v[k,:,:] # dt = 3600 # s dx = np.mean(np.mean(uk,0),0)*dt
xn = len(Xlist) yn = len(Ylist) zn = len(Zlist) dx = np.diff(Xlist) z = 1 for time in range(dayi,dayf,days): print 'time:', time tlabel = str(time) while len(tlabel) < 3: tlabel = '0'+tlabel #Velocity_CG_m_50_6e_9.csv file0 = path+'Tracer_'+str(z)+'_CG_'+label+'_'+str(time)+'.csv' # T = lagrangian_stats.read_Scalar(file0,zn,xn,yn) T = np.sum(T,0)/3. FT = np.zeros((xn/1,yn)) # for j in range(len(Ylist)): tempfft = scipy.fftpack.fft(T[:,j]**2,xn) FT[:,j] = abs(tempfft)**2 w = scipy.fftpack.fftfreq(xn, dx[1]) # w = scipy.fftpack.fftshift(w) FTp = np.mean(FT,1)/xn # ideal t=0 Theory = T*0 + 3 for j in range(len(Ylist)): tempfft = scipy.fftpack.fft(Theory[:,j]**2,xn)
Ylist = np.linspace(0,2000,yn)# y co-ordinates of the desired array shape [X,Y] = np.meshgrid(Xlist,Ylist) dl = [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1] Zlist = np.cumsum(dl) time = range(dayi,dayf,days) t = 0 for tt in time: tlabel = str(tt) while len(tlabel) < 3: tlabel = '0'+tlabel # file1 = basename+'_'+label+'_' + tlabel + '.csv' print file1 f = open(file1,'wr') writer = csv.writer(f) # Temp = lagrangian_stats.read_Scalar('./RST/'+basename+'/'+basename+'_'+label+'_'+str(tt)+'.csv',zn,xn,yn) # for k in range(len(Zlist)): # determine speed direction # for i in range(len(Xlist)): for j in range(len(Ylist)): writer.writerow([Xlist[i],Ylist[j],Zlist[k],Temp[k,i,j]]) f.close()
Zlist = -1*np.cumsum(depths) Xlist = np.linspace(-150000,150000,301) Ylist = np.linspace(-150000,150000,301) [X,Y] = np.meshgrid(Xlist,Ylist) time = range(dayi,dayf,days) t = 0 for tt in time: tlabel = str(tt) while len(tlabel) < 3: tlabel = '0'+tlabel # file1 = label+'_' + tlabel # Rho = lagrangian_stats.read_Scalar('./Density_CG/Density_CG_'+label+'_'+str(tt),zn,xn,yn,[tt]) # for z in range(5,zn,25): # determine speed direction # plt.figure() plt.contourf(Xlist,Ylist,Rho[z,:,:,0],50,extend='both',cmap=plt.cm.PiYG) plt.colorbar() plt.ylabel('Y [Km]',fontsize=18) plt.xlabel('X [Km]',fontsize=18) plt.savefig('./plot/'+label+'/T_'+str(z)+'_'+file1+'.eps') plt.close() print './plot/'+label+'/T_'+str(z)+'_'+file1+'.eps'