#get current time series for plotting tidal ellipses
import netCDF4 as NC
runname = '40d'
grid = NC.Dataset('/ocean/klesouef/meopar/nemo-forcing/grid/bathy_meter_SalishSea2.nc','r')
resultsloc = '/ocean/dlatorne/MEOPAR/SalishSea/results/'+runname
uresults = NC.Dataset('/ocean/dlatorne/MEOPAR/SalishSea/results/40d/SalishSea_3d_20020915_20021024_grid_U.nc','r')
print uresults
<type 'netCDF4.Dataset'> root group (NETCDF3_64BIT file format): Conventions: CF-1.1 production: An IPSL model TimeStamp: 07/01/2014 04:54:27 -0700 file_name: SalishSea_3d_20020915_20021024_grid_U.nc dimensions: x, y, depthu, time_counter, tbnds variables: nav_lon, nav_lat, depthu, time_counter, time_counter_bnds, vozocrtx, u_wind_stress groups:
from salishsea_tools import tidetools
mod_M2_u_amp, mod_M2_u_pha, mod_M2_v_amp, mod_M2_v_pha = tidetools.get_current_harms(runname,resultsloc)
%matplotlib inline
import matplotlib.pylab as plt
i = 381
j = 10
Au = mod_M2_u_amp[i,j]
PHIu = mod_M2_u_pha[i,j]
Av = mod_M2_v_amp[i,j]
PHIv = mod_M2_v_pha[i,j]
SEMA, ECC, INC, PHA = tidetools.ap2ep(Au, PHIu, Av, PHIv)
print Au, PHIu, Av, PHIv
print SEMA, ECC, INC, PHA, SEMA*INC
0.00931887 -104.685 0.00931887 -104.685 0.013178870402 0.0 0.0 300.315261841 0.0
#Wait... it looks like u and v real and imaginary components in Tidal_Harmonics_*.nc are the same?
#Why would this be? Doesn't this imply rectilinear currents...?
runname = '40d'
loc = '/ocean/dlatorne/MEOPAR/SalishSea/results/'
eta = NC.Dataset(loc+runname+'/Tidal_Harmonics_eta.nc','r')
u = NC.Dataset(loc+runname+'/Tidal_Harmonics_U.nc','r')
v = NC.Dataset(loc+runname+'/Tidal_Harmonics_V.nc','r')
print eta.variables['M2_eta_real'][0,i,j]
print u.variables['M2_u_real'][0,i,j]
print v.variables['M2_v_real'][0,i,j]
print eta.variables['M2_eta_imag'][0,i,j]
print u.variables['M2_u_imag'][0,i,j]
print v.variables['M2_v_imag'][0,i,j]
plt.figure(figsize=(15,10))
plt.subplot(2,3,1)
plt.contourf(u.variables['M2_u_real'][0,:,:])
plt.colorbar()
plt.title('u real')
plt.subplot(2,3,2)
plt.contourf(v.variables['M2_v_real'][0,:,:])
plt.colorbar()
plt.title('v real')
plt.subplot(2,3,3)
plt.contourf(eta.variables['M2_eta_real'][0,:,:])
plt.colorbar()
plt.title('eta real')
plt.subplot(2,3,4)
plt.contourf(u.variables['M2_u_imag'][0,:,:])
plt.colorbar()
plt.title('u imag')
plt.subplot(2,3,5)
plt.contourf(v.variables['M2_v_imag'][0,:,:])
plt.colorbar()
plt.title('v imag')
plt.subplot(2,3,6)
plt.contourf(eta.variables['M2_eta_imag'][0,:,:])
plt.colorbar()
plt.title('eta imag')
-0.328695 -0.00236234 -0.00236234 0.836416 0.00901447 0.00901447
<matplotlib.text.Text at 0x63e3e650>