1

#!/usr/bin/env python

     2

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

     3

     4

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

     5

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

     6

#

     7

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

     8

# Copyright (C) 2017: Andrey Skvortsov <andrej.skvortzov@gmail.com>

     9

#

    10

# See COPYING file for copyrights details.

    11

#

    12

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

    13

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

    14

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

    15

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

    16

#

    17

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

    18

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

    19

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    20

# GNU General Public License for more details.

    21

#

    22

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

    23

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

    24

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

    25

    26

    27

from __future__ import absolute_import

    28

import re

    29

    30

import wx

    31

    32

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

    33

#                                  Helpers

    34

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

    35

    36

MICROSECONDS = 0.001

    37

MILLISECONDS = 1

    38

SECOND = 1000

    39

MINUTE = 60 * SECOND

    40

HOUR = 60 * MINUTE

    41

DAY = 24 * HOUR

    42

    43

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

    44

    45

    46

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

    47

#                         Edit Duration Value Dialog

    48

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

    49

    50

    51

class DurationEditorDialog(wx.Dialog):

    52

    53

    def __init__(self, parent):

    54

        wx.Dialog.__init__(self, parent, title=_('Edit Duration'))

    55

    56

        CONTROLS = [

    57

            ("Days", _('Days:')),

    58

            ("Hours", _('Hours:')),

    59

            ("Minutes", _('Minutes:')),

    60

            ("Seconds", _('Seconds:')),

    61

            ("Milliseconds", _('Milliseconds:')),

    62

            ("Microseconds", _('Microseconds:')),

    63

        ]

    64

    65

        main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)

    66

        main_sizer.AddGrowableCol(0)

    67

        main_sizer.AddGrowableRow(0)

    68

    69

        controls_sizer = wx.FlexGridSizer(cols=len(CONTROLS), hgap=10, rows=2, vgap=10)

    70

        main_sizer.AddSizer(controls_sizer, border=20,

    71

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

    72

    73

        controls = []

    74

        for i, (name, label) in enumerate(CONTROLS):

    75

            controls_sizer.AddGrowableCol(i)

    76

    77

            st = wx.StaticText(self, label=label)

    78

            txtctrl = wx.TextCtrl(self, value='0', style=wx.TE_PROCESS_ENTER)

    79

            self.Bind(wx.EVT_TEXT_ENTER,

    80

                      self.GetControlValueTestFunction(txtctrl),

    81

                      txtctrl)

    82

            setattr(self, name, txtctrl)

    83

    84

            controls.append((st, txtctrl))

    85

    86

        for st, txtctrl in controls:

    87

            controls_sizer.AddWindow(st, flag=wx.GROW)

    88

    89

        for st, txtctrl in controls:

    90

            controls_sizer.AddWindow(txtctrl, flag=wx.GROW)

    91

    92

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

    93

        self.Bind(wx.EVT_BUTTON, self.OnOK, button_sizer.GetAffirmativeButton())

    94

        main_sizer.AddSizer(button_sizer, border=20,

    95

                            flag=wx.ALIGN_RIGHT | wx.BOTTOM | wx.LEFT | wx.RIGHT)

    96

    97

        self.SetSizer(main_sizer)

    98

        self.Fit()

    99

        self.Days.SetFocus()

   100

   101

    def SetDuration(self, value):

   102

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

   103

        if result is not None:

   104

            values = result.groups()

   105

            for control, index in [(self.Days, 1), (self.Hours, 2),

   106

                                   (self.Minutes, 3), (self.Seconds, 4)]:

   107

                value = values[index]

   108

                if value is not None:

   109

                    control.SetValue(value)

   110

                else:

   111

                    control.SetValue("0")

   112

            milliseconds = values[5]

   113

            if milliseconds is not None:

   114

                self.Milliseconds.SetValue("%d" % int(float(milliseconds)))

   115

                self.Microseconds.SetValue("%.3f" % ((float(milliseconds) % MILLISECONDS) / MICROSECONDS))

   116

            else:

   117

                self.Milliseconds.SetValue("0")

   118

                self.Microseconds.SetValue("0")

   119

   120

    def GetControlValueTestFunction(self, control):

   121

        def OnValueChanged(event):

   122

            try:

   123

                float(control.GetValue())

   124

            except ValueError:

   125

                message = wx.MessageDialog(self, _("Invalid value!\nYou must fill a numeric value."), _("Error"), wx.OK | wx.ICON_ERROR)

   126

                message.ShowModal()

   127

                message.Destroy()

   128

            event.Skip()

   129

            self.OnCloseDialog()

   130

        return OnValueChanged

   131

   132

    def GetDuration(self):

   133

        milliseconds = 0

   134

        for control, factor in [(self.Days, DAY), (self.Hours, HOUR),

   135

                                (self.Minutes, MINUTE), (self.Seconds, SECOND),

   136

                                (self.Milliseconds, MILLISECONDS), (self.Microseconds, MICROSECONDS)]:

   137

   138

            milliseconds += float(control.GetValue()) * factor

   139

   140

        not_null = False

   141

        duration = "T#"

   142

        for value, format in [((int(milliseconds) / DAY),             "%dd"),

   143

                              ((int(milliseconds) % DAY) / HOUR,      "%dh"),

   144

                              ((int(milliseconds) % HOUR) / MINUTE,   "%dm"),

   145

                              ((int(milliseconds) % MINUTE) / SECOND, "%ds")]:

   146

   147

            if value > 0 or not_null:

   148

                duration += format % value

   149

                not_null = True

   150

   151

        duration += ("%f" % (milliseconds % SECOND)).rstrip("0").rstrip(".") + "ms"

   152

        return duration

   153

   154

    def OnOK(self, event):

   155

        self.OnCloseDialog()

   156

   157

    def OnCloseDialog(self):

   158

        errors = []

   159

        for control, name in [(self.Days, _("days")), (self.Hours, _("hours")),

   160

                              (self.Minutes, _("minutes")), (self.Seconds, _("seconds")),

   161

                              (self.Milliseconds, _("milliseconds")),

   162

                              (self.Microseconds, _("microseconds"))]:

   163

            try:

   164

                float(control.GetValue())

   165

            except ValueError:

   166

                errors.append(name)

   167

        if len(errors) > 0:

   168

            if len(errors) == 1:

   169

                message = _("Field %s hasn't a valid value!") % errors[0]

   170

            else:

   171

                message = _("Fields %s haven't a valid value!") % ",".join(errors)

   172

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

   173

            dialog.ShowModal()

   174

            dialog.Destroy()

   175

        else:

   176

            self.EndModal(wx.ID_OK)