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

CONTROLS = [

    46

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

    47

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

    48

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

    49

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

    50

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

    51

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

    52

]

    53

    54

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

    55

#                         Edit Duration Value Dialog

    56

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

    57

    58

    59

class DurationEditorDialog(wx.Dialog):

    60

    61

    def __init__(self, parent):

    62

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

    63

    64

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

    65

        main_sizer.AddGrowableCol(0)

    66

        main_sizer.AddGrowableRow(0)

    67

    68

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

    69

        main_sizer.AddSizer(controls_sizer, border=20,

    70

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

    71

    72

        controls = []

    73

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

    74

            controls_sizer.AddGrowableCol(i)

    75

    76

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

    77

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

    78

            self.Bind(wx.EVT_TEXT_ENTER,

    79

                      self.GetControlValueTestFunction(txtctrl),

    80

                      txtctrl)

    81

            setattr(self, name, txtctrl)

    82

    83

            controls.append((st, txtctrl))

    84

    85

        for st, txtctrl in controls:

    86

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

    87

    88

        for st, txtctrl in controls:

    89

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

    90

    91

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

    92

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

    93

        main_sizer.AddSizer(button_sizer, border=20,

    94

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

    95

    96

        self.SetSizer(main_sizer)

    97

        self.Fit()

    98

        self.Days.SetFocus()

    99

   100

    def SetDuration(self, value):

   101

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

   102

        if result is not None:

   103

            values = result.groups()

   104

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

   105

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

   106

                value = values[index]

   107

                if value is not None:

   108

                    control.SetValue(value)

   109

                else:

   110

                    control.SetValue("0")

   111

            milliseconds = values[5]

   112

            if milliseconds is not None:

   113

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

   114

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

   115

            else:

   116

                self.Milliseconds.SetValue("0")

   117

                self.Microseconds.SetValue("0")

   118

   119

    def GetControlValueTestFunction(self, control):

   120

        def OnValueChanged(event):

   121

            try:

   122

                float(control.GetValue())

   123

            except ValueError:

   124

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

   125

                message.ShowModal()

   126

                message.Destroy()

   127

            event.Skip()

   128

            self.OnCloseDialog()

   129

        return OnValueChanged

   130

   131

    def GetDuration(self):

   132

        milliseconds = 0

   133

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

   134

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

   135

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

   136

   137

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

   138

   139

        not_null = False

   140

        duration = "T#"

   141

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

   142

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

   143

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

   144

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

   145

   146

            if value > 0 or not_null:

   147

                duration += format % value

   148

                not_null = True

   149

   150

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

   151

        return duration

   152

   153

    def OnOK(self, event):

   154

        self.OnCloseDialog()

   155

   156

    def OnCloseDialog(self):

   157

        errors = []

   158

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

   159

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

   160

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

   161

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

   162

            try:

   163

                float(control.GetValue())

   164

            except ValueError:

   165

                errors.append(name)

   166

        if len(errors) > 0:

   167

            if len(errors) == 1:

   168

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

   169

            else:

   170

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

   171

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

   172

            dialog.ShowModal()

   173

            dialog.Destroy()

   174

        else:

   175

            self.EndModal(wx.ID_OK)