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
|
! This is part of the netCDF package.
! Copyright 2020 University Corporation for Atmospheric Research/Unidata.
! See COPYRIGHT file for conditions of use.
! This program tests netCDF-4 variable functions from fortran.
! Ed Hartnett, 2/1/2020
program f90tst_var_szip
use typeSizes
use netcdf
implicit none
! This is the name of the data file we will create.
character (len = *), parameter :: FILE_NAME = "f90tst_var_szip.nc"
integer, parameter :: MAX_DIMS = 2
integer, parameter :: NX = 40, NY = 4096
integer :: data_out(NY, NX), data_in(NY, NX)
! We need these ids and other gunk for netcdf.
integer :: ncid, varid, dimids(MAX_DIMS), chunksizes(MAX_DIMS)
integer :: x_dimid, y_dimid
integer :: mode_flag
integer :: nvars, ngatts, ndims, unlimdimid, file_format
integer :: x, y
integer, parameter :: CACHE_SIZE = 1000000
integer :: xtype_in, natts_in, dimids_in(MAX_DIMS), chunksizes_in(MAX_DIMS)
logical :: contiguous_in, shuffle_in, fletcher32_in
integer :: deflate_level_in, endianness_in
character (len = NF90_MAX_NAME) :: name_in
print *, ''
print *,'*** Testing szip writes of netCDF-4 var from Fortran 90.'
! Create some pretend data.
do x = 1, NX
do y = 1, NY
data_out(y, x) = (x - 1) * NY + (y - 1)
end do
end do
! Create the netCDF file.
mode_flag = IOR(nf90_netcdf4, nf90_classic_model)
call handle_err(nf90_create(FILE_NAME, mode_flag, ncid, cache_size = CACHE_SIZE))
! Define the dimensions.
call handle_err(nf90_def_dim(ncid, "x", NX, x_dimid))
call handle_err(nf90_def_dim(ncid, "y", NY, y_dimid))
dimids = (/ y_dimid, x_dimid /)
! Define the variable.
chunksizes = (/ 256, 10 /)
call handle_err(nf90_def_var(ncid, 'data', NF90_INT, dimids, &
varid, chunksizes = chunksizes))
! Turn on szip compression.
call handle_err(nf90_def_var_szip(ncid, 1, 32, 4))
! With classic model netCDF-4 file, enddef must be called.
call handle_err(nf90_enddef(ncid))
! Write the pretend data to the file.
call handle_err(nf90_put_var(ncid, varid, data_out))
! Close the file.
call handle_err(nf90_close(ncid))
! Reopen the file.
call handle_err(nf90_open(FILE_NAME, nf90_nowrite, ncid))
! Check some stuff out.
call handle_err(nf90_inquire(ncid, ndims, nvars, ngatts, unlimdimid, file_format))
if (ndims /= 2 .or. nvars /= 1 .or. ngatts /= 0 .or. unlimdimid /= -1 .or. &
file_format /= nf90_format_netcdf4_classic) stop 3
call handle_err(nf90_inquire_variable(ncid, varid, name_in, xtype_in, ndims, dimids_in, &
natts_in, contiguous_in, chunksizes_in, deflate_level_in, shuffle_in, fletcher32_in, &
endianness_in))
if (name_in .ne. 'data' .or. xtype_in .ne. NF90_INT .or. ndims .ne. MAX_DIMS .or. &
dimids_in(1) /= y_dimid .or. dimids_in(2) /= x_dimid .or. &
natts_in .ne. 0 .or. contiguous_in .neqv. .false. .or. &
chunksizes_in(1) /= chunksizes(1) .or. chunksizes_in(2) /= chunksizes(2) .or. &
shuffle_in .neqv. .false. .or. fletcher32_in .neqv. .false.) &
stop 4
! Check the data.
call handle_err(nf90_get_var(ncid, varid, data_in))
do x = 1, NX
do y = 1, NY
if (data_out(y, x) .ne. data_in(y, x)) stop 3
end do
end do
! Close the file.
call handle_err(nf90_close(ncid))
print *,'*** SUCCESS!'
contains
! This subroutine handles errors by printing an error message and
! exiting with a non-zero status.
subroutine handle_err(errcode)
use netcdf
implicit none
integer, intent(in) :: errcode
if(errcode /= nf90_noerr) then
print *, 'Error: ', trim(nf90_strerror(errcode))
stop 2
endif
end subroutine handle_err
end program f90tst_var_szip
|
