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, 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__, 'opticstable.ui')
self.opticsTable.setSelectionBehavior(QAbstractItemView.SelectRows)
self.opticsTable.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.readFocusButton.clicked.connect(self.read_focus)
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__, 'autoproc.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')
