def save_optics(self):
from madgui.widget.filedialog import getSaveFileName
filename = getSaveFileName(self, 'Open file', self.folder,
self.filename_filters)
if filename:
with open(filename, 'wt') as f:
f.write(self.opticsEdit.toPlainText())
def save_orm(self):
from madgui.widget.filedialog import getSaveFileName
filename = getSaveFileName(self.window(), 'Export Orbit Responses',
self.folder, self.exportFilters)
if filename:
self.export_to(filename)
self.folder, _ = os.path.split(filename)
def onExport(self):
"""Export data to YAML file."""
from madgui.widget.filedialog import getSaveFileName
filename = getSaveFileName(
self.window(), 'Export values', self.folder,
self.exporter.exportFilters)
if filename:
self.exporter.exportTo(filename)
self.folder, _ = os.path.split(filename)
def onExport(self):
"""Export data to YAML file."""
from madgui.widget.filedialog import getSaveFileName
filename = getSaveFileName(
self.window(), 'Export values', self.folder,
self.exporter.exportFilters)
if filename:
self.exporter.exportTo(filename)
self.folder, _ = os.path.split(filename)
def change_output_file(self):
if self.running:
return
from madgui.widget.filedialog import getSaveFileName
filename = getSaveFileName(self.window(), 'Raw data file', self.folder,
[("YAML file", "*" + self.extension)])
if filename:
if not filename.endswith(self.extension):
filename += self.extension
self.set_filename(filename)
def _export(self, title, filters, fetch_data, export=None, **kw):
from madgui.widget.filedialog import getSaveFileName
folder = self.str_folder or self.folder
filename = getSaveFileName(self, title, folder, filters)
if filename:
if export is None:
from madgui.widget.params import export_params as export
data = fetch_data()
export(filename, data, **kw)
self.str_folder = os.path.dirname(filename)