From c6b9f65b368b80b8957cd7729101218e88eb36cb Mon Sep 17 00:00:00 2001
From: Valerio Mariani <valerio.mariani@desy.de>
Date: Tue, 23 Apr 2019 11:31:42 +0200
Subject: [PATCH] Added a function to directly apply crystfel geometry to data
---
cfelpyutils/geometry_utils.py | 33 +++++++++++++++++++++++++++++++++
1 file changed, 33 insertions(+)
diff --git a/cfelpyutils/geometry_utils.py b/cfelpyutils/geometry_utils.py
index 2c21743..e3c7348 100644
--- a/cfelpyutils/geometry_utils.py
+++ b/cfelpyutils/geometry_utils.py
@@ -225,3 +225,36 @@ def compute_visualization_pix_maps(geometry):
) + min_shape[0] // 2 - 1
return PixelMaps(new_x_map, new_y_map, None, None, None)
+
+
+def apply_geometry_to_data(data, geometry):
+ """
+ Apply geometry to data.
+
+ Returns a new numpy array containing the input data with the geometry applied.
+ The array is ready to be displayed using a library like matplotlib or pyqtgraph.
+
+ Args:
+
+ data (ndarray): a numpy array storing the data to which geometry should be
+ applied.
+
+ geometry (Dict): A CrystFEL geometry object (A dictionary
+ returned by the
+ :obj:`cfelpyutils.crystfel_utils.load_crystfel_geometry`
+ function).
+
+ Returns:
+
+ ndarray: a new array containing the input data to which the geometry has been
+ applied. The returned array shape is big enough to display all the input data
+ in the reference system of the physical detector. The array is also supposed to
+ be centered at the center of the reference system of the detector (i.e: the
+ beam interaction point).
+ """
+ pixel_maps = compute_pix_maps(geometry)
+ min_array_shape = compute_min_array_size(pixel_maps)
+ visualization_array = numpy.zeros(min_array_shape, dtype=float)
+ visual_pixel_maps = compute_visualization_pix_maps(geometry)
+ visualization_array[visual_pixel_maps.y.flatten(), visual_pixel_maps.x.flatten()] = data.ravel().astype(visualization_array.dtype)
+ return visualization_array
\ No newline at end of file
--
GitLab