1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

#This file is part of CanFestival, a library implementing CanOpen Stack. 

     5

#

     6

#Copyright (C): Edouard TISSERANT, Francis DUPIN and Laurent BESSARD

     7

#

     8

#See COPYING file for copyrights details.

     9

#

    10

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

    11

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

    12

#License as published by the Free Software Foundation; either

    13

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

    14

#

    15

#This library 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 GNU

    18

#Lesser General Public License for more details.

    19

#

    20

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

    21

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

    22

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    23

    24

import wx

    25

import wx.grid

    26

    27

from types import *

    28

from commondialogs import *

    29

    30

from node import OD_Subindex, OD_MultipleSubindexes, OD_IdenticalSubindexes, OD_IdenticalIndexes

    31

    32

ColSizes = [75, 250, 150, 125, 100, 60, 250]

    33

ColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_CENTER, wx.ALIGN_RIGHT, wx.ALIGN_CENTER, wx.ALIGN_CENTER, wx.ALIGN_LEFT]

    83

    84

DictionaryOrganisation = [

    85

    {"minIndex" : 0x0001, "maxIndex" : 0x0FFF, "name" : "Data Type Definitions"},

    86

    {"minIndex" : 0x1000, "maxIndex" : 0x1029, "name" : "Communication Parameters"},

    87

    {"minIndex" : 0x1200, "maxIndex" : 0x12FF, "name" : "SDO Parameters"},

    88

    {"minIndex" : 0x1400, "maxIndex" : 0x15FF, "name" : "Receive PDO Parameters"},

    89

    {"minIndex" : 0x1600, "maxIndex" : 0x17FF, "name" : "Receive PDO Mapping"},

    90

    {"minIndex" : 0x1800, "maxIndex" : 0x19FF, "name" : "Transmit PDO Parameters"},

    91

    {"minIndex" : 0x1A00, "maxIndex" : 0x1BFF, "name" : "Transmit PDO Mapping"},

    92

    {"minIndex" : 0x1C00, "maxIndex" : 0x1FFF, "name" : "Other Communication Parameters"},

    93

    {"minIndex" : 0x2000, "maxIndex" : 0x5FFF, "name" : "Manufacturer Specific"},

    94

    {"minIndex" : 0x6000, "maxIndex" : 0x9FFF, "name" : "Standardized Device Profile"},

    95

    {"minIndex" : 0xA000, "maxIndex" : 0xBFFF, "name" : "Standardized Interface Profile"}]

    96

    97

SizeConversion = {1 : "X", 8 : "B", 16 : "W", 24 : "D", 32 : "D", 40 : "L", 48 : "L", 56 : "L", 64 : "L"}

    98

    99

