--- a/etherlab/ConfigEditor.py Wed Feb 27 22:40:45 2013 +0100
+++ b/etherlab/ConfigEditor.py Tue Mar 05 00:59:34 2013 +0100
@@ -1,10 +1,13 @@
+from types import TupleType +from plcopen.structures import IEC_KEYWORDS, TestIdentifier from controls import CustomGrid, CustomTable, FolderTree
from editors.ConfTreeNodeEditor import ConfTreeNodeEditor, SCROLLBAR_UNIT
from util.BitmapLibrary import GetBitmap
@@ -17,9 +20,12 @@
parent.Append(helpString=help, id=id, kind=kind, item=text)
-def GetVariablesTableColnames():
+def GetVariablesTableColnames(position=False): - return ["#", _("Name"), _("Index"), _("SubIndex"), _("Type"), _("Access")]
+ colname.append(_("Position")) + return colname + [_("Name"), _("Index"), _("SubIndex"), _("Type"), _("Access")] @@ -27,72 +33,78 @@
def GetAccessValue(access, pdo_mapping):
- value = ACCESS_TYPES.get(access)
+ value = ACCESS_TYPES.get(access, "") -class NodeEditor(ConfTreeNodeEditor):
- CONFNODEEDITOR_TABS = [
- (_("Ethercat node"), "_create_EthercatNodeEditor")]
+ (_("All"), (0x0000, 0xffff)), + (_("Communication Parameters"), (0x1000, 0x1fff)), + (_("Manufacturer Specific"), (0x2000, 0x5fff)), + (_("Standardized Device Profile"), (0x6000, 0x9fff))] +ETHERCAT_INDEX_MODEL = re.compile("#x([0-9a-fA-F]{0,4})$") +ETHERCAT_SUBINDEX_MODEL = re.compile("#x([0-9a-fA-F]{0,2})$") +LOCATION_MODEL = re.compile("(?:%[IQM](?:[XBWLD]?([0-9]+(?:\.[0-9]+)*)))$") +class NodeVariablesSizer(wx.FlexGridSizer): - def _create_EthercatNodeEditor(self, prnt):
- self.EthercatNodeEditor = wx.Panel(prnt, style=wx.TAB_TRAVERSAL)
- main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5)
- main_sizer.AddGrowableCol(0)
- main_sizer.AddGrowableRow(1)
+ def __init__(self, parent, controler, position_column=False): + wx.FlexGridSizer.__init__(self, cols=1, hgap=0, rows=2, vgap=5) - variables_label = wx.StaticText(self.EthercatNodeEditor,
- label=_('Variable entries:'))
- main_sizer.AddWindow(variables_label, border=10, flag=wx.TOP|wx.LEFT|wx.RIGHT)
+ self.Controler = controler + self.PositionColumn = position_column + self.VariablesFilter = wx.ComboBox(parent) + self.VariablesFilter.Bind(wx.EVT_COMBOBOX, self.OnVariablesFilterChanged) + self.AddWindow(self.VariablesFilter, flag=wx.GROW) - self.VariablesGrid = wx.gizmos.TreeListCtrl(self.EthercatNodeEditor,
- style=wx.TR_DEFAULT_STYLE |
- wx.TR_FULL_ROW_HIGHLIGHT)
- self.VariablesGrid.GetMainWindow().Bind(wx.EVT_LEFT_DOWN,
+ self.VariablesGrid = wx.gizmos.TreeListCtrl(parent, + style=wx.TR_DEFAULT_STYLE | + wx.TR_FULL_ROW_HIGHLIGHT) + self.VariablesGrid.GetMainWindow().Bind(wx.EVT_LEFT_DOWN, self.OnVariablesGridLeftClick)
- main_sizer.AddWindow(self.VariablesGrid, border=10,
- flag=wx.GROW|wx.BOTTOM|wx.LEFT|wx.RIGHT)
- self.EthercatNodeEditor.SetSizer(main_sizer)
- return self.EthercatNodeEditor
- def __init__(self, parent, controler, window):
- ConfTreeNodeEditor.__init__(self, parent, controler, window)
+ self.AddWindow(self.VariablesGrid, flag=wx.GROW) + for desc, value in VARIABLES_FILTERS: + self.VariablesFilter.Append(desc) + self.Filters.append(value) + self.VariablesFilter.SetSelection(0) + self.CurrentFilter = self.Filters[0] + for colname, colsize, colalign in zip(GetVariablesTableColnames(position_column), + [40, 80, 350, 80, 100, 80, 80], + [wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, wx.ALIGN_LEFT, + wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, wx.ALIGN_LEFT, + self.VariablesGrid.AddColumn(_(colname), colsize, colalign) + self.VariablesGrid.SetMainColumn(2) + for colname, colsize, colalign in zip(GetVariablesTableColnames(), + [40, 350, 80, 100, 80, 80], + [wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_RIGHT, + wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_LEFT]): + self.VariablesGrid.AddColumn(_(colname), colsize, colalign) + self.VariablesGrid.SetMainColumn(1) - for colname, colsize, colalign in zip(GetVariablesTableColnames(),
- [40, 150, 100, 100, 150, 100],
- [wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_RIGHT,
- wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_LEFT]):
- self.VariablesGrid.AddColumn(_(colname), colsize, colalign)
- self.VariablesGrid.SetMainColumn(1)
- def GetBufferState(self):
- ConfTreeNodeEditor.RefreshView(self)
- self.RefreshSlaveInfos()
- def RefreshSlaveInfos(self):
- slave_infos = self.Controler.GetSlaveInfos()
- if slave_infos is not None:
- self.RefreshVariablesGrid(slave_infos["entries"])
- self.RefreshVariablesGrid([])
+ entries = self.Controler.GetSlaveVariables(self.CurrentFilter) + self.RefreshVariablesGrid(entries) def RefreshVariablesGrid(self, entries):
root = self.VariablesGrid.GetRootItem()
- root = self.VariablesGrid.AddRoot("Slave entries")
- self.GenerateVariablesGridBranch(root, entries, GetVariablesTableColnames())
+ root = self.VariablesGrid.AddRoot(_("Slave entries")) + self.GenerateVariablesGridBranch(root, entries, GetVariablesTableColnames(self.PositionColumn)) self.VariablesGrid.Expand(root)
def GenerateVariablesGridBranch(self, root, entries, colnames, idx=0):
@@ -113,6 +125,8 @@
self.VariablesGrid.SetItemText(item, value, col)
if entry["PDOMapping"] == "":
self.VariablesGrid.SetItemBackgroundColour(item, wx.LIGHT_GREY)
+ self.VariablesGrid.SetItemBackgroundColour(item, wx.WHITE) self.VariablesGrid.SetItemPyData(item, entry)
idx = self.GenerateVariablesGridBranch(item, entry["children"], colnames, idx)
@@ -128,70 +142,703 @@
self.VariablesGrid.Delete(item)
+ def OnVariablesFilterChanged(self, event): + filter = self.VariablesFilter.GetSelection() + self.CurrentFilter = self.Filters[filter] + value = self.VariablesFilter.GetValue() + result = ETHERCAT_INDEX_MODEL.match(value) + value = result.group(1) + self.CurrentFilter = (index, index) def OnVariablesGridLeftClick(self, event):
item, flags, col = self.VariablesGrid.HitTest(event.GetPosition())
entry = self.VariablesGrid.GetItemPyData(item)
data_type = entry.get("Type", "")
- pdo_mapping = entry.get("PDOMapping", "")
+ data_size = self.Controler.GetSizeOfType(data_type) - if (col == -1 and pdo_mapping != "" and
- self.Controler.GetSizeOfType(data_type) is not None):
+ if col == -1 and data_size is not None: + pdo_mapping = entry.get("PDOMapping", "") + access = entry.get("Access", "") entry_index = self.Controler.ExtractHexDecValue(entry.get("Index", "0"))
entry_subindex = self.Controler.ExtractHexDecValue(entry.get("SubIndex", "0"))
- var_name = "%s_%4.4x_%2.2x" % (self.Controler.CTNName(), entry_index, entry_subindex)
+ if self.PositionColumn: + slave_pos = self.Controler.ExtractHexDecValue(entry.get("Position", "0"))
- location = "%s%s" % (dir, self.Controler.GetSizeOfType(data_type)) + \
- ".".join(map(lambda x:str(x), self.Controler.GetCurrentLocation() + (self.Controler.GetSlavePos(), entry_index, entry_subindex)))
+ slave_pos = self.Controler.GetSlavePos() + var_name = "%s_%4.4x_%2.2x" % (self.Controler.CTNName(), entry_index, entry_subindex) + location = "%s%s" % (dir, data_size) + \ + ".".join(map(lambda x:str(x), self.Controler.GetCurrentLocation() + + (slave_pos, entry_index, entry_subindex))) + data = wx.TextDataObject(str((location, "location", data_type, var_name, "", access))) + dragSource = wx.DropSource(self.VariablesGrid) + dragSource.SetData(data) + dragSource.DoDragDrop() - data = wx.TextDataObject(str((location, "location", data_type, var_name, "")))
- dragSource = wx.DropSource(self.VariablesGrid)
- dragSource.SetData(data)
- dragSource.DoDragDrop()
+ elif self.ColumnPosition: + location = self.Controler.GetCurrentLocation() +\ + (slave_pos, entry_index, entry_subindex) + data = wx.TextDataObject(str((location, "variable", access))) + dragSource = wx.DropSource(self.VariablesGrid) + dragSource.SetData(data) + dragSource.DoDragDrop() +class NodeEditor(ConfTreeNodeEditor): + CONFNODEEDITOR_TABS = [ + (_("Ethercat node"), "_create_EthercatNodeEditor")] + def _create_EthercatNodeEditor(self, prnt): + self.EthercatNodeEditor = wx.Panel(prnt, style=wx.TAB_TRAVERSAL) + main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5) + main_sizer.AddGrowableCol(0) + main_sizer.AddGrowableRow(1) + variables_label = wx.StaticText(self.EthercatNodeEditor, + label=_('Variable entries:')) + main_sizer.AddWindow(variables_label, border=10, flag=wx.TOP|wx.LEFT|wx.RIGHT) + self.NodeVariables = NodeVariablesSizer(self.EthercatNodeEditor, self.Controler) + main_sizer.AddSizer(self.NodeVariables, border=10, + flag=wx.GROW|wx.BOTTOM|wx.LEFT|wx.RIGHT) + self.EthercatNodeEditor.SetSizer(main_sizer) + return self.EthercatNodeEditor + def __init__(self, parent, controler, window): + ConfTreeNodeEditor.__init__(self, parent, controler, window) + def GetBufferState(self): + ConfTreeNodeEditor.RefreshView(self) + self.NodeVariables.RefreshView() CIA402NodeEditor = NodeEditor
+def GetProcessVariablesTableColnames(): + return ["#", _("Name"), + _("Read from (nodeid, index, subindex)"), + _("Write to (nodeid, index, subindex)"), +class ProcessVariablesTable(CustomTable): + def GetValue(self, row, col): + if row < self.GetNumberRows(): + colname = self.GetColLabelValue(col, False) + if colname.startswith("Read from"): + value = self.data[row].get("ReadFrom", "") + return "%d, #x%0.4x, #x%0.2x" % value + elif colname.startswith("Write to"): + value = self.data[row].get("WriteTo", "") + return "%d, #x%0.4x, #x%0.2x" % value + return self.data[row].get(colname, "") + def SetValue(self, row, col, value): + if col < len(self.colnames): + colname = self.GetColLabelValue(col, False) + if colname.startswith("Read from"): + self.data[row]["ReadFrom"] = value + elif colname.startswith("Write to"): + self.data[row]["WriteTo"] = value + self.data[row][colname] = value + def _updateColAttrs(self, grid): + wx.grid.Grid -> update the column attributes to add the + appropriate renderer given the column name. + Otherwise default to the default renderer. + for row in range(self.GetNumberRows()): + for col in range(self.GetNumberCols()): + colname = self.GetColLabelValue(col, False) + if colname in ["Name", "Description"]: + editor = wx.grid.GridCellTextEditor() + renderer = wx.grid.GridCellStringRenderer() + grid.SetReadOnly(row, col, False) + grid.SetReadOnly(row, col, True) + grid.SetCellEditor(row, col, editor) + grid.SetCellRenderer(row, col, renderer) + self.ResizeRow(grid, row) +class ProcessVariableDropTarget(wx.TextDropTarget): + def __init__(self, parent): + wx.TextDropTarget.__init__(self) + self.ParentWindow = parent + def OnDropText(self, x, y, data): + self.ParentWindow.Select() + x, y = self.ParentWindow.ProcessVariablesGrid.CalcUnscrolledPosition(x, y) + col = self.ParentWindow.ProcessVariablesGrid.XToCol(x) + row = self.ParentWindow.ProcessVariablesGrid.YToRow(y - self.ParentWindow.ProcessVariablesGrid.GetColLabelSize()) + message = _("Invalid value \"%s\" for process variable")%data + if not isinstance(values, TupleType): + message = _("Invalid value \"%s\" for process variable")%data + if values is not None and 2 <= col <= 3: + if values[1] == "location": + result = LOCATION_MODEL.match(values[0]) + location = map(int, result.group(1).split('.')) + master_location = self.ParentWindow.GetMasterLocation() + if (master_location == tuple(location[:len(master_location)]) and + len(location) - len(master_location) == 3): + self.ParentWindow.ProcessVariablesTable.SetValueByName( + row, "ReadFrom", tuple(location[len(master_location):])) + self.ParentWindow.ProcessVariablesTable.SetValueByName( + row, "WriteTo", tuple(location[len(master_location):])) + self.ParentWindow.SaveProcessVariables() + self.ParentWindow.RefreshProcessVariables() + message = _("Invalid value \"%s\" for process variable")%data + if message is not None: + wx.CallAfter(self.ShowMessage, message) + def ShowMessage(self, message): + message = wx.MessageDialog(self.ParentWindow, message, _("Error"), wx.OK|wx.ICON_ERROR) +def GetStartupCommandsTableColnames(): + return [_("Position"), _("Index"), _("Subindex"), _("Value"), _("Description")] +class StartupCommandDropTarget(wx.TextDropTarget): + def __init__(self, parent): + wx.TextDropTarget.__init__(self) + self.ParentWindow = parent + def OnDropText(self, x, y, data): + self.ParentWindow.Select() + message = _("Invalid value \"%s\" for startup command")%data + if not isinstance(values, TupleType): + message = _("Invalid value \"%s\" for startup command")%data + if values[1] == "location": + result = LOCATION_MODEL.match(values[0]) + location = map(int, result.group(1).split('.')) + elif values[1] == "variable": + if location is not None: + master_location = self.ParentWindow.GetMasterLocation() + if (master_location == tuple(location[:len(master_location)]) and + len(location) - len(master_location) == 3): + if access in ["wo", "rw"]: + self.ParentWindow.AddStartupCommand(*location[len(master_location):]) + message = _("Entry can't be write through SDO") + message = _("Invalid value \"%s\" for startup command")%data + if message is not None: + wx.CallAfter(self.ShowMessage, message) + def ShowMessage(self, message): + message = wx.MessageDialog(self.ParentWindow, message, _("Error"), wx.OK|wx.ICON_ERROR) +class StartupCommandsTable(CustomTable): + A custom wx.grid.Grid Table using user supplied data + def __init__(self, parent, data, colnames): + # The base class must be initialized *first* + CustomTable.__init__(self, parent, data, colnames) + def GetValue(self, row, col): + if row < self.GetNumberRows(): + colname = self.GetColLabelValue(col, False) + value = self.data[row].get(colname, "") + return "#x%0.4x" % value + elif colname == "Subindex": + return "#x%0.2x" % value + def SetValue(self, row, col, value): + if col < len(self.colnames): + colname = self.GetColLabelValue(col, False) + if colname in ["Index", "Subindex"]: + result = ETHERCAT_INDEX_MODEL.match(value) + result = ETHERCAT_SUBINDEX_MODEL.match(value) + value = int(result.group(1), 16) + elif colname == "Value": + elif colname == "Position": + self.old_value = self.data[row][colname] + self.data[row][colname] = value + def _updateColAttrs(self, grid): + wx.grid.Grid -> update the column attributes to add the + appropriate renderer given the column name. + Otherwise default to the default renderer. + for row in range(self.GetNumberRows()): + for col in range(self.GetNumberCols()): + colname = self.GetColLabelValue(col, False) + if colname in ["Position", "Value"]: + editor = wx.grid.GridCellNumberEditor() + renderer = wx.grid.GridCellNumberRenderer() + editor = wx.grid.GridCellTextEditor() + renderer = wx.grid.GridCellStringRenderer() + grid.SetCellEditor(row, col, editor) + grid.SetCellRenderer(row, col, renderer) + grid.SetReadOnly(row, col, False) + self.ResizeRow(grid, row) + def GetCommandIndex(self, position, command_idx): + for row, command in enumerate(self.data): + if command["Position"] == position and command["command_idx"] == command_idx: +class MasterNodesVariablesSizer(NodeVariablesSizer): + def __init__(self, parent, controler): + NodeVariablesSizer.__init__(self, parent, controler, True) + self.CurrentNodesFilter = {} + def SetCurrentNodesFilter(self, nodes_filter): + self.CurrentNodesFilter = nodes_filter + if self.CurrentNodesFilter is not None: + args = self.CurrentNodesFilter.copy() + args["limits"] = self.CurrentFilter + entries = self.Controler.GetNodesVariables(**args) + self.RefreshVariablesGrid(entries) +class MasterEditor(ConfTreeNodeEditor): + CONFNODEEDITOR_TABS = [ + (_("Network"), "_create_EthercatMasterEditor")] + def _create_EthercatMasterEditor(self, prnt): + self.EthercatMasterEditor = wx.ScrolledWindow(prnt, + style=wx.TAB_TRAVERSAL|wx.HSCROLL|wx.VSCROLL) + self.EthercatMasterEditor.Bind(wx.EVT_SIZE, self.OnResize) + main_sizer = wx.BoxSizer(wx.VERTICAL) + self.NodesFilter = wx.ComboBox(self.EthercatMasterEditor, + style=wx.TE_PROCESS_ENTER) + self.Bind(wx.EVT_COMBOBOX, self.OnNodesFilterChanged, self.NodesFilter) + self.Bind(wx.EVT_TEXT_ENTER, self.OnNodesFilterChanged, self.NodesFilter) + process_variables_header = wx.BoxSizer(wx.HORIZONTAL) + process_variables_label = wx.StaticText(self.EthercatMasterEditor, + label=_("Process variables mapped between nodes:")) + process_variables_header.AddWindow(process_variables_label, 1, + flag=wx.ALIGN_CENTER_VERTICAL) + for name, bitmap, help in [ + ("AddVariableButton", "add_element", _("Add process variable")), + ("DeleteVariableButton", "remove_element", _("Remove process variable")), + ("UpVariableButton", "up", _("Move process variable up")), + ("DownVariableButton", "down", _("Move process variable down"))]: + button = wx.lib.buttons.GenBitmapButton(self.EthercatMasterEditor, bitmap=GetBitmap(bitmap), + size=wx.Size(28, 28), style=wx.NO_BORDER) + button.SetToolTipString(help) + setattr(self, name, button) + process_variables_header.AddWindow(button, border=5, flag=wx.LEFT) + self.ProcessVariablesGrid = CustomGrid(self.EthercatMasterEditor, style=wx.VSCROLL) + self.ProcessVariablesGrid.SetMinSize(wx.Size(0, 150)) + self.ProcessVariablesGrid.SetDropTarget(ProcessVariableDropTarget(self)) + self.ProcessVariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, + self.OnProcessVariablesGridCellChange) + self.ProcessVariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, + self.OnProcessVariablesGridCellLeftClick) + startup_commands_header = wx.BoxSizer(wx.HORIZONTAL) + startup_commands_label = wx.StaticText(self.EthercatMasterEditor, + label=_("Startup service variables assignments:")) + startup_commands_header.AddWindow(startup_commands_label, 1, + flag=wx.ALIGN_CENTER_VERTICAL) + for name, bitmap, help in [ + ("AddCommandButton", "add_element", _("Add startup service variable")), + ("DeleteCommandButton", "remove_element", _("Remove startup service variable"))]: + button = wx.lib.buttons.GenBitmapButton(self.EthercatMasterEditor, bitmap=GetBitmap(bitmap), + size=wx.Size(28, 28), style=wx.NO_BORDER) + button.SetToolTipString(help) + setattr(self, name, button) + startup_commands_header.AddWindow(button, border=5, flag=wx.LEFT) + self.StartupCommandsGrid = CustomGrid(self.EthercatMasterEditor, style=wx.VSCROLL) + self.StartupCommandsGrid.SetDropTarget(StartupCommandDropTarget(self)) + self.StartupCommandsGrid.SetMinSize(wx.Size(0, 150)) + self.StartupCommandsGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, + self.OnStartupCommandsGridCellChange) + second_staticbox = wx.StaticBox(self.EthercatMasterEditor, label=_("Nodes variables filter:")) + second_staticbox_sizer = wx.StaticBoxSizer(second_staticbox, wx.VERTICAL) + self.NodesVariables = MasterNodesVariablesSizer(self.EthercatMasterEditor, self.Controler) + second_staticbox_sizer.AddSizer(self.NodesVariables, 1, border=5, flag=wx.GROW|wx.ALL) + main_staticbox = wx.StaticBox(self.EthercatMasterEditor, label=_("Node filter:")) + staticbox_sizer = wx.StaticBoxSizer(main_staticbox, wx.VERTICAL) + main_sizer.AddSizer(staticbox_sizer, border=10, flag=wx.GROW|wx.ALL) + main_staticbox_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=5, vgap=0) + main_staticbox_sizer.AddGrowableCol(0) + main_staticbox_sizer.AddGrowableRow(1) + main_staticbox_sizer.AddGrowableRow(3) + staticbox_sizer.AddSizer(main_staticbox_sizer, 1, flag=wx.GROW) + main_staticbox_sizer.AddWindow(self.NodesFilter, border=5, flag=wx.GROW|wx.ALL) + main_staticbox_sizer.AddSizer(process_variables_header, border=5, + flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM) + main_staticbox_sizer.AddWindow(self.ProcessVariablesGrid, 1, + border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM) + main_staticbox_sizer.AddSizer(startup_commands_header, + border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM) + main_staticbox_sizer.AddWindow(self.StartupCommandsGrid, 1, + border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM) + main_staticbox_sizer.AddSizer(second_staticbox_sizer, 1, + border=5, flag=wx.GROW|wx.LEFT|wx.RIGHT|wx.BOTTOM) + self.EthercatMasterEditor.SetSizer(main_sizer) + return self.EthercatMasterEditor + def __init__(self, parent, controler, window): + ConfTreeNodeEditor.__init__(self, parent, controler, window) + self.ProcessVariablesDefaultValue = {"Name": "", "ReadFrom": "", "WriteTo": "", "Description": ""} + self.ProcessVariablesTable = ProcessVariablesTable(self, [], GetProcessVariablesTableColnames()) + self.ProcessVariablesColSizes = [40, 100, 150, 150, 200] + self.ProcessVariablesColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT] + self.ProcessVariablesGrid.SetTable(self.ProcessVariablesTable) + self.ProcessVariablesGrid.SetButtons({"Add": self.AddVariableButton, + "Delete": self.DeleteVariableButton, + "Up": self.UpVariableButton, + "Down": self.DownVariableButton}) + def _AddVariablesElement(new_row): + self.ProcessVariablesTable.InsertRow(new_row, self.ProcessVariablesDefaultValue.copy()) + self.SaveProcessVariables() + self.ProcessVariablesTable.ResetView(self.ProcessVariablesGrid) + setattr(self.ProcessVariablesGrid, "_AddRow", _AddVariablesElement) + def _DeleteVariablesElement(row): + self.ProcessVariablesTable.RemoveRow(row) + self.SaveProcessVariables() + self.ProcessVariablesTable.ResetView(self.ProcessVariablesGrid) + setattr(self.ProcessVariablesGrid, "_DeleteRow", _DeleteVariablesElement) + def _MoveVariablesElement(row, move): + new_row = self.ProcessVariablesTable.MoveRow(row, move) + self.SaveProcessVariables() + self.ProcessVariablesTable.ResetView(self.ProcessVariablesGrid) + setattr(self.ProcessVariablesGrid, "_MoveRow", _MoveVariablesElement) + self.ProcessVariablesGrid.SetRowLabelSize(0) + for col in range(self.ProcessVariablesTable.GetNumberCols()): + attr = wx.grid.GridCellAttr() + attr.SetAlignment(self.ProcessVariablesColAlignements[col], wx.ALIGN_CENTRE) + self.ProcessVariablesGrid.SetColAttr(col, attr) + self.ProcessVariablesGrid.SetColMinimalWidth(col, self.ProcessVariablesColSizes[col]) + self.ProcessVariablesGrid.AutoSizeColumn(col, False) + self.ProcessVariablesGrid.RefreshButtons() + self.StartupCommandsDefaultValue = {"Position": 0, "Index": 0, "Subindex": 0, "Value": 0, "Description": ""} + self.StartupCommandsTable = StartupCommandsTable(self, [], GetStartupCommandsTableColnames()) + self.StartupCommandsColSizes = [100, 100, 50, 100, 200] + self.StartupCommandsColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, wx.ALIGN_LEFT] + self.StartupCommandsGrid.SetTable(self.StartupCommandsTable) + self.StartupCommandsGrid.SetButtons({"Add": self.AddCommandButton, + "Delete": self.DeleteCommandButton}) + def _AddCommandsElement(new_row): + command = self.StartupCommandsDefaultValue.copy() + command_idx = self.Controler.AppendStartupCommand(command) + self.RefreshStartupCommands() + return self.StartupCommandsTable.GetCommandIndex(command["Position"], command_idx) + setattr(self.StartupCommandsGrid, "_AddRow", _AddCommandsElement) + def _DeleteCommandsElement(row): + command = self.StartupCommandsTable.GetRow(row) + self.Controler.RemoveStartupCommand(command["Position"], command["command_idx"]) + self.RefreshStartupCommands() + setattr(self.StartupCommandsGrid, "_DeleteRow", _DeleteCommandsElement) + self.StartupCommandsGrid.SetRowLabelSize(0) + for col in range(self.StartupCommandsTable.GetNumberCols()): + attr = wx.grid.GridCellAttr() + attr.SetAlignment(self.StartupCommandsColAlignements[col], wx.ALIGN_CENTRE) + self.StartupCommandsGrid.SetColAttr(col, attr) + self.StartupCommandsGrid.SetColMinimalWidth(col, self.StartupCommandsColSizes[col]) + self.StartupCommandsGrid.AutoSizeColumn(col, False) + self.StartupCommandsGrid.RefreshButtons() + def RefreshBuffer(self): + self.ParentWindow.RefreshTitle() + self.ParentWindow.RefreshFileMenu() + self.ParentWindow.RefreshEditMenu() + self.ParentWindow.RefreshPageTitles() + ConfTreeNodeEditor.RefreshView(self) + self.RefreshNodesFilter() + self.RefreshProcessVariables() + self.RefreshStartupCommands() + self.NodesVariables.RefreshView() + def RefreshNodesFilter(self): + value = self.NodesFilter.GetValue() + self.NodesFilter.Clear() + self.NodesFilter.Append(_("All")) + self.NodesFilterValues = [{}] + for vendor_id, vendor_name in self.Controler.GetLibraryVendors(): + self.NodesFilter.Append(_("%s's nodes") % vendor_name) + self.NodesFilterValues.append({"vendor": vendor_id}) + self.NodesFilter.Append(_("CIA402 nodes")) + self.NodesFilterValues.append({"slave_profile": 402}) + if value in self.NodesFilter.GetStrings(): + self.NodesFilter.SetStringSelection(value) + self.NodesFilter.SetValue(value) + self.NodesFilter.SetSelection(0) + self.RefreshCurrentNodesFilter() + def RefreshCurrentNodesFilter(self): + filter = self.NodesFilter.GetSelection() + self.CurrentNodesFilter = self.NodesFilterValues[filter] + self.CurrentNodesFilter = {"slave_pos": int(self.NodesFilter.GetValue())} + self.CurrentNodesFilter = None + self.NodesVariables.SetCurrentNodesFilter(self.CurrentNodesFilter) + def RefreshProcessVariables(self): + self.ProcessVariablesTable.SetData( + self.Controler.GetProcessVariables()) + self.ProcessVariablesTable.ResetView(self.ProcessVariablesGrid) + def SaveProcessVariables(self): + self.Controler.SetProcessVariables( + self.ProcessVariablesTable.GetData()) + def RefreshStartupCommands(self): + if self.CurrentNodesFilter is not None: + self.StartupCommandsTable.SetData( + self.Controler.GetStartupCommands(**self.CurrentNodesFilter)) + self.StartupCommandsTable.ResetView(self.StartupCommandsGrid) + def SelectStartupCommand(self, position, command_idx): + self.StartupCommandsGrid.SetSelectedRow( + self.StartupCommandsTable.GetCommandIndex(position, command_idx)) + def GetMasterLocation(self): + return self.Controler.GetCurrentLocation() + def AddStartupCommand(self, position, index, subindex): + command = self.StartupCommandsDefaultValue.copy() + command["Position"] = position + command["Index"] = index + command["Subindex"] = subindex + command_idx = self.Controler.AppendStartupCommand(command) + self.RefreshStartupCommands() + self.StartupCommandsGrid.SetSelectedRow( + self.StartupCommandsTable.GetCommandIndex(position, command_idx)) + def OnNodesFilterChanged(self, event): + self.RefreshCurrentNodesFilter() + if self.CurrentNodesFilter is not None: + self.RefreshStartupCommands() + self.NodesVariables.RefreshView() + def OnProcessVariablesGridCellChange(self, event): + row, col = event.GetRow(), event.GetCol() + colname = self.ProcessVariablesTable.GetColLabelValue(col, False) + value = self.ProcessVariablesTable.GetValue(row, col) + if not TestIdentifier(value): + message = _("\"%s\" is not a valid identifier!") % value + elif value.upper() in IEC_KEYWORDS: + message = _("\"%s\" is a keyword. It can't be used!") % value + elif value.upper() in [var["Name"].upper() for idx, var in enumerate(self.ProcessVariablesTable.GetData()) if idx != row]: + message = _("An variable named \"%s\" already exists!") % value + self.SaveProcessVariables() + wx.CallAfter(self.ProcessVariablesTable.ResetView, self.ProcessVariablesGrid) + dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR) + def OnProcessVariablesGridCellLeftClick(self, event): + def OnStartupCommandsGridCellChange(self, event): + row, col = event.GetRow(), event.GetCol() + colname = self.StartupCommandsTable.GetColLabelValue(col, False) + value = self.StartupCommandsTable.GetValue(row, col) + if colname == "Position": + if value not in self.Controler.GetSlaves(): + message = _("No slave defined at position %d!") % value + self.Controler.RemoveStartupCommand( + self.StartupCommandsTable.GetOldValue(), + self.StartupCommandsTable.GetValueByName(row, "command_idx")) + command = self.StartupCommandsTable.GetRow(row) + command_idx = self.Controler.AppendStartupCommand(command) + wx.CallAfter(self.RefreshStartupCommands) + wx.CallAfter(self.SelectStartupCommand, command["Position"], command_idx) + self.Controler.SetStartupCommandInfos(self.StartupCommandsTable.GetRow(row)) + dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR) + def OnResize(self, event): + self.EthercatMasterEditor.GetBestSize() + xstart, ystart = self.EthercatMasterEditor.GetViewStart() + window_size = self.EthercatMasterEditor.GetClientSize() + maxx, maxy = self.EthercatMasterEditor.GetMinSize() + posx = max(0, min(xstart, (maxx - window_size[0]) / SCROLLBAR_UNIT)) + posy = max(0, min(ystart, (maxy - window_size[1]) / SCROLLBAR_UNIT)) + self.EthercatMasterEditor.Scroll(posx, posy) + self.EthercatMasterEditor.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, + maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, posx, posy) def GetModulesTableColnames():
return [_("Name"), _("PDO alignment (bits)")]
-class LibraryEditorPanel(wx.ScrolledWindow):
+class LibraryEditorSizer(wx.FlexGridSizer): def __init__(self, parent, module_library, buttons):
- wx.ScrolledWindow.__init__(self, parent,
- style=wx.TAB_TRAVERSAL|wx.HSCROLL|wx.VSCROLL)
- self.Bind(wx.EVT_SIZE, self.OnResize)
+ wx.FlexGridSizer.__init__(self, cols=1, hgap=0, rows=4, vgap=5) self.ModuleLibrary = module_library
- main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=4, vgap=5)
- main_sizer.AddGrowableCol(0)
- main_sizer.AddGrowableRow(1)
- main_sizer.AddGrowableRow(3)
- ESI_files_label = wx.StaticText(self,
+ ESI_files_label = wx.StaticText(parent, - main_sizer.AddWindow(ESI_files_label, border=10,
+ self.AddWindow(ESI_files_label, border=10, flag=wx.TOP|wx.LEFT|wx.RIGHT)
folder_tree_sizer = wx.FlexGridSizer(cols=2, hgap=5, rows=1, vgap=0)
folder_tree_sizer.AddGrowableCol(0)
folder_tree_sizer.AddGrowableRow(0)
- main_sizer.AddSizer(folder_tree_sizer, border=10,
+ self.AddSizer(folder_tree_sizer, border=10, flag=wx.GROW|wx.LEFT|wx.RIGHT)
- self.ESIFiles = FolderTree(self, self.GetPath(), editable=False)
+ self.ESIFiles = FolderTree(parent, self.GetPath(), editable=False) self.ESIFiles.SetFilter(".xml")
- self.ESIFiles.SetMinSize(wx.Size(600, 300))
folder_tree_sizer.AddWindow(self.ESIFiles, flag=wx.GROW)
buttons_sizer = wx.BoxSizer(wx.VERTICAL)
@@ -199,7 +846,7 @@
flag=wx.ALIGN_CENTER_VERTICAL)
for idx, (name, bitmap, help, callback) in enumerate(buttons):
- button = wx.lib.buttons.GenBitmapButton(self,
+ button = wx.lib.buttons.GenBitmapButton(parent, bitmap=GetBitmap(bitmap),
size=wx.Size(28, 28), style=wx.NO_BORDER)
button.SetToolTipString(help)
@@ -211,28 +858,25 @@
callback = getattr(self, "On" + name, None)
- self.Bind(wx.EVT_BUTTON, callback, button)
+ parent.Bind(wx.EVT_BUTTON, callback, button) buttons_sizer.AddWindow(button, border=10, flag=flag)
- modules_label = wx.StaticText(self,
+ modules_label = wx.StaticText(parent, label=_("Modules library:"))
- main_sizer.AddSizer(modules_label, border=10,
+ self.AddSizer(modules_label, border=10, - self.ModulesGrid = wx.gizmos.TreeListCtrl(self,
+ self.ModulesGrid = wx.gizmos.TreeListCtrl(parent, style=wx.TR_DEFAULT_STYLE |
wx.TR_FULL_ROW_HIGHLIGHT)
- self.ModulesGrid.SetMinSize(wx.Size(600, 300))
self.ModulesGrid.GetMainWindow().Bind(wx.EVT_LEFT_DCLICK,
self.OnModulesGridLeftDClick)
- main_sizer.AddWindow(self.ModulesGrid, border=10,
+ self.AddWindow(self.ModulesGrid, border=10, flag=wx.GROW|wx.BOTTOM|wx.LEFT|wx.RIGHT)
- self.SetSizer(main_sizer)
for colname, colsize, colalign in zip(GetModulesTableColnames(),
[wx.ALIGN_LEFT, wx.ALIGN_RIGHT]):
@@ -242,6 +886,10 @@
return self.ModuleLibrary.GetPath()
+ def SetControlMinSize(self, size): + self.ESIFiles.SetMinSize(size) + self.ModulesGrid.SetMinSize(size) def GetSelectedFilePath(self):
return self.ESIFiles.GetPath()
@@ -349,18 +997,6 @@
- def OnResize(self, event):
- xstart, ystart = self.GetViewStart()
- window_size = self.GetClientSize()
- maxx, maxy = self.GetMinSize()
- posx = max(0, min(xstart, (maxx - window_size[0]) / SCROLLBAR_UNIT))
- posy = max(0, min(ystart, (maxy - window_size[1]) / SCROLLBAR_UNIT))
- self.Scroll(posx, posy)
- self.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT,
- maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, posx, posy)
class DatabaseManagementDialog(wx.Dialog):
@@ -373,10 +1009,11 @@
main_sizer.AddGrowableCol(0)
main_sizer.AddGrowableRow(0)
- self.DatabaseEditor = LibraryEditorPanel(self, database,
+ self.DatabaseSizer = LibraryEditorSizer(self, database, [("ImportButton", "ImportESI", _("Import file to ESI files database"), None),
("DeleteButton", "remove_element", _("Remove file from database"), None)])
- main_sizer.AddWindow(self.DatabaseEditor, border=10,
+ self.DatabaseSizer.SetControlMinSize(wx.Size(0, 0)) + main_sizer.AddSizer(self.DatabaseSizer, border=10, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
button_sizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
@@ -387,10 +1024,10 @@
self.SetSizer(main_sizer)
- self.DatabaseEditor.RefreshView()
+ self.DatabaseSizer.RefreshView() - return self.DatabaseEditor.GetSelectedFilePath()
+ return self.DatabaseSizer.GetSelectedFilePath() class LibraryEditor(ConfTreeNodeEditor):
@@ -398,11 +1035,17 @@
(_("Modules Library"), "_create_ModuleLibraryEditor")]
def _create_ModuleLibraryEditor(self, prnt):
- self.ModuleLibraryEditor = LibraryEditorPanel(prnt,
+ self.ModuleLibraryEditor = wx.ScrolledWindow(prnt, + style=wx.TAB_TRAVERSAL|wx.HSCROLL|wx.VSCROLL) + self.ModuleLibraryEditor.Bind(wx.EVT_SIZE, self.OnResize) + self.ModuleLibrarySizer = LibraryEditorSizer(self.ModuleLibraryEditor, self.Controler.GetModulesLibraryInstance(),
[("ImportButton", "ImportESI", _("Import ESI file"), None),
("AddButton", "ImportDatabase", _("Add file from ESI files database"), self.OnAddButton),
("DeleteButton", "remove_element", _("Remove file from library"), None)])
+ self.ModuleLibrarySizer.SetControlMinSize(wx.Size(0, 200)) + self.ModuleLibraryEditor.SetSizer(self.ModuleLibrarySizer) return self.ModuleLibraryEditor
@@ -413,7 +1056,7 @@
ConfTreeNodeEditor.RefreshView(self)
- self.ModuleLibraryEditor.RefreshView()
+ self.ModuleLibrarySizer.RefreshView() def OnAddButton(self, event):
dialog = DatabaseManagementDialog(self,
@@ -425,7 +1068,20 @@
- wx.CallAfter(self.ModuleLibraryEditor.RefreshView)
+ wx.CallAfter(self.ModuleLibrarySizer.RefreshView) + def OnResize(self, event): + self.ModuleLibraryEditor.GetBestSize() + xstart, ystart = self.ModuleLibraryEditor.GetViewStart() + window_size = self.ModuleLibraryEditor.GetClientSize() + maxx, maxy = self.ModuleLibraryEditor.GetMinSize() + posx = max(0, min(xstart, (maxx - window_size[0]) / SCROLLBAR_UNIT)) + posy = max(0, min(ystart, (maxy - window_size[1]) / SCROLLBAR_UNIT)) + self.ModuleLibraryEditor.Scroll(posx, posy) + self.ModuleLibraryEditor.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, + maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, posx, posy) --- a/etherlab/etherlab.py Wed Feb 27 22:40:45 2013 +0100
+++ b/etherlab/etherlab.py Tue Mar 05 00:59:34 2013 +0100
@@ -9,7 +9,7 @@
from POULibrary import POULibrary
from ConfigTreeNode import ConfigTreeNode
from PLCControler import UndoBuffer, LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP, LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY
-from ConfigEditor import NodeEditor, CIA402NodeEditor, LibraryEditor, ETHERCAT_VENDOR, ETHERCAT_GROUP, ETHERCAT_DEVICE
+from ConfigEditor import NodeEditor, CIA402NodeEditor, MasterEditor, LibraryEditor, ETHERCAT_VENDOR, ETHERCAT_GROUP, ETHERCAT_DEVICE from MotionLibrary import Headers, AxisXSD
@@ -161,6 +161,9 @@
return self.CTNParent.GetSlaveInfos(self.GetSlavePos())
+ def GetSlaveVariables(self, limits): + return self.CTNParent.GetSlaveVariables(self.GetSlavePos(), limits) def GetVariableLocationTree(self):
return {"name": self.BaseParams.getName(),
"type": LOCATION_CONFNODE,
@@ -470,6 +473,11 @@
raise ValueError, "Not supported base"
+def sort_commands(x, y): + if x["Index"] == y["Index"]: + return cmp(x["Subindex"], y["Subindex"]) + return cmp(x["Index"], y["Index"]) cls = EtherCATConfigClasses.get("Config_Slave", None)
@@ -488,37 +496,146 @@
slave_info.setProductCode(ExtractHexDecValue(type_infos["product_code"]))
slave_info.setRevisionNo(ExtractHexDecValue(type_infos["revision_number"]))
setattr(cls, "setType", setType)
+ def getInitCmds(self, create_default=False): + Mailbox = self.getMailbox() + Mailbox = self.getMailbox() + InitCmds = CoE.getInitCmds() + if InitCmds is None and create_default: + InitCmds = CoE.getInitCmds() + setattr(cls, "getInitCmds", getInitCmds) + def getStartupCommands(self): + pos = self.getInfo().getPhysAddr() + InitCmds = self.getInitCmds() + for idx, InitCmd in enumerate(InitCmds.getInitCmd()): + comment = InitCmd.getComment() + "Index": InitCmd.getIndex(), + "Subindex": InitCmd.getSubIndex(), + "Value": InitCmd.getData(), + "Description": comment}) + commands.sort(sort_commands) + setattr(cls, "getStartupCommands", getStartupCommands) + def appendStartupCommand(self, command_infos): + InitCmds = self.getInitCmds(True) + command = EtherCATConfigClasses["InitCmds_InitCmd"]() + command.setIndex(command_infos["Index"]) + command.setSubIndex(command_infos["Subindex"]) + command.setData(command_infos["Value"]) + command.setComment(command_infos["Description"]) + InitCmds.appendInitCmd(command) + return len(InitCmds.getInitCmd()) - 1 + setattr(cls, "appendStartupCommand", appendStartupCommand) + def setStartupCommand(self, command_infos): + InitCmds = self.getInitCmds() + if InitCmds is not None: + commands = InitCmds.getInitCmd() + if command_infos["command_idx"] < len(commands): + command = commands[command_infos["command_idx"]] + command.setIndex(command_infos["Index"]) + command.setSubIndex(command_infos["Subindex"]) + command.setData(command_infos["Value"]) + command.setComment(command_infos["Description"]) + setattr(cls, "setStartupCommand", setStartupCommand) + def removeStartupCommand(self, command_idx): + InitCmds = self.getInitCmds() + if InitCmds is not None: + if command_idx < len(InitCmds.getInitCmd()): + InitCmds.removeInitCmd(command_idx) + setattr(cls, "removeStartupCommand", removeStartupCommand) +ProcessVariablesXSD = """<?xml version="1.0" encoding="ISO-8859-1" ?> + <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema"> + <xsd:element name="ProcessVariables"> + <xsd:element name="variable" minOccurs="0" maxOccurs="unbounded"> + <xsd:element name="ReadFrom" type="LocationDesc" minOccurs="0"/> + <xsd:element name="WriteTo" type="LocationDesc" minOccurs="0"/> + <xsd:attribute name="Name" type="xsd:string" use="required"/> + <xsd:attribute name="Comment" type="xsd:string" use="required"/> + <xsd:complexType name="LocationDesc"> + <xsd:attribute name="Position" type="xsd:integer" use="required"/> + <xsd:attribute name="Index" type="xsd:integer" use="required"/> + <xsd:attribute name="SubIndex" type="xsd:integer" use="required"/> +ProcessVariablesClasses = GenerateClassesFromXSDstring(ProcessVariablesXSD) - XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>
- <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">
- <xsd:element name="EtherlabNode">
- <xsd:attribute name="MasterNumber" type="xsd:integer" use="optional" default="0"/>
CTNChildrenTypes = [("EthercatSlave", _EthercatSlaveCTN, "Ethercat Slave")]
CTNChildrenTypes.append(("EthercatCIA402Slave", _EthercatCIA402SlaveCTN, "Ethercat CIA402 Slave"))
+ EditorType = MasterEditor - filepath = self.ConfigFileName()
+ config_filepath = self.ConfigFileName() + config_is_saved = False self.Config = EtherCATConfigClasses["EtherCATConfig"]()
- if os.path.isfile(filepath):
- xmlfile = open(filepath, 'r')
- tree = minidom.parse(xmlfile)
+ if os.path.isfile(config_filepath): + config_xmlfile = open(config_filepath, 'r') + config_tree = minidom.parse(config_xmlfile) - for child in tree.childNodes:
- if child.nodeType == tree.ELEMENT_NODE and child.nodeName == "EtherCATConfig":
+ for child in config_tree.childNodes: + if child.nodeType == config_tree.ELEMENT_NODE and child.nodeName == "EtherCATConfig": self.Config.loadXMLTree(child)
- self.CreateConfigBuffer(True)
+ process_filepath = self.ProcessVariablesFileName() + process_is_saved = False + self.ProcessVariables = ProcessVariablesClasses["ProcessVariables"]() + if os.path.isfile(process_filepath): + process_xmlfile = open(process_filepath, 'r') + process_tree = minidom.parse(process_xmlfile) + process_xmlfile.close() + for child in process_tree.childNodes: + if child.nodeType == process_tree.ELEMENT_NODE and child.nodeName == "ProcessVariables": + self.ProcessVariables.loadXMLTree(child) + process_is_saved = True + if config_is_saved and process_is_saved: + self.CreateBuffer(True) - self.CreateConfigBuffer(False)
+ self.CreateBuffer(False) def ExtractHexDecValue(self, value):
@@ -529,7 +646,10 @@
def ConfigFileName(self):
return os.path.join(self.CTNPath(), "config.xml")
+ def ProcessVariablesFileName(self): + return os.path.join(self.CTNPath(), "process_variables.xml") for slave in self.Config.getConfig().getSlave():
@@ -544,6 +664,97 @@
+ def FilterSlave(self, slave, vendor=None, slave_pos=None, slave_profile=None): + if slave_pos is not None and slave.getInfo().getPhysAddr() != slave_pos: + type_infos = slave.getType() + if vendor is not None and ExtractHexDecValue(type_infos["vendor"]) != vendor: + device, alignment = self.GetModuleInfos(type_infos) + if slave_profile is not None and slave_profile not in device.GetProfileNumbers(): + def GetStartupCommands(self, vendor=None, slave_pos=None, slave_profile=None): + for slave in self.Config.getConfig().getSlave(): + if self.FilterSlave(slave, vendor, slave_pos, slave_profile): + commands.append((slave.getInfo().getPhysAddr(), slave.getStartupCommands())) + return reduce(lambda x, y: x + y[1], commands, []) + def AppendStartupCommand(self, command_infos): + slave = self.GetSlave(command_infos["Position"]) + command_idx = slave.appendStartupCommand(command_infos) + def SetStartupCommandInfos(self, command_infos): + slave = self.GetSlave(command_infos["Position"]) + slave.setStartupCommand(command_infos) + def RemoveStartupCommand(self, slave_pos, command_idx): + slave = self.GetSlave(slave_pos) + slave.removeStartupCommand(command_idx) + def SetProcessVariables(self, variables): + variable = ProcessVariablesClasses["ProcessVariables_variable"]() + variable.setName(var["Name"]) + variable.setComment(var["Description"]) + if var["ReadFrom"] != "": + position, index, subindex = var["ReadFrom"] + if variable.getReadFrom() is None: + read_from = variable.getReadFrom() + read_from.setPosition(position) + read_from.setIndex(index) + read_from.setSubIndex(subindex) + elif variable.getReadFrom() is not None: + variable.deleteReadFrom() + if var["WriteTo"] != "": + position, index, subindex = var["WriteTo"] + if variable.getWriteTo() is None: + write_to = variable.getWriteTo() + write_to.setPosition(position) + write_to.setIndex(index) + write_to.setSubIndex(subindex) + elif variable.getWriteTo() is not None: + variable.deleteWriteTo() + self.ProcessVariables.setvariable(vars) + def GetProcessVariables(self): + for variable in self.ProcessVariables.getvariable(): + var = {"Name": variable.getName(), + "Description": variable.getComment()} + read_from = variable.getReadFrom() + if read_from is not None: + var["ReadFrom"] = (read_from.getPosition(), + read_from.getSubIndex()) + write_to = variable.getWriteTo() + if write_to is not None: + var["WriteTo"] = (write_to.getPosition(), + write_to.getSubIndex()) app_frame = self.GetCTRoot().AppFrame
@@ -599,7 +810,7 @@
slave_infos.setPhysAddr(newConfNodeOpj.BaseParams.getIEC_Channel())
slave_infos.setAutoIncAddr(0)
self.Config.getConfig().appendSlave(slave)
@@ -611,7 +822,7 @@
slave_infos = slave.getInfo()
if slave_infos.getPhysAddr() == slave_pos:
ConfigTreeNode._doRemoveChild(self, CTNInstance)
@@ -620,7 +831,7 @@
slave_info = slave.getInfo()
slave_info.setPhysAddr(new_pos)
def GetSlaveAlias(self, slave_pos):
slave = self.GetSlave(slave_pos)
@@ -634,7 +845,7 @@
slave_info = slave.getInfo()
slave_info.setAutoIncAddr(alias)
def GetSlaveType(self, slave_pos):
slave = self.GetSlave(slave_pos)
@@ -646,50 +857,81 @@
slave = self.GetSlave(slave_pos)
slave.setType(type_infos)
def GetSlaveInfos(self, slave_pos):
slave = self.GetSlave(slave_pos)
type_infos = slave.getType()
- device, alignement = self.GetModuleInfos(type_infos)
+ device, alignment = self.GetModuleInfos(type_infos) infos = type_infos.copy()
- entries = device.GetEntriesList()
- entries_list = entries.items()
- for (index, subindex), entry in entries_list:
- if index != current_index:
- elif current_entry is not None:
- current_entry["children"].append(entry)
infos.update({"physics": device.getPhysics(),
"sync_managers": device.GetSyncManagers(),
+ "entries": self.GetSlaveVariables(device)}) + def GetSlaveVariables(self, slave_pos=None, limits=None, device=None): + if device is None and slave_pos is not None: + slave = self.GetSlave(slave_pos) + type_infos = slave.getType() + device, alignment = self.GetModuleInfos(type_infos) + entries = device.GetEntriesList(limits) + entries_list = entries.items() + for (index, subindex), entry in entries_list: + if slave_pos is not None: + entry["Position"] = str(slave_pos) + if index != current_index: + elif current_entry is not None: + current_entry["children"].append(entry) + def GetNodesVariables(self, vendor=None, slave_pos=None, slave_profile=None, limits=None): + for slave_position in self.GetSlaves(): + if slave_pos is not None and slave_position != slave_pos: + slave = self.GetSlave(slave_position) + type_infos = slave.getType() + if vendor is not None and ExtractHexDecValue(type_infos["vendor"]) != vendor: + device, alignment = self.GetModuleInfos(type_infos) + if slave_profile is not None and slave_profile not in device.GetProfileNumbers(): + entries.extend(self.GetSlaveVariables(slave_position, limits, device)) def GetModuleInfos(self, type_infos):
return self.CTNParent.GetModuleInfos(type_infos)
def GetSlaveTypesLibrary(self, profile_filter=None):
return self.CTNParent.GetModulesLibrary(profile_filter)
+ def GetLibraryVendors(self): + return self.CTNParent.GetVendors() def GetDeviceLocationTree(self, slave_pos, current_location, device_name):
slave = self.GetSlave(slave_pos)
type_infos = slave.getType()
- device, alignement = self.GetModuleInfos(type_infos)
+ device, alignment = self.GetModuleInfos(type_infos) for sync_manager in device.getSm():
@@ -725,21 +967,31 @@
def CTNTestModified(self):
- return self.ChangesToSave or not self.ConfigIsSaved()
+ return self.ChangesToSave or not self.ModelIsSaved() - filepath = self.ConfigFileName()
+ config_filepath = self.ConfigFileName() - text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n"
- extras = {"xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance",
+ config_text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n" + config_extras = {"xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance", "xsi:noNamespaceSchemaLocation" : "EtherCATInfo.xsd"}
- text += self.Config.generateXMLText("EtherCATConfig", 0, extras)
+ config_text += self.Config.generateXMLText("EtherCATConfig", 0, config_extras) - xmlfile = open(filepath,"w")
- xmlfile.write(text.encode("utf-8"))
+ config_xmlfile = open(config_filepath,"w") + config_xmlfile.write(config_text.encode("utf-8")) + process_filepath = self.ProcessVariablesFileName() - self.ConfigBuffer.CurrentSaved()
+ process_text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n" + process_extras = {"xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance"} + process_text += self.ProcessVariables.generateXMLText("ProcessVariables", 0, process_extras) + process_xmlfile = open(process_filepath,"w") + process_xmlfile.write(process_text.encode("utf-8")) + process_xmlfile.close() + self.Buffer.CurrentSaved() def _Generate_C(self, buildpath, locations):
@@ -810,27 +1062,27 @@
return cPickle.loads(cPickle.dumps(model))
- def CreateConfigBuffer(self, saved):
- self.ConfigBuffer = UndoBuffer(cPickle.dumps(self.Config), saved)
+ def CreateBuffer(self, saved): + self.Buffer = UndoBuffer(cPickle.dumps((self.Config, self.ProcessVariables)), saved) - def BufferConfig(self):
- self.ConfigBuffer.Buffering(cPickle.dumps(self.Config))
+ self.Buffer.Buffering(cPickle.dumps((self.Config, self.ProcessVariables))) - def ConfigIsSaved(self):
- if self.ConfigBuffer is not None:
- return self.ConfigBuffer.IsCurrentSaved()
+ def ModelIsSaved(self): + if self.Buffer is not None: + return self.Buffer.IsCurrentSaved() - self.Config = cPickle.loads(self.ConfigBuffer.Previous())
+ self.Config, self.ProcessVariables = cPickle.loads(self.Buffer.Previous()) - self.Config = cPickle.loads(self.ConfigBuffer.Next())
+ self.Config, self.ProcessVariables = cPickle.loads(self.Buffer.Next()) def GetBufferState(self):
- first = self.ConfigBuffer.IsFirst()
- last = self.ConfigBuffer.IsLast()
+ first = self.Buffer.IsFirst() + last = self.Buffer.IsLast() return not first, not last
@@ -983,7 +1235,7 @@
# Initialize strings for formatting master code template
"location": location_str,
- "master_number": etherlab_node_infos.getMasterNumber(),
+ "master_number": self.BaseParams.getIEC_Channel(), "located_variables_declaration": [],
"used_pdo_entry_offset_variables_declaration": [],
"used_pdo_entry_configuration": [],
@@ -1017,7 +1269,7 @@
slave_pos = (slave_alias, alias[slave_alias])
# Extract slave device informations
- device, alignement = self.Controler.GetModuleInfos(type_infos)
+ device, alignment = self.Controler.GetModuleInfos(type_infos) # Extract slaves variables to be mapped
@@ -1413,6 +1665,7 @@
for dictionary in self.GetProfileDictionaries():
datatypes = dictionary.getDataTypes()
if datatypes is not None:
@@ -1431,7 +1684,7 @@
setattr(cls, "getCoE", getCoE)
- def GetEntriesList(self):
+ def GetEntriesList(self, limits=None): if self.DataTypes is None:
@@ -1443,62 +1696,63 @@
for object in dictionary.getObjects().getObject():
entry_index = object.getIndex().getcontent()
index = ExtractHexDecValue(entry_index)
- entry_type = object.getType()
- entry_name = ExtractName(object.getName())
- entry_type_infos = self.DataTypes.get(entry_type, None)
- if entry_type_infos is not None:
- content = entry_type_infos.getcontent()
- for subitem in content["value"]:
- entry_subidx = subitem.getSubIdx()
- if entry_subidx is None:
- subidx = ExtractHexDecValue(entry_subidx)
- subitem_pdomapping = ""
- subitem_flags = subitem.getFlags()
- if subitem_flags is not None:
- access = subitem_flags.getAccess()
+ if limits is None or limits[0] <= index <= limits[1]: + entry_type = object.getType() + entry_name = ExtractName(object.getName()) + entry_type_infos = self.DataTypes.get(entry_type, None) + if entry_type_infos is not None: + content = entry_type_infos.getcontent() + for subitem in content["value"]: + entry_subidx = subitem.getSubIdx() + if entry_subidx is None: + subidx = ExtractHexDecValue(entry_subidx) + subitem_pdomapping = "" + subitem_flags = subitem.getFlags() + if subitem_flags is not None: + access = subitem_flags.getAccess() + subitem_access = access.getcontent() + pdomapping = subitem_flags.getPdoMapping() + if pdomapping is not None: + subitem_pdomapping = pdomapping.upper() + entries[(index, subidx)] = { + "SubIndex": entry_subidx, + (entry_name.decode("utf-8"), + ExtractName(subitem.getDisplayName(), + subitem.getName()).decode("utf-8")), + "Type": subitem.getType(), + "BitSize": subitem.getBitSize(), + "Access": subitem_access, + "PDOMapping": subitem_pdomapping} + entry_flags = object.getFlags() + if entry_flags is not None: + access = entry_flags.getAccess() - subitem_access = access.getcontent()
- pdomapping = subitem_flags.getPdoMapping()
+ entry_access = access.getcontent() + pdomapping = entry_flags.getPdoMapping() if pdomapping is not None:
- subitem_pdomapping = pdomapping.upper()
- entries[(index, subidx)] = {
- "SubIndex": entry_subidx,
- (entry_name.decode("utf-8"),
- ExtractName(subitem.getDisplayName(),
- subitem.getName()).decode("utf-8")),
- "Type": subitem.getType(),
- "BitSize": subitem.getBitSize(),
- "Access": subitem_access,
- "PDOMapping": subitem_pdomapping}
- entry_flags = object.getFlags()
- if entry_flags is not None:
- access = entry_flags.getAccess()
- entry_access = access.getcontent()
- pdomapping = entry_flags.getPdoMapping()
- if pdomapping is not None:
- entry_pdomapping = pdomapping.upper()
- entries[(index, 0)] = {
- "BitSize": object.getBitSize(),
- "Access": entry_access,
- "PDOMapping": entry_pdomapping}
+ entry_pdomapping = pdomapping.upper() + entries[(index, 0)] = { + "BitSize": object.getBitSize(), + "Access": entry_access, + "PDOMapping": entry_pdomapping} for TxPdo in self.getTxPdo():
- ExtractPdoInfos(TxPdo, "Transmit", entries)
+ ExtractPdoInfos(TxPdo, "Transmit", entries, limits) for RxPdo in self.getRxPdo():
- ExtractPdoInfos(RxPdo, "Receive", entries)
+ ExtractPdoInfos(RxPdo, "Receive", entries, limits) setattr(cls, "GetEntriesList", GetEntriesList)
@@ -1544,7 +1798,7 @@
-def ExtractPdoInfos(pdo, pdo_type, entries):
+def ExtractPdoInfos(pdo, pdo_type, entries, limits=None): pdo_index = pdo.getIndex().getcontent()
pdo_name = ExtractName(pdo.getName())
for pdo_entry in pdo.getEntry():
@@ -1553,27 +1807,28 @@
index = ExtractHexDecValue(entry_index)
subindex = ExtractHexDecValue(entry_subindex)
- entry = entries.get((index, subindex), None)
- entry["PDO index"] = pdo_index
- entry["PDO name"] = pdo_name
- entry["PDO type"] = pdo_type
- entry_type = pdo_entry.getDataType()
- if entry_type is not None:
- if pdo_type == "Transmit":
- entries[(index, subindex)] = {
- "SubIndex": entry_subindex,
- "Name": ExtractName(pdo_entry.getName()),
- "Type": entry_type.getcontent(),
- "PDOMapping": pdomapping}
+ if limits is None or limits[0] <= index <= limits[1]: + entry = entries.get((index, subindex), None) + entry["PDO index"] = pdo_index + entry["PDO name"] = pdo_name + entry["PDO type"] = pdo_type + entry_type = pdo_entry.getDataType() + if entry_type is not None: + if pdo_type == "Transmit": + entries[(index, subindex)] = { + "SubIndex": entry_subindex, + "Name": ExtractName(pdo_entry.getName()), + "Type": entry_type.getcontent(), + "PDOMapping": pdomapping} @@ -1585,7 +1840,10 @@
self.ParentLibrary = parent_library
+ if parent_library is not None: @@ -1633,6 +1891,8 @@
vendor_category["groups"][device_group]["devices"].append((device.getType().getcontent(), device))
def GetModulesLibrary(self, profile_filter=None):
+ if self.Library is None: for vendor_id, vendor in self.Library.iteritems():
@@ -1678,6 +1938,9 @@
library.sort(lambda x, y: cmp(x["name"], y["name"]))
+ return [(vendor_id, vendor["name"]) for vendor_id, vendor in self.Library.items()] def GetModuleInfos(self, module_infos):
vendor = ExtractHexDecValue(module_infos["vendor"])
vendor_infos = self.Library.get(vendor)
@@ -1785,6 +2048,9 @@
def GetModulesLibrary(self, profile_filter=None):
return self.ModulesLibrary.GetModulesLibrary(profile_filter)
+ return self.ModulesLibrary.GetVendors() def GetModuleInfos(self, module_infos):
return self.ModulesLibrary.GetModuleInfos(module_infos)