diff options
-rw-r--r-- | test/set_extend.c | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/test/set_extend.c b/test/set_extend.c new file mode 100644 index 0000000..9cb2cb6 --- /dev/null +++ b/test/set_extend.c @@ -0,0 +1,113 @@ + + +#include "hdf5.h" + +/*------------------------------------------------------------------------- + * + * Tests the function H5Dset_extend. In the current version of the library + * the dataset MUST be chunked. + * + *------------------------------------------------------------------------- + */ + +#define DATASETNAME "ShrinkArray" +#define RANK 2 + + +int main( void ) +{ + + hid_t file_id; + hid_t dataset_id; + hid_t space_id; + hid_t plist_id; + hsize_t dims[2] = { 3, 3}; + hsize_t dims_new[2] = { 2, 2}; + hsize_t dims_chunk[2] = { 3, 3}; + hsize_t dims_out[2]; + hsize_t maxdims[1] = { H5S_UNLIMITED }; + int data1[3][3] = { {1, 2, 3}, {4, 5, 6}, {7, 8, 9} }; + int buf[2][2]; + herr_t status; + int i, j; + + /* Create a new file using default properties. */ + file_id = H5Fcreate( "set_extend.h5", H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT ); + + /* Create the data space with unlimited dimensions. */ + space_id = H5Screate_simple( RANK, dims, maxdims ); + + /* Modify dataset creation properties, i.e. enable chunking. */ + plist_id = H5Pcreate (H5P_DATASET_CREATE); + status = H5Pset_chunk( plist_id, RANK, dims_chunk); + + /* Create a new dataset within the file using cparms creation properties. */ + dataset_id = H5Dcreate( file_id , DATASETNAME, H5T_NATIVE_INT, space_id, plist_id ); + + /* Write the data. */ + status = H5Dwrite( dataset_id , H5T_NATIVE_INT, space_id, H5S_ALL, H5P_DEFAULT, data1 ); + +/*------------------------------------------------------------------------- + * Set new dimensions for the array + *------------------------------------------------------------------------- + */ + + /* Set new dimensions for the array. */ + status = H5Dset_extend( dataset_id , dims_new ); + + /* Get the space. */ + space_id = H5Dget_space( dataset_id ); + + /* Get dimensions. */ + status = H5Sget_simple_extent_dims( space_id, dims_out, NULL); + + for( i = 0; i < 2; i++ ) + { + if ( dims_out[i] != dims_new[i] ) + goto out; + } + + +/*------------------------------------------------------------------------- + * Read + *------------------------------------------------------------------------- + */ + + /* Read the new dataset. */ + status = H5Dread( dataset_id, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, buf ); + + /* Compare the read array with the original array */ + for( i = 0; i < dims_out[0]; i++ ) + for( j = 0; j < dims_out[1]; j++ ) + { + if ( buf[i][j] != data1[i][j] ) { + goto out; + } + } + + + +/*------------------------------------------------------------------------- + * end + *------------------------------------------------------------------------- + */ + + /* Close/release resources. */ + H5Dclose( dataset_id ); + H5Sclose( space_id ); + H5Pclose( plist_id ); + H5Fclose( file_id ); + printf("%s \n", "H5Dset_extend test PASSED" ); + + return 0; + +out: + H5Dclose( dataset_id ); + H5Sclose( space_id ); + H5Pclose( plist_id ); + H5Fclose( file_id ); + printf("%s \n", "H5Dset_extend FAILED" ); + return 1; + +} + |