import wx
import os
from wx import ToolBar as Tbar
from wx.lib.platebtn import PlateButton
from sas.guiframe.gui_style import GUIFRAME_ID
from sas.guiframe.gui_style import GUIFRAME_ICON
from wx.lib.platebtn import PB_STYLE_DEFAULT, PB_STYLE_DROPARROW,PB_STYLE_NOBG
#Control panel width
import sys
if sys.platform.count("darwin")==0:
FONT_VARIANT = 0
NAME_BOX = wx.DefaultSize
TB_H = 21
IS_MAC = False
else:
FONT_VARIANT = 1
NAME_BOX = (200, 25)
TB_H = 25
IS_MAC = True
def clear_image(image):
[docs] w, h = image.GetSize()
factor = 155
compress = lambda x: int(x * factor/255.) + factor
for y in range(h):
for x in range(w):
grey = compress(image.GetGreen(x, y))
image.SetRGB(x, y, grey, grey, grey)
if image.HasAlpha():
image.ConvertAlphaToMask()
return image
class GUIToolBar(Tbar):
[docs] """
Implement toolbar for guiframe
"""
def __init__(self, parent, *args, **kwds):
Tbar.__init__(self, parent, *args, **kwds)
#Set window's font size
self.SetWindowVariant(variant=FONT_VARIANT)
self.parent = parent
self._bookmark_menu = None
self._bookmark_bt = None
self.do_layout()
self.on_bind_button()
def do_layout(self):
[docs] """
"""
t_size = TB_H
tbar_size = (t_size, t_size)
button_type = wx.ITEM_NORMAL
reset_im = GUIFRAME_ICON.RESET_ICON
reset_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
reset_bmp = reset_im.ConvertToBitmap()
#disable_reset_bmp = clear_image(reset_im).ConvertToBitmap()
disable_reset_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.RESET_ID, 'Reset', reset_bmp,
disable_reset_bmp, button_type,'Reset')
self.AddSeparator()
save_im = GUIFRAME_ICON.SAVE_ICON
save_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
save_bmp = save_im.ConvertToBitmap()
disable_save_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.SAVE_ID, 'Save', save_bmp,
disable_save_bmp, button_type,'Save')
self.AddSeparator()
report_im = GUIFRAME_ICON.REPORT_ICON
report_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
report_bmp = report_im.ConvertToBitmap()
#disable_report_bmp = clear_image(report_im).ConvertToBitmap()
disable_report_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.PREVIEW_ID, 'Report', report_bmp,
disable_report_bmp, button_type,'Report')
self.AddSeparator()
undo_im = GUIFRAME_ICON.UNDO_ICON
undo_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
undo_bmp = undo_im.ConvertToBitmap()
disable_undo_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.UNDO_ID, 'Undo', undo_bmp,
disable_undo_bmp, button_type,'Undo')
self.AddSeparator()
redo_im = GUIFRAME_ICON.REDO_ICON
redo_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
redo_bmp = redo_im.ConvertToBitmap()
disable_redo_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.REDO_ID, 'Redo', redo_bmp,
disable_redo_bmp, button_type,'Redo')
self.AddSeparator()
copy_im = GUIFRAME_ICON.COPY_ICON
copy_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
copy_bmp = copy_im.ConvertToBitmap()
disable_copy_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.COPY_ID, 'Copy', copy_bmp,
disable_copy_bmp, button_type,'Copy parameter values')
self.AddSeparator()
paste_im = GUIFRAME_ICON.PASTE_ICON
paste_im.Rescale(tbar_size[0], tbar_size[1], wx.IMAGE_QUALITY_HIGH)
paste_bmp = paste_im.ConvertToBitmap()
disable_paste_bmp = wx.NullBitmap
self.AddLabelTool(GUIFRAME_ID.PASTE_ID, 'Paste', paste_bmp,
disable_paste_bmp, button_type,'Paste parameter values')
self.AddSeparator()
self._bookmark_bt = PlateButton(self, -1, 'Bookmarks',
GUIFRAME_ICON.BOOKMARK_ICON.ConvertToBitmap(),
style=PB_STYLE_DEFAULT|PB_STYLE_DROPARROW|PB_STYLE_NOBG)
self._bookmark_bt.SetWindowVariant(FONT_VARIANT)
self._bookmark_bt.Disable()
self._bookmark_menu = wx.Menu()
self.add_bookmark_default()
self._bookmark_bt.SetMenu(self._bookmark_menu)
self.AddControl(self._bookmark_bt)
self.SetToolBitmapSize(tbar_size)
self.AddSeparator()
#add button for the panel on focus
self.button_panel = wx.StaticText(self, -1, 'No Panel',
pos=wx.DefaultPosition,
size=NAME_BOX,
style=wx.SUNKEN_BORDER|wx.ALIGN_LEFT)
button_panel_font = self.button_panel.GetFont()
button_panel_font.SetWeight(wx.BOLD)
self.button_panel.SetFont(button_panel_font)
hint = 'Control Panel on Focus'
self.button_panel.SetToolTipString(hint)
self.AddControl(self.button_panel)
self.AddSeparator()
self.Realize()
def add_bookmark_default(self):
[docs] """
Add default items in bookmark menu
"""
id = wx.NewId()
self._bookmark_menu.Append(id, 'Bookmark This Page State')
self._bookmark_menu.AppendSeparator()
wx.EVT_MENU(self, id, self.on_bookmark)
def on_bind_button(self):
[docs] """
Bind the buttons
"""
if self.parent is not None:
self.parent.Bind(wx.EVT_TOOL, self.parent.on_redo_panel,
id=GUIFRAME_ID.REDO_ID)
self.parent.Bind(wx.EVT_TOOL, self.parent.on_undo_panel,
id=GUIFRAME_ID.UNDO_ID)
self.parent.Bind(wx.EVT_TOOL, self.parent.on_copy_panel,
id=GUIFRAME_ID.COPY_ID)
self.parent.Bind(wx.EVT_TOOL, self.parent.on_paste_panel,
id=GUIFRAME_ID.PASTE_ID)
self.parent.Bind(wx.EVT_TOOL, self.parent.on_reset_panel,
id=GUIFRAME_ID.RESET_ID)
self.parent.Bind(wx.EVT_TOOL, self.parent.on_save_panel,
id=GUIFRAME_ID.SAVE_ID)
self.parent.Bind(wx.EVT_TOOL, self.parent.on_preview_panel,
id=GUIFRAME_ID.PREVIEW_ID)
#self.parent.Bind(wx.EVT_TOOL, self.parent.on_print_panel,
# id=GUIFRAME_ID.PRINT_ID)
def update_button(self, application_name='', panel_name=''):
[docs] """
"""
#self.button_application.SetLabel(str(application_name))
self.button_panel.SetLabel(str(panel_name))
self.button_panel.SetToolTipString(str(panel_name))
def update_toolbar(self, panel=None):
[docs] """
"""
if panel is None:
#self.EnableTool(GUIFRAME_ID.PRINT_ID, False)
self.EnableTool(GUIFRAME_ID.UNDO_ID,False)
self.EnableTool(GUIFRAME_ID.REDO_ID, False)
self.EnableTool(GUIFRAME_ID.COPY_ID,False)
self.EnableTool(GUIFRAME_ID.PASTE_ID, False)
self.EnableTool(GUIFRAME_ID.PREVIEW_ID, False)
self.EnableTool(GUIFRAME_ID.RESET_ID, False)
self.EnableTool(GUIFRAME_ID.SAVE_ID, False)
self._bookmark_bt.Disable()
else:
#self.EnableTool(GUIFRAME_ID.PRINT_ID, panel.get_print_flag())
self.EnableTool(GUIFRAME_ID.UNDO_ID, panel.get_undo_flag())
self.EnableTool(GUIFRAME_ID.REDO_ID, panel.get_redo_flag())
self.EnableTool(GUIFRAME_ID.COPY_ID, panel.get_copy_flag())
self.EnableTool(GUIFRAME_ID.PASTE_ID, panel.get_paste_flag())
self.EnableTool(GUIFRAME_ID.PREVIEW_ID, panel.get_preview_flag())
self.EnableTool(GUIFRAME_ID.RESET_ID, panel.get_reset_flag())
self.EnableTool(GUIFRAME_ID.SAVE_ID, panel.get_save_flag())
self._bookmark_bt.Enable(panel.get_bookmark_flag())
self.Realize()
def enable_undo(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.UNDO_ID, panel.get_undo_flag())
self.Realize()
def enable_redo(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.REDO_ID, panel.get_redo_flag())
self.Realize()
def enable_copy(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.COPY_ID, panel.get_copy_flag())
self.Realize()
def enable_paste(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.PASTE_ID, panel.get_paste_flag())
self.Realize()
def enable_print(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.PRINT_ID, panel.get_print_flag())
self.Realize()
def enable_zoom(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.ZOOM_ID, panel.get_zoom_flag())
self.Realize()
def enable_zoom_in(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.ZOOM_IN_ID, panel.get_zoom_in_flag())
self.Realize()
def enable_zoom_out(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.ZOOM_OUT_ID, panel.get_zoom_out_flag())
self.Realize()
def enable_bookmark(self, panel):
[docs] flag = panel.get_bookmark_flag()
self._bookmark_bt.Enable(flag)
self.Realize()
def enable_save(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.SAVE_ID, panel.get_save_flag())
self.Realize()
def enable_reset(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.RESET_ID, panel.get_reset_flag())
self.Realize()
def enable_preview(self, panel):
[docs] self.EnableTool(GUIFRAME_ID.PREVIEW_ID, panel.get_preview_flag())
self.Realize()
def on_bookmark(self, event):
[docs] """
add book mark
"""
if self.parent is not None:
self.parent.on_bookmark_panel(event)
def append_bookmark(self, event):
[docs] """
receive item to append on the toolbar button bookmark
"""
title = event.title
hint = event.hint
handler = event.handler
id = wx.NewId()
self._bookmark_menu.Append(id, str(title), str(hint))
wx.EVT_MENU(self, id, handler)
def remove_bookmark_item(self, item):
[docs] """
Remove a bookmark item
"""
self._bookmark_menu.DestroyItem(item)
def get_bookmark_items(self):
[docs] """
Get bookmark menu items
"""
return self._bookmark_menu.GetMenuItems()
def append_bookmark_item(self, id, label):