Read image and generate templates

1 files changed, 58 insertions, 0 deletions

diff --git a/src/hdf5-file.c b/src/hdf5-file.c
index 9ece8ada..8a7ffbab 100644
--- a/src/hdf5-file.c
+++ b/src/hdf5-file.c
@@ -19,6 +19,8 @@
 #include <stdint.h>
 #include <hdf5.h>
 
+#include "image.h"
+
 
 int hdf5_write(const char *filename, const uint16_t *data,
                int width, int height)
@@ -74,3 +76,59 @@ int hdf5_write(const char *filename, const uint16_t *data,
 
 	return 0;
 }
+
+
+int hdf5_read(struct image *image, const char *filename)
+{
+	hid_t fh, sh, dh;	/* File, dataspace and data handles */
+	herr_t r;
+	hsize_t size[2];
+	hsize_t max_size[2];
+	uint16_t *buf;
+
+	fh = H5Fopen(filename, H5F_ACC_RDONLY, H5P_DEFAULT);
+	if ( fh < 0 ) {
+		/* TODO: Try other formats here. */
+		fprintf(stderr, "Couldn't open file: %s\n", filename);
+		return 1;
+	}
+
+	dh = H5Dopen(fh, "/data/data", H5P_DEFAULT);
+	if ( dh < 0 ) {
+		fprintf(stderr, "Couldn't open dataset\n");
+		H5Fclose(fh);
+		return 1;
+	}
+
+	sh = H5Dget_space(dh);
+	if ( H5Sget_simple_extent_ndims(sh) != 2 ) {
+		fprintf(stderr, "Dataset is not two-dimensional\n");
+		H5Fclose(fh);
+		return 1;
+	}
+
+	H5Sget_simple_extent_dims(sh, size, max_size);
+	printf("Data dimensions %i %i (max %i %i)\n",
+	                           (int)size[1], (int)size[0],
+	                           (int)max_size[1], (int)max_size[0]);
+
+	buf = malloc(sizeof(float)*size[0]*size[1]);
+
+	r = H5Dread(dh, H5T_NATIVE_UINT16, H5S_ALL, H5S_ALL, H5P_DEFAULT, buf);
+	if ( r < 0 ) {
+		fprintf(stderr, "Couldn't read data\n");
+		H5Dclose(dh);
+		H5Fclose(fh);
+		return 1;
+	}
+
+	image->data = buf;
+	image->height = size[0];
+	image->width = size[1];
+	image->x_centre = image->width/2;
+	image->y_centre = image->height/2;
+
+	H5Fclose(fh);
+
+	return 0;
+}