Initital git repo forked of SVN revision 5235

Author: Richard Hofmeister

Plot/plot_bdyvelocity.py

@@ -0,0 +1,21 @@
+from pylab import *
+import netCDF4
+
+nc = netCDF4.Dataset('outputs/1_hvel.nc')
+ncv=nc.variables
+
+ul = ncv['u'][:,-1,22]
+ur = ncv['u'][:,-1,42]
+
+time = ncv['time'][:]/86400.
+
+figure(figsize=(10,3))
+
+plot(time,ul,'k-',lw=2.0,label='left')
+plot(time,ur,'-',lw=2.0,color='orange',label='right')
+ylabel('[m/s]')
+xlabel('days')
+legend()
+
+show()
+nc.close()

Plot/plot_salinities.py

@@ -0,0 +1,47 @@
+from pylab import *
+import netCDF4
+
+runs = {}
+
+runs['upwind'] = 'o001'
+runs['tvd'] = 'o002'
+runs['tvd 0.1*dt'] = 'o003'
+runs['tvd 10*dt'] = 'outputs'
+
+lss = ['-','--',':','-.']
+
+fig = figure(figsize=(8,5))
+
+for label,folder in runs.iteritems():
+
+  nc = netCDF4.Dataset('%s/1_salt.nc'%folder)
+  ncv=nc.variables
+
+  t = ncv['time'][:]
+  y = ncv['y'][:]
+  yidx = where(y==0.0)
+  s = ncv['salt'][-1,-1][yidx]-35.
+  x = ncv['x'][yidx]/1000.
+
+  
+  if label=='upwind':
+    col = 'r'
+    ls = '-'
+    s0 = ncv['salt'][0,-1][yidx]-35.
+    plot(x,s0,'-',lw=3.0,color=(0.6,0.6,0.6),label='initial condition')
+  else:
+    col = 'k'
+    ls = lss.pop()
+    
+
+  plot(x,s,ls=ls,color=col,lw=2.0,label=label)
+
+  nc.close()
+
+ylim(-0.1,1.1)
+ylabel('Tracer [1/1]')
+xlabel('km')
+legend(loc='upper left',frameon=False)
+
+savefig('transport_comparison.pdf')
+show()

Plot/plot_timesteps.py

