def __init__(self, corrector): super().__init__() load_ui(self, __package__, self.ui_file) self.corrector = Corrector(corrector.session, False) self.corrector.setup({ 'monitors': corrector.monitors, 'optics': corrector.variables, }) self.corrector.start() self.bot = ProcBot(self, self.corrector) elem_by_knob = {} for elem in corrector.model.elements: for knob in corrector.model.get_elem_knobs(elem): elem_by_knob.setdefault(knob.lower(), elem) self.steerers = [ elem_by_knob[v.lower()] for v in corrector.variables ] self.corrector.add_record = self.add_record self.raw_records = [] self.init_controls() self.set_initial_values() self.connect_signals()
def __init__(self, session): super().__init__() load_ui(self, __package__, self.ui_file) self.corrector = Corrector(session, direct=not self.multi_step) self.corrector.start() self.init_controls() self.set_initial_values() self.connect_signals()
def __init__(self, scene): super().__init__() self.scene = scene self.available = scene.user_tables self.plotted = scene.scene_graph.node('user_curves') self.folder = scene.model.path load_ui(self, __package__, self.ui_file) self.init_controls() self.connect_signals()
def __init__(self, session, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, self.ui_file) session.model_args = self.model_args self.session = session self.config = session.config self.control = session.control self.model = session.model self.user_ns = session.user_ns self.exec_folder = self.config.exec_folder self.str_folder = self.config.str_folder self.model.changed2.connect(self._on_model_changed) self.initUI() logging.info('Welcome to madgui. Type <Ctrl>+O to open a file.')
def __init__(self, parent, monitors): super().__init__() load_ui(self, __package__, self.ui_file) self.opticsEdit.setFont(monospace()) self.control = parent.control self.model = parent.model self.monitors = monitors self._parent = parent self.fit_results = List() last_monitor = max(map(self.model.elements.index, monitors)) quads = [ el.name for el in self.model.elements if el.base_name.lower() == 'quadrupole' and el.index < last_monitor ] self.quad_knobs = { name.lower(): knob for name in quads for knob in self.model._get_knobs(self.model.elements[name], 'k1') } self.quadsListWidget.addItems(list(quads)) self.quadsListWidget.setCurrentItem( self.quadsListWidget.item(len(quads) - 1), QItemSelectionModel.SelectCurrent) self.quadsListWidget.setCurrentItem( self.quadsListWidget.item(len(quads) - 2), QItemSelectionModel.Select) self.startButton = self.dialogButtonBox.button(Button.Ok) self.btn_abort = self.dialogButtonBox.button(Button.Abort) self.btn_close = self.dialogButtonBox.button(Button.Close) self.btn_reset = self.dialogButtonBox.button(Button.Reset) self.applyButton = self.dialogButtonBox.button(Button.Apply) self.startButton.clicked.connect(self.start) self.btn_abort.clicked.connect(self.cancel) self.btn_close.clicked.connect(self._close) self.btn_reset.clicked.connect(self.reset) self.applyButton.clicked.connect(self.apply) self.loadButton = self.saveButtonBox.button(Button.Open) self.saveButton = self.saveButtonBox.button(Button.Save) self.loadButton.clicked.connect(self.load_optics) self.saveButton.clicked.connect(self.save_optics) self.focusButton.clicked.connect(self.read_focus) self.opticsEdit.textChanged.connect(self.update_ui) self.quadsListWidget.itemSelectionChanged.connect(self.update_ui) self.resultsTable.set_viewmodel(self.get_result_row, self.fit_results, unit=True) self.update_filename() self.fileButton.clicked.connect(self.change_output_file) self.update_ui()
def __init__(self, session): super().__init__() load_ui(self, __package__, self.ui_file) self.control = session.control self.model = session.model() self.corrector = Corrector(session, False) self.corrector.setup({ 'monitors': [], }) self.corrector.start() self.bot = ProcBot(self, self.corrector) self.init_controls() self.set_initial_values() self.connect_signals()
def __init__(self, matcher): super().__init__() load_ui(self, __package__, self.ui_file) self.matcher = matcher self.model = model = matcher.model local_constraints = ['envx', 'envy'] + [ cmdpar.name for cmdpar in model.madx.command.constraint.cmdpar.values() if cmdpar.dtype == PARAM_TYPE_CONSTRAINT ] self.elem_enum = make_enum('Elem', model.elements.names) self.lcon_enum = make_enum('Local', local_constraints, strict=False) self.knob_enum = make_enum('Knobs', matcher.knobs, strict=False) self.init_controls() self.set_initial_values() self.connect_signals()
def __init__(self, matcher, control): super().__init__() load_ui(self, __package__, self.ui_file) self.matcher = matcher self.control = control self.model = model = matcher.model local_constraints = ['envx', 'envy'] + [ cmdpar.name for cmdpar in model.madx.command.constraint.cmdpar.values() if cmdpar.dtype == PARAM_TYPE_CONSTRAINT ] self.elem_enum = make_enum('Elem', model.elements.names) self.lcon_enum = make_enum('Local', local_constraints, strict=False) self.knob_enum = make_enum('Knobs', matcher.knobs, strict=False) self.init_controls() self.set_initial_values() self.connect_signals()
def __init__(self, session): super().__init__(session.window()) load_ui(self, __package__, self.ui_file) self.session = session self.control = session.control self.model = session.model() self.frame = session.window() self.readouts = self.control.sampler.readouts_list # TODO: we should eventually load this from model-specific session # file, but it's fine like this for now: self._monconf = session.config['online_control']['monitors'] self._offsets = session.config['online_control']['offsets'] self.monitorTable.set_viewmodel(self.get_monitor_row, self.readouts, unit=True) self.monitorTable.setSelectionBehavior(QAbstractItemView.SelectRows) self.monitorTable.setSelectionMode(QAbstractItemView.ExtendedSelection) self.buttonBox.button(Button.Ok).clicked.connect(self.accept) self.buttonBox.button(Button.Save).clicked.connect(self.export)
def __init__(self, corrector): super().__init__() load_ui(self, __package__, self.ui_file) self.corrector = Corrector(corrector.session, False) self.corrector.setup({ 'monitors': corrector.monitors, 'optics': corrector.variables, }) self.corrector.start() self.bot = ProcBot(self, self.corrector) elem_by_knob = {} for elem in corrector.model.elements: for knob in corrector.model.get_elem_knobs(elem): elem_by_knob.setdefault(knob.lower(), elem) self.steerers = [elem_by_knob[v.lower()] for v in corrector.variables] self.corrector.add_record = self.add_record self.raw_records = [] self.init_controls() self.set_initial_values() self.connect_signals()
def __init__(self, version_info, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'about.ui') self.textBrowser.setHtml(version_info.to_html()) self.setWindowTitle("About {}".format(version_info.name))
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'opticstable.ui') self.opticsTable.setSelectionBehavior(QAbstractItemView.SelectRows) self.opticsTable.setSelectionMode(QAbstractItemView.ExtendedSelection) self.readFocusButton.clicked.connect(self.read_focus)
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'fitsettings.ui')
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'autoproc.ui')
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'responsetable.ui')
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'recordstable.ui') self.recordsTable.setSelectionBehavior(QAbstractItemView.SelectRows) self.recordsTable.setSelectionMode(QAbstractItemView.ExtendedSelection)
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) load_ui(self, __package__, 'configselect.ui')