def load_optics(self):
from madgui.widget.filedialog import getOpenFileName
filename = getOpenFileName(self, 'Open file', self.folder,
self.filename_filters)
if filename:
with open(filename) as f:
self.opticsEdit.setPlainText(f.read())
def onImport(self):
"""Import data from JSON/YAML file."""
from madgui.widget.filedialog import getOpenFileName
filename = getOpenFileName(
self.window(), 'Import values', self.folder,
self.exporter.importFilters)
if filename:
self.exporter.importFrom(filename)
self.folder, _ = os.path.split(filename)
def on_btn_load(self):
filename = getOpenFileName(
self.window(), 'Open data file for comparison',
self.folder, self.dataFileFilters)
if filename:
self.folder, basename = os.path.split(filename)
data = self.load_file(filename)
style = self.scene.config['reference_style']
self.scene.add_curve(basename, data, style)
def onImport(self):
"""Import data from JSON/YAML file."""
from madgui.widget.filedialog import getOpenFileName
filename = getOpenFileName(
self.window(), 'Import values', self.folder,
self.exporter.importFilters)
if filename:
self.exporter.importFrom(filename)
self.folder, _ = os.path.split(filename)
def _import(self, title, filters, callback, data_key):
from madgui.widget.filedialog import getOpenFileName
folder = self.str_folder or self.folder
filename = getOpenFileName(self, title, folder, filters)
if filename:
from madgui.widget.params import import_params
data = import_params(filename)
callback(data)
self.str_folder = os.path.dirname(filename)
def fileOpen(self):
from madgui.widget.filedialog import getOpenFileName
filters = [
("Model files", "*.cpymad.yml"),
("MAD-X files", "*.madx", "*.str", "*.seq"),
("All files", "*"),
]
filename = getOpenFileName(
self, 'Open file', self.folder, filters)
if filename:
self.session.load_model(filename)
def execFile(self):
from madgui.widget.filedialog import getOpenFileName
filters = [
("All MAD-X files", "*.madx", "*.str", "*.seq"),
("Strength files", "*.str"),
("All files", "*"),
]
folder = self.exec_folder or self.folder
filename = getOpenFileName(
self, 'Open MAD-X file', folder, filters)
if filename:
self.model().call(filename)
self.exec_folder = os.path.dirname(filename)
def _open_float_values(self):
from madgui.widget.filedialog import getOpenFileName
filters = [
("STR files", "*.str"),
("All files", "*"),
]
folder = self.window.str_folder or self.window.folder
if self.str_file:
folder = os.path.dirname(self.str_file)
filename = getOpenFileName(self.window, 'Open strength export', folder,
filters)
if filename:
self.load_float_values(filename)
def _open_sd_values(self):
from madgui.widget.filedialog import getOpenFileName
filters = [
("YAML files", "*.yml", "*.yaml"),
("All files", "*"),
]
folder = self.window.str_folder or self.window.folder
if self.sd_file:
folder = os.path.dirname(self.sd_file)
filename = getOpenFileName(self.window, 'Open monitor export', folder,
filters)
if filename:
self.load_sd_values(filename)
def load_orm(self):
from madgui.widget.filedialog import getOpenFileName
filename = getOpenFileName(self.window(), 'Load Orbit Responses',
self.folder, self.exportFilters)
if filename:
self.load_from(filename)
def load(self):
from madgui.widget.filedialog import getOpenFileName
filename = getOpenFileName(self.window(), 'Load offsets', self.folder,
self.exportFilters)
if filename:
self.load_from(filename)