@@ -0,0 +1,116 @@
+from pylab import *
+import netCDF4
+import sys
+
+runs = {}
+if len(sys.argv)>1:
+  name = sys.argv[1]
+else:
+  name='test'
+
+if False:
+ runs[5] = name+'_5s'
+ runs[10] = name+'_10s'
+ runs[25] = name+'_25s'
+ runs[37] = name+'_37s'
+ runs[50] = name+'_50s'
+ runs[100] = name+'_100s'
+ runs[200] = name+'_200s'
+else:
+ runs[5] = name+'_5s'
+ runs[10] = name+'_10s'
+ runs[15] = name+'_15s'
+ runs[20] = name+'_20s'
+ runs[40] = name+'_40s'
+ runs[80] = name+'_80s'
+ runs[160] = name+'_160s'
+
+refname='tvd_20m'
+
+lss = ['-','--',':','-.','-..']
+tidx = 477
+
+fig = figure(figsize=(8,5))
+tax=axes()
+vf = figure(figsize=(8,5))
+vax=axes()
+
+keys = sorted(runs)
+
+for dt in keys:
+  folder = runs[dt]
+  cfl = dt*10./200.
+  #label = '%d s'%dt
+  label = '%0.1f'%cfl
+  print(folder)
+  nc = netCDF4.Dataset('%s/outputs/1_salt.nc'%folder)
+  ncv=nc.variables
+
+  t = ncv['time'][:]
+  y = ncv['y'][:]
+  yidx = where(y==0.0)
+  s = ncv['salt'][tidx,-1][yidx]-35.
+  x = ncv['x'][yidx]/1000.
+
+  vnc = netCDF4.Dataset('%s/outputs/1_hvel.nc'%folder)
+  vncv = vnc.variables
+  xidx=21+11
+  v = vncv['u'][-25:,-1,xidx]
+  vt = vncv['time'][-25:]
+  hours = (vt-vt[0])/3600.
+  
+  if label=='50 s':
+    col = 'r'
+    ls = '-'
+    s0 = ncv['salt'][0,-1][yidx]-35.
+    tax.plot(x,s0,'-',lw=3.0,color='orange',label='initial condition')
+  else:
+    col = (log(int(dt))/log(5)/3.-0.3)*ones((3))
+    ls = '-'#lss.pop()
+    
+
+  tax.plot(x,s,ls=ls,color=col,lw=2.0,label=label)
+  vax.plot(hours,v,ls=ls,color=col,lw=2.0,label=label)
+
+  nc.close()
+  vnc.close()
+
+# reference
+if True:
+  nc =  netCDF4.Dataset('%s/outputs/1_salt.nc'%refname)
+  ncv=nc.variables
+
+  t = ncv['time'][:]
+  y = ncv['y'][:]
+  yidx = where(y==0.0)
+  s = ncv['salt'][tidx,-1][yidx]-35.
+  x = ncv['x'][yidx]/1000.
+
+  vnc = netCDF4.Dataset('%s/outputs/1_hvel.nc'%refname)
+  vncv = vnc.variables
+  tnum,znum,xnum = vncv['u'].shape
+  xidx=xnum/3+xnum/6
+  v = vncv['u'][-25:,-1,xidx]
+  vt = vncv['time'][-25:]
+  hours = (vt-vt[0])/3600.
+  nc.close()
+  vnc.close()
+
+  tax.plot(x,s,'-',lw=3.0,color='red',label='reference')
+  vax.plot(hours,v,'-',color='red',lw=2.0,label='reference')
+
+tax.set_ylim(-0.1,1.1)
+tax.set_ylabel('Tracer [1/1]')
+tax.set_xlabel('km')
+tax.legend(loc='upper left',frameon=False,ncol=2)
+
+vax.set_ylim(-0.2,0.2)
+vax.set_ylabel('surface current [m/s]')
+vax.set_xlabel('time [h]')
+vax.legend(loc='upper left',frameon=False,ncol=2)
+
+fig.savefig('transport_comparison_%s.pdf'%name)
+show(fig)
+
+vf.savefig('velocity_comparison_%s.pdf'%name)
+show(vf)

Plot/plot_tracers.py

@@ -0,0 +1,61 @@
+from pylab import *
+import netCDF4
+import sys
+
+runs = {}
+if len(sys.argv)>1:
+  name = sys.argv[1]
+else:
+  name='test'
+
+
+runs[1000] = name+'_1000m'
+runs[500] = name+'_500m'
+runs[200] = name+'_200m'
+runs[100] = name+'_100m'
+runs[50] = name+'_50m'
+runs[20] = name+'_20m'
+
+lss = ['-','--',':','-.','-..']
+tidx = 477
+
+fig = figure(figsize=(8,5))
+
+keys = sorted(runs)
+
+for dx in keys:
+  folder = runs[dx]
+  label = '%d m'%dx
+
+  print(label)
+  nc = netCDF4.Dataset('%s/outputs/1_salt.nc'%folder)
+  ncv=nc.variables
+
+  t = ncv['time'][:]
+  y = ncv['y'][:]
+  yidx = where(y==0.0)
+  s = ncv['salt'][tidx,-1][yidx]-35.
+  x = ncv['x'][yidx]/1000.
+
+  
+  if label=='20 m':
+    col = 'r'
+    ls = '-'
+    s0 = ncv['salt'][0,-1][yidx]-35.
+    plot(x,s0,'-',lw=3.0,color='orange',label='initial condition')
+  else:
+    col = (log10(int(label[:-2]))/3.-0.2)*ones((3))
+    ls = '-'#lss.pop()
+    
+
+  plot(x,s,ls=ls,color=col,lw=2.0,label=label)
+
+  nc.close()
+
+ylim(-0.1,1.1)
+ylabel('Tracer [1/1]')
+xlabel('km')
+legend(loc='upper left',frameon=False)
+
+savefig('transport_comparison_%s.pdf'%name)
+show()