1

import wx

     2

import wx.grid

     3

import wx.gizmos

     4

     5

from controls import CustomGrid, CustomTable

     6

from ConfTreeNodeEditor import ConfTreeNodeEditor, SCROLLBAR_UNIT

     7

     8

[ETHERCAT_VENDOR, ETHERCAT_GROUP, ETHERCAT_DEVICE] = range(3)

     9

    10

def AppendMenu(parent, help, id, kind, text):

    11

    if wx.VERSION >= (2, 6, 0):

    12

        parent.Append(help=help, id=id, kind=kind, text=text)

    13

    else:

    14

        parent.Append(helpString=help, id=id, kind=kind, item=text)

    15

    16

def GetSyncManagersTableColnames():

    17

    _ = lambda x : x

    18

    return ["#", _("Name"), _("Start Address"), _("Default Size"), _("Control Byte"), _("Enable")]

    19

    20

class SyncManagersTable(CustomTable):

    21

    22

    def GetValue(self, row, col):

    23

        if row < self.GetNumberRows():

    24

            if col == 0:

    25

                return row

    26

            return self.data[row].get(self.GetColLabelValue(col, False), "")

    27

    28

def GetVariablesTableColnames():

    29

    _ = lambda x : x

    30

    return ["#", _("Name"), _("Index"), _("SubIndex"), _("Type"), _("PDO index"), _("PDO name"), _("PDO type")]

    31

    32

[ID_NODEEDITOR, ID_NODEEDITORVENDORLABEL, 

    33

 ID_NODEEDITORVENDOR, ID_NODEEDITORPRODUCTCODELABEL, 

    34

 ID_NODEEDITORPRODUCTCODE, ID_NODEEDITORREVISIONNUMBERLABEL, 

    35

 ID_NODEEDITORREVISIONNUMBER, ID_NODEEDITORPHYSICSLABEL, 

    36

 ID_NODEEDITORPHYSICS, ID_NODEEDITORSYNCMANAGERSLABEL, 

    37

 ID_NODEEDITORSYNCMANAGERSGRID, ID_NODEEDITORVARIABLESLABEL, 

    38

 ID_NODEEDITORVARIABLESGRID, 

    39

] = [wx.NewId() for _init_ctrls in range(13)]

    40

    41

