PythonCode/create_passtracers.py at master · PaulSpence/PythonCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
"""
create_passtracers.py
-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
"""


import os
import glob
import numpy as np

from scipy import interpolate
import sys
sys.path.insert(1,'/projects/v45/python')
import netCDF4 as nc

def cpt():

    vals=nc.Dataset('/short/v45/pas561/mom/input/gfdl_nyf_1080/ocean_temp_salt.res.nc','r')
    zt = vals.variables['ZT'][:]
    grid_x_t = vals.variables['GRID_X_T'][:]
    grid_y_t = vals.variables['GRID_Y_T'][:]
    temp = vals.variables['temp'][:]
    vals.close()

    km=temp.shape[0]
    jm=temp.shape[1]
    im=temp.shape[2]
    print 'km,jm,im',km,jm,im

    #create land/sea masks
    msk = np.ones(np.shape(temp))

    save_field(grid_x_t,grid_y_t,zt,temp,'ocean_passive.res.nc','passive_patch_1')

def save_field(x, y, z, field,name,var):

    field_nc = nc.Dataset(name,'w')

    nz, ny, nx = field.shape
    field_nc.createDimension('ZT', nz)
    field_nc.createDimension('LAT', ny)
    field_nc.createDimension('LON', nx)

    z_nc = field_nc.createVariable('ZT', 'f8', ('ZT',))
    z_nc[:] = z
    z_nc.long_name = 'depth'
    z_nc.units = 'm'
    z_nc.axis = 'Cartesian Z'

    lat_nc = field_nc.createVariable('LAT', 'f8', ('LAT',))
    lat_nc[:] = y
    lat_nc.long_name = 'Mercator Gridded Latitude'
    lat_nc.units = 'degrees north'
    lat_nc.axis = 'Cartesian Y'

    lon_nc = field_nc.createVariable('LON', 'f8', ('LON',))
    lon_nc[:] = x
    lon_nc.long_name = 'Mercator Gridded Longitude'
    lon_nc.units = 'degrees east'
    lon_nc.axis = 'Cartesian X'

    temp_nc = field_nc.createVariable(var, 'f8', ('ZT', 'LAT', 'LON'))
    temp_nc[:] = field

    field_nc.close()


    #output_file = 'ocean_passive.res.nc'
    #print "Writing output file ",output_file
    ## Output to NetCDF file
    #output = Dataset(output_file, 'w', format='NETCDF4')

    #output.createDimension('GRID_X_T', jm)
    #y_var = output.createVariable('GRID_X_T', 'f8', ('GRID_X_T',))
    #y_var.units = 'degree_east'

    #y_var[:] = yt_ocean[:]

    #output.createDimension('xt_ocean', im)
    #x_var = output.createVariable('xt_ocean', 'f8', ('xt_ocean',))
    #x_var.units = 'meters'
    #x_var[:] = xt_ocean[:]

    #dx_var = output.createVariable('dxt', 'f8', ('yt_ocean', 'xt_ocean',))
    #dx_var.units = 'm'
    #dx_var[:,:] = dxt

    #output.close()


    if __name__ == "__main":
       cpt()