def setUp(self):
self.integration_widget = IntegrationWidget()
self.move_widget = self.integration_widget.move_widget
self.setup_widget = self.integration_widget.move_widget.motors_setup_widget
self.img_model = ImgModel()
self.epics_controller = EpicsController(self.integration_widget,
self.img_model)
def setUp(self):
# creating Data objects
self.img_data = ImgModel()
self.img_data.load("Data/CbnCorrectionOptimization/Mg2SiO4_091.tif")
self.calibration_data = CalibrationModel(self.img_data)
self.calibration_data.load("Data/CbnCorrectionOptimization/LaB6_40keV side.poni")
self.mask_data = MaskModel()
self.mask_data.load_mask("Data/CbnCorrectionOptimization/Mg2SiO4_91_combined.mask")
# creating the ObliqueAngleDetectorAbsorptionCorrection
_, fit2d_parameter = self.calibration_data.get_calibration_parameter()
detector_tilt = fit2d_parameter['tilt']
detector_tilt_rotation = fit2d_parameter['tiltPlanRotation']
self.tth_array = self.calibration_data.spectrum_geometry.twoThetaArray((2048, 2048))
self.azi_array = self.calibration_data.spectrum_geometry.chiArray((2048, 2048))
self.oiadac_correction = ObliqueAngleDetectorAbsorptionCorrection(
self.tth_array, self.azi_array,
detector_thickness=40,
absorption_length=465.5,
tilt=detector_tilt,
rotation=detector_tilt_rotation,
)
self.img_data.add_img_correction(self.oiadac_correction, "oiadac")
def setUp(self):
self.working_dir = {}
self.img_model = ImgModel()
self.mask_model = MaskModel()
self.mask_widget = MaskWidget()
self.mask_controller = MaskController(self.working_dir,
self.mask_widget, self.img_model,
self.mask_model)
def setUp(self):
self.working_dir = {'image': ''}
self.widget = IntegrationWidget()
self.image_model = ImgModel()
self.mask_model = MaskModel()
self.spectrum_model = PatternModel()
self.calibration_model = CalibrationModel(self.image_model)
self.controller = ImageController(
working_dir=self.working_dir,
widget=self.widget,
img_model=self.image_model,
mask_model=self.mask_model,
spectrum_model=self.spectrum_model,
calibration_model=self.calibration_model)
def setUp(self):
self.img_model = ImgModel()
self.mask_model = MaskModel()
self.calibration_model = CalibrationModel(self.img_model)
self.calibration_model._calibrants_working_dir = os.path.join(
data_path, 'calibrants')
self.calibration_model.integrate_1d = MagicMock()
self.calibration_model.integrate_2d = MagicMock()
self.calibration_widget = CalibrationWidget()
self.working_dir = {}
self.calibration_controller = CalibrationController(
working_dir=self.working_dir,
img_model=self.img_model,
mask_model=self.mask_model,
widget=self.calibration_widget,
calibration_model=self.calibration_model)
def setUp(self):
self.image_model = ImgModel()
self.calibration_model = CalibrationModel()
self.calibration_model.is_calibrated = True
self.calibration_model.spectrum_geometry.wavelength = 0.31E-10
self.calibration_model.integrate_1d = MagicMock(return_value=(self.calibration_model.tth,
self.calibration_model.int))
self.spectrum_model = PatternModel()
self.phase_model = PhaseModel()
self.widget = IntegrationWidget()
self.widget.pattern_widget._auto_range = True
self.phase_tw = self.widget.phase_tw
self.spectrum_controller = PatternController({}, self.widget, self.image_model, None,
self.calibration_model, self.spectrum_model)
self.controller = PhaseController({}, self.widget, self.calibration_model, self.spectrum_model,
self.phase_model)
self.spectrum_controller.load(os.path.join(data_path, 'spectrum_001.xy'))
def __init__(self, use_settings=True):
self.use_settings = use_settings
self.widget = MainWidget()
# create data
self.img_model = ImgModel()
self.calibration_model = CalibrationModel(self.img_model)
self.mask_model = MaskModel()
self.spectrum_model = PatternModel()
self.phase_model = PhaseModel()
self.settings_directory = os.path.join(os.path.expanduser("~"),
'.Dioptas')
self.working_directories = {
'calibration': '',
'mask': '',
'image': '',
'spectrum': '',
'overlay': '',
'phase': ''
}
if use_settings:
self.load_settings()
self.calibration_controller = CalibrationController(
self.working_directories, self.widget.calibration_widget,
self.img_model, self.mask_model, self.calibration_model)
self.mask_controller = MaskController(self.working_directories,
self.widget.mask_widget,
self.img_model, self.mask_model)
self.integration_controller = IntegrationController(
self.working_directories, self.widget.integration_widget,
self.img_model, self.mask_model, self.calibration_model,
self.spectrum_model, self.phase_model)
self.create_signals()
self.update_title()
self.current_tab_index = 0
def setUp(self):
self.img_model = ImgModel()
self.calibration_model = CalibrationModel(self.img_model)
def setUp(self):
self.img_model = ImgModel()
self.img_model.load(os.path.join(data_path, 'image_001.tif'))
