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
98
99
100
101
102
103
104
105
106
107
|
from __future__ import print_function
import cmor
import numpy
vars = {
'hfls': [
'W.m-2',
25.,
40.],
'tas': [
'K',
25,
268.15],
'clt': [
'%',
10000.,
0.],
'ta': [
'K',
25,
273.15]}
nlat = 90
dlat = 180 / nlat
nlon = 180
dlon = 360. / nlon
nlev = 19
ntimes = 12
lats = numpy.arange(-90 + dlat / 2., 90, dlat)
blats = numpy.arange(-90, 90 + dlat, dlat)
lons = numpy.arange(0 + dlon / 2., 360., dlon)
blons = numpy.arange(0, 360. + dlon, dlon)
tvars = ['hfls', 'tas', 'clt', 'ta']
cmor.setup(inpath='Tables', netcdf_file_action=cmor.CMOR_REPLACE)
cmor.dataset_json("Test/CMOR_input_example.json")
table = 'CMIP6_Amon.json'
cmor.load_table(table)
for var in tvars:
ilat = cmor.axis(
table_entry='latitude',
coord_vals=lats,
cell_bounds=blats,
units='degrees_north')
ilon = cmor.axis(
table_entry='longitude',
coord_vals=lons,
cell_bounds=blons,
units='degrees_east')
itim = cmor.axis(
table_entry='time', coord_vals=numpy.arange(
ntimes, dtype=float), cell_bounds=numpy.arange(
ntimes + 1, dtype=float), units='months since 2000')
ilev = cmor.axis(table_entry='plev19',
coord_vals=numpy.array([1000.,
925,
850,
700,
600,
500,
400,
300,
250,
200,
150,
100,
70,
50,
30,
20,
10,
5,
1]),
units='hPa')
if var != 'ta':
axes = [itim, ilat, ilon]
data = numpy.random.random(
(ntimes, nlat, nlon)) * vars[var][1] + vars[var][2]
else:
axes = [itim, ilev, ilat, ilon]
data = numpy.random.random(
(ntimes, nlev, nlat, nlon)) * vars[var][1] + vars[var][2]
kw = {}
if var in ['hfss', 'hfls']:
kw['positive'] = 'up'
var = cmor.variable(
table_entry=var,
units=vars[var][0],
axis_ids=axes,
**kw)
cmor.write(var, data)
path = cmor.close(var, file_name=True)
print('Saved in:', path)
cmor.close()
print('hello')
|
