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

import re

    26

import datetime

    27

    28

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

    29

#                            Force Variable Dialog

    30

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

    31

    32

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

    33

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

    34

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

    35

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

    36

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

    37

    38

def gen_get_function(f):

    39

    def get_function(v):

    40

        try:

    41

            return f(v)

    42

        except:

    43

            return None

    44

    return get_function

    45

    46

getinteger = gen_get_function(int)

    47

getfloat = gen_get_function(float)

    48

getstring = gen_get_function(str)

    49

    50

SECOND = 1000000

    51

MINUTE = 60 * SECOND

    52

HOUR = 60 * MINUTE

    53

DAY = 24 * HOUR

    54

    55

IEC_TIME_MODEL = re.compile("(?:(?:T|TIME)#)?(-)?(?:(%(float)s)D_?)?(?:(%(float)s)H_?)?(?:(%(float)s)M(?!S)_?)?(?:(%(float)s)S_?)?(?:(%(float)s)MS)?" % {"float": "[0-9]+(?:\.[0-9]+)?"})

    56

IEC_DATE_MODEL = re.compile("(?:(?:D|DATE)#)?([0-9]{4})-([0-9]{2})-([0-9]{2})")

    57

IEC_DATETIME_MODEL = re.compile("(?:(?:DT|DATE_AND_TIME)#)?([0-9]{4})-([0-9]{2})-([0-9]{2})-([0-9]{2}):([0-9]{2}):([0-9]{2}(?:\.[0-9]+)?)")

    58

IEC_TIMEOFDAY_MODEL = re.compile("(?:(?:TOD|TIME_OF_DAY)#)?([0-9]{2}):([0-9]{2}):([0-9]{2}(?:\.[0-9]+)?)")

    59

    60

def gettime(v):    

    61

    result = IEC_TIME_MODEL.match(v.upper())

    62

    if result is not None:

    63

        negative, days, hours, minutes, seconds, milliseconds = result.groups()

    64

        microseconds = 0

    65

        not_null = False

    66

        for value, factor in [(days, DAY),

    67

                              (hours, HOUR),

    68

                              (minutes, MINUTE),

    69

                              (seconds, SECOND),

    70

                              (milliseconds, 1000)]:

    71

            if value is not None:

    72

                microseconds += float(value) * factor

    73

                not_null = True

    74

        if not not_null:

    75

            return None

    76

        if negative is not None:

    77

            microseconds = -microseconds

    78

        return datetime.timedelta(microseconds=microseconds)

    79

    80

    else: 

    81

        return None

    82

    83

def getdate(v):

    84

    result = IEC_DATE_MODEL.match(v.upper())

    85

    if result is not None:

    86

        year, month, day = result.groups()

    87

        try:

    88

            date = datetime.datetime(int(year), int(month), int(day))

    89

        except ValueError, e:

    90

            return None

    91

        base_date = datetime.datetime(1970, 1, 1)

    92

        return date - base_date

    93

    else: 

    94

        return None

    95

    96

def getdatetime(v):

    97

    result = IEC_DATETIME_MODEL.match(v.upper())

    98

    if result is not None:

    99

        year, month, day, hours, minutes, seconds = result.groups()

   100

        try:

   101

            date = datetime.datetime(int(year), int(month), int(day), int(hours), int(minutes), int(float(seconds)), int((float(second) * SECOND) % SECOND))

   102

        except ValueError, e:

   103

            return None

   104

        base_date = datetime.datetime(1970, 1, 1)

   105

        return date - base_date

   106

    else: 

   107

        return None

   108

   109

def gettimeofday(v):

   110

    result = IEC_TIMEOFDAY_MODEL.match(v.upper())

   111

    if result is not None:

   112

        hours, minutes, seconds = result.groups()

   113

        microseconds = 0

   114

        for value, factor in [(hours, HOUR),

   115

                              (minutes, MINUTE),

   116

                              (seconds, SECOND)]:

   117

            microseconds += float(value) * factor

   118

        return datetime.timedelta(microseconds=microseconds)

   119

    else:

   120

        return None

   121

   122

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

   123

                "SINT": getinteger,

   124

                "INT": getinteger,

   125

                "DINT": getinteger,

   126

                "LINT": getinteger,

   127

                "USINT": getinteger,

   128

                "UINT": getinteger,

   129

                "UDINT": getinteger,

   130

                "ULINT": getinteger,

   131

                "BYTE": getinteger,

   132

                "WORD": getinteger,

   133

                "DWORD": getinteger,

   134

                "LWORD": getinteger,

   135

                "REAL": getfloat,

   136

                "LREAL": getfloat,

   137

                "STRING": getstring,

   138

                "WSTRING": getstring,

   139

                "TIME": gettime,

   140

                "DATE": getdate,

   141

                "DT": getdatetime,

   142

                "TOD": gettimeofday}

   143

   144

   145

class ForceVariableDialog(wx.TextEntryDialog):

   146

   147

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

   148

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

   149

            if id is not None:

   150

                event(self, id, function)

   151

            else:

   152

                event(self, function)

   153

   154

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

   155

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

   156

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

   157

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

   158

   159

        self.IEC_Type = iec_type 

   160

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

   161

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

   162

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

   163

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

   164

        else:

   165

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

   166

   167

    def OnOK(self, event):

   168

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

   169

        if value == "":

   170

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

   171

            message.ShowModal()

   172

            message.Destroy()

   173

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

   174

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

   175

            message.ShowModal()

   176

            message.Destroy()

   177

        else:

   178

            self.EndModal(wx.ID_OK)

   179

   180

    def GetValue(self):

   181

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