#include "hdf5.h"
#include "RLE.h"

int main()
{
	hid_t file, data_space, dataset;
	hid_t rle_plist, status;
	hsize_t dimsf[2];
	dimsf[0] = 16;
	dimsf[1] = 16;
	hsize_t chunk_size[2];
	chunk_size[0] = 16;
	chunk_size[1] = 16;
    
	//writing random data to inp_data
	uint8_t *inp_data = new uint8_t[INP_DATA_SIZE];
	for(int i=0;i<INP_DATA_SIZE;i+=10)
		for( int j=i+0; j<i+10;j++)
			inp_data[j]= i;

	// creating an empty HDF5 file with default access/creation properties
	file = H5Fcreate("rle_file.h5", H5F_ACC_TRUNC, H5P_DEFAULT,H5P_DEFAULT);

	/*
	 * Create a property list and setting a default value to fill the dataset
	 */
	rle_plist = H5Pcreate(H5P_DATASET_CREATE);	// Property list for using rle filter
	H5Pset_chunk(rle_plist, 2, chunk_size);
	
	/*
	 * Setting the new filter to our property list for creating the dataset
	 */
	unsigned int cd_values [1] = {0};
	status = H5Pset_filter(rle_plist, RLE_FILTER_ID, H5Z_FLAG_MANDATORY, 1, cd_values);
	if(status<0)
	{
		cout<<"failed in setting the filter\n";
		return -1;
	}
	/*
	 * Create a new dataset within the file.  The datatype
	 * and data space describe the data on disk, which may
	 * be different than the format used in the application's
	 * memory.
	 */
	data_space = H5Screate_simple(2, dimsf, NULL);
	std::string blk_dataset = "data";

	dataset = H5Dcreate (file, blk_dataset.c_str(), H5T_NATIVE_UCHAR, data_space, H5P_DEFAULT, rle_plist, H5P_DEFAULT);
	cout<<"size of H5T_NATIVE_UCHAR = "<<sizeof(H5T_NATIVE_UCHAR)<<endl;

	/*
	 * Checking if the RLE filter is available, confirming that it is registered successfully
	 */
	int rle_avail = H5Zfilter_avail(RLE_FILTER_ID);
	unsigned int filter_config;
	if(rle_avail)
	{
		status = H5Zget_filter_info(RLE_FILTER_ID, &filter_config);
		if((filter_config & H5Z_FILTER_CONFIG_ENCODE_ENABLED) && (filter_config && H5Z_FILTER_CONFIG_DECODE_ENABLED))
			cout<<"rle_filter is available\n";
	}
	/*
	 * Write the array to the file.  The datatype and data
	 * space describe the format of the data in the `dd'
	 * buffer.  The raw data is translated to the format
	 * required on disk defined above.  We use default raw
	 * data transfer properties.
	 */
	int err = H5Dwrite (dataset, H5T_NATIVE_UCHAR, H5S_ALL, H5S_ALL, H5P_DEFAULT, inp_data);

	delete [] inp_data;
	H5Pclose (rle_plist);
	H5Dclose (dataset);
	H5Sclose (data_space);
	H5Fclose (file);
	return 0;
}