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

from types import TupleType

    28

    29

import wx

    30

    31

from controls import CustomEditableListBox

    32

    33

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

    34

#                                  Helpers

    35

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

    36

    37

DIMENSION_MODEL = re.compile("([0-9]+)\.\.([0-9]+)$")

    38

    39

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

    40

#                             Array Type Dialog

    41

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

    42

    43

    44

class ArrayTypeDialog(wx.Dialog):

    45

    46

    def __init__(self, parent, datatypes, infos):

    47

        wx.Dialog.__init__(self, parent, title=_('Edit array type properties'))

    48

    49

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

    50

        main_sizer.AddGrowableCol(0)

    51

        main_sizer.AddGrowableRow(1)

    52

    53

        top_sizer = wx.BoxSizer(wx.HORIZONTAL)

    54

        main_sizer.AddSizer(top_sizer, border=20,

    55

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

    56

    57

        basetype_label = wx.StaticText(self, label=_('Base Type:'))

    58

        top_sizer.AddWindow(basetype_label, 1, flag=wx.ALIGN_BOTTOM)

    59

    60

        self.BaseType = wx.ComboBox(self, style=wx.CB_READONLY)

    61

        top_sizer.AddWindow(self.BaseType, 1, flag=wx.GROW)

    62

    63

        self.Dimensions = CustomEditableListBox(self, label=_("Dimensions:"),

    64

                                                style=(wx.gizmos.EL_ALLOW_NEW |

    65

                                                       wx.gizmos.EL_ALLOW_EDIT |

    66

                                                       wx.gizmos.EL_ALLOW_DELETE))

    67

        for func in ["_OnLabelEndEdit",

    68

                     "_OnAddButton",

    69

                     "_OnDelButton",

    70

                     "_OnUpButton",

    71

                     "_OnDownButton"]:

    72

            setattr(self.Dimensions, func, self.OnDimensionsChanged)

    73

        main_sizer.AddSizer(self.Dimensions, border=20,

    74

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

    75

    76

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

    77

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

    78

        main_sizer.AddSizer(button_sizer, border=20,

    79

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

    80

    81

        self.SetSizer(main_sizer)

    82

    83

        for datatype in datatypes:

    84

            self.BaseType.Append(datatype)

    85

    86

        if isinstance(infos, TupleType) and infos[0] == "array":

    87

            self.BaseType.SetStringSelection(infos[1])

    88

            self.Dimensions.SetStrings(map("..".join, infos[2]))

    89

        elif infos in datatypes:

    90

            self.BaseType.SetStringSelection(infos)

    91

    92

        self.BaseType.SetFocus()

    93

        self.Fit()

    94

    95

    def GetDimensions(self):

    96

        message = None

    97

        dimensions_list = []

    98

        dimension_strings = self.Dimensions.GetStrings()

    99

        if len(dimension_strings) == 0:

   100

            message = _("Empty dimension isn't allowed.")

   101

   102

        for dimensions in dimension_strings:

   103

            result = DIMENSION_MODEL.match(dimensions)

   104

            if result is None:

   105

                message = _("\"%s\" value isn't a valid array dimension!") % dimensions

   106

                break

   107

            bounds = result.groups()

   108

            if int(bounds[0]) >= int(bounds[1]):

   109

                message = _("\"%s\" value isn't a valid array dimension!\nRight value must be greater than left value.") % dimensions

   110

                break

   111

            dimensions_list.append(bounds)

   112

   113

        if message is not None:

   114

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

   115

            dlg.ShowModal()

   116

            dlg.Destroy()

   117

            return None

   118

        return dimensions_list

   119

   120

    def OnDimensionsChanged(self, event):

   121

        wx.CallAfter(self.GetDimensions)

   122

        event.Skip()

   123

   124

    def OnOK(self, event):

   125

        if self.GetDimensions() is not None:

   126

            self.EndModal(wx.ID_OK)

   127

   128

    def GetValue(self):

   129

        return "array", self.BaseType.GetStringSelection(), self.GetDimensions()