dialogs/ForceVariableDialog.py
changeset 469 17411b970353
child 475 9ca03fdff80f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/dialogs/ForceVariableDialog.py	Sun Dec 06 21:20:55 2009 +0100
@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+
+#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor
+#based on the plcopen standard. 
+#
+#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
+#
+#See COPYING file for copyrights details.
+#
+#This library is free software; you can redistribute it and/or
+#modify it under the terms of the GNU General Public
+#License as published by the Free Software Foundation; either
+#version 2.1 of the License, or (at your option) any later version.
+#
+#This library is distributed in the hope that it will be useful,
+#but WITHOUT ANY WARRANTY; without even the implied warranty of
+#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#General Public License for more details.
+#
+#You should have received a copy of the GNU General Public
+#License along with this library; if not, write to the Free Software
+#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+import wx
+
+#-------------------------------------------------------------------------------
+#                            Force Variable Dialog
+#-------------------------------------------------------------------------------
+
+LOCATIONDATATYPES = {"X" : ["BOOL"],
+                     "B" : ["SINT", "USINT", "BYTE", "STRING"],
+                     "W" : ["INT", "UINT", "WORD", "WSTRING"],
+                     "D" : ["DINT", "UDINT", "REAL", "DWORD"],
+                     "L" : ["LINT", "ULINT", "LREAL", "LWORD"]} 
+
+def checkbool(v):
+    return v in ["TRUE", "FALSE"]
+
+def gen_check_function(f):
+    def check_function(v):
+        try:
+            f(v)
+            return True
+        except:
+            return False
+    return check_function
+
+checkinteger = gen_check_function(int)
+checkfloat = gen_check_function(float)
+checkstring = gen_check_function(str)
+
+CheckTypeValue = {"BOOL": lambda x: x in ["TRUE", "FALSE"],
+                  "SINT": checkinteger,
+                  "INT": checkinteger,
+                  "DINT": checkinteger,
+                  "LINT": checkinteger,
+                  "USINT": checkinteger,
+                  "UINT": checkinteger,
+                  "UDINT": checkinteger,
+                  "ULINT": checkinteger,
+                  "BYTE": checkinteger,
+                  "WORD": checkinteger,
+                  "DWORD": checkinteger,
+                  "LWORD": checkinteger,
+                  "REAL": checkfloat,
+                  "LREAL": checkfloat,
+                  "STRING": checkstring,
+                  "WSTRING": checkstring,}
+
+
+class ForceVariableDialog(wx.TextEntryDialog):
+
+    if wx.VERSION < (2, 6, 0):
+        def Bind(self, event, function, id = None):
+            if id is not None:
+                event(self, id, function)
+            else:
+                event(self, function)
+
+    def __init__(self, parent, iec_type):
+        wx.TextEntryDialog.__init__(self, parent, message = _("Forcing Variable Value"), 
+                caption = _("Please enter value for a \"%s\" variable:"%iec_type), defaultValue = "", 
+                style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition)
+        
+        self.IEC_Type = iec_type 
+        if wx.VERSION >= (2, 8, 0):
+            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(2).GetSizer().GetItem(1).GetSizer().GetAffirmativeButton().GetId())
+        elif wx.VERSION >= (2, 6, 0):
+            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())
+        else:
+            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())
+    
+    def OnOK(self, event):
+        value = self.GetSizer().GetItem(1).GetWindow().GetValue()
+        if value == "":
+            message = wx.MessageDialog(self, _("You must type a value!"), _("Error"), wx.OK|wx.ICON_ERROR)
+            message.ShowModal()
+            message.Destroy()
+        elif not CheckTypeValue[self.IEC_Type](value):
+            message = wx.MessageDialog(self, _("Invalid value \"%s\" for \"%s\" variable!")%(value, self.IEC_Type), _("Error"), wx.OK|wx.ICON_ERROR)
+            message.ShowModal()
+            message.Destroy()
+        else:
+            self.EndModal(wx.ID_OK)
+
+    def GetValue(self):
+        return self.GetSizer().GetItem(1).GetWindow().GetValue()