def __init__(self, frame, menubar):
"""
Add plugin to the frame.
Add a menu that can be used to connect to the online control. When
connected, the plugin can be used to access parameters in the online
database. This works only if the corresponding parameters were named
exactly as in the database and are assigned with the ":=" operator.
"""
self._frame = frame
self._plugin = None
loaders = [
loader
for loader in EntryPoint('madgui.online.PluginLoader').slots
if loader.check_avail()
]
if loaders:
submenu = self.create_menu(loaders)
menu.extend(frame, menubar, [submenu])
def _CreateMenu(self):
"""Create a menubar."""
# TODO: this needs to be done more dynamically. E.g. use resource
# files and/or a plugin system to add/enable/disable menu elements.
MenuItem = menu.Item
Menu = menu.Menu
Separator = menu.Separator
menubar = self.menubar = wx.MenuBar()
menu.extend(self, menubar, [
Menu('&Session', [
MenuItem('&New session window\tCtrl+N',
'Open a new session window',
self.OnNewWindow),
MenuItem('&Python shell\tCtrl+P',
'Open a tab with a python shell',
self._NewCommandTab),
Separator,
MenuItem('&Open model\tCtrl+O',
'Load model or open new model from a MAD-X file.',
self._LoadFile),
MenuItem('&Save model as..\tCtrl+S',
'Save the current model (beam + twiss) to a file',
self._SaveModel,
self._CanSaveModel),
Separator,
MenuItem('&Reset session',
'Clear the MAD-X session state.',
lambda _: self._ResetSession()),
Separator,
MenuItem('&Close',
'Close window',
self.OnQuit,
id=wx.ID_CLOSE),
]),
Menu('&View', [
MenuItem('&Envelope',
'Open new tab with beam envelopes.',
lambda _: TwissView.create(self.session,
self, basename='env')),
MenuItem('&Position',
'Open new tab with beam position.',
lambda _: TwissView.create(self.session,
self, basename='pos')),
]),
Menu('&Manage', [
MenuItem('&Initial conditions',
'Add/remove/edit TWISS initial conditions.',
self._EditTwiss),
MenuItem('&Beam',
'Set beam.',
self._SetBeam),
Separator,
MenuItem('Show &element indicators',
'Show indicators for beam line elements.',
self._ShowIndicators,
self._UpdateShowIndicators,
wx.ITEM_CHECK),
]),
Menu('&Help', [
MenuItem('&About',
'Show about dialog.',
lambda _: show_about_dialog(self)),
]),
])
# Create menu items
self.hook.menu(self, menubar)
self.Bind(wx.EVT_UPDATE_UI, self.OnUpdateMenu, menubar)
self._IsEnabledTop = {self.ViewMenuIndex: True,
self.TabMenuIndex: True}
return menubar
