mirror/netcdf-c
2
0
Fork 0
mirror of https://github.com/Unidata/netcdf-c.git synced 2024-11-21 03:13:42 +08:00
Code Issues Packages Projects Releases Wiki Activity
v4.9.2-release.wif
netcdf-c/h5_test/tst_h_vars.c
Dennis Heimbigner 3ffe7be446 Enhance/Fix filter support 
re: Discussion https://github.com/Unidata/netcdf-c/discussions/2214

The primary change is to support so-called "standard filters".
A standard filter is one that is defined by the following
netcdf-c API:
````
int nc_def_var_XXX(int ncid, int varid, size_t nparams, unsigned* params);
int nc_inq_var_XXXX(int ncid, int varid, int* usefilterp, unsigned* params);
````
So for example, zstandard would be a standard filter by defining
the functions *nc_def_var_zstandard* and *nc_inq_var_zstandard*.

In order to define these functions, we need a new dispatch function:
````
int nc_inq_filter_avail(int ncid, unsigned filterid);
````
This function, combined with the existing filter API can be used
to implement arbitrary standard filters using a simple code pattern.
Note that I would have preferred that this function return a list
of all available filters, but HDF5 does not support that functionality.

So this PR implements the dispatch function and implements
the following standard functions:
    + bzip2
    + zstandard
    + blosc
Specific test cases are also provided for HDF5 and NCZarr.
Over time, other specific standard filters will be defined.

## Primary Changes
* Add nc_inq_filter_avail() to netcdf-c API.
* Add standard filter implementations to test use of *nc_inq_filter_avail*.
* Bump the dispatch table version number and add to all the relevant
   dispatch tables (libsrc, libsrcp, etc).
* Create a program to invoke nc_inq_filter_avail so that it is accessible
  to shell scripts.
* Cleanup szip support to properly support szip
  when HDF5 is disabled. This involves detecting
  libsz separately from testing if HDF5 supports szip.
* Integrate shuffle and fletcher32 into the existing
  filter API. This means that, for example, nc_def_var_fletcher32
  is now a wrapper around nc_def_var_filter.
* Extend the Codec defaulting to allow multiple default shared libraries.

## Misc. Changes
* Modify configure.ac/CMakeLists.txt to look for the relevant
  libraries implementing standard filters.
* Modify libnetcdf.settings to list available standard filters
  (including deflate and szip).
* Add CMake test modules to locate libbz2 and libzstd.
* Cleanup the HDF5 memory manager function use in the plugins.
* remove unused file include//ncfilter.h
* remove tests for the HDF5 memory operations e.g. H5allocate_memory.
* Add flag to ncdump to force use of _Filter instead of _Deflate
  or _Shuffle or _Fletcher32. Used for testing.
2022-03-14 12:39:37 -06:00

602 lines
23 KiB
C
Raw Permalink Blame History

