PROTON_CHARGE_UNCERTAINTY = 0.005
[docs]
class Roi:
[docs]
def __init__(self, x1=0, y1=0, x2=None, y2=None, width=None, height=None):
"""x1 and x2 are mandatory
y2 or height are mandatory
x2 or width are mandatory"""
if x1 < 0 or y1 < 0:
raise ValueError("ROI top left corner coordinates must be non-negative")
if x2 is not None and width is not None:
raise ValueError("ROI cannot have both x2 and width defined")
if y2 is not None and height is not None:
raise ValueError("ROI cannot have both y2 and height defined")
if x2 is not None:
if x2 < 0:
raise ValueError("ROI x2 must be non-negative")
if x2 < x1:
raise ValueError("ROI x2 must be greater than or equal to x1")
if width is not None:
raise ValueError("ROI cannot have both x2 and width defined")
else:
if width is None or width <= 0:
raise ValueError("ROI width must be positive when x2 is not defined")
x2 = x1 + width
if y2 is not None:
if y2 < 0:
raise ValueError("ROI y2 must be non-negative")
if y2 < y1:
raise ValueError("ROI y2 must be greater than or equal to y1")
if height is not None:
raise ValueError("ROI cannot have both y2 and height defined")
else:
if height is None or height <= 0:
raise ValueError("ROI height must be positive when y2 is not defined")
y2 = y1 + height
self.x1 = x1
self.y1 = y1
self.x2 = x2
self.y2 = y2
[docs]
def get_roi(self):
return (self.x1, self.y1, self.x2, self.y2)
[docs]
def set_roi(self, x1, y1, x2, y2):
if x1 < 0 or y1 < 0 or x2 < 0 or y2 < 0:
raise ValueError("ROI coordinates must be non-negative")
self.x1 = x1
self.y1 = y1
self.x2 = x2
self.y2 = y2
def __repr__(self):
return f"Roi(x1={self.x1}, y1={self.y1}, x2={self.x2}, y2={self.y2})"
def __str__(self):
return f"Roi: (x1={self.x1}, y1={self.y1}) to (x2={self.x2}, y2={self.y2})"
[docs]
class DataType:
sample = "sample"
ob = "ob"
normalization = "normalization"
[docs]
class MasterDictKeys:
data_type = "data_type"
list_folders = "list_folders"
obs_data_combined = "obs_data_combined"
sample_data = "sample_data"
uncertainties_obs_data_combined = "uncertainties_obs_data_combined"
uncertainties_sample_data = "uncertainties_sample_data"
frame_number = "frame_number"
proton_charge = "proton_charge"
matching_ob = "matching_ob"
list_images = "list_images"
data = "data"
nexus_path = "nexus_path"
data_path = "data_path"
shutter_counts = "shutter_counts"
list_spectra = "list_spectra"
list_shutters = "list_shutters"
ext = "ext"
[docs]
class Facility:
ornl = "ornl"
lanl = "lanl"
[docs]
class NormalizationStatus:
all_nexus_file_found = False
all_spectra_file_found = False
all_shutter_counts_file_found = False
all_proton_charge_value_found = False
all_list_shutter_values_for_each_image_found = False