Add tests/stream_roundtrip

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