#include "hdf5.h"
#include "h5test.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[2] = {H5S_UNLIMITED, H5S_UNLIMITED};
  int     data1[3][3] = { {1, 2, 3}, {4, 5, 6}, {7, 8, 9} }; 
#if 0
  int     buf1[3][3];
#endif
  int     buf2[2][2];
  herr_t  status; 
  hsize_t i, j;

  TESTING("extend dataset");
  
  /* 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; shrink it to (2,2)
 *-------------------------------------------------------------------------
 */
  
  /* 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, buf2 );

  /* 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 (  buf2[i][j] != data1[i][j] ) {
     goto out;
    }
   }


#if 0

/*-------------------------------------------------------------------------
 * Set new dimensions for the array; expand it again to (3,3)
 *-------------------------------------------------------------------------
 */
  
  /* Set new dimensions for the array. */
  status = H5Dset_extend( dataset_id , dims );

   /* 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[i] ) 
    goto out;
  }


/*-------------------------------------------------------------------------
 * Read
 *-------------------------------------------------------------------------
 */
 
  /* Read the new dataset. */
  status = H5Dread( dataset_id, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, buf1 );

  /* 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 ( i > 1 || j > 1 )
	{
     if (  buf1[i][j] != 0 ) 
      goto out;
	}
	else
	{
     if (  buf1[i][j] != data1[i][j] ) 
      goto out;
    }
   }

#endif
    
/*-------------------------------------------------------------------------
 * end
 *-------------------------------------------------------------------------
 */
 
 /* Close/release resources. */
 H5Dclose( dataset_id );
 H5Sclose( space_id );
 H5Pclose( plist_id  );
 H5Fclose( file_id );
 PASSED();

 return 0;

out:
 H5Dclose( dataset_id );
 H5Sclose( space_id );
 H5Pclose( plist_id  );
 H5Fclose( file_id );
 H5_FAILED();
 return 1;

}