Charles Mita
482ab573af
This helps prevent potential segfaults due to null-pointer dereferencing in the face of errors when the plugin is accessed via multiple threads. If multiple threads are pushing errors onto the error stack then its possible for elements of the various string pointer arrays to be left NULL, causing a segfault when the stack is printed. Whilst this does nothing to address the "correctness" of the plugin in a multi-threaded context, at least it shouldn't crash. Error messages may still be corrupted.
108 lines
2.3 KiB
C
108 lines
2.3 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <hdf5.h>
|
|
#include "file.h"
|
|
#include "err.h"
|
|
|
|
|
|
void apply_mask(int *data, int *mask, int size) {
|
|
int *dptr, *mptr;
|
|
dptr = data;
|
|
mptr = mask;
|
|
while (dptr < data + size && mptr < mask + size) {
|
|
if (*mptr & 0x01) *dptr = -1;
|
|
if (*mptr & 0xFE) *dptr = -2;
|
|
dptr++;
|
|
mptr++;
|
|
}
|
|
}
|
|
|
|
|
|
int parse_args(int argc, char **argv, char **file_name, int *frame_idx) {
|
|
int retval = 0;
|
|
if (argc == 2) {
|
|
*frame_idx = 0;
|
|
*file_name = argv[1];
|
|
} else if (argc >= 2) {
|
|
*file_name = argv[1];
|
|
*frame_idx = atoi(argv[2]);
|
|
} else {
|
|
ERROR_JUMP(-1, done, "Require filename argument");
|
|
}
|
|
done:
|
|
return retval;
|
|
}
|
|
|
|
|
|
int main(int argc, char **argv) {
|
|
int err = 0;
|
|
int retval = 0;
|
|
char *test_file = "";
|
|
struct data_description_t desc = {0};
|
|
struct dataset_properties_t prop = {0};
|
|
int dims[3] = {0};
|
|
hid_t fid = 0;
|
|
int frame_idx = 0;
|
|
int *mask = NULL;
|
|
int *data = NULL;
|
|
|
|
init_error_handling();
|
|
if (init_h5_error_handling() < 0) {
|
|
ERROR_JUMP(-1, done, "");
|
|
}
|
|
|
|
if (parse_args(argc, argv, &test_file, &frame_idx) < 0) {
|
|
ERROR_JUMP(-1, done, "Failure parsing arguments");
|
|
}
|
|
|
|
fid = H5Fopen(test_file, H5F_ACC_RDONLY, H5P_DEFAULT);
|
|
if (fid < 0) ERROR_JUMP(-1, done, "Error opening file");
|
|
|
|
err = extract_detector_info(fid, &desc, &prop);
|
|
if (err < 0) {
|
|
ERROR_JUMP(err, done, "");
|
|
}
|
|
dims[0] = prop.dims[0];
|
|
dims[1] = prop.dims[1];
|
|
dims[2] = prop.dims[2];
|
|
|
|
printf("Dims: %d, %d, %d\n", dims[0], dims[1], dims[2]);
|
|
|
|
mask = malloc(dims[1] * dims[2] * sizeof(*mask));
|
|
if (!mask) {
|
|
ERROR_JUMP(err, done, "Failed to allocate space for pixel mask");
|
|
}
|
|
err = desc.get_pixel_mask(&desc, mask);
|
|
if (err < 0) {
|
|
ERROR_JUMP(err, done, "");
|
|
}
|
|
|
|
data = malloc(dims[1] * dims[2] * sizeof(*data));
|
|
err = desc.get_data_frame(&desc, &prop, frame_idx, sizeof(*data), data);
|
|
if (err < 0) {
|
|
ERROR_JUMP(err, done, "");
|
|
}
|
|
|
|
apply_mask(data, mask, dims[1] * dims[2]);
|
|
{
|
|
int i, j;
|
|
int max_i = 30;
|
|
int max_j = 10;
|
|
max_j = max_j < dims[1] ? max_j : dims[1];
|
|
max_i = max_i < dims[2] ? max_i : dims[2];
|
|
for (j = 0; j < max_j; j++) {
|
|
for (i = 0; i < max_i; i++) {
|
|
printf("%3d ", data[i + j*dims[2]]);
|
|
}
|
|
printf("\n");
|
|
}
|
|
}
|
|
|
|
done:
|
|
if (fid > 0) H5Fclose(fid);
|
|
if (data) free(data);
|
|
if (mask) free(mask);
|
|
if (retval != 0) dump_error_stack(stderr);
|
|
return retval;
|
|
}
|