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

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

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

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

    12

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

    13

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

    14

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

    15

#

    16

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

    17

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

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

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

    22

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

    23

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

    24

    25

    26

from __future__ import absolute_import

    27

import wx

    28

import wx.grid

    29

import wx.lib.buttons

    30

    31

from controls import CustomGrid, CustomTable

    32

from plcopen.BlockInstanceCollector import _ActionInfos

    33

from util.BitmapLibrary import GetBitmap

    34

from util.TranslationCatalogs import NoTranslate

    35

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

    36

#                                  Helpers

    37

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

    38

    39

    40

def GetActionTableColnames():

    41

    _ = NoTranslate

    42

    return [_("Qualifier"), _("Duration"), _("Type"), _("Value"), _("Indicator")]

    43

    44

    45

def GetTypeList():

    46

    _ = NoTranslate

    47

    return [_("Action"), _("Variable"), _("Inline")]

    48

    49

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

    50

#                               Action Table

    51

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

    52

    53

    54

class ActionTable(CustomTable):

    55

    56

    def GetValue(self, row, col):

    57

        if row < self.GetNumberRows():

    58

            colname = self.GetColLabelValue(col, False)

    59

            value = getattr(self.data[row], colname.lower())

    60

            if colname == "Type":

    61

                return _(value)

    62

            return value

    63

    64

    def SetValue(self, row, col, value):

    65

        if col < len(self.colnames):

    66

            colname = self.GetColLabelValue(col, False)

    67

            if colname == "Type":

    68

                value = self.Parent.TranslateType[value]

    69

            elif colname == "Qualifier" and not self.Parent.DurationList[value]:

    70

                self.data[row].duration = ""

    71

            setattr(self.data[row], colname.lower(), value)

    72

    73

    def _updateColAttrs(self, grid):

    74

        """

    75

        wx.Grid -> update the column attributes to add the

    76

        appropriate renderer given the column name.

    77

    78

        Otherwise default to the default renderer.

    79

        """

    80

    81

        for row in range(self.GetNumberRows()):

    82

            for col in range(self.GetNumberCols()):

    83

                editor = None

    84

                renderer = None

    85

                readonly = False

    86

                colname = self.GetColLabelValue(col, False)

    87

                if colname == "Qualifier":

    88

                    editor = wx.grid.GridCellChoiceEditor()

    89

                    editor.SetParameters(self.Parent.QualifierList)

    90

                if colname == "Duration":

    91

                    editor = wx.grid.GridCellTextEditor()

    92

                    renderer = wx.grid.GridCellStringRenderer()

    93

                    readonly = not self.Parent.DurationList[self.data[row].qualifier]

    94

                elif colname == "Type":

    95

                    editor = wx.grid.GridCellChoiceEditor()

    96

                    editor.SetParameters(self.Parent.TypeList)

    97

                elif colname == "Value":

    98

                    value_type = self.data[row].type

    99

                    if value_type == "Action":

   100

                        editor = wx.grid.GridCellChoiceEditor()

   101

                        editor.SetParameters(self.Parent.ActionList)

   102

                    elif value_type == "Variable":

   103

                        editor = wx.grid.GridCellChoiceEditor()

   104

                        editor.SetParameters(self.Parent.VariableList)

   105

                    elif value_type == "Inline":

   106

                        editor = wx.grid.GridCellTextEditor()

   107

                        renderer = wx.grid.GridCellStringRenderer()

   108

                elif colname == "Indicator":

   109

                    editor = wx.grid.GridCellChoiceEditor()

   110

                    editor.SetParameters(self.Parent.VariableList)

   111

   112

                grid.SetCellEditor(row, col, editor)

   113

                grid.SetCellRenderer(row, col, renderer)

   114

                grid.SetReadOnly(row, col, readonly)

   115

   116

                grid.SetCellBackgroundColour(row, col, wx.WHITE)

   117

            self.ResizeRow(grid, row)

   118

   119

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

   120

#                            Action Block Dialog

   121

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

   122

   123

   124

