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): 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): """ 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): """ """ 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): """ 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): """ 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=''): """ """ #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): """ """ 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): self.EnableTool(GUIFRAME_ID.UNDO_ID, panel.get_undo_flag()) self.Realize() def enable_redo(self, panel): self.EnableTool(GUIFRAME_ID.REDO_ID, panel.get_redo_flag()) self.Realize() def enable_copy(self, panel): self.EnableTool(GUIFRAME_ID.COPY_ID, panel.get_copy_flag()) self.Realize() def enable_paste(self, panel): self.EnableTool(GUIFRAME_ID.PASTE_ID, panel.get_paste_flag()) self.Realize() def enable_print(self, panel): self.EnableTool(GUIFRAME_ID.PRINT_ID, panel.get_print_flag()) self.Realize() def enable_zoom(self, panel): self.EnableTool(GUIFRAME_ID.ZOOM_ID, panel.get_zoom_flag()) self.Realize() def enable_zoom_in(self, panel): self.EnableTool(GUIFRAME_ID.ZOOM_IN_ID, panel.get_zoom_in_flag()) self.Realize() def enable_zoom_out(self, panel): self.EnableTool(GUIFRAME_ID.ZOOM_OUT_ID, panel.get_zoom_out_flag()) self.Realize() def enable_bookmark(self, panel): flag = panel.get_bookmark_flag() self._bookmark_bt.Enable(flag) self.Realize() def enable_save(self, panel): self.EnableTool(GUIFRAME_ID.SAVE_ID, panel.get_save_flag()) self.Realize() def enable_reset(self, panel): self.EnableTool(GUIFRAME_ID.RESET_ID, panel.get_reset_flag()) self.Realize() def enable_preview(self, panel): self.EnableTool(GUIFRAME_ID.PREVIEW_ID, panel.get_preview_flag()) self.Realize() def on_bookmark(self, event): """ add book mark """ if self.parent is not None: self.parent.on_bookmark_panel(event) def append_bookmark(self, event): """ 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): """ Remove a bookmark item """ self._bookmark_menu.DestroyItem(item) def get_bookmark_items(self): """ Get bookmark menu items """ return self._bookmark_menu.GetMenuItems() def append_bookmark_item(self, id, label): """ Append a item in bookmark """ self._bookmark_menu.Append(id, label)