class SubindexTable(wx.grid.PyGridTableBase):

   100

   101

    """

   102

    A custom wxGrid Table using user supplied data

   103

    """

   104

    def __init__(self, parent, data, editors, colnames):

   105

        # The base class must be initialized *first*

   106

        wx.grid.PyGridTableBase.__init__(self)

   107

        self.data = data

   108

        self.editors = editors

   109

        self.CurrentIndex = 0

   110

        self.colnames = colnames

   111

        self.Parent = parent

   112

        self.Editable = True

   113

        # XXX

   114

        # we need to store the row length and collength to

   115

        # see if the table has changed size

   116

        self._rows = self.GetNumberRows()

   117

        self._cols = self.GetNumberCols()

   118

   119

    def Disable(self):

   120

        self.Editable = False

   121

   122

    def Enable(self):

   123

        self.Editable = True

   124

   125

    def GetNumberCols(self):

   126

        return len(self.colnames)

   127

   128

    def GetNumberRows(self):

   129

        return len(self.data)

   130

   132

        if col < len(self.colnames):

   135

            return self.colnames[col]

   136

   138

        return row

   139

   140

    def GetValue(self, row, col):

   141

        if row < self.GetNumberRows():

   147

   148

    def GetEditor(self, row, col):

   149

        if row < self.GetNumberRows():

   151

   152

    def GetValueByName(self, row, colname):

   153

        return self.data[row].get(colname)

   154

   155

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

   156

        if col < len(self.colnames):

   165

   166

    def ResetView(self, grid):

   167

        """

   168

        (wx.grid.Grid) -> Reset the grid view.   Call this to

   169

        update the grid if rows and columns have been added or deleted

   170

        """

   171

        grid.BeginBatch()

   172

        for current, new, delmsg, addmsg in [

   173

            (self._rows, self.GetNumberRows(), wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED),

   174

            (self._cols, self.GetNumberCols(), wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED),

   175

        ]:

   176

            if new < current:

   177

                msg = wx.grid.GridTableMessage(self,delmsg,new,current-new)

   178

                grid.ProcessTableMessage(msg)

   179

            elif new > current:

   180

                msg = wx.grid.GridTableMessage(self,addmsg,new-current)

   181

                grid.ProcessTableMessage(msg)

   182

                self.UpdateValues(grid)

   183

        grid.EndBatch()

   184

   185

        self._rows = self.GetNumberRows()

   186

        self._cols = self.GetNumberCols()

   187

        # update the column rendering scheme

   188

        self._updateColAttrs(grid)

   189

   190

        # update the scrollbars and the displayed part of the grid

   191

        grid.AdjustScrollbars()

   192

        grid.ForceRefresh()

   193

   194

   195

    def UpdateValues(self, grid):

   196

        """Update all displayed values"""

   197

        # This sends an event to the grid table to update all of the values

   198

        msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_REQUEST_VIEW_GET_VALUES)

   199

        grid.ProcessTableMessage(msg)

   200

   201

    def _updateColAttrs(self, grid):

   202

        """

   203

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

   204

        appropriate renderer given the column name.

   205

   206

        Otherwise default to the default renderer.

   207

        """

   208

   209

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

   210

            attr = wx.grid.GridCellAttr()

   211

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

   212

            grid.SetColAttr(col, attr)

   215

   216

        typelist = None

   217

        maplist = None

   218

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

   219

            editors = self.editors[row]

   220

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

   221

                editor = None

   222

                renderer = None

   223

   225

                editortype = editors[colname]

   226

                if editortype and self.Editable:

   227

                    grid.SetReadOnly(row, col, False)

   228

                    if editortype == "string":

   229

                        editor = wx.grid.GridCellTextEditor()

   230

                        renderer = wx.grid.GridCellStringRenderer()

   231

                        if colname == "value" and "length" in editors:

   232

                            editor.SetParameters(editors["length"]) 

   233

                    elif editortype == "number":

   234

                        editor = wx.grid.GridCellNumberEditor()

   235

                        renderer = wx.grid.GridCellNumberRenderer()

   236

                        if colname == "value" and "min" in editors and "max" in editors:

   237

                            editor.SetParameters("%s,%s"%(editors["min"],editors["max"]))

   238

                    elif editortype == "float":

   239

                        editor = wx.grid.GridCellTextEditor()

   240

                        renderer = wx.grid.GridCellStringRenderer()

   241

                    elif editortype == "bool":

   242

                        editor = wx.grid.GridCellChoiceEditor()

   243

                        editor.SetParameters(BoolList)

   244

                    elif editortype == "access":

   245

                        editor = wx.grid.GridCellChoiceEditor()

   246

                        editor.SetParameters(AccessList)

   247

                    elif editortype == "raccess":

   248

                        editor = wx.grid.GridCellChoiceEditor()

   249

                        editor.SetParameters(RAccessList)

   250

                    elif editortype == "option":

   251

                        editor = wx.grid.GridCellChoiceEditor()

   252

                        editor.SetParameters(OptionList)

   253

                    elif editortype == "type":

   254

                        editor = wx.grid.GridCellChoiceEditor()

   255

                        if typelist == None:

   256

                            typelist = self.Parent.Manager.GetCurrentTypeList()

   257

                        editor.SetParameters(typelist)

   258

                    elif editortype == "map":

   259

                        editor = wx.grid.GridCellChoiceEditor()

   260

                        if maplist == None:

   261

                            maplist = self.Parent.Manager.GetCurrentMapList()

   262

                        editor.SetParameters(maplist)

   263

                    elif editortype == "time":

   264

                        editor = wx.grid.GridCellTextEditor()

   265

                        renderer = wx.grid.GridCellStringRenderer()

   266

                    elif editortype == "domain":

   267

                        editor = wx.grid.GridCellTextEditor()

   268

                        renderer = wx.grid.GridCellStringRenderer()

   269

                    elif editortype == "dcf":

   270

                        editor = wx.grid.GridCellTextEditor()

   271

                        renderer = wx.grid.GridCellStringRenderer()

   272

                else:

   273

                    grid.SetReadOnly(row, col, True)

   274

   275

                grid.SetCellEditor(row, col, editor)

   276

                grid.SetCellRenderer(row, col, renderer)

   277

   278

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

   279

   280

    def SetData(self, data):

   281

        self.data = data

   282

   283

    def SetEditors(self, editors):

   284

        self.editors = editors

   285

   286

    def GetCurrentIndex(self):

   287

        return self.CurrentIndex

   288

   289

    def SetCurrentIndex(self, index):

   290

        self.CurrentIndex = index

   291

   292

    def AppendRow(self, row_content):

   293

        self.data.append(row_content)

   294

   295

    def Empty(self):

   296

        self.data = []

   297

        self.editors = []

   298

   299