class ActionBlockDialog(wx.Dialog):

   125

   126

    def __init__(self, parent):

   127

        wx.Dialog.__init__(self, parent, title=_('Edit action block properties'))

   128

   129

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

   130

        main_sizer.AddGrowableCol(0)

   131

        main_sizer.AddGrowableRow(1)

   132

   133

        top_sizer = wx.FlexGridSizer(cols=5, hgap=5, rows=1, vgap=0)

   134

        top_sizer.AddGrowableCol(0)

   135

        top_sizer.AddGrowableRow(0)

   136

        main_sizer.AddSizer(top_sizer, border=20,

   137

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

   138

   139

        actions_label = wx.StaticText(self, label=_('Actions:'))

   140

        top_sizer.AddWindow(actions_label, flag=wx.ALIGN_BOTTOM)

   141

   142

        for name, bitmap, help in [

   143

                ("AddButton", "add_element", _("Add action")),

   144

                ("DeleteButton", "remove_element", _("Remove action")),

   145

                ("UpButton", "up", _("Move action up")),

   146

                ("DownButton", "down", _("Move action down"))]:

   147

            button = wx.lib.buttons.GenBitmapButton(

   148

                self, bitmap=GetBitmap(bitmap),

   149

                size=wx.Size(28, 28), style=wx.NO_BORDER)

   150

            button.SetToolTipString(help)

   151

            setattr(self, name, button)

   152

            top_sizer.AddWindow(button)

   153

   154

        self.ActionsGrid = CustomGrid(self, size=wx.Size(-1, 250), style=wx.VSCROLL)

   155

        self.ActionsGrid.DisableDragGridSize()

   156

        self.ActionsGrid.EnableScrolling(False, True)

   157

        self.ActionsGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE,

   158

                              self.OnActionsGridCellChange)

   159

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

   160

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

   161

   162

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

   163

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

   164

        main_sizer.AddSizer(button_sizer, border=20,

   165

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

   166

   167

        self.SetSizer(main_sizer)

   168

   169

        self.Table = ActionTable(self, [], GetActionTableColnames())

   170

        typelist = GetTypeList()

   171

        self.TypeList = ",".join(map(_, typelist))

   172

        self.TranslateType = dict([(_(value), value) for value in typelist])

   173

        self.ColSizes = [60, 90, 130, 200, 50]

   174

        self.ColAlignements = [wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT]

   175

   176

        self.ActionsGrid.SetTable(self.Table)

   177

        self.ActionsGrid.SetDefaultValue(_ActionInfos("N", "Action", "", "", ""))

   178

        self.ActionsGrid.SetButtons({"Add": self.AddButton,

   179

                                     "Delete": self.DeleteButton,

   180

                                     "Up": self.UpButton,

   181

                                     "Down": self.DownButton})

   182

        self.ActionsGrid.SetRowLabelSize(0)

   183

   184

        for col in range(self.Table.GetNumberCols()):

   185

            attr = wx.grid.GridCellAttr()

   186

            attr.SetAlignment(self.ColAlignements[col], wx.ALIGN_CENTRE)

   187

            self.ActionsGrid.SetColAttr(col, attr)

   188

            self.ActionsGrid.SetColMinimalWidth(col, self.ColSizes[col])

   189

            self.ActionsGrid.AutoSizeColumn(col, False)

   190

   191

        self.Table.ResetView(self.ActionsGrid)

   192

        self.ActionsGrid.SetFocus()

   193

        self.ActionsGrid.RefreshButtons()

   194

        self.Fit()

   195

   196

    def OnOK(self, event):

   197

        self.ActionsGrid.CloseEditControl()

   198

        self.EndModal(wx.ID_OK)

   199

   200

    def OnActionsGridCellChange(self, event):

   201

        wx.CallAfter(self.Table.ResetView, self.ActionsGrid)

   202

        event.Skip()

   203

   204

    def SetQualifierList(self, list):

   205

        self.QualifierList = ",".join(list)

   206

        self.DurationList = list

   207

   208

    def SetVariableList(self, list):

   209

        self.VariableList = "," + ",".join([variable.Name for variable in list])

   210

   211

    def SetActionList(self, list):

   212

        self.ActionList = "," + ",".join(list)

   213

   214

    def SetValues(self, actions):

   215

        for action in actions:

   216

            row = action.copy()

   217

            if row.type == "reference" and row.value in self.ActionList:

   218

                row.type = "Action"

   219

            elif row.type == "reference" and row.value in self.VariableList:

   220

                row.type = "Variable"

   221

            else:

   222

                row.type = "Inline"

   223

            self.Table.AppendRow(row)

   224

        self.Table.ResetView(self.ActionsGrid)

   225

        if len(actions) > 0:

   226

            self.ActionsGrid.SetGridCursor(0, 0)

   227

        self.ActionsGrid.RefreshButtons()

   228

   229

    def GetValues(self):

   230

        actions = self.Table.GetData()

   231

        for action in actions:

   232

            if action.type in ["Action", "Variable"]:

   233

                action.type = "reference"

   234

            else:

   235

                action.type = "inline"

   236

        return actions