path: root/test/chunk_info.c

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 * Copyright by The HDF Group.                                               *
 * Copyright by the Board of Trustees of the University of Illinois.         *
 * All rights reserved.                                                      *
 *                                                                           *
 * This file is part of HDF5.  The full HDF5 copyright notice, including     *
 * terms governing use, modification, and redistribution, is contained in    *
 * the COPYING file, which can be found at the root of the source code       *
 * distribution tree, or in https://support.hdfgroup.org/ftp/HDF5/releases.  *
 * If you do not have access to either file, you may request a copy from     *
 * help@hdfgroup.org.                                                        *
 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

/*
 * Programmer:  Pedro Vicente <pvn@hdfgroup.edu>
 *              April 7, 2008
 *
 * Purpose:     Tests chunk query API functions
 */

#include "h5test.h"

 /* #define PRINT_DATA
 */ 
#define H5FILE_NAME "chunk_info.h5"
#define DATASETNAME "2d"
#define RANK         2
#define FILENAME_BUF_SIZE       1024

/* Parameters for testing chunk querying */
#define FILENAME                "tchunk_info"
#define DSET_SIMPLE_CHUNKED     "Chunked Dataset"
#define DSET_EMPTY              "Empty Dataset"
#define NX          16
#define NY          16
#define CHUNK_NX     4
#define CHUNK_NY     4
#define CHUNK_SIZE  64
#define NUM_CHUNKS_WRITTEN  4

void reinit_vars(unsigned *read_filter_mask, haddr_t *addr, hsize_t *size);

/*-------------------------------------------------------------------------
 * Function:    read_each_chunk (helper function)
 *
 * Purpose:     Reads the chunk specified by its offset and verifies that
 *              it contains the same data as what was written.  This function
 *              is used in test_get_chunk_info.
 *
 * Return:      Success:    SUCCEED
 *              Failure:    FAIL
 *
 * Date:        September 2018
 *
 *-------------------------------------------------------------------------
 */
static herr_t read_each_chunk(hid_t dset_id, hsize_t offset1, hsize_t offset2, void *direct_buf)
{
    int      read_buf[CHUNK_NX][CHUNK_NY];
    hsize_t  offset[2] = {offset1, offset2};
    unsigned read_filter_mask = 0;

    /* Read the raw chunk back */
    HDmemset(&read_buf, 0, sizeof(read_buf));

    /* Read the chunk specified by its offset */
    if (H5Dread_chunk(dset_id, H5P_DEFAULT, offset, &read_filter_mask, read_buf) < 0)
        return(FAIL);

    /* Verify that read chunk is the same as the corresponding written one */
    if (HDmemcmp(direct_buf, read_buf, CHUNK_NX*CHUNK_NY) != 0)
    {
        HDfprintf(stderr, "Read chunk differs from written chunk at offset (%d,%d)\n", offset1, offset2);
        return(FAIL);
    }

    return(SUCCEED);
}

/*-------------------------------------------------------------------------
 * Function:    reinit_vars (helper function)
 *
 * Purpose:     Helper function to wipe out variables for the next use,
 *              used in test_get_chunk_info.
 *
 * Return:      Won't fail
 *
 * Date:        September 2018
 *
 *-------------------------------------------------------------------------
 */
void reinit_vars(unsigned *read_filter_mask, haddr_t *addr, hsize_t *size)
{
    if (read_filter_mask)
        *read_filter_mask = 0;
    if (addr)
        *addr = 0;
    if (size)
        *size = 0;
}

/*-------------------------------------------------------------------------
 * Function:	test_get_chunk_info
 *
 * Purpose:	    Test getting various chunk information
 *
 * Return:	    Success:	SUCCEED
 *		        Failure:	FAIL
 *
 * Note:        Note that the dataspace argument in these new functions are
 *              currently not used.  The functionality involved the dataspace
 *              will be implemented in the next version.
 *
 * Description:
 *              This function tests the new API functions added for HDFFV-10615:
 *              H5Dget_num_chunks, H5Dget_chunk_info, and H5Dget_chunk_info_by_coord.
 *
 * Date:        September 2018
 *
 *-------------------------------------------------------------------------
 */
