--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/confnodes/c_ext/c_ext.py Mon May 07 18:47:29 2012 +0200
@@ -0,0 +1,315 @@
+import wx
+import os
+from xml.dom import minidom
+import cPickle
+
+from xmlclass import *
+
+from ConfigTree import ConfigTreeNode, opjimg
+from CFileEditor import CFileEditor
+from PLCControler import PLCControler, UndoBuffer, LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP, LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY
+
+CFileClasses = GenerateClassesFromXSD(os.path.join(os.path.dirname(__file__), "cext_xsd.xsd"))
+
+TYPECONVERSION = {"BOOL" : "X", "SINT" : "B", "INT" : "W", "DINT" : "D", "LINT" : "L",
+ "USINT" : "B", "UINT" : "W", "UDINT" : "D", "ULINT" : "L", "REAL" : "D", "LREAL" : "L",
+ "STRING" : "B", "BYTE" : "B", "WORD" : "W", "DWORD" : "D", "LWORD" : "L", "WSTRING" : "W"}
+
+class _Cfile:
+ XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>
+ <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">
+ <xsd:element name="CExtension">
+ <xsd:complexType>
+ <xsd:attribute name="CFLAGS" type="xsd:string" use="required"/>
+ <xsd:attribute name="LDFLAGS" type="xsd:string" use="required"/>
+ </xsd:complexType>
+ </xsd:element>
+ </xsd:schema>
+ """
+ EditorType = CFileEditor
+
+ def __init__(self):
+ filepath = self.CFileName()
+
+ self.CFile = CFileClasses["CFile"]()
+ if os.path.isfile(filepath):
+ xmlfile = open(filepath, 'r')
+ tree = minidom.parse(xmlfile)
+ xmlfile.close()
+
+ for child in tree.childNodes:
+ if child.nodeType == tree.ELEMENT_NODE and child.nodeName == "CFile":
+ self.CFile.loadXMLTree(child, ["xmlns", "xmlns:xsi", "xsi:schemaLocation"])
+ self.CreateCFileBuffer(True)
+ else:
+ self.CreateCFileBuffer(False)
+ self.OnPlugSave()
+
+ def CFileName(self):
+ return os.path.join(self.PlugPath(), "cfile.xml")
+
+ def GetBaseTypes(self):
+ return self.GetPlugRoot().GetBaseTypes()
+
+ def GetDataTypes(self, basetypes = False, only_locatables = False):
+ return self.GetPlugRoot().GetDataTypes(basetypes=basetypes, only_locatables=only_locatables)
+
+ def GetSizeOfType(self, type):
+ return TYPECONVERSION.get(self.GetPlugRoot().GetBaseType(type), None)
+
+ def SetVariables(self, variables):
+ self.CFile.variables.setvariable([])
+ for var in variables:
+ variable = CFileClasses["variables_variable"]()
+ variable.setname(var["Name"])
+ variable.settype(var["Type"])
+ variable.setclass(var["Class"])
+ self.CFile.variables.appendvariable(variable)
+
+ def GetVariables(self):
+ datas = []
+ for var in self.CFile.variables.getvariable():
+ datas.append({"Name" : var.getname(), "Type" : var.gettype(), "Class" : var.getclass()})
+ return datas
+
+ def GetVariableLocationTree(self):
+ '''See ConfigTreeNode.GetVariableLocationTree() for a description.'''
+
+ current_location = ".".join(map(str, self.GetCurrentLocation()))
+
+ vars = []
+ input = memory = output = 0
+ for var in self.CFile.variables.getvariable():
+ var_size = self.GetSizeOfType(var.gettype())
+ var_location = ""
+ if var.getclass() == "input":
+ var_class = LOCATION_VAR_INPUT
+ if var_size is not None:
+ var_location = "%%I%s%s.%d"%(var_size, current_location, input)
+ input += 1
+ elif var.getclass() == "memory":
+ var_class = LOCATION_VAR_INPUT
+ if var_size is not None:
+ var_location = "%%M%s%s.%d"%(var_size, current_location, memory)
+ memory += 1
+ else:
+ var_class = LOCATION_VAR_OUTPUT
+ if var_size is not None:
+ var_location = "%%Q%s%s.%d"%(var_size, current_location, output)
+ output += 1
+ vars.append({"name": var.getname(),
+ "type": var_class,
+ "size": var_size,
+ "IEC_type": var.gettype(),
+ "var_name": var.getname(),
+ "location": var_location,
+ "description": "",
+ "children": []})
+
+ return {"name": self.BaseParams.getName(),
+ "type": LOCATION_CONFNODE,
+ "location": self.GetFullIEC_Channel(),
+ "children": vars}
+
+ def SetPartText(self, name, text):
+ if name == "Includes":
+ self.CFile.includes.settext(text)
+ elif name == "Globals":
+ self.CFile.globals.settext(text)
+ elif name == "Init":
+ self.CFile.initFunction.settext(text)
+ elif name == "CleanUp":
+ self.CFile.cleanUpFunction.settext(text)
+ elif name == "Retrieve":
+ self.CFile.retrieveFunction.settext(text)
+ elif name == "Publish":
+ self.CFile.publishFunction.settext(text)
+
+ def GetPartText(self, name):
+ if name == "Includes":
+ return self.CFile.includes.gettext()
+ elif name == "Globals":
+ return self.CFile.globals.gettext()
+ elif name == "Init":
+ return self.CFile.initFunction.gettext()
+ elif name == "CleanUp":
+ return self.CFile.cleanUpFunction.gettext()
+ elif name == "Retrieve":
+ return self.CFile.retrieveFunction.gettext()
+ elif name == "Publish":
+ return self.CFile.publishFunction.gettext()
+ return ""
+
+ ConfNodeMethods = [
+ {"bitmap" : os.path.join("images", "EditCfile"),
+ "name" : _("Edit C File"),
+ "tooltip" : _("Edit C File"),
+ "method" : "_OpenView"},
+ ]
+
+ def PlugTestModified(self):
+ return self.ChangesToSave or not self.CFileIsSaved()
+
+ def OnPlugSave(self):
+ filepath = self.CFileName()
+
+ text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n"
+ extras = {"xmlns":"http://www.w3.org/2001/XMLSchema",
+ "xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance",
+ "xsi:schemaLocation" : "cext_xsd.xsd"}
+ text += self.CFile.generateXMLText("CFile", 0, extras)
+
+ xmlfile = open(filepath,"w")
+ xmlfile.write(text.encode("utf-8"))
+ xmlfile.close()
+
+ self.MarkCFileAsSaved()
+ return True
+
+ def PlugGenerate_C(self, buildpath, locations):
+ """
+ Generate C code
+ @param current_location: Tupple containing confnode 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))
+
+ text = "/* Code generated by Beremiz c_ext confnode */\n\n"
+
+ # Adding includes
+ text += "/* User includes */\n"
+ text += self.CFile.includes.gettext()
+ text += "\n"
+
+ text += """/* Beremiz c_ext confnode includes */
+#ifdef _WINDOWS_H
+ #include "iec_types.h"
+#else
+ #include "iec_std_lib.h"
+#endif
+
+"""
+
+ # Adding variables
+ vars = []
+ inputs = memories = outputs = 0
+ for variable in self.CFile.variables.variable:
+ var = {"Name" : variable.getname(), "Type" : variable.gettype()}
+ if variable.getclass() == "input":
+ var["location"] = "__I%s%s_%d"%(self.GetSizeOfType(var["Type"]), location_str, inputs)
+ inputs += 1
+ elif variable.getclass() == "memory":
+ var["location"] = "__M%s%s_%d"%(self.GetSizeOfType(var["Type"]), location_str, memories)
+ memories += 1
+ else:
+ var["location"] = "__Q%s%s_%d"%(self.GetSizeOfType(var["Type"]), location_str, outputs)
+ outputs += 1
+ vars.append(var)
+ text += "/* Beremiz c_ext confnode user variables definition */\n"
+ base_types = self.GetPlugRoot().GetBaseTypes()
+ for var in vars:
+ if var["Type"] in base_types:
+ prefix = "IEC_"
+ else:
+ prefix = ""
+ text += "%s%s beremiz%s;\n"%(prefix, var["Type"], var["location"])
+ text += "%s%s *%s = &beremiz%s;\n"%(prefix, var["Type"], var["location"], var["location"])
+ text += "/* User variables reference */\n"
+ for var in vars:
+ text += "#define %s beremiz%s\n"%(var["Name"], var["location"])
+ text += "\n"
+
+ # Adding user global variables and routines
+ text += "/* User internal user variables and routines */\n"
+ text += self.CFile.globals.gettext()
+
+ # Adding Beremiz confnode functions
+ text += "/* Beremiz confnode functions */\n"
+ text += "int __init_%s(int argc,char **argv)\n{\n"%location_str
+ text += self.CFile.initFunction.gettext()
+ text += " return 0;\n"
+ text += "\n}\n\n"
+
+ text += "void __cleanup_%s(void)\n{\n"%location_str
+ text += self.CFile.cleanUpFunction.gettext()
+ text += "\n}\n\n"
+
+ text += "void __retrieve_%s(void)\n{\n"%location_str
+ text += self.CFile.retrieveFunction.gettext()
+ text += "\n}\n\n"
+
+ text += "void __publish_%s(void)\n{\n"%location_str
+ text += self.CFile.publishFunction.gettext()
+ text += "\n}\n\n"
+
+ Gen_Cfile_path = os.path.join(buildpath, "CFile_%s.c"%location_str)
+ cfile = open(Gen_Cfile_path,'w')
+ cfile.write(text)
+ cfile.close()
+
+ matiec_flags = '"-I%s"'%os.path.abspath(self.GetPlugRoot().GetIECLibPath())
+
+ return [(Gen_Cfile_path, str(self.CExtension.getCFLAGS() + matiec_flags))],str(self.CExtension.getLDFLAGS()),True
+
+
+#-------------------------------------------------------------------------------
+# Current Buffering Management Functions
+#-------------------------------------------------------------------------------
+
+ """
+ Return a copy of the cfile model
+ """
+ def Copy(self, model):
+ return cPickle.loads(cPickle.dumps(model))
+
+ def CreateCFileBuffer(self, saved):
+ self.Buffering = False
+ self.CFileBuffer = UndoBuffer(cPickle.dumps(self.CFile), saved)
+
+ def BufferCFile(self):
+ self.CFileBuffer.Buffering(cPickle.dumps(self.CFile))
+
+ def StartBuffering(self):
+ self.Buffering = True
+
+ def EndBuffering(self):
+ if self.Buffering:
+ self.CFileBuffer.Buffering(cPickle.dumps(self.CFile))
+ self.Buffering = False
+
+ def MarkCFileAsSaved(self):
+ self.EndBuffering()
+ self.CFileBuffer.CurrentSaved()
+
+ def CFileIsSaved(self):
+ return self.CFileBuffer.IsCurrentSaved() and not self.Buffering
+
+ def LoadPrevious(self):
+ self.EndBuffering()
+ self.CFile = cPickle.loads(self.CFileBuffer.Previous())
+
+ def LoadNext(self):
+ self.CFile = cPickle.loads(self.CFileBuffer.Next())
+
+ def GetBufferState(self):
+ first = self.CFileBuffer.IsFirst() and not self.Buffering
+ last = self.CFileBuffer.IsLast()
+ return not first, not last
+
+class RootClass:
+
+ PlugChildsTypes = [("C_File",_Cfile, "C file")]
+
+ def PlugGenerate_C(self, buildpath, locations):
+ return [],"",False
+
+