[ID_EDITINGPANEL, ID_EDITINGPANELADDBUTTON, ID_EDITINGPANELINDEXCHOICE, 

   300

 ID_EDITINGPANELINDEXLIST, ID_EDITINGPANELINDEXLISTPANEL, ID_EDITINGPANELPARTLIST, 

   301

 ID_EDITINGPANELSECONDSPLITTER, ID_EDITINGPANELSUBINDEXGRID,

   302

 ID_EDITINGPANELSUBINDEXGRIDPANEL, ID_EDITINGPANELCALLBACKCHECK,

   303

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

   304

   305

[ID_EDITINGPANELINDEXLISTMENUITEMS0, ID_EDITINGPANELINDEXLISTMENUITEMS1, 

   306

 ID_EDITINGPANELINDEXLISTMENUITEMS2, 

   307

] = [wx.NewId() for _init_coll_IndexListMenu_Items in range(3)]

   308

   309

[ID_EDITINGPANELMENU1ITEMS0, ID_EDITINGPANELMENU1ITEMS1, 

   310

 ID_EDITINGPANELMENU1ITEMS3, ID_EDITINGPANELMENU1ITEMS4, 

   311

] = [wx.NewId() for _init_coll_SubindexGridMenu_Items in range(4)]

   312

   313

class EditingPanel(wx.SplitterWindow):

   314

    def _init_coll_AddToListSizer_Items(self, parent):

   315

        parent.AddWindow(self.AddButton, 0, border=0, flag=0)

   316

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

   317

   318

    def _init_coll_SubindexGridSizer_Items(self, parent):

   319

        parent.AddWindow(self.CallbackCheck, 0, border=0, flag=0)

   320

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

   321

   322

    def _init_coll_IndexListSizer_Items(self, parent):

   323

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

   324

        parent.AddSizer(self.AddToListSizer, 0, border=0, flag=wx.GROW)

   325

   326

    def _init_coll_AddToListSizer_Growables(self, parent):

   327

        parent.AddGrowableCol(1)

   328

   329

    def _init_coll_SubindexGridSizer_Growables(self, parent):

   330

        parent.AddGrowableCol(0)

   331

        parent.AddGrowableRow(1)

   332

   333

    def _init_coll_IndexListSizer_Growables(self, parent):

   334

        parent.AddGrowableCol(0)

   335

        parent.AddGrowableRow(0)

   336

   337

    def _init_coll_SubindexGridMenu_Items(self, parent):

   338

        parent.Append(help='', id=ID_EDITINGPANELMENU1ITEMS0,

   340

        parent.Append(help='', id=ID_EDITINGPANELMENU1ITEMS1,

   342

        parent.AppendSeparator()

   343

        parent.Append(help='', id=ID_EDITINGPANELMENU1ITEMS3,

   345

        if not self.Editable:

   346

            parent.Append(help='', id=ID_EDITINGPANELMENU1ITEMS4,

   348

        self.Bind(wx.EVT_MENU, self.OnAddSubindexMenu,

   349

              id=ID_EDITINGPANELMENU1ITEMS0)

   350

        self.Bind(wx.EVT_MENU, self.OnDeleteSubindexMenu,

   351

              id=ID_EDITINGPANELMENU1ITEMS1)

   352

        self.Bind(wx.EVT_MENU, self.OnDefaultValueSubindexMenu,

   353

              id=ID_EDITINGPANELMENU1ITEMS3)

   354

        if not self.Editable:

   355

            self.Bind(wx.EVT_MENU, self.OnAddToDCFSubindexMenu,

   356

                  id=ID_EDITINGPANELMENU1ITEMS4)

   357

   358

    def _init_coll_IndexListMenu_Items(self, parent):

   359

        parent.Append(help='', id=ID_EDITINGPANELINDEXLISTMENUITEMS0,

   361

        parent.Append(help='', id=ID_EDITINGPANELINDEXLISTMENUITEMS2,

   363

        parent.Append(help='', id=ID_EDITINGPANELINDEXLISTMENUITEMS1,

   365

        self.Bind(wx.EVT_MENU, self.OnRenameIndexMenu,

   366

              id=ID_EDITINGPANELINDEXLISTMENUITEMS0)

   367

        self.Bind(wx.EVT_MENU, self.OnDeleteIndexMenu,

   368

              id=ID_EDITINGPANELINDEXLISTMENUITEMS1)

   369

        self.Bind(wx.EVT_MENU, self.OnModifyIndexMenu,

   370

              id=ID_EDITINGPANELINDEXLISTMENUITEMS2)

   371

   372

    def _init_utils(self):

   373

        self.IndexListMenu = wx.Menu(title='')

   374

        self.SubindexGridMenu = wx.Menu(title='')

   375

   376

        self._init_coll_IndexListMenu_Items(self.IndexListMenu)

   377

        self._init_coll_SubindexGridMenu_Items(self.SubindexGridMenu)

   378

   379

    def _init_sizers(self):

   380

        self.IndexListSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)

   381

        self.SubindexGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)

   382

        self.AddToListSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=1, vgap=0)

   383

   384

        self._init_coll_IndexListSizer_Growables(self.IndexListSizer)

   385

        self._init_coll_IndexListSizer_Items(self.IndexListSizer)

   386

        self._init_coll_SubindexGridSizer_Growables(self.SubindexGridSizer)

   387

        self._init_coll_SubindexGridSizer_Items(self.SubindexGridSizer)

   388

        self._init_coll_AddToListSizer_Growables(self.AddToListSizer)

   389

        self._init_coll_AddToListSizer_Items(self.AddToListSizer)

   390

   391

        self.SubindexGridPanel.SetSizer(self.SubindexGridSizer)

   392

        self.IndexListPanel.SetSizer(self.IndexListSizer)

   393

   394

    def _init_ctrls(self, prnt):

   395

        wx.SplitterWindow.__init__(self, id=ID_EDITINGPANEL,

   396

              name='MainSplitter', parent=prnt, point=wx.Point(0, 0),

   397

              size=wx.Size(-1, -1), style=wx.SP_3D)

   398

        self._init_utils()

   399

        self.SetNeedUpdating(True)

   400

        self.SetMinimumPaneSize(1)

   401

   402

        self.PartList = wx.ListBox(choices=[], id=ID_EDITINGPANELPARTLIST,

   403

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

   404

              size=wx.Size(-1, -1), style=0)

   405

        self.PartList.Bind(wx.EVT_LISTBOX, self.OnPartListBoxClick,

   406

              id=ID_EDITINGPANELPARTLIST)

   407

   408

        self.SecondSplitter = wx.SplitterWindow(id=ID_EDITINGPANELSECONDSPLITTER,

   409

              name='SecondSplitter', parent=self, point=wx.Point(0, 0), 

   410

              size=wx.Size(-1, -1), style=wx.SP_3D)

   411

        self.SecondSplitter.SetMinimumPaneSize(1)

   412

        self.SplitHorizontally(self.PartList, self.SecondSplitter,

   413

              110)

   414

   415

        self.SubindexGridPanel = wx.Panel(id=ID_EDITINGPANELSUBINDEXGRIDPANEL,

   416

              name='SubindexGridPanel', parent=self.SecondSplitter, 

   417

              pos=wx.Point(0, 0), size=wx.Size(-1, -1), style=wx.TAB_TRAVERSAL)

   418

   419

        self.IndexListPanel = wx.Panel(id=ID_EDITINGPANELINDEXLISTPANEL,

   420

              name='IndexListPanel', parent=self.SecondSplitter, 

   421

              pos=wx.Point(0, 0), size=wx.Size(-1, -1), style=wx.TAB_TRAVERSAL)

   422

        self.SecondSplitter.SplitVertically(self.IndexListPanel,

   423

              self.SubindexGridPanel, 280)

   424

   425

        self.SubindexGrid = wx.grid.Grid(id=ID_EDITINGPANELSUBINDEXGRID,

   426

              name='SubindexGrid', parent=self.SubindexGridPanel, pos=wx.Point(0,

   427

              0), size=wx.Size(-1, -1), style=0)

   428

        self.SubindexGrid.SetFont(wx.Font(12, 77, wx.NORMAL, wx.NORMAL, False,

   429

              'Sans'))

   430

        self.SubindexGrid.SetLabelFont(wx.Font(10, 77, wx.NORMAL, wx.NORMAL,

   431

              False, 'Sans'))

   432

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

   433

              self.OnSubindexGridCellChange)

   434

        self.SubindexGrid.Bind(wx.grid.EVT_GRID_CELL_RIGHT_CLICK,

   435

              self.OnSubindexGridRightClick)

   436

        self.SubindexGrid.Bind(wx.grid.EVT_GRID_SELECT_CELL,

   437

              self.OnSubindexGridSelectCell)

   438

        self.SubindexGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, 

   439

              self.OnSubindexGridCellLeftClick)

   440

        self.SubindexGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, 

   441

              self.OnSubindexGridEditorShown)

   442

   443

        self.CallbackCheck = wx.CheckBox(id=ID_EDITINGPANELCALLBACKCHECK,

   445

              parent=self.SubindexGridPanel, pos=wx.Point(0, 0), size=wx.Size(152,

   446

              24), style=0)

   447

        self.CallbackCheck.Bind(wx.EVT_CHECKBOX, self.OnCallbackCheck,

   448

              id=ID_EDITINGPANELCALLBACKCHECK)

   449

   450

        self.IndexList = wx.ListBox(choices=[], id=ID_EDITINGPANELINDEXLIST,

   451

              name='IndexList', parent=self.IndexListPanel, pos=wx.Point(0, 0),

   452

              size=wx.Size(-1, -1), style=0)

   453

        self.IndexList.Bind(wx.EVT_LISTBOX, self.OnIndexListClick,

   454

              id=ID_EDITINGPANELINDEXLIST)

   455

        self.IndexList.Bind(wx.EVT_RIGHT_UP, self.OnIndexListRightUp)

   456

   458

              name='AddButton', parent=self.IndexListPanel, pos=wx.Point(0, 0),

   460

        self.AddButton.Bind(wx.EVT_BUTTON, self.OnAddButtonClick,

   461

              id=ID_EDITINGPANELADDBUTTON)

   462

   463

        self.IndexChoice = wx.Choice(choices=[], id=ID_EDITINGPANELINDEXCHOICE,

   464

              name='IndexChoice', parent=self.IndexListPanel, pos=wx.Point(50,

   465

              0), size=wx.Size(-1, 30), style=0)

   466

   467

        self._init_sizers()

   468

   469

    def __init__(self, parent, window, manager, editable = True):

   470

        self.Editable = editable

   471

        self._init_ctrls(parent)

   472

        self.ParentWindow = window

   473

        self.Manager = manager

   474

        self.ListIndex = []

   475

        self.ChoiceIndex = []

   476

        self.FirstCall = False

   477

        self.Index = None

   478

   479

        for values in DictionaryOrganisation:

   480

            text = "   0x%04X-0x%04X      %s"%(values["minIndex"], values["maxIndex"], values["name"])

   481

            self.PartList.Append(text)

   483

        self.SubindexGrid.SetTable(self.Table)

   484

        self.SubindexGrid.SetRowLabelSize(0)

   485

        self.CallbackCheck.Disable()

   486

        self.Table.ResetView(self.SubindexGrid)

   487

   488

        if not self.Editable:

   489

            self.AddButton.Disable()

   490

            self.IndexChoice.Disable()

   491

            self.CallbackCheck.Disable()

   492

            self.Table.Disable()

   493

   494

    def GetIndex(self):

   495

        return self.Index

   496

   497

    def SetIndex(self, index):

   498

        self.Index = index

   499

   500

    def GetSelection(self):

   501

        selected = self.IndexList.GetSelection()

   502

        if selected != wx.NOT_FOUND:

   503

            index = self.ListIndex[selected]

   504

            subIndex = self.SubindexGrid.GetGridCursorRow()

   505

            return index, subIndex

   506

        return None

   507