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 __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.mask_model = MaskModel() self.img = np.zeros((10, 10)) self.mask_model.set_dimension(self.img.shape)