aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorThomas White <taw@physics.org>2022-03-08 11:57:55 +0100
committerThomas White <taw@physics.org>2022-03-08 14:14:53 +0100
commit59e8a7ef447d0c8c56d33bfb7baf0010b5873fd7 (patch)
treec0f5551dd980b2ef56e3e56170046643ab874f62
parenta40d7a356c3a71c83b9f0556f878aea1f607c0af (diff)
Add tests/stream_roundtrip
-rw-r--r--tests/meson.build6
-rw-r--r--tests/stream_roundtrip.c140
-rw-r--r--tests/stream_roundtrip.geom29
3 files changed, 175 insertions, 0 deletions
diff --git a/tests/meson.build b/tests/meson.build
index aa425707..0109ec9d 100644
--- a/tests/meson.build
+++ b/tests/meson.build
@@ -68,6 +68,12 @@ endforeach
# Less easy unit tests for libcrystfel functions
+exe = executable('stream_roundtrip',
+ ['stream_roundtrip.c'],
+ dependencies : [libcrystfeldep])
+test('stream_roundtrip', exe,
+ args: [files('stream_roundtrip.geom')])
+
exe = executable('stream_read',
['stream_read.c'],
dependencies : [libcrystfeldep])
diff --git a/tests/stream_roundtrip.c b/tests/stream_roundtrip.c
new file mode 100644
index 00000000..29099b1a
--- /dev/null
+++ b/tests/stream_roundtrip.c
@@ -0,0 +1,140 @@
+/*
+ * stream_roundtrip.c
+ *
+ * Check that peaks and reflections can be sent on a stream round-trip
+ *
+ * Copyright © 2020-2022 Deutsches Elektronen-Synchrotron DESY,
+ * a research centre of the Helmholtz Association.
+ *
+ * Authors:
+ * 2020-2022 Thomas White <taw@physics.org>
+ *
+ * This file is part of CrystFEL.
+ *
+ * CrystFEL is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * CrystFEL is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with CrystFEL. If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+
+#include <stdio.h>
+
+#include <stream.h>
+#include <image.h>
+#include <datatemplate.h>
+
+#define N_PEAKS (10)
+
+int main(int argc, char *argv[])
+{
+ gsl_rng *rng;
+ int i;
+ float peak_fs[N_PEAKS];
+ float peak_ss[N_PEAKS];
+ float peak_i[N_PEAKS];
+ int peak_pn[N_PEAKS] = { 0, 0, 1, 1, 0, 1, 0, 1, 0, 1 };
+ struct image *image;
+ DataTemplate *dtempl;
+ Stream *st;
+ int fail = 0;
+
+ /* Create test data ................................................. */
+
+ rng = gsl_rng_alloc(gsl_rng_mt19937);
+ for ( i=0; i<N_PEAKS; i++ ) {
+ peak_fs[i] = gsl_rng_uniform(rng) * 100.0;
+ peak_ss[i] = gsl_rng_uniform(rng) * 100.0;
+ peak_i[i] = 10.0*(i+1);
+ STATUS("%f %f %i %f\n",
+ peak_fs[i], peak_ss[i], peak_pn[i], peak_i[i]);
+ }
+ gsl_rng_free(rng);
+
+ /* Write stream ..................................................... */
+
+ dtempl = data_template_new_from_file(argv[1]);
+ if ( dtempl == NULL ) {
+ ERROR("Failed to load data template\n");
+ return 1;
+ }
+
+ image = image_create_for_simulation(dtempl);
+ if ( image == NULL ) {
+ ERROR("Failed to create image\n");
+ return 1;
+ }
+
+ image->features = image_feature_list_new();
+ for ( i=0; i<N_PEAKS; i++ ) {
+ image_add_feature(image->features, peak_fs[i], peak_ss[i],
+ peak_pn[i], image, peak_i[i], NULL);
+ }
+
+ st = stream_open_for_write("stream_roundtrip.stream", dtempl);
+ if ( st == NULL ) {
+ ERROR("Failed to open stream for writing\n");
+ return 1;
+ }
+
+ stream_write_geometry_file(st, argv[1]);
+
+ if ( stream_write_chunk(st, image, STREAM_PEAKS) ) {
+ ERROR("Failed to write stream chunk\n");
+ return 1;
+ }
+
+ stream_close(st);
+ image_free(image);
+
+ /* Read stream ...................................................... */
+
+ st = stream_open_for_read("stream_roundtrip.stream");
+ if ( st == NULL ) {
+ ERROR("Failed to open stream for reading\n");
+ return 1;
+ }
+
+ image = stream_read_chunk(st, STREAM_PEAKS);
+ if ( image == NULL ) {
+ ERROR("Failed to read stream chunk\n");
+ return 1;
+ }
+
+ stream_close(st);
+
+ for ( i=0; i<image_feature_count(image->features); i++ ) {
+ struct imagefeature *f;
+ f = image_get_feature(image->features, i);
+ STATUS("%f %f %i %f\n", f->fs, f->ss, f->pn, f->intensity);
+ if ( f->pn != peak_pn[i] ) {
+ STATUS("Panel number doesn't match (should be %i)\n",
+ peak_pn[i]);
+ fail = 1;
+ }
+ if ( !within_tolerance(f->fs, peak_fs[i], 0.1) ) {
+ ERROR("fs doesn't match (should be %f)\n", peak_fs[i]);
+ fail = 1;
+ }
+ if ( !within_tolerance(f->ss, peak_ss[i], 0.1) ) {
+ ERROR("ss doesn't match (should be %f)\n", peak_ss[i]);
+ fail = 1;
+ }
+ if ( !within_tolerance(f->intensity, peak_i[i], 0.1) ) {
+ ERROR("Intensity doesn't match (should be %f)\n",
+ peak_i[i]);
+ fail = 1;
+ }
+ }
+
+ return fail;
+}
diff --git a/tests/stream_roundtrip.geom b/tests/stream_roundtrip.geom
new file mode 100644
index 00000000..79383178
--- /dev/null
+++ b/tests/stream_roundtrip.geom
@@ -0,0 +1,29 @@
+photon_energy = 10000 eV
+data = /data/data_array
+clen = 50 mm
+res = 1000000
+adu_per_photon = 1
+
+panelA/min_fs = 0
+panelA/max_fs = 99
+panelA/min_ss = 0
+panelA/max_ss = 99
+panelA/fs = x
+panelA/ss = y
+panelA/dim0 = ss
+panelA/dim1 = fs
+panelA/dim2 = 0
+panelA/corner_x = -100
+panelA/corner_y = -100
+
+panelB/min_fs = 0
+panelB/max_fs = 99
+panelB/min_ss = 0
+panelB/max_ss = 99
+panelB/fs = x
+panelB/ss = y
+panelB/dim0 = ss
+panelB/dim1 = fs
+panelB/dim2 = 1
+panelB/corner_x = -100
+panelB/corner_y = 100