static herr_t
test_get_chunk_info(void)
{
    hid_t       chunkfile = -1, fapl = -1;
    hid_t       fspace = -1, dset = -1;
    hid_t       mspace = -1;
    hid_t       cparms = -1, dxpl = -1;
    hsize_t     dims[2]  = {NX, NY};
    hsize_t     maxdims[2] = {H5S_UNLIMITED, H5S_UNLIMITED};
    hsize_t     chunk_dims[2] ={CHUNK_NX, CHUNK_NY};
    int         fillvalue = -1;
    char        filename[FILENAME_BUF_SIZE];
    unsigned    filter_mask = 0;
    int         direct_buf[16][CHUNK_NX][CHUNK_NY];
    int         out_buf[NX][NY];
    size_t      buf_size = CHUNK_NX*CHUNK_NY*sizeof(int);
    int         aggression = 9;         /* Compression aggression setting */
    unsigned    read_filter_mask = 0;   /* filter mask after direct read */
    H5F_libver_t low, high;             /* File format bounds */
    hsize_t offset[2];
    hsize_t out_offset[2] = {0, 0};
    hsize_t size = 0;
    hsize_t nchunks = 0;
    haddr_t addr = 0;
    hsize_t index = 0;
    hsize_t i, j;
    int     n; /* for use on buffer, to avoid conversion warning */

    TESTING("getting chunk information");

    /* Create a file */
    h5_fixname(FILENAME, H5P_DEFAULT, filename, sizeof filename);
    if((chunkfile = H5Fcreate(filename, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT)) < 0)
        TEST_ERROR

    /* Create the data space */
    if((fspace = H5Screate_simple(RANK, dims, maxdims)) < 0)
        TEST_ERROR

    if((mspace = H5Screate_simple(RANK, chunk_dims, NULL)) < 0)
        TEST_ERROR

    /* Modify dataset creation properties, i.e. enable chunking and compression */
    if((cparms = H5Pcreate(H5P_DATASET_CREATE)) < 0)
        TEST_ERROR

    if(H5Pset_chunk(cparms, RANK, chunk_dims) < 0)
        TEST_ERROR

    if(H5Pset_deflate(cparms, (unsigned ) aggression) < 0)
        TEST_ERROR

    if (H5Pset_fill_value(cparms, H5T_NATIVE_INT, &fillvalue) < 0)
        TEST_ERROR;

    /* Create a new dataset using cparms creation properties */
    if((dset = H5Dcreate2(chunkfile, DSET_SIMPLE_CHUNKED, H5T_NATIVE_INT, fspace,
            H5P_DEFAULT, cparms, H5P_DEFAULT)) < 0) TEST_ERROR

    if((dxpl = H5Pcreate(H5P_DATASET_XFER)) < 0)
        TEST_ERROR

    /* Indicate skipping the compression filter.     */
    filter_mask = 0x00000001;

    /* Initialize the array of chunk data, that is for all 16 chunks */
    for(n = 0; n < 16; n++)
        for(i = 0; i < CHUNK_NX; i++)
            for(j = 0; j < CHUNK_NY; j++)
                direct_buf[n][i][j] = n + 1;

    /* Write NUM_CHUNKS_WRITTEN chunks: (0,2) (0,3) (1,2) (1,3) */
    n = 0;
    for (i = 0; i < 2; i++)
        for (j = 2; j < 4; j++, n++)
        {
            offset[0] = i * CHUNK_NX;
            offset[1] = j * CHUNK_NY;
            if (H5Dwrite_chunk(dset, dxpl, filter_mask, offset, buf_size, (void*)direct_buf[n]) < 0)
                TEST_ERROR
        }

    /* Close the dataset then... */
    if(H5Dclose(dset) < 0) TEST_ERROR

    /* ...open it again to test the chunk query functions */
    if((dset = H5Dopen2(chunkfile, DSET_SIMPLE_CHUNKED, H5P_DEFAULT)) < 0) TEST_ERROR

    /* Get and verify the number of chunks written */
    if (H5Dget_num_chunks(dset, mspace, &nchunks) < 0) TEST_ERROR;
    if (nchunks != NUM_CHUNKS_WRITTEN) TEST_ERROR;

    /* Read the entire dataset back */
    if(H5Dread(dset, H5T_NATIVE_INT, fspace, fspace, H5P_DEFAULT, out_buf) < 0)
        TEST_ERROR

    /* Get and verify info of the first chunk */
    index = 0;
    reinit_vars(&read_filter_mask, &addr, &size);
    if (H5Dget_chunk_info(dset, fspace, index, out_offset, &read_filter_mask, &addr, &size) < 0)
        TEST_ERROR
    if (read_filter_mask != filter_mask) TEST_ERROR;
    if (size != CHUNK_SIZE) TEST_ERROR;
    if (out_offset[0] != 0 || out_offset[1] != 8) TEST_ERROR;

    /* Get and verify info of the second chunk */
    index = 1;
    reinit_vars(&read_filter_mask, &addr, &size);
    if (H5Dget_chunk_info(dset, fspace, index, out_offset, &read_filter_mask, &addr, &size) < 0)
        TEST_ERROR
    if (read_filter_mask != filter_mask) TEST_ERROR;
    if (size != CHUNK_SIZE) TEST_ERROR;
    if (out_offset[0] != 0 || out_offset[1] != 12) TEST_ERROR;

    /* Get and verify info of the third chunk */
    index = 2;
    reinit_vars(&read_filter_mask, &addr, &size);
    if (H5Dget_chunk_info(dset, fspace, index, out_offset, &read_filter_mask, &addr, &size) < 0)
        TEST_ERROR
 
    if (read_filter_mask != filter_mask) TEST_ERROR;
    if (size != CHUNK_SIZE) TEST_ERROR;
    if (out_offset[0] != 4 || out_offset[1] != 8) TEST_ERROR;

    /* Get and verify info of the last chunk */
    index = 3;
    reinit_vars(&read_filter_mask, &addr, &size);
    if (H5Dget_chunk_info(dset, fspace, index, out_offset, &read_filter_mask, &addr, &size) < 0)
        TEST_ERROR
    if (read_filter_mask != filter_mask) TEST_ERROR;
    if (size != CHUNK_SIZE) TEST_ERROR;
    if (out_offset[0] != 4 || out_offset[1] != 12) TEST_ERROR;

    /* Attempt to get info of empty chunk and verify the returned address and size */
    index = 5;
    reinit_vars(&read_filter_mask, &addr, &size);
    if (H5Dget_chunk_info(dset, fspace, index, out_offset, &read_filter_mask, &addr, &size) < 0)
        TEST_ERROR
    if (addr != HADDR_UNDEF) TEST_ERROR;
    if (size != 0) TEST_ERROR;

    /* Get info of the chunk at logical coordinates (0,2) */
    offset[0] = 0;
    offset[1] = 2 * CHUNK_NY;
    if (H5Dget_chunk_info_by_coord(dset, offset, &read_filter_mask, &addr, &size) < 0) TEST_ERROR;
    if (read_filter_mask != filter_mask) TEST_ERROR;
    if (size != CHUNK_SIZE) TEST_ERROR;

    /* Get info of the chunk at logical coordinates (1,3) */
    offset[0] = 1 * CHUNK_NX;
    offset[1] = 3 * CHUNK_NY;
    if (H5Dget_chunk_info_by_coord(dset, offset, &read_filter_mask, &addr, &size) < 0) TEST_ERROR;
    if (read_filter_mask != filter_mask) TEST_ERROR;
    if (size != CHUNK_SIZE) TEST_ERROR;

    /* Attempt to get info of empty chunks, verify the returned address and size */
    offset[0] = 0;
    offset[1] = 0;
    if (H5Dget_chunk_info_by_coord(dset, offset, &read_filter_mask, &addr, &size) < 0) TEST_ERROR;
    if (addr != HADDR_UNDEF) TEST_ERROR;
    if (size != 0) TEST_ERROR;

    offset[0] = 3 * CHUNK_NX;
    offset[1] = 3 * CHUNK_NY;
    if (H5Dget_chunk_info_by_coord(dset, offset, &read_filter_mask, &addr, &size) < 0) TEST_ERROR;
    if (addr != HADDR_UNDEF) TEST_ERROR;
    if (size != 0) TEST_ERROR;

    /* Read each chunk and verify the values */
    n = 0;
    for (i = 0; i < 2; i++)
        for (j = 2; j < 4; j++, n++)
            if (read_each_chunk(dset, i*CHUNK_NX, j*CHUNK_NY, (void*)direct_buf[n]) < 0)
                TEST_ERROR

    /* Close the first dataset */
    if (H5Dclose(dset) < 0) TEST_ERROR

    /* Create an empty dataset */
    if((dset = H5Dcreate2(chunkfile, DSET_EMPTY, H5T_NATIVE_INT, fspace,
            H5P_DEFAULT, cparms, H5P_DEFAULT)) < 0) TEST_ERROR
    if (H5Dclose(dset) < 0) TEST_ERROR

    /* Reopen the empty dataset to verify the chunk query functions on it */
    if((dset = H5Dopen2(chunkfile, DSET_EMPTY, H5P_DEFAULT)) < 0) TEST_ERROR

    /* Verify that the number of chunks is 0 */
    if (H5Dget_num_chunks(dset, mspace, &nchunks) < 0) TEST_ERROR;
    if (nchunks != 0) TEST_ERROR;

    /* Attempt to get info of a chunk from an empty dataset, verify the
       returned address and size */
    index = 0;
    reinit_vars(&read_filter_mask, &addr, &size);
    if (H5Dget_chunk_info(dset, fspace, index, out_offset, &read_filter_mask, &addr, &size) < 0)
        TEST_ERROR
    if (addr != HADDR_UNDEF) TEST_ERROR;
    if (size != 0) TEST_ERROR;

    /* Attempt to get info of a chunk given its coords from an empty dataset,
       verify the returned address and size */
    offset[0] = 0;
    offset[1] = 0;
    if (H5Dget_chunk_info_by_coord(dset, offset, &read_filter_mask, &addr, &size) < 0) TEST_ERROR;
    if (addr != HADDR_UNDEF) TEST_ERROR;
    if (size != 0) TEST_ERROR;

    /* Close/release resources. */
    if (H5Sclose(mspace) < 0) TEST_ERROR
    if (H5Sclose(fspace) < 0) TEST_ERROR
    if (H5Pclose(cparms) < 0) TEST_ERROR
    if (H5Pclose(dxpl) < 0) TEST_ERROR
    if (H5Fclose(chunkfile) < 0) TEST_ERROR

    PASSED();
    return SUCCEED;

error:
    H5E_BEGIN_TRY {
        H5Dclose(dset);
        H5Sclose(mspace);
        H5Sclose(fspace);
        H5Pclose(cparms);
        H5Pclose(dxpl);
        H5Fclose(chunkfile);
    } H5E_END_TRY;

    H5_FAILED();
    return FAIL;
} /* test_get_chunk_info() */

