def init(self, ui, parent, style):
self.is_modal = style == self.MODAL
window_style = 0
view = ui.view
if view.resizable:
window_style |= wx.RESIZE_BORDER
title = view.title
if title == "":
title = DefaultTitle
history = ui.history
window = ui.control
if window is not None:
if history is not None:
history.observe(
self._on_undoable, "undoable", remove=True, dispatch="ui"
)
history.observe(
self._on_redoable, "redoable", remove=True, dispatch="ui"
)
history.observe(
self._on_revertable, "undoable", remove=True, dispatch="ui"
)
window.SetSizer(None)
ui.reset()
else:
self.ui = ui
if style == self.MODAL:
if view.resizable:
window_style |= wx.MAXIMIZE_BOX | wx.MINIMIZE_BOX
window = wx.Dialog(
parent,
-1,
title,
style=window_style | wx.DEFAULT_DIALOG_STYLE,
)
elif style == self.NONMODAL:
if parent is not None:
window_style |= (
wx.FRAME_FLOAT_ON_PARENT | wx.FRAME_NO_TASKBAR
)
window = wx.Frame(
parent,
-1,
title,
style=window_style
| (wx.DEFAULT_FRAME_STYLE & (~wx.RESIZE_BORDER)),
)
else:
if window_style == 0:
window_style = wx.SIMPLE_BORDER
if parent is not None:
window_style |= (
wx.FRAME_FLOAT_ON_PARENT | wx.FRAME_NO_TASKBAR
)
if isinstance(parent, tuple):
window = wx.Frame(None, -1, "", style=window_style)
window._control_region = parent
else:
window = wx.Frame(parent, -1, "", style=window_style)
window._kind = ui.view.kind
self._monitor = MouseMonitor(ui)
# Set the correct default window background color:
window.SetBackgroundColour(WindowColor)
self.control = window
window.Bind(wx.EVT_CLOSE, self._on_close_page)
window.Bind(wx.EVT_CHAR, self._on_key)
self.set_icon(view.icon)
buttons = [self.coerce_button(button) for button in view.buttons]
nbuttons = len(buttons)
no_buttons = (nbuttons == 1) and self.is_button(buttons[0], "")
has_buttons = (not no_buttons) and (
(nbuttons > 0)
or view.undo
or view.revert
or view.ok
or view.cancel
)
if has_buttons or (view.menubar is not None):
if history is None:
history = UndoHistory()
else:
history = None
ui.history = history
# Create the actual trait sheet panel and imbed it in a scrollable
# window (if requested):
sw_sizer = wx.BoxSizer(wx.VERTICAL)
if ui.scrollable:
sizer = wx.BoxSizer(wx.VERTICAL)
sw = TraitsUIScrolledPanel(window)
trait_sheet = panel(ui, sw)
sizer.Add(trait_sheet, 1, wx.EXPAND)
tsdx, tsdy = trait_sheet.GetSize()
sw.SetScrollRate(16, 16)
max_dy = (2 * SystemMetrics().screen_height) // 3
sw.SetSizer(sizer)
sw.SetSize(
wx.Size(
tsdx + ((tsdy > max_dy) * scrollbar_dx), min(tsdy, max_dy)
)
)
else:
sw = panel(ui, window)
sw_sizer.Add(sw, 1, wx.EXPAND)
sw_sizer.SetMinSize(sw.GetSize())
# Check to see if we need to add any of the special function buttons:
if (not no_buttons) and (has_buttons or view.help):
sw_sizer.Add(wx.StaticLine(window, -1), 0, wx.EXPAND)
b_sizer = wx.BoxSizer(wx.HORIZONTAL)
# Convert all button flags to actual button actions if no buttons
# were specified in the 'buttons' trait:
if nbuttons == 0:
if view.undo:
self.check_button(buttons, UndoButton)
if view.revert:
self.check_button(buttons, RevertButton)
if view.ok:
self.check_button(buttons, OKButton)
if view.cancel:
self.check_button(buttons, CancelButton)
if view.help:
self.check_button(buttons, HelpButton)
# Create a button for each button action:
for raw_button, button in zip(view.buttons, buttons):
button = self.coerce_button(button)
default = raw_button == view.default_button
if self.is_button(button, "Undo"):
self.undo = self.add_button(
button, b_sizer, self._on_undo, False, default=default
)
self.redo = self.add_button(
button, b_sizer, self._on_redo, False, "Redo"
)
history.observe(
self._on_undoable, "undoable", dispatch="ui"
)
history.observe(
self._on_redoable, "redoable", dispatch="ui"
)
if history.can_undo:
self._on_undoable(True)
if history.can_redo:
self._on_redoable(True)
elif self.is_button(button, "Revert"):
self.revert = self.add_button(
button,
b_sizer,
self._on_revert,
False,
default=default,
)
history.observe(
self._on_revertable, "undoable", dispatch="ui"
)
if history.can_undo:
self._on_revertable(True)
elif self.is_button(button, "OK"):
self.ok = self.add_button(
button, b_sizer, self._on_ok, default=default
)
ui.observe(self._on_error, "errors", dispatch="ui")
elif self.is_button(button, "Cancel"):
self.add_button(
button, b_sizer, self._on_cancel, default=default
)
elif self.is_button(button, "Help"):
self.add_button(
button, b_sizer, self._on_help, default=default
)
elif not self.is_button(button, ""):
self.add_button(button, b_sizer, default=default)
sw_sizer.Add(b_sizer, 0, wx.ALIGN_RIGHT | wx.ALL, 5)
# Add the menu bar, tool bar and status bar (if any):
self.add_menubar()
self.add_toolbar()
self.add_statusbar()
# Lay all of the dialog contents out:
window.SetSizer(sw_sizer)
window.Fit()
def __init__(self, ui, parent, allow_buttons):
""" Initializes the object.
"""
self.ui = ui
history = None
view = ui.view
title = view.title
# Reset any existing history listeners:
history = ui.history
if history is not None:
history.observe(
self._on_undoable, "undoable", remove=True, dispatch="ui"
)
history.observe(
self._on_redoable, "redoable", remove=True, dispatch="ui"
)
history.observe(
self._on_revertable, "undoable", remove=True, dispatch="ui"
)
# Determine if we need any buttons or an 'undo' history:
buttons = [self.coerce_button(button) for button in view.buttons]
nbuttons = len(buttons)
if nbuttons == 0:
if view.undo:
self.check_button(buttons, UndoButton)
if view.revert:
self.check_button(buttons, RevertButton)
if view.help:
self.check_button(buttons, HelpButton)
if allow_buttons and (history is None):
for button in buttons:
if self.is_button(button, "Undo") or self.is_button(
button, "Revert"
):
history = UndoHistory()
break
ui.history = history
# Create a container panel to put everything in:
cpanel = getattr(self, "control", None)
if cpanel is not None:
cpanel.SetSizer(None)
cpanel.DestroyChildren()
else:
self.control = cpanel = TraitsUIPanel(parent, -1)
# Create the actual trait sheet panel and embed it in a scrollable
# window (if requested):
sw_sizer = wx.BoxSizer(wx.VERTICAL)
if ui.scrollable:
sizer = wx.BoxSizer(wx.VERTICAL)
sw = TraitsUIScrolledPanel(cpanel)
sizer.Add(panel(ui, sw), 1, wx.EXPAND)
sw.SetSizerAndFit(sizer)
sw.SetScrollRate(16, 16)
else:
sw = panel(ui, cpanel)
if (title != "") and (
not isinstance(getattr(parent, "owner", None), DockWindow)
):
sw_sizer.Add(
heading_text(cpanel, text=title).control, 0, wx.EXPAND
)
self.add_toolbar(sw_sizer)
sw_sizer.Add(sw, 1, wx.EXPAND)
if allow_buttons and (
(nbuttons != 1) or (not self.is_button(buttons[0], ""))
):
# Add the special function buttons:
sw_sizer.Add(wx.StaticLine(cpanel, -1), 0, wx.EXPAND)
b_sizer = wx.BoxSizer(wx.HORIZONTAL)
for button in buttons:
if self.is_button(button, "Undo"):
self.undo = self.add_button(
button, b_sizer, self._on_undo, False
)
self.redo = self.add_button(
button, b_sizer, self._on_redo, False, "Redo"
)
history.observe(
self._on_undoable, "undoable", dispatch="ui"
)
history.observe(
self._on_redoable, "redoable", dispatch="ui"
)
elif self.is_button(button, "Revert"):
self.revert = self.add_button(
button, b_sizer, self._on_revert, False
)
history.observe(
self._on_revertable, "undoable", dispatch="ui"
)
elif self.is_button(button, "Help"):
self.add_button(button, b_sizer, self._on_help)
elif not self.is_button(button, ""):
self.add_button(button, b_sizer)
sw_sizer.Add(b_sizer, 0, wx.ALIGN_RIGHT | wx.ALL, 5)
cpanel.SetSizerAndFit(sw_sizer)
def init(self, ui, parent, style):
"""Initialise the object.
FIXME: Note that we treat MODAL and POPUP as equivalent until we
have an example that demonstrates how POPUP is supposed to work.
"""
self.ui = ui
self.control = ui.control
view = ui.view
history = ui.history
if self.control is not None:
if history is not None:
history.observe(
self._on_undoable, "undoable", remove=True, dispatch="ui"
)
history.observe(
self._on_redoable, "redoable", remove=True, dispatch="ui"
)
history.observe(
self._on_revertable, "undoable", remove=True, dispatch="ui"
)
ui.reset()
else:
self.create_dialog(parent, style)
self.set_icon(view.icon)
# Convert the buttons to actions.
buttons = [self.coerce_button(button) for button in view.buttons]
nr_buttons = len(buttons)
no_buttons = (nr_buttons == 1) and self.is_button(buttons[0], "")
has_buttons = (not no_buttons) and (
(nr_buttons > 0)
or view.undo
or view.revert
or view.ok
or view.cancel
)
if has_buttons or (view.menubar is not None):
if history is None:
history = UndoHistory()
else:
history = None
ui.history = history
if (not no_buttons) and (has_buttons or view.help):
bbox = QtGui.QDialogButtonBox()
# Create the necessary special function buttons.
if nr_buttons == 0:
if view.undo:
self.check_button(buttons, UndoButton)
if view.revert:
self.check_button(buttons, RevertButton)
if view.ok:
self.check_button(buttons, OKButton)
if view.cancel:
self.check_button(buttons, CancelButton)
if view.help:
self.check_button(buttons, HelpButton)
for raw_button, button in zip(view.buttons, buttons):
default = raw_button == view.default_button
if self.is_button(button, "Undo"):
self.undo = self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.ActionRole,
self._on_undo,
False,
default=default,
)
history.observe(
self._on_undoable, "undoable", dispatch="ui"
)
if history.can_undo:
self._on_undoable(True)
self.redo = self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.ActionRole,
self._on_redo,
False,
"Redo",
)
history.observe(
self._on_redoable, "redoable", dispatch="ui"
)
if history.can_redo:
self._on_redoable(True)
elif self.is_button(button, "Revert"):
self.revert = self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.ResetRole,
self._on_revert,
False,
default=default,
)
history.observe(
self._on_revertable, "undoable", dispatch="ui"
)
if history.can_undo:
self._on_revertable(True)
elif self.is_button(button, "OK"):
self.ok = self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.AcceptRole,
self.control.accept,
default=default,
)
ui.observe(self._on_error, "errors", dispatch="ui")
elif self.is_button(button, "Cancel"):
self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.RejectRole,
self.control.reject,
default=default,
)
elif self.is_button(button, "Help"):
self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.HelpRole,
self._on_help,
default=default,
)
elif not self.is_button(button, ""):
self.add_button(
button,
bbox,
QtGui.QDialogButtonBox.ActionRole,
default=default,
)
else:
bbox = None
self.add_contents(panel(ui), bbox)
