def test_read_write_xfel_file(tmpdir):
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS
)
path = str(tmpdir / 'dssc_geom.h5')
geom.to_h5_file_and_quad_positions(path)
assert_isfile(path)
loaded = DSSC_1MGeometry.from_h5_file(path)
assert_geom_close(loaded, geom)
def test_inspect():
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS
)
# Smoketest
ax = geom.inspect()
assert isinstance(ax, Axes)
def test_quad_positions_with_file():
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS
)
quad_pos_out = geom.quad_positions(sample_xfel_geom)
np.testing.assert_allclose(quad_pos_out, QUAD_POS)
def test_quad_positions_no_file():
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS)
# Smoketest - the results without passing a file in aren't usable yet
quad_pos_out = geom.quad_positions()
assert quad_pos_out.shape == (4, 2)
def test_snap_assemble_data():
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS)
stacked_data = np.zeros((16, 128, 512))
img, centre = geom.position_modules_fast(stacked_data)
assert img.shape == (1281, 1099)
assert tuple(centre) == (656, 552)
assert np.isnan(img[0, 0])
assert img[50, 50] == 0
def test_to_distortion_array():
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS)
# Smoketest
distortion = geom.to_distortion_array()
assert isinstance(distortion, np.ndarray)
assert distortion.shape == (2048, 512, 6, 3)
# Coordinates in m, origin at corner; max x & y should be ~ 25cm
assert 0.20 < distortion[..., 1].max() < 0.35
assert 0.20 < distortion[..., 2].max() < 0.35
assert 0.0 <= distortion[..., 1].min() < 0.01
assert 0.0 <= distortion[..., 2].min() < 0.01
def test_get_pixel_positions():
geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
sample_xfel_geom, QUAD_POS)
pixelpos = geom.get_pixel_positions()
assert pixelpos.shape == (16, 128, 512, 3)
px = pixelpos[..., 0]
py = pixelpos[..., 1]
assert -0.15 < px.min() < -0.1
assert 0.15 > px.max() > 0.1
assert -0.2 < py.min() < -0.12
assert 0.2 > py.max() > 0.12
# Odd-numbered rows in Q1 & Q2 should have 0.5 pixel higher x than the even.
np.testing.assert_allclose(px[0, 1::2, 0] - px[0, 0::2, 0], 236e-6 / 2)
def _load_geometry(self, filename, quad_positions):
"""Override."""
if self._assembler_type == GeomAssembler.OWN \
or self._stack_only:
from ...geometries import DSSC_1MGeometryFast
if self._stack_only:
self._geom = DSSC_1MGeometryFast()
else:
self._geom = DSSC_1MGeometryFast.from_h5_file_and_quad_positions(
filename, quad_positions)
else:
from extra_geom import DSSC_1MGeometry
try:
self._geom = DSSC_1MGeometry.from_h5_file_and_quad_positions(
filename, quad_positions)
except OSError as e:
raise AssemblingError(e)
def from_h5_file_and_quad_positions(cls, geom_file, quad_pos=None):
"""Create geometry from geometry file or quad positions."""
quad_pos = quad_pos or Defaults.fallback_quad_pos[cls.detector_name]
exgeom_obj = DSSC_1MGeometry.from_h5_file_and_quad_positions(
geom_file, quad_pos)
return cls(exgeom_obj, geom_file)
def test_quad_positions_no_file():
geom = DSSC_1MGeometry.from_quad_positions(QUAD_POS)
quad_pos_out = geom.quad_positions()
np.testing.assert_allclose(quad_pos_out, QUAD_POS)