#include "hdf5.h"
#include "H5LDprivate.h"
#include <time.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <stdlib.h>

/*
 *  WATCH.h5: file with various types of datasets for testing--                                   
 *
 *  The following datasets are chunked, H5D_ALLOC_TIME_INCR, max. dimensional setting:       
 *      DSET_ONE: one-dimensional dataset                                                   
 *      DSET_TWO: two-dimensional dataset                                                  
 *      DSET_CMPD: one-dimensional dataset with compound type                             
 *      DSET_CMPD_ESC: one-dimensional dataset with compound type and member names with
 *		       escape/separator characters 
 *      DSET_CMPD_TWO: two-dimensional dataset with compound type                             
 *                                                                                           
 *  The following datasets are one-dimensional, chunked, max. dimension setting:              
 *      DSET_ALLOC_EARLY: dataset with H5D_ALLOC_TIME_EARLY                                  
 *      DSET_ALLOC_LATE: dataset H5D_ALLOC_TIME_LATE                                        
 *                                                                                       
 *  The following datasets are one-dimensional:                                           
 *      DSET_NONE: fixed dimension setting, contiguous, H5D_ALLOC_TIME_LATE              
 *      DSET_NOMAX: fixed dimension setting, chunked, H5D_ALLOC_TIME_INCR      
 */
#define ONE_DIMS0	10
#define MAX_ONE_DIMS0	100

#define DSET_ONE 	"DSET_ONE"
#define DSET_NONE	"DSET_NONE"
#define DSET_NOMAX	"DSET_NOMAX"
#define DSET_ALLOC_LATE "DSET_ALLOC_LATE"
#define DSET_ALLOC_EARLY "DSET_ALLOC_EARLY"
#define DSET_CMPD 	"DSET_CMPD"
#define DSET_CMPD_ESC 	"DSET_CMPD_ESC"
#define DSET_NULL 	"DSET_NULL"
#define DSET_SCALAR 	"DSET_SCALAR"

#define TWO_DIMS0	4
#define TWO_DIMS1	10
#define MAX_TWO_DIMS0	60
#define MAX_TWO_DIMS1	100

#define DSET_TWO 	"DSET_TWO"
#define DSET_CMPD_TWO 	"DSET_CMPD_TWO"

#define CHUNK_SIZE	2

#define FILE "test_ld.h5"

/* Data structures for datasets with compound types */
typedef struct sub22_t {
    unsigned int a;
    unsigned int b;
    unsigned int c;
} sub22_t;

typedef struct sub2_t {
    unsigned int a;
    sub22_t b;
    unsigned int c;
} sub2_t;

typedef struct sub4_t {
    unsigned int a;
    unsigned int b;
} sub4_t;

typedef struct set_t {
    unsigned int field1;
    sub2_t field2;
    double field3;
    sub4_t field4;
} set_t;

/*
 **************************************************************************************
 *
 * Create a dataset with the given input parameters
 * Write to the dataset with the given "data"
 *
 **************************************************************************************
 */
static int
generate_dset(hid_t fid, const char *dname, int ndims, hsize_t *dims, hsize_t *maxdims, hid_t dtid, void *data)
{
    hid_t dcpl;		/* Dataset creation property */
    hid_t did;		/* Dataset id */
    hid_t sid;		/* Dataspace id */
    int i;		/* Local index variable */
    hsize_t chunk_dims[H5S_MAX_RANK];	/* Dimension sizes for chunks */

    /* Create the dataspace */
    if((sid = H5Screate_simple(ndims, dims, maxdims)) < 0)
	goto done;

    /* Set up dataset's creation properties */
    if(!HDstrcmp(dname, DSET_NONE))
	dcpl = H5P_DEFAULT;
    else {
	if((dcpl = H5Pcreate(H5P_DATASET_CREATE)) < 0)
	    goto done;
	for(i = 0; i < ndims; i++)
	    chunk_dims[i] = CHUNK_SIZE;
	if(H5Pset_chunk(dcpl, ndims, chunk_dims) < 0)
	    goto done;
    }

    if(!HDstrcmp(dname, DSET_ALLOC_LATE)) {
	if(H5Pset_alloc_time(dcpl, H5D_ALLOC_TIME_LATE) < 0)
	    goto done;
    } else if(!HDstrcmp(dname, DSET_ALLOC_EARLY)) {
	if(H5Pset_alloc_time(dcpl, H5D_ALLOC_TIME_EARLY) < 0)
	    goto done;
    }

    /* Create the dataset */
    if((did = H5Dcreate2(fid, dname, dtid, sid, H5P_DEFAULT, dcpl, H5P_DEFAULT)) < 0)
	goto done;

    /* Write to the dataset */
    if(H5Dwrite(did, dtid, H5S_ALL, H5S_ALL, H5P_DEFAULT, data) < 0)
	goto done;

    /* Closing */
    if(H5Pclose(dcpl) < 0) goto done;
    if(H5Sclose(sid) < 0) goto done;
    if(H5Dclose(did) < 0) goto done;

    return(SUCCEED);

done:
    H5E_BEGIN_TRY
	H5Sclose(sid);
	H5Pclose(dcpl);
	H5Dclose(did);
    H5E_END_TRY

    return(FAIL);
} /* generate_dset() */

