__version__ = "$Revision$"
import os, sys, getopt, wx, tempfile
from types import TupleType, StringType, UnicodeType
CWD = os.path.split(os.path.realpath(__file__))[0]
return os.path.join(CWD,*args)
if __name__ == '__main__':
print "\nUsage of LPCBeremiz.py :"
print "\n %s Projectpath Buildpath port\n"%sys.argv[0]
opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
except getopt.GetoptError:
# print help information and exit:
if o in ("-h", "--help"):
app.SetAppName('beremiz')
wx.InitAllImageHandlers()
# Import module for internationalization
if __name__ == '__main__':
__builtin__.__dict__['_'] = wx.GetTranslation#unicode_translation
from plugger import PluginsRoot, PlugTemplate, opjimg, connectors
from plcopen.structures import LOCATIONDATATYPES
from PLCControler import LOCATION_PLUGIN, LOCATION_MODULE, LOCATION_GROUP,\
LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY
from PLCOpenEditor import IDEFrame, ProjectDialog
#-------------------------------------------------------------------------------
# CANFESTIVAL PLUGIN HACK
#-------------------------------------------------------------------------------
from plugins.canfestival import canfestival
class LPC_canfestival_config:
def getCFLAGS(self, *args):
def getLDFLAGS(self, *args):
canfestival.local_canfestival_config = LPC_canfestival_config()
#-------------------------------------------------------------------------------
#-------------------------------------------------------------------------------
LOCATION_TYPES = {"I": LOCATION_VAR_INPUT,
"Q": LOCATION_VAR_OUTPUT,
"M": LOCATION_VAR_MEMORY}
LOCATION_DIRS = dict([(dir, size) for size, dir in LOCATION_TYPES.iteritems()])
for size, types in LOCATIONDATATYPES.iteritems():
LOCATION_SIZES[type] = size
def _GetModuleChildren(module):
for child in module["children"]:
if child["type"] == LOCATION_GROUP:
children.extend(child["children"])
def _GetVariables(module):
for child in module["children"]:
if child["type"] in [LOCATION_GROUP, LOCATION_MODULE]:
variables.extend(_GetVariables(child))
def _GetLastModuleGroup(module):
for child in module["children"]:
if child["type"] == LOCATION_GROUP:
def _GetModuleBySomething(module, something, toks):
for child in _GetModuleChildren(module):
if child.get(something) == toks[0]:
return _GetModuleBySomething(child, something, toks[1:])
def _GetModuleVariable(module, location, direction):
for child in _GetModuleChildren(module):
if child["location"] == location and child["type"] == LOCATION_TYPES[direction]:
def _RemoveModuleChild(module, child):
if child in module["children"]:
module["children"].remove(child)
for group in module["children"]:
if group["type"] == LOCATION_GROUP and child in group["children"]:
group["children"].remove(child)
BUS_TEXT = """/* Code generated by LPCBus plugin */
/* LPCBus plugin includes */
/* LPCBus plugin user variables definition */
/* LPCBus plugin functions */
int __init_%(location_str)s(int argc,char **argv)
void __cleanup_%(location_str)s(void)
void __retrieve_%(location_str)s(void)
void __publish_%(location_str)s(void)
self.VariableLocationTree = []
self.ResetUsedLocations()
def __getitem__(self, key):
return self.VariableLocationTree
raise KeyError, "Only 'children' key is available"
def _GetChildBySomething(self, something, toks):
return _GetModuleBySomething({"children" : self.VariableLocationTree}, something, toks)
return self.GetPlugRoot().GetBaseTypes()
def GetSizeOfType(self, type):
return LOCATION_SIZES[self.GetPlugRoot().GetBaseType(type)]
def _GetVariableLocationTree(self, current_location, infos):
if infos["type"] == LOCATION_MODULE:
location = current_location + (infos["IEC_Channel"],)
return {"name": infos["name"],
"location": ".".join(map(str, location + ("x",))),
"children": [self._GetVariableLocationTree(location, child) for child in infos["children"]]}
elif infos["type"] == LOCATION_GROUP:
return {"name": infos["name"],
"children": [self._GetVariableLocationTree(current_location, child) for child in infos["children"]]}
size = self.GetSizeOfType(infos["IEC_type"])
location = "%" + LOCATION_DIRS[infos["type"]] + size + ".".join(map(str, current_location + infos["location"]))
return {"name": infos["name"],
"IEC_type": infos["IEC_type"],
"description": infos["description"],
def GetVariableLocationTree(self):
return {"name": self.BaseParams.getName(),
"location": self.GetFullIEC_Channel(),
"children": [self._GetVariableLocationTree(self.GetCurrentLocation(), child)
for child in self.VariableLocationTree]}
def PlugTestModified(self):
def PlugRequestSave(self):
def ResetUsedLocations(self):
def _AddUsedLocation(self, parent, location):
if not parent.has_key(num):
parent[num] = {"used": False, "children": {}}
self._AddUsedLocation(parent[num]["children"], location)
parent[num]["used"] = True
def AddUsedLocation(self, location):
self._AddUsedLocation(self.UsedLocations, list(location))
def _CheckLocationConflicts(self, parent, location):
if not parent.has_key(num):
return self._CheckLocationConflicts(parent[num]["children"], location)
elif len(parent[num]["children"]) > 0:
def CheckLocationConflicts(self, location):
return self._CheckLocationConflicts(self.UsedLocations, list(location))
def PlugGenerate_C(self, buildpath, locations):
@param current_location: Tupple containing plugin IEC location : %I0.0.4.5 => (0,0,4,5)
@param locations: List of complete variables locations \
[{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)
"NAME" : name of the variable (generally "__IW0_1_2" style)
"DIR" : direction "Q","I" or "M"
"SIZE" : size "X", "B", "W", "D", "L"
"LOC" : tuple of interger for IEC location (0,1,2,...)
@return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND
current_location = self.GetCurrentLocation()
# define a unique name for the generated C file
location_str = "_".join(map(str, current_location))
code_str = {"location_str": location_str,
for module in _GetModuleChildren(self):
code_str["init_code"] += " %s\n" % module["init"]
self.ResetUsedLocations()
for location in locations:
loc = location["LOC"][len(current_location):]
while next is not None and i < len(loc):
next = self._GetChildBySomething("IEC_Channel", loc[:i + 1])
for variable in _GetModuleChildren(group):
if variable["location"] == var_loc:
# if location["DIR"] != LOCATION_DIRS[variable["type"]]:
# raise Exception, "Direction conflict in variable definition"
# if location["IEC_TYPE"] != variable["IEC_type"]:
# raise Exception, "Type conflict in variable definition"
if location["DIR"] == "Q":
if self.CheckLocationConflicts(location["LOC"]):
raise Exception, "BYTE and BIT from the same BYTE can't be used together"
self.AddUsedLocation(location["LOC"])
vars.append({"location": location["NAME"],
"Type": variable["IEC_type"],
"Retrieve": variable["retrieve"],
"Publish": variable["publish"],
base_types = self.GetPlugRoot().GetBaseTypes()
if var["Type"] in base_types:
code_str["var_decl"] += "%s%s beremiz%s;\n"%(prefix, var["Type"], var["location"])
code_str["var_decl"] += "%s%s *%s = &beremiz%s;\n"%(prefix, var["Type"], var["location"], var["location"])
if var["Retrieve"] != "":
code_str["retrieve_code"] += " " + var["Retrieve"] % ("*" + var["location"]) + "\n"
code_str["publish_code"] += " " + var["Publish"] % ("*" + var["location"]) + "\n"
Gen_Module_path = os.path.join(buildpath, "Bus_%s.c"%location_str)
module = open(Gen_Module_path,'w')
module.write(BUS_TEXT % code_str)
matiec_flags = '"-I%s"'%os.path.abspath(self.GetPlugRoot().GetIECLibPath())
return [(Gen_Module_path, matiec_flags)],"",True
#-------------------------------------------------------------------------------
#-------------------------------------------------------------------------------
def mycopytree(src, dst):
Copy content of a directory to an other, omit hidden files
@param src: source directory
@param dst: destination directory
for i in os.listdir(src):
if not i.startswith('.'):
srcpath = os.path.join(src,i)
dstpath = os.path.join(dst,i)
if os.path.isdir(srcpath):
if os.path.exists(dstpath):
mycopytree(srcpath, dstpath)
elif os.path.isfile(srcpath):
shutil.copy2(srcpath, dstpath)
[SIMULATION_MODE, TRANSFER_MODE] = range(2)
class LPCPluginsRoot(PluginsRoot):
{"bitmap" : opjimg("Debug"),
"tooltip" : _("Simulate PLC"),
{"bitmap" : opjimg("Run"),
"tooltip" : _("Start PLC"),
{"bitmap" : opjimg("Stop"),
"tooltip" : _("Stop Running PLC"),
{"bitmap" : opjimg("Build"),
"tooltip" : _("Build project into build folder"),
{"bitmap" : opjimg("Transfer"),
"tooltip" : _("Transfer PLC"),
def __init__(self, frame, logger):
PluginsRoot.__init__(self, frame, logger)
self.PlugChildsTypes += [("LPCBus", LPCBus, "LPC bus")]
self.LPCConnector = False
self.previous_mode = None
self.SimulationBuildPath = None
self.AbortTransferTimer = None
def GetProjectName(self):
return self.Project.getname()
def GetDefaultTargetName(self):
if self.CurrentMode == SIMULATION_MODE:
return PluginsRoot.GetDefaultTargetName(self)
target = PluginsRoot.GetTarget(self)
if self.CurrentMode != SIMULATION_MODE:
target.getcontent()["value"].setBuildPath(self.BuildPath)
if self.CurrentMode == SIMULATION_MODE:
if self.SimulationBuildPath is None:
self.SimulationBuildPath = os.path.join(tempfile.mkdtemp(), os.path.basename(self.ProjectPath), "build")
return self.SimulationBuildPath
return PluginsRoot._getBuildPath(self)
if self.BuildPath is not None:
mycopytree(self.OrigBuildPath, self.BuildPath)
def SetProjectName(self, name):
return self.Project.setname(name)
def SetOnlineMode(self, mode, path=None):
if self.OnlineMode != mode.upper():
self.OnlineMode = mode.upper()
if self.OnlineMode != "OFF":
uri = "LPC://%s/%s" % (self.OnlineMode,path)
self.LPCConnector = connectors.ConnectorFactory(uri, self)
self.logger.write_error(_("Exception while connecting %s!\n")%uri)
self.logger.write_error(traceback.format_exc())
# Did connection success ?
if self.LPCConnector is None:
self.logger.write_error(_("Connection failed to %s!\n")%uri)
def ApplyOnlineMode(self):
if self.CurrentMode != SIMULATION_MODE:
self._connector = self.LPCConnector
# Init with actual PLC status and print it
self.UpdateMethodsFromPLCStatus()
if self.LPCConnector is not None and self.OnlineMode == "APPLICATION":
self.CompareLocalAndRemotePLC()
if self.previous_plcstate is not None:
status = _(self.previous_plcstate)
self.logger.write(_("PLC is %s\n")%status)
if self.StatusTimer and not self.StatusTimer.IsRunning():
self.StatusTimer.Start(milliseconds=500, oneShot=False)
if self.previous_plcstate=="Started":
if self.DebugAvailable() and self.GetIECProgramsAndVariables():
self.logger.write(_("Debug connect matching running PLC\n"))
self.logger.write_warning(_("Debug do not match PLC - stop/transfert/start to re-enable\n"))
elif self.StatusTimer and self.StatusTimer.IsRunning():
if self.CurrentMode == TRANSFER_MODE:
if self.OnlineMode == "BOOTLOADER":
elif self.OnlineMode == "APPLICATION":
self.AbortTransferTimer.Stop()
self.AbortTransferTimer = None
self.logger.write(_("PLC transferred successfully\n"))
# Update a PLCOpenEditor Pou variable name
def UpdateProjectVariableName(self, old_name, new_name):
self.Project.updateElementName(old_name, new_name)
def RemoveProjectVariableByAddress(self, address):
self.Project.removeVariableByAddress(address)
def RemoveProjectVariableByFilter(self, leading):
self.Project.removeVariableByFilter(leading)
def LoadProject(self, ProjectPath, BuildPath=None):
Load a project contained in a folder
@param ProjectPath: path of the project folder
if os.path.basename(ProjectPath) == "":
ProjectPath = os.path.dirname(ProjectPath)
# Verify that project contains a PLCOpen program
plc_file = os.path.join(ProjectPath, "plc.xml")
if os.path.isfile(plc_file):
result = self.OpenXMLFile(plc_file)
self.CreateNewProject({"companyName": "",
# Change XSD into class members
# Keep track of the root plugin (i.e. project path)
self.ProjectPath = ProjectPath
self.BuildPath = self._getBuildPath()
self.OrigBuildPath = BuildPath
if BuildPath is not None:
mycopytree(BuildPath, self.BuildPath)
# If dir have already be made, and file exist
if os.path.isdir(self.PlugPath()) and os.path.isfile(self.PluginXmlFilePath()):
#Load the plugin.xml file into parameters members
result = self.LoadXMLParams()
#Load and init all the childs
if self.PlugTestModified():
self.RefreshPluginsBlockLists()
wx.CallAfter(self.RefreshPluginsBlockLists)
############# Real PLC object access #############
def UpdateMethodsFromPLCStatus(self):
# Get PLC state : Running or Stopped
# TODO : use explicit status instead of boolean
if self.OnlineMode == "OFF":
elif self.OnlineMode == "BOOTLOADER":
if self._connector is not None:
status = self._connector.GetPLCstatus()
if self.previous_plcstate != status or self.previous_mode != self.CurrentMode:
simulating = self.CurrentMode == SIMULATION_MODE
"Started" : [("_Simulate", False),
"Stopped" : [("_Simulate", False),
"Connected" : [("_Simulate", not simulating),
"Disconnected" :[("_Simulate", not simulating),
self.previous_plcstate = status
self.previous_mode = self.CurrentMode
def Generate_plc_declare_locations(self):
Declare used locations in order to simulatePLC in a black box
return """#include "iec_types_all.h"
#define __LOCATED_VAR(type, name, ...) \
type *name = &beremiz_##name;
#include "LOCATED_VARIABLES.h"
Method called by user to Simulate PLC
self.CurrentMode = SIMULATION_MODE
self._connector = connectors.ConnectorFactory(uri, self)
self.logger.write_error(_("Exception while connecting %s!\n")%uri)
self.logger.write_error(traceback.format_exc())
# Did connection success ?
if self._connector is None:
self.logger.write_error(_("Connection failed to %s!\n")%uri)
buildpath = self._getBuildPath()
# Eventually create build dir
if not os.path.exists(buildpath):
# Generate SoftPLC IEC code
IECGenRes = self._Generate_SoftPLC()
# If IEC code gen fail, bail out.
self.logger.write_error(_("IEC-61131-3 code generation failed !\n"))
# Reset variable and program list that are parsed from
# CSV file generated by IEC2C compiler.
self.ResetIECProgramsAndVariables()
gen_result = self.PlugGenerate_C(buildpath, self.PLCGeneratedLocatedVars)
PlugCFilesAndCFLAGS, PlugLDFLAGS, DoCalls = gen_result[:3]
# if some files have been generated put them in the list with their location
self.LocationCFilesAndCFLAGS = [(self.GetCurrentLocation(), PlugCFilesAndCFLAGS, DoCalls)]
self.LocationCFilesAndCFLAGS = []
# plugin asks for some LDFLAGS
# LDFLAGS can be either string
if type(PlugLDFLAGS)==type(str()):
self.LDFLAGS=[PlugLDFLAGS]
elif type(PlugLDFLAGS)==type(list()):
self.LDFLAGS=PlugLDFLAGS[:]
# Template based part of C code generation
# files are stacked at the beginning, as files of plugin tree root
for generator, filename, name in [
(self.Generate_plc_debugger, "plc_debugger.c", "Debugger"),
# init/cleanup/retrieve/publish, run and align code
(self.Generate_plc_common_main,"plc_common_main.c","Common runtime"),
# declare located variables for simulate in a black box
(self.Generate_plc_declare_locations,"plc_declare_locations.c","Declare Locations")]:
code_path = os.path.join(buildpath,filename)
open(code_path, "w").write(code)
# Insert this file as first file to be compiled at root plugin
self.LocationCFilesAndCFLAGS[0][1].insert(0,(code_path, self.plcCFLAGS))
self.logger.write_error(name+_(" generation failed !\n"))
self.logger.write_error(traceback.format_exc())
builder = self.GetBuilder()
self.logger.write_error(_("Fatal : cannot get builder.\n"))
self.logger.write_error(_("C Build failed.\n"))
self.logger.write_error(_("C Build crashed !\n"))
self.logger.write_error(traceback.format_exc())
data = builder.GetBinaryCode()
if self._connector.NewPLC(builder.GetBinaryCodeMD5(), data, []):
if self.AppFrame is not None:
self.AppFrame.CloseDebugTabs()
self.AppFrame.RefreshInstancesTree()
self.UnsubscribeAllDebugIECVariable()
self.ProgramTransferred()
self.logger.write(_("Transfer completed successfully.\n"))
self.logger.write_error(_("Transfer failed\n"))
if not self.StatusTimer.IsRunning():
self.StatusTimer.Start(milliseconds=500, oneShot=False)
def StopSimulation(self):
if self.CurrentMode == SIMULATION_MODE:
def CompareLocalAndRemotePLC(self):
if self.LPCConnector is None:
# We are now connected. Update button status
MD5 = self.GetLastBuildMD5()
# Check remote target PLC correspondance to that md5
if MD5 is not None and self.LPCConnector.MatchMD5(MD5):
# warns controller that program match
self.ProgramTransferred()
if self.CurrentMode is None and self.OnlineMode != "OFF":
self.CurrentMode = TRANSFER_MODE
ID_ABORTTRANSFERTIMER = wx.NewId()
self.AbortTransferTimer = wx.Timer(self.AppFrame, ID_ABORTTRANSFERTIMER)
self.AppFrame.Bind(wx.EVT_TIMER, self.AbortTransfer, self.AbortTransferTimer)
if self.OnlineMode == "BOOTLOADER":
self.logger.write(_("Resetting PLC\n"))
self.LPCConnector.ResetPLC()
self.AbortTransferTimer.Start(milliseconds=5000, oneShot=True)
self.logger.write(_("Start PLC transfer\n"))
self.AbortTransferTimer.Stop()
PluginsRoot._Transfer(self)
self.AbortTransferTimer.Start(milliseconds=5000, oneShot=True)
def AbortTransfer(self, event):
self.logger.write_warning(_("Timeout waiting PLC to recover\n"))
self.AbortTransferTimer.Stop()
self.AbortTransferTimer = None
#-------------------------------------------------------------------------------
#-------------------------------------------------------------------------------
class LPCBeremiz(Beremiz):
def _init_coll_FileMenu_Items(self, parent):
AppendMenu(parent, help='', id=wx.ID_SAVE,
kind=wx.ITEM_NORMAL, text=_(u'Save\tCTRL+S'))
AppendMenu(parent, help='', id=wx.ID_CLOSE,
kind=wx.ITEM_NORMAL, text=_(u'Close Tab\tCTRL+W'))
AppendMenu(parent, help='', id=wx.ID_PAGE_SETUP,
kind=wx.ITEM_NORMAL, text=_(u'Page Setup'))
AppendMenu(parent, help='', id=wx.ID_PREVIEW,
kind=wx.ITEM_NORMAL, text=_(u'Preview'))
AppendMenu(parent, help='', id=wx.ID_PRINT,
kind=wx.ITEM_NORMAL, text=_(u'Print'))
AppendMenu(parent, help='', id=wx.ID_PROPERTIES,
kind=wx.ITEM_NORMAL, text=_(u'Properties'))
AppendMenu(parent, help='', id=wx.ID_EXIT,
kind=wx.ITEM_NORMAL, text=_(u'Quit\tCTRL+Q'))
self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=wx.ID_SAVE)
self.Bind(wx.EVT_MENU, self.OnCloseTabMenu, id=wx.ID_CLOSE)
self.Bind(wx.EVT_MENU, self.OnPageSetupMenu, id=wx.ID_PAGE_SETUP)
self.Bind(wx.EVT_MENU, self.OnPreviewMenu, id=wx.ID_PREVIEW)
self.Bind(wx.EVT_MENU, self.OnPrintMenu, id=wx.ID_PRINT)
self.Bind(wx.EVT_MENU, self.OnPropertiesMenu, id=wx.ID_PROPERTIES)
self.Bind(wx.EVT_MENU, self.OnQuitMenu, id=wx.ID_EXIT)
def _init_ctrls(self, prnt):
IDEFrame._init_ctrls(self, prnt)
self.Bind(wx.EVT_MENU, self.OnOpenWidgetInspector, id=ID_BEREMIZINSPECTOR)
accel = wx.AcceleratorTable([wx.AcceleratorEntry(wx.ACCEL_CTRL|wx.ACCEL_ALT, ord('I'), ID_BEREMIZINSPECTOR)])
self.SetAcceleratorTable(accel)
self.PLCConfig = wx.ScrolledWindow(id=ID_BEREMIZPLCCONFIG,
name='PLCConfig', parent=self.LeftNoteBook, pos=wx.Point(0, 0),
size=wx.Size(-1, -1), style=wx.TAB_TRAVERSAL|wx.SUNKEN_BORDER|wx.HSCROLL|wx.VSCROLL)
self.PLCConfig.SetBackgroundColour(wx.WHITE)
self.PLCConfig.Bind(wx.EVT_LEFT_DOWN, self.OnPanelLeftDown)
self.PLCConfig.Bind(wx.EVT_SIZE, self.OnMoveWindow)
self.LeftNoteBook.InsertPage(0, self.PLCConfig, _("Topology"), True)
self.LogConsole = wx.TextCtrl(id=ID_BEREMIZLOGCONSOLE, value='',
name='LogConsole', parent=self.BottomNoteBook, pos=wx.Point(0, 0),
size=wx.Size(0, 0), style=wx.TE_MULTILINE|wx.TE_RICH2)
self.LogConsole.Bind(wx.EVT_LEFT_DCLICK, self.OnLogConsoleDClick)
self.BottomNoteBook.AddPage(self.LogConsole, _("Log Console"))
self._init_beremiz_sizers()
def OnCloseFrame(self, event):
self.PluginRoot.ResetAppFrame(lpcberemiz_cmd.Log)
if self.PluginRoot.OnlineMode == 0:
self.PluginRoot._connector = None
self.PluginRoot.KillDebugThread()
lpcberemiz_cmd.Log.write("Closed\n")
def ShowProperties(self):
old_values = self.Controler.GetProjectProperties()
dialog = ProjectDialog(self ,False)
dialog.SetValues(old_values)
if dialog.ShowModal() == wx.ID_OK:
new_values = dialog.GetValues()
new_values["creationDateTime"] = old_values["creationDateTime"]
if new_values != old_values:
self.Controler.SetProjectProperties(None, new_values)
self._Refresh(TITLE, TOOLBAR, FILEMENU, EDITMENU, DISPLAYMENU,
TYPESTREE, INSTANCESTREE, SCALING)
def RefreshFileMenu(self):
if self.PluginRoot is not None:
selected = self.TabsOpened.GetSelection()
graphic_viewer = isinstance(self.TabsOpened.GetPage(selected), Viewer)
if self.TabsOpened.GetPageCount() > 0:
self.FileMenu.Enable(wx.ID_CLOSE, True)
self.FileMenu.Enable(wx.ID_PREVIEW, True)
self.FileMenu.Enable(wx.ID_PRINT, True)
self.FileMenu.Enable(wx.ID_PREVIEW, False)
self.FileMenu.Enable(wx.ID_PRINT, False)
self.FileMenu.Enable(wx.ID_CLOSE, False)
self.FileMenu.Enable(wx.ID_PREVIEW, False)
self.FileMenu.Enable(wx.ID_PRINT, False)
self.FileMenu.Enable(wx.ID_PAGE_SETUP, True)
self.FileMenu.Enable(wx.ID_SAVE, self.PluginRoot.PlugTestModified())
self.FileMenu.Enable(wx.ID_PROPERTIES, True)
self.FileMenu.Enable(wx.ID_CLOSE, False)
self.FileMenu.Enable(wx.ID_PAGE_SETUP, False)
self.FileMenu.Enable(wx.ID_PREVIEW, False)
self.FileMenu.Enable(wx.ID_PRINT, False)
self.FileMenu.Enable(wx.ID_SAVE, False)
self.FileMenu.Enable(wx.ID_PROPERTIES, False)
def RefreshPLCParams(self):
self.ClearSizer(self.PLCParamsSizer)
if self.PluginRoot is not None:
plcwindow = wx.Panel(self.PLCConfig, -1, size=wx.Size(-1, -1))
if self.PluginRoot.PlugTestModified():
bkgdclr = CHANGED_TITLE_COLOUR
if self.PluginRoot not in self.PluginInfos:
self.PluginInfos[self.PluginRoot] = {"right_visible" : False}
plcwindow.SetBackgroundColour(TITLE_COLOUR)
plcwindow.Bind(wx.EVT_LEFT_DOWN, self.OnPanelLeftDown)
self.PLCParamsSizer.AddWindow(plcwindow, 0, border=0, flag=wx.GROW)
plcwindowsizer = wx.BoxSizer(wx.HORIZONTAL)
plcwindow.SetSizer(plcwindowsizer)
st = wx.StaticText(plcwindow, -1)
st.SetFont(wx.Font(faces["size"], wx.DEFAULT, wx.NORMAL, wx.BOLD, faceName = faces["helv"]))
st.SetLabel(self.PluginRoot.GetProjectName())
plcwindowsizer.AddWindow(st, 0, border=5, flag=wx.ALL|wx.ALIGN_CENTER)
plcwindowmainsizer = wx.BoxSizer(wx.VERTICAL)
plcwindowsizer.AddSizer(plcwindowmainsizer, 0, border=5, flag=wx.ALL)
plcwindowbuttonsizer = wx.BoxSizer(wx.HORIZONTAL)
plcwindowmainsizer.AddSizer(plcwindowbuttonsizer, 0, border=0, flag=wx.ALIGN_CENTER)
msizer = self.GenerateMethodButtonSizer(self.PluginRoot, plcwindow, not self.PluginInfos[self.PluginRoot]["right_visible"])
plcwindowbuttonsizer.AddSizer(msizer, 0, border=0, flag=wx.GROW)
self.PLCConfigMainSizer.Layout()
def GenerateTreeBranch(self, plugin):
leftwindow = wx.Panel(self.PLCConfig, -1, size=wx.Size(-1, -1))
if plugin.PlugTestModified():
bkgdclr=CHANGED_WINDOW_COLOUR
leftwindow.SetBackgroundColour(bkgdclr)
if plugin not in self.PluginInfos:
self.PluginInfos[plugin] = {"expanded" : False, "left_visible" : False, "right_visible" : False}
self.PluginInfos[plugin]["children"] = plugin.IECSortedChilds()
plugin_infos = plugin.GetVariableLocationTree()
if len(self.PluginInfos[plugin]["children"]) == 0:
plugin_locations = plugin_infos["children"]
if not self.PluginInfos[plugin].has_key("locations_infos"):
self.PluginInfos[plugin]["locations_infos"] = {"root": {"expanded" : False}}
self.PluginInfos[plugin]["locations_infos"]["root"]["children"] = []
self.PluginTreeSizer.AddWindow(leftwindow, 0, border=0, flag=wx.GROW)
leftwindowsizer = wx.BoxSizer(wx.HORIZONTAL)
leftwindow.SetSizer(leftwindowsizer)
st = wx.StaticText(leftwindow, -1)
st.SetFont(wx.Font(faces["size"], wx.DEFAULT, wx.NORMAL, wx.BOLD, faceName = faces["helv"]))
st.SetLabel(plugin.GetFullIEC_Channel())
leftwindowsizer.AddWindow(st, 0, border=5, flag=wx.RIGHT)
expandbutton_id = wx.NewId()
expandbutton = wx.lib.buttons.GenBitmapToggleButton(id=expandbutton_id, bitmap=wx.Bitmap(Bpath( 'images', 'plus.png')),
name='ExpandButton', parent=leftwindow, pos=wx.Point(0, 0),
size=wx.Size(13, 13), style=wx.NO_BORDER)
expandbutton.labelDelta = 0
expandbutton.SetBezelWidth(0)
expandbutton.SetUseFocusIndicator(False)
expandbutton.SetBitmapSelected(wx.Bitmap(Bpath( 'images', 'minus.png')))
if len(self.PluginInfos[plugin]["children"]) > 0:
expandbutton.SetToggle(self.PluginInfos[plugin]["expanded"])
if expandbutton.GetToggle():
self.ExpandPlugin(plugin)
self.CollapsePlugin(plugin)
self.PluginInfos[plugin]["expanded"] = expandbutton.GetToggle()
self.PLCConfigMainSizer.Layout()
expandbutton.Bind(wx.EVT_BUTTON, togglebutton, id=expandbutton_id)
elif len(plugin_locations) > 0:
locations_infos = self.PluginInfos[plugin]["locations_infos"]
expandbutton.SetToggle(locations_infos["root"]["expanded"])
if expandbutton.GetToggle():
self.ExpandLocation(locations_infos, "root")
self.CollapseLocation(locations_infos, "root")
self.PluginInfos[plugin]["expanded"] = expandbutton.GetToggle()
locations_infos["root"]["expanded"] = expandbutton.GetToggle()
self.PLCConfigMainSizer.Layout()
expandbutton.Bind(wx.EVT_BUTTON, togglebutton, id=expandbutton_id)
expandbutton.Enable(False)
leftwindowsizer.AddWindow(expandbutton, 0, border=5, flag=wx.RIGHT|wx.ALIGN_CENTER_VERTICAL)
sb = wx.StaticBitmap(leftwindow, -1)
icon = plugin_infos.get("icon", None)
icon = os.path.join(base_folder, "plcopeneditor", 'Images', '%s.png' % self.LOCATION_BITMAP[plugin_infos["type"]])
sb.SetBitmap(wx.Bitmap(icon))
leftwindowsizer.AddWindow(sb, 0, border=5, flag=wx.RIGHT|wx.ALIGN_CENTER_VERTICAL)
st = wx.StaticText(leftwindow, st_id, size=wx.DefaultSize, style=wx.NO_BORDER)
st.SetFont(wx.Font(faces["size"] * 0.75, wx.DEFAULT, wx.NORMAL, wx.BOLD, faceName = faces["helv"]))
st.SetLabel(plugin.MandatoryParams[1].getName())
leftwindowsizer.AddWindow(st, 0, border=5, flag=wx.RIGHT|wx.ALIGN_CENTER_VERTICAL)
rightwindow = wx.Panel(self.PLCConfig, -1, size=wx.Size(-1, -1))
rightwindow.SetBackgroundColour(bkgdclr)
self.PluginTreeSizer.AddWindow(rightwindow, 0, border=0, flag=wx.GROW)
self.PluginInfos[plugin]["left"] = leftwindow
self.PluginInfos[plugin]["right"] = rightwindow
for child in self.PluginInfos[plugin]["children"]:
self.GenerateTreeBranch(child)
if not self.PluginInfos[child]["expanded"]:
self.CollapsePlugin(child)
if len(plugin_locations) > 0:
locations_infos = self.PluginInfos[plugin]["locations_infos"]
for location in plugin_locations:
locations_infos["root"]["children"].append("root.%s" % location["name"])
self.GenerateLocationTreeBranch(locations_infos, "root", location)
if not locations_infos["root"]["expanded"]:
self.CollapseLocation(locations_infos, "root")
def __init__(self, port):
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket.connect(('localhost', port))
idx = self.Buffer.find("\n")
self.Buffer += self.socket.recv(2048)
idx = self.Buffer.find("\n")
line = self.Buffer[:idx+1]
self.Buffer = self.Buffer[idx+1:]
""" Base class for file like objects to facilitate StdOut for the Shell."""
def write(self, s, style = None):
def write_warning(self, s):
def write_error(self, s):
if __name__ == '__main__':
from threading import Thread, Timer, Semaphore
wx_eval_lock = Semaphore(0)
def wx_evaluator(callable, *args, **kwargs):
eval_res=callable(*args,**kwargs)
def evaluator(callable, *args, **kwargs):
wx.CallAfter(wx_evaluator,callable,*args,**kwargs)
# Command log for debug, for viewing from wxInspector
#cmdlogf=open("bmzcmdlog.txt","w")
class LPCBeremiz_Cmd(cmd.Cmd):
def __init__(self, PluginRoot, Log):
cmd.Cmd.__init__(self, stdin=Log, stdout=Log)
self.use_rawinput = False
self.PluginRoot = PluginRoot
if self.RefreshTimer is not None:
self.RefreshTimer.cancel()
self.RefreshTimer = Timer(0.1, wx.CallAfter, args = [self.Refresh])
self.RefreshTimer.start()
self.PluginRoot.SetAppFrame(frame, frame.Log)
frame._Refresh(TITLE, INSTANCESTREE, FILEMENU, EDITMENU)
self.PluginRoot.ResetAppFrame(self.Log)
def SetProjectProperties(self, projectname, productname, productversion, companyname):
properties = self.PluginRoot.GetProjectProperties()
properties["projectName"] = projectname
properties["productName"] = productname
properties["productVersion"] = productversion
properties["companyName"] = companyname
self.PluginRoot.SetProjectProperties(properties=properties)
def SetOnlineMode(self, mode, path=None):
self.PluginRoot.SetOnlineMode(mode, path)
def AddBus(self, iec_channel, name, icon=None):
for child in self.PluginRoot.IterChilds():
if child.BaseParams.getName() == name:
return "Error: A bus named %s already exists\n" % name
elif child.BaseParams.getIEC_Channel() == iec_channel:
return "Error: A bus with IEC_channel %d already exists\n" % iec_channel
bus = self.PluginRoot.PlugAddChild(name, "LPCBus", iec_channel)
return "Error: Unable to create bus\n"
def RenameBus(self, iec_channel, name):
bus = self.PluginRoot.GetChildByIECLocation((iec_channel,))
return "Error: No bus found\n"
for child in self.PluginRoot.IterChilds():
if child != bus and child.BaseParams.getName() == name:
return "Error: A bus named %s already exists\n" % name
bus.BaseParams.setName(name)
def ChangeBusIECChannel(self, old_iec_channel, new_iec_channel):
bus = self.PluginRoot.GetChildByIECLocation((old_iec_channel,))
return "Error: No bus found\n"
for child in self.PluginRoot.IterChilds():
if child != bus and child.BaseParams.getIEC_Channel() == new_iec_channel:
return "Error: A bus with IEC_channel %d already exists\n" % new_iec_channel
self.PluginRoot.UpdateProjectVariableLocation(str(old_iec_channel),
self.PluginRoot.UpdateProjectVariableLocation(
bus.BaseParams.setIEC_Channel(new_iec_channel)
def RemoveBus(self, iec_channel):
bus = self.PluginRoot.GetChildByIECLocation((iec_channel,))
return "Error: No bus found\n"
self.PluginRoot.RemoveProjectVariableByFilter(str(iec_channel))
self.PluginRoot.PluggedChilds["LPCBus"].remove(bus)
def AddModule(self, parent, iec_channel, name, icode, icon=None):
module = self.PluginRoot.GetChildByIECLocation(parent)
return "Error: No parent found\n"
for child in _GetModuleChildren(module):
if child["name"] == name:
return "Error: A module named %s already exists\n" % name
elif child["IEC_Channel"] == iec_channel:
return "Error: A module with IEC_channel %d already exists\n" % iec_channel
_GetLastModuleGroup(module).append({"name": name,
"IEC_Channel": iec_channel,
def RenameModule(self, iec_location, name):
module = self.PluginRoot.GetChildByIECLocation(iec_location)
return "Error: No module found\n"
parent = self.PluginRoot.GetChildByIECLocation(iec_location[:-1])
if parent is self.PluginRoot:
return "Error: No module found\n"
if module["name"] != name:
for child in _GetModuleChildren(parent):
if child["name"] == name:
return "Error: A module named %s already exists\n" % name
def ChangeModuleIECChannel(self, old_iec_location, new_iec_channel):
module = self.PluginRoot.GetChildByIECLocation(old_iec_location)
return "Error: No module found\n"
parent = self.PluginRoot.GetChildByIECLocation(old_iec_location[:-1])
if parent is self.PluginRoot:
return "Error: No module found\n"
if module["IEC_Channel"] != new_iec_channel:
for child in _GetModuleChildren(parent):
if child["IEC_Channel"] == new_iec_channel:
return "Error: A module with IEC_channel %d already exists\n" % new_iec_channel
self.PluginRoot.UpdateProjectVariableLocation(".".join(map(str, old_iec_location)), ".".join(map(str, old_iec_location[:1] + (new_iec_channel,))))
module["IEC_Channel"] = new_iec_channel
def ChangeModuleInitCode(self, iec_location, icode):
module = self.PluginRoot.GetChildByIECLocation(iec_location)
return "Error: No module found\n"
def RemoveModule(self, parent, iec_channel):
module = self.PluginRoot.GetChildByIECLocation(parent)
return "Error: No parent found\n"
child = _GetModuleBySomething(module, "IEC_Channel", (iec_channel,))
return "Error: No module found\n"
self.PluginRoot.RemoveProjectVariableByFilter(".".join(map(str, parent + (iec_channel,))))
_RemoveModuleChild(module, child)
def StartGroup(self, parent, name, icon=None):
module = self.PluginRoot.GetChildByIECLocation(parent)
return "Error: No parent found\n"
for child in module["children"]:
if child["type"] == LOCATION_GROUP and child["name"] == name:
return "Error: A group named %s already exists\n" % name
module["children"].append({"name": name,
def AddVariable(self, parent, location, name, direction, type, rcode, pcode, description=""):
module = self.PluginRoot.GetChildByIECLocation(parent)
return "Error: No parent found\n"
for child in _GetModuleChildren(module):
if child["name"] == name:
return "Error: A variable named %s already exists\n" % name
if child["location"] == location and child["type"] == LOCATION_TYPES[direction]:
return "Error: A variable with location %s already exists\n" % ".".join(map(str, location))
_GetLastModuleGroup(module).append({"name": name,
"type": LOCATION_TYPES[direction],
"description": description,
def ChangeVariableParams(self, parent, location, new_name, new_direction, new_type, new_rcode, new_pcode, new_description=None):
module = self.PluginRoot.GetChildByIECLocation(parent)
return "Error: No parent found\n"
for child in _GetModuleChildren(module):
if child["location"] == location and child["type"] == LOCATION_TYPES[new_direction]:
elif child["name"] == new_name:
return "Error: A variable named %s already exists\n" % new_name
return "Error: No variable found\n"
if variable["name"] != new_name:
self.PluginRoot.UpdateProjectVariableName(variable["name"], new_name)
variable["name"] = new_name
variable["type"] = LOCATION_TYPES[new_direction]
variable["IEC_type"] = new_type
variable["retrieve"] = new_rcode
variable["publish"] = new_pcode
if new_description is not None:
variable["description"] = new_description
def RemoveVariable(self, parent, location, direction):
module = self.PluginRoot.GetChildByIECLocation(parent)
return "Error: No parent found\n"
child = _GetModuleVariable(module, location, direction)
return "Error: No variable found\n"
size = LOCATION_SIZES[self.PluginRoot.GetBaseType(child["IEC_type"])]
address = "%" + LOCATION_DIRS[child["type"]] + size + ".".join(map(str, parent + location))
self.PluginRoot.RemoveProjectVariableByAddress(address)
_RemoveModuleChild(module, child)
return tuple(map(int, loc.split(".")))
def GetCmdFunction(function, arg_types, opt=0):
arg_number = len(arg_types)
def CmdFunction(self, line):
args_toks = line.split('"')
if len(args_toks) % 2 == 0:
self.Log.write("Error: Invalid command\n")
for num, arg in enumerate(args_toks):
args.extend(stripped.split(" "))
if opt == 0 and len(args) != arg_number:
elif len(args) > arg_number:
elif len(args) < arg_number - opt:
number = arg_number - opt
self.Log.write("Error: No argument%s expected\n" % extra)
self.Log.write("Error: 1 argument%s expected\n" % extra)
self.Log.write("Error: %d arguments%s expected\n" % (number, extra))
for num, arg in enumerate(args):
args[num] = arg_types[num](arg)
self.Log.write("Error: Invalid value for argument %d\n" % (num + 1))
#cmdlogf.write(str((function,line))+'\n')
func = getattr(self, function)
res = evaluator(func,*args)
#cmdlogf.write("--->"+str(res)+'\n')
cmdlog.append((function,line,res))
if len(cmdlog) > 100: #prevent debug log to grow too much
if isinstance(res, (StringType, UnicodeType)):
def CmdThreadProc(PluginRoot, Log):
for function, (arg_types, opt) in {"Exit": ([], 0),
"SetProjectProperties": ([str, str, str, str], 0),
"SetOnlineMode": ([str, str], 1),
"AddBus": ([int, str, str], 1),
"RenameBus": ([int, str], 0),
"ChangeBusIECChannel": ([int, int], 0),
"AddModule": ([location, int, str, str, str], 1),
"RenameModule": ([location, str], 0),
"ChangeModuleIECChannel": ([location, int], 0),
"ChangeModuleInitCode": ([location, str], 0),
"RemoveModule": ([location, int], 0),
"StartGroup": ([location, str, str], 1),
"AddVariable": ([location, location, str, str, str, str, str, str], 1),
"ChangeVariableParams": ([location, location, str, str, str, str, str, str], 1),
"RemoveVariable": ([location, location], 0)}.iteritems():
setattr(LPCBeremiz_Cmd, "do_%s" % function, GetCmdFunction(function, arg_types, opt))
lpcberemiz_cmd = LPCBeremiz_Cmd(PluginRoot, Log)
Log = StdoutPseudoFile(port)
PluginRoot = LPCPluginsRoot(None, Log)
if projectOpen is not None and os.path.isdir(projectOpen):
result = PluginRoot.LoadProject(projectOpen, buildpath)
Log.write("Error: Invalid project directory", result)
Log.write("Error: No such file or directory")
cmd_thread=Thread(target=CmdThreadProc, args=[PluginRoot, Log])
# Install a exception handle for bug reports
AddExceptHook(os.getcwd(),__version__)
frame = LPCBeremiz(None, plugin_root=PluginRoot, debug=True)