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

    28

    29

import re

    30

    31

import wx

    32

    33

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

    34

#                                  Helpers

    35

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

    36

    37

MICROSECONDS = 0.001

    38

MILLISECONDS = 1

    39

SECOND = 1000

    40

MINUTE = 60 * SECOND

    41

HOUR = 60 * MINUTE

    42

DAY = 24 * HOUR

    43

    44

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

    45

    46

    47

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

    48

#                         Edit Duration Value Dialog

    49

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

    50

    51

    52

class DurationEditorDialog(wx.Dialog):

    53

    54

    def __init__(self, parent):

    55

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

    56

    57

        CONTROLS = [

    58

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

    59

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

    60

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

    61

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

    62

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

    63

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

    64

        ]

    65

    66

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

    67

        main_sizer.AddGrowableCol(0)

    68

        main_sizer.AddGrowableRow(0)

    69

    70

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

    71

        main_sizer.Add(controls_sizer, border=20,

    72

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

    73

    74

        controls = []

    75

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

    76

            controls_sizer.AddGrowableCol(i)

    77

    78

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

    79

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

    80

            self.Bind(wx.EVT_TEXT_ENTER,

    81

                      self.GetControlValueTestFunction(txtctrl),

    82

                      txtctrl)

    83

            setattr(self, name, txtctrl)

    84

    85

            controls.append((st, txtctrl))

    86

    87

        for st, txtctrl in controls:

    88

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

    89

    90

        for st, txtctrl in controls:

    91

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

    92

    93

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

    94

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

    95

        main_sizer.Add(button_sizer, border=20,

    96

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

    97

    98

        self.SetSizer(main_sizer)

    99

        self.Fit()

   100

        self.Days.SetFocus()

   101

   102

    def SetDuration(self, value):

   103

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

   104

        if result is not None:

   105

            values = result.groups()

   106

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

   107

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

   108

                value = values[index]

   109

                if value is not None:

   110

                    control.SetValue(value)

   111

                else:

   112

                    control.SetValue("0")

   113

            milliseconds = values[5]

   114

            if milliseconds is not None:

   115

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

   116

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

   117

            else:

   118

                self.Milliseconds.SetValue("0")

   119

                self.Microseconds.SetValue("0")

   120

   121

    def GetControlValueTestFunction(self, control):

   122

        def OnValueChanged(event):

   123

            try:

   124

                float(control.GetValue())

   125

            except ValueError:

   126

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

   127

                message.ShowModal()

   128

                message.Destroy()

   129

            event.Skip()

   130

            self.OnCloseDialog()

   131

        return OnValueChanged

   132

   133

    def GetDuration(self):

   134

        milliseconds = 0

   135

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

   136

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

   137

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

   138

   139

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

   140

   141

        not_null = False

   142

        duration = "T#"

   143

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

   144

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

   145

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

   146

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

   147

   148

            if value > 0 or not_null:

   149

                duration += format % value

   150

                not_null = True

   151

   152

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

   153

        return duration

   154

   155

    def OnOK(self, event):

   156

        event.Skip()

   157

        self.OnCloseDialog()

   158

   159

    def OnCloseDialog(self):

   160

        errors = []

   161

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

   162

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

   163

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

   164

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

   165

            try:

   166

                float(control.GetValue())

   167

            except ValueError:

   168

                errors.append(name)

   169

        if len(errors) > 0:

   170

            if len(errors) == 1:

   171

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

   172

            else:

   173

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

   174

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

   175

            dialog.ShowModal()

   176

            dialog.Destroy()

   177

        else:

   178

            self.EndModal(wx.ID_OK)