1 #!/usr/bin/env python

     2 # -*- coding: utf-8 -*-

     3 

     4 #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor

     5 #based on the plcopen standard. 

     6 #

     7 #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8 #

     9 #See COPYING file for copyrights details.

    10 #

    11 #This library is free software; you can redistribute it and/or

    12 #modify it under the terms of the GNU General Public

    13 #License as published by the Free Software Foundation; either

    14 #version 2.1 of the License, or (at your option) any later version.

    15 #

    16 #This library is distributed in the hope that it will be useful,

    17 #but WITHOUT ANY WARRANTY; without even the implied warranty of

    18 #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    19 #General Public License for more details.

    20 #

    21 #You should have received a copy of the GNU General Public

    22 #License along with this library; if not, write to the Free Software

    23 #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    24 

    25 import wx

    26 import wx.grid

    27 

    28 from graphics import *

    29 

    30 #-------------------------------------------------------------------------------

    31 #                          Create New Block Dialog

    32 #-------------------------------------------------------------------------------

    33 

    34 [ID_BLOCKPROPERTIESDIALOG, ID_BLOCKPROPERTIESDIALOGNAME, 

    35  ID_BLOCKPROPERTIESDIALOGTYPETREE, ID_BLOCKPROPERTIESDIALOGTYPEDESC, 

    36  ID_BLOCKPROPERTIESDIALOGINPUTS, ID_BLOCKPROPERTIESDIALOGPREVIEW, 

    37  ID_BLOCKPROPERTIESDIALOGEXECUTIONORDER, ID_BLOCKPROPERTIESDIALOGEXECUTIONCONTROL, 

    38  ID_BLOCKPROPERTIESDIALOGSTATICTEXT1, ID_BLOCKPROPERTIESDIALOGSTATICTEXT2, 

    39  ID_BLOCKPROPERTIESDIALOGSTATICTEXT3, ID_BLOCKPROPERTIESDIALOGSTATICTEXT4, 

    40  ID_BLOCKPROPERTIESDIALOGSTATICTEXT5, ID_BLOCKPROPERTIESDIALOGSTATICTEXT6, 

    41 ] = [wx.NewId() for _init_ctrls in range(14)]

    42 

    43 [CATEGORY, BLOCK] = range(2)

    44 

    45 class BlockPropertiesDialog(wx.Dialog):

    46     

    47     if wx.VERSION < (2, 6, 0):

    48         def Bind(self, event, function, id = None):

    49             if id is not None:

    50                 event(self, id, function)

    51             else:

    52                 event(self, function)

    53     

    54     def _init_coll_flexGridSizer1_Items(self, parent):

    55         parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

    56         parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

    57 

    58     def _init_coll_flexGridSizer1_Growables(self, parent):

    59         parent.AddGrowableCol(0)

    60         parent.AddGrowableRow(0)

    61 

    62     def _init_coll_MainSizer_Items(self, parent):

    63         parent.AddSizer(self.LeftBoxSizer, 1, border=5, flag=wx.GROW|wx.RIGHT)

    64         parent.AddSizer(self.RightGridSizer, 1, border=5, flag=wx.GROW|wx.LEFT)

    65 

    66     def _init_coll_LeftBoxSizer_Items(self, parent):

    67         parent.AddWindow(self.TypeTree, 3, border=5, flag=wx.GROW|wx.BOTTOM)

    68         parent.AddWindow(self.TypeDesc, 1, border=0, flag=wx.GROW)

    69 

    70     def _init_coll_RightGridSizer_Items(self, parent):

    71         parent.AddSizer(self.RightUpGridSizer, 0, border=0, flag=wx.GROW)

    72         parent.AddWindow(self.staticText6, 0, border=0, flag=wx.GROW)

    73         parent.AddWindow(self.Preview, 0, border=0, flag=wx.GROW)

    74 

    75     def _init_coll_RightGridSizer_Growables(self, parent):

    76         parent.AddGrowableCol(0)

    77         parent.AddGrowableRow(2)

    78 

    79     def _init_coll_RightUpGridSizer_Items(self, parent):

    80         parent.AddWindow(self.staticText2, 0, border=4, flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP)

    81         parent.AddWindow(self.BlockName, 0, border=0, flag=wx.GROW)

    82         parent.AddWindow(self.staticText3, 0, border=4, flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP)

    83         parent.AddWindow(self.Inputs, 0, border=0, flag=wx.GROW)

    84         parent.AddWindow(self.staticText4, 0, border=4, flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP)

    85         parent.AddWindow(self.ExecutionOrder, 0, border=0, flag=wx.GROW)

    86         parent.AddWindow(self.staticText5, 0, border=4, flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP)

    87         parent.AddWindow(self.ExecutionControl, 0, border=0, flag=wx.GROW)

    88     

    89     def _init_coll_RightUpGridSizer_Growables(self, parent):

    90         parent.AddGrowableCol(1)

    91     

    92     def _init_sizers(self):

    93         self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)

    94         self.MainSizer = wx.BoxSizer(wx.HORIZONTAL)

    95         self.LeftBoxSizer = wx.StaticBoxSizer(self.staticbox1, wx.VERTICAL)

    96         self.RightGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=5)

    97         self.RightUpGridSizer = wx.FlexGridSizer(cols=2, hgap=5, rows=4, vgap=5)

    98         

    99         self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

   100         self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

   101         self._init_coll_MainSizer_Items(self.MainSizer)

   102         self._init_coll_LeftBoxSizer_Items(self.LeftBoxSizer)

   103         self._init_coll_RightGridSizer_Items(self.RightGridSizer)

   104         self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

   105         self._init_coll_RightUpGridSizer_Items(self.RightUpGridSizer)

   106         self._init_coll_RightUpGridSizer_Growables(self.RightUpGridSizer)

   107         

   108         self.SetSizer(self.flexGridSizer1)

   109 

   110     def _init_ctrls(self, prnt):

   111         wx.Dialog.__init__(self, id=ID_BLOCKPROPERTIESDIALOG,

   112               name='BlockPropertiesDialog', parent=prnt, pos=wx.Point(376, 223),

   113               size=wx.Size(600, 400), style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER,

   114               title=_('Block Properties'))

   115         self.SetClientSize(wx.Size(600, 400))

   116 

   117         self.staticbox1 = wx.StaticBox(id=ID_BLOCKPROPERTIESDIALOGSTATICTEXT1,

   118               label=_('Type:'), name='staticBox1', parent=self,

   119               pos=wx.Point(0, 0), size=wx.Size(0, 0), style=0)

   120 

   121         self.staticText2 = wx.StaticText(id=ID_BLOCKPROPERTIESDIALOGSTATICTEXT2,

   122               label=_('Name:'), name='staticText2', parent=self,

   123               pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   124 

   125         self.staticText3 = wx.StaticText(id=ID_BLOCKPROPERTIESDIALOGSTATICTEXT2,

   126               label=_('Inputs:'), name='staticText4', parent=self,

   127               pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   128 

   129         self.staticText4 = wx.StaticText(id=ID_BLOCKPROPERTIESDIALOGSTATICTEXT4,

   130               label=_('Execution Order:'), name='staticText4', parent=self,

   131               pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   132 

   133         self.staticText5 = wx.StaticText(id=ID_BLOCKPROPERTIESDIALOGSTATICTEXT5,

   134               label=_('Execution Control:'), name='staticText5', parent=self,

   135               pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   136 

   137         self.staticText6 = wx.StaticText(id=ID_BLOCKPROPERTIESDIALOGSTATICTEXT6,

   138               label=_('Preview:'), name='staticText6', parent=self,

   139               pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   140 

   141         if wx.Platform == '__WXMSW__':

   142             treestyle = wx.TR_HAS_BUTTONS|wx.TR_SINGLE|wx.SUNKEN_BORDER

   143         else:

   144             treestyle = wx.TR_HAS_BUTTONS|wx.TR_HIDE_ROOT|wx.TR_SINGLE|wx.SUNKEN_BORDER

   145         self.TypeTree = wx.TreeCtrl(id=ID_BLOCKPROPERTIESDIALOGTYPETREE,

   146               name='TypeTree', parent=self, pos=wx.Point(0, 0),

   147               size=wx.Size(0, 0), style=treestyle)

   148         self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTypeTreeItemSelected,

   149               id=ID_BLOCKPROPERTIESDIALOGTYPETREE)

   150 

   151         self.TypeDesc = wx.TextCtrl(id=ID_BLOCKPROPERTIESDIALOGTYPEDESC,

   152               name='TypeDesc', parent=self, pos=wx.Point(0, 0),

   153               size=wx.Size(0, 0), style=wx.TE_READONLY|wx.TE_MULTILINE)

   154 

   155         self.BlockName = wx.TextCtrl(id=ID_BLOCKPROPERTIESDIALOGNAME, value='',

   156               name='BlockName', parent=self, pos=wx.Point(0, 0),

   157               size=wx.Size(0, 24), style=0)

   158         self.Bind(wx.EVT_TEXT, self.OnNameChanged, id=ID_BLOCKPROPERTIESDIALOGNAME)

   159 

   160         self.Inputs = wx.SpinCtrl(id=ID_BLOCKPROPERTIESDIALOGINPUTS,

   161               name='Inputs', parent=self, pos=wx.Point(0, 0),

   162               size=wx.Size(0, 24), style=wx.SP_ARROW_KEYS, min=2, max=20)

   163         self.Bind(wx.EVT_SPINCTRL, self.OnInputsChanged, id=ID_BLOCKPROPERTIESDIALOGINPUTS)

   164 

   165         self.ExecutionOrder = wx.SpinCtrl(id=ID_BLOCKPROPERTIESDIALOGEXECUTIONORDER,

   166               name='ExecutionOrder', parent=self, pos=wx.Point(0, 0),

   167               size=wx.Size(0, 24), style=wx.SP_ARROW_KEYS, min=0)

   168         self.Bind(wx.EVT_SPINCTRL, self.OnExecutionOrderChanged, id=ID_BLOCKPROPERTIESDIALOGEXECUTIONORDER)

   169 

   170         self.ExecutionControl = wx.CheckBox(id=ID_BLOCKPROPERTIESDIALOGEXECUTIONCONTROL,

   171               name='ExecutionControl', parent=self, pos=wx.Point(0, 0),

   172               size=wx.Size(0, 24), style=0)

   173         self.Bind(wx.EVT_CHECKBOX, self.OnExecutionOrderChanged, id=ID_BLOCKPROPERTIESDIALOGEXECUTIONCONTROL)

   174 

   175         self.Preview = wx.Panel(id=ID_BLOCKPROPERTIESDIALOGPREVIEW,

   176               name='Preview', parent=self, pos=wx.Point(0, 0),

   177               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   178         self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

   179         setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

   180         setattr(self.Preview, "GetScaling", lambda:None)

   181         setattr(self.Preview, "GetBlockType", self.Controler.GetBlockType)

   182         setattr(self.Preview, "IsOfType", self.Controler.IsOfType)

   183 

   184         self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   185         if wx.VERSION >= (2, 5, 0):

   186             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())

   187             self.Preview.Bind(wx.EVT_PAINT, self.OnPaint)

   188         else:

   189             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())

   190             wx.EVT_PAINT(self.Preview, self.OnPaint)

   191         

   192         self._init_sizers()

   193 

   194     def __init__(self, parent, controler):

   195         self.Controler = controler

   196         self._init_ctrls(parent)

   197         self.BlockName.SetValue("")

   198         self.BlockName.Enable(False)

   199         self.Inputs.Enable(False)

   200         self.Block = None

   201         self.MinBlockSize = None

   202         self.First = True

   203         

   204         self.PouNames = []

   205         self.PouElementNames = []

   206     

   207     def SetPreviewFont(self, font):

   208         self.Preview.SetFont(font)

   209     

   210     def FindTreeItem(self, root, name, inputs = None):

   211         if root.IsOk():

   212             pydata = self.TypeTree.GetPyData(root)

   213             type_inputs = pydata.get("inputs", None)

   214             type_extension = pydata.get("extension", None)

   215             if inputs is not None and type_inputs is not None:

   216                 if type_extension is not None:

   217                     same_inputs = type_inputs == inputs[:type_extension]

   218                 else:

   219                     same_inputs = type_inputs == inputs

   220             else:

   221                 same_inputs = True

   222             if self.TypeTree.GetItemText(root) == name and same_inputs:

   223                 return root

   224             else:

   225                 if wx.VERSION < (2, 6, 0):

   226                     item, root_cookie = self.TypeTree.GetFirstChild(root, 0)

   227                 else:

   228                     item, root_cookie = self.TypeTree.GetFirstChild(root)

   229                 while item.IsOk():

   230                     result = self.FindTreeItem(item, name, inputs)

   231                     if result:

   232                         return result

   233                     item, root_cookie = self.TypeTree.GetNextChild(root, root_cookie)

   234         return None

   235     

   236     def OnOK(self, event):

   237         selected = self.TypeTree.GetSelection()

   238         block_name = self.BlockName.GetValue()

   239         name_enabled = self.BlockName.IsEnabled()

   240         if not selected.IsOk() or self.TypeTree.GetItemParent(selected) == self.TypeTree.GetRootItem() or selected == self.TypeTree.GetRootItem():

   241             message = wx.MessageDialog(self, _("Form isn't complete. Valid block type must be selected!"), _("Error"), wx.OK|wx.ICON_ERROR)

   242             message.ShowModal()

   243             message.Destroy()

   244         elif name_enabled and block_name == "":

   245             message = wx.MessageDialog(self, _("Form isn't complete. Name must be filled!"), _("Error"), wx.OK|wx.ICON_ERROR)

   246             message.ShowModal()

   247             message.Destroy()

   248         elif name_enabled and not TestIdentifier(block_name):

   249             message = wx.MessageDialog(self, _("\"%s\" is not a valid identifier!")%block_name, _("Error"), wx.OK|wx.ICON_ERROR)

   250             message.ShowModal()

   251             message.Destroy()

   252         elif name_enabled and block_name.upper() in IEC_KEYWORDS:

   253             message = wx.MessageDialog(self, _("\"%s\" is a keyword. It can't be used!")%block_name, _("Error"), wx.OK|wx.ICON_ERROR)

   254             message.ShowModal()

   255             message.Destroy()

   256         elif name_enabled and block_name.upper() in self.PouNames:

   257             message = wx.MessageDialog(self, _("\"%s\" pou already exists!")%block_name, _("Error"), wx.OK|wx.ICON_ERROR)

   258             message.ShowModal()

   259             message.Destroy()

   260         elif name_enabled and block_name.upper() in self.PouElementNames:

   261             message = wx.MessageDialog(self, _("\"%s\" element for this pou already exists!")%block_name, _("Error"), wx.OK|wx.ICON_ERROR)

   262             message.ShowModal()

   263             message.Destroy()

   264         else:

   265             self.EndModal(wx.ID_OK)

   266 

   267     def SetBlockList(self, blocktypes):

   268         if wx.Platform == '__WXMSW__':

   269             root = self.TypeTree.AddRoot(_("Block Types"))

   270         else:

   271             root = self.TypeTree.AddRoot("")

   272         self.TypeTree.SetPyData(root, {"type" : CATEGORY})

   273         for category in blocktypes:

   274             category_name = category["name"]

   275             category_item = self.TypeTree.AppendItem(root, _(category_name))

   276             self.TypeTree.SetPyData(category_item, {"type" : CATEGORY})

   277             for blocktype in category["list"]:

   278                 blocktype_item = self.TypeTree.AppendItem(category_item, blocktype["name"])

   279                 block_data = {"type" : BLOCK, 

   280                               "inputs" : tuple([type for name, type, modifier in blocktype["inputs"]]),

   281                               "extension" : None}

   282                 if blocktype["extensible"]:

   283                     block_data["extension"] = len(blocktype["inputs"])

   284                 self.TypeTree.SetPyData(blocktype_item, block_data)

   285         if wx.Platform == '__WXMSW__':

   286             self.TypeTree.Expand(root)

   287 

   288     def SetMinBlockSize(self, size):

   289         self.MinBlockSize = size

   290 

   291     def SetPouNames(self, pou_names):

   292         self.PouNames = [pou_name.upper() for pou_name in pou_names]

   293         

   294     def SetPouElementNames(self, element_names):

   295         self.PouElementNames = [element_name.upper() for element_name in element_names]

   296         

   297     def SetValues(self, values):

   298         blocktype = values.get("type", None)

   299         if blocktype is not None:

   300             inputs = values.get("inputs", None)

   301             item = self.FindTreeItem(self.TypeTree.GetRootItem(), blocktype, inputs)

   302             if item:

   303                 self.TypeTree.SelectItem(item)

   304         for name, value in values.items():

   305             if name == "name":

   306                 self.BlockName.SetValue(value)

   307             elif name == "extension":

   308                 self.Inputs.SetValue(value)

   309             elif name == "executionOrder":

   310                 self.ExecutionOrder.SetValue(value)

   311             elif name == "executionControl":

   312            	    self.ExecutionControl.SetValue(value)

   313         self.RefreshPreview()

   314 

   315     def GetValues(self):

   316         values = {}

   317         item = self.TypeTree.GetSelection()

   318         values["type"] = self.TypeTree.GetItemText(item)

   319         values["inputs"] = self.TypeTree.GetPyData(item)["inputs"]

   320         if self.BlockName.GetValue() != "":

   321             values["name"] = self.BlockName.GetValue()

   322         values["width"], values["height"] = self.Block.GetSize()

   323         values["extension"] = self.Inputs.GetValue()

   324         values["executionOrder"] = self.ExecutionOrder.GetValue()

   325         values["executionControl"] = self.ExecutionControl.GetValue()

   326         return values

   327 

   328     def OnTypeTreeItemSelected(self, event):

   329         self.BlockName.SetValue("")

   330         selected = event.GetItem()

   331         pydata = self.TypeTree.GetPyData(selected)

   332         if pydata["type"] != CATEGORY:

   333             blocktype = self.Controler.GetBlockType(self.TypeTree.GetItemText(selected), pydata["inputs"])

   334             if blocktype:

   335                 self.Inputs.SetValue(len(blocktype["inputs"]))

   336                 self.Inputs.Enable(blocktype["extensible"])

   337                 self.BlockName.Enable(blocktype["type"] != "function")

   338                 comment = blocktype["comment"]

   339                 self.TypeDesc.SetValue(_(comment) + blocktype.get("usage", ""))

   340                 wx.CallAfter(self.RefreshPreview)

   341             else:

   342                 self.BlockName.Enable(False)

   343                 self.Inputs.Enable(False)

   344                 self.Inputs.SetValue(2)

   345                 self.TypeDesc.SetValue("")

   346                 wx.CallAfter(self.ErasePreview)

   347         else:

   348             self.BlockName.Enable(False)

   349             self.Inputs.Enable(False)

   350             self.Inputs.SetValue(2)

   351             self.TypeDesc.SetValue("")

   352             wx.CallAfter(self.ErasePreview)

   353         event.Skip()

   354 

   355     def OnNameChanged(self, event):

   356         if self.BlockName.IsEnabled():

   357             self.RefreshPreview()

   358         event.Skip()

   359     

   360     def OnInputsChanged(self, event):

   361         if self.Inputs.IsEnabled():

   362             self.RefreshPreview()

   363         event.Skip()

   364     

   365     def OnExecutionOrderChanged(self, event):

   366         self.RefreshPreview()

   367         event.Skip()

   368     

   369     def OnExecutionControlChanged(self, event):

   370         self.RefreshPreview()

   371         event.Skip()

   372     

   373     def ErasePreview(self):

   374         dc = wx.ClientDC(self.Preview)

   375         dc.Clear()

   376         self.Block = None

   377         

   378     def RefreshPreview(self):

   379         dc = wx.ClientDC(self.Preview)

   380         dc.SetFont(self.Preview.GetFont())

   381         dc.Clear()

   382         item = self.TypeTree.GetSelection()

   383         if item.IsOk():

   384             pydata = self.TypeTree.GetPyData(item)

   385             if pydata["type"] == CATEGORY:

   386                 self.Block = None

   387             else:

   388                 blocktype = self.TypeTree.GetItemText(item)

   389                 if blocktype:

   390                     self.Block = FBD_Block(self.Preview, blocktype, 

   391                             self.BlockName.GetValue(), 

   392                             extension = self.Inputs.GetValue(), 

   393                             inputs = pydata["inputs"], 

   394                             executionControl = self.ExecutionControl.GetValue(), 

   395                             executionOrder = self.ExecutionOrder.GetValue())

   396                     width, height = self.MinBlockSize

   397                     min_width, min_height = self.Block.GetMinSize()

   398                     width, height = max(min_width, width), max(min_height, height)

   399                     self.Block.SetSize(width, height)

   400                     clientsize = self.Preview.GetClientSize()

   401                     x = (clientsize.width - width) / 2

   402                     y = (clientsize.height - height) / 2

   403                     self.Block.SetPosition(x, y)

   404                     self.Block.Draw(dc)

   405                 else:

   406                     self.Block = None

   407 

   408     def OnPaint(self, event):

   409         if self.Block is not None:

   410             self.RefreshPreview()

   411         event.Skip()

   412 

   413 #-------------------------------------------------------------------------------

   414 #                          Create New Variable Dialog

   415 #-------------------------------------------------------------------------------

   416 

   417 [ID_VARIABLEPROPERTIESDIALOG, ID_VARIABLEPROPERTIESDIALOGSPACER, 

   418  ID_VARIABLEPROPERTIESDIALOGNAME, ID_VARIABLEPROPERTIESDIALOGCLASS, 

   419  ID_VARIABLEPROPERTIESDIALOGPREVIEW, ID_VARIABLEPROPERTIESDIALOGEXPRESSION,

   420  ID_VARIABLEPROPERTIESDIALOGEXECUTIONORDER, ID_VARIABLEPROPERTIESDIALOGSTATICTEXT1, 

   421  ID_VARIABLEPROPERTIESDIALOGSTATICTEXT2, ID_VARIABLEPROPERTIESDIALOGSTATICTEXT3, 

   422  ID_VARIABLEPROPERTIESDIALOGSTATICTEXT4, ID_VARIABLEPROPERTIESDIALOGSTATICTEXT5

   423 ] = [wx.NewId() for _init_ctrls in range(12)]

   424 

   425 VARIABLE_CLASSES_DICT = {INPUT : _("Input"),

   426                          INOUT : _("InOut"),

   427                          OUTPUT : _("Output")}

   428 VARIABLE_CLASSES_DICT_REVERSE = dict([(value, key) for key, value in VARIABLE_CLASSES_DICT.iteritems()])

   429 

   430 class VariablePropertiesDialog(wx.Dialog):

   431     

   432     if wx.VERSION < (2, 6, 0):

   433         def Bind(self, event, function, id = None):

   434             if id is not None:

   435                 event(self, id, function)

   436             else:

   437                 event(self, function)

   438     

   439     def _init_coll_flexGridSizer1_Items(self, parent):

   440         parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

   441         parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

   442         

   443     def _init_coll_flexGridSizer1_Growables(self, parent):

   444         parent.AddGrowableCol(0)

   445         parent.AddGrowableRow(0)

   446     

   447     def _init_coll_MainSizer_Items(self, parent):

   448         parent.AddSizer(self.TopSizer, 0, border=0, flag=wx.GROW)

   449         parent.AddWindow(self.staticText5, 0, border=0, flag=wx.GROW)

   450         parent.AddWindow(self.Preview, 0, border=0, flag=wx.GROW)

   451         

   452     def _init_coll_MainSizer_Growables(self, parent):

   453         parent.AddGrowableCol(0)

   454         parent.AddGrowableRow(2)

   455     

   456     def _init_coll_TopSizer_Items(self, parent):

   457         parent.AddSizer(self.LeftGridSizer, 1, border=5, flag=wx.GROW|wx.RIGHT)

   458         parent.AddSizer(self.RightGridSizer, 1, border=5, flag=wx.GROW|wx.LEFT)

   459     

   460     def _init_coll_LeftGridSizer_Items(self, parent):

   461         parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)

   462         parent.AddWindow(self.Class, 0, border=0, flag=wx.GROW)

   463         parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)

   464         parent.AddWindow(self.Expression, 0, border=0, flag=wx.GROW)

   465         parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW)

   466         parent.AddWindow(self.ExecutionOrder, 0, border=0, flag=wx.GROW)

   467         parent.AddWindow(self.Spacer, 0, border=0, flag=wx.GROW)

   468     

   469     def _init_coll_LeftGridSizer_Growables(self, parent):

   470         parent.AddGrowableCol(0)

   471         parent.AddGrowableRow(2)

   472             

   473     def _init_coll_RightGridSizer_Items(self, parent):

   474         parent.AddWindow(self.staticText4, 0, border=0, flag=wx.GROW)

   475         parent.AddWindow(self.VariableName, 0, border=0, flag=wx.GROW)

   476         

   477     def _init_coll_RightGridSizer_Growables(self, parent):

   478         parent.AddGrowableCol(0)

   479         parent.AddGrowableRow(1)

   480         

   481     def _init_sizers(self):

   482         self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)

   483         self.MainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=5)

   484         self.TopSizer = wx.BoxSizer(wx.HORIZONTAL)

   485         self.LeftGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=7, vgap=5)

   486         self.RightGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5)

   487 

   488         self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

   489         self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

   490         self._init_coll_MainSizer_Items(self.MainSizer)

   491         self._init_coll_MainSizer_Growables(self.MainSizer)

   492         self._init_coll_TopSizer_Items(self.TopSizer)

   493         self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)

   494         self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)

   495         self._init_coll_RightGridSizer_Items(self.RightGridSizer)

   496         self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

   497         

   498         self.SetSizer(self.flexGridSizer1)

   499 

   500     def _init_ctrls(self, prnt, ctrler):

   501         wx.Dialog.__init__(self, id=ID_VARIABLEPROPERTIESDIALOG,

   502               name='VariablePropertiesDialog', parent=prnt, pos=wx.Point(376, 223),

   503               size=wx.Size(400, 380), style=wx.DEFAULT_DIALOG_STYLE,

   504               title=_('Variable Properties'))

   505         self.SetClientSize(wx.Size(400, 380))

   506 

   507         self.staticText1 = wx.StaticText(id=ID_VARIABLEPROPERTIESDIALOGSTATICTEXT1,

   508               label=_('Class:'), name='staticText1', parent=self,

   509               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   510 

   511         self.staticText2 = wx.StaticText(id=ID_VARIABLEPROPERTIESDIALOGSTATICTEXT2,

   512               label=_('Expression:'), name='staticText2', parent=self,

   513               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   514 

   515         self.staticText3 = wx.StaticText(id=ID_VARIABLEPROPERTIESDIALOGSTATICTEXT3,

   516               label=_('Execution Order:'), name='staticText3', parent=self,

   517               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   518 

   519         self.staticText4 = wx.StaticText(id=ID_VARIABLEPROPERTIESDIALOGSTATICTEXT4,

   520               label=_('Name:'), name='staticText4', parent=self,

   521               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   522 

   523         self.staticText5 = wx.StaticText(id=ID_VARIABLEPROPERTIESDIALOGSTATICTEXT5,

   524               label=_('Preview:'), name='staticText5', parent=self,

   525               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   526 

   527         self.Class = wx.ComboBox(id=ID_VARIABLEPROPERTIESDIALOGCLASS,

   528               name='Class', parent=self, pos=wx.Point(0, 0),

   529               size=wx.Size(0, 28), style=wx.CB_READONLY)

   530         self.Bind(wx.EVT_COMBOBOX, self.OnClassChanged, id=ID_VARIABLEPROPERTIESDIALOGCLASS)

   531         

   532         self.VariableName = wx.ListBox(id=ID_VARIABLEPROPERTIESDIALOGNAME,

   533               name='Name', parent=self, pos=wx.Point(0, 0),

   534               size=wx.Size(0, 90), style=wx.LB_SINGLE|wx.LB_SORT)

   535         self.Bind(wx.EVT_LISTBOX, self.OnNameChanged, id=ID_VARIABLEPROPERTIESDIALOGNAME)

   536 

   537         self.Expression = wx.TextCtrl(id=ID_VARIABLEPROPERTIESDIALOGEXPRESSION,

   538               name='Expression', parent=self, pos=wx.Point(0, 0),

   539               size=wx.Size(0, 24), style=0)

   540         self.Bind(wx.EVT_TEXT, self.OnExpressionChanged, id=ID_VARIABLEPROPERTIESDIALOGEXPRESSION)

   541 

   542         self.ExecutionOrder = wx.SpinCtrl(id=ID_VARIABLEPROPERTIESDIALOGEXECUTIONORDER,

   543               name='ExecutionOrder', parent=self, pos=wx.Point(0, 0),

   544               size=wx.Size(0, 24), style=wx.SP_ARROW_KEYS, min=0)

   545         self.Bind(wx.EVT_SPINCTRL, self.OnExecutionOrderChanged, id=ID_VARIABLEPROPERTIESDIALOGEXECUTIONORDER)

   546 

   547         self.Spacer = wx.Panel(id=ID_VARIABLEPROPERTIESDIALOGSPACER,

   548               name='Spacer', parent=self, pos=wx.Point(0, 0),

   549               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

   550 

   551         self.Preview = wx.Panel(id=ID_VARIABLEPROPERTIESDIALOGPREVIEW,

   552               name='Preview', parent=self, pos=wx.Point(0, 0),

   553               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   554         self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

   555         setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

   556         setattr(self.Preview, "GetScaling", lambda:None)

   557         setattr(self.Preview, "IsOfType", ctrler.IsOfType)

   558 

   559         self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   560         if wx.VERSION >= (2, 5, 0):

   561             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())

   562             self.Preview.Bind(wx.EVT_PAINT, self.OnPaint)

   563         else:

   564             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())

   565             wx.EVT_PAINT(self.Preview, self.OnPaint)

   566         

   567         self._init_sizers()

   568 

   569     def __init__(self, parent, controler, transition = ""):

   570         self._init_ctrls(parent, controler)

   571         self.Transition = transition

   572         self.Variable = None

   573         self.VarList = []

   574         self.MinVariableSize = None

   575         

   576         for choice in VARIABLE_CLASSES_DICT.itervalues():

   577             self.Class.Append(choice)

   578         self.Class.SetStringSelection(VARIABLE_CLASSES_DICT[INPUT])

   579 

   580         self.RefreshNameList()

   581 

   582     def SetPreviewFont(self, font):

   583         self.Preview.SetFont(font)

   584 

   585     def RefreshNameList(self):

   586         selected = self.VariableName.GetStringSelection()

   587         var_class = VARIABLE_CLASSES_DICT_REVERSE[self.Class.GetStringSelection()]

   588         self.VariableName.Clear()

   589         self.VariableName.Append("")

   590         for name, var_type, value_type in self.VarList:

   591             if var_type != "Input" or var_class == INPUT:

   592                 self.VariableName.Append(name)

   593         if selected != "" and self.VariableName.FindString(selected) != wx.NOT_FOUND:

   594             self.VariableName.SetStringSelection(selected)

   595             self.Expression.Enable(False)

   596         else:

   597             self.VariableName.SetStringSelection("")

   598             self.Expression.Enable(var_class == INPUT)

   599         self.VariableName.Enable(self.VariableName.GetCount() > 0)

   600             

   601     def SetMinVariableSize(self, size):

   602         self.MinVariableSize = size

   603 

   604     def SetVariables(self, vars):

   605         self.VarList = vars

   606         self.RefreshNameList()

   607 

   608     def SetValues(self, values):

   609         value_type = values.get("type", None)

   610         value_name = values.get("name", None)

   611         if value_type:

   612             self.Class.SetStringSelection(VARIABLE_CLASSES_DICT[value_type])

   613             self.RefreshNameList()

   614         if value_name:

   615             if self.VariableName.FindString(value_name) != wx.NOT_FOUND:

   616                 self.VariableName.SetStringSelection(value_name)

   617                 self.Expression.Enable(False)

   618             else:

   619                 self.Expression.SetValue(value_name)

   620                 self.VariableName.Enable(False)

   621         if "executionOrder" in values:

   622             self.ExecutionOrder.SetValue(values["executionOrder"])

   623         self.RefreshPreview()

   624         

   625     def GetValues(self):

   626         values = {}

   627         values["type"] = VARIABLE_CLASSES_DICT_REVERSE[self.Class.GetStringSelection()]

   628         expression = self.Expression.GetValue()

   629         if self.Expression.IsEnabled() and expression != "":

   630             values["name"] = expression

   631         else:

   632             values["name"] = self.VariableName.GetStringSelection()

   633         values["value_type"] = None

   634         for var_name, var_type, value_type in self.VarList:

   635             if var_name == values["name"]:

   636                 values["value_type"] = value_type

   637         values["width"], values["height"] = self.Variable.GetSize()

   638         values["executionOrder"] = self.ExecutionOrder.GetValue()

   639         return values

   640 

   641     def OnOK(self, event):

   642         expression = self.Expression.GetValue()

   643         if self.Expression.IsEnabled():

   644             value = expression

   645         else:

   646             value = self.VariableName.GetStringSelection()

   647         if value == "":

   648             message = wx.MessageDialog(self, _("At least a variable or an expression must be selected!"), _("Error"), wx.OK|wx.ICON_ERROR)

   649             message.ShowModal()

   650             message.Destroy()

   651         elif value.upper() in IEC_KEYWORDS:

   652             message = wx.MessageDialog(self, _("\"%s\" is a keyword. It can't be used!")%value, _("Error"), wx.OK|wx.ICON_ERROR)

   653             message.ShowModal()

   654             message.Destroy()

   655         else:

   656             self.EndModal(wx.ID_OK)

   657 

   658     def OnClassChanged(self, event):

   659         self.RefreshNameList()

   660         self.RefreshPreview()

   661         event.Skip()

   662 

   663     def OnNameChanged(self, event):

   664         if self.VariableName.GetStringSelection() != "":

   665             self.Expression.Enable(False)

   666         elif VARIABLE_CLASSES_DICT_REVERSE[self.Class.GetStringSelection()] == INPUT:

   667             self.Expression.Enable(True)

   668         self.RefreshPreview()

   669         event.Skip()

   670     

   671     def OnExpressionChanged(self, event):

   672         if self.Expression.GetValue() != "":

   673             self.VariableName.Enable(False)

   674         else:

   675             self.VariableName.Enable(True)

   676         self.RefreshPreview()

   677         event.Skip()

   678     

   679     def OnExecutionOrderChanged(self, event):

   680         self.RefreshPreview()

   681         event.Skip()

   682     

   683     def RefreshPreview(self):

   684         dc = wx.ClientDC(self.Preview)

   685         dc.SetFont(self.Preview.GetFont())

   686         dc.Clear()

   687         expression = self.Expression.GetValue()

   688         if self.Expression.IsEnabled() and expression != "":

   689             name = expression

   690         else:

   691             name = self.VariableName.GetStringSelection()

   692         type = ""

   693         for var_name, var_type, value_type in self.VarList:

   694             if var_name == name:

   695                 type = value_type

   696         classtype = VARIABLE_CLASSES_DICT_REVERSE[self.Class.GetStringSelection()]

   697         self.Variable = FBD_Variable(self.Preview, classtype, name, type, executionOrder = self.ExecutionOrder.GetValue())

   698         width, height = self.MinVariableSize

   699         min_width, min_height = self.Variable.GetMinSize()

   700         width, height = max(min_width, width), max(min_height, height)

   701         self.Variable.SetSize(width, height)

   702         clientsize = self.Preview.GetClientSize()

   703         x = (clientsize.width - width) / 2

   704         y = (clientsize.height - height) / 2

   705         self.Variable.SetPosition(x, y)

   706         self.Variable.Draw(dc)

   707 

   708     def OnPaint(self, event):

   709         self.RefreshPreview()

   710         event.Skip()

   711 

   712 #-------------------------------------------------------------------------------

   713 #                          Create New Connection Dialog

   714 #-------------------------------------------------------------------------------

   715 

   716 [ID_CONNECTIONPROPERTIESDIALOG, ID_CONNECTIONPROPERTIESDIALOGSPACER, 

   717  ID_CONNECTIONPROPERTIESDIALOGNAME, ID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON1, 

   718  ID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON2, ID_CONNECTIONPROPERTIESDIALOGPREVIEW,

   719  ID_CONNECTIONPROPERTIESDIALOGSTATICTEXT1, ID_CONNECTIONPROPERTIESDIALOGSTATICTEXT2, 

   720  ID_CONNECTIONPROPERTIESDIALOGSTATICTEXT3, 

   721 ] = [wx.NewId() for _init_ctrls in range(9)]

   722 

   723 class ConnectionPropertiesDialog(wx.Dialog):

   724     

   725     if wx.VERSION < (2, 6, 0):

   726         def Bind(self, event, function, id = None):

   727             if id is not None:

   728                 event(self, id, function)

   729             else:

   730                 event(self, function)

   731     

   732     def _init_coll_flexGridSizer1_Items(self, parent):

   733         parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

   734         parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

   735         

   736     def _init_coll_flexGridSizer1_Growables(self, parent):

   737         parent.AddGrowableCol(0)

   738         parent.AddGrowableRow(0)

   739     

   740     def _init_coll_MainSizer_Items(self, parent):

   741         parent.AddSizer(self.LeftGridSizer, 1, border=5, flag=wx.GROW|wx.RIGHT)

   742         parent.AddSizer(self.RightGridSizer, 1, border=5, flag=wx.GROW|wx.LEFT)

   743     

   744     def _init_coll_LeftGridSizer_Items(self, parent):

   745         parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)

   746         parent.AddWindow(self.radioButton1, 0, border=0, flag=wx.GROW)

   747         parent.AddWindow(self.radioButton2, 0, border=0, flag=wx.GROW)

   748         parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)

   749         parent.AddWindow(self.ConnectionName, 0, border=0, flag=wx.GROW)

   750         parent.AddWindow(self.Spacer, 0, border=0, flag=wx.GROW)

   751     

   752     def _init_coll_LeftGridSizer_Growables(self, parent):

   753         parent.AddGrowableCol(0)

   754         parent.AddGrowableRow(5)

   755             

   756     def _init_coll_RightGridSizer_Items(self, parent):

   757         parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW)

   758         parent.AddWindow(self.Preview, 0, border=0, flag=wx.GROW)

   759         

   760     def _init_coll_RightGridSizer_Growables(self, parent):

   761         parent.AddGrowableCol(0)

   762         parent.AddGrowableRow(1)

   763 

   764     def _init_sizers(self):

   765         self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)

   766         self.MainSizer = wx.BoxSizer(wx.HORIZONTAL)

   767         self.LeftGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=6, vgap=5)

   768         self.RightGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5)

   769 

   770         self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

   771         self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

   772         self._init_coll_MainSizer_Items(self.MainSizer)

   773         self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)

   774         self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)

   775         self._init_coll_RightGridSizer_Items(self.RightGridSizer)

   776         self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

   777 

   778         self.SetSizer(self.flexGridSizer1)

   779 

   780     def _init_ctrls(self, prnt, ctrler):

   781         wx.Dialog.__init__(self, id=ID_CONNECTIONPROPERTIESDIALOG,

   782               name='ConnectionPropertiesDialog', parent=prnt, pos=wx.Point(376, 223),

   783               size=wx.Size(350, 220), style=wx.DEFAULT_DIALOG_STYLE,

   784               title=_('Connection Properties'))

   785         self.SetClientSize(wx.Size(350, 220))

   786 

   787         self.staticText1 = wx.StaticText(id=ID_CONNECTIONPROPERTIESDIALOGSTATICTEXT1,

   788               label=_('Type:'), name='staticText1', parent=self,

   789               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   790 

   791         self.staticText2 = wx.StaticText(id=ID_CONNECTIONPROPERTIESDIALOGSTATICTEXT2,

   792               label=_('Name:'), name='staticText2', parent=self,

   793               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   794 

   795         self.staticText3 = wx.StaticText(id=ID_CONNECTIONPROPERTIESDIALOGSTATICTEXT3,

   796               label=_('Preview:'), name='staticText3', parent=self,

   797               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

   798 

   799         self.radioButton1 = wx.RadioButton(id=ID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON1,

   800               label=_('Connector'), name='radioButton1', parent=self,

   801               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=wx.RB_GROUP)

   802         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON1)

   803         self.radioButton1.SetValue(True)

   804 

   805         self.radioButton2 = wx.RadioButton(id=ID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON2,

   806               label=_('Continuation'), name='radioButton2', parent=self, 

   807               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=0)

   808         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_CONNECTIONPROPERTIESDIALOGRADIOBUTTON2)

   809         self.radioButton2.SetValue(False)

   810         

   811         self.ConnectionName = wx.TextCtrl(id=ID_CONNECTIONPROPERTIESDIALOGNAME,

   812               name='Name', parent=self, pos=wx.Point(0, 0),

   813               size=wx.Size(0, 24), style=0)

   814         self.Bind(wx.EVT_TEXT, self.OnNameChanged, id=ID_CONNECTIONPROPERTIESDIALOGNAME)

   815 

   816         self.Preview = wx.Panel(id=ID_CONNECTIONPROPERTIESDIALOGPREVIEW,

   817               name='Preview', parent=self, pos=wx.Point(0, 0),

   818               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   819         self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

   820         setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

   821         setattr(self.Preview, "GetScaling", lambda:None)

   822         setattr(self.Preview, "IsOfType", ctrler.IsOfType)

   823 

   824         self.Spacer = wx.Panel(id=ID_CONNECTIONPROPERTIESDIALOGSPACER,

   825               name='Spacer', parent=self, pos=wx.Point(0, 0),

   826               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

   827 

   828         self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   829         if wx.VERSION >= (2, 5, 0):

   830             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())

   831             self.Preview.Bind(wx.EVT_PAINT, self.OnPaint)

   832         else:

   833             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())

   834             wx.EVT_PAINT(self.Preview, self.OnPaint)

   835         

   836         self._init_sizers()

   837 

   838     def __init__(self, parent, controler):

   839         self._init_ctrls(parent, controler)

   840         self.Connection = None

   841         self.MinConnectionSize = None

   842         

   843         self.PouNames = []

   844         self.PouElementNames = []

   845     

   846     def SetPreviewFont(self, font):

   847         self.Preview.SetFont(font)

   848     

   849     def SetMinConnectionSize(self, size):

   850         self.MinConnectionSize = size

   851     

   852     def SetValues(self, values):

   853         for name, value in values.items():

   854             if name == "type":

   855                 if value == CONNECTOR:

   856                     self.radioButton1.SetValue(True)

   857                 elif value == CONTINUATION:

   858                     self.radioButton2.SetValue(True)

   859             elif name == "name":

   860                 self.ConnectionName.SetValue(value)

   861         self.RefreshPreview()

   862     

   863     def GetValues(self):

   864         values = {}

   865         if self.radioButton1.GetValue():

   866             values["type"] = CONNECTOR

   867         else:

   868             values["type"] = CONTINUATION

   869         values["name"] = self.ConnectionName.GetValue()

   870         values["width"], values["height"] = self.Connection.GetSize()

   871         return values

   872 

   873     def SetPouNames(self, pou_names):

   874         self.PouNames = [pou_name.upper() for pou_name in pou_names]

   875         

   876     def SetPouElementNames(self, element_names):

   877         self.PouElementNames = [element_name.upper() for element_name in element_names]

   878 

   879     def OnOK(self, event):

   880         connection_name = self.ConnectionName.GetValue()

   881         if connection_name == "":

   882             message = wx.MessageDialog(self, _("Form isn't complete. Name must be filled!"), _("Error"), wx.OK|wx.ICON_ERROR)

   883             message.ShowModal()

   884             message.Destroy()

   885         elif not TestIdentifier(connection_name):

   886             message = wx.MessageDialog(self, _("\"%s\" is not a valid identifier!")%connection_name, _("Error"), wx.OK|wx.ICON_ERROR)

   887             message.ShowModal()

   888             message.Destroy()

   889         elif connection_name.upper() in IEC_KEYWORDS:

   890             message = wx.MessageDialog(self, _("\"%s\" is a keyword. It can't be used!")%connection_name, _("Error"), wx.OK|wx.ICON_ERROR)

   891             message.ShowModal()

   892             message.Destroy()

   893         elif connection_name.upper() in self.PouNames:

   894             message = wx.MessageDialog(self, _("\"%s\" pou already exists!")%connection_name, _("Error"), wx.OK|wx.ICON_ERROR)

   895             message.ShowModal()

   896             message.Destroy()

   897         elif connection_name.upper() in self.PouElementNames:

   898             message = wx.MessageDialog(self, _("\"%s\" element for this pou already exists!")%connection_name, _("Error"), wx.OK|wx.ICON_ERROR)

   899             message.ShowModal()

   900             message.Destroy()

   901         else:

   902             self.EndModal(wx.ID_OK)

   903 

   904     def OnTypeChanged(self, event):

   905         self.RefreshPreview()

   906         event.Skip()

   907 

   908     def OnNameChanged(self, event):

   909         self.RefreshPreview()

   910         event.Skip()

   911         

   912     def RefreshPreview(self):

   913         dc = wx.ClientDC(self.Preview)

   914         dc.SetFont(self.Preview.GetFont())

   915         dc.Clear()

   916         if self.radioButton1.GetValue():

   917             self.Connection = FBD_Connector(self.Preview, CONNECTOR, self.ConnectionName.GetValue())

   918         else:

   919             self.Connection = FBD_Connector(self.Preview, CONTINUATION, self.ConnectionName.GetValue())

   920         width, height = self.MinConnectionSize

   921         min_width, min_height = self.Connection.GetMinSize()

   922         width, height = max(min_width, width), max(min_height, height)

   923         self.Connection.SetSize(width, height)

   924         clientsize = self.Preview.GetClientSize()

   925         x = (clientsize.width - width) / 2

   926         y = (clientsize.height - height) / 2

   927         self.Connection.SetPosition(x, y)

   928         self.Connection.Draw(dc)

   929 

   930     def OnPaint(self, event):

   931         self.RefreshPreview()

   932         event.Skip()

   933 

   934 #-------------------------------------------------------------------------------

   935 #                        Edit Ladder Element Properties Dialog

   936 #-------------------------------------------------------------------------------

   937 

   938 

   939 [ID_LDELEMENTDIALOG, ID_LDELEMENTDIALOGSPACER, 

   940  ID_LDELEMENTDIALOGNAME, ID_LDELEMENTDIALOGRADIOBUTTON1, 

   941  ID_LDELEMENTDIALOGRADIOBUTTON2, ID_LDELEMENTDIALOGRADIOBUTTON3,

   942  ID_LDELEMENTDIALOGRADIOBUTTON4, ID_LDELEMENTDIALOGRADIOBUTTON5,

   943  ID_LDELEMENTDIALOGRADIOBUTTON6, ID_LDELEMENTDIALOGPREVIEW,

   944  ID_LDELEMENTDIALOGSTATICTEXT1, ID_LDELEMENTDIALOGSTATICTEXT2, 

   945  ID_LDELEMENTDIALOGSTATICTEXT3, 

   946 ] = [wx.NewId() for _init_ctrls in range(13)]

   947 

   948 class LDElementDialog(wx.Dialog):

   949     

   950     if wx.VERSION < (2, 6, 0):

   951         def Bind(self, event, function, id = None):

   952             if id is not None:

   953                 event(self, id, function)

   954             else:

   955                 event(self, function)

   956     

   957     def _init_coll_flexGridSizer1_Items(self, parent):

   958         parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

   959         parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

   960         

   961     def _init_coll_flexGridSizer1_Growables(self, parent):

   962         parent.AddGrowableCol(0)

   963         parent.AddGrowableRow(0)

   964     

   965     def _init_coll_MainSizer_Items(self, parent):

   966         parent.AddSizer(self.LeftGridSizer, 1, border=5, flag=wx.GROW|wx.RIGHT)

   967         parent.AddSizer(self.RightGridSizer, 1, border=5, flag=wx.GROW|wx.LEFT)

   968     

   969     def _init_coll_LeftGridSizer_Items(self, parent):

   970         parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)

   971         parent.AddSizer(self.RadioButtonSizer, 0, border=0, flag=wx.GROW)

   972         parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)

   973         parent.AddWindow(self.ElementName, 0, border=0, flag=wx.GROW)

   974         parent.AddWindow(self.Spacer, 0, border=0, flag=wx.GROW)

   975         

   976     def _init_coll_LeftGridSizer_Growables(self, parent):

   977         parent.AddGrowableCol(0)

   978         parent.AddGrowableRow(7)

   979     

   980     def _init_coll_RadioButtonSizer_Items(self, parent):

   981         parent.AddWindow(self.radioButton1, 0, border=0, flag=wx.GROW)

   982         parent.AddWindow(self.radioButton2, 0, border=0, flag=wx.GROW)

   983         parent.AddWindow(self.radioButton3, 0, border=0, flag=wx.GROW)

   984         parent.AddWindow(self.radioButton4, 0, border=0, flag=wx.GROW)

   985         parent.AddWindow(self.radioButton5, 0, border=0, flag=wx.GROW)

   986         parent.AddWindow(self.radioButton6, 0, border=0, flag=wx.GROW)

   987         

   988     def _init_coll_RightGridSizer_Items(self, parent):

   989         parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW)

   990         parent.AddWindow(self.Preview, 0, border=0, flag=wx.GROW)

   991         

   992     def _init_coll_RightGridSizer_Growables(self, parent):

   993         parent.AddGrowableCol(0)

   994         parent.AddGrowableRow(1)

   995 

   996     def _init_sizers(self):

   997         self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)

   998         self.MainSizer = wx.BoxSizer(wx.HORIZONTAL)

   999         self.LeftGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=5, vgap=5)

  1000         self.RadioButtonSizer = wx.BoxSizer(wx.VERTICAL)

  1001         self.RightGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5)

  1002 

  1003         self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

  1004         self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

  1005         self._init_coll_MainSizer_Items(self.MainSizer)

  1006         self._init_coll_LeftGridSizer_Items(self.LeftGridSizer)

  1007         self._init_coll_LeftGridSizer_Growables(self.LeftGridSizer)

  1008         self._init_coll_RadioButtonSizer_Items(self.RadioButtonSizer)

  1009         self._init_coll_RightGridSizer_Items(self.RightGridSizer)

  1010         self._init_coll_RightGridSizer_Growables(self.RightGridSizer)

  1011 

  1012         self.SetSizer(self.flexGridSizer1)

  1013 

  1014     def _init_ctrls(self, prnt, ctrler, title, extra_size = 0):

  1015         wx.Dialog.__init__(self, id=ID_LDELEMENTDIALOG,

  1016               name='LDElementDialog', parent=prnt, pos=wx.Point(376, 223),

  1017               size=wx.Size(350, 260 + extra_size), style=wx.DEFAULT_DIALOG_STYLE,

  1018               title=title)

  1019         self.SetClientSize(wx.Size(350, 260 + extra_size))

  1020 

  1021         self.staticText1 = wx.StaticText(id=ID_LDELEMENTDIALOGSTATICTEXT1,

  1022               label=_('Modifier:'), name='staticText1', parent=self,

  1023               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

  1024 

  1025         self.staticText2 = wx.StaticText(id=ID_LDELEMENTDIALOGSTATICTEXT2,

  1026               label=_('Name:'), name='staticText2', parent=self,

  1027               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

  1028 

  1029         self.staticText3 = wx.StaticText(id=ID_LDELEMENTDIALOGSTATICTEXT3,

  1030               label=_('Preview:'), name='staticText3', parent=self,

  1031               pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

  1032 

  1033         self.radioButton1 = wx.RadioButton(id=ID_LDELEMENTDIALOGRADIOBUTTON1,

  1034               label=_("Normal"), name='radioButton1', parent=self,

  1035               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=wx.RB_GROUP)

  1036         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_LDELEMENTDIALOGRADIOBUTTON1)

  1037         self.radioButton1.SetValue(True)

  1038 

  1039         self.radioButton2 = wx.RadioButton(id=ID_LDELEMENTDIALOGRADIOBUTTON2,

  1040               label=_("Negated"), name='radioButton2', parent=self, 

  1041               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=0)

  1042         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_LDELEMENTDIALOGRADIOBUTTON2)

  1043 

  1044         self.radioButton3 = wx.RadioButton(id=ID_LDELEMENTDIALOGRADIOBUTTON3,

  1045               label=_("Set"), name='radioButton3', parent=self,

  1046               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=0)

  1047         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_LDELEMENTDIALOGRADIOBUTTON3)

  1048 

  1049         self.radioButton4 = wx.RadioButton(id=ID_LDELEMENTDIALOGRADIOBUTTON4,

  1050               label=_("Reset"), name='radioButton4', parent=self, 

  1051               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=0)

  1052         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_LDELEMENTDIALOGRADIOBUTTON4)

  1053 

  1054         self.radioButton5 = wx.RadioButton(id=ID_LDELEMENTDIALOGRADIOBUTTON5,

  1055               label=_("Rising Edge"), name='radioButton5', parent=self, 

  1056               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=0)

  1057         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_LDELEMENTDIALOGRADIOBUTTON5)

  1058 

  1059         self.radioButton6 = wx.RadioButton(id=ID_LDELEMENTDIALOGRADIOBUTTON6,

  1060               label=_("Falling Edge"), name='radioButton6', parent=self, 

  1061               pos=wx.Point(0, 0), size=wx.Size(0, 24), style=0)

  1062         self.Bind(wx.EVT_RADIOBUTTON, self.OnTypeChanged, id=ID_LDELEMENTDIALOGRADIOBUTTON6)

  1063 

  1064         self.ElementName = wx.ComboBox(id=ID_LDELEMENTDIALOGNAME,

  1065               name='Name', parent=self, pos=wx.Point(0, 0),

  1066               size=wx.Size(0, 28), style=wx.CB_READONLY)

  1067         self.Bind(wx.EVT_COMBOBOX, self.OnNameChanged, id=ID_LDELEMENTDIALOGNAME)

  1068 

  1069         self.Preview = wx.Panel(id=ID_LDELEMENTDIALOGPREVIEW,

  1070               name='Preview', parent=self, pos=wx.Point(0, 0),

  1071               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

  1072         self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

  1073         setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

  1074         setattr(self.Preview, "GetScaling", lambda:None)

  1075         setattr(self.Preview, "IsOfType", ctrler.IsOfType)

  1076 

  1077         self.Spacer = wx.Panel(id=ID_LDELEMENTDIALOGSPACER,

  1078               name='Spacer', parent=self, pos=wx.Point(0, 0),

  1079               size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

  1080 

  1081         self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

  1082         

  1083         if wx.VERSION >= (2, 5, 0):

  1084             self.Preview.Bind(wx.EVT_PAINT, self.OnPaint)

  1085         else:

  1086             wx.EVT_PAINT(self.Preview, self.OnPaint)

  1087         

  1088         self._init_sizers()

  1089 

  1090     def __init__(self, parent, controler, type):

  1091         self.Type = type

  1092         if type == "contact":

  1093             self._init_ctrls(parent, controler, _("Edit Contact Values"))

  1094             self.Element = LD_Contact(self.Preview, CONTACT_NORMAL, "")

  1095             self.radioButton3.Hide()

  1096             self.radioButton4.Hide()

  1097         elif type == "coil":

  1098             self._init_ctrls(parent, controler, _("Edit Coil Values"), 50)

  1099             self.Element = LD_Coil(self.Preview, COIL_NORMAL, "")

  1100             

  1101     

  1102     def SetPreviewFont(self, font):

  1103         self.Preview.SetFont(font)

  1104     

  1105     def SetElementSize(self, size):

  1106         min_width, min_height = self.Element.GetMinSize()

  1107         width, height = max(min_width, size[0]), max(min_height, size[1])

  1108         self.Element.SetSize(width, height)

  1109         

  1110     def SetVariables(self, vars):

  1111         self.ElementName.Clear()

  1112         for name in vars:

  1113             self.ElementName.Append(name)

  1114         self.ElementName.Enable(self.ElementName.GetCount() > 0)

  1115 

  1116     def SetValues(self, values):

  1117         for name, value in values.items():

  1118             if name == "name":

  1119                 self.Element.SetName(value)

  1120                 self.ElementName.SetStringSelection(value)

  1121             elif name == "type":

  1122                 self.Element.SetType(value)

  1123                 if self.Type == "contact":

  1124                     if value == CONTACT_NORMAL:

  1125                         self.radioButton1.SetValue(True)

  1126                     elif value == CONTACT_REVERSE:

  1127                         self.radioButton2.SetValue(True)

  1128                     elif value == CONTACT_RISING:

  1129                         self.radioButton5.SetValue(True)

  1130                     elif value == CONTACT_FALLING:

  1131                         self.radioButton6.SetValue(True)

  1132                 elif self.Type == "coil":

  1133                     if value == COIL_NORMAL:

  1134                         self.radioButton1.SetValue(True)

  1135                     elif value == COIL_REVERSE:

  1136                         self.radioButton2.SetValue(True)