/* This is part of the netCDF package. Copyright 2020 University
Corporation for Atmospheric Research/Unidata See COPYRIGHT file for
conditions of use.
Test HDF5 file code. These are not intended to be exhaustive tests,
but they use HDF5 the same way that netCDF-4 does, so if these
tests don't work, than netCDF-4 won't work either.
This files tests dataset creation and writing.
Ed Hartnett
*/
#include "h5_err_macros.h"
#include <hdf5.h>
#define FILE_NAME "tst_h_vars.h5"
#define TEST_NAME "tst_h_vars"
#define GRP_NAME "Henry_V"
#define VAR_BOOL_NAME "Southhamptons_Battle_Record"
#define GRP2_NAME "Some_3D_Met_Data"
#define DIM1_LEN 3
#define MAX_DIMS 255
#define NDIM1 1
int
main()
{
hid_t fileid, grpid, spaceid, datasetid;
int bool_out[DIM1_LEN] = {0, 1, 0};
hsize_t dims[1];
printf("\n*** Checking HDF5 variable functions.\n");
printf("*** Checking HDF5 boolean variables...");
/* Open file and create group. */
if ((fileid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP_NAME, 0)) < 0) ERR;
/* Write an array of bools. */
dims[0] = DIM1_LEN;
if ((spaceid = H5Screate_simple(1, dims, dims)) < 0) ERR;
if ((datasetid = H5Dcreate1(grpid, VAR_BOOL_NAME, H5T_NATIVE_HBOOL,
spaceid, H5P_DEFAULT)) < 0) ERR;
if (H5Dwrite(datasetid, H5T_NATIVE_HBOOL, H5S_ALL, H5S_ALL, H5P_DEFAULT,
bool_out) < 0) ERR;
if (H5Dclose(datasetid) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
SUMMARIZE_ERR;
printf("*** Checking HDF5 variable with unlimited dimension...");
{
#define LAT_LEN 2
#define LON_LEN 3
#define NDIMS 3
#define LAT_NAME "Lat"
#define LON_NAME "Lon"
#define TIME_NAME "Time"
#define PRES_NAME "Pressure"
#define TEMP_NAME "Temperature"
hid_t fileid, grpid, spaceid, write_spaceid, spaceid_in, mem_spaceid;
hid_t pres_dsid, temp_dsid, cparmsid;
float float_data_out[LAT_LEN][LON_LEN];
hsize_t dims[NDIMS], max_dims[NDIMS];
hsize_t dims_in[NDIMS], max_dims_in[NDIMS];
hsize_t start[MAX_DIMS], count[MAX_DIMS];
int lat, lon;
/* Set up some phoney data, 1 record's worth. In C, first
* dimension varies most slowly. */
for (lat = 0; lat < LAT_LEN; lat++)
for (lon = 0; lon < LON_LEN; lon++)
float_data_out[lat][lon] = -666.666;
/* Create file and group. */
if ((fileid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP2_NAME, 0)) < 0) ERR;
/* Create a space corresponding to these three dimensions. */
dims[0] = 0;
dims[1] = LAT_LEN;
dims[2] = LON_LEN;
max_dims[0] = H5S_UNLIMITED;
max_dims[1] = LAT_LEN;
max_dims[2] = LON_LEN;
if ((spaceid = H5Screate_simple(NDIMS, dims, max_dims)) < 0) ERR;
/* Enable chunking for unlimited datasets. */
if ((cparmsid = H5Pcreate(H5P_DATASET_CREATE)) < 0) ERR;
dims[0] = 1;
dims[1] = 1;
dims[2] = 1;
if (H5Pset_chunk(cparmsid, NDIMS, dims) < 0) ERR;
/* Create two variables which use this space. */
if ((pres_dsid = H5Dcreate1(grpid, PRES_NAME, H5T_NATIVE_FLOAT,
spaceid, cparmsid)) < 0) ERR;
if ((temp_dsid = H5Dcreate1(grpid, TEMP_NAME, H5T_NATIVE_FLOAT,
spaceid, cparmsid)) < 0) ERR;
/* Get the spaceid and check various things. */
if ((spaceid_in = H5Dget_space(pres_dsid)) < 0) ERR;
if (H5Sget_simple_extent_dims(spaceid_in, dims_in,
max_dims_in) < 0) ERR;
if (dims_in[0] != 0 || dims_in[1] != LAT_LEN || dims_in[2] != LON_LEN) ERR;
if (max_dims_in[0] != H5S_UNLIMITED || max_dims_in[1] != LAT_LEN || max_dims_in[2] != LON_LEN) ERR;
/* Extend each of them to hold one record. */
dims[0] = 1;
dims[1] = LAT_LEN;
dims[2] = LON_LEN;
if (H5Dextend(pres_dsid, dims) < 0) ERR;
if (H5Dextend(temp_dsid, dims) < 0) ERR;
/* Create a space to deal with one record at a time in memory. */
if ((mem_spaceid = H5Screate_simple(NDIMS, dims, NULL)) < 0) ERR;
/* Create a space to write one record. */
if ((write_spaceid = H5Screate_simple(NDIMS, dims, NULL)) < 0) ERR;
/* Write one record of data to each dataset. */
if (H5Dwrite(pres_dsid, H5T_IEEE_F32BE, mem_spaceid, write_spaceid,
H5P_DEFAULT, float_data_out) < 0) ERR;
if (H5Dwrite(temp_dsid, H5T_IEEE_F32LE, mem_spaceid, write_spaceid,
H5P_DEFAULT, float_data_out) < 0) ERR;
/* Get the spaceid and check various things. */
if ((spaceid_in = H5Dget_space(temp_dsid)) < 0) ERR;
if (H5Sget_simple_extent_dims(spaceid_in, dims_in,
max_dims_in) < 0) ERR;
if (dims_in[0] != 1 || dims_in[1] != LAT_LEN || dims_in[2] != LON_LEN) ERR;
if (max_dims_in[0] != H5S_UNLIMITED || max_dims_in[1] != LAT_LEN || max_dims_in[2] != LON_LEN) ERR;
/* Extend each of them to hold a second record. */
dims[0] = 2;
dims[1] = LAT_LEN;
dims[2] = LON_LEN;
if (H5Dextend(pres_dsid, dims) < 0) ERR;
if (H5Dextend(temp_dsid, dims) < 0) ERR;
/* Create a space to write second record. */
if ((write_spaceid = H5Screate_simple(NDIMS, dims, NULL)) < 0) ERR;
count[0] = 1;
count[1] = LAT_LEN;
count[2] = LON_LEN;
start[0] = 1;
start[1] = 0;
start[2] = 0;
if (H5Sselect_hyperslab(write_spaceid, H5S_SELECT_SET,
start, NULL, count, NULL) < 0) ERR;
/* Write second record of data to each dataset. */
if (H5Dwrite(pres_dsid, H5T_IEEE_F32LE, mem_spaceid, write_spaceid,
H5P_DEFAULT, float_data_out) < 0) ERR;
if (H5Dwrite(temp_dsid, H5T_IEEE_F32LE, mem_spaceid, write_spaceid,
H5P_DEFAULT, float_data_out) < 0) ERR;
/* Get the spaceid and check various things. */
if ((spaceid_in = H5Dget_space(pres_dsid)) < 0) ERR;
if (H5Sget_simple_extent_dims(spaceid_in, dims_in,
max_dims_in) < 0) ERR;
if (dims_in[0] != 2 || dims_in[1] != LAT_LEN || dims_in[2] != LON_LEN) ERR;
if (max_dims_in[0] != H5S_UNLIMITED || max_dims_in[1] != LAT_LEN || max_dims_in[2] != LON_LEN) ERR;
/* Close up the shop. */
if (H5Dclose(pres_dsid) < 0 ||
H5Dclose(temp_dsid) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0) ERR;
}
SUMMARIZE_ERR;
printf("*** Checking HDF5 deflate filter setting and getting...");
#define DEFLATE_LEVEL 9
#define MAX_NAME 100
#define NUM_CD_ELEM 10
/* HDF5 defines this... */
#define DEFLATE_NAME "deflate"
{
H5Z_filter_t filter;
int num_filters;
hid_t propid;
unsigned int flags, cd_values[NUM_CD_ELEM], filter_config;
size_t cd_nelems = NUM_CD_ELEM;
size_t namelen = MAX_NAME;
char name[MAX_NAME + 1];
/* Open file and create group. */
if ((fileid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP_NAME, 0)) < 0) ERR;
/* Write an array of bools, with compression. */
dims[0] = DIM1_LEN;
if ((propid = H5Pcreate(H5P_DATASET_CREATE)) < 0) ERR;
if (H5Pset_layout(propid, H5D_CHUNKED)) ERR;
if (H5Pset_chunk(propid, 1, dims)) ERR;
if (H5Pset_deflate(propid, DEFLATE_LEVEL)) ERR;
if ((spaceid = H5Screate_simple(1, dims, dims)) < 0) ERR;
if ((datasetid = H5Dcreate1(grpid, VAR_BOOL_NAME, H5T_NATIVE_HBOOL,
spaceid, propid)) < 0) ERR;
if (H5Dwrite(datasetid, H5T_NATIVE_HBOOL, H5S_ALL, H5S_ALL, H5P_DEFAULT,
bool_out) < 0) ERR;
if (H5Dclose(datasetid) < 0 ||
H5Pclose(propid) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
/* Now reopen the file and check. */
if ((fileid = H5Fopen(FILE_NAME, H5F_ACC_RDONLY, H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gopen1(fileid, GRP_NAME)) < 0) ERR;
if ((datasetid = H5Dopen1(grpid, VAR_BOOL_NAME)) < 0) ERR;
if ((propid = H5Dget_create_plist(datasetid)) < 0) ERR;
/* The possible values of filter (which is just an int) can be
* found in H5Zpublic.h. */
if ((num_filters = H5Pget_nfilters(propid)) < 0) ERR;
if (num_filters != 1) ERR;
if ((filter = H5Pget_filter2(propid, 0, &flags, &cd_nelems, cd_values,
namelen, name, &filter_config)) < 0) ERR;
if (filter != H5Z_FILTER_DEFLATE || cd_nelems != 1 ||
cd_values[0] != DEFLATE_LEVEL || strcmp(name, DEFLATE_NAME)) ERR;
if (H5Dclose(datasetid) < 0 ||
H5Pclose(propid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
}
SUMMARIZE_ERR;
printf("*** Checking HDF5 deflate, fletcher32, shuffle filter setting and getting...");
{
H5Z_filter_t filter;
int num_filters;
hid_t propid;
unsigned int flags, cd_values[NUM_CD_ELEM], filter_config;
size_t cd_nelems = NUM_CD_ELEM;
size_t namelen = MAX_NAME;
char name[MAX_NAME + 1];
int found_shuffle = 0, found_fletcher32 = 0, found_deflate = 0;
int f;
/* Open file and create group. */
if ((fileid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP_NAME, 0)) < 0) ERR;
/* Write an array of bools, with compression, fletcher32
* checksum, shuffle filters. Like a hoogie with "the works." */
dims[0] = DIM1_LEN;
if ((propid = H5Pcreate(H5P_DATASET_CREATE)) < 0) ERR;
if (H5Pset_layout(propid, H5D_CHUNKED)) ERR;
if (H5Pset_chunk(propid, 1, dims)) ERR;
if (H5Pset_shuffle(propid)) ERR;
if (H5Pset_deflate(propid, DEFLATE_LEVEL)) ERR;
if (H5Pset_fletcher32(propid)) ERR;
if ((spaceid = H5Screate_simple(1, dims, dims)) < 0) ERR;
if ((datasetid = H5Dcreate1(grpid, VAR_BOOL_NAME, H5T_NATIVE_HBOOL,
spaceid, propid)) < 0) ERR;
if (H5Dwrite(datasetid, H5T_NATIVE_HBOOL, H5S_ALL, H5S_ALL, H5P_DEFAULT,
bool_out) < 0) ERR;
if (H5Dclose(datasetid) < 0 ||
H5Pclose(propid) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
/* Now reopen the file and check. */
if ((fileid = H5Fopen(FILE_NAME, H5F_ACC_RDONLY, H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gopen1(fileid, GRP_NAME)) < 0) ERR;
if ((datasetid = H5Dopen1(grpid, VAR_BOOL_NAME)) < 0) ERR;
if ((propid = H5Dget_create_plist(datasetid)) < 0) ERR;
/* The possible values of filter (which is just an int) can be
* found in H5Zpublic.h. */
if ((num_filters = H5Pget_nfilters(propid)) < 0) ERR;
if (num_filters != 3) ERR;
for (f = 0; f < num_filters; f++)
{
if ((filter = H5Pget_filter2(propid, f, &flags, &cd_nelems, cd_values,
namelen, name, &filter_config)) < 0) ERR;
switch (filter)
{
case H5Z_FILTER_SHUFFLE:
found_shuffle++;
break;
case H5Z_FILTER_FLETCHER32:
found_fletcher32++;
break;
case H5Z_FILTER_DEFLATE:
found_deflate++;
if (cd_nelems != 1 || cd_values[0] != DEFLATE_LEVEL ||
strcmp(name, DEFLATE_NAME)) ERR;
break;
default:
break;
}
}
if (!found_fletcher32 || !found_deflate || !found_shuffle) ERR;
if (H5Dclose(datasetid) < 0 ||
H5Pclose(propid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
}
SUMMARIZE_ERR;
printf("*** Checking HDF5 endianness control...");
#define NATIVE_VAR_NAME "native_var"
#define LE_VAR_NAME "le_var"
#define BE_VAR_NAME "be_var"
{
int data[DIM1_LEN], data_in[DIM1_LEN];
hid_t typeid, native_typeid;
hid_t native_did, le_did, be_did;
H5T_order_t order;
htri_t equal;
int i;
for (i = 0; i < DIM1_LEN; i++)
data[i] = i;
/* Open file and create group. */
if ((fileid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP_NAME, 0)) < 0) ERR;
/* Create a dataset of native endian. */
dims[0] = DIM1_LEN;
if ((spaceid = H5Screate_simple(1, dims, dims)) < 0) ERR;
if ((native_did = H5Dcreate1(grpid, NATIVE_VAR_NAME, H5T_NATIVE_INT,
spaceid, H5P_DEFAULT)) < 0) ERR;
if ((le_did = H5Dcreate1(grpid, LE_VAR_NAME, H5T_STD_I32LE,
spaceid, H5P_DEFAULT)) < 0) ERR;
if ((be_did = H5Dcreate1(grpid, BE_VAR_NAME, H5T_STD_I32BE,
spaceid, H5P_DEFAULT)) < 0) ERR;
if (H5Dwrite(native_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data) < 0) ERR;
if (H5Dwrite(le_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data) < 0) ERR;
if (H5Dwrite(be_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data) < 0) ERR;
if (H5Dclose(native_did) < 0 ||
H5Dclose(le_did) < 0 ||
H5Dclose(be_did) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
/* Now reopen the file and check. */
if ((fileid = H5Fopen(FILE_NAME, H5F_ACC_RDONLY, H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gopen1(fileid, GRP_NAME)) < 0) ERR;
if ((native_did = H5Dopen1(grpid, NATIVE_VAR_NAME)) < 0) ERR;
if ((typeid = H5Dget_type(native_did)) < 0) ERR;
if ((native_typeid = H5Tget_native_type(typeid, H5T_DIR_DESCEND)) < 0) ERR;
if ((order = H5Tget_order(typeid)) < 0) ERR;
if ((equal = H5Tequal(typeid, native_typeid)) < 0) ERR;
if (!equal) ERR;
if (H5Dread(native_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data_in) < 0) ERR;
for (i = 0; i < DIM1_LEN; i++)
if (data[i] != data_in[i]) ERR;
if ((le_did = H5Dopen1(grpid, LE_VAR_NAME)) < 0) ERR;
if ((typeid = H5Dget_type(le_did)) < 0) ERR;
if ((order = H5Tget_order(typeid)) < 0) ERR;
if (order != H5T_ORDER_LE) ERR;
if (H5Dread(le_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data_in) < 0) ERR;
for (i = 0; i < DIM1_LEN; i++)
if (data[i] != data_in[i]) ERR;
if ((be_did = H5Dopen1(grpid, BE_VAR_NAME)) < 0) ERR;
if ((typeid = H5Dget_type(be_did)) < 0) ERR;
if ((order = H5Tget_order(typeid)) < 0) ERR;
if (order != H5T_ORDER_BE) ERR;
if (H5Dread(be_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data_in) < 0) ERR;
for (i = 0; i < DIM1_LEN; i++)
if (data[i] != data_in[i]) ERR;
if (H5Dclose(native_did) < 0 ||
H5Dclose(le_did) < 0 ||
H5Dclose(be_did) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
}
SUMMARIZE_ERR;
#ifdef HAVE_H5Z_SZIP
printf("*** Checking szip functionality...");
#define SZIP_VAR_NAME "szip_var"
#define SZIP_DIM1_LEN 32
{
int data[SZIP_DIM1_LEN];
hid_t plistid;
hsize_t chunksize[NDIM1] = {SZIP_DIM1_LEN};
int options_mask = 32, pixels_per_block = 4;
hsize_t my_dims[NDIM1];
int i;
/* For info about HDF5 and szip, see
* https://support.hdfgroup.org/HDF5/doc/RM/RM_H5P.html#Property-SetSzip
* and
* https://support.hdfgroup.org/doc_resource/SZIP/index.html. */
for (i = 0; i < SZIP_DIM1_LEN; i++)
data[i] = i;
/* Open file and create group. */
if ((fileid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP_NAME, 0)) < 0) ERR;
/* Create dataset creation property list. */
if ((plistid = H5Pcreate(H5P_DATASET_CREATE)) < 0) ERR;
/* Turn on chunking. */
if (H5Pset_chunk(plistid, NDIM1, chunksize) < 0) ERR;
/* Turn off object tracking times in HDF5 (as netcdf-4 does). */
if (H5Pset_obj_track_times(plistid, 0) < 0) ERR;
/* Turn on szip compression. */
if (H5Pset_szip(plistid, options_mask, pixels_per_block) < 0) ERR;
/* Create a space. */
my_dims[0] = SZIP_DIM1_LEN;
if ((spaceid = H5Screate_simple(1, my_dims, my_dims)) < 0) ERR;
/* Create a dataset. */
if ((datasetid = H5Dcreate2(grpid, SZIP_VAR_NAME, H5T_NATIVE_INT,
spaceid, H5P_DEFAULT, plistid,
H5P_DEFAULT)) < 0) ERR;
/* Write data. */
if (H5Dwrite(datasetid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data) < 0) ERR;
/* Release resources. */
if (H5Dclose(datasetid) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Pclose(plistid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
{
/* Now reopen the file and check. */
int data_in[SZIP_DIM1_LEN];
hid_t native_did;
int i;
if ((fileid = H5Fopen(FILE_NAME, H5F_ACC_RDONLY, H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gopen1(fileid, GRP_NAME)) < 0) ERR;
if ((native_did = H5Dopen1(grpid, SZIP_VAR_NAME)) < 0) ERR;
if (H5Dread(native_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data_in) < 0) ERR;
for (i = 0; i < SZIP_DIM1_LEN; i++)
if (data[i] != data_in[i]) ERR;
if (H5Dclose(native_did) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
}
}
SUMMARIZE_ERR;
printf("*** Checking using szip and zlib on same var...");
#define BOTH_VAR_NAME "szip_var"
#define BOTH_DIM1_LEN 50
#define NUM_FILE 5
#define MAX_STR 80
{
int data[BOTH_DIM1_LEN];
hid_t plistid;
hsize_t chunksize[NDIM1] = {BOTH_DIM1_LEN};
int options_mask = 32, pixels_per_block = 4;
int deflate_level = 3;
hsize_t my_dims[NDIM1];
int i, f;
/* Create data. */
for (i = 0; i < BOTH_DIM1_LEN; i++)
data[i] = i;
/* Run test 4 times. */
for (f = 0; f < NUM_FILE; f++)
{
char file_name[MAX_STR * 2 + 1];
char desc[NUM_FILE][MAX_STR + 1] = {"uncompressed", "zlib",
"szip", "zlib_and_szip",
"szip_and_zlib"};
/* Open file and create group. */
sprintf(file_name, "%s_%s.h5", TEST_NAME, desc[f]);
if ((fileid = H5Fcreate(file_name, H5F_ACC_TRUNC, H5P_DEFAULT,
H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gcreate1(fileid, GRP_NAME, 0)) < 0) ERR;
/* Create dataset creation property list. */
if ((plistid = H5Pcreate(H5P_DATASET_CREATE)) < 0) ERR;
/* Turn on chunking. */
if (H5Pset_chunk(plistid, NDIM1, chunksize) < 0) ERR;
/* Turn off object tracking times in HDF5 (as netcdf-4 does). */
if (H5Pset_obj_track_times(plistid, 0) < 0) ERR;
/* Turn on compression for some files. */
switch (f)
{
case 1:
if (H5Pset_deflate(plistid, deflate_level) < 0)
break;
case 2:
if (H5Pset_szip(plistid, options_mask, pixels_per_block) < 0) ERR;
break;
case 3:
if (H5Pset_deflate(plistid, deflate_level) < 0)
if (H5Pset_szip(plistid, options_mask, pixels_per_block) < 0) ERR;
break;
case 4:
if (H5Pset_szip(plistid, options_mask, pixels_per_block) < 0) ERR;
if (H5Pset_deflate(plistid, deflate_level) < 0)
break;
}
/* Create a space. */
my_dims[0] = BOTH_DIM1_LEN;
if ((spaceid = H5Screate_simple(1, my_dims, my_dims)) < 0) ERR;
/* Create a dataset. */
if ((datasetid = H5Dcreate2(grpid, BOTH_VAR_NAME, H5T_NATIVE_INT,
spaceid, H5P_DEFAULT, plistid,
H5P_DEFAULT)) < 0) ERR;
/* Write data. */
if (H5Dwrite(datasetid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data) < 0) ERR;
/* Release resources. */
if (H5Dclose(datasetid) < 0 ||
H5Sclose(spaceid) < 0 ||
H5Pclose(plistid) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
{
/* Now reopen the file and check. */
int data_in[BOTH_DIM1_LEN];
hid_t native_did;
int i;
if ((fileid = H5Fopen(file_name, H5F_ACC_RDONLY, H5P_DEFAULT)) < 0) ERR;
if ((grpid = H5Gopen1(fileid, GRP_NAME)) < 0) ERR;
if ((native_did = H5Dopen1(grpid, BOTH_VAR_NAME)) < 0) ERR;
if (H5Dread(native_did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT,
data_in) < 0) ERR;
for (i = 0; i < BOTH_DIM1_LEN; i++)
if (data[i] != data_in[i]) ERR;
if (H5Dclose(native_did) < 0 ||
H5Gclose(grpid) < 0 ||
H5Fclose(fileid) < 0)
ERR;
}
} /* next file */
}
SUMMARIZE_ERR;
#endif /* HAVE_H5Z_SZIP */
FINAL_RESULTS;
}
Reference in New Issue View Git Blame Copy Permalink