class DataTypeEditor(EditorPanel): def _init_Editor(self, parent): self.Editor = wx.Panel(parent, style=wx.SUNKEN_BORDER) self.MainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10) self.MainSizer.AddGrowableCol(0) self.MainSizer.AddGrowableRow(1) top_sizer = wx.BoxSizer(wx.HORIZONTAL) self.MainSizer.AddSizer(top_sizer, border=5, flag=wx.GROW | wx.TOP | wx.LEFT | wx.RIGHT) derivation_type_label = wx.StaticText(self.Editor, label=_('Derivation Type:')) top_sizer.AddWindow(derivation_type_label, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT) self.DerivationType = wx.ComboBox(self.Editor, size=wx.Size(200, -1), style=wx.CB_READONLY) self.Bind(wx.EVT_COMBOBOX, self.OnDerivationTypeChanged, self.DerivationType) top_sizer.AddWindow(self.DerivationType, border=5, flag=wx.GROW | wx.RIGHT) typeinfos_staticbox = wx.StaticBox(self.Editor, label=_('Type infos:')) typeinfos_sizer = wx.StaticBoxSizer(typeinfos_staticbox, wx.HORIZONTAL) self.MainSizer.AddSizer(typeinfos_sizer, border=5, flag=wx.GROW | wx.BOTTOM | wx.LEFT | wx.RIGHT) # Panel for Directly derived data types self.DirectlyPanel = wx.Panel(self.Editor, style=wx.TAB_TRAVERSAL) typeinfos_sizer.AddWindow(self.DirectlyPanel, 1) directly_panel_sizer = wx.BoxSizer(wx.HORIZONTAL) directly_basetype_label = wx.StaticText(self.DirectlyPanel, label=_('Base Type:')) directly_panel_sizer.AddWindow(directly_basetype_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.DirectlyBaseType = wx.ComboBox(self.DirectlyPanel, style=wx.CB_READONLY) self.Bind(wx.EVT_COMBOBOX, self.OnInfosChanged, self.DirectlyBaseType) directly_panel_sizer.AddWindow(self.DirectlyBaseType, 1, border=5, flag=wx.GROW | wx.ALL) directly_initialvalue_label = wx.StaticText(self.DirectlyPanel, label=_('Initial Value:')) directly_panel_sizer.AddWindow(directly_initialvalue_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.DirectlyInitialValue = wx.TextCtrl(self.DirectlyPanel, style=wx.TE_PROCESS_ENTER | wx.TE_RICH) self.Bind(wx.EVT_TEXT_ENTER, self.OnReturnKeyPressed, self.DirectlyInitialValue) directly_panel_sizer.AddWindow(self.DirectlyInitialValue, 1, border=5, flag=wx.ALL) self.DirectlyPanel.SetSizer(directly_panel_sizer) # Panel for Subrange data types self.SubrangePanel = wx.Panel(self.Editor, style=wx.TAB_TRAVERSAL) typeinfos_sizer.AddWindow(self.SubrangePanel, 1) subrange_panel_sizer = wx.GridSizer(cols=4, hgap=5, rows=3, vgap=0) subrange_basetype_label = wx.StaticText(self.SubrangePanel, label=_('Base Type:')) subrange_panel_sizer.AddWindow(subrange_basetype_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.SubrangeBaseType = wx.ComboBox(self.SubrangePanel, style=wx.CB_READONLY) self.Bind(wx.EVT_COMBOBOX, self.OnSubrangeBaseTypeChanged, self.SubrangeBaseType) subrange_panel_sizer.AddWindow(self.SubrangeBaseType, 1, border=5, flag=wx.GROW | wx.ALL) subrange_initialvalue_label = wx.StaticText(self.SubrangePanel, label=_('Initial Value:')) subrange_panel_sizer.AddWindow(subrange_initialvalue_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.SubrangeInitialValue = wx.SpinCtrl(self.SubrangePanel, style=wx.TAB_TRAVERSAL) self.Bind(wx.EVT_SPINCTRL, self.OnInfosChanged, self.SubrangeInitialValue) subrange_panel_sizer.AddWindow(self.SubrangeInitialValue, 1, border=5, flag=wx.GROW | wx.ALL) subrange_minimum_label = wx.StaticText(self.SubrangePanel, label=_('Minimum:')) subrange_panel_sizer.AddWindow(subrange_minimum_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.SubrangeMinimum = wx.SpinCtrl(self.SubrangePanel, style=wx.TAB_TRAVERSAL) self.Bind(wx.EVT_SPINCTRL, self.OnSubrangeMinimumChanged, self.SubrangeMinimum) subrange_panel_sizer.AddWindow(self.SubrangeMinimum, 1, border=5, flag=wx.GROW | wx.ALL) for i in xrange(2): subrange_panel_sizer.AddWindow(wx.Size(0, 0), 1) subrange_maximum_label = wx.StaticText(self.SubrangePanel, label=_('Maximum:')) subrange_panel_sizer.AddWindow(subrange_maximum_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.SubrangeMaximum = wx.SpinCtrl(self.SubrangePanel, style=wx.TAB_TRAVERSAL) self.Bind(wx.EVT_SPINCTRL, self.OnSubrangeMaximumChanged, self.SubrangeMaximum) subrange_panel_sizer.AddWindow(self.SubrangeMaximum, 1, border=5, flag=wx.GROW | wx.ALL) self.SubrangePanel.SetSizer(subrange_panel_sizer) # Panel for Enumerated data types self.EnumeratedPanel = wx.Panel(self.Editor, style=wx.TAB_TRAVERSAL) typeinfos_sizer.AddWindow(self.EnumeratedPanel, 1) enumerated_panel_sizer = wx.BoxSizer(wx.HORIZONTAL) self.EnumeratedValues = CustomEditableListBox( self.EnumeratedPanel, label=_("Values:"), style=(wx.gizmos.EL_ALLOW_NEW | wx.gizmos.EL_ALLOW_EDIT | wx.gizmos.EL_ALLOW_DELETE)) setattr(self.EnumeratedValues, "_OnLabelEndEdit", self.OnEnumeratedValueEndEdit) for func in [ "_OnAddButton", "_OnDelButton", "_OnUpButton", "_OnDownButton" ]: setattr(self.EnumeratedValues, func, self.OnEnumeratedValuesChanged) enumerated_panel_sizer.AddWindow(self.EnumeratedValues, 1, border=5, flag=wx.GROW | wx.ALL) enumerated_panel_rightsizer = wx.BoxSizer(wx.HORIZONTAL) enumerated_panel_sizer.AddSizer(enumerated_panel_rightsizer, 1) enumerated_initialvalue_label = wx.StaticText( self.EnumeratedPanel, label=_('Initial Value:')) enumerated_panel_rightsizer.AddWindow(enumerated_initialvalue_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.EnumeratedInitialValue = wx.ComboBox(self.EnumeratedPanel, style=wx.CB_READONLY) self.Bind(wx.EVT_COMBOBOX, self.OnInfosChanged, self.EnumeratedInitialValue) enumerated_panel_rightsizer.AddWindow(self.EnumeratedInitialValue, 1, border=5, flag=wx.ALL) self.EnumeratedPanel.SetSizer(enumerated_panel_sizer) # Panel for Array data types self.ArrayPanel = wx.Panel(self.Editor, style=wx.TAB_TRAVERSAL) typeinfos_sizer.AddWindow(self.ArrayPanel, 1) array_panel_sizer = wx.FlexGridSizer(cols=2, hgap=5, rows=2, vgap=0) array_panel_sizer.AddGrowableCol(0) array_panel_sizer.AddGrowableCol(1) array_panel_sizer.AddGrowableRow(1) array_panel_leftSizer = wx.BoxSizer(wx.HORIZONTAL) array_panel_sizer.AddSizer(array_panel_leftSizer, flag=wx.GROW) array_basetype_label = wx.StaticText(self.ArrayPanel, label=_('Base Type:')) array_panel_leftSizer.AddWindow(array_basetype_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.ArrayBaseType = wx.ComboBox(self.ArrayPanel, style=wx.CB_READONLY) self.Bind(wx.EVT_COMBOBOX, self.OnInfosChanged, self.ArrayBaseType) array_panel_leftSizer.AddWindow(self.ArrayBaseType, 1, border=5, flag=wx.GROW | wx.ALL) array_panel_rightsizer = wx.BoxSizer(wx.HORIZONTAL) array_panel_sizer.AddSizer(array_panel_rightsizer, flag=wx.GROW) array_initialvalue_label = wx.StaticText(self.ArrayPanel, label=_('Initial Value:')) array_panel_rightsizer.AddWindow(array_initialvalue_label, 1, border=5, flag=wx.ALIGN_CENTER_VERTICAL | wx.ALL) self.ArrayInitialValue = wx.TextCtrl(self.ArrayPanel, style=wx.TE_PROCESS_ENTER | wx.TE_RICH) self.Bind(wx.EVT_TEXT_ENTER, self.OnReturnKeyPressed, self.ArrayInitialValue) array_panel_rightsizer.AddWindow(self.ArrayInitialValue, 1, border=5, flag=wx.ALL) self.ArrayDimensions = CustomEditableListBox( self.ArrayPanel, label=_("Dimensions:"), style=(wx.gizmos.EL_ALLOW_NEW | wx.gizmos.EL_ALLOW_EDIT | wx.gizmos.EL_ALLOW_DELETE)) for func in [ "_OnLabelEndEdit", "_OnAddButton", "_OnDelButton", "_OnUpButton", "_OnDownButton" ]: setattr(self.ArrayDimensions, func, self.OnDimensionsChanged) array_panel_sizer.AddWindow(self.ArrayDimensions, 0, border=5, flag=wx.GROW | wx.ALL) self.ArrayPanel.SetSizer(array_panel_sizer) # Panel for Structure data types self.StructurePanel = wx.Panel(self.Editor, style=wx.TAB_TRAVERSAL) typeinfos_sizer.AddWindow(self.StructurePanel, 1) structure_panel_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0) structure_panel_sizer.AddGrowableCol(0) structure_panel_sizer.AddGrowableRow(1) structure_button_sizer = wx.FlexGridSizer(cols=5, hgap=5, rows=1, vgap=0) structure_button_sizer.AddGrowableCol(0) structure_button_sizer.AddGrowableRow(0) structure_panel_sizer.AddSizer(structure_button_sizer, 0, border=5, flag=wx.ALL | wx.GROW) structure_elements_label = wx.StaticText(self.StructurePanel, label=_('Elements :')) structure_button_sizer.AddWindow(structure_elements_label, flag=wx.ALIGN_BOTTOM) for name, bitmap, help in [ ("StructureAddButton", "add_element", _("Add element")), ("StructureDeleteButton", "remove_element", _("Remove element")), ("StructureUpButton", "up", _("Move element up")), ("StructureDownButton", "down", _("Move element down")) ]: button = wx.lib.buttons.GenBitmapButton(self.StructurePanel, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) structure_button_sizer.AddWindow(button) self.StructureElementsGrid = CustomGrid(self.StructurePanel, size=wx.Size(0, 150), style=wx.VSCROLL) self.StructureElementsGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnStructureElementsGridCellChange) self.StructureElementsGrid.Bind( wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnStructureElementsGridEditorShown) structure_panel_sizer.AddWindow(self.StructureElementsGrid, flag=wx.GROW) self.StructurePanel.SetSizer(structure_panel_sizer) self.Editor.SetSizer(self.MainSizer) def __init__(self, parent, tagname, window, controler): EditorPanel.__init__(self, parent, tagname, window, controler) self.StructureElementDefaultValue = { "Name": "", "Type": DefaultType, "Initial Value": "" } self.StructureElementsTable = ElementsTable(self, [], GetElementsTableColnames()) self.StructureColSizes = [40, 150, 100, 250] self.StructureColAlignements = [ wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT ] self.StructureElementsGrid.SetTable(self.StructureElementsTable) self.StructureElementsGrid.SetButtons({ "Add": self.StructureAddButton, "Delete": self.StructureDeleteButton, "Up": self.StructureUpButton, "Down": self.StructureDownButton }) def _AddStructureElement(new_row): self.StructureElementsTable.InsertRow( new_row, self.StructureElementDefaultValue.copy()) self.RefreshTypeInfos() self.StructureElementsTable.ResetView(self.StructureElementsGrid) return new_row setattr(self.StructureElementsGrid, "_AddRow", _AddStructureElement) def _DeleteStructureElement(row): self.StructureElementsTable.RemoveRow(row) self.RefreshTypeInfos() self.StructureElementsTable.ResetView(self.StructureElementsGrid) setattr(self.StructureElementsGrid, "_DeleteRow", _DeleteStructureElement) def _MoveStructureElement(row, move): new_row = self.StructureElementsTable.MoveRow(row, move) if new_row != row: self.RefreshTypeInfos() self.StructureElementsTable.ResetView( self.StructureElementsGrid) return new_row setattr(self.StructureElementsGrid, "_MoveRow", _MoveStructureElement) self.StructureElementsGrid.SetRowLabelSize(0) for col in range(self.StructureElementsTable.GetNumberCols()): attr = wx.grid.GridCellAttr() attr.SetAlignment(self.StructureColAlignements[col], wx.ALIGN_CENTRE) self.StructureElementsGrid.SetColAttr(col, attr) self.StructureElementsGrid.SetColMinimalWidth( col, self.StructureColSizes[col]) self.StructureElementsGrid.AutoSizeColumn(col, False) self.StructureElementsGrid.RefreshButtons() for datatype in GetDatatypeTypes(): self.DerivationType.Append(_(datatype)) self.SubrangePanel.Hide() self.EnumeratedPanel.Hide() self.ArrayPanel.Hide() self.StructurePanel.Hide() self.CurrentPanel = "Directly" self.Highlights = [] self.Initializing = False self.HighlightControls = { ("Directly", "base"): self.DirectlyBaseType, ("Directly", "initial"): self.DirectlyInitialValue, ("Subrange", "base"): self.SubrangeBaseType, ("Subrange", "lower"): self.SubrangeMinimum, ("Subrange", "upper"): self.SubrangeMaximum, ("Subrange", "initial"): self.SubrangeInitialValue, ("Enumerated", "value"): self.EnumeratedValues, ("Enumerated", "initial"): self.EnumeratedInitialValue, ("Array", "initial"): self.ArrayInitialValue, ("Array", "base"): self.ArrayBaseType, ("Array", "range"): self.ArrayDimensions, } self.RefreshHighlightsTimer = wx.Timer(self, -1) self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer) def __del__(self): self.RefreshHighlightsTimer.Stop() def GetBufferState(self): return self.Controler.GetBufferState() def Undo(self): self.Controler.LoadPrevious() self.ParentWindow.CloseTabsWithoutModel() def Redo(self): self.Controler.LoadNext() self.ParentWindow.CloseTabsWithoutModel() def HasNoModel(self): return self.Controler.GetEditedElement(self.TagName) is None def RefreshView(self): self.Initializing = True self.DirectlyBaseType.Clear() self.ArrayBaseType.Clear() for datatype in self.Controler.GetDataTypes(self.TagName): self.DirectlyBaseType.Append(datatype) self.ArrayBaseType.Append(datatype) self.DirectlyBaseType.SetSelection(0) self.SubrangeBaseType.Clear() words = self.TagName.split("::") for base_type in self.Controler.GetSubrangeBaseTypes(words[1]): self.SubrangeBaseType.Append(base_type) self.SubrangeBaseType.SetSelection(0) self.RefreshBoundsRange() type_infos = self.Controler.GetDataTypeInfos(self.TagName) if type_infos is not None: datatype = type_infos["type"] self.DerivationType.SetStringSelection(_(datatype)) if type_infos["type"] == "Directly": self.DirectlyBaseType.SetStringSelection( type_infos["base_type"]) self.DirectlyInitialValue.SetValue(type_infos["initial"]) elif type_infos["type"] == "Subrange": self.SubrangeBaseType.SetStringSelection( type_infos["base_type"]) self.RefreshBoundsRange() self.SubrangeMinimum.SetValue(int(type_infos["min"])) self.SubrangeMaximum.SetValue(int(type_infos["max"])) self.RefreshSubrangeInitialValueRange() if type_infos["initial"] != "": self.SubrangeInitialValue.SetValue( int(type_infos["initial"])) else: self.SubrangeInitialValue.SetValue(type_infos["min"]) elif type_infos["type"] == "Enumerated": self.EnumeratedValues.SetStrings(type_infos["values"]) self.RefreshEnumeratedValues() self.EnumeratedInitialValue.SetStringSelection( type_infos["initial"]) elif type_infos["type"] == "Array": self.ArrayBaseType.SetStringSelection(type_infos["base_type"]) self.ArrayDimensions.SetStrings( map(lambda x: "..".join(x), type_infos["dimensions"])) self.ArrayInitialValue.SetValue(type_infos["initial"]) elif type_infos["type"] == "Structure": self.StructureElementsTable.SetData(type_infos["elements"]) self.RefreshDisplayedInfos() self.ShowHighlights() self.StructureElementsTable.ResetView(self.StructureElementsGrid) self.StructureElementsGrid.RefreshButtons() self.Initializing = False def OnDerivationTypeChanged(self, event): wx.CallAfter(self.RefreshDisplayedInfos) wx.CallAfter(self.RefreshTypeInfos) event.Skip() def OnReturnKeyPressed(self, event): self.RefreshTypeInfos() def OnInfosChanged(self, event): self.RefreshTypeInfos() event.Skip() def OnSubrangeBaseTypeChanged(self, event): self.RefreshBoundsRange() self.RefreshTypeInfos() event.Skip() def OnSubrangeMinimumChanged(self, event): if not self.Initializing: wx.CallAfter( self.SubrangeMinimum.SetValue, min(self.SubrangeMaximum.GetValue(), self.SubrangeMinimum.GetValue())) wx.CallAfter(self.RefreshSubrangeInitialValueRange) wx.CallAfter(self.RefreshTypeInfos) event.Skip() def OnSubrangeMaximumChanged(self, event): if not self.Initializing: wx.CallAfter( self.SubrangeMaximum.SetValue, max(self.SubrangeMinimum.GetValue(), self.SubrangeMaximum.GetValue())) wx.CallAfter(self.RefreshSubrangeInitialValueRange) wx.CallAfter(self.RefreshTypeInfos) event.Skip() def OnDimensionsChanged(self, event): wx.CallAfter(self.RefreshTypeInfos) event.Skip() def OnEnumeratedValueEndEdit(self, event): text = event.GetText() values = self.EnumeratedValues.GetStrings() index = event.GetIndex() if index >= len(values) or values[index].upper() != text.upper(): if text.upper() in [value.upper() for value in values]: message = wx.MessageDialog( self, _("\"%s\" value already defined!") % text, _("Error"), wx.OK | wx.ICON_ERROR) message.ShowModal() message.Destroy() event.Veto() elif text.upper() in IEC_KEYWORDS: message = wx.MessageDialog( self, _("\"%s\" is a keyword. It can't be used!") % text, _("Error"), wx.OK | wx.ICON_ERROR) message.ShowModal() message.Destroy() else: initial_selected = None if index < len( values ) and self.EnumeratedInitialValue.GetStringSelection( ) == values[index]: initial_selected = text wx.CallAfter(self.RefreshEnumeratedValues, initial_selected) wx.CallAfter(self.RefreshTypeInfos) event.Skip() else: event.Skip() def OnEnumeratedValuesChanged(self, event): wx.CallAfter(self.RefreshEnumeratedValues) wx.CallAfter(self.RefreshTypeInfos) event.Skip() def ShowErrorMessage(self, message): dialog = wx.MessageDialog(self, message, _("Error"), wx.OK | wx.ICON_ERROR) dialog.ShowModal() dialog.Destroy() def OnStructureElementsGridCellChange(self, event): row, col = event.GetRow(), event.GetCol() colname = self.StructureElementsTable.GetColLabelValue(col, False) value = self.StructureElementsTable.GetValue(row, col) if colname == "Name": message = None 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 self.PouNames: # message = _("A pou with \"%s\" as name exists!")%value elif value.upper() in [ var["Name"].upper() for idx, var in enumerate( self.StructureElementsTable.GetData()) if idx != row ]: message = _( "An element named \"%s\" already exists in this structure!" ) % value else: self.RefreshTypeInfos() wx.CallAfter(self.StructureElementsTable.ResetView, self.StructureElementsGrid) # old_value = self.Table.GetOldValue() # if old_value != "": # self.Controler.UpdateEditedElementUsedVariable(self.TagName, old_value, value) # self.Controler.BufferProject() event.Skip() if message is not None: event.Veto() wx.CallAfter(self.ShowErrorMessage, message) else: self.RefreshTypeInfos() wx.CallAfter(self.StructureElementsTable.ResetView, self.StructureElementsGrid) event.Skip() def OnStructureElementsGridSelectCell(self, event): wx.CallAfter(self.RefreshStructureButtons) event.Skip() def OnStructureElementsGridEditorShown(self, event): row, col = event.GetRow(), event.GetCol() if self.StructureElementsTable.GetColLabelValue(col, False) == "Type": type_menu = wx.Menu(title='') base_menu = wx.Menu(title='') for base_type in self.Controler.GetBaseTypes(): new_id = wx.NewId() AppendMenu(base_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=base_type) self.Bind(wx.EVT_MENU, self.GetElementTypeFunction(base_type), id=new_id) type_menu.AppendMenu(wx.NewId(), _("Base Types"), base_menu) datatype_menu = wx.Menu(title='') for datatype in self.Controler.GetDataTypes(self.TagName, False): new_id = wx.NewId() AppendMenu(datatype_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=datatype) self.Bind(wx.EVT_MENU, self.GetElementTypeFunction(datatype), id=new_id) type_menu.AppendMenu(wx.NewId(), _("User Data Types"), datatype_menu) new_id = wx.NewId() AppendMenu(type_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=_("Array")) self.Bind(wx.EVT_MENU, self.ElementArrayTypeFunction, id=new_id) # functionblock_menu = wx.Menu(title='') # bodytype = self.Controler.GetEditedElementBodyType(self.TagName) # pouname, poutype = self.Controler.GetEditedElementType(self.TagName) # if classtype in ["Input","Output","InOut","External","Global"] or poutype != "function" and bodytype in ["ST", "IL"]: # for functionblock_type in self.Controler.GetFunctionBlockTypes(self.TagName): # new_id = wx.NewId() # AppendMenu(functionblock_menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=functionblock_type) # self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(functionblock_type), id=new_id) # type_menu.AppendMenu(wx.NewId(), _("Function Block Types"), functionblock_menu) rect = self.StructureElementsGrid.BlockToDeviceRect((row, col), (row, col)) self.StructureElementsGrid.PopupMenuXY( type_menu, rect.x + rect.width, rect.y + self.StructureElementsGrid.GetColLabelSize()) type_menu.Destroy() event.Veto() else: event.Skip() def GetElementTypeFunction(self, base_type): def ElementTypeFunction(event): row = self.StructureElementsGrid.GetGridCursorRow() self.StructureElementsTable.SetValueByName(row, "Type", base_type) self.RefreshTypeInfos() self.StructureElementsTable.ResetView(self.StructureElementsGrid) return ElementTypeFunction def ElementArrayTypeFunction(self, event): row = self.StructureElementsGrid.GetGridCursorRow() dialog = ArrayTypeDialog( self, self.Controler.GetDataTypes(self.TagName), self.StructureElementsTable.GetValueByName(row, "Type")) if dialog.ShowModal() == wx.ID_OK: self.StructureElementsTable.SetValueByName(row, "Type", dialog.GetValue()) self.RefreshTypeInfos() self.StructureElementsTable.ResetView(self.StructureElementsGrid) dialog.Destroy() def RefreshDisplayedInfos(self): selected = DATATYPE_TYPES_DICT[ self.DerivationType.GetStringSelection()] if selected != self.CurrentPanel: if self.CurrentPanel == "Directly": self.DirectlyPanel.Hide() elif self.CurrentPanel == "Subrange": self.SubrangePanel.Hide() elif self.CurrentPanel == "Enumerated": self.EnumeratedPanel.Hide() elif self.CurrentPanel == "Array": self.ArrayPanel.Hide() elif self.CurrentPanel == "Structure": self.StructurePanel.Hide() self.CurrentPanel = selected if selected == "Directly": self.DirectlyPanel.Show() elif selected == "Subrange": self.SubrangePanel.Show() elif selected == "Enumerated": self.EnumeratedPanel.Show() elif selected == "Array": self.ArrayPanel.Show() elif selected == "Structure": self.StructurePanel.Show() self.MainSizer.Layout() def RefreshEnumeratedValues(self, initial_selected=None): if initial_selected is None: initial_selected = self.EnumeratedInitialValue.GetStringSelection() self.EnumeratedInitialValue.Clear() self.EnumeratedInitialValue.Append("") for value in self.EnumeratedValues.GetStrings(): self.EnumeratedInitialValue.Append(value) self.EnumeratedInitialValue.SetStringSelection(initial_selected) def RefreshBoundsRange(self): range = self.Controler.GetDataTypeRange( self.SubrangeBaseType.GetStringSelection()) if range is not None: min_value, max_value = range self.SubrangeMinimum.SetRange(min_value, max_value) self.SubrangeMinimum.SetValue( min(max(min_value, self.SubrangeMinimum.GetValue()), max_value)) self.SubrangeMaximum.SetRange(min_value, max_value) self.SubrangeMaximum.SetValue( min(max(min_value, self.SubrangeMaximum.GetValue()), max_value)) def RefreshSubrangeInitialValueRange(self): self.SubrangeInitialValue.SetRange(self.SubrangeMinimum.GetValue(), self.SubrangeMaximum.GetValue()) def RefreshTypeInfos(self): selected = DATATYPE_TYPES_DICT[ self.DerivationType.GetStringSelection()] infos = {"type": selected} if selected == "Directly": infos["base_type"] = self.DirectlyBaseType.GetStringSelection() infos["initial"] = self.DirectlyInitialValue.GetValue() elif selected == "Subrange": infos["base_type"] = self.SubrangeBaseType.GetStringSelection() infos["min"] = str(self.SubrangeMinimum.GetValue()) infos["max"] = str(self.SubrangeMaximum.GetValue()) initial_value = self.SubrangeInitialValue.GetValue() if initial_value == infos["min"]: infos["initial"] = "" else: infos["initial"] = str(initial_value) elif selected == "Enumerated": infos["values"] = self.EnumeratedValues.GetStrings() infos["initial"] = self.EnumeratedInitialValue.GetStringSelection() elif selected == "Array": infos["base_type"] = self.ArrayBaseType.GetStringSelection() infos["dimensions"] = [] for dimensions in self.ArrayDimensions.GetStrings(): result = DIMENSION_MODEL.match(dimensions) if result is None: message = wx.MessageDialog( self, _("\"%s\" value isn't a valid array dimension!") % dimensions, _("Error"), wx.OK | wx.ICON_ERROR) message.ShowModal() message.Destroy() self.RefreshView() return bounds = result.groups() if int(bounds[0]) >= int(bounds[1]): message = wx.MessageDialog( self, _("\"%s\" value isn't a valid array dimension!\nRight value must be greater than left value." ) % dimensions, _("Error"), wx.OK | wx.ICON_ERROR) message.ShowModal() message.Destroy() self.RefreshView() return infos["dimensions"].append(bounds) infos["initial"] = self.ArrayInitialValue.GetValue() elif selected == "Structure": infos["elements"] = self.StructureElementsTable.GetData() infos["initial"] = "" self.Controler.SetDataTypeInfos(self.TagName, infos) self.ParentWindow.RefreshTitle() self.ParentWindow.RefreshFileMenu() self.ParentWindow.RefreshEditMenu() # ------------------------------------------------------------------------------- # Highlights showing functions # ------------------------------------------------------------------------------- def OnRefreshHighlightsTimer(self, event): self.RefreshView() event.Skip() def ClearHighlights(self, highlight_type=None): if highlight_type is None: self.Highlights = [] else: self.Highlights = [(infos, start, end, highlight) for (infos, start, end, highlight) in self.Highlights if highlight != highlight_type] for control in self.HighlightControls.itervalues(): if isinstance(control, (wx.ComboBox, wx.SpinCtrl)): control.SetBackgroundColour(wx.NullColour) control.SetForegroundColour(wx.NullColour) elif isinstance(control, wx.TextCtrl): value = control.GetValue() control.SetStyle(0, len(value), wx.TextAttr(wx.NullColour)) elif isinstance(control, wx.gizmos.EditableListBox): listctrl = control.GetListCtrl() for i in xrange(listctrl.GetItemCount()): listctrl.SetItemBackgroundColour(i, wx.NullColour) listctrl.SetItemTextColour(i, wx.NullColour) self.StructureElementsTable.ClearHighlights(highlight_type) self.RefreshView() def AddHighlight(self, infos, start, end, highlight_type): self.Highlights.append((infos, start, end, highlight_type)) self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True) def ShowHighlights(self): type_infos = self.Controler.GetDataTypeInfos(self.TagName) for infos, start, end, highlight_type in self.Highlights: if infos[0] == "struct": self.StructureElementsTable.AddHighlight( infos[1:], highlight_type) else: control = self.HighlightControls.get( (type_infos["type"], infos[0]), None) if control is not None: if isinstance(control, (wx.ComboBox, wx.SpinCtrl)): control.SetBackgroundColour(highlight_type[0]) control.SetForegroundColour(highlight_type[1]) elif isinstance(control, wx.TextCtrl): control.SetStyle( start[1], end[1] + 1, wx.TextAttr(highlight_type[1], highlight_type[0])) elif isinstance(control, wx.gizmos.EditableListBox): listctrl = control.GetListCtrl() listctrl.SetItemBackgroundColour( infos[1], highlight_type[0]) listctrl.SetItemTextColour(infos[1], highlight_type[1]) listctrl.Select(listctrl.FocusedItem, False)
class ResourceEditor(EditorPanel): VARIABLE_PANEL_TYPE = "resource" def _init_Editor(self, parent): self.Editor = wx.Panel(parent, style=wx.SUNKEN_BORDER | wx.TAB_TRAVERSAL) main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5) main_sizer.AddGrowableCol(0) main_sizer.AddGrowableRow(0) main_sizer.AddGrowableRow(1) tasks_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5) tasks_sizer.AddGrowableCol(0) tasks_sizer.AddGrowableRow(1) main_sizer.AddSizer(tasks_sizer, border=5, flag=wx.GROW | wx.TOP | wx.LEFT | wx.RIGHT) tasks_buttons_sizer = wx.FlexGridSizer(cols=5, hgap=5, rows=1, vgap=0) tasks_buttons_sizer.AddGrowableCol(0) tasks_buttons_sizer.AddGrowableRow(0) tasks_sizer.AddSizer(tasks_buttons_sizer, flag=wx.GROW) tasks_label = wx.StaticText(self.Editor, label=_(u'Tasks:')) tasks_buttons_sizer.AddWindow(tasks_label, flag=wx.ALIGN_BOTTOM) for name, bitmap, help in [ ("AddTaskButton", "add_element", _("Add task")), ("DeleteTaskButton", "remove_element", _("Remove task")), ("UpTaskButton", "up", _("Move task up")), ("DownTaskButton", "down", _("Move task down")) ]: button = wx.lib.buttons.GenBitmapButton(self.Editor, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) tasks_buttons_sizer.AddWindow(button) self.TasksGrid = CustomGrid(self.Editor, style=wx.VSCROLL) self.TasksGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnTasksGridCellChange) tasks_sizer.AddWindow(self.TasksGrid, flag=wx.GROW) instances_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5) instances_sizer.AddGrowableCol(0) instances_sizer.AddGrowableRow(1) main_sizer.AddSizer(instances_sizer, border=5, flag=wx.GROW | wx.BOTTOM | wx.LEFT | wx.RIGHT) instances_buttons_sizer = wx.FlexGridSizer(cols=5, hgap=5, rows=1, vgap=0) instances_buttons_sizer.AddGrowableCol(0) instances_buttons_sizer.AddGrowableRow(0) instances_sizer.AddSizer(instances_buttons_sizer, flag=wx.GROW) instances_label = wx.StaticText(self.Editor, label=_(u'Instances:')) instances_buttons_sizer.AddWindow(instances_label, flag=wx.ALIGN_BOTTOM) for name, bitmap, help in [ ("AddInstanceButton", "add_element", _("Add instance")), ("DeleteInstanceButton", "remove_element", _("Remove instance")), ("UpInstanceButton", "up", _("Move instance up")), ("DownInstanceButton", "down", _("Move instance down")) ]: button = wx.lib.buttons.GenBitmapButton(self.Editor, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) instances_buttons_sizer.AddWindow(button) self.InstancesGrid = CustomGrid(self.Editor, style=wx.VSCROLL) self.InstancesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnInstancesGridCellChange) instances_sizer.AddWindow(self.InstancesGrid, flag=wx.GROW) self.Editor.SetSizer(main_sizer) def __init__(self, parent, tagname, window, controler): EditorPanel.__init__(self, parent, tagname, window, controler) self.RefreshHighlightsTimer = wx.Timer(self, -1) self.Bind(wx.EVT_TIMER, self.OnRefreshHighlightsTimer, self.RefreshHighlightsTimer) self.TasksDefaultValue = { "Name": "", "Triggering": "", "Single": "", "Interval": "", "Priority": 0 } self.TasksTable = ResourceTable(self, [], GetTasksTableColnames()) self.TasksTable.SetColAlignements([ wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_RIGHT, wx.ALIGN_RIGHT ]) self.TasksTable.SetColSizes([200, 100, 100, 150, 100]) self.TasksGrid.SetTable(self.TasksTable) self.TasksGrid.SetButtons({ "Add": self.AddTaskButton, "Delete": self.DeleteTaskButton, "Up": self.UpTaskButton, "Down": self.DownTaskButton }) def _AddTask(new_row): self.TasksTable.InsertRow(new_row, self.TasksDefaultValue.copy()) self.RefreshModel() self.RefreshView() return new_row setattr(self.TasksGrid, "_AddRow", _AddTask) def _DeleteTask(row): self.TasksTable.RemoveRow(row) self.RefreshModel() self.RefreshView() setattr(self.TasksGrid, "_DeleteRow", _DeleteTask) def _MoveTask(row, move): new_row = self.TasksTable.MoveRow(row, move) if new_row != row: self.RefreshModel() self.RefreshView() return new_row setattr(self.TasksGrid, "_MoveRow", _MoveTask) self.TasksGrid.SetRowLabelSize(0) self.TasksTable.ResetView(self.TasksGrid) self.TasksGrid.RefreshButtons() self.InstancesDefaultValue = {"Name": "", "Type": "", "Task": ""} self.InstancesTable = ResourceTable(self, [], GetInstancesTableColnames()) self.InstancesTable.SetColAlignements( [wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT]) self.InstancesTable.SetColSizes([200, 150, 150]) self.InstancesGrid.SetTable(self.InstancesTable) self.InstancesGrid.SetButtons({ "Add": self.AddInstanceButton, "Delete": self.DeleteInstanceButton, "Up": self.UpInstanceButton, "Down": self.DownInstanceButton }) def _AddInstance(new_row): self.InstancesTable.InsertRow(new_row, self.InstancesDefaultValue.copy()) self.RefreshModel() self.RefreshView() return new_row setattr(self.InstancesGrid, "_AddRow", _AddInstance) def _DeleteInstance(row): self.InstancesTable.RemoveRow(row) self.RefreshModel() self.RefreshView() setattr(self.InstancesGrid, "_DeleteRow", _DeleteInstance) def _MoveInstance(row, move): new_row = max( 0, min(row + move, self.InstancesTable.GetNumberRows() - 1)) if new_row != row: if self.InstancesTable.GetValueByName( row, "Task") != self.InstancesTable.GetValueByName( new_row, "Task"): return row self.InstancesTable.MoveRow(row, move) self.RefreshModel() self.RefreshView() return new_row setattr(self.InstancesGrid, "_MoveRow", _MoveInstance) def _RefreshInstanceButtons(): if self: rows = self.InstancesTable.GetNumberRows() row = self.InstancesGrid.GetGridCursorRow() self.DeleteInstanceButton.Enable(rows > 0) self.UpInstanceButton.Enable( row > 0 and self.InstancesTable.GetValueByName(row, "Task") == self.InstancesTable.GetValueByName(row - 1, "Task")) self.DownInstanceButton.Enable( 0 <= row < rows - 1 and self.InstancesTable.GetValueByName(row, "Task") == self.InstancesTable.GetValueByName(row + 1, "Task")) setattr(self.InstancesGrid, "RefreshButtons", _RefreshInstanceButtons) self.InstancesGrid.SetRowLabelSize(0) self.InstancesTable.ResetView(self.InstancesGrid) self.InstancesGrid.RefreshButtons() self.TasksGrid.SetFocus() def __del__(self): self.RefreshHighlightsTimer.Stop() def RefreshTypeList(self): self.TypeList = "" blocktypes = self.Controler.GetBlockResource() for blocktype in blocktypes: self.TypeList += ",%s" % blocktype def RefreshTaskList(self): self.TaskList = "" for row in xrange(self.TasksTable.GetNumberRows()): self.TaskList += ",%s" % self.TasksTable.GetValueByName( row, "Name") def RefreshVariableList(self): self.VariableList = "" for variable in self.Controler.GetEditedResourceVariables( self.TagName): self.VariableList += ",%s" % variable def RefreshModel(self): self.Controler.SetEditedResourceInfos(self.TagName, self.TasksTable.GetData(), self.InstancesTable.GetData()) self.RefreshBuffer() # Buffer the last model state def RefreshBuffer(self): self.Controler.BufferProject() self.ParentWindow.RefreshTitle() self.ParentWindow.RefreshFileMenu() self.ParentWindow.RefreshEditMenu() def GetBufferState(self): return self.Controler.GetBufferState() def Undo(self): self.Controler.LoadPrevious() self.ParentWindow.CloseTabsWithoutModel() def Redo(self): self.Controler.LoadNext() self.ParentWindow.CloseTabsWithoutModel() def HasNoModel(self): return self.Controler.GetEditedElement(self.TagName) is None def RefreshView(self, variablepanel=True): EditorPanel.RefreshView(self, variablepanel) tasks, instances = self.Controler.GetEditedResourceInfos(self.TagName) self.TasksTable.SetData(tasks) self.InstancesTable.SetData(instances) self.RefreshTypeList() self.RefreshTaskList() self.RefreshVariableList() self.TasksTable.ResetView(self.TasksGrid) self.InstancesTable.ResetView(self.InstancesGrid) self.TasksGrid.RefreshButtons() self.InstancesGrid.RefreshButtons() def OnTasksGridCellChange(self, event): row, col = event.GetRow(), event.GetCol() if self.TasksTable.GetColLabelValue(col) == "Name": tasklist = [ name for name in self.TaskList.split(",") if name != "" ] for i in xrange(self.TasksTable.GetNumberRows()): task = self.TasksTable.GetValueByName(i, "Name") if task in tasklist: tasklist.remove(task) if len(tasklist) > 0: old_name = tasklist[0] new_name = self.TasksTable.GetValue(row, col) for i in xrange(self.InstancesTable.GetNumberRows()): if self.InstancesTable.GetValueByName(i, "Task") == old_name: self.InstancesTable.SetValueByName(i, "Task", new_name) self.RefreshModel() colname = self.TasksTable.GetColLabelValue(col, False) if colname in ["Triggering", "Name", "Single", "Interval"]: wx.CallAfter(self.RefreshView, False) event.Skip() def OnInstancesGridCellChange(self, event): self.RefreshModel() self.ParentWindow.RefreshPouInstanceVariablesPanel() self.InstancesGrid.RefreshButtons() event.Skip() #------------------------------------------------------------------------------- # Highlights showing functions #------------------------------------------------------------------------------- def OnRefreshHighlightsTimer(self, event): self.RefreshView() event.Skip() def AddHighlight(self, infos, start, end, highlight_type): EditorPanel.AddHighlight(self, infos, start, end, highlight_type) if infos[0] == "task": self.TasksTable.AddHighlight(infos[1:], highlight_type) elif infos[0] == "instance": self.InstancesTable.AddHighlight(infos[1:], highlight_type) self.RefreshHighlightsTimer.Start(int(REFRESH_HIGHLIGHT_PERIOD * 1000), oneShot=True) def ClearHighlights(self, highlight_type=None): EditorPanel.ClearHighlights(self, highlight_type) self.TasksTable.ClearHighlights(highlight_type) self.InstancesTable.ClearHighlights(highlight_type) self.TasksTable.ResetView(self.TasksGrid) self.InstancesTable.ResetView(self.InstancesGrid)
class VariablesEditor(wx.Panel): def __init__(self, parent, window, controler): wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL) main_sizer = wx.FlexGridSizer(cols=2, hgap=0, rows=1, vgap=4) main_sizer.AddGrowableCol(1) main_sizer.AddGrowableRow(0) controls_sizer = wx.BoxSizer(wx.VERTICAL) main_sizer.AddSizer(controls_sizer, border=5, flag=wx.ALL) for name, bitmap, help in [ ("AddVariableButton", "add_element", _("Add variable")), ("DeleteVariableButton", "remove_element", _("Remove variable")), ("UpVariableButton", "up", _("Move variable up")), ("DownVariableButton", "down", _("Move variable down")) ]: button = wx.lib.buttons.GenBitmapButton(self, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) controls_sizer.AddWindow(button, border=5, flag=wx.BOTTOM) self.VariablesGrid = CustomGrid(self, style=wx.VSCROLL) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick) self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown) main_sizer.AddWindow(self.VariablesGrid, flag=wx.GROW) self.SetSizer(main_sizer) self.ParentWindow = window self.Controler = controler self.VariablesDefaultValue = { "Name": "LocalVar0", "Type": DefaultType, "Initial": "", "Description": "", "OnChange": "", "Options": "" } self.Table = VariablesTable(self, [], self.GetVariableTableColnames()) self.ColAlignements = [wx.ALIGN_RIGHT] + \ [wx.ALIGN_LEFT]*(len(self.VariablesDefaultValue)) self.ColSizes = [20, 150 ] + [130] * (len(self.VariablesDefaultValue) - 1) self.VariablesGrid.SetTable(self.Table) self.VariablesGrid.SetButtons({ "Add": self.AddVariableButton, "Delete": self.DeleteVariableButton, "Up": self.UpVariableButton, "Down": self.DownVariableButton }) def _AddVariable(new_row): if new_row > 0: row_content = self.Table.data[new_row - 1].copy() old_name = row_content['Name'] row_content['Name'] =\ self.Controler.GenerateNewName(old_name, old_name+'%d') else: row_content = self.VariablesDefaultValue.copy() self.Table.InsertRow(new_row, row_content) self.RefreshModel() self.RefreshView() return new_row setattr(self.VariablesGrid, "_AddRow", _AddVariable) def _DeleteVariable(row): self.Table.RemoveRow(row) self.RefreshModel() self.RefreshView() setattr(self.VariablesGrid, "_DeleteRow", _DeleteVariable) def _MoveVariable(row, move): new_row = self.Table.MoveRow(row, move) if new_row != row: self.RefreshModel() self.RefreshView() return new_row setattr(self.VariablesGrid, "_MoveRow", _MoveVariable) self.VariablesGrid.SetRowLabelSize(0) for col in range(self.Table.GetNumberCols()): attr = wx.grid.GridCellAttr() attr.SetAlignment(self.ColAlignements[col], wx.ALIGN_CENTRE) self.VariablesGrid.SetColAttr(col, attr) self.VariablesGrid.SetColSize(col, self.ColSizes[col]) self.Table.ResetView(self.VariablesGrid) def GetVariableTableColnames(self): _ = NoTranslate return [ "#", _("Name"), _("Type"), _("Initial"), _("Description"), _("OnChange"), _("Options") ] def RefreshModel(self): self.Controler.SetVariables(self.Table.GetData()) self.RefreshBuffer() # Buffer the last model state def RefreshBuffer(self): self.Controler.BufferCodeFile() self.ParentWindow.RefreshTitle() self.ParentWindow.RefreshFileMenu() self.ParentWindow.RefreshEditMenu() self.ParentWindow.RefreshPageTitles() def RefreshView(self): self.Table.SetData(self.Controler.GetVariables()) self.Table.ResetView(self.VariablesGrid) self.VariablesGrid.RefreshButtons() def DoGetBestSize(self): return self.ParentWindow.GetPanelBestSize() def ShowErrorMessage(self, message): dialog = wx.MessageDialog(self, message, _("Error"), wx.OK | wx.ICON_ERROR) dialog.ShowModal() dialog.Destroy() def OnVariablesGridCellChange(self, event): row, col = event.GetRow(), event.GetCol() colname = self.Table.GetColLabelValue(col, False) value = self.Table.GetValue(row, col) message = None if colname == "Name" and value != "": 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 var_row, var in enumerate(self.Table.data) if var_row != row ]: message = _( "A variable with \"%s\" as name already exists!") % value else: self.RefreshModel() wx.CallAfter(self.RefreshView) else: self.RefreshModel() wx.CallAfter(self.RefreshView) if message is not None: event.Veto() wx.CallAfter(self.ShowErrorMessage, message) else: event.Skip() def OnVariablesGridEditorShown(self, event): row, col = event.GetRow(), event.GetCol() if self.Table.GetColLabelValue(col, False) == "Type": type_menu = wx.Menu(title='') base_menu = wx.Menu(title='') for base_type in self.Controler.GetBaseTypes(): new_id = wx.NewId() base_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=base_type) self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(base_type), id=new_id) type_menu.AppendMenu(wx.NewId(), "Base Types", base_menu) datatype_menu = wx.Menu(title='') for datatype in self.Controler.GetDataTypes(): new_id = wx.NewId() datatype_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=datatype) self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(datatype), id=new_id) type_menu.AppendMenu(wx.NewId(), "User Data Types", datatype_menu) rect = self.VariablesGrid.BlockToDeviceRect((row, col), (row, col)) self.VariablesGrid.PopupMenuXY( type_menu, rect.x + rect.width, rect.y + self.VariablesGrid.GetColLabelSize()) type_menu.Destroy() event.Veto() else: event.Skip() def GetVariableTypeFunction(self, base_type): def VariableTypeFunction(event): row = self.VariablesGrid.GetGridCursorRow() self.Table.SetValueByName(row, "Type", base_type) self.Table.ResetView(self.VariablesGrid) self.RefreshModel() self.RefreshView() event.Skip() return VariableTypeFunction def OnVariablesGridCellLeftClick(self, event): if event.GetCol() == 0: row = event.GetRow() data_type = self.Table.GetValueByName(row, "Type") var_name = self.Table.GetValueByName(row, "Name") data = wx.TextDataObject( str((var_name, "Global", data_type, self.Controler.GetCurrentLocation()))) dragSource = wx.DropSource(self.VariablesGrid) dragSource.SetData(data) dragSource.DoDragDrop() return event.Skip()
class MBRequestDataPanel(wx.Panel): def __init__(self, parent, window, controler): wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL) """ main sizer """ requestPanelSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=1, vgap=4) requestPanelSizer.AddGrowableCol(1) requestPanelSizer.AddGrowableRow(0) """ sizer with control buttons """ controls_sizer = wx.BoxSizer(wx.VERTICAL) requestPanelSizer.AddSizer(controls_sizer, border=5, flag=wx.ALL) for name, bitmap, help in [ ("AddVariableButton", "add_element", _("Add variable")), ("DeleteVariableButton", "remove_element", _("Remove variable")), ("UpVariableButton", "up", _("Move variable up")), ("DownVariableButton", "down", _("Move variable down")) ]: button = wx.lib.buttons.GenBitmapButton(self, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) controls_sizer.AddWindow(button, border=5, flag=wx.BOTTOM) """ variable grid """ self.VariablesGrid = CustomGrid(self, style=wx.VSCROLL) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick) self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown) requestPanelSizer.AddWindow(self.VariablesGrid, flag=wx.GROW) """ add header to grid """ mainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=4) mainSizer.AddGrowableRow(1) mainSizer.AddGrowableCol(0) headerSizer = wx.BoxSizer(wx.HORIZONTAL) label = wx.StaticText(self, label=" Request data") font = wx.Font(14, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD) label.SetFont(font) headerSizer.Add(label) self.masterSelectCmbx = ComboBoxWithLabel( self, "", ("Master0", "Master1", "Master2")) wx.EVT_COMBOBOX(self, self.masterSelectCmbx.cmbbox.GetId(), self.OnChangeMasterOption) headerSizer.Add(self.masterSelectCmbx, flag=wx.ALIGN_CENTER_VERTICAL) mainSizer.AddSizer(headerSizer, flag=wx.GROW) mainSizer.AddSizer(requestPanelSizer, flag=wx.GROW) self.SetSizer(mainSizer) self.ParentWindow = window self.Controler = controler """ request defaule data value """ self.VariablesDefaultValue = { "Name": "", "Address": "1", "Len": "1", "Data type": "WORD", "Modbus type": HOLDING_REGISTER_READ, "Description": DESCRIPTION, "Device ID": "1", "Transfer method": "Periodic", "Period (ms)": "100", "Timeout (ms)": "100", } self.Table = MBRequestDataTable(self) self.ColAlignements = [wx.ALIGN_RIGHT] + \ [wx.ALIGN_LEFT]*(len(self.VariablesDefaultValue)) self.ColSizes = [20, 150 ] + [100] * (len(self.VariablesDefaultValue) - 1) self.VariablesGrid.SetTable(self.Table) self.VariablesGrid.SetButtons({ "Add": self.AddVariableButton, "Delete": self.DeleteVariableButton, "Up": self.UpVariableButton, "Down": self.DownVariableButton }) """ handlers for adding variable """ def _AddVariable(new_row): if new_row > 0: row_content = self.Table.data[new_row - 1].copy() newAddr = int(row_content["Address"]) newLen = int(row_content["Len"]) if row_content["Data type"] in ("REAL", "INT"): newLen = newLen * 2 newAddr = newAddr + newLen row_content["Address"] = str(newAddr) result = VARIABLE_NAME_SUFFIX_MODEL.search(row_content["Name"]) if result is not None: name = row_content["Name"][:result.start(1)] suffix = result.group(1) if suffix != "": start_idx = int(suffix) else: start_idx = 0 else: name = row_content["Name"] start_idx = 0 row_content["Name"] = self.Controler.GenerateNewName( name + "%d", start_idx) else: row_content = self.VariablesDefaultValue.copy() self.Table.InsertRow(new_row, row_content) self.RefreshModel() self.RefreshView() return new_row setattr(self.VariablesGrid, "_AddRow", _AddVariable) """ handlers for deleting variable """ def _DeleteVariable(row): self.Table.RemoveRow(row) self.RefreshModel() self.RefreshView() setattr(self.VariablesGrid, "_DeleteRow", _DeleteVariable) """ handlers for moving varible """ def _MoveVariable(row, move): new_row = self.Table.MoveRow(row, move) if new_row != row: self.RefreshModel() self.RefreshView() return new_row setattr(self.VariablesGrid, "_MoveRow", _MoveVariable) self.VariablesGrid.SetRowLabelSize(0) for col in range(self.Table.GetNumberCols()): attr = wx.grid.GridCellAttr() attr.SetAlignment(self.ColAlignements[col], wx.ALIGN_CENTRE) self.VariablesGrid.SetColAttr(col, attr) self.VariablesGrid.SetColSize(col, self.ColSizes[col]) self.Table.ResetView(self.VariablesGrid) def OnChangeMasterOption(self, event): masterOptionOnFrame = self.masterSelectCmbx.cmbbox.GetStringSelection() masterOptionInXml = [ i for i in self.Controler.GetVariables() if i["Name"] == MASTER_OPTION ] if len(masterOptionInXml) > 0: if masterOptionOnFrame != masterOptionInXml[0]: self.RefreshModel() else: self.RefreshModel() event.Skip() def GetMasterOption(self): return { "Name": "Master option", "Address": "", "Len": "", "Device ID": "", "Data type": u"WORD", "Transfer method": "", "Period (ms)": "100", "Timeout (ms)": "100", "Description": MASTER_OPTION, "Modbus type": self.masterSelectCmbx.cmbbox.GetStringSelection() } def RefreshModel(self): controllerVariables = self.Controler.GetVariables() controllerVariables = [ i for i in controllerVariables if i["Description"] not in (DESCRIPTION, MASTER_OPTION) ] controllerVariables += self.Table.GetData() controllerVariables.append(self.GetMasterOption()) self.Controler.SetVariables(controllerVariables) self.RefreshBuffer() def RefreshView(self): varForTable = self.Controler.GetVariables() varForTable = [ i for i in varForTable if i["Description"] == DESCRIPTION ] self.Table.SetData(varForTable) self.Table.ResetView(self.VariablesGrid) varForTable = self.Controler.GetVariables() varForTable = [i for i in varForTable if i["Name"] == MASTER_OPTION] if len(varForTable) > 0: self.masterSelectCmbx.cmbbox.SetStringSelection( varForTable[0]["Modbus type"]) self.VariablesGrid.RefreshButtons() def OnVariablesGridCellChange(self, event): row, col = event.GetRow(), event.GetCol() colname = self.Table.GetColLabelValue(col, False) value = self.Table.GetValue(row, col) message = None if colname == "Modbus type": self.VariablesGrid.SetCellValue(row, col + 1, "WORD") if colname == "Len": modbusType = self.Table.GetValue(row, 4) # Modubs type value if modbusType in (HOLDING_REGISTER_READ, INPUT_REGISTER_READ): if int(value) > MAX_SIMULTANEOUS_REGISTERS_READ: value = MAX_SIMULTANEOUS_REGISTERS_READ self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_REGISTERS_READ)) if modbusType == HOLDING_REGISTER_WRITE: if int(value) > MAX_SIMULTANEOUS_REGISTERS_WRITE: self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_REGISTERS_WRITE)) if modbusType in (COIL_READ, DISCRETE_INPUT_READ): if int(value) > MAX_SIMULTANEOUS_COIL_DISC_READ: self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_COIL_DISC_READ)) if modbusType == COIL_WRITE: if int(value) > MAX_SIMULTANEOUS_COIL_WRITE: self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_COIL_WRITE)) if colname == "Name" and value != "": 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 var_row, var in enumerate(self.Table.data) if var_row != row ]: message = _( "A variable with \"%s\" as name already exists!") % value else: self.RefreshModel() wx.CallAfter(self.RefreshView) else: self.RefreshModel() wx.CallAfter(self.RefreshView) if message is not None: dialog = wx.MessageDialog(self, message, _("Error"), wx.OK | wx.ICON_ERROR) dialog.ShowModal() dialog.Destroy() event.Veto() else: event.Skip() def DoGetBestSize(self): return self.ParentWindow.GetPanelBestSize() def OnVariablesGridEditorShown(self, event): event.Skip() def GetVariableTypeFunction(self, base_type): def VariableTypeFunction(event): row = self.VariablesGrid.GetGridCursorRow() self.Table.SetValueByName(row, "Data type", base_type) self.Table.ResetView(self.VariablesGrid) self.RefreshModel() self.RefreshView() event.Skip() return VariableTypeFunction def OnVariablesGridCellLeftClick(self, event): if event.GetCol() == 0: row = event.GetRow() data_type = self.Table.GetValueByName(row, "Data type") var_name = self.Table.GetValueByName(row, "Name") data = wx.TextDataObject( str((var_name, "Global", data_type, self.Controler.GetCurrentLocation()))) dragSource = wx.DropSource(self.VariablesGrid) dragSource.SetData(data) dragSource.DoDragDrop() return event.Skip() def RefreshBuffer(self): self.Controler.BufferCodeFile() self.ParentWindow.RefreshTitle() self.ParentWindow.RefreshFileMenu() self.ParentWindow.RefreshEditMenu() self.ParentWindow.RefreshPageTitles()
class RequestTablePanel(wx.Panel): def __init__(self, parent, window, controler): wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL) """ main sizer """ requestPanelSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=1, vgap=4) requestPanelSizer.AddGrowableCol(1) requestPanelSizer.AddGrowableRow(0) """ sizer with control buttons """ controls_sizer = wx.BoxSizer(wx.VERTICAL) requestPanelSizer.AddSizer(controls_sizer, border=5, flag=wx.ALL) for name, bitmap, help in [ ("AddVariableButton", "add_element", _("Add variable")), ("DeleteVariableButton", "remove_element", _("Remove variable")), ("UpVariableButton", "up", _("Move variable up")), ("DownVariableButton", "down", _("Move variable down")) ]: button = wx.lib.buttons.GenBitmapButton(self, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) controls_sizer.AddWindow(button, border=5, flag=wx.BOTTOM) """ variable grid """ self.VariablesGrid = CustomGrid(self, style=wx.VSCROLL) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick) self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown) requestPanelSizer.AddWindow(self.VariablesGrid, flag=wx.GROW) """ add header to grid """ mainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=4) mainSizer.AddGrowableRow(1) mainSizer.AddGrowableCol(0) headerSizer = wx.BoxSizer(wx.HORIZONTAL) self.ipAddressDefault = { "Name": "ipAddres", "Address": "10.10.10.10", "Device ID": "", "Data type": "", "Transfer method": "", "Len": "1", "Modbus type": "", "Device ID": "1", "Period (ms)": "100", "Description": DESCRIPTION_IP, } self.portDefault = { "Name": "port", "Address": "502", "Device ID": "", "Data type": "", "Transfer method": "", "Len": "1", "Modbus type": "", "Device ID": "1", "Period (ms)": "100", "Description": DESCRIPTION_PORT } # виджеты для текста IP (label) labelIP = wx.StaticText(self, label=" Ip address: ") font = wx.Font(13, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD) labelIP.SetFont(font) headerSizer.Add(labelIP) """ поле для IP """ self.ip_address = IpAddrCtrl(self, size=(120, 22)) wx.EVT_TEXT(self, self.ip_address.GetId(), self.OnChangeIp) """ wx.EVT_COMBOBOX(self, self.masterSelectCmbx.cmbbox.GetId(), self.OnChangeMasterOption) """ headerSizer.Add(self.ip_address, flag=wx.ALIGN_CENTER_VERTICAL) """ виджеты для текста порта (label) """ labelPort = wx.StaticText(self, label=" Port: ") font = wx.Font(13, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD) labelPort.SetFont(font) headerSizer.Add(labelPort) """ поле для порта """ self.port = IntCtrl(self, min=0, max=65535, limited=True, size=(70, 21)) wx.EVT_TEXT(self, self.port.GetId(), self.OnChangePort) wx.EVT_TEXT(self, self.port.GetId(), self.OnChangePort) headerSizer.Add(self.port, flag=wx.ALIGN_CENTER_VERTICAL) mainSizer.AddSizer(headerSizer, flag=wx.GROW) mainSizer.AddSizer(requestPanelSizer, flag=wx.GROW) self.SetSizer(mainSizer) self.ParentWindow = window self.Controler = controler self.VariablesDefaultValue = { "Name": "", "Address": "1", "Len": "1", "Data type": "WORD", "Modbus type": HOLDING_REGISTER_READ, "Description": DESCRIPTION, "Device ID": "1", "Transfer method": "Periodic", "Period (ms)": "100" } self.Table = MBRequestDataTable(self) self.ColAlignements = [wx.ALIGN_RIGHT] + \ [wx.ALIGN_LEFT]*(len(self.VariablesDefaultValue)) self.ColSizes = [20, 150 ] + [100] * (len(self.VariablesDefaultValue) - 1) self.VariablesGrid.SetTable(self.Table) self.VariablesGrid.SetButtons({ "Add": self.AddVariableButton, "Delete": self.DeleteVariableButton, "Up": self.UpVariableButton, "Down": self.DownVariableButton }) """ handlers for adding variable """ def _AddVariable(new_row): if new_row > 0: row_content = self.Table.data[new_row - 1].copy() newAddr = int(row_content["Address"]) newLen = int(row_content["Len"]) if row_content["Data type"] in ("REAL", "INT"): newLen = newLen * 2 newAddr = newAddr + newLen row_content["Address"] = str(newAddr) result = VARIABLE_NAME_SUFFIX_MODEL.search(row_content["Name"]) if result is not None: name = row_content["Name"][:result.start(1)] suffix = result.group(1) if suffix != "": start_idx = int(suffix) else: start_idx = 0 else: name = row_content["Name"] start_idx = 0 row_content["Name"] = self.Controler.GenerateNewName( name + "%d", start_idx) else: row_content = self.VariablesDefaultValue.copy() self.Table.InsertRow(new_row, row_content) self.RefreshModel() self.RefreshView() return new_row setattr(self.VariablesGrid, "_AddRow", _AddVariable) """ handlers for deleting variable """ def _DeleteVariable(row): self.Table.RemoveRow(row) self.RefreshModel() self.RefreshView() setattr(self.VariablesGrid, "_DeleteRow", _DeleteVariable) """ handlers for moving varible """ def _MoveVariable(row, move): new_row = self.Table.MoveRow(row, move) if new_row != row: self.RefreshModel() self.RefreshView() return new_row setattr(self.VariablesGrid, "_MoveRow", _MoveVariable) self.VariablesGrid.SetRowLabelSize(0) for col in range(self.Table.GetNumberCols()): attr = wx.grid.GridCellAttr() attr.SetAlignment(self.ColAlignements[col], wx.ALIGN_CENTRE) self.VariablesGrid.SetColAttr(col, attr) self.VariablesGrid.SetColSize(col, self.ColSizes[col]) self.Table.ResetView(self.VariablesGrid) self.Bind(wx.EVT_SHOW, self.OnShow) def OnChangeMasterOption(self, event): masterOptionOnFrame = self.masterSelectCmbx.cmbbox.GetStringSelection() masterOptionInXml = [ i for i in self.Controler.GetVariables() if i["Name"] == MASTER_OPTION ] if len(masterOptionInXml) > 0: if masterOptionOnFrame != masterOptionInXml[0]: self.RefreshModel() else: self.RefreshModel() event.Skip() def GetMasterOption(self): return { "Name": "Master option", "Address": "", "Len": "", "Device ID": "", "Data type": u"WORD", "Transfer method": "", "Period (ms)": "", "Description": MASTER_OPTION, "Modbus type": "0" } def RefreshModel(self): ipAddress = self.ipAddressDefault.copy() ipAddress["Address"] = self.ip_address.GetValue() controllerVariables = self.Controler.GetVariables() controllerVariables = [ i for i in controllerVariables if i["Description"] not in (DESCRIPTION, MASTER_OPTION) ] controllerVariables += self.Table.GetData() controllerVariables.append(self.GetMasterOption()) controllerVariables.append(ipAddress) controllerVariables.append(self.portDefault) self.Controler.SetVariables(controllerVariables) self.RefreshBuffer() def RefreshView(self): varForTable = self.Controler.GetVariables() varForTable = [ i for i in varForTable if i["Description"] == DESCRIPTION ] self.Table.SetData(varForTable) self.Table.ResetView(self.VariablesGrid) varForTable = self.Controler.GetVariables() varForTable = [i for i in varForTable if i["Name"] == MASTER_OPTION] #if len(varForTable) > 0: #self.masterSelectCmbx.cmbbox.SetStringSelection(varForTable[0]["Modbus type"]) self.VariablesGrid.RefreshButtons() varForTable = self.Controler.GetVariables() varForTable = [ i for i in varForTable if i["Description"] == DESCRIPTION_IP ] if len(varForTable) == 0: varForTable = [] varForTable.append(self.ipAddressDefault) self.ip_address.SetValue(varForTable[0]["Address"]) varForTable = self.Controler.GetVariables() vars = [i for i in varForTable if i["Description"] == DESCRIPTION_PORT] if len(vars) == 0: vars = [] vars.append(self.portDefault) # if self.port.GetLineText != vars[0]["Address"]: # print vars self.port.SetValue(int(vars[0]["Address"])) def OnShow(self, event): pass # print 'kek' # используется для событий, когда изменяются поля ip-адреса и порта def OnChangeIp(self, event): ipAddress = self.ipAddressDefault.copy() ipAddress["Address"] = self.ip_address.GetValue().replace(' ', '') # port = self.portDefault.copy() # port["Address"] = self.port.GetLineText(0) controllerVariables = self.Controler.GetVariables() # valuesOnFrame = self.GetData() valueIPOnWidget = self.ip_address.GetValue() controllerVariablesIP = [ i for i in controllerVariables if i["Description"] == (DESCRIPTION_IP) ] # valuePortOnWidget = self.port.GetLineText() controllerVariablesIP = [ i for i in controllerVariables if i["Description"] == (DESCRIPTION_PORT) ] if not (ipAddress in controllerVariables): controllerVariables = [ i for i in controllerVariables if i["Description"] != DESCRIPTION_IP ] controllerVariables.append(ipAddress) # controllerVariables.append(port) self.Controler.SetVariables(controllerVariables) self.RefreshBuffer() event.Skip() def OnChangePort(self, event): portAddress = self.portDefault.copy() portAddress["Address"] = self.port.GetLineText(0) controllerVariables = self.Controler.GetVariables() if not (portAddress in controllerVariables): controllerVariables = [ i for i in controllerVariables if i["Description"] != DESCRIPTION_PORT ] controllerVariables.append(portAddress) self.Controler.SetVariables(controllerVariables) self.RefreshBuffer() event.Skip() # def GetData(self): # """ # Считывает с GUI настроенные пользователем параметры и возвращает их в виде списка # словарей # """ def OnVariablesGridCellChange(self, event): row, col = event.GetRow(), event.GetCol() colname = self.Table.GetColLabelValue(col, False) value = self.Table.GetValue(row, col) message = None if colname == "Modbus type": self.VariablesGrid.SetCellValue(row, col + 1, "WORD") if colname == "Len": modbusType = self.Table.GetValue(row, 4) # Modubs type value if modbusType in (HOLDING_REGISTER_READ, INPUT_REGISTER_READ): if int(value) > MAX_SIMULTANEOUS_REGISTERS_READ: value = MAX_SIMULTANEOUS_REGISTERS_READ self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_REGISTERS_READ)) if modbusType == HOLDING_REGISTER_WRITE: if int(value) > MAX_SIMULTANEOUS_REGISTERS_WRITE: self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_REGISTERS_WRITE)) if modbusType in (COIL_READ, DISCRETE_INPUT_READ): if int(value) > MAX_SIMULTANEOUS_COIL_DISC_READ: self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_COIL_DISC_READ)) if modbusType == COIL_WRITE: if int(value) > MAX_SIMULTANEOUS_COIL_WRITE: self.VariablesGrid.SetCellValue( row, col, str(MAX_SIMULTANEOUS_COIL_WRITE)) if colname == "Name" and value != "": 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 var_row, var in enumerate(self.Table.data) if var_row != row ]: message = _( "A variable with \"%s\" as name already exists!") % value else: self.RefreshModel() wx.CallAfter(self.RefreshView) else: self.RefreshModel() wx.CallAfter(self.RefreshView) if message is not None: dialog = wx.MessageDialog(self, message, _("Error"), wx.OK | wx.ICON_ERROR) dialog.ShowModal() dialog.Destroy() event.Veto() else: event.Skip() def DoGetBestSize(self): return self.ParentWindow.GetPanelBestSize() def OnVariablesGridEditorShown(self, event): event.Skip() def GetVariableTypeFunction(self, base_type): def VariableTypeFunction(event): row = self.VariablesGrid.GetGridCursorRow() self.Table.SetValueByName(row, "Data type", base_type) self.Table.ResetView(self.VariablesGrid) self.RefreshModel() self.RefreshView() event.Skip() return VariableTypeFunction def OnVariablesGridCellLeftClick(self, event): if event.GetCol() == 0: row = event.GetRow() data_type = self.Table.GetValueByName(row, "Data type") var_name = self.Table.GetValueByName(row, "Name") data = wx.TextDataObject( str((var_name, "Global", data_type, self.Controler.GetCurrentLocation()))) dragSource = wx.DropSource(self.VariablesGrid) dragSource.SetData(data) dragSource.DoDragDrop() return event.Skip() def RefreshBuffer(self): self.Controler.BufferCodeFile() self.ParentWindow.RefreshTitle() self.ParentWindow.RefreshFileMenu() self.ParentWindow.RefreshEditMenu() self.ParentWindow.RefreshPageTitles()
class DebugVariablePanel(wx.Panel, DebugViewer): def __init__(self, parent, producer): wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL) DebugViewer.__init__(self, producer, True) main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0) main_sizer.AddGrowableCol(0) main_sizer.AddGrowableRow(1) button_sizer = wx.BoxSizer(wx.HORIZONTAL) main_sizer.AddSizer(button_sizer, border=5, flag=wx.ALIGN_RIGHT | wx.ALL) for name, bitmap, help in [ ("DeleteButton", "remove_element", _("Remove debug variable")), ("UpButton", "up", _("Move debug variable up")), ("DownButton", "down", _("Move debug variable down")) ]: button = wx.lib.buttons.GenBitmapButton(self, bitmap=GetBitmap(bitmap), size=wx.Size(28, 28), style=wx.NO_BORDER) button.SetToolTipString(help) setattr(self, name, button) button_sizer.AddWindow(button, border=5, flag=wx.LEFT) self.VariablesGrid = CustomGrid(self, size=wx.Size(0, 150), style=wx.VSCROLL) self.VariablesGrid.SetDropTarget(DebugVariableDropTarget(self)) self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_RIGHT_CLICK, self.OnVariablesGridCellRightClick) main_sizer.AddWindow(self.VariablesGrid, flag=wx.GROW) self.SetSizer(main_sizer) self.HasNewData = False self.Table = DebugVariableTable(self, [], GetDebugVariablesTableColnames()) self.VariablesGrid.SetTable(self.Table) self.VariablesGrid.SetButtons({ "Delete": self.DeleteButton, "Up": self.UpButton, "Down": self.DownButton }) def _AddVariable(new_row): return self.VariablesGrid.GetGridCursorRow() setattr(self.VariablesGrid, "_AddRow", _AddVariable) def _DeleteVariable(row): item = self.Table.GetItem(row) self.RemoveDataConsumer(item) self.Table.RemoveItem(row) self.RefreshGrid() setattr(self.VariablesGrid, "_DeleteRow", _DeleteVariable) def _MoveVariable(row, move): new_row = max(0, min(row + move, self.Table.GetNumberRows() - 1)) if new_row != row: self.Table.MoveItem(row, new_row) self.RefreshGrid() return new_row setattr(self.VariablesGrid, "_MoveRow", _MoveVariable) self.VariablesGrid.SetRowLabelSize(0) for col in range(self.Table.GetNumberCols()): attr = wx.grid.GridCellAttr() attr.SetAlignment(wx.ALIGN_RIGHT, wx.ALIGN_CENTER) self.VariablesGrid.SetColAttr(col, attr) self.VariablesGrid.SetColSize(col, 100) self.Table.ResetView(self.VariablesGrid) self.VariablesGrid.RefreshButtons() def RefreshNewData(self): if self.HasNewData: self.HasNewData = False self.RefreshGrid() DebugViewer.RefreshNewData(self) def RefreshGrid(self): self.Freeze() self.Table.ResetView(self.VariablesGrid) self.VariablesGrid.RefreshButtons() self.Thaw() def UnregisterObsoleteData(self): items = [(idx, item) for idx, item in enumerate(self.Table.GetData())] items.reverse() for idx, item in items: iec_path = item.GetVariable().upper() if self.GetDataType(iec_path) is None: self.RemoveDataConsumer(item) self.Table.RemoveItem(idx) else: self.AddDataConsumer(iec_path, item) self.Freeze() self.Table.ResetView(self.VariablesGrid) self.VariablesGrid.RefreshButtons() self.Thaw() def ResetGrid(self): self.DeleteDataConsumers() self.Table.Empty() self.Freeze() self.Table.ResetView(self.VariablesGrid) self.VariablesGrid.RefreshButtons() self.Thaw() def GetForceVariableMenuFunction(self, iec_path, item): iec_type = self.GetDataType(iec_path) def ForceVariableFunction(event): if iec_type is not None: dialog = ForceVariableDialog(self, iec_type, str(item.GetValue())) if dialog.ShowModal() == wx.ID_OK: self.ForceDataValue(iec_path, dialog.GetValue()) return ForceVariableFunction def GetReleaseVariableMenuFunction(self, iec_path): def ReleaseVariableFunction(event): self.ReleaseDataValue(iec_path) return ReleaseVariableFunction def OnVariablesGridCellRightClick(self, event): row, col = event.GetRow(), event.GetCol() if self.Table.GetColLabelValue(col, False) == "Value": iec_path = self.Table.GetValueByName(row, "Variable").upper() menu = wx.Menu(title='') new_id = wx.NewId() AppendMenu(menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=_("Force value")) self.Bind(wx.EVT_MENU, self.GetForceVariableMenuFunction( iec_path.upper(), self.Table.GetItem(row)), id=new_id) new_id = wx.NewId() AppendMenu(menu, help='', id=new_id, kind=wx.ITEM_NORMAL, text=_("Release value")) self.Bind(wx.EVT_MENU, self.GetReleaseVariableMenuFunction(iec_path.upper()), id=new_id) if self.Table.IsForced(row): menu.Enable(new_id, True) else: menu.Enable(new_id, False) self.PopupMenu(menu) menu.Destroy() event.Skip() def InsertValue(self, iec_path, idx=None, force=False): if idx is None: idx = self.Table.GetNumberRows() for item in self.Table.GetData(): if iec_path == item.GetVariable(): return item = VariableTableItem(self, iec_path, "") result = self.AddDataConsumer(iec_path.upper(), item) if result is not None or force: self.Table.InsertItem(idx, item) self.RefreshGrid() def GetDebugVariables(self): return [item.GetVariable() for item in self.Table.GetData()]