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
|
/************************************************************
This example shows how to read and write array datatypes
to a dataset. The program first writes integers arrays of
dimension ADIM0xADIM1 to a dataset with a dataspace of
DIM0, then closes the file. Next, it reopens the file,
reads back the data, and outputs it to the screen.
************************************************************/
#include "hdf5.h"
#include <stdio.h>
#include <stdlib.h>
#define FILE "h5ex_t_array.h5"
#define DATASET "DS1"
#define DIM0 4
#define ADIM0 3
#define ADIM1 5
int
main(void)
{
hid_t file = H5I_INVALID_HID; /* File Handle */
hid_t space = H5I_INVALID_HID; /* Dataspace Handle */
hid_t dset = H5I_INVALID_HID; /* Dataset Handle */
hid_t filetype = H5I_INVALID_HID;
hid_t memtype = H5I_INVALID_HID;
/* Handles */
herr_t status;
hsize_t dims[1] = {DIM0};
hsize_t adims[2] = {ADIM0, ADIM1};
int wdata[DIM0][ADIM0][ADIM1]; /* Write buffer */
int ***rdata = NULL; /* Read buffer */
int ndims;
hsize_t i, j, k;
/*
* Initialize data. i is the element in the dataspace, j and k the
* elements within the array datatype.
*/
for (i = 0; i < DIM0; i++)
for (j = 0; j < ADIM0; j++)
for (k = 0; k < ADIM1; k++)
wdata[i][j][k] = i * j - j * k + i * k;
/*
* Create a new file using the default properties.
*/
file = H5Fcreate(FILE, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
/*
* Create array datatypes for file and memory.
*/
filetype = H5Tarray_create(H5T_STD_I64LE, 2, adims);
memtype = H5Tarray_create(H5T_NATIVE_INT, 2, adims);
/*
* Create dataspace. Setting maximum size to NULL sets the maximum
* size to be the current size.
*/
space = H5Screate_simple(1, dims, NULL);
/*
* Create the dataset and write the array data to it.
*/
dset = H5Dcreate(file, DATASET, filetype, space, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT);
status = H5Dwrite(dset, memtype, H5S_ALL, H5S_ALL, H5P_DEFAULT, wdata[0][0]);
/*
* Close and release resources.
*/
status = H5Dclose(dset);
status = H5Sclose(space);
status = H5Tclose(filetype);
status = H5Tclose(memtype);
status = H5Fclose(file);
/*
* Now we begin the read section of this example. Here we assume
* the dataset and array have the same name and rank, but can have
* any size. Therefore we must allocate a new array to read in
* data using malloc().
*/
/*
* Open file and dataset.
*/
file = H5Fopen(FILE, H5F_ACC_RDONLY, H5P_DEFAULT);
dset = H5Dopen(file, DATASET, H5P_DEFAULT);
/*
* Get the datatype and its dimensions.
*/
filetype = H5Dget_type(dset);
ndims = H5Tget_array_dims(filetype, adims);
/*
* Get dataspace and allocate memory for read buffer. This is a
* three dimensional dataset when the array datatype is included so
* the dynamic allocation must be done in steps.
*/
space = H5Dget_space(dset);
ndims = H5Sget_simple_extent_dims(space, dims, NULL);
/*
* Allocate array of pointers to two-dimensional arrays (the
* elements of the dataset.
*/
rdata = (int ***)malloc(dims[0] * sizeof(int **));
/*
* Allocate two dimensional array of pointers to rows in the data
* elements.
*/
rdata[0] = (int **)malloc(dims[0] * adims[0] * sizeof(int *));
/*
* Allocate space for integer data.
*/
rdata[0][0] = (int *)malloc(dims[0] * adims[0] * adims[1] * sizeof(int));
/*
* Set the members of the pointer arrays allocated above to point
* to the correct locations in their respective arrays.
*/
for (i = 0; i < dims[0]; i++) {
rdata[i] = rdata[0] + i * adims[0];
for (j = 0; j < adims[0]; j++)
rdata[i][j] = rdata[0][0] + (adims[0] * adims[1] * i) + (adims[1] * j);
}
/*
* Create the memory datatype.
*/
memtype = H5Tarray_create(H5T_NATIVE_INT, 2, adims);
/*
* Read the data.
*/
status = H5Dread(dset, memtype, H5S_ALL, H5S_ALL, H5P_DEFAULT, rdata[0][0]);
/*
* Output the data to the screen.
*/
for (i = 0; i < dims[0]; i++) {
printf("%s[%" PRIuHSIZE "]:\n", DATASET, i);
for (j = 0; j < adims[0]; j++) {
printf(" [");
for (k = 0; k < adims[1]; k++)
printf(" %3d", rdata[i][j][k]);
printf("]\n");
}
printf("\n");
}
/*
* Close and release resources.
*/
free(rdata[0][0]);
free(rdata[0]);
free(rdata);
status = H5Dclose(dset);
status = H5Sclose(space);
status = H5Tclose(filetype);
status = H5Tclose(memtype);
status = H5Fclose(file);
return 0;
}
|