mirror of
https://github.com/Unidata/netcdf-c.git
synced 2024-11-27 07:30:33 +08:00
88 lines
2.4 KiB
C
88 lines
2.4 KiB
C
#include <hdf5.h>
|
|
#include <hdf5_hl.h>
|
|
#include <nc_tests.h>
|
|
#include "err_macros.h"
|
|
|
|
#define FILE_NAME_A "tst_virtual_a.nc"
|
|
#define FILE_NAME_B "tst_virtual_b.nc"
|
|
#define VARIABLE_SIZE 5
|
|
#define VARIABLE_NAME "v"
|
|
|
|
static void
|
|
create_dataset_a() {
|
|
hsize_t dims[1] = {VARIABLE_SIZE};
|
|
float data[VARIABLE_SIZE];
|
|
for(size_t i = 0; i < VARIABLE_SIZE; ++i) {
|
|
data[i] = (float)i;
|
|
}
|
|
|
|
hid_t file = H5Fcreate(FILE_NAME_A, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
|
|
hid_t space = H5Screate_simple(1, dims, NULL);
|
|
hid_t dset = H5Dcreate2(file, VARIABLE_NAME, H5T_IEEE_F32LE, space, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT);
|
|
|
|
H5Dwrite(dset, H5T_IEEE_F32LE, H5S_ALL, H5S_ALL, H5P_DEFAULT, data);
|
|
H5DSset_scale(dset, VARIABLE_NAME);
|
|
|
|
H5Dclose(dset);
|
|
H5Sclose(space);
|
|
H5Fclose(file);
|
|
}
|
|
|
|
static void
|
|
create_dataset_b() {
|
|
hsize_t dims[1] = {VARIABLE_SIZE};
|
|
|
|
hid_t file = H5Fcreate(FILE_NAME_B, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
|
|
hid_t src_space = H5Screate_simple(1, dims, NULL);
|
|
hid_t space = H5Screate_simple(1, dims, NULL);
|
|
|
|
hid_t dcpl = H5Pcreate(H5P_DATASET_CREATE);
|
|
H5Pset_virtual(dcpl, space, FILE_NAME_A, VARIABLE_NAME, src_space);
|
|
hid_t dset = H5Dcreate2(file, VARIABLE_NAME, H5T_IEEE_F32LE, space, H5P_DEFAULT, dcpl, H5P_DEFAULT);
|
|
|
|
H5DSset_scale(dset, VARIABLE_NAME);
|
|
|
|
H5Dclose(dset);
|
|
H5Pclose(dcpl);
|
|
H5Sclose(space);
|
|
H5Sclose(src_space);
|
|
H5Fclose(file);
|
|
}
|
|
|
|
int read_back_contents(const char* filename) {
|
|
int ncid, varid, ndims;
|
|
float data[VARIABLE_SIZE];
|
|
|
|
char var_name[NC_MAX_NAME+1];
|
|
int dimids_var[1], var_type, natts;
|
|
|
|
printf("\treading back %s.\n", filename);
|
|
|
|
if (nc_open(filename, 0, &ncid)) ERR;
|
|
if (nc_inq_varid(ncid, VARIABLE_NAME, &varid)) ERR;
|
|
if (nc_inq_var(ncid, varid, var_name, &var_type, &ndims, dimids_var, &natts)) ERR;
|
|
if (nc_get_var_float(ncid, varid, data)) ERR;
|
|
printf("\t %s: ", var_name);
|
|
for (size_t i = 0; i < VARIABLE_SIZE; ++i) {
|
|
printf("%f, ", data[i]);
|
|
}
|
|
printf("\n");
|
|
if (nc_close(ncid)) ERR;
|
|
SUMMARIZE_ERR;
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
|
|
printf("\n*** Testing reading of virtual datasets.\n");
|
|
|
|
create_dataset_a();
|
|
create_dataset_b();
|
|
|
|
int status;
|
|
if((status = read_back_contents(FILE_NAME_A))) return status;
|
|
if((status = read_back_contents(FILE_NAME_B))) return status;
|
|
}
|