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.phase_model = PhaseModel()