1

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

     2

     3

# This file is part of Beremiz, a Integrated Development Environment for

     4

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     5

#

     6

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

     7

#

     8

# See COPYING file for copyrights details.

     9

#

    10

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

    11

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

    12

# as published by the Free Software Foundation; either version 2

    13

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

    14

#

    15

# This program 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

    18

# GNU General Public License for more details.

    19

#

    20

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

    21

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

    22

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    23

    24

    25

from __future__ import absolute_import

    26

import re

    27

import datetime

    28

    29

import wx

    30

    31

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

    32

#                                Helpers

    33

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

    34

    35

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

    36

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

    37

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

    38

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

    39

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

    40

    41

    42

def gen_get_function(f):

    43

    def get_function(v):

    44

        try:

    45

            return f(v)

    46

        except Exception:

    47

            return None

    48

    return get_function

    49

    50

    51

def gen_get_string(delimiter):

    52

    STRING_MODEL = re.compile("%(delimiter)s([^%(delimiter)s]*)%(delimiter)s$" % {"delimiter": delimiter})

    53

    54

    def get_string(v):

    55

        result = STRING_MODEL.match(v)

    56

        if result is not None:

    57

            return result.group(1)

    58

        return None

    59

    return get_string

    60

    61

    62

getinteger = gen_get_function(int)

    63

getfloat = gen_get_function(float)

    64

getstring = gen_get_string("'")

    65

getwstring = gen_get_string('"')

    66

    67

SECOND = 1000000

    68

MINUTE = 60 * SECOND

    69

HOUR = 60 * MINUTE

    70

DAY = 24 * HOUR

    71

    72

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]+)?"})

    73

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

    74

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]+)?)$")

    75

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

    76

    77

    78

def gettime(v):

    79

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

    80

    if result is not None:

    81

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

    82

        microseconds = 0

    83

        not_null = False

    84

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

    85

                              (hours, HOUR),

    86

                              (minutes, MINUTE),

    87

                              (seconds, SECOND),

    88

                              (milliseconds, 1000)]:

    89

            if value is not None:

    90

                microseconds += float(value) * factor

    91

                not_null = True

    92

        if not not_null:

    93

            return None

    94

        if negative is not None:

    95

            microseconds = -microseconds

    96

        return datetime.timedelta(microseconds=microseconds)

    97

    98

    else:

    99

        return None

   100

   101

   102

def getdate(v):

   103

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

   104

    if result is not None:

   105

        year, month, day = result.groups()

   106

        try:

   107

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

   108

        except ValueError:

   109

            return None

   110

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

   111

        return date - base_date

   112

    else:

   113

        return None

   114

   115

   116

def getdatetime(v):

   117

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

   118

    if result is not None:

   119

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

   120

        try:

   121

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

   122

        except ValueError:

   123

            return None

   124

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

   125

        return date - base_date

   126

    else:

   127

        return None

   128

   129

   130

def gettimeofday(v):

   131

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

   132

    if result is not None:

   133

        hours, minutes, seconds = result.groups()

   134

        microseconds = 0

   135

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

   136

                              (minutes, MINUTE),

   137

                              (seconds, SECOND)]:

   138

            microseconds += float(value) * factor

   139

        return datetime.timedelta(microseconds=microseconds)

   140

    else:

   141

        return None

   142

   143

   144

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

   145

                "SINT": getinteger,

   146

                "INT": getinteger,

   147

                "DINT": getinteger,

   148

                "LINT": getinteger,

   149

                "USINT": getinteger,

   150

                "UINT": getinteger,

   151

                "UDINT": getinteger,

   152

                "ULINT": getinteger,

   153

                "BYTE": getinteger,

   154

                "WORD": getinteger,

   155

                "DWORD": getinteger,

   156

                "LWORD": getinteger,

   157

                "REAL": getfloat,

   158

                "LREAL": getfloat,

   159

                "STRING": getstring,

   160

                "WSTRING": getwstring,

   161

                "TIME": gettime,

   162

                "DATE": getdate,

   163

                "DT": getdatetime,

   164

                "TOD": gettimeofday}

   165

   166

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

   167

#                            Force Variable Dialog

   168

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

   169

   170

   171

class ForceVariableDialog(wx.TextEntryDialog):

   172

   173

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

   174

        wx.TextEntryDialog.__init__(

   175

            self, parent,

   176

            message=_("Forcing Variable Value"),

   177

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

   178

            defaultValue=defaultValue,

   179

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

   180

   181

        self.IEC_Type = iec_type

   182

   183

        self.Bind(wx.EVT_BUTTON, self.OnOK,

   184

                  self.GetSizer().GetItem(2).GetSizer().GetItem(1).

   185

                  GetSizer().GetAffirmativeButton())

   186

        self.ValueTextCtrl = self.GetSizer().GetItem(1).GetWindow()

   187

        if self.IEC_Type == "BOOL":

   188

            self.ToggleButton = wx.ToggleButton(self, label=_("Toggle value"))

   189

            value = GetTypeValue[self.IEC_Type](defaultValue)

   190

            if value is not None:

   191

                self.ToggleButton.SetValue(value)

   192

   193

            border = self.GetSizer().GetItem(1).GetBorder()

   194

            self.GetSizer().Insert(before=2, item=self.ToggleButton,

   195

                                   border=border,

   196

                                   flag=wx.LEFT | wx.RIGHT | wx.EXPAND)

   197

            self.Bind(wx.EVT_TOGGLEBUTTON, self.ToggleBoolValue, self.ToggleButton)

   198

   199

        self.Fit()

   200

   201

    def ToggleBoolValue(self, event):

   202

        value = self.ToggleButton.GetValue()

   203

        self.ValueTextCtrl.SetValue(unicode(value))

   204

   205

    def OnOK(self, event):

   206

        message = None

   207

        value = self.ValueTextCtrl.GetValue()

   208

        if value == "":

   209

            message = _("You must type a value!")

   210

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

   211

            message = _("Invalid value \"{a1}\" for \"{a2}\" variable!").format(a1=value, a2=self.IEC_Type)

   212

        if message is not None:

   213

            dialog = wx.MessageDialog(self, message, _("Error"), wx.OK | wx.ICON_ERROR)

   214

            dialog.ShowModal()

   215

            dialog.Destroy()

   216

        else:

   217

            self.EndModal(wx.ID_OK)

   218

        event.Skip()

   219

   220

    def GetValue(self):

   221

        return GetTypeValue[self.IEC_Type](wx.TextEntryDialog.GetValue(self))