Skip to content
Snippets Groups Projects
Commit 4683ca2f authored by David Hammer's avatar David Hammer
Browse files

Update JungfrauCorrection to use unsafe_get

parent 2a8fe93e
No related branches found
No related tags found
1 merge request!12Snapshot: field test deployed version as of end of run 202201
...@@ -307,9 +307,9 @@ class JungfrauCorrection(BaseCorrection): ...@@ -307,9 +307,9 @@ class JungfrauCorrection(BaseCorrection):
@property @property
def input_data_shape(self): def input_data_shape(self):
return ( return (
self._schema_cache["dataFormat.memoryCells"], self.unsafe_get("dataFormat.memoryCells"),
self._schema_cache["dataFormat.pixelsY"], self.unsafe_get("dataFormat.pixelsY"),
self._schema_cache["dataFormat.pixelsX"], self.unsafe_get("dataFormat.pixelsX"),
) )
def __init__(self, config): def __init__(self, config):
...@@ -352,7 +352,7 @@ class JungfrauCorrection(BaseCorrection): ...@@ -352,7 +352,7 @@ class JungfrauCorrection(BaseCorrection):
buffer_handle, buffer_array = self._shmem_buffer.next_slot() buffer_handle, buffer_array = self._shmem_buffer.next_slot()
self.kernel_runner.correct(self._correction_flag_enabled) self.kernel_runner.correct(self._correction_flag_enabled)
self.kernel_runner.reshape( self.kernel_runner.reshape(
output_order=self._schema_cache["dataFormat.outputAxisOrder"], output_order=self.unsafe_get("dataFormat.outputAxisOrder"),
out=buffer_array, out=buffer_array,
) )
...@@ -364,8 +364,8 @@ class JungfrauCorrection(BaseCorrection): ...@@ -364,8 +364,8 @@ class JungfrauCorrection(BaseCorrection):
preview_cell, preview_cell,
preview_pulse, preview_pulse,
) = utils.pick_frame_index( ) = utils.pick_frame_index(
self._schema_cache["preview.selectionMode"], self.unsafe_get("preview.selectionMode"),
self._schema_cache["preview.index"], self.unsafe_get("preview.index"),
cell_table, cell_table,
_pretend_pulse_table, _pretend_pulse_table,
warn_func=self.log_status_warn, warn_func=self.log_status_warn,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment