HGKeeper

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

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

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

# See COPYING file for copyrights details.

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

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

# This program 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 program; if not, write to the Free Software

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

from __future__ import absolute_import

import re

import datetime

from builtins import str as text

import wx

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

# Helpers

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

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

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

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

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

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

def gen_get_function(f):

def get_function(v):

try:

return f(v)

except Exception:

return None

return get_function

def gen_get_string(delimiter):

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

def get_string(v):

result = STRING_MODEL.match(v)

if result is not None:

return result.group(1)

return None

return get_string

getinteger = gen_get_function(int)

getfloat = gen_get_function(float)

getstring = gen_get_string("'")

getwstring = gen_get_string('"')

SECOND = 1000000

MINUTE = 60 * SECOND

HOUR = 60 * MINUTE

DAY = 24 * HOUR

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

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

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

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

def gettime(v):

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

if result is not None:

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

microseconds = 0

not_null = False

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

(hours, HOUR),

(minutes, MINUTE),

(seconds, SECOND),

(milliseconds, 1000)]:

if value is not None:

microseconds += float(value) * factor

not_null = True

if not not_null:

return None

if negative is not None:

microseconds = -microseconds

return datetime.timedelta(microseconds=microseconds)

else:

return None

def getdate(v):

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

if result is not None:

year, month, day = result.groups()

try:

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

except ValueError:

return None

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

return date - base_date

else:

return None

def getdatetime(v):

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

if result is not None:

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

try:

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

except ValueError:

return None

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

return date - base_date

else:

return None

def gettimeofday(v):

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

if result is not None:

hours, minutes, seconds = result.groups()

microseconds = 0

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

(minutes, MINUTE),

(seconds, SECOND)]:

microseconds += float(value) * factor

return datetime.timedelta(microseconds=microseconds)

else:

return None

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

"SINT": getinteger,

"INT": getinteger,

"DINT": getinteger,

"LINT": getinteger,

"USINT": getinteger,

"UINT": getinteger,

"UDINT": getinteger,

"ULINT": getinteger,

"BYTE": getinteger,

"WORD": getinteger,

"DWORD": getinteger,

"LWORD": getinteger,

"REAL": getfloat,

"LREAL": getfloat,

"STRING": getstring,

"WSTRING": getwstring,

"TIME": gettime,

"DATE": getdate,

"DT": getdatetime,

"TOD": gettimeofday}

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

# Force Variable Dialog

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

class ForceVariableDialog(wx.Dialog):

"""Dialog to enforce new value for variables in debug panel"""

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

"""

Constructor

@param parent: Parent wx.Window of dialog for modal

@param iec_type: IEC type of variable (string). For example 'BOOL', 'LREAL'.

@param defaultValue: current variable value as string. Default is empty string.

"""

wx.Dialog.__init__(

self, parent,

name='ForceVariableDialog',

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

style=wx.DEFAULT_DIALOG_STYLE, pos=wx.DefaultPosition)

self.IEC_Type = iec_type

info_sizer = wx.BoxSizer(wx.VERTICAL)

message_label = wx.StaticText(self, label=_("Forcing Variable Value"))

info_sizer.Add(message_label, border=10,

flag=wx.ALIGN_LEFT | wx.GROW | wx.TOP | wx.LEFT | wx.RIGHT)

if GetTypeValue[self.IEC_Type] in [getinteger, getfloat]:

self.InitCtrlNumber(info_sizer, defaultValue)

elif self.IEC_Type == "BOOL":

self.InitCtrlBool(info_sizer, defaultValue)

else:

self.InitCtrlDefault(info_sizer, defaultValue)

self.GetEnteredValue = self.GetValueDefault

button_sizer = self.CreateButtonSizer(wx.OK | wx.CANCEL | wx.CENTRE)

self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetAffirmativeId())

info_sizer.Add(button_sizer, border=10, flag=wx.ALIGN_RIGHT | wx.ALL)

self.SetSizer(info_sizer)

self.Fit()

self.ValueCtrl.SetFocus()

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

# default type methods

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

def InitCtrlDefault(self, info_sizer, defaultValue):

"""Add simple text control to change variable of any type"""

self.ValueCtrl = wx.TextCtrl(self)

self.ValueCtrl.SetValue(defaultValue)

info_sizer.Add(self.ValueCtrl, border=10, proportion=1,

flag=wx.ALIGN_LEFT | wx.GROW | wx.TOP | wx.LEFT | wx.RIGHT)

def GetValueDefault(self):

"""

Returns text representation for a variable value

@return: variable value as a string

"""

return text(self.ValueCtrl.GetValue())

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

# integer and floating point number type methods

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

def InitCtrlNumber(self, info_sizer, defaultValue):

"""Add controls to change float and integer variables"""

sizer = wx.BoxSizer(wx.HORIZONTAL)

self.InitCtrlDefault(sizer, defaultValue)

self.SpinButtonCtrl = wx.SpinButton(self, style=wx.HORIZONTAL | wx.SP_WRAP)

sizer.Add(self.SpinButtonCtrl, border=10,

self.Bind(wx.EVT_SPIN_UP, self.SpinButtonChanged)

self.Bind(wx.EVT_SPIN_DOWN, self.SpinButtonChanged)

info_sizer.Add(sizer, proportion=1, flag=wx.EXPAND)

def SpinButtonChanged(self, evt):

"""Increment/decrement variable value"""

value = self.GetValue()

if value is not None:

up = evt.GetEventType() == wx.EVT_SPIN_UP._getEvtType()

value = value + 1 if up else value - 1

self.ValueCtrl.SetValue(text(value))

evt.Skip()

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

# bool type related methods

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

def InitCtrlBool(self, info_sizer, defaultValue):

"""Add button to change value of boolean variable"""

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

value = GetTypeValue[self.IEC_Type](defaultValue)

if value is not None:

self.ValueCtrl.SetValue(value)

info_sizer.Add(self.ValueCtrl, border=10,

def OnOK(self, event):

"""

Checks new entered value

before closing dialog window

"""

message = None

ret = True

value = self.GetEnteredValue()

if value == "":

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

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

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

if message is not None:

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

dialog.ShowModal()

dialog.Destroy()

ret = False

else:

self.EndModal(wx.ID_OK)

event.Skip(ret)

def GetValue(self):

"""

Return new enforce value of particular type

"""

return GetTypeValue[self.IEC_Type](self.GetEnteredValue())

beremiz