added a first version of the time tool processing script
This commit is contained in:
86
procprof/spectrometer.py
Normal file
86
procprof/spectrometer.py
Normal file
@ -0,0 +1,86 @@
|
||||
import sys
|
||||
import json
|
||||
|
||||
|
||||
def process_image(image, pulse_id, timestamp, x_axis, y_axis, parameters, bsdata=None):
|
||||
image = image.astype(int)
|
||||
|
||||
camera_name = parameters["camera_name"]
|
||||
|
||||
background = parameters.get("background_data")
|
||||
background_name = parameters.get("image_background")
|
||||
background_mode = parameters.get("image_background_enable")
|
||||
|
||||
roi_signal = parameters.get("roi_signal")
|
||||
roi_background = parameters.get("roi_background")
|
||||
project_axis = parameters.get("project_axis", 0)
|
||||
threshold = parameters.get("threshold")
|
||||
|
||||
|
||||
# maintain the structure of processing_parameters
|
||||
background_shape = None
|
||||
|
||||
# maintain the structure of res
|
||||
projection_signal = projection_background = None
|
||||
|
||||
|
||||
try:
|
||||
|
||||
if background is not None:
|
||||
background_shape = background.shape
|
||||
image -= background
|
||||
|
||||
if threshold is not None:
|
||||
image -= threshold
|
||||
image[image < 0] = 0
|
||||
|
||||
if roi_signal is not None:
|
||||
projection_signal = get_roi_projection(image, roi_signal, project_axis)
|
||||
|
||||
if roi_background is not None:
|
||||
projection_background = get_roi_projection(image, roi_background, project_axis)
|
||||
|
||||
except Exception as e:
|
||||
lineno = sys.exc_info()[2].tb_lineno
|
||||
tn = type(e).__name__
|
||||
status = f"Error in line number {lineno}: {tn}: {e}"
|
||||
else:
|
||||
status = "OK"
|
||||
|
||||
|
||||
processing_parameters = {
|
||||
"image_shape": image.shape,
|
||||
"background_shape": background_shape,
|
||||
|
||||
"background_name": background_name,
|
||||
"background_mode": background_mode,
|
||||
|
||||
"roi_signal": roi_signal,
|
||||
"roi_background": roi_background,
|
||||
"project_axis": project_axis,
|
||||
"threshold": threshold,
|
||||
|
||||
"status": status
|
||||
}
|
||||
|
||||
processing_parameters = json.dumps(processing_parameters)
|
||||
|
||||
res = {
|
||||
camera_name + ".processing_parameters": processing_parameters,
|
||||
|
||||
camera_name + ".projection_signal": projection_signal,
|
||||
camera_name + ".projection_background": projection_background
|
||||
}
|
||||
|
||||
return res
|
||||
|
||||
|
||||
|
||||
def get_roi_projection(image, roi, axis):
|
||||
x_start, x_stop, y_start, y_stop = roi
|
||||
cropped = image[x_start:x_stop, y_start:y_stop]
|
||||
project = cropped.mean(axis=axis)
|
||||
return project
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user