class NodeEditor(ConfTreeNodeEditor):

    42

    43

    ID = ID_NODEEDITOR

    44

    45

    def _init_coll_MainSizer_Items(self, parent):

    46

        parent.AddSizer(self.SlaveInfosDetailsSizer, 0, border=5, flag=wx.TOP|wx.LEFT|wx.RIGHT|wx.GROW)

    47

        parent.AddWindow(self.SyncManagersLabel, 0, border=5, flag=wx.LEFT|wx.RIGHT|wx.GROW)

    48

        parent.AddWindow(self.SyncManagersGrid, 0, border=5, flag=wx.LEFT|wx.RIGHT|wx.GROW)

    49

        parent.AddWindow(self.VariablesLabel, 0, border=5, flag=wx.LEFT|wx.RIGHT|wx.GROW)

    50

        parent.AddWindow(self.VariablesGrid, 0, border=5, flag=wx.BOTTOM|wx.LEFT|wx.RIGHT|wx.GROW)

    51

    52

    def _init_coll_MainSizer_Growables(self, parent):

    53

        parent.AddGrowableCol(0)

    54

        parent.AddGrowableRow(2, 1)

    55

        parent.AddGrowableRow(4, 2)

    56

    57

    def _init_coll_SlaveInfosDetailsSizer_Items(self, parent):

    58

        parent.AddWindow(self.VendorLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL|wx.GROW)

    59

        parent.AddWindow(self.Vendor, 0, border=0, flag=wx.GROW)

    60

        parent.AddWindow(self.ProductCodeLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL|wx.GROW)

    61

        parent.AddWindow(self.ProductCode, 0, border=0, flag=wx.GROW)

    62

        parent.AddWindow(self.RevisionNumberLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL|wx.GROW)

    63

        parent.AddWindow(self.RevisionNumber, 0, border=0, flag=wx.GROW)

    64

        parent.AddWindow(self.PhysicsLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL|wx.GROW)

    65

        parent.AddWindow(self.Physics, 0, border=0, flag=wx.GROW)

    66

    67

    def _init_coll_SlaveInfosDetailsSizer_Growables(self, parent):

    68

        parent.AddGrowableCol(1)

    69

        parent.AddGrowableCol(3)

    70

    71

    def _init_sizers(self):

    72

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

    73

        self.SlaveInfosDetailsSizer = wx.FlexGridSizer(cols=4, hgap=5, rows=2, vgap=5)

    74

    75

        self._init_coll_MainSizer_Growables(self.MainSizer)

    76

        self._init_coll_MainSizer_Items(self.MainSizer)

    77

        self._init_coll_SlaveInfosDetailsSizer_Growables(self.SlaveInfosDetailsSizer)

    78

        self._init_coll_SlaveInfosDetailsSizer_Items(self.SlaveInfosDetailsSizer)

    79

    80

        self.ConfNodeEditor.SetSizer(self.MainSizer)

    81

    82

    def _init_ConfNodeEditor(self, prnt):

    83

        self.ConfNodeEditor = wx.ScrolledWindow(id=-1, name='SlavePanel', parent=prnt,

    84

              size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL|wx.SUNKEN_BORDER|wx.HSCROLL|wx.VSCROLL)

    85

        self.ConfNodeEditor.Bind(wx.EVT_SIZE, self.OnConfNodeEditorResize)

    86

    87

        self.VendorLabel = wx.StaticText(id=ID_NODEEDITORVENDORLABEL,

    88

              label=_('Vendor:'), name='VendorLabel', parent=self.ConfNodeEditor,

    89

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

    90

    91

        self.Vendor = wx.TextCtrl(id=ID_NODEEDITORVENDOR, value='',

    92

              name='Vendor', parent=self.ConfNodeEditor, pos=wx.Point(0, 0),

    93

              size=wx.Size(0, 24), style=wx.TE_READONLY)

    94

    95

        self.ProductCodeLabel = wx.StaticText(id=ID_NODEEDITORPRODUCTCODELABEL,

    96

              label=_('Product code:'), name='ProductCodeLabel', parent=self.ConfNodeEditor,

    97

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

    98

    99

        self.ProductCode = wx.TextCtrl(id=ID_NODEEDITORPRODUCTCODE, value='',

   100

              name='ProductCode', parent=self.ConfNodeEditor, pos=wx.Point(0, 0),

   101

              size=wx.Size(0, 24), style=wx.TE_READONLY)

   102

   103

        self.RevisionNumberLabel = wx.StaticText(id=ID_NODEEDITORREVISIONNUMBERLABEL,

   104

              label=_('Revision number:'), name='RevisionNumberLabel', parent=self.ConfNodeEditor,

   105

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   106

   107

        self.RevisionNumber = wx.TextCtrl(id=ID_NODEEDITORREVISIONNUMBER, value='',

   108

              name='RevisionNumber', parent=self.ConfNodeEditor, pos=wx.Point(0, 0),

   109

              size=wx.Size(0, 24), style=wx.TE_READONLY)

   110

   111

        self.PhysicsLabel = wx.StaticText(id=ID_NODEEDITORPHYSICSLABEL,

   112

              label=_('Physics:'), name='PhysicsLabel', parent=self.ConfNodeEditor,

   113

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   114

   115

        self.Physics = wx.TextCtrl(id=ID_NODEEDITORPHYSICS, value='',

   116

              name='Physics', parent=self.ConfNodeEditor, pos=wx.Point(0, 0),

   117

              size=wx.Size(0, 24), style=wx.TE_READONLY)

   118

   119

        self.SyncManagersLabel =  wx.StaticText(id=ID_NODEEDITORSYNCMANAGERSLABEL,

   120

              label=_('Sync managers:'), name='SyncManagersLabel', parent=self.ConfNodeEditor,

   121

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   122

   123

        self.SyncManagersGrid = CustomGrid(id=ID_NODEEDITORSYNCMANAGERSGRID,

   124

              name='SyncManagersGrid', parent=self.ConfNodeEditor, pos=wx.Point(0, 0), 

   125

              size=wx.Size(0, 200), style=wx.VSCROLL)

   126

   127

        self.VariablesLabel =  wx.StaticText(id=ID_NODEEDITORVARIABLESLABEL,

   128

              label=_('Variable entries:'), name='VariablesLabel', parent=self.ConfNodeEditor,

   129

              pos=wx.Point(0, 0), size=wx.DefaultSize, style=0)

   130

   131

        self.VariablesGrid = wx.gizmos.TreeListCtrl(id=ID_NODEEDITORVARIABLESGRID,

   132

              name='VariablesGrid', parent=self.ConfNodeEditor, pos=wx.Point(0, 0), 

   133

              size=wx.Size(0, 400), style=wx.TR_DEFAULT_STYLE |

   134

                                          wx.TR_ROW_LINES |

   135

                                          wx.TR_COLUMN_LINES |

   136

                                          wx.TR_HIDE_ROOT |

   137

                                          wx.TR_FULL_ROW_HIGHLIGHT)

   138

        self.VariablesGrid.GetMainWindow().Bind(wx.EVT_LEFT_DOWN, self.OnVariablesGridLeftClick)

   139

   140

        self._init_sizers()

   141

   142

    def __init__(self, parent, controler, window):

   143

        ConfTreeNodeEditor.__init__(self, parent, controler, window)

   144

   145

        self.SyncManagersTable = SyncManagersTable(self, [], GetSyncManagersTableColnames())

   146

        self.SyncManagersGrid.SetTable(self.SyncManagersTable)

   147

        self.SyncManagersGridColAlignements = [wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_RIGHT, 

   148

                                               wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, wx.ALIGN_RIGHT]

   149

        self.SyncManagersGridColSizes = [40, 150, 100, 100, 100, 100]

   150

        self.SyncManagersGrid.SetRowLabelSize(0)

   151

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

   152

            attr = wx.grid.GridCellAttr()

   153

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

   154

            self.SyncManagersGrid.SetColAttr(col, attr)

   155

            self.SyncManagersGrid.SetColMinimalWidth(col, self.SyncManagersGridColSizes[col])

   156

            self.SyncManagersGrid.AutoSizeColumn(col, False)

   157

   158

        for colname, colsize, colalign in zip(GetVariablesTableColnames(),

   159

                                              [40, 150, 100, 100, 150, 100, 150, 100],

   160

                                              [wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_RIGHT, 

   161

                                               wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_RIGHT, 

   162

                                               wx.ALIGN_LEFT, wx.ALIGN_LEFT]):

   163

            self.VariablesGrid.AddColumn(colname, colsize, colalign)

   164

        self.VariablesGrid.SetMainColumn(1)

   165

   166

    def GetBufferState(self):

   167

        return False, False

   168

   169

    def RefreshView(self):

   172

        slave_infos = self.Controler.GetSlaveInfos()

   173

        if slave_infos is not None:

   174

            self.Vendor.SetValue(slave_infos["vendor"])

   175

            self.ProductCode.SetValue(slave_infos["product_code"])

   176

            self.RevisionNumber.SetValue(slave_infos["revision_number"])

   177

            self.Physics.SetValue(slave_infos["physics"])

   178

            self.SyncManagersTable.SetData(slave_infos["sync_managers"])

   179

            self.SyncManagersTable.ResetView(self.SyncManagersGrid)

   180

            self.RefreshVariablesGrid(slave_infos["entries"])

   181

        else:

   182

            self.Vendor.SetValue("")

   183

            self.ProductCode.SetValue("")

   184

            self.RevisionNumber.SetValue("")

   185

            self.Physics.SetValue("")

   186

            self.SyncManagersTable.SetData([])

   187

            self.SyncManagersTable.ResetView(self.SyncManagersGrid)

   188

            self.RefreshVariablesGrid([])

   189

   190

    def RefreshVariablesGrid(self, entries):

   191

        root = self.VariablesGrid.GetRootItem()

   192

        if not root.IsOk():

   193

            root = self.VariablesGrid.AddRoot("Slave entries")

   194

        self.GenerateVariablesGridBranch(root, entries, GetVariablesTableColnames())

   195

        self.VariablesGrid.Expand(root)

   196

   197

    def GenerateVariablesGridBranch(self, root, entries, colnames, idx=0):

   198

        if wx.VERSION >= (2, 6, 0):

   199

            item, root_cookie = self.VariablesGrid.GetFirstChild(root)

   200

        else:

   201

            item, root_cookie = self.VariablesGrid.GetFirstChild(root, 0)

   202

   203

        no_more_items = not item.IsOk()

   204

        for entry in entries:

   205

            idx += 1

   206

            if no_more_items:

   207

                item = self.VariablesGrid.AppendItem(root, "")

   208

            for col, colname in enumerate(colnames):

   209

                if col == 0:

   210

                    self.VariablesGrid.SetItemText(item, str(idx), 0)

   211

                else:

   212

                    self.VariablesGrid.SetItemText(item, entry.get(colname, ""), col)

   213

            if entry["PDOMapping"] == "":

   214

                self.VariablesGrid.SetItemBackgroundColour(item, wx.LIGHT_GREY)

   215

            self.VariablesGrid.SetItemPyData(item, entry)

   216

            idx = self.GenerateVariablesGridBranch(item, entry["children"], colnames, idx)

   217

            if not no_more_items:

   218

                item, root_cookie = self.VariablesGrid.GetNextChild(root, root_cookie)

   219

                no_more_items = not item.IsOk()

   220

   221

        if not no_more_items:

   222

            to_delete = []

   223

            while item.IsOk():

   224

                to_delete.append(item)

   225

                item, root_cookie = self.VariablesGrid.GetNextChild(root, root_cookie)

   226

            for item in to_delete:

   227

                self.VariablesGrid.Delete(item)

   228

   229

        return idx

   230

   231

    def OnVariablesGridLeftClick(self, event):

   232

        item, flags, col = self.VariablesGrid.HitTest(event.GetPosition())

   233

        if item.IsOk():

   234

            entry = self.VariablesGrid.GetItemPyData(item)

   235

            data_type = entry.get("Type", "")

   236

            pdo_mapping = entry.get("PDOMapping", "")

   237

   238

            if (col == -1 and pdo_mapping != "" and

   239

                self.Controler.GetSizeOfType(data_type) is not None):

   240

   241

                entry_index = self.Controler.ExtractHexDecValue(entry.get("Index", "0"))

   242

                entry_subindex = self.Controler.ExtractHexDecValue(entry.get("SubIndex", "0"))

   243

                var_name = "%s_%4.4x_%2.2x" % (self.Controler.CTNName(), entry_index, entry_subindex)

   244

                if pdo_mapping == "R":

   245

                    dir = "%I"

   246

                else:

   247

                    dir = "%Q"

   248

                location = "%s%s" % (dir, self.Controler.GetSizeOfType(data_type)) + \

   249

                           ".".join(map(lambda x:str(x), self.Controler.GetCurrentLocation() + (self.Controler.GetSlavePos(), entry_index, entry_subindex)))

   250

   251

                data = wx.TextDataObject(str((location, "location", data_type, var_name, "")))

   252

                dragSource = wx.DropSource(self.VariablesGrid)

   253

                dragSource.SetData(data)

   254

                dragSource.DoDragDrop()

   255

   256

        event.Skip()

   257

   258

    def OnConfNodeEditorResize(self, event):

   259

        self.ConfNodeEditor.GetBestSize()

   260

        xstart, ystart = self.ConfNodeEditor.GetViewStart()

   261

        window_size = self.ConfNodeEditor.GetClientSize()

   262

        maxx, maxy = self.ConfNodeEditor.GetMinSize()

   263

        posx = max(0, min(xstart, (maxx - window_size[0]) / SCROLLBAR_UNIT))

   264

        posy = max(0, min(ystart, (maxy - window_size[1]) / SCROLLBAR_UNIT))

   265

        self.ConfNodeEditor.Scroll(posx, posy)

   266

        self.ConfNodeEditor.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, 

   267

                maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, posx, posy)

   268

        event.Skip()

   269

   270

CIA402NodeEditor = NodeEditor