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
|
from __future__ import print_function
from test_python_common import * # common subroutines
import cmor._cmor
import os
pth = os.path.split(os.path.realpath(os.curdir))
if pth[-1] == 'Test':
ipth = opth = '.'
else:
ipth = opth = 'Test'
myaxes = numpy.zeros(9, dtype='i')
myaxes2 = numpy.zeros(9, dtype='i')
myvars = numpy.zeros(9, dtype='i')
cmor.setup(
inpath=ipth,
set_verbosity=cmor.CMOR_NORMAL,
netcdf_file_action=cmor.CMOR_REPLACE,
exit_control=cmor.CMOR_EXIT_ON_MAJOR)
cmor.dataset_json("Test/CMOR_input_example.json")
tables = []
a = cmor.load_table("Tables/CMIP6_grids.json")
tables.append(a)
tables.append(cmor.load_table("Tables/CMIP6_Lmon.json"))
print('Tables ids:', tables)
cmor.set_table(tables[0])
x, y, lon_coords, lat_coords, lon_vertices, lat_vertices = gen_irreg_grid(
lon, lat)
myaxes[0] = cmor.axis(table_entry='y',
units='m',
coord_vals=y)
myaxes[1] = cmor.axis(table_entry='x',
units='m',
coord_vals=x)
grid_id = cmor.grid(axis_ids=myaxes[:2],
latitude=lat_coords,
longitude=lon_coords,
latitude_vertices=lat_vertices,
longitude_vertices=lon_vertices)
print('got grid_id:', grid_id)
myaxes[2] = grid_id
## mapnm = 'lambert_conformal_conic'
# params = [ "standard_parallel1",
# "longitude_of_central_meridian","latitude_of_projection_origin",
# "false_easting","false_northing","standard_parallel2" ]
## punits = ["","","","","","" ]
## pvalues = [-20.,175.,13.,8.,0.,20. ]
# cmor.set_grid_mapping(grid_id=myaxes[2],
## mapping_name = mapnm,
## parameter_names = params,
## parameter_values = pvalues,
# parameter_units = punits)
cmor.set_table(tables[1])
myaxes[3] = cmor.axis(table_entry='time',
units='months since 1980')
myaxes[4] = cmor.axis(table_entry='vegtype',
units='',
coord_vals="""grass marijuana opium""".split())
pass_axes = [myaxes[2], myaxes[3], myaxes[4]]
print('ok going to cmorvar')
myvars[0] = cmor.variable(table_entry='landCoverFrac',
units='%',
axis_ids=pass_axes,
history='no history',
comment='no future'
)
for i in range(ntimes):
data2d = numpy.random.random((3, 4, 3))
print('writing time: ', i, data2d.shape, data2d)
print(Time[i], bnds_time[2 * i:2 * i + 2])
cmor.write(myvars[0], data2d, 1, time_vals=Time[i],
time_bnds=bnds_time[2 * i:2 * i + 2])
cmor.close()
|
