Add new examples, h5rados_dset_wss.c and h5rados_dset_rss.c, to test

partial I/O with "shape same" selections. This is supported by the
chunked layout implementation.

2 files changed, 269 insertions, 0 deletions

diff --git a/examples/h5rados_dset_rss.c b/examples/h5rados_dset_rss.c
new file mode 100644
index 0000000..456da0c
--- /dev/null
+++ b/examples/h5rados_dset_rss.c
@@ -0,0 +1,134 @@
+#include "h5rados_example.h"
+
+int main(int argc, char *argv[]) {
+    rados_t cluster;
+    char *pool = "mypool";
+    hid_t file = -1, dset = -1, file_space = -1, mem_space = -1, fapl = -1;
+    hsize_t dims[2] = {4, 6};
+    hsize_t fstart[2], count[2], mstart[2];
+    int buf[4][6];
+    int i, j;
+
+    (void)MPI_Init(&argc, &argv);
+
+
+    if(argc != 9)
+        PRINTF_ERROR("argc != 9\n");
+
+    fstart[0] = (hsize_t)atoi(argv[3]);
+    fstart[1] = (hsize_t)atoi(argv[4]);
+    count[0] = (hsize_t)atoi(argv[5]);
+    count[1] = (hsize_t)atoi(argv[6]);
+    mstart[0] = (hsize_t)atoi(argv[7]);
+    mstart[1] = (hsize_t)atoi(argv[8]);
+
+    if(rados_create(&cluster, NULL) < 0)
+        ERROR;
+    if(rados_conf_read_file(cluster, "ceph.conf") < 0)
+        ERROR;
+
+    /* Initialize VOL */
+    if(H5VLrados_init(cluster, pool) < 0)
+        ERROR;
+
+    /* Set up FAPL */
+    if((fapl = H5Pcreate(H5P_FILE_ACCESS)) < 0)
+        ERROR;
+    if(H5Pset_fapl_rados(fapl, MPI_COMM_WORLD, MPI_INFO_NULL) < 0)
+        ERROR;
+    if(H5Pset_all_coll_metadata_ops(fapl, true) < 0)
+        ERROR;
+
+    /* Open file */
+    if((file = H5Fopen(argv[1], H5F_ACC_RDWR, fapl)) < 0)
+        ERROR;
+
+    /* Open dataset */
+    if((dset = H5Dopen2(file, argv[2], H5P_DEFAULT)) < 0)
+        ERROR;
+
+    printf("Selecting elements denoted with X\n");
+    printf("File:\n");
+    for(i = 0; i < 4; i++) {
+        if((i >= fstart[0]) && (i < fstart[0] + count[0])) {
+            for(j = 0; j < 6; j++)
+                if((j >= fstart[1]) && (j < fstart[1] + count[1]))
+                    printf("X");
+                else
+                    printf(".");
+            printf("\n");
+        }
+        else
+            printf("......\n");
+    }
+    printf("\nMemory:\n");
+    for(i = 0; i < 4; i++) {
+        if((i >= mstart[0]) && (i < mstart[0] + count[0])) {
+            for(j = 0; j < 6; j++)
+                if((j >= mstart[1]) && (j < mstart[1] + count[1]))
+                    printf("X");
+                else
+                    printf(".");
+            printf("\n");
+        }
+        else
+            printf("......\n");
+    }
+
+    /* Set up dataspaces */
+    if((file_space = H5Screate_simple(2, dims, NULL)) < 0)
+        ERROR;
+    if((mem_space = H5Screate_simple(2, dims, NULL)) < 0)
+        ERROR;
+    if(H5Sselect_hyperslab(file_space, H5S_SELECT_SET, fstart, NULL, count, NULL) < 0)
+        ERROR;
+    if(H5Sselect_hyperslab(mem_space, H5S_SELECT_SET, mstart, NULL, count, NULL) < 0)
+        ERROR;
+
+    /* Initialize buffer */
+    for(i = 0; i < 4; i++)
+        for(j = 0; j < 6; j++)
+            buf[i][j] = -1;
+
+    /* Read data */
+    if(H5Dread(dset, H5T_NATIVE_INT, mem_space, file_space, H5P_DEFAULT, buf) < 0)
+        ERROR;
+
+    /* Fill and print buffer */
+    printf("Successfully read data. Buffer is:\n");
+    for(i = 0; i < 4; i++) {
+        for(j = 0; j < 6; j++)
+            printf("%d ", buf[i][j]);
+        printf("\n");
+    }
+
+    /* Close */
+    if(H5Dclose(dset) < 0)
+        ERROR;
+    if(H5Fclose(file) < 0)
+        ERROR;
+    if(H5Sclose(file_space) < 0)
+        ERROR;
+    if(H5Sclose(mem_space) < 0)
+        ERROR;
+    if(H5Pclose(fapl) < 0)
+        ERROR;
+
+    printf("Success\n");
+
+    (void)MPI_Finalize();
+    return 0;
+
+error:
+    H5E_BEGIN_TRY {
+        H5Dclose(dset);
+        H5Fclose(file);
+        H5Sclose(file_space);
+        H5Sclose(mem_space);
+        H5Pclose(fapl);
+    } H5E_END_TRY;
+
+    (void)MPI_Finalize();
+    return 1;
+}
+
diff --git a/examples/h5rados_dset_wss.c b/examples/h5rados_dset_wss.c
new file mode 100644
index 0000000..cefc1ed
--- /dev/null
+++ b/examples/h5rados_dset_wss.c
@@ -0,0 +1,135 @@
+#include "h5rados_example.h"
+#include <time.h>
+
+int main(int argc, char *argv[]) {
+    rados_t cluster;
+    char *pool = "mypool";
+    hid_t file = -1, dset = -1, file_space = -1, mem_space = -1, fapl = -1;
+    hsize_t dims[2] = {4, 6};
+    hsize_t mstart[2], count[2], fstart[2];
+    int buf[4][6];
+    int i, j;
+
+    (void)MPI_Init(&argc, &argv);
+
+    /* Seed random number generator */
+    srand(time(NULL));
+
+    if(argc != 9)
+        PRINTF_ERROR("argc != 9\n");
+
+    mstart[0] = (hsize_t)atoi(argv[3]);
+    mstart[1] = (hsize_t)atoi(argv[4]);
+    count[0] = (hsize_t)atoi(argv[5]);
+    count[1] = (hsize_t)atoi(argv[6]);
+    fstart[0] = (hsize_t)atoi(argv[7]);
+    fstart[1] = (hsize_t)atoi(argv[8]);
+
+    if(rados_create(&cluster, NULL) < 0)
+        ERROR;
+    if(rados_conf_read_file(cluster, "ceph.conf") < 0)
+        ERROR;
+
+    /* Initialize VOL */
+    if(H5VLrados_init(cluster, pool) < 0)
+        ERROR;
+
+    /* Set up FAPL */
+    if((fapl = H5Pcreate(H5P_FILE_ACCESS)) < 0)
+        ERROR;
+    if(H5Pset_fapl_rados(fapl, MPI_COMM_WORLD, MPI_INFO_NULL) < 0)
+        ERROR;
+    if(H5Pset_all_coll_metadata_ops(fapl, true) < 0)
+        ERROR;
+
+    /* Open file */
+    if((file = H5Fopen(argv[1], H5F_ACC_RDWR, fapl)) < 0)
+        ERROR;
+
+    /* Open dataset */
+    if((dset = H5Dopen2(file, argv[2], H5P_DEFAULT)) < 0)
+        ERROR;
+
+    printf("Selecting elements denoted with X\n");
+    printf("Memory:\n");
+    for(i = 0; i < 4; i++) {
+        if((i >= mstart[0]) && (i < mstart[0] + count[0])) {
+            for(j = 0; j < 6; j++)
+                if((j >= mstart[1]) && (j < mstart[1] + count[1]))
+                    printf("X");
+                else
+                    printf(".");
+            printf("\n");
+        }
+        else
+            printf("......\n");
+    }
+    printf("\nFile:\n");
+    for(i = 0; i < 4; i++) {
+        if((i >= fstart[0]) && (i < fstart[0] + count[0])) {
+            for(j = 0; j < 6; j++)
+                if((j >= fstart[1]) && (j < fstart[1] + count[1]))
+                    printf("X");
+                else
+                    printf(".");
+            printf("\n");
+        }
+        else
+            printf("......\n");
+    }
+
+    /* Fill and print buffer */
+    printf("Writing data. Buffer is:\n");
+    for(i = 0; i < 4; i++) {
+        for(j = 0; j < 6; j++) {
+            buf[i][j] = rand() % 10;
+            printf("%d ", buf[i][j]);
+        }
+        printf("\n");
+    }
+
+
+    /* Set up dataspaces */
+    if((file_space = H5Screate_simple(2, dims, NULL)) < 0)
+        ERROR;
+    if((mem_space = H5Screate_simple(2, dims, NULL)) < 0)
+        ERROR;
+    if(H5Sselect_hyperslab(file_space, H5S_SELECT_SET, fstart, NULL, count, NULL) < 0)
+        ERROR;
+    if(H5Sselect_hyperslab(mem_space, H5S_SELECT_SET, mstart, NULL, count, NULL) < 0)
+        ERROR;
+
+    /* Write data */
+    if(H5Dwrite(dset, H5T_NATIVE_INT, mem_space, file_space, H5P_DEFAULT, buf) < 0)
+        ERROR;
+
+    /* Close */
+    if(H5Dclose(dset) < 0)
+        ERROR;
+    if(H5Fclose(file) < 0)
+        ERROR;
+    if(H5Sclose(file_space) < 0)
+        ERROR;
+    if(H5Sclose(mem_space) < 0)
+        ERROR;
+    if(H5Pclose(fapl) < 0)
+        ERROR;
+
+    printf("Success\n");
+
+    (void)MPI_Finalize();
+    return 0;
+
+error:
+    H5E_BEGIN_TRY {
+        H5Dclose(dset);
+        H5Fclose(file);
+        H5Sclose(file_space);
+        H5Sclose(mem_space);
+        H5Pclose(fapl);
+    } H5E_END_TRY;
+
+    (void)MPI_Finalize();
+    return 1;
+}
+