diff options
Diffstat (limited to 'test/swmr_addrem_writer.c')
| -rw-r--r-- | test/swmr_addrem_writer.c | 182 |
1 files changed, 90 insertions, 92 deletions
diff --git a/test/swmr_addrem_writer.c b/test/swmr_addrem_writer.c index 7ad74ed..fd3096f 100644 --- a/test/swmr_addrem_writer.c +++ b/test/swmr_addrem_writer.c @@ -39,18 +39,16 @@ /****************/ /* The maximum # of records to add/remove from the dataset in one step */ -#define MAX_SIZE_CHANGE 10 +#define MAX_SIZE_CHANGE 10 /********************/ /* Local Prototypes */ /********************/ static hid_t open_skeleton(const char *filename, unsigned verbose); -static int addrem_records(hid_t fid, unsigned verbose, unsigned long nops, - unsigned long flush_count); -static void usage(void); +static int addrem_records(hid_t fid, unsigned verbose, unsigned long nops, unsigned long flush_count); +static void usage(void); - /*------------------------------------------------------------------------- * Function: open_skeleton * @@ -72,44 +70,44 @@ static void usage(void); static hid_t open_skeleton(const char *filename, unsigned verbose) { - hid_t fid; /* File ID for new HDF5 file */ - hid_t fapl; /* File access property list */ - hid_t sid; /* Dataspace ID */ - hsize_t dim[2]; /* Dataspace dimension */ - unsigned u, v; /* Local index variable */ + hid_t fid; /* File ID for new HDF5 file */ + hid_t fapl; /* File access property list */ + hid_t sid; /* Dataspace ID */ + hsize_t dim[2]; /* Dataspace dimension */ + unsigned u, v; /* Local index variable */ HDassert(filename); /* Create file access property list */ - if((fapl = h5_fileaccess()) < 0) + if ((fapl = h5_fileaccess()) < 0) return -1; /* Set to use the latest library format */ - if(H5Pset_libver_bounds(fapl, H5F_LIBVER_LATEST, H5F_LIBVER_LATEST) < 0) + if (H5Pset_libver_bounds(fapl, H5F_LIBVER_LATEST, H5F_LIBVER_LATEST) < 0) return -1; /* Open the file */ - if((fid = H5Fopen(filename, H5F_ACC_RDWR | H5F_ACC_SWMR_WRITE, fapl)) < 0) + if ((fid = H5Fopen(filename, H5F_ACC_RDWR | H5F_ACC_SWMR_WRITE, fapl)) < 0) return -1; /* Close file access property list */ - if(H5Pclose(fapl) < 0) + if (H5Pclose(fapl) < 0) return -1; /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Opening datasets\n"); /* Open the datasets */ - for(u = 0; u < NLEVELS; u++) - for(v = 0; v < symbol_count[u]; v++) { - if((symbol_info[u][v].dsid = H5Dopen2(fid, symbol_info[u][v].name, H5P_DEFAULT)) < 0) + for (u = 0; u < NLEVELS; u++) + for (v = 0; v < symbol_count[u]; v++) { + if ((symbol_info[u][v].dsid = H5Dopen2(fid, symbol_info[u][v].name, H5P_DEFAULT)) < 0) return -1; - if((sid = H5Dget_space(symbol_info[u][v].dsid)) < 0) + if ((sid = H5Dget_space(symbol_info[u][v].dsid)) < 0) return -1; - if(2 != H5Sget_simple_extent_ndims(sid)) + if (2 != H5Sget_simple_extent_ndims(sid)) return -1; - if(H5Sget_simple_extent_dims(sid, dim, NULL) < 0) + if (H5Sget_simple_extent_dims(sid, dim, NULL) < 0) return -1; symbol_info[u][v].nrecords = dim[1]; } /* end for */ @@ -117,7 +115,6 @@ open_skeleton(const char *filename, unsigned verbose) return fid; } - /*------------------------------------------------------------------------- * Function: addrem_records * @@ -144,13 +141,13 @@ open_skeleton(const char *filename, unsigned verbose) static int addrem_records(hid_t fid, unsigned verbose, unsigned long nops, unsigned long flush_count) { - hid_t tid; /* Datatype ID for records */ - hid_t mem_sid; /* Memory dataspace ID */ - hsize_t start[2] = {0, 0}, count[2] = {1, 1}; /* Hyperslab selection values */ - hsize_t dim[2] = {1, 0}; /* Dataspace dimensions */ - symbol_t buf[MAX_SIZE_CHANGE]; /* Write buffer */ - unsigned long op_to_flush; /* # of operations before flush */ - unsigned long u, v; /* Local index variables */ + hid_t tid; /* Datatype ID for records */ + hid_t mem_sid; /* Memory dataspace ID */ + hsize_t start[2] = {0, 0}, count[2] = {1, 1}; /* Hyperslab selection values */ + hsize_t dim[2] = {1, 0}; /* Dataspace dimensions */ + symbol_t buf[MAX_SIZE_CHANGE]; /* Write buffer */ + unsigned long op_to_flush; /* # of operations before flush */ + unsigned long u, v; /* Local index variables */ HDassert(fid > 0); @@ -158,19 +155,19 @@ addrem_records(hid_t fid, unsigned verbose, unsigned long nops, unsigned long fl HDmemset(&buf, 0, sizeof(buf)); /* Create a dataspace for the record to add */ - if((mem_sid = H5Screate_simple(2, count, NULL)) < 0) + if ((mem_sid = H5Screate_simple(2, count, NULL)) < 0) return -1; /* Create datatype for appending records */ - if((tid = create_symbol_datatype()) < 0) + if ((tid = create_symbol_datatype()) < 0) return -1; /* Add and remove records to random datasets, according to frequency * distribution */ op_to_flush = flush_count; - for(u=0; u<nops; u++) { - symbol_info_t *symbol; /* Symbol to write record to */ - hid_t file_sid; /* Dataset's space ID */ + for (u = 0; u < nops; u++) { + symbol_info_t *symbol; /* Symbol to write record to */ + hid_t file_sid; /* Dataset's space ID */ /* Get a random dataset, according to the symbol distribution */ symbol = choose_dataset(); @@ -178,16 +175,16 @@ addrem_records(hid_t fid, unsigned verbose, unsigned long nops, unsigned long fl /* Decide whether to shrink or expand, and by how much */ count[1] = (hsize_t)HDrandom() % (MAX_SIZE_CHANGE * 2) + 1; - if(count[1] > MAX_SIZE_CHANGE) { + if (count[1] > MAX_SIZE_CHANGE) { /* Add records */ count[1] -= MAX_SIZE_CHANGE; /* Set the buffer's IDs (equal to its position) */ - for(v=0; v<count[1]; v++) + for (v = 0; v < count[1]; v++) buf[v].rec_id = (uint64_t)symbol->nrecords + (uint64_t)v; /* Set the memory space to the correct size */ - if(H5Sset_extent_simple(mem_sid, 2, count, NULL) < 0) + if (H5Sset_extent_simple(mem_sid, 2, count, NULL) < 0) return -1; /* Get the coordinates to write */ @@ -195,71 +192,71 @@ addrem_records(hid_t fid, unsigned verbose, unsigned long nops, unsigned long fl /* Cork the metadata cache, to prevent the object header from being * flushed before the data has been written */ - if(H5Odisable_mdc_flushes(symbol->dsid) < 0) + if (H5Odisable_mdc_flushes(symbol->dsid) < 0) return -1; - /* Extend the dataset's dataspace to hold the new record */ - symbol->nrecords+= count[1]; + /* Extend the dataset's dataspace to hold the new record */ + symbol->nrecords += count[1]; dim[1] = symbol->nrecords; - if(H5Dset_extent(symbol->dsid, dim) < 0) + if (H5Dset_extent(symbol->dsid, dim) < 0) return -1; /* Get the dataset's dataspace */ - if((file_sid = H5Dget_space(symbol->dsid)) < 0) + if ((file_sid = H5Dget_space(symbol->dsid)) < 0) return -1; /* Choose the last record in the dataset */ - if(H5Sselect_hyperslab(file_sid, H5S_SELECT_SET, start, NULL, count, NULL) < 0) + if (H5Sselect_hyperslab(file_sid, H5S_SELECT_SET, start, NULL, count, NULL) < 0) return -1; /* Write record to the dataset */ - if(H5Dwrite(symbol->dsid, tid, mem_sid, file_sid, H5P_DEFAULT, &buf) < 0) + if (H5Dwrite(symbol->dsid, tid, mem_sid, file_sid, H5P_DEFAULT, &buf) < 0) return -1; /* Uncork the metadata cache */ - if(H5Oenable_mdc_flushes(symbol->dsid) < 0) + if (H5Oenable_mdc_flushes(symbol->dsid) < 0) return -1; /* Close the dataset's dataspace */ - if(H5Sclose(file_sid) < 0) + if (H5Sclose(file_sid) < 0) return -1; } /* end if */ else { /* Shrink the dataset's dataspace */ - if(count[1] > symbol->nrecords) + if (count[1] > symbol->nrecords) symbol->nrecords = 0; else symbol->nrecords -= count[1]; dim[1] = symbol->nrecords; - if(H5Dset_extent(symbol->dsid, dim) < 0) + if (H5Dset_extent(symbol->dsid, dim) < 0) return -1; } /* end else */ /* Check for flushing file */ - if(flush_count > 0) { + if (flush_count > 0) { /* Decrement count of records to write before flushing */ op_to_flush--; /* Check for counter being reached */ - if(0 == op_to_flush) { + if (0 == op_to_flush) { /* Flush contents of file */ - if(H5Fflush(fid, H5F_SCOPE_GLOBAL) < 0) + if (H5Fflush(fid, H5F_SCOPE_GLOBAL) < 0) return -1; /* Reset flush counter */ op_to_flush = flush_count; } /* end if */ - } /* end if */ - } /* end for */ + } /* end if */ + } /* end for */ /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Closing datasets\n"); /* Close the datasets */ - for(u = 0; u < NLEVELS; u++) - for(v = 0; v < symbol_count[u]; v++) - if(H5Dclose(symbol_info[u][v].dsid) < 0) + for (u = 0; u < NLEVELS; u++) + for (v = 0; v < symbol_count[u]; v++) + if (H5Dclose(symbol_info[u][v].dsid) < 0) return -1; return 0; @@ -285,29 +282,30 @@ usage(void) HDexit(EXIT_FAILURE); } -int main(int argc, const char *argv[]) +int +main(int argc, const char *argv[]) { - hid_t fid; /* File ID for file opened */ - long nops = 0; /* # of times to grow or shrink the dataset */ - long flush_count = 1000; /* # of records to write between flushing file */ - unsigned verbose = 1; /* Whether to emit some informational messages */ - unsigned use_seed = 0; /* Set to 1 if a seed was set on the command line */ - unsigned random_seed = 0; /* Random # seed */ - unsigned u; /* Local index variable */ - int temp; + hid_t fid; /* File ID for file opened */ + long nops = 0; /* # of times to grow or shrink the dataset */ + long flush_count = 1000; /* # of records to write between flushing file */ + unsigned verbose = 1; /* Whether to emit some informational messages */ + unsigned use_seed = 0; /* Set to 1 if a seed was set on the command line */ + unsigned random_seed = 0; /* Random # seed */ + unsigned u; /* Local index variable */ + int temp; /* Parse command line options */ - if(argc < 2) + if (argc < 2) usage(); - if(argc > 1) { + if (argc > 1) { u = 1; - while(u < (unsigned)argc) { - if(argv[u][0] == '-') { - switch(argv[u][1]) { + while (u < (unsigned)argc) { + if (argv[u][0] == '-') { + switch (argv[u][1]) { /* # of records to write between flushing file */ case 'f': flush_count = HDatol(argv[u + 1]); - if(flush_count < 0) + if (flush_count < 0) usage(); u += 2; break; @@ -321,8 +319,8 @@ int main(int argc, const char *argv[]) /* Random # seed */ case 'r': use_seed = 1; - temp = HDatoi(argv[u + 1]); - if(temp < 0) + temp = HDatoi(argv[u + 1]); + if (temp < 0) usage(); else random_seed = (unsigned)temp; @@ -333,31 +331,31 @@ int main(int argc, const char *argv[]) usage(); break; } /* end switch */ - } /* end if */ + } /* end if */ else { /* Get the number of records to append */ nops = HDatol(argv[u]); - if(nops <= 0) + if (nops <= 0) usage(); u++; } /* end else */ - } /* end while */ - } /* end if */ - if(nops <= 0) + } /* end while */ + } /* end if */ + if (nops <= 0) usage(); - if(flush_count >= nops) + if (flush_count >= nops) usage(); /* Emit informational message */ - if(verbose) { + if (verbose) { HDfprintf(stderr, "Parameters:\n"); HDfprintf(stderr, "\t# of operations between flushes = %ld\n", flush_count); HDfprintf(stderr, "\t# of operations = %ld\n", nops); } /* end if */ /* Set the random seed */ - if(0 == use_seed) { + if (0 == use_seed) { struct timeval t; HDgettimeofday(&t, NULL); random_seed = (unsigned)(t.tv_usec); @@ -367,19 +365,19 @@ int main(int argc, const char *argv[]) HDfprintf(stderr, "Using writer random seed: %u\n", random_seed); /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Generating symbol names\n"); /* Generate dataset names */ - if(generate_symbols() < 0) + if (generate_symbols() < 0) return -1; /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Opening skeleton file: %s\n", FILENAME); /* Open file skeleton */ - if((fid = open_skeleton(FILENAME, verbose)) < 0) { + if ((fid = open_skeleton(FILENAME, verbose)) < 0) { HDfprintf(stderr, "Error opening skeleton file!\n"); HDexit(EXIT_FAILURE); } /* end if */ @@ -388,31 +386,31 @@ int main(int argc, const char *argv[]) h5_send_message(WRITER_MESSAGE, NULL, NULL); /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Adding and removing records\n"); /* Grow and shrink datasets */ - if(addrem_records(fid, verbose, (unsigned long)nops, (unsigned long)flush_count) < 0) { + if (addrem_records(fid, verbose, (unsigned long)nops, (unsigned long)flush_count) < 0) { HDfprintf(stderr, "Error adding and removing records from datasets!\n"); HDexit(EXIT_FAILURE); } /* end if */ /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Releasing symbols\n"); /* Clean up the symbols */ - if(shutdown_symbols() < 0) { + if (shutdown_symbols() < 0) { HDfprintf(stderr, "Error releasing symbols!\n"); HDexit(EXIT_FAILURE); } /* end if */ /* Emit informational message */ - if(verbose) + if (verbose) HDfprintf(stderr, "Closing objects\n"); /* Close objects opened */ - if(H5Fclose(fid) < 0) { + if (H5Fclose(fid) < 0) { HDfprintf(stderr, "Error closing file!\n"); HDexit(EXIT_FAILURE); } /* end if */ |