|
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) 2012: 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 |
|
27 from plcopen.structures import TestIdentifier, IEC_KEYWORDS |
|
28 |
|
29 def GetPouTypes(): |
|
30 _ = lambda x : x |
|
31 return [_("function"), _("functionBlock"), _("program")] |
|
32 POU_TYPES_DICT = dict([(_(pou_type), pou_type) for pou_type in GetPouTypes()]) |
|
33 |
|
34 def GetPouLanguages(): |
|
35 _ = lambda x : x |
|
36 return [_("IL"), _("ST"), _("LD"), _("FBD"), _("SFC")] |
|
37 POU_LANGUAGES_DICT = dict([(_(language), language) for language in GetPouLanguages()]) |
|
38 |
|
39 class PouDialog(wx.Dialog): |
|
40 |
|
41 def __init__(self, parent, pou_type = None): |
|
42 wx.Dialog.__init__(self, id=-1, parent=parent, |
|
43 name='PouDialog', title=_('Create a new POU'), |
|
44 size=wx.Size(300, 200), style=wx.DEFAULT_DIALOG_STYLE) |
|
45 self.SetClientSize(wx.Size(300, 200)) |
|
46 |
|
47 main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10) |
|
48 main_sizer.AddGrowableCol(0) |
|
49 main_sizer.AddGrowableRow(0) |
|
50 |
|
51 infos_sizer = wx.FlexGridSizer(cols=2, hgap=5, rows=3, vgap=15) |
|
52 infos_sizer.AddGrowableCol(1) |
|
53 main_sizer.AddSizer(infos_sizer, border=20, |
|
54 flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT) |
|
55 |
|
56 pouname_label = wx.StaticText(self, label=_('POU Name:')) |
|
57 infos_sizer.AddWindow(pouname_label, border=4, |
|
58 flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP) |
|
59 |
|
60 self.PouName = wx.TextCtrl(self) |
|
61 infos_sizer.AddWindow(self.PouName, flag=wx.GROW) |
|
62 |
|
63 poutype_label = wx.StaticText(self, label=_('POU Type:')) |
|
64 infos_sizer.AddWindow(poutype_label, border=4, |
|
65 flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP) |
|
66 |
|
67 self.PouType = wx.ComboBox(self, style=wx.CB_READONLY) |
|
68 self.Bind(wx.EVT_COMBOBOX, self.OnTypeChanged, self.PouType) |
|
69 infos_sizer.AddWindow(self.PouType, flag=wx.GROW) |
|
70 |
|
71 language_label = wx.StaticText(self, label=_('Language:')) |
|
72 infos_sizer.AddWindow(language_label, border=4, |
|
73 flag=wx.ALIGN_CENTER_VERTICAL|wx.TOP) |
|
74 |
|
75 self.Language = wx.ComboBox(self, style=wx.CB_READONLY) |
|
76 infos_sizer.AddWindow(self.Language, flag=wx.GROW) |
|
77 |
|
78 button_sizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE) |
|
79 self.Bind(wx.EVT_BUTTON, self.OnOK, button_sizer.GetAffirmativeButton()) |
|
80 main_sizer.AddSizer(button_sizer, border=20, |
|
81 flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT) |
|
82 |
|
83 self.SetSizer(main_sizer) |
|
84 |
|
85 for option in GetPouTypes(): |
|
86 self.PouType.Append(_(option)) |
|
87 if pou_type is not None: |
|
88 self.PouType.SetStringSelection(_(pou_type)) |
|
89 self.RefreshLanguage() |
|
90 |
|
91 self.PouNames = [] |
|
92 self.PouElementNames = [] |
|
93 |
|
94 def OnOK(self, event): |
|
95 error = [] |
|
96 pou_name = self.PouName.GetValue() |
|
97 if pou_name == "": |
|
98 error.append(_("POU Name")) |
|
99 if self.PouType.GetSelection() == -1: |
|
100 error.append(_("POU Type")) |
|
101 if self.Language.GetSelection() == -1: |
|
102 error.append(_("Language")) |
|
103 message = None |
|
104 question = False |
|
105 if len(error) > 0: |
|
106 text = "" |
|
107 for i, item in enumerate(error): |
|
108 if i == 0: |
|
109 text += item |
|
110 elif i == len(error) - 1: |
|
111 text += _(" and %s")%item |
|
112 else: |
|
113 text += _(", %s")%item |
|
114 message = _("Form isn't complete. %s must be filled!") % text |
|
115 elif not TestIdentifier(pou_name): |
|
116 message = _("\"%s\" is not a valid identifier!") % pou_name |
|
117 elif pou_name.upper() in IEC_KEYWORDS: |
|
118 message = _("\"%s\" is a keyword. It can't be used!") % pou_name |
|
119 elif pou_name.upper() in self.PouNames: |
|
120 message = _("\"%s\" pou already exists!") % pou_name |
|
121 elif pou_name.upper() in self.PouElementNames: |
|
122 message = _("A POU has an element named \"%s\". This could cause a conflict. Do you wish to continue?") % pou_name |
|
123 question = True |
|
124 if message is not None: |
|
125 if question: |
|
126 dialog = wx.MessageDialog(self, message, _("Warning"), wx.YES_NO|wx.ICON_EXCLAMATION) |
|
127 result = dialog.ShowModal() |
|
128 dialog.Destroy() |
|
129 if result == wx.ID_YES: |
|
130 self.EndModal(wx.ID_OK) |
|
131 else: |
|
132 dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR) |
|
133 dialog.ShowModal() |
|
134 dialog.Destroy() |
|
135 else: |
|
136 self.EndModal(wx.ID_OK) |
|
137 |
|
138 def RefreshLanguage(self): |
|
139 selection = POU_LANGUAGES_DICT.get(self.Language.GetStringSelection(), "") |
|
140 self.Language.Clear() |
|
141 for language in GetPouLanguages(): |
|
142 if language != "SFC" or POU_TYPES_DICT[self.PouType.GetStringSelection()] != "function": |
|
143 self.Language.Append(_(language)) |
|
144 if self.Language.FindString(_(selection)) != wx.NOT_FOUND: |
|
145 self.Language.SetStringSelection(_(selection)) |
|
146 |
|
147 def OnTypeChanged(self, event): |
|
148 self.RefreshLanguage() |
|
149 event.Skip() |
|
150 |
|
151 def SetPouNames(self, pou_names): |
|
152 self.PouNames = [pou_name.upper() for pou_name in pou_names] |
|
153 |
|
154 def SetPouElementNames(self, element_names): |
|
155 self.PouElementNames = [element_name.upper() for element_name in element_names] |
|
156 |
|
157 def SetValues(self, values): |
|
158 for item, value in values.items(): |
|
159 if item == "pouName": |
|
160 self.PouName.SetValue(value) |
|
161 elif item == "pouType": |
|
162 self.PouType.SetStringSelection(_(value)) |
|
163 elif item == "language": |
|
164 self.Language.SetStringSelection(_(value)) |
|
165 |
|
166 def GetValues(self): |
|
167 values = {} |
|
168 values["pouName"] = self.PouName.GetValue() |
|
169 values["pouType"] = POU_TYPES_DICT[self.PouType.GetStringSelection()] |
|
170 values["language"] = POU_LANGUAGES_DICT[self.Language.GetStringSelection()] |
|
171 return values |