1

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

     2

     3

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

     4

#based on the plcopen standard. 

     5

#

     6

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

     7

#

     8

#See COPYING file for copyrights details.

     9

#

    10

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

    11

#modify it under the terms of the GNU General Public

    12

#License as published by the Free Software Foundation; either

    13

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

    14

#

    15

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

    16

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

    17

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

    18

#General Public License for more details.

    19

#

    20

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

    21

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

    22

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

    23

    24

import wx

    25

    26

#-------------------------------------------------------------------------------

    27

#                            Force Variable Dialog

    28

#-------------------------------------------------------------------------------

    29

    30

LOCATIONDATATYPES = {"X" : ["BOOL"],

    31

                     "B" : ["SINT", "USINT", "BYTE", "STRING"],

    32

                     "W" : ["INT", "UINT", "WORD", "WSTRING"],

    33

                     "D" : ["DINT", "UDINT", "REAL", "DWORD"],

    34

                     "L" : ["LINT", "ULINT", "LREAL", "LWORD"]} 

    35

    36

def gen_get_function(f):

    37

    def get_function(v):

    38

        try:

    39

            return f(v)

    40

        except:

    41

            return None

    42

    return get_function

    43

    44

getinteger = gen_get_function(int)

    45

getfloat = gen_get_function(float)

    46

getstring = gen_get_function(str)

    47

    48

GetTypeValue = {"BOOL": lambda x: {"TRUE": True, "FALSE": False}.get(x.upper(), None),

    49

                "SINT": getinteger,

    50

                "INT": getinteger,

    51

                "DINT": getinteger,

    52

                "LINT": getinteger,

    53

                "USINT": getinteger,

    54

                "UINT": getinteger,

    55

                "UDINT": getinteger,

    56

                "ULINT": getinteger,

    57

                "BYTE": getinteger,

    58

                "WORD": getinteger,

    59

                "DWORD": getinteger,

    60

                "LWORD": getinteger,

    61

                "REAL": getfloat,

    62

                "LREAL": getfloat,

    63

                "STRING": getstring,

    64

                "WSTRING": getstring}

    65

    66

    67

class ForceVariableDialog(wx.TextEntryDialog):

    68

    69

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

    70

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

    71

            if id is not None:

    72

                event(self, id, function)

    73

            else:

    74

                event(self, function)

    75

    76

    def __init__(self, parent, iec_type, defaultValue=""):

    77

        wx.TextEntryDialog.__init__(self, parent, message = _("Forcing Variable Value"), 

    78

                caption = _("Please enter value for a \"%s\" variable:"%iec_type), defaultValue = defaultValue, 

    79

                style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition)

    80

    81

        self.IEC_Type = iec_type 

    82

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

    83

            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(2).GetSizer().GetItem(1).GetSizer().GetAffirmativeButton().GetId())

    84

        elif wx.VERSION >= (2, 6, 0):

    85

            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())

    86

        else:

    87

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

    88

    89

    def OnOK(self, event):

    90

        value = self.GetSizer().GetItem(1).GetWindow().GetValue()

    91

        if value == "":

    92

            message = wx.MessageDialog(self, _("You must type a value!"), _("Error"), wx.OK|wx.ICON_ERROR)

    93

            message.ShowModal()

    94

            message.Destroy()

    95

        elif GetTypeValue[self.IEC_Type](value) is None:

    96

            message = wx.MessageDialog(self, _("Invalid value \"%s\" for \"%s\" variable!")%(value, self.IEC_Type), _("Error"), wx.OK|wx.ICON_ERROR)

    97

            message.ShowModal()

    98

            message.Destroy()

    99

        else:

   100

            self.EndModal(wx.ID_OK)

   101

   102

    def GetValue(self):

   103

        return GetTypeValue[self.IEC_Type](self.GetSizer().GetItem(1).GetWindow().GetValue())