[2895] | 1 | PROGRAM rebuild_nemo |
---|
| 2 | |
---|
| 3 | !!========================================================================= |
---|
| 4 | !! *** rebuild_nemo *** |
---|
| 5 | !!========================================================================= |
---|
| 6 | !! |
---|
| 7 | !! A routine to rebuild NEMO files from multiple processors into one file. |
---|
| 8 | !! This routine is designed to be much quicker than the old IOIPSL rebuild |
---|
| 9 | !! but at the cost of an increased memory usage. |
---|
| 10 | !! |
---|
| 11 | !! NEMO rebuild has the following features: |
---|
| 12 | !! * dynamically works out what variables require rebuilding |
---|
| 13 | !! * does not copy subdomain halo regions |
---|
| 14 | !! * works for 1,2,3 and 4d arrays or types for all valid NetCDF types |
---|
| 15 | !! * utilises OMP shared memory parallelisation where applicable |
---|
[3025] | 16 | !! * time 'chunking' for lower memory use |
---|
| 17 | !! (only for 4D vars with unlimited dimension) |
---|
[2895] | 18 | !! |
---|
| 19 | !! Ed Blockley - August 2011 |
---|
| 20 | !! (based on original code by Matt Martin) |
---|
| 21 | !! |
---|
| 22 | !!------------------------------------------------------------------------- |
---|
| 23 | !! |
---|
| 24 | !! The code reads the filestem and number of subdomains from the namelist file nam_rebuild. |
---|
| 25 | !! |
---|
| 26 | !! The 1st subdomain file is used to determine the dimensions and variables in all the input files. |
---|
| 27 | !! It is also used to find which dimensions (and hence which variables) require rebuilding |
---|
| 28 | !! as well as information about the global domain. |
---|
| 29 | !! |
---|
| 30 | !! It then opens all the input files (unbuffered) and creates an array of netcdf identifiers |
---|
| 31 | !! before looping through all the variables and updating the rebuilt output file (either by direct |
---|
| 32 | !! copying or looping over the number of domains and rebuilding as appropriate). |
---|
| 33 | !! |
---|
| 34 | !! The code looks more complicated than it is because it has lots of case statements to deal with all |
---|
| 35 | !! the various NetCDF data types and with various data dimensions (up to 4d). |
---|
| 36 | !! |
---|
| 37 | !! Diagnostic output is written to numout (default 6 - stdout) |
---|
| 38 | !! and errors are written to numerr (default 0 - stderr). |
---|
| 39 | !! |
---|
[3025] | 40 | !! If time chunking is specified the code will use less memory but take a little longer. |
---|
| 41 | !! It does this by breaking down the 4D input variables over their 4th dimension |
---|
| 42 | !! (generally time) by way of a while loop. |
---|
| 43 | !! |
---|
[2895] | 44 | !!------------------------------------------------------------------------------- |
---|
| 45 | |
---|
| 46 | USE netcdf |
---|
| 47 | |
---|
| 48 | !$ USE omp_lib ! Note OpenMP sentinel |
---|
| 49 | |
---|
| 50 | IMPLICIT NONE |
---|
| 51 | |
---|
| 52 | ! kind specifications |
---|
| 53 | INTEGER,PARAMETER :: i1=SELECTED_INT_KIND(2) ! NF90_BYTE |
---|
| 54 | INTEGER,PARAMETER :: i2=SELECTED_INT_KIND(4) ! NF90_SHORT |
---|
| 55 | INTEGER,PARAMETER :: i4=SELECTED_INT_KIND(9) ! NF90_INT |
---|
| 56 | INTEGER,PARAMETER :: sp=SELECTED_REAL_KIND(6,37) ! NF90_FLOAT |
---|
| 57 | INTEGER,PARAMETER :: dp=SELECTED_REAL_KIND(12,307) ! NF90_DOUBLE |
---|
| 58 | |
---|
| 59 | INTEGER,PARAMETER :: numnam = 11 |
---|
| 60 | INTEGER,PARAMETER :: numout = 6 |
---|
| 61 | INTEGER,PARAMETER :: numerr = 0 |
---|
| 62 | |
---|
| 63 | LOGICAL, PARAMETER :: l_verbose = .true. |
---|
| 64 | |
---|
| 65 | CHARACTER(LEN=nf90_max_name) :: filebase, suffix, attname, dimname, varname, time, date, zone, timestamp |
---|
| 66 | CHARACTER(LEN=nf90_max_name), ALLOCATABLE :: filenames(:), indimnames(:) |
---|
| 67 | CHARACTER(LEN=nf90_max_name), DIMENSION(2) :: dims |
---|
| 68 | |
---|
[3025] | 69 | INTEGER :: ndomain, ifile, ndomain_file, nchunksize |
---|
[2895] | 70 | INTEGER :: ncid, outid, idim, istop |
---|
| 71 | INTEGER :: natts, attid, xtype, varid, rbdims |
---|
| 72 | INTEGER :: jv, ndims, nvars, dimlen, dimids(4) |
---|
| 73 | INTEGER :: dimid, unlimitedDimId, di, dj, dr |
---|
[3025] | 74 | INTEGER :: nmax_unlimited, nt, ntchunk |
---|
[2895] | 75 | INTEGER :: chunksize = 32000000 |
---|
| 76 | INTEGER :: nthreads = 1 |
---|
| 77 | INTEGER, ALLOCATABLE :: outdimids(:), outdimlens(:), indimlens(:), inncids(:) |
---|
| 78 | INTEGER, ALLOCATABLE :: global_sizes(:), rebuild_dims(:) |
---|
| 79 | INTEGER, DIMENSION(2) :: halo_start, halo_end, local_sizes |
---|
| 80 | INTEGER, DIMENSION(2) :: idomain, jdomain, rdomain, start_pos |
---|
| 81 | INTEGER :: ji, jj, jk, jl, jr |
---|
| 82 | |
---|
| 83 | REAL(sp) :: ValMin, ValMax, InMin, InMax |
---|
| 84 | |
---|
| 85 | ! NF90_BYTE local data arrays |
---|
| 86 | INTEGER(i1), ALLOCATABLE, SAVE, DIMENSION(:) :: localdata_1d_i1 |
---|
| 87 | INTEGER(i1), ALLOCATABLE, SAVE, DIMENSION(:,:) :: localdata_2d_i1 |
---|
| 88 | INTEGER(i1), ALLOCATABLE, SAVE, DIMENSION(:,:,:) :: localdata_3d_i1 |
---|
| 89 | INTEGER(i1), ALLOCATABLE, SAVE, DIMENSION(:,:,:,:) :: localdata_4d_i1 |
---|
| 90 | |
---|
| 91 | ! NF90_SHORT local data arrays |
---|
| 92 | INTEGER(i2), ALLOCATABLE, SAVE, DIMENSION(:) :: localdata_1d_i2 |
---|
| 93 | INTEGER(i2), ALLOCATABLE, SAVE, DIMENSION(:,:) :: localdata_2d_i2 |
---|
| 94 | INTEGER(i2), ALLOCATABLE, SAVE, DIMENSION(:,:,:) :: localdata_3d_i2 |
---|
| 95 | INTEGER(i2), ALLOCATABLE, SAVE, DIMENSION(:,:,:,:) :: localdata_4d_i2 |
---|
| 96 | |
---|
| 97 | ! NF90_INT local data arrays |
---|
| 98 | INTEGER(i4), ALLOCATABLE, SAVE, DIMENSION(:) :: localdata_1d_i4 |
---|
| 99 | INTEGER(i4), ALLOCATABLE, SAVE, DIMENSION(:,:) :: localdata_2d_i4 |
---|
| 100 | INTEGER(i4), ALLOCATABLE, SAVE, DIMENSION(:,:,:) :: localdata_3d_i4 |
---|
| 101 | INTEGER(i4), ALLOCATABLE, SAVE, DIMENSION(:,:,:,:) :: localdata_4d_i4 |
---|
| 102 | |
---|
| 103 | ! NF90_FLOAT local data arrays |
---|
| 104 | REAL(sp), ALLOCATABLE, SAVE, DIMENSION(:) :: localdata_1d_sp |
---|
| 105 | REAL(sp), ALLOCATABLE, SAVE, DIMENSION(:,:) :: localdata_2d_sp |
---|
| 106 | REAL(sp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) :: localdata_3d_sp |
---|
| 107 | REAL(sp), ALLOCATABLE, SAVE, DIMENSION(:,:,:,:) :: localdata_4d_sp |
---|
| 108 | |
---|
| 109 | ! NF90_DOUBLE local data arrays |
---|
| 110 | REAL(dp), ALLOCATABLE, SAVE, DIMENSION(:) :: localdata_1d_dp |
---|
| 111 | REAL(dp), ALLOCATABLE, SAVE, DIMENSION(:,:) :: localdata_2d_dp |
---|
| 112 | REAL(dp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) :: localdata_3d_dp |
---|
| 113 | REAL(dp), ALLOCATABLE, SAVE, DIMENSION(:,:,:,:) :: localdata_4d_dp |
---|
| 114 | |
---|
| 115 | ! NF90_BYTE global data arrays |
---|
| 116 | INTEGER(i1) :: globaldata_0d_i1 |
---|
| 117 | INTEGER(i1), ALLOCATABLE, DIMENSION(:) :: globaldata_1d_i1 |
---|
| 118 | INTEGER(i1), ALLOCATABLE, DIMENSION(:,:) :: globaldata_2d_i1 |
---|
| 119 | INTEGER(i1), ALLOCATABLE, DIMENSION(:,:,:) :: globaldata_3d_i1 |
---|
| 120 | INTEGER(i1), ALLOCATABLE, DIMENSION(:,:,:,:) :: globaldata_4d_i1 |
---|
| 121 | |
---|
| 122 | ! NF90_SHORT global data arrays |
---|
| 123 | INTEGER(i2) :: globaldata_0d_i2 |
---|
| 124 | INTEGER(i2), ALLOCATABLE, DIMENSION(:) :: globaldata_1d_i2 |
---|
| 125 | INTEGER(i2), ALLOCATABLE, DIMENSION(:,:) :: globaldata_2d_i2 |
---|
| 126 | INTEGER(i2), ALLOCATABLE, DIMENSION(:,:,:) :: globaldata_3d_i2 |
---|
| 127 | INTEGER(i2), ALLOCATABLE, DIMENSION(:,:,:,:) :: globaldata_4d_i2 |
---|
| 128 | |
---|
| 129 | ! NF90_INT global data arrays |
---|
| 130 | INTEGER(i4) :: globaldata_0d_i4 |
---|
| 131 | INTEGER(i4), ALLOCATABLE, DIMENSION(:) :: globaldata_1d_i4 |
---|
| 132 | INTEGER(i4), ALLOCATABLE, DIMENSION(:,:) :: globaldata_2d_i4 |
---|
| 133 | INTEGER(i4), ALLOCATABLE, DIMENSION(:,:,:) :: globaldata_3d_i4 |
---|
| 134 | INTEGER(i4), ALLOCATABLE, DIMENSION(:,:,:,:) :: globaldata_4d_i4 |
---|
| 135 | |
---|
| 136 | ! NF90_FLOAT global data arrays |
---|
| 137 | REAL(sp) :: globaldata_0d_sp |
---|
| 138 | REAL(sp), ALLOCATABLE, DIMENSION(:) :: globaldata_1d_sp |
---|
| 139 | REAL(sp), ALLOCATABLE, DIMENSION(:,:) :: globaldata_2d_sp |
---|
| 140 | REAL(sp), ALLOCATABLE, DIMENSION(:,:,:) :: globaldata_3d_sp |
---|
| 141 | REAL(sp), ALLOCATABLE, DIMENSION(:,:,:,:) :: globaldata_4d_sp |
---|
| 142 | |
---|
| 143 | ! NF90_DOUBLE global data arrays |
---|
| 144 | REAL(dp) :: globaldata_0d_dp |
---|
| 145 | REAL(dp), ALLOCATABLE, DIMENSION(:) :: globaldata_1d_dp |
---|
| 146 | REAL(dp), ALLOCATABLE, DIMENSION(:,:) :: globaldata_2d_dp |
---|
| 147 | REAL(dp), ALLOCATABLE, DIMENSION(:,:,:) :: globaldata_3d_dp |
---|
| 148 | REAL(dp), ALLOCATABLE, DIMENSION(:,:,:,:) :: globaldata_4d_dp |
---|
| 149 | |
---|
| 150 | LOGICAL :: l_valid = .false. |
---|
| 151 | LOGICAL :: l_noRebuild = .false. |
---|
| 152 | LOGICAL :: l_findDims = .true. |
---|
| 153 | |
---|
[3025] | 154 | NAMELIST/nam_rebuild/ filebase, ndomain, dims, nchunksize |
---|
[2895] | 155 | |
---|
| 156 | !End of definitions |
---|
| 157 | |
---|
| 158 | !-------------------------------------------------------------------------------- |
---|
| 159 | !0. OMP setup |
---|
| 160 | |
---|
| 161 | !$OMP PARALLEL DEFAULT(NONE) SHARED(nthreads) |
---|
| 162 | !$OMP MASTER |
---|
| 163 | !$ nthreads = omp_get_num_threads() |
---|
| 164 | !$ WRITE(numout,*) 'Running OMP with ',nthreads,' thread(s).' |
---|
| 165 | !$OMP END MASTER |
---|
| 166 | !$OMP END PARALLEL |
---|
| 167 | |
---|
| 168 | !-------------------------------------------------------------------------------- |
---|
| 169 | !1. Read in the namelist |
---|
| 170 | |
---|
| 171 | dims(:) = "" |
---|
[3025] | 172 | nchunksize = 0 |
---|
[2895] | 173 | OPEN( UNIT=numnam, FILE='nam_rebuild', FORM='FORMATTED', STATUS='OLD' ) |
---|
| 174 | READ( numnam, nam_rebuild ) |
---|
| 175 | CLOSE( numnam ) |
---|
| 176 | IF( .NOT. ALL(dims(:) == "") ) l_findDims = .false. |
---|
| 177 | |
---|
| 178 | !1.1 Set up the filenames and fileids |
---|
| 179 | |
---|
| 180 | ALLOCATE(filenames(ndomain)) |
---|
| 181 | IF (l_verbose) WRITE(numout,*) 'Rebuilding the following files:' |
---|
| 182 | DO ifile = 1, ndomain |
---|
| 183 | WRITE(suffix,'(i4.4)') ifile-1 |
---|
| 184 | filenames(ifile) = TRIM(filebase)//'_'//TRIM(suffix)//'.nc' |
---|
| 185 | IF (l_verbose) WRITE(numout,*) TRIM(filenames(ifile)) |
---|
| 186 | END DO |
---|
| 187 | ALLOCATE(inncids(ndomain)) |
---|
| 188 | |
---|
| 189 | !--------------------------------------------------------------------------- |
---|
| 190 | !2. Read in the global dimensions from the first input file and set up the output file |
---|
| 191 | |
---|
| 192 | CALL check_nf90( nf90_open( TRIM(filenames(1)), nf90_share, ncid ) ) |
---|
| 193 | CALL check_nf90( nf90_inquire( ncid, ndims, nvars, natts ) ) |
---|
| 194 | |
---|
| 195 | !2.0 Read in the total number of processors the file is expecting and check its correct |
---|
| 196 | |
---|
| 197 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_number_total', ndomain_file ) ) |
---|
| 198 | IF( ndomain /= ndomain_file ) THEN |
---|
| 199 | WRITE(numerr,*) 'ERROR! : number of files to rebuild in file does not agree with namelist' |
---|
| 200 | WRITE(numerr,*) 'Attribute DOMAIN_number_total is : ', ndomain_file |
---|
| 201 | WRITE(numerr,*) 'Number of files specified in namelist is: ', ndomain |
---|
[6415] | 202 | STOP 9 |
---|
[2895] | 203 | ENDIF |
---|
| 204 | |
---|
| 205 | !2.1 Set up the output file |
---|
| 206 | |
---|
| 207 | CALL check_nf90( nf90_create( TRIM(filebase)//'.nc', nf90_64bit_offset, outid, chunksize=chunksize ) ) |
---|
| 208 | |
---|
| 209 | !2.2 Set up dimensions in output file |
---|
| 210 | |
---|
| 211 | !2.2.0 Find out how many dimensions are required to be rebuilt and which ones they are |
---|
| 212 | CALL check_nf90( nf90_inquire_attribute( ncid, nf90_global, 'DOMAIN_dimensions_ids', xtype, rbdims, attid ) ) |
---|
| 213 | |
---|
| 214 | ALLOCATE(rebuild_dims(rbdims)) |
---|
| 215 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_dimensions_ids', rebuild_dims ) ) |
---|
| 216 | |
---|
| 217 | ALLOCATE(global_sizes(rbdims)) |
---|
| 218 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_size_global', global_sizes ) ) |
---|
| 219 | IF (l_verbose) WRITE(numout,*) 'Size of global arrays: ', global_sizes |
---|
| 220 | |
---|
| 221 | |
---|
| 222 | !2.2.1 Copy the dimensions into the output file apart from rebuild_dims() which are dimensioned globally |
---|
| 223 | ALLOCATE(indimlens(ndims), indimnames(ndims), outdimlens(ndims)) |
---|
| 224 | CALL check_nf90( nf90_inquire( ncid, unlimitedDimId = unlimitedDimId ) ) |
---|
| 225 | istop = 0 |
---|
| 226 | DO idim = 1, ndims |
---|
| 227 | CALL check_nf90( nf90_inquire_dimension( ncid, idim, dimname, dimlen ) ) |
---|
| 228 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_size_local', local_sizes ) ) |
---|
| 229 | indimlens(idim) = dimlen |
---|
| 230 | indimnames(idim) = dimname |
---|
| 231 | IF (l_findDims) THEN |
---|
| 232 | IF( idim == rebuild_dims(1) ) THEN |
---|
| 233 | IF( dimlen == local_sizes(1) ) THEN |
---|
| 234 | dimlen = global_sizes(1) |
---|
| 235 | ELSE |
---|
| 236 | istop = 1 |
---|
| 237 | ENDIF |
---|
| 238 | ENDIF |
---|
| 239 | IF( rbdims > 1 .AND. idim == rebuild_dims(2) ) THEN |
---|
| 240 | IF( dimlen == local_sizes(2) ) THEN |
---|
| 241 | dimlen = global_sizes(2) |
---|
| 242 | ELSE |
---|
| 243 | istop = 1 |
---|
| 244 | ENDIF |
---|
| 245 | ENDIF |
---|
| 246 | ELSE ! l_findDims = false |
---|
[3019] | 247 | IF( TRIM(dimname) == TRIM(dims(1))) THEN |
---|
| 248 | dimlen = global_sizes(1) |
---|
| 249 | rebuild_dims(1) = idim |
---|
| 250 | ENDIF |
---|
| 251 | IF( rbdims > 1 .AND. TRIM(dimname) == TRIM(dims(2))) THEN |
---|
| 252 | dimlen = global_sizes(2) |
---|
| 253 | rebuild_dims(2) = idim |
---|
| 254 | ENDIF |
---|
[2895] | 255 | ENDIF |
---|
| 256 | |
---|
| 257 | IF( idim == unlimitedDimId ) THEN |
---|
| 258 | CALL check_nf90( nf90_def_dim( outid, dimname, nf90_unlimited, dimid) ) |
---|
[3025] | 259 | nmax_unlimited = dimlen |
---|
[2895] | 260 | ELSE |
---|
| 261 | CALL check_nf90( nf90_def_dim( outid, dimname, dimlen, dimid) ) |
---|
| 262 | ENDIF |
---|
| 263 | outdimlens(idim) = dimlen |
---|
| 264 | END DO |
---|
| 265 | |
---|
| 266 | IF( istop == 1 ) THEN |
---|
| 267 | WRITE(numerr,*) 'ERROR! : DOMAIN_local_sizes attribute does not match rebuild dimension lengths in the first file' |
---|
| 268 | WRITE(numerr,*) 'Attribute DOMAIN_local_sizes is : ', local_sizes |
---|
| 269 | WRITE(numerr,*) 'Dimensions to be rebuilt are of size : ', outdimlens(rebuild_dims(1)), outdimlens(rebuild_dims(2)) |
---|
[6415] | 270 | STOP 9 |
---|
[2895] | 271 | ENDIF |
---|
| 272 | |
---|
| 273 | IF (l_findDims) THEN |
---|
| 274 | IF (l_verbose) WRITE(numout,*) 'Finding rebuild dimensions from the first file...' |
---|
| 275 | ELSE |
---|
| 276 | IF (l_verbose) WRITE(numout,*) 'Using rebuild dimensions given in namelist...' |
---|
| 277 | ENDIF |
---|
| 278 | |
---|
| 279 | IF( rbdims > 1 ) THEN |
---|
| 280 | IF (l_verbose) WRITE(numout,*) 'Rebuilding across dimensions '//TRIM(indimnames(rebuild_dims(1)))// & |
---|
| 281 | & ' and '//TRIM(indimnames(rebuild_dims(2))) |
---|
| 282 | ELSE |
---|
| 283 | IF (l_verbose) WRITE(numout,*) 'Rebuilding across dimension '//TRIM(indimnames(rebuild_dims(1))) |
---|
| 284 | ENDIF |
---|
| 285 | |
---|
| 286 | !2.2.2 Copy the global attributes into the output file, apart from those beginning with DOMAIN_ |
---|
| 287 | ! Also need to change the file_name attribute and the TimeStamp attribute. |
---|
| 288 | DO attid = 1, natts |
---|
| 289 | CALL check_nf90( nf90_inq_attname( ncid, nf90_global, attid, attname ) ) |
---|
| 290 | IF( INDEX( attname, "DOMAIN_" ) == 1 ) CYCLE |
---|
| 291 | IF( INDEX( attname, "file_name") == 1 ) CYCLE |
---|
| 292 | IF( INDEX( attname, "associate_file") == 1 ) CYCLE |
---|
| 293 | IF (l_verbose) WRITE(numout,*) 'Copying attribute '//TRIM(attname)//' into destination file...' |
---|
| 294 | CALL check_nf90( nf90_copy_att( ncid, nf90_global, attname, outid, nf90_global ) ) |
---|
| 295 | END DO |
---|
| 296 | CALL check_nf90( nf90_put_att( outid, nf90_global, "file_name", TRIM(filebase)//'.nc') ) |
---|
| 297 | IF (l_verbose) WRITE(numout,*) 'Writing new file_name attribute' |
---|
| 298 | CALL DATE_AND_TIME ( date=date, time=time, zone=zone ) |
---|
| 299 | timestamp = date(7:8) // "/" // date(5:6) // "/" // date(1:4) // " " // & |
---|
| 300 | time(1:2) // ":" // time(3:4) // ":" // time(5:6) // " " // & |
---|
| 301 | zone |
---|
| 302 | CALL check_nf90( nf90_put_att( outid, nf90_global, "TimeStamp", timestamp ) ) |
---|
| 303 | IF (l_verbose) WRITE(numout,*) 'Writing new TimeStamp attribute' |
---|
| 304 | |
---|
| 305 | !2.2.3 Copy the variable definitions and attributes into the output file. |
---|
| 306 | DO jv = 1, nvars |
---|
| 307 | CALL check_nf90( nf90_inquire_variable( ncid, jv, varname, xtype, ndims, dimids, natts ) ) |
---|
| 308 | ALLOCATE(outdimids(ndims)) |
---|
| 309 | DO idim = 1, ndims |
---|
| 310 | outdimids(idim) = dimids(idim) |
---|
| 311 | END DO |
---|
| 312 | CALL check_nf90( nf90_def_var( outid, varname, xtype, outdimids, varid ) ) |
---|
| 313 | DEALLOCATE(outdimids) |
---|
| 314 | IF (l_verbose) WRITE(numout,*) 'Defining variable '//TRIM(varname)//'...' |
---|
| 315 | IF( natts > 0 ) THEN |
---|
| 316 | DO attid = 1, natts |
---|
| 317 | CALL check_nf90( nf90_inq_attname( ncid, varid, attid, attname ) ) |
---|
| 318 | CALL check_nf90( nf90_copy_att( ncid, varid, attname, outid, varid ) ) |
---|
| 319 | END DO |
---|
| 320 | ENDIF |
---|
| 321 | END DO |
---|
| 322 | |
---|
| 323 | !2.3 End definitions in output file and copy 1st file ncid to the inncids array |
---|
| 324 | |
---|
| 325 | CALL check_nf90( nf90_enddef( outid ) ) |
---|
| 326 | inncids(1) = ncid |
---|
| 327 | IF (l_verbose) WRITE(numout,*) 'Finished defining output file.' |
---|
| 328 | |
---|
| 329 | !--------------------------------------------------------------------------- |
---|
| 330 | !3. Read in data from each file for each variable |
---|
| 331 | |
---|
| 332 | !3.1 Open each file and store the ncid in inncids array |
---|
| 333 | |
---|
| 334 | IF (l_verbose) WRITE(numout,*) 'Opening input files...' |
---|
| 335 | DO ifile = 2, ndomain |
---|
| 336 | CALL check_nf90( nf90_open( TRIM(filenames(ifile)), nf90_share, ncid, chunksize=chunksize ) ) |
---|
| 337 | inncids(ifile) = ncid |
---|
| 338 | END DO |
---|
| 339 | IF (l_verbose) WRITE(numout,*) 'All input files open.' |
---|
| 340 | |
---|
| 341 | DO jv = 1, nvars |
---|
| 342 | |
---|
| 343 | ValMin = 1.e10 |
---|
| 344 | ValMax = -1.e10 |
---|
| 345 | l_valid = .false. |
---|
| 346 | istop = nf90_noerr |
---|
[3025] | 347 | nt = 1 |
---|
| 348 | ntchunk = nmax_unlimited |
---|
| 349 | IF( nchunksize == 0 ) nchunksize = nmax_unlimited |
---|
[2895] | 350 | |
---|
| 351 | !3.2 Inquire variable to find out name and how many dimensions it has |
---|
| 352 | ! and importantly whether it contains the dimensions in rebuild_dims() |
---|
| 353 | |
---|
| 354 | ncid = inncids(1) |
---|
| 355 | CALL check_nf90( nf90_inquire_variable( ncid, jv, varname, xtype, ndims, dimids, natts ) ) |
---|
| 356 | |
---|
| 357 | l_noRebuild = .true. |
---|
| 358 | IF( ANY( dimids(1:ndims) == rebuild_dims(1) )) l_noRebuild = .false. |
---|
| 359 | IF( rbdims > 1 ) THEN |
---|
| 360 | IF( ANY( dimids(1:ndims) == rebuild_dims(2) )) l_noRebuild = .false. |
---|
| 361 | ENDIF |
---|
| 362 | |
---|
[3025] | 363 | !3.2.0 start while loop for time chunking |
---|
| 364 | |
---|
| 365 | DO WHILE( nt <= nmax_unlimited ) |
---|
| 366 | |
---|
| 367 | IF( ndims > 3 ) THEN |
---|
| 368 | ntchunk = MIN( nchunksize, nmax_unlimited + 1 - nt ) |
---|
| 369 | ENDIF |
---|
| 370 | |
---|
[2895] | 371 | IF (l_noRebuild) THEN |
---|
| 372 | |
---|
[3025] | 373 | IF( nchunksize == nmax_unlimited .OR. ndims <= 3 ) THEN |
---|
| 374 | IF (l_verbose) WRITE(numout,*) 'Copying data from variable '//TRIM(varname)//'...' |
---|
| 375 | ELSE |
---|
| 376 | IF (l_verbose) WRITE(numout,'(A,I3,A,I3,A)') ' Copying data from variable ' & |
---|
| 377 | & //TRIM(varname)//' for chunks ',nt,' to ',nt+ntchunk-1,' ...' |
---|
| 378 | ENDIF |
---|
[2895] | 379 | |
---|
| 380 | !3.2.1 If rebuilding not required then just need to read in variable |
---|
| 381 | ! for copying direct into output file after the OMP (files) loop. |
---|
| 382 | IF( ndims == 0 ) THEN |
---|
| 383 | |
---|
| 384 | SELECT CASE( xtype ) |
---|
| 385 | CASE( NF90_BYTE ) |
---|
[6415] | 386 | globaldata_0d_i1 = 0 |
---|
[2895] | 387 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_0d_i1 ) ) |
---|
| 388 | CASE( NF90_SHORT ) |
---|
[6415] | 389 | globaldata_0d_i2 = 0 |
---|
[2895] | 390 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_0d_i2 ) ) |
---|
| 391 | CASE( NF90_INT ) |
---|
[6415] | 392 | globaldata_0d_i4 = 0 |
---|
[2895] | 393 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_0d_i4 ) ) |
---|
| 394 | CASE( NF90_FLOAT ) |
---|
[6415] | 395 | globaldata_0d_sp = 0. |
---|
[2895] | 396 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_0d_sp ) ) |
---|
| 397 | CASE( NF90_DOUBLE ) |
---|
[6415] | 398 | globaldata_0d_dp = 0. |
---|
[2895] | 399 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_0d_dp ) ) |
---|
| 400 | CASE DEFAULT |
---|
| 401 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 402 | STOP 9 |
---|
[2895] | 403 | END SELECT |
---|
| 404 | |
---|
| 405 | ELSEIF( ndims == 1 ) THEN |
---|
| 406 | |
---|
| 407 | SELECT CASE( xtype ) |
---|
| 408 | CASE( NF90_BYTE ) |
---|
| 409 | ALLOCATE(globaldata_1d_i1(indimlens(dimids(1)))) |
---|
[6415] | 410 | globaldata_1d_i1(:) = 0 |
---|
[2895] | 411 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_1d_i1 ) ) |
---|
| 412 | CASE( NF90_SHORT ) |
---|
| 413 | ALLOCATE(globaldata_1d_i2(indimlens(dimids(1)))) |
---|
[6415] | 414 | globaldata_1d_i2(:) = 0 |
---|
[2895] | 415 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_1d_i2 ) ) |
---|
| 416 | CASE( NF90_INT ) |
---|
| 417 | ALLOCATE(globaldata_1d_i4(indimlens(dimids(1)))) |
---|
[6415] | 418 | globaldata_1d_i4(:) = 0 |
---|
[2895] | 419 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_1d_i4 ) ) |
---|
| 420 | CASE( NF90_FLOAT ) |
---|
| 421 | ALLOCATE(globaldata_1d_sp(indimlens(dimids(1)))) |
---|
[6415] | 422 | globaldata_1d_sp(:) = 0. |
---|
[2895] | 423 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_1d_sp ) ) |
---|
| 424 | CASE( NF90_DOUBLE ) |
---|
| 425 | ALLOCATE(globaldata_1d_dp(indimlens(dimids(1)))) |
---|
[6415] | 426 | globaldata_1d_dp(:) = 0. |
---|
[2895] | 427 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_1d_dp ) ) |
---|
| 428 | CASE DEFAULT |
---|
| 429 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 430 | STOP 9 |
---|
[2895] | 431 | END SELECT |
---|
| 432 | |
---|
| 433 | ELSEIF( ndims == 2 ) THEN |
---|
| 434 | |
---|
| 435 | SELECT CASE( xtype ) |
---|
| 436 | CASE( NF90_BYTE ) |
---|
| 437 | ALLOCATE(globaldata_2d_i1(indimlens(dimids(1)),indimlens(dimids(2)))) |
---|
[6415] | 438 | globaldata_2d_i1(:,:) = 0 |
---|
[2895] | 439 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_2d_i1 ) ) |
---|
| 440 | CASE( NF90_SHORT ) |
---|
| 441 | ALLOCATE(globaldata_2d_i2(indimlens(dimids(1)),indimlens(dimids(2)))) |
---|
[6415] | 442 | globaldata_2d_i2(:,:) = 0 |
---|
[2895] | 443 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_2d_i2 ) ) |
---|
| 444 | CASE( NF90_INT ) |
---|
| 445 | ALLOCATE(globaldata_2d_i4(indimlens(dimids(1)),indimlens(dimids(2)))) |
---|
[6415] | 446 | globaldata_2d_i4(:,:) = 0 |
---|
[2895] | 447 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_2d_i4 ) ) |
---|
| 448 | CASE( NF90_FLOAT ) |
---|
| 449 | ALLOCATE(globaldata_2d_sp(indimlens(dimids(1)),indimlens(dimids(2)))) |
---|
[6415] | 450 | globaldata_2d_sp(:,:) = 0. |
---|
[2895] | 451 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_2d_sp ) ) |
---|
| 452 | CASE( NF90_DOUBLE ) |
---|
| 453 | ALLOCATE(globaldata_2d_dp(indimlens(dimids(1)),indimlens(dimids(2)))) |
---|
[6415] | 454 | globaldata_2d_dp(:,:) = 0. |
---|
[2895] | 455 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_2d_dp ) ) |
---|
| 456 | CASE DEFAULT |
---|
| 457 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 458 | STOP 9 |
---|
[2895] | 459 | END SELECT |
---|
| 460 | |
---|
| 461 | ELSEIF( ndims == 3 ) THEN |
---|
| 462 | |
---|
| 463 | SELECT CASE( xtype ) |
---|
| 464 | CASE( NF90_BYTE ) |
---|
| 465 | ALLOCATE(globaldata_3d_i1(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
| 466 | & indimlens(dimids(3)))) |
---|
[6415] | 467 | globaldata_3d_i1(:,:,:) = 0 |
---|
[2895] | 468 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_3d_i1 ) ) |
---|
| 469 | CASE( NF90_SHORT ) |
---|
| 470 | ALLOCATE(globaldata_3d_i2(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
| 471 | & indimlens(dimids(3)))) |
---|
[6415] | 472 | globaldata_3d_i2(:,:,:) = 0 |
---|
[2895] | 473 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_3d_i2 ) ) |
---|
| 474 | CASE( NF90_INT ) |
---|
| 475 | ALLOCATE(globaldata_3d_i4(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
| 476 | & indimlens(dimids(3)))) |
---|
[6415] | 477 | globaldata_3d_i4(:,:,:) = 0 |
---|
[2895] | 478 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_3d_i4 ) ) |
---|
| 479 | CASE( NF90_FLOAT ) |
---|
| 480 | ALLOCATE(globaldata_3d_sp(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
| 481 | & indimlens(dimids(3)))) |
---|
[6415] | 482 | globaldata_3d_sp(:,:,:) = 0. |
---|
[2895] | 483 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_3d_sp ) ) |
---|
| 484 | CASE( NF90_DOUBLE ) |
---|
| 485 | ALLOCATE(globaldata_3d_dp(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
| 486 | & indimlens(dimids(3)))) |
---|
[6415] | 487 | globaldata_3d_dp(:,:,:) = 0. |
---|
[2895] | 488 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_3d_dp ) ) |
---|
| 489 | CASE DEFAULT |
---|
| 490 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 491 | STOP 9 |
---|
[2895] | 492 | END SELECT |
---|
| 493 | |
---|
| 494 | ELSEIF( ndims == 4 ) THEN |
---|
| 495 | |
---|
| 496 | SELECT CASE( xtype ) |
---|
| 497 | CASE( NF90_BYTE ) |
---|
| 498 | ALLOCATE(globaldata_4d_i1(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
[3025] | 499 | & indimlens(dimids(3)),ntchunk)) |
---|
[6415] | 500 | globaldata_4d_i1(:,:,:,:) = 0 |
---|
[3025] | 501 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_4d_i1, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 502 | CASE( NF90_SHORT ) |
---|
| 503 | ALLOCATE(globaldata_4d_i2(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
[3025] | 504 | & indimlens(dimids(3)),ntchunk)) |
---|
[6415] | 505 | globaldata_4d_i2(:,:,:,:) = 0 |
---|
[3025] | 506 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_4d_i2, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 507 | CASE( NF90_INT ) |
---|
| 508 | ALLOCATE(globaldata_4d_i4(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
[3025] | 509 | & indimlens(dimids(3)),ntchunk)) |
---|
[6415] | 510 | globaldata_4d_i4(:,:,:,:) = 0 |
---|
[3025] | 511 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_4d_i4, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 512 | CASE( NF90_FLOAT ) |
---|
| 513 | ALLOCATE(globaldata_4d_sp(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
[3025] | 514 | & indimlens(dimids(3)),ntchunk)) |
---|
[6415] | 515 | globaldata_4d_sp(:,:,:,:) = 0. |
---|
[3025] | 516 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_4d_sp, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 517 | CASE( NF90_DOUBLE ) |
---|
| 518 | ALLOCATE(globaldata_4d_dp(indimlens(dimids(1)),indimlens(dimids(2)), & |
---|
[3025] | 519 | & indimlens(dimids(3)),ntchunk)) |
---|
[6415] | 520 | globaldata_4d_dp(:,:,:,:) = 0. |
---|
[3025] | 521 | CALL check_nf90( nf90_get_var( ncid, jv, globaldata_4d_dp, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 522 | CASE DEFAULT |
---|
| 523 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 524 | STOP 9 |
---|
[2895] | 525 | END SELECT |
---|
| 526 | |
---|
| 527 | ENDIF |
---|
| 528 | |
---|
| 529 | ELSE ! l_noRebuild = .false. |
---|
| 530 | |
---|
| 531 | !3.2.2 For variables that require rebuilding we need to read in from all ndomain files |
---|
| 532 | ! Here we allocate global variables ahead of looping over files |
---|
[3025] | 533 | IF( nchunksize == nmax_unlimited .OR. ndims <= 3 ) THEN |
---|
| 534 | IF (l_verbose) WRITE(numout,*) 'Rebuilding data from variable '//TRIM(varname)//'...' |
---|
| 535 | ELSE |
---|
| 536 | IF (l_verbose) WRITE(numout,'(A,I3,A,I3,A)') ' Rebuilding data from variable ' & |
---|
| 537 | & //TRIM(varname)//' for chunks ',nt,' to ',nt+ntchunk-1,' ...' |
---|
| 538 | ENDIF |
---|
[2895] | 539 | IF( ndims == 1 ) THEN |
---|
| 540 | |
---|
| 541 | SELECT CASE( xtype ) |
---|
| 542 | CASE( NF90_BYTE ) |
---|
| 543 | ALLOCATE(globaldata_1d_i1(outdimlens(dimids(1)))) |
---|
[6415] | 544 | globaldata_1d_i1(:) = 0 |
---|
[2895] | 545 | CASE( NF90_SHORT ) |
---|
| 546 | ALLOCATE(globaldata_1d_i2(outdimlens(dimids(1)))) |
---|
[6415] | 547 | globaldata_1d_i2(:) = 0 |
---|
[2895] | 548 | CASE( NF90_INT ) |
---|
| 549 | ALLOCATE(globaldata_1d_i4(outdimlens(dimids(1)))) |
---|
[6415] | 550 | globaldata_1d_i4(:) = 0 |
---|
[2895] | 551 | CASE( NF90_FLOAT ) |
---|
| 552 | ALLOCATE(globaldata_1d_sp(outdimlens(dimids(1)))) |
---|
[6415] | 553 | globaldata_1d_sp(:) = 0. |
---|
[2895] | 554 | CASE( NF90_DOUBLE ) |
---|
| 555 | ALLOCATE(globaldata_1d_dp(outdimlens(dimids(1)))) |
---|
[6415] | 556 | globaldata_1d_dp(:) = 0. |
---|
[2895] | 557 | CASE DEFAULT |
---|
| 558 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 559 | STOP 9 |
---|
[2895] | 560 | END SELECT |
---|
| 561 | |
---|
| 562 | ELSEIF( ndims == 2 ) THEN |
---|
| 563 | |
---|
| 564 | SELECT CASE( xtype ) |
---|
| 565 | CASE( NF90_BYTE ) |
---|
| 566 | ALLOCATE(globaldata_2d_i1(outdimlens(dimids(1)),outdimlens(dimids(2)))) |
---|
[6415] | 567 | globaldata_2d_i1(:,:) = 0 |
---|
[2895] | 568 | CASE( NF90_SHORT ) |
---|
| 569 | ALLOCATE(globaldata_2d_i2(outdimlens(dimids(1)),outdimlens(dimids(2)))) |
---|
[6415] | 570 | globaldata_2d_i2(:,:) = 0 |
---|
[2895] | 571 | CASE( NF90_INT ) |
---|
| 572 | ALLOCATE(globaldata_2d_i4(outdimlens(dimids(1)),outdimlens(dimids(2)))) |
---|
[6415] | 573 | globaldata_2d_i4(:,:) = 0 |
---|
[2895] | 574 | CASE( NF90_FLOAT ) |
---|
| 575 | ALLOCATE(globaldata_2d_sp(outdimlens(dimids(1)),outdimlens(dimids(2)))) |
---|
[6415] | 576 | globaldata_2d_sp(:,:) = 0. |
---|
[2895] | 577 | CASE( NF90_DOUBLE ) |
---|
| 578 | ALLOCATE(globaldata_2d_dp(outdimlens(dimids(1)),outdimlens(dimids(2)))) |
---|
[6415] | 579 | globaldata_2d_dp(:,:) = 0. |
---|
[2895] | 580 | CASE DEFAULT |
---|
| 581 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 582 | STOP 9 |
---|
[2895] | 583 | END SELECT |
---|
| 584 | |
---|
| 585 | ELSEIF( ndims == 3 ) THEN |
---|
| 586 | |
---|
| 587 | SELECT CASE( xtype ) |
---|
| 588 | CASE( NF90_BYTE ) |
---|
| 589 | ALLOCATE(globaldata_3d_i1(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
| 590 | & outdimlens(dimids(3)))) |
---|
[6415] | 591 | globaldata_3d_i1(:,:,:) = 0 |
---|
[2895] | 592 | CASE( NF90_SHORT ) |
---|
| 593 | ALLOCATE(globaldata_3d_i2(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
| 594 | & outdimlens(dimids(3)))) |
---|
[6415] | 595 | globaldata_3d_i2(:,:,:) = 0 |
---|
[2895] | 596 | CASE( NF90_INT ) |
---|
| 597 | ALLOCATE(globaldata_3d_i4(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
| 598 | & outdimlens(dimids(3)))) |
---|
[6415] | 599 | globaldata_3d_i4(:,:,:) = 0 |
---|
[2895] | 600 | CASE( NF90_FLOAT ) |
---|
| 601 | ALLOCATE(globaldata_3d_sp(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
| 602 | & outdimlens(dimids(3)))) |
---|
[6415] | 603 | globaldata_3d_sp(:,:,:) = 0. |
---|
[2895] | 604 | CASE( NF90_DOUBLE ) |
---|
| 605 | ALLOCATE(globaldata_3d_dp(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
| 606 | & outdimlens(dimids(3)))) |
---|
[6415] | 607 | globaldata_3d_dp(:,:,:) = 0. |
---|
[2895] | 608 | CASE DEFAULT |
---|
| 609 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 610 | STOP 9 |
---|
[2895] | 611 | END SELECT |
---|
| 612 | |
---|
| 613 | ELSEIF( ndims == 4 ) THEN |
---|
| 614 | |
---|
| 615 | SELECT CASE( xtype ) |
---|
| 616 | CASE( NF90_BYTE ) |
---|
| 617 | ALLOCATE(globaldata_4d_i1(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
[3025] | 618 | & outdimlens(dimids(3)),ntchunk)) |
---|
[6415] | 619 | globaldata_4d_i1(:,:,:,:) = 0 |
---|
[2895] | 620 | CASE( NF90_SHORT ) |
---|
| 621 | ALLOCATE(globaldata_4d_i2(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
[3025] | 622 | & outdimlens(dimids(3)),ntchunk)) |
---|
[6415] | 623 | globaldata_4d_i2(:,:,:,:) = 0 |
---|
[2895] | 624 | CASE( NF90_INT ) |
---|
| 625 | ALLOCATE(globaldata_4d_i4(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
[3025] | 626 | & outdimlens(dimids(3)),ntchunk)) |
---|
[6415] | 627 | globaldata_4d_i4(:,:,:,:) = 0 |
---|
[2895] | 628 | CASE( NF90_FLOAT ) |
---|
| 629 | ALLOCATE(globaldata_4d_sp(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
[3025] | 630 | & outdimlens(dimids(3)),ntchunk)) |
---|
[6415] | 631 | globaldata_4d_sp(:,:,:,:) = 0. |
---|
[2895] | 632 | CASE( NF90_DOUBLE ) |
---|
| 633 | ALLOCATE(globaldata_4d_dp(outdimlens(dimids(1)),outdimlens(dimids(2)), & |
---|
[3025] | 634 | & outdimlens(dimids(3)),ntchunk)) |
---|
[6415] | 635 | globaldata_4d_dp(:,:,:,:) = 0. |
---|
[2895] | 636 | CASE DEFAULT |
---|
| 637 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 638 | STOP 9 |
---|
[2895] | 639 | END SELECT |
---|
| 640 | ELSE |
---|
| 641 | WRITE(numerr,*) 'ERROR! : A netcdf variable has more than 4 dimensions which is not taken into account' |
---|
[6415] | 642 | STOP 9 |
---|
[2895] | 643 | ENDIF |
---|
| 644 | |
---|
| 645 | !$OMP PARALLEL DO DEFAULT(NONE) & |
---|
| 646 | !$OMP& PRIVATE(ifile,ncid,xtype,start_pos,local_sizes,InMin,InMax,natts, & |
---|
[3019] | 647 | !$OMP& ndims,attid,attname,dimids,idim,dimname,dimlen,unlimitedDimId, & |
---|
[2895] | 648 | !$OMP& halo_start,halo_end,idomain,jdomain,rdomain,di,dj,dr, & |
---|
| 649 | !$OMP& localdata_1d_i2,localdata_1d_i4,localdata_1d_sp,localdata_1d_dp, & |
---|
| 650 | !$OMP& localdata_2d_i2,localdata_2d_i4,localdata_2d_sp,localdata_2d_dp, & |
---|
| 651 | !$OMP& localdata_3d_i2,localdata_3d_i4,localdata_3d_sp,localdata_3d_dp, & |
---|
| 652 | !$OMP& localdata_4d_i2,localdata_4d_i4,localdata_4d_sp,localdata_4d_dp, & |
---|
| 653 | !$OMP& localdata_1d_i1,localdata_2d_i1,localdata_3d_i1,localdata_4d_i1) & |
---|
[3019] | 654 | !$OMP& SHARED(jv,nvars,varname,filenames,ValMin,ValMax,indimlens,outdimlens,rbdims, & |
---|
[2895] | 655 | !$OMP& ndomain,outid,chunksize,istop,l_valid,nthreads,inncids,rebuild_dims, & |
---|
| 656 | !$OMP& globaldata_1d_i2,globaldata_1d_i4,globaldata_1d_sp,globaldata_1d_dp, & |
---|
| 657 | !$OMP& globaldata_2d_i2,globaldata_2d_i4,globaldata_2d_sp,globaldata_2d_dp, & |
---|
| 658 | !$OMP& globaldata_3d_i2,globaldata_3d_i4,globaldata_3d_sp,globaldata_3d_dp, & |
---|
| 659 | !$OMP& globaldata_4d_i2,globaldata_4d_i4,globaldata_4d_sp,globaldata_4d_dp, & |
---|
[3025] | 660 | !$OMP& globaldata_1d_i1,globaldata_2d_i1,globaldata_3d_i1,globaldata_4d_i1, & |
---|
| 661 | !$OMP& ntchunk,nt,nmax_unlimited) |
---|
[2895] | 662 | |
---|
| 663 | DO ifile = 1, ndomain |
---|
| 664 | |
---|
| 665 | ncid = inncids(ifile) |
---|
| 666 | !$OMP CRITICAL |
---|
| 667 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_size_local', local_sizes ), istop ) |
---|
| 668 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_position_first', start_pos ), istop ) |
---|
| 669 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_halo_size_start', halo_start ), istop ) |
---|
| 670 | CALL check_nf90( nf90_get_att( ncid, nf90_global, 'DOMAIN_halo_size_end', halo_end ), istop ) |
---|
| 671 | CALL check_nf90( nf90_inquire_variable( ncid, jv, varname, xtype, ndims, dimids, natts ), istop ) |
---|
| 672 | CALL check_nf90( nf90_inquire( ncid, unlimitedDimId = unlimitedDimId ), istop ) |
---|
| 673 | !$OMP END CRITICAL |
---|
| 674 | |
---|
| 675 | ! set defaults for rebuilding so that i is 1st, j 2nd |
---|
| 676 | di=1 |
---|
| 677 | dj=2 |
---|
| 678 | |
---|
| 679 | IF( rbdims == 1 ) THEN |
---|
| 680 | ! override defaults above and set other variables |
---|
| 681 | start_pos(2) = 1 |
---|
| 682 | local_sizes(2) = outdimlens(3-dimids(2)) |
---|
| 683 | halo_end(2) = 0 |
---|
| 684 | halo_start(2) = 0 |
---|
| 685 | di=rebuild_dims(1) |
---|
| 686 | dj=3-di |
---|
| 687 | ENDIF |
---|
| 688 | |
---|
| 689 | !3.3.1 Generate local domain interior sizes from local_sizes and halo sizes |
---|
| 690 | ! idomain defines the 1st and last interior points in the i direction and |
---|
| 691 | ! jdomain defines the 1st and last interior points in the j direction |
---|
| 692 | |
---|
| 693 | idomain(1) = 1 + halo_start(di) |
---|
| 694 | idomain(2) = local_sizes(di) - halo_end(di) |
---|
| 695 | jdomain(1) = 1 + halo_start(dj) |
---|
| 696 | jdomain(2) = local_sizes(dj) - halo_end(dj) |
---|
| 697 | |
---|
| 698 | !3.3.2 For rbdims or more dimensions put the data array from this input file into the correct |
---|
| 699 | ! part of the output data array. Assume the first dimensions are those to be rebuilt. |
---|
| 700 | |
---|
| 701 | IF( ndims == 1 ) THEN |
---|
| 702 | |
---|
| 703 | IF( rebuild_dims(1) == 1 ) THEN |
---|
| 704 | dr = di |
---|
| 705 | rdomain = idomain |
---|
| 706 | ELSE |
---|
| 707 | dr = dj |
---|
| 708 | rdomain = jdomain |
---|
| 709 | ENDIF |
---|
| 710 | |
---|
| 711 | SELECT CASE( xtype ) |
---|
| 712 | CASE( NF90_BYTE ) |
---|
| 713 | ALLOCATE(localdata_1d_i1(local_sizes(dr))) |
---|
| 714 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_1d_i1 ), istop ) |
---|
| 715 | DO jr = rdomain(1), rdomain(2) |
---|
| 716 | globaldata_1d_i1(start_pos(dr) + jr - 1) = localdata_1d_i1(jr) |
---|
| 717 | END DO |
---|
| 718 | DEALLOCATE(localdata_1d_i1) |
---|
| 719 | CASE( NF90_SHORT ) |
---|
| 720 | ALLOCATE(localdata_1d_i2(local_sizes(dr))) |
---|
| 721 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_1d_i2 ), istop ) |
---|
| 722 | DO jr = rdomain(1), rdomain(2) |
---|
| 723 | globaldata_1d_i2(start_pos(dr) + jr - 1) = localdata_1d_i2(jr) |
---|
| 724 | END DO |
---|
| 725 | DEALLOCATE(localdata_1d_i2) |
---|
| 726 | CASE( NF90_INT ) |
---|
| 727 | ALLOCATE(localdata_1d_i4(local_sizes(dr))) |
---|
| 728 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_1d_i4 ), istop ) |
---|
| 729 | DO jr = rdomain(1), rdomain(2) |
---|
| 730 | globaldata_1d_i4(start_pos(dr) + jr - 1) = localdata_1d_i4(jr) |
---|
| 731 | END DO |
---|
| 732 | DEALLOCATE(localdata_1d_i4) |
---|
| 733 | CASE( NF90_FLOAT ) |
---|
| 734 | ALLOCATE(localdata_1d_sp(local_sizes(dr))) |
---|
| 735 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_1d_sp ), istop ) |
---|
| 736 | DO jr = rdomain(1), rdomain(2) |
---|
| 737 | globaldata_1d_sp(start_pos(dr) + jr - 1) = localdata_1d_sp(jr) |
---|
| 738 | END DO |
---|
| 739 | DEALLOCATE(localdata_1d_sp) |
---|
| 740 | CASE( NF90_DOUBLE ) |
---|
| 741 | ALLOCATE(localdata_1d_dp(local_sizes(dr))) |
---|
| 742 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_1d_dp ), istop ) |
---|
| 743 | DO jr = rdomain(1), rdomain(2) |
---|
| 744 | globaldata_1d_dp(start_pos(dr) + jr - 1) = localdata_1d_dp(jr) |
---|
| 745 | END DO |
---|
| 746 | DEALLOCATE(localdata_1d_dp) |
---|
| 747 | CASE DEFAULT |
---|
| 748 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
| 749 | istop = istop + 1 |
---|
| 750 | END SELECT |
---|
| 751 | |
---|
| 752 | ELSEIF( ndims == 2 ) THEN |
---|
| 753 | |
---|
| 754 | SELECT CASE( xtype ) |
---|
| 755 | CASE( NF90_BYTE ) |
---|
| 756 | ALLOCATE(localdata_2d_i1(local_sizes(di),local_sizes(dj))) |
---|
| 757 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_2d_i1 ), istop ) |
---|
| 758 | DO jj = jdomain(1), jdomain(2) |
---|
| 759 | DO ji = idomain(1), idomain(2) |
---|
| 760 | globaldata_2d_i1(start_pos(di) + ji - 1, start_pos(dj) + jj - 1) = localdata_2d_i1(ji,jj) |
---|
| 761 | END DO |
---|
| 762 | END DO |
---|
| 763 | DEALLOCATE(localdata_2d_i1) |
---|
| 764 | CASE( NF90_SHORT ) |
---|
| 765 | ALLOCATE(localdata_2d_i2(local_sizes(di),local_sizes(dj))) |
---|
| 766 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_2d_i2 ), istop ) |
---|
| 767 | DO jj = jdomain(1), jdomain(2) |
---|
| 768 | DO ji = idomain(1), idomain(2) |
---|
| 769 | globaldata_2d_i2(start_pos(di) + ji - 1, start_pos(dj) + jj - 1) = localdata_2d_i2(ji,jj) |
---|
| 770 | END DO |
---|
| 771 | END DO |
---|
| 772 | DEALLOCATE(localdata_2d_i2) |
---|
| 773 | CASE( NF90_INT ) |
---|
| 774 | ALLOCATE(localdata_2d_i4(local_sizes(di),local_sizes(dj))) |
---|
| 775 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_2d_i4 ), istop ) |
---|
| 776 | DO jj = jdomain(1), jdomain(2) |
---|
| 777 | DO ji = idomain(1), idomain(2) |
---|
| 778 | globaldata_2d_i4(start_pos(di) + ji - 1, start_pos(dj) + jj - 1) = localdata_2d_i4(ji,jj) |
---|
| 779 | END DO |
---|
| 780 | END DO |
---|
| 781 | DEALLOCATE(localdata_2d_i4) |
---|
| 782 | CASE( NF90_FLOAT ) |
---|
| 783 | ALLOCATE(localdata_2d_sp(local_sizes(di),local_sizes(dj))) |
---|
| 784 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_2d_sp ), istop ) |
---|
| 785 | DO jj = jdomain(1), jdomain(2) |
---|
| 786 | DO ji = idomain(1), idomain(2) |
---|
| 787 | globaldata_2d_sp(start_pos(di) + ji - 1, start_pos(dj) + jj - 1) = localdata_2d_sp(ji,jj) |
---|
| 788 | END DO |
---|
| 789 | END DO |
---|
| 790 | DEALLOCATE(localdata_2d_sp) |
---|
| 791 | CASE( NF90_DOUBLE ) |
---|
| 792 | ALLOCATE(localdata_2d_dp(local_sizes(di),local_sizes(dj))) |
---|
| 793 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_2d_dp ), istop ) |
---|
| 794 | DO jj = jdomain(1), jdomain(2) |
---|
| 795 | DO ji = idomain(1), idomain(2) |
---|
| 796 | globaldata_2d_dp(start_pos(di) + ji - 1, start_pos(dj) + jj - 1) = localdata_2d_dp(ji,jj) |
---|
| 797 | END DO |
---|
| 798 | END DO |
---|
| 799 | DEALLOCATE(localdata_2d_dp) |
---|
| 800 | CASE DEFAULT |
---|
| 801 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
| 802 | istop = istop + 1 |
---|
| 803 | END SELECT |
---|
| 804 | |
---|
| 805 | ELSEIF( ndims == 3 ) THEN |
---|
| 806 | |
---|
| 807 | SELECT CASE( xtype ) |
---|
| 808 | CASE( NF90_BYTE ) |
---|
| 809 | ALLOCATE(localdata_3d_i1(local_sizes(di),local_sizes(dj),indimlens(dimids(3)))) |
---|
| 810 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_3d_i1 ), istop ) |
---|
| 811 | !$OMP PARALLEL DO DEFAULT(NONE) PRIVATE(ji,jj,jk) & |
---|
| 812 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_3d_i1,localdata_3d_i1,di,dj) |
---|
| 813 | DO jk = 1, indimlens(dimids(3)) |
---|
| 814 | DO jj = jdomain(1), jdomain(2) |
---|
| 815 | DO ji = idomain(1), idomain(2) |
---|
| 816 | globaldata_3d_i1(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk) = localdata_3d_i1(ji,jj,jk) |
---|
| 817 | END DO |
---|
| 818 | END DO |
---|
| 819 | END DO |
---|
| 820 | !$OMP END PARALLEL DO |
---|
| 821 | DEALLOCATE(localdata_3d_i1) |
---|
| 822 | CASE( NF90_SHORT ) |
---|
| 823 | ALLOCATE(localdata_3d_i2(local_sizes(di),local_sizes(dj),indimlens(dimids(3)))) |
---|
| 824 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_3d_i2 ), istop ) |
---|
| 825 | !$OMP PARALLEL DO DEFAULT(NONE) PRIVATE(ji,jj,jk) & |
---|
| 826 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_3d_i2,localdata_3d_i2,di,dj) |
---|
| 827 | DO jk = 1, indimlens(dimids(3)) |
---|
| 828 | DO jj = jdomain(1), jdomain(2) |
---|
| 829 | DO ji = idomain(1), idomain(2) |
---|
| 830 | globaldata_3d_i2(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk) = localdata_3d_i2(ji,jj,jk) |
---|
| 831 | END DO |
---|
| 832 | END DO |
---|
| 833 | END DO |
---|
| 834 | !$OMP END PARALLEL DO |
---|
| 835 | DEALLOCATE(localdata_3d_i2) |
---|
| 836 | CASE( NF90_INT ) |
---|
| 837 | ALLOCATE(localdata_3d_i4(local_sizes(di),local_sizes(dj),indimlens(dimids(3)))) |
---|
| 838 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_3d_i4 ), istop ) |
---|
| 839 | !$OMP PARALLEL DO DEFAULT(NONE) PRIVATE(ji,jj,jk) & |
---|
| 840 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_3d_i4,localdata_3d_i4,di,dj) |
---|
| 841 | DO jk = 1, indimlens(dimids(3)) |
---|
| 842 | DO jj = jdomain(1), jdomain(2) |
---|
| 843 | DO ji = idomain(1), idomain(2) |
---|
| 844 | globaldata_3d_i4(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk) = localdata_3d_i4(ji,jj,jk) |
---|
| 845 | END DO |
---|
| 846 | END DO |
---|
| 847 | END DO |
---|
| 848 | !$OMP END PARALLEL DO |
---|
| 849 | DEALLOCATE(localdata_3d_i4) |
---|
| 850 | CASE( NF90_FLOAT ) |
---|
| 851 | ALLOCATE(localdata_3d_sp(local_sizes(di),local_sizes(dj),indimlens(dimids(3)))) |
---|
| 852 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_3d_sp ), istop ) |
---|
| 853 | !$OMP PARALLEL DO DEFAULT(NONE) PRIVATE(ji,jj,jk) & |
---|
| 854 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_3d_sp,localdata_3d_sp,di,dj) |
---|
| 855 | DO jk = 1, indimlens(dimids(3)) |
---|
| 856 | DO jj = jdomain(1), jdomain(2) |
---|
| 857 | DO ji = idomain(1), idomain(2) |
---|
| 858 | globaldata_3d_sp(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk) = localdata_3d_sp(ji,jj,jk) |
---|
| 859 | END DO |
---|
| 860 | END DO |
---|
| 861 | END DO |
---|
| 862 | !$OMP END PARALLEL DO |
---|
| 863 | DEALLOCATE(localdata_3d_sp) |
---|
| 864 | CASE( NF90_DOUBLE ) |
---|
| 865 | ALLOCATE(localdata_3d_dp(local_sizes(di),local_sizes(dj),indimlens(dimids(3)))) |
---|
| 866 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_3d_dp ), istop ) |
---|
| 867 | !$OMP PARALLEL DO DEFAULT(NONE) PRIVATE(ji,jj,jk) & |
---|
| 868 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_3d_dp,localdata_3d_dp,di,dj) |
---|
| 869 | DO jk = 1, indimlens(dimids(3)) |
---|
| 870 | DO jj = jdomain(1), jdomain(2) |
---|
| 871 | DO ji = idomain(1), idomain(2) |
---|
| 872 | globaldata_3d_dp(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk) = localdata_3d_dp(ji,jj,jk) |
---|
| 873 | END DO |
---|
| 874 | END DO |
---|
| 875 | END DO |
---|
| 876 | !$OMP END PARALLEL DO |
---|
| 877 | DEALLOCATE(localdata_3d_dp) |
---|
| 878 | CASE DEFAULT |
---|
| 879 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
| 880 | istop = istop + 1 |
---|
| 881 | END SELECT |
---|
| 882 | |
---|
| 883 | ELSEIF (ndims == 4) THEN |
---|
| 884 | |
---|
| 885 | SELECT CASE( xtype ) |
---|
| 886 | CASE( NF90_BYTE ) |
---|
| 887 | ALLOCATE(localdata_4d_i1(local_sizes(di),local_sizes(dj), & |
---|
[3025] | 888 | & indimlens(dimids(3)),ntchunk)) |
---|
| 889 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_4d_i1, start=(/1,1,1,nt/) ), istop ) |
---|
[2895] | 890 | !$OMP PARALLEL DEFAULT(NONE) PRIVATE(ji,jj,jk,jl) & |
---|
[3025] | 891 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_4d_i1,localdata_4d_i1,di,dj,nt,ntchunk) |
---|
| 892 | DO jl = 1, ntchunk |
---|
[2895] | 893 | !$OMP DO |
---|
| 894 | DO jk = 1, indimlens(dimids(3)) |
---|
| 895 | DO jj = jdomain(1), jdomain(2) |
---|
| 896 | DO ji = idomain(1), idomain(2) |
---|
| 897 | globaldata_4d_i1(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk, jl) = localdata_4d_i1(ji,jj,jk,jl) |
---|
| 898 | END DO |
---|
| 899 | END DO |
---|
| 900 | END DO |
---|
| 901 | !$OMP END DO nowait |
---|
| 902 | END DO |
---|
| 903 | !$OMP END PARALLEL |
---|
| 904 | DEALLOCATE(localdata_4d_i1) |
---|
| 905 | CASE( NF90_SHORT ) |
---|
| 906 | ALLOCATE(localdata_4d_i2(local_sizes(di),local_sizes(dj), & |
---|
[3025] | 907 | & indimlens(dimids(3)),ntchunk)) |
---|
| 908 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_4d_i2, start=(/1,1,1,nt/) ), istop ) |
---|
[2895] | 909 | !$OMP PARALLEL DEFAULT(NONE) PRIVATE(ji,jj,jk,jl) & |
---|
[3025] | 910 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_4d_i2,localdata_4d_i2,di,dj,nt,ntchunk) |
---|
| 911 | DO jl = 1, ntchunk |
---|
[2895] | 912 | !$OMP DO |
---|
| 913 | DO jk = 1, indimlens(dimids(3)) |
---|
| 914 | DO jj = jdomain(1), jdomain(2) |
---|
| 915 | DO ji = idomain(1), idomain(2) |
---|
| 916 | globaldata_4d_i2(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk, jl) = localdata_4d_i2(ji,jj,jk,jl) |
---|
| 917 | END DO |
---|
| 918 | END DO |
---|
| 919 | END DO |
---|
| 920 | !$OMP END DO nowait |
---|
| 921 | END DO |
---|
| 922 | !$OMP END PARALLEL |
---|
| 923 | DEALLOCATE(localdata_4d_i2) |
---|
| 924 | CASE( NF90_INT ) |
---|
| 925 | ALLOCATE(localdata_4d_i4(local_sizes(di),local_sizes(dj), & |
---|
[3025] | 926 | & indimlens(dimids(3)),ntchunk)) |
---|
| 927 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_4d_i4, start=(/1,1,1,nt/) ), istop ) |
---|
[2895] | 928 | !$OMP PARALLEL DEFAULT(NONE) PRIVATE(ji,jj,jk,jl) & |
---|
[3025] | 929 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_4d_i4,localdata_4d_i4,di,dj,nt,ntchunk) |
---|
| 930 | DO jl = 1, ntchunk |
---|
[2895] | 931 | !$OMP DO |
---|
| 932 | DO jk = 1, indimlens(dimids(3)) |
---|
| 933 | DO jj = jdomain(1), jdomain(2) |
---|
| 934 | DO ji = idomain(1), idomain(2) |
---|
| 935 | globaldata_4d_i4(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk, jl) = localdata_4d_i4(ji,jj,jk,jl) |
---|
| 936 | END DO |
---|
| 937 | END DO |
---|
| 938 | END DO |
---|
| 939 | !$OMP END DO nowait |
---|
| 940 | END DO |
---|
| 941 | !$OMP END PARALLEL |
---|
| 942 | DEALLOCATE(localdata_4d_i4) |
---|
| 943 | CASE( NF90_FLOAT ) |
---|
| 944 | ALLOCATE(localdata_4d_sp(local_sizes(di),local_sizes(dj), & |
---|
[3025] | 945 | & indimlens(dimids(3)),ntchunk)) |
---|
| 946 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_4d_sp, start=(/1,1,1,nt/) ), istop ) |
---|
[2895] | 947 | !$OMP PARALLEL DEFAULT(NONE) PRIVATE(ji,jj,jk,jl) & |
---|
[3025] | 948 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_4d_sp,localdata_4d_sp,di,dj,nt,ntchunk) |
---|
| 949 | DO jl = 1, ntchunk |
---|
[2895] | 950 | !$OMP DO |
---|
| 951 | DO jk = 1, indimlens(dimids(3)) |
---|
| 952 | DO jj = jdomain(1), jdomain(2) |
---|
| 953 | DO ji = idomain(1), idomain(2) |
---|
| 954 | globaldata_4d_sp(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk, jl) = localdata_4d_sp(ji,jj,jk,jl) |
---|
| 955 | END DO |
---|
| 956 | END DO |
---|
| 957 | END DO |
---|
| 958 | !$OMP END DO nowait |
---|
| 959 | END DO |
---|
| 960 | !$OMP END PARALLEL |
---|
| 961 | DEALLOCATE(localdata_4d_sp) |
---|
| 962 | CASE( NF90_DOUBLE ) |
---|
| 963 | ALLOCATE(localdata_4d_dp(local_sizes(di),local_sizes(dj), & |
---|
[3025] | 964 | & indimlens(dimids(3)),ntchunk)) |
---|
| 965 | CALL check_nf90( nf90_get_var( ncid, jv, localdata_4d_dp, start=(/1,1,1,nt/) ), istop ) |
---|
[2895] | 966 | !$OMP PARALLEL DEFAULT(NONE) PRIVATE(ji,jj,jk,jl) & |
---|
[3025] | 967 | !$OMP& SHARED(idomain,jdomain,indimlens,dimids,start_pos,globaldata_4d_dp,localdata_4d_dp,di,dj,nt,ntchunk) |
---|
| 968 | DO jl = 1, ntchunk |
---|
[2895] | 969 | !$OMP DO |
---|
| 970 | DO jk = 1, indimlens(dimids(3)) |
---|
| 971 | DO jj = jdomain(1), jdomain(2) |
---|
| 972 | DO ji = idomain(1), idomain(2) |
---|
| 973 | globaldata_4d_dp(start_pos(di) + ji - 1, start_pos(dj) + jj - 1, jk, jl) = localdata_4d_dp(ji,jj,jk,jl) |
---|
| 974 | END DO |
---|
| 975 | END DO |
---|
| 976 | END DO |
---|
| 977 | !$OMP END DO nowait |
---|
| 978 | END DO |
---|
| 979 | !$OMP END PARALLEL |
---|
| 980 | DEALLOCATE(localdata_4d_dp) |
---|
| 981 | CASE DEFAULT |
---|
| 982 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
| 983 | istop = istop + 1 |
---|
| 984 | END SELECT |
---|
| 985 | |
---|
| 986 | ENDIF ! l_noRebuild false |
---|
| 987 | |
---|
| 988 | !3.4 Work out if the valid_min and valid_max attributes exist for this variable. |
---|
| 989 | ! If they do then calculate the extrema over all input files. |
---|
| 990 | |
---|
| 991 | DO attid = 1, natts |
---|
| 992 | CALL check_nf90( nf90_inq_attname( ncid, jv, attid, attname ), istop ) |
---|
| 993 | IF( INDEX( attname, "valid_min" ) == 1 ) THEN |
---|
| 994 | CALL check_nf90( nf90_get_att( ncid, jv, attname, InMin), istop ) |
---|
| 995 | l_valid = .true. |
---|
| 996 | ENDIF |
---|
| 997 | IF( INDEX( attname, "valid_max" ) == 1 ) THEN |
---|
| 998 | CALL check_nf90( nf90_get_att( ncid, jv, attname, InMax ), istop ) |
---|
| 999 | l_valid = .true. |
---|
| 1000 | ENDIF |
---|
| 1001 | END DO |
---|
| 1002 | |
---|
| 1003 | IF (l_valid) THEN |
---|
| 1004 | !$OMP CRITICAL |
---|
| 1005 | IF( InMin < ValMin ) ValMin = InMin |
---|
| 1006 | IF( InMax > ValMax ) ValMax = InMax |
---|
| 1007 | !$OMP END CRITICAL |
---|
| 1008 | ENDIF |
---|
| 1009 | |
---|
| 1010 | !3.5 Abort if failure and only 1 thread |
---|
| 1011 | |
---|
| 1012 | IF( nthreads == 1 .AND. istop /= nf90_noerr ) THEN |
---|
[3019] | 1013 | WRITE(numerr,*) '*** NEMO rebuild failed! ***' |
---|
[6415] | 1014 | STOP 9 |
---|
[2895] | 1015 | ENDIF |
---|
| 1016 | |
---|
| 1017 | END DO ! loop over files |
---|
| 1018 | !$OMP END PARALLEL DO |
---|
| 1019 | |
---|
| 1020 | !3.6 Abort if any of the OMP threads failed |
---|
| 1021 | IF( istop /= nf90_noerr ) THEN |
---|
[3019] | 1022 | WRITE(numerr,*) '*** NEMO rebuild failed! ***' |
---|
[6415] | 1023 | STOP 9 |
---|
[2895] | 1024 | ENDIF |
---|
| 1025 | |
---|
| 1026 | ENDIF ! ndims > 2 |
---|
| 1027 | |
---|
| 1028 | !--------------------------------------------------------------------------- |
---|
| 1029 | !4. Write data to output file |
---|
| 1030 | |
---|
| 1031 | IF (l_verbose) WRITE(numout,*) 'Writing variable '//TRIM(varname)//'...' |
---|
| 1032 | |
---|
| 1033 | !4.1 If the valid min and max attributes exist then update them in the file |
---|
| 1034 | |
---|
| 1035 | IF( l_valid ) THEN |
---|
| 1036 | CALL check_nf90( nf90_put_att( outid, jv, "valid_min", ValMin ) ) |
---|
| 1037 | CALL check_nf90( nf90_put_att( outid, jv, "valid_max", ValMax ) ) |
---|
| 1038 | ENDIF |
---|
| 1039 | |
---|
| 1040 | !4.2 Write the data to the output file depending on how many dimensions it has |
---|
| 1041 | |
---|
| 1042 | IF( ndims == 0 ) THEN |
---|
| 1043 | |
---|
| 1044 | SELECT CASE( xtype ) |
---|
| 1045 | CASE( NF90_BYTE ) |
---|
| 1046 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_0d_i1 ) ) |
---|
| 1047 | CASE( NF90_SHORT ) |
---|
| 1048 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_0d_i2 ) ) |
---|
| 1049 | CASE( NF90_INT ) |
---|
| 1050 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_0d_i4 ) ) |
---|
| 1051 | CASE( NF90_FLOAT ) |
---|
| 1052 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_0d_sp ) ) |
---|
| 1053 | CASE( NF90_DOUBLE ) |
---|
| 1054 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_0d_dp ) ) |
---|
| 1055 | END SELECT |
---|
| 1056 | |
---|
| 1057 | ELSEIF( ndims == 1 ) THEN |
---|
| 1058 | |
---|
| 1059 | SELECT CASE( xtype ) |
---|
| 1060 | CASE( NF90_BYTE ) |
---|
| 1061 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_1d_i1 ) ) |
---|
| 1062 | DEALLOCATE(globaldata_1d_i1) |
---|
| 1063 | CASE( NF90_SHORT ) |
---|
| 1064 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_1d_i2 ) ) |
---|
| 1065 | DEALLOCATE(globaldata_1d_i2) |
---|
| 1066 | CASE( NF90_INT ) |
---|
| 1067 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_1d_i4 ) ) |
---|
| 1068 | DEALLOCATE(globaldata_1d_i4) |
---|
| 1069 | CASE( NF90_FLOAT ) |
---|
| 1070 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_1d_sp ) ) |
---|
| 1071 | DEALLOCATE(globaldata_1d_sp) |
---|
| 1072 | CASE( NF90_DOUBLE ) |
---|
| 1073 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_1d_dp ) ) |
---|
| 1074 | DEALLOCATE(globaldata_1d_dp) |
---|
| 1075 | END SELECT |
---|
| 1076 | |
---|
| 1077 | ELSEIF( ndims == 2 ) THEN |
---|
| 1078 | |
---|
| 1079 | SELECT CASE( xtype ) |
---|
| 1080 | CASE( NF90_BYTE ) |
---|
| 1081 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_2d_i1 ) ) |
---|
| 1082 | DEALLOCATE(globaldata_2d_i1) |
---|
| 1083 | CASE( NF90_SHORT ) |
---|
| 1084 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_2d_i2 ) ) |
---|
| 1085 | DEALLOCATE(globaldata_2d_i2) |
---|
| 1086 | CASE( NF90_INT ) |
---|
| 1087 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_2d_i4 ) ) |
---|
| 1088 | DEALLOCATE(globaldata_2d_i4) |
---|
| 1089 | CASE( NF90_FLOAT ) |
---|
| 1090 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_2d_sp ) ) |
---|
| 1091 | DEALLOCATE(globaldata_2d_sp) |
---|
| 1092 | CASE( NF90_DOUBLE ) |
---|
| 1093 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_2d_dp ) ) |
---|
| 1094 | DEALLOCATE(globaldata_2d_dp) |
---|
| 1095 | CASE DEFAULT |
---|
| 1096 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 1097 | STOP 9 |
---|
[2895] | 1098 | END SELECT |
---|
| 1099 | |
---|
| 1100 | ELSEIF( ndims == 3 ) THEN |
---|
| 1101 | |
---|
| 1102 | SELECT CASE( xtype ) |
---|
| 1103 | CASE( NF90_BYTE ) |
---|
| 1104 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_3d_i1 ) ) |
---|
| 1105 | DEALLOCATE(globaldata_3d_i1) |
---|
| 1106 | CASE( NF90_SHORT ) |
---|
| 1107 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_3d_i2 ) ) |
---|
| 1108 | DEALLOCATE(globaldata_3d_i2) |
---|
| 1109 | CASE( NF90_INT ) |
---|
| 1110 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_3d_i4 ) ) |
---|
| 1111 | DEALLOCATE(globaldata_3d_i4) |
---|
| 1112 | CASE( NF90_FLOAT ) |
---|
| 1113 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_3d_sp ) ) |
---|
| 1114 | DEALLOCATE(globaldata_3d_sp) |
---|
| 1115 | CASE( NF90_DOUBLE ) |
---|
| 1116 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_3d_dp ) ) |
---|
| 1117 | DEALLOCATE(globaldata_3d_dp) |
---|
| 1118 | CASE DEFAULT |
---|
| 1119 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 1120 | STOP 9 |
---|
[2895] | 1121 | END SELECT |
---|
| 1122 | |
---|
| 1123 | ELSEIF( ndims == 4 ) THEN |
---|
[3025] | 1124 | |
---|
[2895] | 1125 | SELECT CASE( xtype ) |
---|
| 1126 | CASE( NF90_BYTE ) |
---|
[3025] | 1127 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_4d_i1, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 1128 | DEALLOCATE(globaldata_4d_i1) |
---|
| 1129 | CASE( NF90_SHORT ) |
---|
[3025] | 1130 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_4d_i2, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 1131 | DEALLOCATE(globaldata_4d_i2) |
---|
| 1132 | CASE( NF90_INT ) |
---|
[3025] | 1133 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_4d_i4, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 1134 | DEALLOCATE(globaldata_4d_i4) |
---|
| 1135 | CASE( NF90_FLOAT ) |
---|
[3025] | 1136 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_4d_sp, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 1137 | DEALLOCATE(globaldata_4d_sp) |
---|
| 1138 | CASE( NF90_DOUBLE ) |
---|
[3025] | 1139 | CALL check_nf90( nf90_put_var( outid, jv, globaldata_4d_dp, start=(/1,1,1,nt/) ) ) |
---|
[2895] | 1140 | DEALLOCATE(globaldata_4d_dp) |
---|
| 1141 | CASE DEFAULT |
---|
| 1142 | WRITE(numerr,*) 'Unknown nf90 type: ', xtype |
---|
[6415] | 1143 | STOP 9 |
---|
[2895] | 1144 | END SELECT |
---|
| 1145 | |
---|
| 1146 | ENDIF |
---|
[3025] | 1147 | |
---|
| 1148 | nt = nt + ntchunk |
---|
| 1149 | |
---|
| 1150 | END DO ! WHILE loop |
---|
[2895] | 1151 | |
---|
| 1152 | END DO ! loop over variables |
---|
| 1153 | |
---|
| 1154 | !--------------------------------------------------------------------------- |
---|
| 1155 | !5. Close files |
---|
| 1156 | |
---|
| 1157 | !5.1 Close input files |
---|
| 1158 | |
---|
| 1159 | IF (l_verbose) WRITE(numout,*) 'Closing input files...' |
---|
| 1160 | DO ifile = 1, ndomain |
---|
| 1161 | ncid = inncids(ifile) |
---|
| 1162 | CALL check_nf90( nf90_close( ncid ) ) |
---|
| 1163 | END DO |
---|
| 1164 | |
---|
| 1165 | !5.2 Close output file |
---|
| 1166 | |
---|
| 1167 | IF (l_verbose) WRITE(numout,*) 'Closing output file...' |
---|
| 1168 | CALL check_nf90( nf90_close( outid ) ) |
---|
| 1169 | |
---|
| 1170 | IF (l_verbose) WRITE(numout,*) 'NEMO rebuild completed successfully' |
---|
| 1171 | IF (l_verbose) WRITE(numout,*) |
---|
| 1172 | |
---|
| 1173 | CONTAINS |
---|
| 1174 | |
---|
| 1175 | |
---|
| 1176 | SUBROUTINE check_nf90(status, errorFlag) |
---|
| 1177 | !--------------------------------------------------------------------- |
---|
| 1178 | ! Checks return code from nf90 library calls and warns if needed |
---|
| 1179 | ! If errorFlag is present then it just increments this flag (OMP use) |
---|
| 1180 | ! |
---|
| 1181 | !--------------------------------------------------------------------- |
---|
| 1182 | INTEGER, INTENT(IN ) :: status |
---|
| 1183 | INTEGER, INTENT(INOUT), OPTIONAL :: errorFlag |
---|
| 1184 | !--------------------------------------------------------------------- |
---|
| 1185 | |
---|
| 1186 | IF( status /= nf90_noerr ) THEN |
---|
| 1187 | WRITE(numerr,*) 'ERROR! : '//TRIM(nf90_strerror(status)) |
---|
| 1188 | IF( PRESENT( errorFlag ) ) THEN |
---|
| 1189 | errorFlag = errorFlag + status |
---|
| 1190 | ELSE |
---|
| 1191 | WRITE(numerr,*) "*** NEMO rebuild failed ***" |
---|
| 1192 | WRITE(numerr,*) |
---|
[6415] | 1193 | STOP 9 |
---|
[2895] | 1194 | ENDIF |
---|
| 1195 | ENDIF |
---|
| 1196 | |
---|
| 1197 | END SUBROUTINE check_nf90 |
---|
| 1198 | |
---|
| 1199 | |
---|
| 1200 | END PROGRAM rebuild_nemo |
---|