#!/usr/bin/env python
# -*- coding: utf-8 -*-
#This file is part of Beremiz, a Integrated Development Environment for
#programming IEC 61131-3 automates supporting plcopen standard and CanFestival.
#
#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
#
#See COPYING file for copyrights details.
#
#This library is free software; you can redistribute it and/or
#modify it under the terms of the GNU General Public
#License as published by the Free Software Foundation; either
#version 2.1 of the License, or (at your option) any later version.
#
#This library is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
#General Public License for more details.
#
#You should have received a copy of the GNU General Public
#License along with this library; if not, write to the Free Software
#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
__version__ = "$Revision$"
import wx
import wx.lib.buttons
if wx.VERSION >= (2, 8, 0):
import wx.lib.customtreectrl as CT
import types
import time
import os, re, platform, sys, time, traceback, getopt, commands
from plugger import PluginsRoot
from wxPopen import wxPopen3
CWD = os.path.split(os.path.realpath(__file__))[0]
class LogPseudoFile:
""" Base class for file like objects to facilitate StdOut for the Shell."""
def __init__(self, output = None):
self.red_white = wx.TextAttr("RED", "WHITE")
self.red_yellow = wx.TextAttr("RED", "YELLOW")
self.black_white = wx.TextAttr("BLACK", "WHITE")
self.default_style = None
self.output = output
def writelines(self, l):
map(self.write, l)
def write(self, s, style = None):
if not style : style=self.black_white
if self.default_style != style:
self.output.SetDefaultStyle(style)
self.default_style = style
self.output.AppendText(s)
def write_warning(self, s):
self.write(s,self.red_white)
def write_error(self, s):
self.write(s,self.red_yellow)
def flush(self):
self.output.SetValue("")
def isatty(self):
return false
def LogCommand(self, Command, sz_limit = 100, no_stdout=False):
self.errlen = 0
self.exitcode = None
self.outdata = ""
self.errdata = ""
def output(v):
self.outdata += v
if not no_stdout:
self.write(v)
def errors(v):
self.errdata += v
self.errlen += 1
if self.errlen > sz_limit:
p.kill()
self.write_warning(v)
def fin(pid,ecode):
self.exitcode = ecode
if self.exitcode != 0:
self.write(Command + "\n")
self.write_warning("exited with status %d (pid %d)\n"%(ecode,pid))
def spin(p):
while not p.finished:
wx.Yield()
time.sleep(0.01)
input = []
p = wxPopen3(Command, input, output, errors, fin, self.output)
if p.pid:
spin(p)
return (self.exitcode, self.outdata, self.errdata)
[ID_BEREMIZ, ID_BEREMIZMAINSPLITTER,
ID_BEREMIZSECONDSPLITTER, ID_BEREMIZLEFTPANEL,
ID_BEREMIZPARAMSPANEL, ID_BEREMIZLOGCONSOLE,
ID_BEREMIZPLUGINTREE, ID_BEREMIZPLUGINCHILDS,
ID_BEREMIZADDBUTTON, ID_BEREMIZDELETEBUTTON,
ID_BEREMIZPARAMSSTATICBOX, ID_BEREMIZPARAMSENABLE,
ID_BEREMIZPARAMSTARGETTYPE, ID_BEREMIZPARAMSIECCHANNEL,
ID_BEREMIZPARAMSSTATICTEXT1, ID_BEREMIZPARAMSSTATICTEXT2,
ID_BEREMIZPARAMSATTRIBUTESGRID,
] = [wx.NewId() for _init_ctrls in range(17)]
[ID_BEREMIZFILEMENUITEMS0, ID_BEREMIZFILEMENUITEMS1,
ID_BEREMIZFILEMENUITEMS2, ID_BEREMIZFILEMENUITEMS3,
ID_BEREMIZFILEMENUITEMS5, ID_BEREMIZFILEMENUITEMS7,
] = [wx.NewId() for _init_coll_FileMenu_Items in range(6)]
[ID_BEREMIZEDITMENUITEMS0, ID_BEREMIZEDITMENUITEMS2,
ID_BEREMIZEDITMENUITEMS3,
] = [wx.NewId() for _init_coll_EditMenu_Items in range(3)]
[ID_BEREMIZRUNMENUITEMS0, ID_BEREMIZRUNMENUITEMS2,
ID_BEREMIZRUNMENUITEMS3, ID_BEREMIZRUNMENUITEMS5,
] = [wx.NewId() for _init_coll_EditMenu_Items in range(4)]
[ID_BEREMIZHELPMENUITEMS0, ID_BEREMIZHELPMENUITEMS1,
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(2)]
class Beremiz(wx.Frame):
def _init_coll_FileMenu_Items(self, parent):
parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'New\tCTRL+N')
parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'Open\tCTRL+O')
parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'Save\tCTRL+S')
parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS3,
kind=wx.ITEM_NORMAL, text=u'Close Project')
parent.AppendSeparator()
parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS5,
kind=wx.ITEM_NORMAL, text=u'Properties')
parent.AppendSeparator()
parent.Append(help='', id=ID_BEREMIZFILEMENUITEMS7,
kind=wx.ITEM_NORMAL, text=u'Quit\tCTRL+Q')
self.Bind(wx.EVT_MENU, self.OnNewProjectMenu,
id=ID_BEREMIZFILEMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu,
id=ID_BEREMIZFILEMENUITEMS1)
self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu,
id=ID_BEREMIZFILEMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu,
id=ID_BEREMIZFILEMENUITEMS3)
self.Bind(wx.EVT_MENU, self.OnPropertiesMenu,
id=ID_BEREMIZFILEMENUITEMS5)
self.Bind(wx.EVT_MENU, self.OnQuitMenu,
id=ID_BEREMIZFILEMENUITEMS7)
def _init_coll_EditMenu_Items(self, parent):
parent.Append(help='', id=ID_BEREMIZEDITMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'Edit PLC\tCTRL+R')
parent.AppendSeparator()
parent.Append(help='', id=ID_BEREMIZEDITMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'Add Plugin')
parent.Append(help='', id=ID_BEREMIZEDITMENUITEMS3,
kind=wx.ITEM_NORMAL, text=u'Delete Plugin')
self.Bind(wx.EVT_MENU, self.OnEditPLCMenu,
id=ID_BEREMIZEDITMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnAddMenu,
id=ID_BEREMIZEDITMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnDeleteMenu,
id=ID_BEREMIZEDITMENUITEMS3)
def _init_coll_RunMenu_Items(self, parent):
parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'Build\tCTRL+R')
parent.AppendSeparator()
parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS2,
kind=wx.ITEM_NORMAL, text=u'Simulate')
parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS3,
kind=wx.ITEM_NORMAL, text=u'Run')
parent.AppendSeparator()
parent.Append(help='', id=ID_BEREMIZRUNMENUITEMS5,
kind=wx.ITEM_NORMAL, text=u'Save Log')
self.Bind(wx.EVT_MENU, self.OnBuildMenu,
id=ID_BEREMIZRUNMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnSimulateMenu,
id=ID_BEREMIZRUNMENUITEMS2)
self.Bind(wx.EVT_MENU, self.OnRunMenu,
id=ID_BEREMIZRUNMENUITEMS3)
self.Bind(wx.EVT_MENU, self.OnSaveLogMenu,
id=ID_BEREMIZRUNMENUITEMS5)
def _init_coll_HelpMenu_Items(self, parent):
parent.Append(help='', id=ID_BEREMIZHELPMENUITEMS0,
kind=wx.ITEM_NORMAL, text=u'Beremiz\tF1')
parent.Append(help='', id=ID_BEREMIZHELPMENUITEMS1,
kind=wx.ITEM_NORMAL, text=u'About')
self.Bind(wx.EVT_MENU, self.OnBeremizMenu,
id=ID_BEREMIZHELPMENUITEMS0)
self.Bind(wx.EVT_MENU, self.OnAboutMenu,
id=ID_BEREMIZHELPMENUITEMS1)
def _init_coll_menuBar1_Menus(self, parent):
parent.Append(menu=self.FileMenu, title=u'File')
#parent.Append(menu=self.EditMenu, title=u'Edit')
#parent.Append(menu=self.RunMenu, title=u'Run')
parent.Append(menu=self.HelpMenu, title=u'Help')
def _init_utils(self):
self.menuBar1 = wx.MenuBar()
self.FileMenu = wx.Menu(title=u'')
#self.EditMenu = wx.Menu(title=u'')
#self.RunMenu = wx.Menu(title=u'')
self.HelpMenu = wx.Menu(title=u'')
self._init_coll_menuBar1_Menus(self.menuBar1)
self._init_coll_FileMenu_Items(self.FileMenu)
#self._init_coll_EditMenu_Items(self.EditMenu)
#self._init_coll_RunMenu_Items(self.RunMenu)
self._init_coll_HelpMenu_Items(self.HelpMenu)
def _init_coll_LeftGridSizer_Items(self, parent):
parent.AddWindow(self.PluginTree, 0, border=0, flag=wx.GROW)
parent.AddSizer(self.ButtonGridSizer, 0, border=0, flag=wx.GROW)
def _init_coll_LeftGridSizer_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_coll_RightGridSizer_Items(self, parent):
parent.AddSizer(self.MenuSizer, 0, border=0, flag=wx.GROW)
if wx.VERSION < (2, 8, 0):
parent.AddWindow(self.SecondSplitter, 0, border=0, flag=wx.GROW)
else:
parent.AddWindow(self.ParamsPanel, 0, border=0, flag=wx.GROW)
def _init_coll_RightGridSizer_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(1)
def _init_coll_ButtonGridSizer_Items(self, parent):
parent.AddWindow(self.PluginChilds, 0, border=0, flag=wx.GROW)
parent.AddWindow(self.AddButton, 0, border=0, flag=0)
parent.AddWindow(self.DeleteButton, 0, border=0, flag=0)
def _init_coll_ButtonGridSizer_Growables(self, parent):
parent.AddGrowableCol(0)
parent.AddGrowableRow(0)
def _init_sizers(self):
self.LeftGridSizer = wx.FlexGridSizer(cols=1, hgap=2, rows=1, vgap=2)
self.ButtonGridSizer = wx.FlexGridSizer(cols=3, hgap=2, rows=1, vgap=2)
self.RightGridSizer = wx.FlexGridSizer(cols=1, hgap=2, rows=2, vgap=2)
self.MenuSizer = wx.BoxSizer(wx.VERTICAL)
self.ParamsPanelMainSizer = wx.BoxSizer(wx.VERTICAL)
self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)
self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)
self._init_coll_ButtonGridSizer_Growables(self.ButtonGridSizer)
self._init_coll_ButtonGridSizer_Items(self.ButtonGridSizer)
self._init_coll_RightGridSizer_Growables(self.RightGridSizer)
self._init_coll_RightGridSizer_Items(self.RightGridSizer)
self.LeftPanel.SetSizer(self.LeftGridSizer)
self.RightPanel.SetSizer(self.RightGridSizer)
self.ParamsPanel.SetSizer(self.ParamsPanelMainSizer)
def _init_ctrls(self, prnt):
wx.Frame.__init__(self, id=ID_BEREMIZ, name=u'Beremiz',
parent=prnt, pos=wx.Point(0, 0), size=wx.Size(1000, 600),
style=wx.DEFAULT_FRAME_STYLE|wx.CLIP_CHILDREN, title=u'Beremiz')
self._init_utils()
self.SetClientSize(wx.Size(1000, 600))
self.SetMenuBar(self.menuBar1)
self.Bind(wx.EVT_ACTIVATE, self.OnFrameActivated)
if wx.VERSION < (2, 8, 0):
self.MainSplitter = wx.SplitterWindow(id=ID_BEREMIZMAINSPLITTER,
name='MainSplitter', parent=self, point=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.SP_3D)
self.MainSplitter.SetNeedUpdating(True)
self.MainSplitter.SetMinimumPaneSize(1)
self.LeftPanel = wx.Panel(id=ID_BEREMIZLEFTPANEL,
name='LeftPanel', parent=self.MainSplitter, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)
self.PluginTree = wx.TreeCtrl(id=ID_BEREMIZPLUGINTREE,
name='PluginTree', parent=self.LeftPanel, pos=wx.Point(0, 0),
size=wx.Size(-1, -1), style=wx.TR_HAS_BUTTONS|wx.TR_SINGLE|wx.SUNKEN_BORDER)
self.PluginTree.Bind(wx.EVT_RIGHT_UP, self.OnPluginTreeRightUp)
self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnPluginTreeItemSelected,
id=ID_BEREMIZPLUGINTREE)
self.PluginChilds = wx.Choice(id=ID_BEREMIZPLUGINCHILDS,
name='PluginChilds', parent=self.LeftPanel, pos=wx.Point(0, 0),
size=wx.Size(-1, -1), style=0)
self.AddButton = wx.lib.buttons.GenBitmapButton(ID=ID_BEREMIZADDBUTTON, bitmap=wx.Bitmap(os.path.join(CWD, 'images', 'Add32x32.png')),
name='AddBusButton', parent=self.LeftPanel, pos=wx.Point(0, 0),
size=wx.Size(32, 32), style=wx.NO_BORDER)
self.AddButton.SetToolTipString("Add a plugin of the type selected")
self.AddButton.Bind(wx.EVT_BUTTON, self.OnAddButton,
id=ID_BEREMIZADDBUTTON)
self.DeleteButton = wx.lib.buttons.GenBitmapButton(ID=ID_BEREMIZDELETEBUTTON, bitmap=wx.Bitmap(os.path.join(CWD, 'images', 'Delete32x32.png')),
name='DeleteBusButton', parent=self.LeftPanel, pos=wx.Point(0, 0),
size=wx.Size(32, 32), style=wx.NO_BORDER)
self.DeleteButton.SetToolTipString("Delete the current selected plugin")
self.DeleteButton.Bind(wx.EVT_BUTTON, self.OnDeleteButton,
id=ID_BEREMIZDELETEBUTTON)
self.RightPanel = wx.Panel(id=ID_BEREMIZLEFTPANEL,
name='RightPanel', parent=self.MainSplitter, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)
self.SecondSplitter = wx.SplitterWindow(id=ID_BEREMIZSECONDSPLITTER,
name='SecondSplitter', parent=self.RightPanel, point=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.SP_3D)
self.SecondSplitter.SetNeedUpdating(True)
self.SecondSplitter.SetMinimumPaneSize(1)
self.MainSplitter.SplitVertically(self.LeftPanel, self.RightPanel,
300)
self.ParamsPanel = wx.ScrolledWindow(id=ID_BEREMIZPARAMSPANEL,
name='ParamsPanel', parent=self.SecondSplitter, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)
self.ParamsPanel.SetScrollbars(10, 10, 0, 0, 0, 0)
self.LogConsole = wx.TextCtrl(id=ID_BEREMIZLOGCONSOLE, value='',
name='LogConsole', parent=self.SecondSplitter, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TE_MULTILINE|wx.TE_RICH2)
self.SecondSplitter.SplitHorizontally(self.ParamsPanel, self.LogConsole,
-250)
self._init_sizers()
else:
self.AUIManager = wx.aui.AuiManager(self)
self.AUIManager.SetDockSizeConstraint(0.5, 0.5)
self.Panes = {}
self.PluginTree = CT.CustomTreeCtrl(id=ID_BEREMIZPLUGINTREE,
name='PluginTree', parent=self, pos=wx.Point(0, 0),
size=wx.Size(-1, -1), style=CT.TR_HAS_BUTTONS|CT.TR_EDIT_LABELS|CT.TR_HAS_VARIABLE_ROW_HEIGHT|wx.TR_SINGLE|wx.SUNKEN_BORDER)
self.PluginTree.Bind(wx.EVT_RIGHT_UP, self.OnPluginTreeRightUp)
self.PluginTree.Bind(wx.EVT_IDLE, self.OnPluginTreeIdle)
self.Bind(CT.EVT_TREE_SEL_CHANGED, self.OnPluginTreeItemSelected,
id=ID_BEREMIZPLUGINTREE)
self.Bind(CT.EVT_TREE_ITEM_CHECKED, self.OnPluginTreeItemChecked,
id=ID_BEREMIZPLUGINTREE)
self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnPluginTreeItemBeginEdit,
id=ID_BEREMIZPLUGINTREE)
self.Bind(CT.EVT_TREE_END_LABEL_EDIT, self.OnPluginTreeItemEndEdit,
id=ID_BEREMIZPLUGINTREE)
self.Bind(CT.EVT_TREE_ITEM_EXPANDED, self.OnPluginTreeItemExpanded,
id=ID_BEREMIZPLUGINTREE)
self.Bind(CT.EVT_TREE_ITEM_COLLAPSED, self.OnPluginTreeItemCollapsed,
id=ID_BEREMIZPLUGINTREE)
self.AUIManager.AddPane(self.PluginTree, wx.aui.AuiPaneInfo().CenterPane())
self.LogConsole = wx.TextCtrl(id=ID_BEREMIZLOGCONSOLE, value='',
name='LogConsole', parent=self, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TE_MULTILINE|wx.TE_RICH2)
self.AUIManager.AddPane(self.LogConsole, wx.aui.AuiPaneInfo().Caption("Log Console").Bottom().Layer(1).BestSize(wx.Size(800, 200)).CloseButton(False))
self.AUIManager.Update()
def __init__(self, parent, projectOpen):
self._init_ctrls(parent)
self.Log = LogPseudoFile(self.LogConsole)
self.PluginRoot = PluginsRoot(self)
self.DisableEvents = False
if projectOpen:
self.PluginRoot.LoadProject(projectOpen, self.Log)
self.RefreshPluginTree()
self.RefreshPluginToolBar()
self.PluginTree.SelectItem(self.PluginTree.GetRootItem())
if wx.VERSION < (2, 8, 0):
self.RefreshPluginParams()
self.RefreshButtons()
self.RefreshMainMenu()
def OnFrameActivated(self, event):
if not event.GetActive():
self.PluginRoot.RefreshPluginsBlockLists()
def RefreshButtons(self):
if wx.VERSION < (2, 8, 0):
if self.PluginRoot.HasProjectOpened():
self.PluginChilds.Enable(True)
self.AddButton.Enable(True)
self.DeleteButton.Enable(True)
else:
self.PluginChilds.Enable(False)
self.AddButton.Enable(False)
self.DeleteButton.Enable(False)
def RefreshMainMenu(self):
if self.menuBar1:
if self.PluginRoot.HasProjectOpened():
## self.menuBar1.EnableTop(1, True)
## self.menuBar1.EnableTop(2, True)
self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, True)
self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, True)
self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, True)
else:
## self.menuBar1.EnableTop(1, False)
## self.menuBar1.EnableTop(2, False)
self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS2, False)
self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS3, False)
self.FileMenu.Enable(ID_BEREMIZFILEMENUITEMS5, False)
def RefreshPluginTree(self):
self.DisableEvents = True
infos = self.PluginRoot.GetPlugInfos()
root = self.PluginTree.GetRootItem()
if root is None or not root.IsOk():
root = self.PluginTree.AddRoot(infos["name"])
last_selected = self.GetSelectedPluginName()
self.GenerateTreeBranch(root, infos, True)
self.PluginTree.Expand(self.PluginTree.GetRootItem())
self.SelectedPluginByName(root, last_selected)
if wx.VERSION < (2, 8, 0):
self.RefreshPluginParams()
self.DisableEvents = False
self.MustRecalTreeSizes = True
def SelectedPluginByName(self, root, name):
if name:
toks = name.split('.', 1)
item, root_cookie = self.PluginTree.GetFirstChild(root)
while item is not None and item.IsOk():
if self.PluginTree.GetPyData(item) == toks[0]:
if len(toks)>1:
return self.SelectedPluginByName(item, toks[1])
else:
self.PluginTree.SelectItem(item, True)
return True
item, root_cookie = self.PluginTree.GetNextChild(root, root_cookie)
return False
def GenerateTreeBranch(self, root, infos, first = False):
to_delete = []
self.PluginTree.SetItemText(root, infos["name"])
self.PluginTree.SetPyData(root, infos["type"])
if wx.VERSION >= (2, 8, 0):
plugin = self.GetSelectedPlugin(root)
if plugin is not None:
old_window = self.PluginTree.GetItemWindow(root)
if old_window is not None:
old_window.GetSizer().Clear(True)
old_window.Destroy()
outside = wx.Panel(self.PluginTree, -1, size=wx.Size(-1, -1))
outside.SetBackgroundColour(wx.WHITE)
insidesizer = wx.FlexGridSizer(cols=2,rows=1)
insidesizer.AddGrowableCol(1)
outside.SetSizer(insidesizer)
insidesizer.AddSpacer(-1)
window = wx.Panel(outside, -1, size=wx.Size(-1, -1))
window.SetBackgroundColour(wx.Colour(250,250,255))
insidesizer.AddWindow(window, border=1, flag=wx.GROW|wx.ALL)
tcsizer = wx.FlexGridSizer(cols=3, hgap=0, rows=1, vgap=0)
if "channel" in infos:
sc = wx.SpinCtrl(window, -1, size=wx.Size(50, 25))
sc.SetValue(infos["channel"])
sc.Bind(wx.EVT_SPINCTRL, self.GetItemChannelChangedFunction(root))
tcsizer.AddWindow(sc, 0, border=5, flag=wx.LEFT|wx.TOP|wx.BOTTOM|wx.ALIGN_CENTER)
bsizer = wx.BoxSizer(wx.HORIZONTAL)
if "parent" in infos or plugin == self.PluginRoot:
addbutton_id = wx.NewId()
addbutton = wx.lib.buttons.GenBitmapButton(id=addbutton_id, bitmap=wx.Bitmap(os.path.join(CWD, 'images', 'Add24x24.png')),
name='AddBusButton', parent=window, pos=wx.Point(0, 0),
size=wx.Size(24, 24), style=wx.NO_BORDER)
addbutton.SetToolTipString("Add a plugin to this one")
addbutton.Bind(wx.EVT_BUTTON, self.GetAddButtonFunction(root, window), id=addbutton_id)
bsizer.AddWindow(addbutton, 0, border=5, flag=wx.LEFT|wx.RIGHT|wx.TOP|wx.BOTTOM|wx.ALIGN_CENTER)
if plugin != self.PluginRoot:
deletebutton_id = wx.NewId()
deletebutton = wx.lib.buttons.GenBitmapButton(id=deletebutton_id, bitmap=wx.Bitmap(os.path.join(CWD, 'images', 'Delete24x24.png')),
name='DeleteBusButton', parent=window, pos=wx.Point(0, 0),
size=wx.Size(24, 24), style=wx.NO_BORDER)
deletebutton.SetToolTipString("Delete this plugin")
deletebutton.Bind(wx.EVT_BUTTON, self.GetDeleteButtonFunction(root), id=deletebutton_id)
bsizer.AddWindow(deletebutton, 0, border=5, flag=wx.RIGHT|wx.TOP|wx.BOTTOM|wx.ALIGN_CENTER)
plugin_infos = plugin.GetParamsAttributes()
psizer = wx.BoxSizer(wx.HORIZONTAL)
self.RefreshSizerElement(window, psizer, plugin, plugin_infos, None, True, root)
msizer = wx.BoxSizer(wx.VERTICAL)
msizer.AddSizer(bsizer, 0, border=0, flag=wx.GROW)
if len(plugin.PluginMethods) > 0: #and plugin != self.PluginRoot
for plugin_method in plugin.PluginMethods:
if "method" in plugin_method:
id = wx.NewId()
button = wx.lib.buttons.GenBitmapTextButton(id=id, parent=window,
bitmap=wx.Bitmap(os.path.join(CWD, "%s24x24.png"%plugin_method.get("bitmap", os.path.join("images", "Unknown")))), label=plugin_method["name"],
name=plugin_method["name"], pos=wx.DefaultPosition, style=wx.NO_BORDER)
button.SetToolTipString(plugin_method["tooltip"])
button.Bind(wx.EVT_BUTTON, self.GetButtonCallBackFunction(plugin, plugin_method["method"]), id=id)
#hack to force size to mini
gen_mini_GetLabelSize = lambda obj:lambda:(wx.lib.buttons.GenBitmapTextButton._GetLabelSize(obj)[:-1] + (False,))
button._GetLabelSize = gen_mini_GetLabelSize(button)
#button._GetLabelSize = lambda :(-1,-1,False)
msizer.AddWindow(button, 0, border=5, flag=0)
tcsizer.AddSizer(msizer, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)
tcsizer.AddSizer(psizer, 0, border=0, flag=wx.GROW)
if plugin == self.PluginRoot:
tcsizer.AddGrowableCol(0)
else:
tcsizer.AddGrowableCol(1)
window.SetSizer(tcsizer)
insidesizer.Fit(outside)
self.PluginTree.SetItemWindow(root, outside)
if "enabled" in infos:
self.PluginTree.CheckItem(root, infos["enabled"])
self.PluginTree.SetItemWindowEnabled(root, infos["enabled"])
item, root_cookie = self.PluginTree.GetFirstChild(root)
for values in infos["values"]:
if item is None or not item.IsOk():
item = self.PluginTree.AppendItem(root, "")
# 2.6.x returns bad item on gtk
if wx.Platform != '__WXMSW__' or wx.VERSION >= (2, 8, 0):
item, root_cookie = self.PluginTree.GetNextChild(root, root_cookie)
self.GenerateTreeBranch(item, values)
item, root_cookie = self.PluginTree.GetNextChild(root, root_cookie)
while item is not None and item.IsOk():
to_delete.append(item)
item, root_cookie = self.PluginTree.GetNextChild(root, root_cookie)
for item in to_delete:
self.PluginTree.Delete(item)
MustRecalTreeSizes = False
def OnPluginTreeIdle(self,event):
if self.MustRecalTreeSizes:
self.ResizePluginTreeWindow()
event.RequestMore()
event.Skip()
def ResizePluginTreeWindow(self):
if getattr(self, "PluginRoot", None):
root = self.PluginTree.GetRootItem()
if root is not None and root.IsOk():
minimalsz = wx.Size(-1,-1)
minimalpos = wx.Point(-1,-1)
itemswindows = self.ResizePluginTreeWindow_r(root, minimalsz, minimalpos)
for window, posx in itemswindows:
spacer_width = minimalpos.x - posx
window.GetSizer().GetItem(0).SetSpacer(wx.Size(spacer_width,-1))
window.SetSize(wx.Size(spacer_width + minimalsz.x, -1))
window.GetSizer().Layout()
#wx.MessageBox(str(itemswindows) + "\n" + str(minimalpos) + "\n" + str(minimalsz) + "\n" + str(window.GetSizer().GetItem(0).GetSpacer()))
self.MustRecalTreeSizes = False
def ResizePluginTreeWindow_r(self, root, sz, pt):
window = self.PluginTree.GetItemWindow(root)
posx = window.GetPosition().x
res = [(window,posx)]
window.GetSizer().GetItem(0).SetSpacer(wx.Size(0,0))
bestsz = window.GetBestSize()
pt.x = max(pt.x, posx)
sz.x = max(sz.x, bestsz.x)
if root.IsExpanded():
item, root_cookie = self.PluginTree.GetFirstChild(root)
while item is not None and item.IsOk() :
res.extend(self.ResizePluginTreeWindow_r(item, sz, pt))
item, root_cookie = self.PluginTree.GetNextChild(root, root_cookie)
return res
def GetSelectedPluginName(self, selected = None):
if selected is None:
selected = self.PluginTree.GetSelection()
if selected is None or not selected.IsOk():
return None
if selected == self.PluginTree.GetRootItem():
return ""
else:
name = self.PluginTree.GetPyData(selected)
item = self.PluginTree.GetItemParent(selected)
while item.IsOk() and item != self.PluginTree.GetRootItem():
name = "%s.%s"%(self.PluginTree.GetPyData(item), name)
item = self.PluginTree.GetItemParent(item)
return name
def GetSelectedPlugin(self, selected = None):
name = self.GetSelectedPluginName(selected)
if name is None:
return None
elif name == "":
return self.PluginRoot
else:
return self.PluginRoot.GetChildByName(name)
def OnPluginTreeItemSelected(self, event):
if wx.VERSION < (2, 8, 0):
wx.CallAfter(self.RefreshPluginParams)
event.Skip()
def OnPluginTreeItemChecked(self, event):
if not self.DisableEvents:
item = event.GetItem()
plugin = self.GetSelectedPlugin(item)
if plugin:
res, StructChanged = plugin.SetParamsAttribute("BaseParams.Enabled", self.PluginTree.IsItemChecked(item), self.Log)
wx.CallAfter(self.RefreshPluginTree)
event.Skip()
def OnPluginTreeItemBeginEdit(self, event):
if event.GetItem() == self.PluginTree.GetRootItem():
event.Veto()
else:
event.Skip()
def OnPluginTreeItemEndEdit(self, event):
if event.GetLabel() == "":
event.Veto()
elif not self.DisableEvents:
plugin = self.GetSelectedPlugin()
if plugin:
res, StructChanged = plugin.SetParamsAttribute("BaseParams.Name", event.GetLabel(), self.Log)
wx.CallAfter(self.RefreshPluginTree)
event.Skip()
def ShowChildrenWindows(self, root, show = True):
item, root_cookie = self.PluginTree.GetFirstChild(root)
while item is not None and item.IsOk():
window = self.PluginTree.GetItemWindow(item)
if show:
window.Show()
else:
window.Hide()
if self.PluginTree.IsExpanded(item):
self.ShowChildrenWindows(item, show)
item, root_cookie = self.PluginTree.GetNextChild(root, root_cookie)
def OnPluginTreeItemExpanded(self, event):
self.ShowChildrenWindows(event.GetItem(), True)
self.MustRecalTreeSizes = True
event.Skip()
def OnPluginTreeItemCollapsed(self, event):
self.ShowChildrenWindows(event.GetItem(), False)
event.Skip()
def GetItemChannelChangedFunction(self, item):
def OnPluginTreeItemChannelChanged(event):
if not self.DisableEvents:
plugin = self.GetSelectedPlugin(item)
if plugin:
res, StructChanged = plugin.SetParamsAttribute("BaseParams.IEC_Channel", event.GetInt(), self.Log)
wx.CallAfter(self.RefreshPluginTree)
event.Skip()
return OnPluginTreeItemChannelChanged
def _GetAddPluginFunction(self, name, selected = None):
def OnPluginMenu(event):
self.AddPlugin(name, selected)
event.Skip()
return OnPluginMenu
def OnPluginTreeRightUp(self, event):
plugin = self.GetSelectedPlugin()
if plugin:
main_menu = wx.Menu(title='')
if len(plugin.PlugChildsTypes) > 0:
plugin_menu = wx.Menu(title='')
for name, XSDClass in plugin.PlugChildsTypes:
new_id = wx.NewId()
plugin_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=name)
self.Bind(wx.EVT_MENU, self._GetAddPluginFunction(name), id=new_id)
main_menu.AppendMenu(-1, "Add", plugin_menu, '')
new_id = wx.NewId()
main_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text="Delete")
self.Bind(wx.EVT_MENU, self.OnDeleteButton, id=new_id)
pos_x, pos_y = event.GetPosition()
self.PluginTree.PopupMenuXY(main_menu, pos_x, pos_y)
event.Skip()
def RefreshPluginToolBar(self):
if wx.VERSION < (2, 8, 0):
self.ClearSizer(self.MenuSizer)
else:
# toolbar temporarely disabled.
return
if "ToolBar" in self.Panes:
self.AUIManager.DetachPane(self.Panes["ToolBar"])
self.Panes["ToolBar"].Destroy()
if self.PluginRoot.HasOpenedProject() and len(self.PluginRoot.PluginMethods) > 0:
if wx.VERSION > (2, 8, 0):
toolbar = wx.ToolBar(self, -1, wx.DefaultPosition, wx.DefaultSize,
wx.TB_FLAT | wx.TB_NODIVIDER | wx.NO_BORDER | wx.TB_TEXT)
toolbar.SetToolBitmapSize(wx.Size(48, 48))
else:
boxsizer = wx.BoxSizer(wx.HORIZONTAL)
width = 0
for plugin_infos in self.PluginRoot.PluginMethods:
if "method" in plugin_infos:
id = wx.NewId()
if "bitmap" in plugin_infos:
if wx.VERSION < (2, 8, 0):
button = wx.lib.buttons.GenBitmapTextButton(ID=id, parent=self.RightPanel,
bitmap=wx.Bitmap(os.path.join(CWD, "%s32x32.png"%plugin_infos["bitmap"])), label=plugin_infos["name"],
name=plugin_infos["name"], pos=wx.Point(0, 0), style=wx.BU_EXACTFIT|wx.NO_BORDER)
else:
button = wx.lib.buttons.GenBitmapTextButton(id=id, parent=toolbar,
bitmap=wx.Bitmap(os.path.join(CWD, "%s32x32.png"%plugin_infos["bitmap"])), label=plugin_infos["name"],
name=plugin_infos["name"], pos=wx.Point(0, 0), style=wx.BU_EXACTFIT|wx.NO_BORDER)
button.SetSize(button.GetBestSize())
else:
if wx.VERSION < (2, 8, 0):
button = wx.Button(id=id, label=plugin_infos["name"],
name=plugin_infos["name"], parent=self.RightPanel,
pos=wx.Point(0, 0), style=wx.BU_EXACTFIT)
else:
button = wx.Button(id=id, label=plugin_infos["name"],
name=plugin_infos["name"], parent=toolbar,
pos=wx.Point(0, 0), size=(-1, 48), style=wx.BU_EXACTFIT)
button.SetToolTipString(plugin_infos["tooltip"])
button.Bind(wx.EVT_BUTTON, self.GetButtonCallBackFunction(self.PluginRoot, plugin_infos["method"]), id=id)
width += button.GetSize()[0]
if wx.VERSION < (2, 8, 0):
boxsizer.AddWindow(button, 0, border=5, flag=wx.GROW|wx.RIGHT)
else:
toolbar.AddControl(button)
if wx.VERSION < (2, 8, 0):
self.MenuSizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.ALL)
self.RightGridSizer.Layout()
else:
toolbar.Realize()
self.Panes["ToolBar"] = toolbar
self.AUIManager.AddPane(toolbar, wx.aui.AuiPaneInfo().
Name("ToolBar").Caption("Toolbar").
ToolbarPane().Top().
LeftDockable(False).RightDockable(False))
if wx.Platform == '__WXMSW__':
self.AUIManager.GetPane("ToolBar").BestSize(wx.Size(width + 3, 55))
self.AUIManager.Update()
def RefreshPluginParams(self):
plugin = self.GetSelectedPlugin()
if not plugin:
# Refresh ParamsPanel
self.ParamsPanel.Hide()
# Refresh PluginChilds
self.PluginChilds.Clear()
self.PluginChilds.Enable(False)
self.AddButton.Enable(False)
self.DeleteButton.Enable(False)
else:
# Refresh ParamsPanel
self.ParamsPanel.Show()
infos = plugin.GetParamsAttributes()
if len(self.MenuSizer.GetChildren()) > 1:
self.ClearSizer(self.MenuSizer.GetChildren()[1].GetSizer())
self.MenuSizer.Remove(1)
self.ClearSizer(self.ParamsPanelMainSizer)
self.RefreshSizerElement(self.ParamsPanel, self.ParamsPanelMainSizer, plugin, infos, None, False)
if plugin != self.PluginRoot and len(plugin.PluginMethods) > 0:
boxsizer = wx.BoxSizer(wx.HORIZONTAL)
self.MenuSizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.ALL)
for plugin_infos in plugin.PluginMethods:
if "method" in plugin_infos:
id = wx.NewId()
if "bitmap" in plugin_infos:
button = wx.lib.buttons.GenBitmapTextButton(ID=id, parent=self.RightPanel,
bitmap=wx.Bitmap(os.path.join(CWD, "%s32x32.png"%plugin_infos["bitmap"])), label=plugin_infos["name"],
name=plugin_infos["name"], pos=wx.Point(0, 0), style=wx.BU_EXACTFIT|wx.NO_BORDER)
else:
button = wx.Button(id=id, label=plugin_infos["name"],
name=plugin_infos["name"], parent=self.RightPanel,
pos=wx.Point(0, 0), style=wx.BU_EXACTFIT)
button.SetToolTipString(plugin_infos["tooltip"])
button.Bind(wx.EVT_BUTTON, self.GetButtonCallBackFunction(plugin, plugin_infos["method"]), id=id)
boxsizer.AddWindow(button, 0, border=5, flag=wx.GROW|wx.RIGHT)
self.RightGridSizer.Layout()
self.ParamsPanelMainSizer.Layout()
self.ParamsPanel.SetClientSize(self.ParamsPanel.GetClientSize())
# Refresh PluginChilds
self.PluginChilds.Clear()
if len(plugin.PlugChildsTypes) > 0:
self.PluginChilds.Append("")
for name, XSDClass in plugin.PlugChildsTypes:
self.PluginChilds.Append(name)
self.PluginChilds.Enable(True)
self.AddButton.Enable(True)
else:
self.PluginChilds.Enable(False)
self.AddButton.Enable(False)
self.DeleteButton.Enable(True)
def GetButtonCallBackFunction(self, plugin, method):
def OnButtonClick(event):
method(plugin, self.Log)
event.Skip()
return OnButtonClick
def GetChoiceCallBackFunction(self, choicectrl, plugin, path, selected=None):
def OnChoiceChanged(event):
if plugin:
res, StructChanged = plugin.SetParamsAttribute(path, choicectrl.GetStringSelection(), self.Log)
if StructChanged: wx.CallAfter(self.RefreshPluginTree)
choicectrl.SetStringSelection(res)
event.Skip()
return OnChoiceChanged
def GetChoiceContentCallBackFunction(self, choicectrl, staticboxsizer, plugin, path, selected=None):
def OnChoiceContentChanged(event):
if plugin:
res, StructChanged = plugin.SetParamsAttribute(path, choicectrl.GetStringSelection(), self.Log)
if StructChanged: wx.CallAfter(self.RefreshPluginTree)
choicectrl.SetStringSelection(res)
infos = self.PluginRoot.GetParamsAttributes(path)
staticbox = staticboxsizer.GetStaticBox()
staticbox.SetLabel("%(name)s - %(value)s"%infos)
if wx.VERSION < (2, 8, 0):
self.ParamsPanel.Freeze()
self.RefreshSizerElement(self.ParamsPanel, staticboxsizer, plugin, infos["children"], "%s.%s"%(path, infos["name"]), selected=selected)
self.ParamsPanelMainSizer.Layout()
self.ParamsPanel.Thaw()
self.ParamsPanel.Refresh()
else:
wx.CallAfter(self.RefreshPluginTree)
event.Skip()
return OnChoiceContentChanged
def GetTextCtrlCallBackFunction(self, textctrl, plugin, path, selected=None):
def OnTextCtrlChanged(event):
if plugin:
res, StructChanged = plugin.SetParamsAttribute(path, textctrl.GetValue(), self.Log)
if StructChanged: wx.CallAfter(self.RefreshPluginTree)
textctrl.SetValue(res)
event.Skip()
return OnTextCtrlChanged
def GetCheckBoxCallBackFunction(self, chkbx, plugin, path, selected=None):
def OnCheckBoxChanged(event):
if plugin:
res, StructChanged = plugin.SetParamsAttribute(path, chkbx.IsChecked(), self.Log)
if StructChanged: wx.CallAfter(self.RefreshPluginTree)
chkbx.SetValue(res)
event.Skip()
return OnCheckBoxChanged
def ClearSizer(self, sizer):
staticboxes = []
for item in sizer.GetChildren():
if item.IsSizer():
item_sizer = item.GetSizer()
self.ClearSizer(item_sizer)
if isinstance(item_sizer, wx.StaticBoxSizer):
staticboxes.append(item_sizer.GetStaticBox())
sizer.Clear(True)
for staticbox in staticboxes:
staticbox.Destroy()
def RefreshSizerElement(self, parent, sizer, plugin, elements, path, clean = True, selected = None):
if clean:
sizer.Clear(True)
first = True
for element_infos in elements:
if path:
element_path = "%s.%s"%(path, element_infos["name"])
else:
element_path = element_infos["name"]
if isinstance(element_infos["type"], types.ListType):
boxsizer = wx.BoxSizer(wx.HORIZONTAL)
if first:
sizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.ALL)
else:
sizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM)
bitmappath = os.path.join("images", "%s.png"%element_infos["name"])
if os.path.isfile(bitmappath):
staticbitmap = wx.StaticBitmap(id=-1, bitmap=wx.Bitmap(bitmappath),
name="%s_bitmap"%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(24, 24), style=0)
boxsizer.AddWindow(staticbitmap, 0, border=5, flag=wx.RIGHT)
statictext = wx.StaticText(id=-1, label="%s:"%element_infos["name"],
name="%s_label"%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
boxsizer.AddWindow(statictext, 0, border=4, flag=wx.TOP)
id = wx.NewId()
choicectrl = wx.Choice(id=id, name=element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(150, 25), style=0)
boxsizer.AddWindow(choicectrl, 0, border=0, flag=0)
choicectrl.Append("")
if len(element_infos["type"]) > 0 and isinstance(element_infos["type"][0], types.TupleType):
for choice, xsdclass in element_infos["type"]:
choicectrl.Append(choice)
staticbox = wx.StaticBox(id=-1, label="%(name)s - %(value)s"%element_infos,
name='%s_staticbox'%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(0, 0), style=0)
staticboxsizer = wx.StaticBoxSizer(staticbox, wx.VERTICAL)
sizer.AddSizer(staticboxsizer, 0, border=5, flag=wx.GROW|wx.BOTTOM)
self.RefreshSizerElement(parent, staticboxsizer, plugin, element_infos["children"], element_path, selected)
callback = self.GetChoiceContentCallBackFunction(choicectrl, staticboxsizer, plugin, element_path, selected)
else:
for choice in element_infos["type"]:
choicectrl.Append(choice)
callback = self.GetChoiceCallBackFunction(choicectrl, plugin, element_path, selected)
if element_infos["value"]:
choicectrl.SetStringSelection(element_infos["value"])
choicectrl.Bind(wx.EVT_CHOICE, callback, id=id)
elif isinstance(element_infos["type"], types.DictType):
boxsizer = wx.BoxSizer(wx.HORIZONTAL)
if first:
sizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.ALL)
else:
sizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM)
bitmappath = os.path.join("images", "%s.png"%element_infos["name"])
if os.path.isfile(bitmappath):
staticbitmap = wx.StaticBitmap(id=-1, bitmap=wx.Bitmap(bitmappath),
name="%s_bitmap"%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(24, 24), style=0)
boxsizer.AddWindow(staticbitmap, 0, border=5, flag=wx.RIGHT)
statictext = wx.StaticText(id=-1, label="%s:"%element_infos["name"],
name="%s_label"%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
boxsizer.AddWindow(statictext, 0, border=4, flag=wx.TOP)
id = wx.NewId()
scmin = -(2**31)
scmax = 2**31-1
if "min" in element_infos["type"]:
scmin = element_infos["type"]["min"]
if "max" in element_infos["type"]:
scmax = element_infos["type"]["max"]
spinctrl = wx.SpinCtrl(id=id, name=element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(150, 25), style=wx.SP_ARROW_KEYS|wx.ALIGN_RIGHT)
spinctrl.SetRange(scmin,scmax)
boxsizer.AddWindow(spinctrl, 0, border=0, flag=0)
spinctrl.SetValue(element_infos["value"])
spinctrl.Bind(wx.EVT_SPINCTRL, self.GetTextCtrlCallBackFunction(spinctrl, plugin, element_path, selected), id=id)
elif element_infos["type"] == "element":
staticbox = wx.StaticBox(id=-1, label=element_infos["name"],
name='%s_staticbox'%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(0, 0), style=0)
staticboxsizer = wx.StaticBoxSizer(staticbox, wx.VERTICAL)
if first:
sizer.AddSizer(staticboxsizer, 0, border=0, flag=wx.GROW|wx.TOP)
else:
sizer.AddSizer(staticboxsizer, 0, border=0, flag=wx.GROW)
self.RefreshSizerElement(parent, staticboxsizer, plugin, element_infos["children"], element_path, False, selected)
else:
boxsizer = wx.BoxSizer(wx.HORIZONTAL)
if first:
sizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.ALL)
else:
sizer.AddSizer(boxsizer, 0, border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM)
bitmappath = os.path.join("images", "%s.png"%element_infos["name"])
if os.path.isfile(bitmappath):
staticbitmap = wx.StaticBitmap(id=-1, bitmap=wx.Bitmap(bitmappath),
name="%s_bitmap"%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(24, 24), style=0)
boxsizer.AddWindow(staticbitmap, 0, border=5, flag=wx.RIGHT)
statictext = wx.StaticText(id=-1, label="%s:"%element_infos["name"],
name="%s_label"%element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)
boxsizer.AddWindow(statictext, 0, border=4, flag=wx.TOP)
id = wx.NewId()
if element_infos["type"] == "boolean":
checkbox = wx.CheckBox(id=id, name=element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(17, 25), style=0)
boxsizer.AddWindow(checkbox, 0, border=0, flag=0)
checkbox.SetValue(element_infos["value"])
checkbox.Bind(wx.EVT_CHECKBOX, self.GetCheckBoxCallBackFunction(checkbox, plugin, element_path, selected), id=id)
elif element_infos["type"] in ["unsignedLong", "long","integer"]:
if element_infos["type"].startswith("unsigned"):
scmin = 0
else:
scmin = -(2**31)
scmax = 2**31-1
spinctrl = wx.SpinCtrl(id=id, name=element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(150, 25), style=wx.SP_ARROW_KEYS|wx.ALIGN_RIGHT)
spinctrl.SetRange(scmin, scmax)
boxsizer.AddWindow(spinctrl, 0, border=0, flag=0)
spinctrl.SetValue(element_infos["value"])
spinctrl.Bind(wx.EVT_SPINCTRL, self.GetTextCtrlCallBackFunction(spinctrl, plugin, element_path, selected), id=id)
else:
textctrl = wx.TextCtrl(id=id, name=element_infos["name"], parent=parent,
pos=wx.Point(0, 0), size=wx.Size(150, 25), style=wx.TE_PROCESS_ENTER)
boxsizer.AddWindow(textctrl, 0, border=0, flag=0)
textctrl.SetValue(str(element_infos["value"]))
textctrl.Bind(wx.EVT_TEXT_ENTER, self.GetTextCtrlCallBackFunction(textctrl, plugin, element_path, selected), id=id)
textctrl.Bind(wx.EVT_KILL_FOCUS, self.GetTextCtrlCallBackFunction(textctrl, plugin, element_path, selected))
first = False
def OnNewProjectMenu(self, event):
defaultpath = self.PluginRoot.GetProjectPath()
if not defaultpath:
defaultpath = os.getcwd()
dialog = wx.DirDialog(self , "Choose a project", defaultpath, wx.DD_NEW_DIR_BUTTON)
if dialog.ShowModal() == wx.ID_OK:
projectpath = dialog.GetPath()
dialog.Destroy()
res = self.PluginRoot.NewProject(projectpath)
if not res :
self.RefreshPluginToolBar()
self.RefreshPluginTree()
self.RefreshButtons()
self.RefreshMainMenu()
else:
message = wx.MessageDialog(self, res, "ERROR", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
event.Skip()
def OnOpenProjectMenu(self, event):
defaultpath = self.PluginRoot.GetProjectPath()
if not defaultpath:
defaultpath = os.getcwd()
dialog = wx.DirDialog(self , "Choose a project", defaultpath, wx.DD_NEW_DIR_BUTTON)
if dialog.ShowModal() == wx.ID_OK:
projectpath = dialog.GetPath()
if os.path.isdir(projectpath):
result = self.PluginRoot.LoadProject(projectpath, self.Log)
if not result:
self.RefreshPluginToolBar()
self.RefreshPluginTree()
self.PluginTree.SelectItem(self.PluginTree.GetRootItem())
self.RefreshButtons()
self.RefreshMainMenu()
self.PluginTree.ScrollWindow(0, self.PluginTree.GetClientSize()[1])
else:
message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
else:
message = wx.MessageDialog(self, "\"%s\" folder is not a valid Beremiz project\n"%projectpath, "Error", wx.OK|wx.ICON_ERROR)
message.ShowModal()
message.Destroy()
dialog.Destroy()
event.Skip()
def OnCloseProjectMenu(self, event):
self.RefreshPluginToolBar()
self.RefreshPluginTree()
self.RefreshButtons()
self.RefreshMainMenu()
event.Skip()
def OnSaveProjectMenu(self, event):
if self.PluginRoot.HasProjectOpened():
self.PluginRoot.SaveProject()
event.Skip()
def OnPropertiesMenu(self, event):
event.Skip()
def OnQuitMenu(self, event):
self.Close()
event.Skip()
def OnEditPLCMenu(self, event):
self.EditPLC()
event.Skip()
def OnAddMenu(self, event):
self.AddPlugin()
event.Skip()
def OnDeleteMenu(self, event):
self.DeletePlugin()
event.Skip()
def OnBuildMenu(self, event):
#self.BuildAutom()
event.Skip()
def OnSimulateMenu(self, event):
event.Skip()
def OnRunMenu(self, event):
event.Skip()
def OnSaveLogMenu(self, event):
event.Skip()
def OnBeremizMenu(self, event):
event.Skip()
def OnAboutMenu(self, event):
event.Skip()
def OnAddButton(self, event):
PluginType = self.PluginChilds.GetStringSelection()
if PluginType != "":
self.AddPlugin(PluginType)
event.Skip()
def OnDeleteButton(self, event):
self.DeletePlugin()
event.Skip()
def CloseEditor(self, bus_id):
if self.BusManagers.get(bus_id, None) != None:
self.BusManagers[bus_id]["Editor"] = None
def GetAddButtonFunction(self, item, window):
def AddButtonFunction(event):
plugin = self.GetSelectedPlugin(item)
if plugin and len(plugin.PlugChildsTypes) > 0:
plugin_menu = wx.Menu(title='')
for name, XSDClass in plugin.PlugChildsTypes:
new_id = wx.NewId()
plugin_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=name)
self.Bind(wx.EVT_MENU, self._GetAddPluginFunction(name, item), id=new_id)
window_pos = window.GetPosition()
wx.CallAfter(self.PluginTree.PopupMenu, plugin_menu)
event.Skip()
return AddButtonFunction
def GetDeleteButtonFunction(self, item):
def DeleteButtonFunction(event):
wx.CallAfter(self.DeletePlugin, item)
event.Skip()
return DeleteButtonFunction
def AddPlugin(self, PluginType, selected = None):
dialog = wx.TextEntryDialog(self, "Please enter a name for plugin:", "Add Plugin", "", wx.OK|wx.CANCEL)
if dialog.ShowModal() == wx.ID_OK:
PluginName = dialog.GetValue()
plugin = self.GetSelectedPlugin(selected)
plugin.PlugAddChild(PluginName, PluginType, self.Log)
self.RefreshPluginTree()
dialog.Destroy()
def DeletePlugin(self, selected = None):
dialog = wx.MessageDialog(self, "Really delete plugin ?", "Remove plugin", wx.YES_NO|wx.NO_DEFAULT)
if dialog.ShowModal() == wx.ID_YES:
plugin = self.GetSelectedPlugin(selected)
plugin.PlugRemove()
del plugin
self.RefreshPluginTree()
dialog.Destroy()
#-------------------------------------------------------------------------------
# Exception Handler
#-------------------------------------------------------------------------------
Max_Traceback_List_Size = 20
def Display_Exception_Dialog(e_type,e_value,e_tb):
trcbck_lst = []
for i,line in enumerate(traceback.extract_tb(e_tb)):
trcbck = " " + str(i+1) + ". "
if line[0].find(os.getcwd()) == -1:
trcbck += "file : " + str(line[0]) + ", "
else:
trcbck += "file : " + str(line[0][len(os.getcwd()):]) + ", "
trcbck += "line : " + str(line[1]) + ", " + "function : " + str(line[2])
trcbck_lst.append(trcbck)
# Allow clicking....
cap = wx.Window_GetCapture()
if cap:
cap.ReleaseMouse()
dlg = wx.SingleChoiceDialog(None,
"""
An error happens.
Click on OK for saving an error report.
Please contact LOLITech at:
+33 (0)3 29 52 95 67
bugs_Beremiz@lolitech.fr
Error:
""" +
str(e_type) + " : " + str(e_value),
"Error",
trcbck_lst)
try:
res = (dlg.ShowModal() == wx.ID_OK)
finally:
dlg.Destroy()
return res
def Display_Error_Dialog(e_value):
message = wxMessageDialog(None, str(e_value), "Error", wxOK|wxICON_ERROR)
message.ShowModal()
message.Destroy()
def get_last_traceback(tb):
while tb.tb_next:
tb = tb.tb_next
return tb
def format_namespace(d, indent=' '):
return '\n'.join(['%s%s: %s' % (indent, k, repr(v)[:10000]) for k, v in d.iteritems()])
ignored_exceptions = [] # a problem with a line in a module is only reported once per session
def AddExceptHook(path, app_version='[No version]'):#, ignored_exceptions=[]):
def handle_exception(e_type, e_value, e_traceback):
traceback.print_exception(e_type, e_value, e_traceback) # this is very helpful when there's an exception in the rest of this func
last_tb = get_last_traceback(e_traceback)
ex = (last_tb.tb_frame.f_code.co_filename, last_tb.tb_frame.f_lineno)
if str(e_value).startswith("!!!"):
Display_Error_Dialog(e_value)
elif ex not in ignored_exceptions:
result = Display_Exception_Dialog(e_type,e_value,e_traceback)
if result:
ignored_exceptions.append(ex)
info = {
'app-title' : wx.GetApp().GetAppName(), # app_title
'app-version' : app_version,
'wx-version' : wx.VERSION_STRING,
'wx-platform' : wx.Platform,
'python-version' : platform.python_version(), #sys.version.split()[0],
'platform' : platform.platform(),
'e-type' : e_type,
'e-value' : e_value,
'date' : time.ctime(),
'cwd' : os.getcwd(),
}
if e_traceback:
info['traceback'] = ''.join(traceback.format_tb(e_traceback)) + '%s: %s' % (e_type, e_value)
last_tb = get_last_traceback(e_traceback)
exception_locals = last_tb.tb_frame.f_locals # the locals at the level of the stack trace where the exception actually occurred
info['locals'] = format_namespace(exception_locals)
if 'self' in exception_locals:
info['self'] = format_namespace(exception_locals['self'].__dict__)
output = open(path+os.sep+"bug_report_"+info['date'].replace(':','-').replace(' ','_')+".txt",'w')
lst = info.keys()
lst.sort()
for a in lst:
output.write(a+":\n"+str(info[a])+"\n\n")
#sys.excepthook = lambda *args: wx.CallAfter(handle_exception, *args)
sys.excepthook = handle_exception
if __name__ == '__main__':
def usage():
print "\nUsage of Beremiz.py :"
print "\n %s [Projectpath]\n"%sys.argv[0]
try:
opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
except getopt.GetoptError:
# print help information and exit:
usage()
sys.exit(2)
for o, a in opts:
if o in ("-h", "--help"):
usage()
sys.exit()
if len(args) > 1:
usage()
sys.exit()
elif len(args) == 1:
projectOpen = args[0]
else:
projectOpen = None
app = wx.PySimpleApp()
wx.InitAllImageHandlers()
# Install a exception handle for bug reports
AddExceptHook(os.getcwd(),__version__)
frame = Beremiz(None, projectOpen)
frame.Show()
app.MainLoop()