int
main(void)
{
    hid_t fid;			/* File id */
    hid_t fapl;			/* File access property list */
    hsize_t cur_dims[1];	/* Dimension sizes */
    hsize_t max_dims[1];	/* Maximum dimension sizes */
    hsize_t cur2_dims[2];	/* Current dimension sizes */
    hsize_t max2_dims[2];	/* Maximum dimension sizes */
    hid_t set_tid, esc_set_tid;	/* Compound type id */
    hid_t sub22_tid;		/* Compound type id */
    hid_t sub2_tid, esc_sub2_tid;	/* Compound type id */
    hid_t sub4_tid, esc_sub4_tid;	/* Compound type id */
    hid_t null_did, null_sid;		/* H5S_NULL dataset & dataspace ids */
    hid_t scalar_did, scalar_sid;	/* H5S_SCALAR dataset & dataspace ids */
    int one_data[ONE_DIMS0];		/* Buffer for data */
    int two_data[TWO_DIMS0*TWO_DIMS1];	/* Buffer for data */
    set_t one_cbuf[ONE_DIMS0];		/* Buffer for data with compound type */
    set_t two_cbuf[TWO_DIMS0*TWO_DIMS1];	/* Buffer for data with compound type */
    int i;			/* Local index variable */

    /* Create a file access property list */
    if((fapl = H5Pcreate(H5P_FILE_ACCESS)) < 0)
        goto done;

    /* Set to use latest library format */
    if((H5Pset_libver_bounds(fapl, H5F_LIBVER_LATEST, H5F_LIBVER_LATEST)) < 0)
        goto done;

    /* Create a file */
    if((fid = H5Fcreate(FILE, H5F_ACC_TRUNC, H5P_DEFAULT, fapl)) < 0)
	goto done;

    /* Initialization for one-dimensional dataset */
    cur_dims[0] = ONE_DIMS0;
    max_dims[0] = MAX_ONE_DIMS0;
    for(i = 0; i < ONE_DIMS0; i++)
        one_data[i] = i;

    /* Generate DSET_ONE, DSET_NONE, DSET_NOMAX, DSET_ALLOC_LATE, DSET_EARLY */
    if(generate_dset(fid, DSET_ONE, 1, cur_dims, max_dims, H5T_NATIVE_INT, one_data) < 0)
	goto done;
    if(generate_dset(fid, DSET_NONE, 1, cur_dims, NULL, H5T_NATIVE_INT, one_data) < 0)
	goto done;
    if(generate_dset(fid, DSET_NOMAX, 1, cur_dims, NULL, H5T_NATIVE_INT, one_data) < 0)
	goto done;
    if(generate_dset(fid, DSET_ALLOC_LATE, 1, cur_dims, max_dims, H5T_NATIVE_INT, one_data) < 0)
	goto done;
    if(generate_dset(fid, DSET_ALLOC_EARLY, 1, cur_dims, max_dims, H5T_NATIVE_INT, one_data) < 0)
	goto done;

    /* Initialization for two-dimensional dataset */
    cur2_dims[0] = TWO_DIMS0;
    cur2_dims[1] = TWO_DIMS1;
    max2_dims[0] = MAX_TWO_DIMS0;
    max2_dims[1] = MAX_TWO_DIMS1;

    for(i = 0; i < (TWO_DIMS0 * TWO_DIMS1); i++)
        two_data[i] = i;

    /* Generate DSET_TWO */
    if(generate_dset(fid, DSET_TWO, 2, cur2_dims, max2_dims, H5T_NATIVE_INT, two_data) < 0)
	goto done;

    /* Initialization for one-dimensional compound typed dataset */
    cur_dims[0] = ONE_DIMS0;
    max_dims[0] = MAX_ONE_DIMS0;

    for (i = 0; i < ONE_DIMS0; i++) {
        one_cbuf[i].field1 = 1;
        one_cbuf[i].field2.a = 2;
        one_cbuf[i].field2.c = 4;
        one_cbuf[i].field2.b.a = 20;
        one_cbuf[i].field2.b.b = 40;
        one_cbuf[i].field2.b.c = 80;
        one_cbuf[i].field3 = 3.0;
        one_cbuf[i].field4.a = 4;
        one_cbuf[i].field4.b = 8;
    }

    /* Create the compound type */
    if((sub22_tid = H5Tcreate(H5T_COMPOUND, sizeof(sub22_t))) < 0)
	goto done;
    if(H5Tinsert(sub22_tid, "a", HOFFSET(sub22_t, a), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(sub22_tid, "b", HOFFSET(sub22_t, b), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(sub22_tid, "c", HOFFSET(sub22_t, c), H5T_NATIVE_INT) < 0)
	goto done;

    if((sub2_tid = H5Tcreate(H5T_COMPOUND, sizeof(sub2_t))) < 0)
	goto done;
    if(H5Tinsert(sub2_tid, "a", HOFFSET(sub2_t, a), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(sub2_tid, "b", HOFFSET(sub2_t, b), sub22_tid) < 0)
	goto done;
    if(H5Tinsert(sub2_tid, "c", HOFFSET(sub2_t, c), H5T_NATIVE_INT) < 0)
	goto done;

    if((sub4_tid = H5Tcreate(H5T_COMPOUND, sizeof(sub4_t))) < 0)
	goto done;
    if(H5Tinsert(sub4_tid, "a", HOFFSET(sub4_t, a), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(sub4_tid, "b", HOFFSET(sub4_t, b), H5T_NATIVE_INT) < 0)
	goto done;

    if((set_tid = H5Tcreate(H5T_COMPOUND, sizeof(set_t))) < 0)
	goto done;
    if(H5Tinsert(set_tid, "field1", HOFFSET(set_t, field1), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(set_tid, "field2", HOFFSET(set_t, field2), sub2_tid) < 0)
	goto done;
    if(H5Tinsert(set_tid, "field3", HOFFSET(set_t, field3), H5T_NATIVE_DOUBLE) < 0)
	goto done;
    if(H5Tinsert(set_tid, "field4", HOFFSET(set_t, field4), sub4_tid) < 0)
	goto done;

    /* Create the compound type with escape/separator characters */
    if((esc_sub2_tid = H5Tcreate(H5T_COMPOUND, sizeof(sub2_t))) < 0)
	goto done;
    if(H5Tinsert(esc_sub2_tid, ".a", HOFFSET(sub2_t, a), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(esc_sub2_tid, ",b", HOFFSET(sub2_t, b), sub22_tid) < 0)
	goto done;
    if(H5Tinsert(esc_sub2_tid, "\\c", HOFFSET(sub2_t, c), H5T_NATIVE_INT) < 0)
	goto done;

    if((esc_sub4_tid = H5Tcreate(H5T_COMPOUND, sizeof(sub4_t))) < 0)
	goto done;
    if(H5Tinsert(esc_sub4_tid, "a.", HOFFSET(sub4_t, a), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(esc_sub4_tid, "b,", HOFFSET(sub4_t, b), H5T_NATIVE_INT) < 0)
	goto done;

    if((esc_set_tid = H5Tcreate(H5T_COMPOUND, sizeof(set_t))) < 0)
	goto done;
    if(H5Tinsert(esc_set_tid, "field,1", HOFFSET(set_t, field1), H5T_NATIVE_INT) < 0)
	goto done;
    if(H5Tinsert(esc_set_tid, "field2.", HOFFSET(set_t, field2), esc_sub2_tid) < 0)
	goto done;
    if(H5Tinsert(esc_set_tid, "field\\3", HOFFSET(set_t, field3), H5T_NATIVE_DOUBLE) < 0)
	goto done;
    if(H5Tinsert(esc_set_tid, "field4,", HOFFSET(set_t, field4), esc_sub4_tid) < 0)
	goto done;

    /* Generate DSET_CMPD, DSET_CMPD_ESC */
    if(generate_dset(fid, DSET_CMPD, 1, cur_dims, max_dims, set_tid, one_cbuf) < 0)
	goto done;
    if(generate_dset(fid, DSET_CMPD_ESC, 1, cur_dims, max_dims, esc_set_tid, one_cbuf) < 0)
	goto done;

    /* Initialization for two-dimensional compound typed dataset */
    cur2_dims[0] = TWO_DIMS0;
    cur2_dims[1] = TWO_DIMS1;
    max2_dims[0] = MAX_TWO_DIMS0;
    max2_dims[0] = MAX_TWO_DIMS1;

    for (i = 0; i < (TWO_DIMS0 * TWO_DIMS1); i++) {
        two_cbuf[i].field1 = 1;
        two_cbuf[i].field2.a = 2;
        two_cbuf[i].field2.c = 4;
        two_cbuf[i].field2.b.a = 20;
        two_cbuf[i].field2.b.b = 40;
        two_cbuf[i].field2.b.c = 80;
        two_cbuf[i].field3 = 3.0;
        two_cbuf[i].field4.a = 4;
        two_cbuf[i].field4.b = 8;
    }

    /* Generate DSET_CMPD_TWO */
    if(generate_dset(fid, DSET_CMPD_TWO, 2, cur2_dims, max2_dims, set_tid, two_cbuf) < 0)
	goto done;

    /* Create NULL dataspace */
    if((null_sid = H5Screate(H5S_NULL)) < 0)
	goto done;

    /* Create the NULL dataset */
    if((null_did = H5Dcreate2(fid, DSET_NULL, H5T_NATIVE_UINT, null_sid, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT)) < 0)
	goto done;

    /* Create SCALAR dataspace */
    if((scalar_sid = H5Screate(H5S_SCALAR)) < 0)
	goto done;

    /* Create the SCALAR dataset */
    if((scalar_did = H5Dcreate2(fid, DSET_SCALAR, H5T_NATIVE_INT, scalar_sid, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT)) < 0)
	goto done;

    /* Closing */
    if(H5Dclose(scalar_did) < 0) goto done;
    if(H5Sclose(scalar_sid) < 0) goto done;

    if(H5Dclose(null_did) < 0) goto done;
    if(H5Sclose(null_sid) < 0) goto done;

    if(H5Tclose(sub22_tid) < 0) goto done;
    if(H5Tclose(sub2_tid) < 0) goto done;
    if(H5Tclose(sub4_tid) < 0) goto done;
    if(H5Tclose(set_tid) < 0) goto done;
    if(H5Tclose(esc_sub2_tid) < 0) goto done;
    if(H5Tclose(esc_sub4_tid) < 0) goto done;
    if(H5Tclose(esc_set_tid) < 0) goto done;

    if(H5Pclose(fapl) < 0) goto done;
    if(H5Fclose(fid) < 0) goto done;

    exit(EXIT_SUCCESS);

done:
    H5E_BEGIN_TRY
	H5Tclose(sub22_tid);
	H5Tclose(sub2_tid);
	H5Tclose(sub4_tid);
	H5Tclose(set_tid);
	H5Tclose(esc_sub2_tid);
	H5Tclose(esc_sub4_tid);
	H5Tclose(esc_set_tid);

	H5Dclose(null_did);
	H5Sclose(null_sid);
	H5Dclose(scalar_did);
	H5Sclose(scalar_sid);

	H5Pclose(fapl);
	H5Fclose(fid);
    H5E_END_TRY

    exit(EXIT_FAILURE);
} /* main() */