/*-------------------------------------------------------------------------
 * Function:    create_4x4_dset
 *
 * Purpose:     Tests functions related to chunk information
 *
 * Return:      Success:    SUCCEED
 *              Failure:    FAIL
 *
 * Description:
 *              The code in this create_4x4_dset function was originally written by
 *              Pedro in main() that writes a 4x4 dataset by iterating on
 *              2x2 chunks at a time, with the intention of making a frame
 *              work to test H5Dget_chunk_info, which was not in the library,
 *              until now.  For the work in HDFFV-10615, the test function
 *              test_get_chunk_info was added to test the new query chunk
 *              API functions: H5Dget_num_chunk, H5Dget_chunk_info, and
 *              H5Dget_chunk_info_by_coord.  This code can be used at a
 *              later time, so it is kept here.
 *              -BMR, November 5, 2018
 *
 * Programmer:  Pedro Vicente <pvn@hdfgroup.edu>
 *              April 7, 2008
 *
 *-------------------------------------------------------------------------
 */
int
create_4x4_dset(void)
{
    hid_t   fid;      /* file ID */
    hid_t   did;      /* dataset ID */
    hid_t   f_sid;    /* file space ID */
    hid_t   m_sid;    /* memory space ID */
    hid_t   pid;      /* property list ID */
    hsize_t start[2]; /* chunk location to start writing */
    hsize_t dims[2]  = { 4, 4};
    hsize_t chunk_dims[2] = { 2, 2 };
    int     chunk_data[2][2] = { {1, 1}, {1, 1} };
    int     buf[4][4];
    int     fillvalue = 0;
    hsize_t i, j;

    /* create a new file using default properties. */
    if ((fid = H5Fcreate(H5FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT)) < 0) TEST_ERROR;

    /* create the file space */
    if ((f_sid = H5Screate_simple(RANK, dims, dims)) < 0) TEST_ERROR;

    /* create the memory space with chunk dimensions */
    if ((m_sid = H5Screate_simple(RANK, chunk_dims, chunk_dims)) < 0) TEST_ERROR;
    start[0] = 0;
    start[1] = 0;
    if (H5Sselect_hyperslab(m_sid, H5S_SELECT_SET, start, NULL, chunk_dims, NULL) < 0) TEST_ERROR;

   /*-------------------------------------------------------------------------
    * create a dataset
    *-------------------------------------------------------------------------
    */

    /* modify dataset creation properties, i.e. enable chunking. */
    if ((pid = H5Pcreate (H5P_DATASET_CREATE)) < 0) TEST_ERROR;
    if (H5Pset_chunk(pid, RANK, chunk_dims) < 0) TEST_ERROR;
    if (H5Pset_fill_value(pid, H5T_NATIVE_INT, &fillvalue) < 0) TEST_ERROR;

    /* create a new dataset */
    if((did = H5Dcreate2(fid , DATASETNAME, H5T_NATIVE_INT, f_sid, H5P_DEFAULT, pid, H5P_DEFAULT)) < 0) TEST_ERROR;


   /*-------------------------------------------------------------------------
    * write the dataset in 2x2 chunks
    *-------------------------------------------------------------------------
    */

    /* iterate in dim 0 */
    for (j = 0; j < chunk_dims[0]; j++) {
        /* reset start in dim 1 */
        start[1] = 0;

        /* iterate in dim 1 */
        for (i = 0; i < chunk_dims[1]; i++) {
            /* select file hyperslab to save a 2x2 chunk */
            if (H5Sselect_hyperslab(f_sid, H5S_SELECT_SET, start, NULL, chunk_dims, NULL) < 0) TEST_ERROR;

            /* write the data to the hyperslab. */
            if (H5Dwrite(did, H5T_NATIVE_INT, m_sid, f_sid, H5P_DEFAULT, chunk_data) < 0) TEST_ERROR;

            /* read back and display complete dataset 4x4 */
            if (H5Dread(did, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, buf) < 0) TEST_ERROR;

#if defined (PRINT_DATA)
            printf("\n");
            printf("dataset: \n");
            for (jj = 0; jj < dims[0]; jj++) {
                for (ii = 0; ii < dims[1]; ii++) printf("%d ", buf[jj][ii]);
                printf("\n");
            }
#endif
            /* increment start in dim 1 */
            start[1] += 2;
        }
        /* increment start in dim 0 */
        start[0] += 2;
    }

   /*-------------------------------------------------------------------------
    * close
    *-------------------------------------------------------------------------
    */
    if (H5Dclose(did) < 0) TEST_ERROR
    if (H5Sclose(f_sid) < 0) TEST_ERROR
    if (H5Sclose(m_sid) < 0) TEST_ERROR
    if (H5Pclose(pid) < 0) TEST_ERROR
    if (H5Fclose(fid) < 0) TEST_ERROR

    return SUCCEED;

error:
    H5Dclose(did);
    H5Sclose(f_sid);
    H5Sclose(m_sid);
    H5Pclose(pid);
    H5Fclose(fid);

    return FAIL;
} /* end create_4x4_dset */

/*-------------------------------------------------------------------------
 * Function:    main
 *
 * Purpose:     Tests functions related to chunk information
 *
 * Return:      Success:    SUCCEED
 *              Failure:    FAIL
 *
 * Programmer:  Binh-Minh Ribler
 *              November 5, 2018
 *
 *-------------------------------------------------------------------------
 */
int
main(void)
{
    int     nerrors = 0;

    /* Tests getting chunk information */
    nerrors += (test_get_chunk_info() < 0 ? 1 : 0);

    /* Create a 4x4 dataset (using the existing code to avoid compilation
       warnings for now) */
    nerrors += (create_4x4_dset() < 0 ? 1 : 0);

    if(nerrors) {
        goto error;

    } /* end if */

    HDprintf("All chunk query tests passed.\n");

    return SUCCEED;

error:
    nerrors = MAX(1, nerrors);
    HDprintf("***** %d QUERY CHUNK INFO TEST%s FAILED! *****\n",
              nerrors, 1 == nerrors ? "" : "S");
    return FAIL;
}