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 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419
|
MODULE local_subs
USE cmor_users_functions
!!$ PRIVATE
!!$ PUBLIC read_coords, read_time, read_3d_input_files, read_2d_input_files
CONTAINS
SUBROUTINE read_coords(alats, alons, plevs, bnds_lat, bnds_lon)
IMPLICIT NONE
DOUBLE PRECISION, INTENT(OUT), DIMENSION(:) :: alats
DOUBLE PRECISION, INTENT(OUT), DIMENSION(:) :: alons
DOUBLE PRECISION, INTENT(OUT), DIMENSION(:) :: plevs
DOUBLE PRECISION, INTENT(OUT), DIMENSION(:,:) :: bnds_lat
DOUBLE PRECISION, INTENT(OUT), DIMENSION(:,:) :: bnds_lon
INTEGER :: i
DO i = 1, SIZE(alons)
alons(i) = (i-1)*360./SIZE(alons)
bnds_lon(1,i) = (i - 1.5)*360./SIZE(alons)
bnds_lon(2,i) = (i - 0.5)*360./SIZE(alons)
END DO
DO i = 1, SIZE(alats)
alats(i) = (size(alats)+1-i)*10
bnds_lat(1,i) = (size(alats)+1-i)*10 + 5.
bnds_lat(2,i) = (size(alats)+1-i)*10 - 5.
END DO
DO i = 1, SIZE(plevs)
plevs(i) = i*1.0e4
END DO
plevs = (/100000., 92500., 85000., 70000.,&
60000., 50000., 40000., 30000., 25000., 20000.,&
15000., 10000., 7000., 5000., 3000., 2000., 1000., 500., 100. /)
RETURN
END SUBROUTINE read_coords
SUBROUTINE read_time(it, time, time_bnds)
IMPLICIT NONE
INTEGER, INTENT(IN) :: it
DOUBLE PRECISION, INTENT(OUT) :: time
DOUBLE PRECISION, INTENT(OUT), DIMENSION(2,1) :: time_bnds
time = (it-0.5)*30.
time_bnds(1,1) = (it-1)*30.
time_bnds(2,1) = it*30.
RETURN
END SUBROUTINE read_time
include "reader_2D_3D.f90"
END MODULE local_subs
program testing
use cmor_users_functions
use local_subs
implicit none
integer error_flag
INTEGER, PARAMETER :: ntimes = 2 ! number of time samples to process
INTEGER, PARAMETER :: lon = 4 ! number of longitude grid cells
INTEGER, PARAMETER :: lat = 3 ! number of latitude grid cells
INTEGER, PARAMETER :: lev = 5 ! number of standard pressure levels
INTEGER, PARAMETER :: lev2 = 19 ! number of standard pressure levels
! INTEGER, PARAMETER :: n2d = 4 ! number of IPCC Table A1a fields to be
INTEGER, PARAMETER :: n2d = 1 ! number of IPCC Table A1a fields to be
! output.
INTEGER, PARAMETER :: n3d = 3 ! number of IPCC Table A1c fields to
! be output.
! My variable names for IPCC Table A1c fields
CHARACTER (LEN=5), DIMENSION(n3d) :: &
varin3d=(/'CLOUD', 'U ', 'T '/)
! Units appropriate to my data
CHARACTER (LEN=6), DIMENSION(n3d) :: &
units3d=(/ '% ', 'm s-1 ', 'K ' /)
! Corresponding IPCC Table A1c entry (variable name)
CHARACTER (LEN=2), DIMENSION(n3d) :: entry3d = (/ 'cl', 'ua', 'ta' /)
! My variable names for IPCC Table A1a fields
CHARACTER (LEN=8), DIMENSION(n2d) :: &
! varin2d=(/ 'LATENT ', 'TSURF ', 'SOIL_WET', 'PSURF ' /)
varin2d=(/ 'SFTLF ' /)
! Units appropriate to my data
CHARACTER (LEN=6), DIMENSION(n2d) :: &
! units2d=(/ 'W m-2 ', 'K ', 'kg m-2', 'Pa ' /)
units2d=(/ '% ' /)
CHARACTER (LEN=4), DIMENSION(n2d) :: &
! positive2d= (/ 'down', ' ', ' ', ' ' /)
positive2d= (/ ' ' /)
! Corresponding IPCC Table A1a entry (variable name)
CHARACTER (LEN=5), DIMENSION(n2d) :: &
! entry2d = (/ 'hfls ', 'tas ', 'mrsos', 'ps ' /)
entry2d = (/ 'sftof ' /)
! uninitialized variables used in communicating with CMOR:
! ---------------------------------------------------------
INTEGER :: znondim_id, zfactor_id
INTEGER, DIMENSION(n2d) :: var2d_ids
INTEGER, DIMENSION(n3d) :: var3d_ids
REAL, DIMENSION(lon,lat) :: data2d
real, DIMENSION(lon*lat) :: data1dtest
REAL, DIMENSION(lon,lat,lev2) :: data3d
DOUBLE PRECISION, DIMENSION(lat) :: alats
DOUBLE PRECISION, DIMENSION(lon) :: alons
DOUBLE PRECISION, DIMENSION(lev2) :: plevs
DOUBLE PRECISION, DIMENSION(1) :: time
DOUBLE PRECISION, DIMENSION(2,1):: bnds_time
DOUBLE PRECISION, DIMENSION(2,lat) :: bnds_lat
DOUBLE PRECISION, DIMENSION(2,lon) :: bnds_lon
DOUBLE PRECISION, DIMENSION(lev) :: zlevs
DOUBLE PRECISION, DIMENSION(lev+1) :: zlev_bnds
real, DIMENSION(lev) :: a_coeff
DOUBLE PRECISION, DIMENSION(lev) :: b_coeff
DOUBLE PRECISION :: p0
DOUBLE PRECISION :: p0array(1)
real, DIMENSION(lev+1) :: a_coeff_bnds
DOUBLE PRECISION, DIMENSION(lev+1) :: b_coeff_bnds
INTEGER :: ilon, ilat, ipres, ilev, itim,i,ilon2,ilat2
real :: missing
double precision bt
! Other variables:
! ---------------------
INTEGER :: it, m, j,k
bt=0.
! format classic(NETCDF-3)
! j = CMOR_REPLACE_3
! format NETCDF-4
j = CMOR_REPLACE
k = CMOR_EXIT_ON_MAJOR
error_flag = cmor_setup(inpath='Test', netcdf_file_action=j,&
exit_control=k)
print*,'Test code: done'
error_flag = cmor_dataset_json("Test/CMOR_input_example.json")
print*, 'Test code: done 2 lalala'
call read_coords(alats, alons, plevs, bnds_lat, bnds_lon)
print*, 'Test code: ok calling axis stuff lat',alats
ilat = cmor_axis( &
table='Tables/CMIP6_Amon.json', &
table_entry='latitude', &
units='degrees_north', &
length=lat, &
coord_vals=alats, &
cell_bounds=bnds_lat)
print*, 'Test code: ok calling axis stuff lon',ilat
ilon = cmor_axis( &
table='Tables/CMIP6_Amon.json', &
table_entry='longitude', &
length=lon, &
units='degrees_east', &
coord_vals=alons, &
cell_bounds=bnds_lon)
ilat2 = cmor_axis( &
table='Tables/CMIP6_Ofx.json', &
table_entry='latitude', &
units='degrees_north', &
length=lat, &
coord_vals=alats, &
cell_bounds=bnds_lat)
print*, 'Test code: ok calling axis stuff lon',ilat
ilon2 = cmor_axis( &
table='Tables/CMIP6_Ofx.json', &
table_entry='longitude', &
length=lon, &
units='degrees_east', &
coord_vals=alons, &
cell_bounds=bnds_lon)
print*, 'Test code: ok calling axis stuff pressure',ilon
ipres = cmor_axis( &
table='Tables/CMIP6_Amon.json', &
table_entry='plev19', &
units='Pa', &
length=lev2, &
coord_vals=plevs)
! note that the time axis is defined next, but the time coordinate
! values and bounds will be passed to cmor through function
! cmor_write (later, below).
print*, 'Test code: ok calling axis stuff time',ipres,ntimes
itim = cmor_axis( &
table='Tables/CMIP6_Amon.json', &
table_entry='time', &
units='days since 2030-1-1', &
length=ntimes, &
interval='30 days')
! define model eta levels (although these must be provided, they will
! actually be replaced by a+b before writing the netCDF file)
zlevs = (/ 0.1, 0.3, 0.55, 0.7, 0.9 /)
zlev_bnds=(/ 0.,.2, .42, .62, .8, 1. /)
print*, 'Test code: ok calling axis stuff lev2',itim
ilev = cmor_axis( &
table='Tables/CMIP6_Amon.json', &
table_entry='standard_hybrid_sigma', &
units="1", &
length=lev, &
coord_vals=zlevs, &
cell_bounds=zlev_bnds)
print*, 'Test code: ok called stuff lev2',ilev
! define z-factors needed to transform from model level to pressure
p0 = 1.e5
p0array(1)=p0
a_coeff = (/ 0.1, 0.2, 0.3, 0.22, 0.1 /)
b_coeff = (/ 0.0, 0.1, 0.2, 0.5, 0.8 /)
a_coeff_bnds=(/0.,.15, .25, .25, .16, 0./)
b_coeff_bnds=(/0.,.05, .15, .35, .65, 1./)
print*, 'Test code: zfactor p0'
error_flag = cmor_zfactor( &
zaxis_id=ilev, &
zfactor_name='p0', &
units='Pa', &
zfactor_values = p0)
print*, 'Test code: result',error_flag
print*, 'Test code: zfactor b'
error_flag = cmor_zfactor( &
zaxis_id=ilev, &
zfactor_name='b', &
axis_ids= (/ ilev /), &
zfactor_values = b_coeff, &
zfactor_bounds = b_coeff_bnds )
print*, 'Test code: result',error_flag
print*, 'Test code: zfactor a'
error_flag = cmor_zfactor( &
zaxis_id=ilev, &
zfactor_name='a', &
axis_ids= (/ ilev /), &
zfactor_values = a_coeff, &
zfactor_bounds = a_coeff_bnds )
print*, 'Test code: result',error_flag
print*, 'Test code: zfactor ps'
zfactor_id = cmor_zfactor( &
zaxis_id=ilev, &
zfactor_name='ps', &
axis_ids=(/ ilon, ilat, itim /), &
units='Pa' )
print*, 'Test code: result',zfactor_id
print*, 'Test code: var3d'
missing = 1.e28
var3d_ids(1) = cmor_variable( &
table='Tables/CMIP6_Amon.json', &
table_entry=entry3d(1), &
units=units3d(1), &
axis_ids=(/ ilon, ilat, ilev, itim /), &
missing_value=missing, &
original_name=varin3d(1))
! Define variables appearing in IPCC table A1c that are a function of pressure
! (3-d variables)
print*, 'Test code: result',var3d_ids(1)
DO m=2,n3d
print*, 'Test code: var: ',entry3d(m)
var3d_ids(m) = cmor_variable( &
table='Tables/CMIP6_Amon.json', &
table_entry=entry3d(m), &
units=units3d(m), &
axis_ids=(/ ilon, ilat, ipres, itim /), &
missing_value=missing, &
original_name=varin3d(m))
print*, 'Test code: result',var3d_ids(m)
ENDDO
! Define variables appearing in IPCC table A1a (2-d variables)
DO m=1,n2d
print*, 'Test code: var: ',entry2d(m)
var2d_ids(m) = cmor_variable( &
table='Tables/CMIP6_Ofx.json', &
table_entry=entry2d(m), &
units=units2d(m), &
! axis_ids=(/ ilon, ilat, itim /), &
axis_ids=(/ ilon2, ilat2 /), &
missing_value=missing, &
positive=positive2d(m), &
original_name=varin2d(m))
print*, 'Test code: result',var2d_ids(m)
ENDDO
PRINT*, 'Test code: '
PRINT*, 'Test code: completed everything up to writing output fields '
PRINT*, 'Test code: '
time_loop: DO it=1, ntimes
! In the following loops over the 3d and 2d fields, the user-written
! subroutines (read_3d_input_files and read_2d_input_files) retrieve
! the requested IPCC table A1c and table A1a fields and store them in
! data3d and data2d, respectively. In addition a user-written code
! (read_time) retrieves the time and time-bounds associated with the
! time sample (in units of 'days since 1970-1-1', consistent with the
! axis definitions above). The bounds are set to the beginning and
! the end of the month retrieved, indicating the averaging period.
! The user must write a code to obtain the times and time-bounds for
! the time slice. The following line is simply a place-holder for
! the user's code, which should replace it.
call read_time(it, time(1), bnds_time)
call read_3d_input_files(it, varin3d(1), data3d)
error_flag = cmor_write( &
var_id = var3d_ids(1), &
data = data3d, &
ntimes_passed = 1, &
time_vals = time, &
time_bnds = bnds_time )
call read_2d_input_files(it, 'PSURF', data2d)
error_flag = cmor_write( &
var_id = zfactor_id, &
data = data2d, &
ntimes_passed = 1, &
time_vals = time, &
time_bnds = bnds_time, &
store_with = var3d_ids(1) )
print*, 'Test code: result',error_flag
! Cycle through the 3-d fields (stored on pressure levels),
! and retrieve the requested variable and append each to the
! appropriate netCDF file.
DO m=2,n3d
! The user must write the code that fills the arrays of data
! that will be passed to CMOR. The following line is simply a
! a place-holder for the user's code, which should replace it.
call read_3d_input_files(it, varin3d(m), data3d)
! append a single time sample of data for a single field to
! the appropriate netCDF file.
print*, ' Writing 3D variable:',var3d_ids(m)
error_flag = cmor_write( &
var_id = var3d_ids(m), &
data = data3d, &
ntimes_passed = 1, &
time_vals = time, &
time_bnds = bnds_time )
IF (error_flag < 0) THEN
! write diagnostic messages to standard output device
write(*,*) 'Test code: Error encountered writing IPCC Table A1c ' &
// 'field ', entry3d(m), ', which I call ', varin3d(m)
write(*,*) 'Test code: Was processing time sample: ', time
END IF
END DO
!!$
!!$ ! Cycle through the 2-d fields, retrieve the requested variable and
!!$ ! append each to the appropriate netCDF file.
!!$
DO m=1,n2d
! The user must write the code that fills the arrays of data
! that will be passed to CMOR. The following line is simply a
! a place-holder for the user's code, which should replace it.
print*, 'varinf',varin2d(m)
call read_2d_input_files(it, varin2d(m), data2d)
! append a single time sample of data for a single field to
! the appropriate netCDF file.
print*, 'Writing var 2d:',var2d_ids(m),entry2d(m)
error_flag = cmor_write( &
var_id = var2d_ids(m), &
data = data2d, &
ntimes_passed = 0 )
!ntimes_passed = 1, &
!time_vals = time, &
!time_bnds = bnds_time )
IF (error_flag < 0) THEN
! write diagnostic messages to standard output device
write(*,*) 'Test code: Error encountered writing IPCC Table A1a ' &
// 'field ', entry2d(m), ', which I call ', varin2d(m)
write(*,*) 'Test code: Was processing time sample: ', time
END IF
END DO
END DO time_loop
error_flag = cmor_close()
end program testing
|