1

import wx

     3

     4

from controls import CustomGrid, CustomTable, EditorPanel

     5

     6

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

     7

     8

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

     9

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

    10

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

    11

    else:

    12

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

    13

    14

[ID_SLAVETYPECHOICEDIALOG, ID_SLAVETYPECHOICEDIALOGSTATICTEXT1,

    15

 ID_SLAVETYPECHOICEDIALOGSLAVETYPESLIBRARY

    16

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

    17

    18

class SlaveTypeChoiceDialog(wx.Dialog):

    19

    20

    if wx.VERSION < (2, 6, 0):

    21

        def Bind(self, event, function, id = None):

    22

            if id is not None:

    23

                event(self, id, function)

    24

            else:

    25

                event(self, function)

    26

    27

    def _init_coll_flexGridSizer1_Items(self, parent):

    28

        parent.AddWindow(self.staticText1, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

    29

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

    30

        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

    31

    32

    def _init_coll_flexGridSizer1_Growables(self, parent):

    33

        parent.AddGrowableCol(0)

    34

        parent.AddGrowableRow(1)

    35

    36

    def _init_sizers(self):

    37

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

    38

    39

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)

    40

        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)

    41

    42

        self.SetSizer(self.flexGridSizer1)

    43

    44

    def _init_ctrls(self, prnt):

    45

        wx.Dialog.__init__(self, id=ID_SLAVETYPECHOICEDIALOG,

    46

              name='SlaveTypeChoiceDialog', parent=prnt,

    47

              size=wx.Size(600, 400), style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER,

    48

              title=_('Browse slave types library'))

    49

        self.SetClientSize(wx.Size(600, 400))

    50

    51

        self.staticText1 = wx.StaticText(id=ID_SLAVETYPECHOICEDIALOGSTATICTEXT1,

    52

              label=_('Choose a slave type:'), name='staticText1', parent=self,

    53

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

    54

    55

        self.SlaveTypesLibrary = wx.TreeCtrl(id=ID_SLAVETYPECHOICEDIALOGSLAVETYPESLIBRARY,

    56

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

    57

              size=wx.Size(0, 0), style=wx.TR_HAS_BUTTONS|wx.TR_SINGLE|wx.SUNKEN_BORDER|wx.TR_HIDE_ROOT|wx.TR_LINES_AT_ROOT)

    58

    59

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

    60

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

    61

            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())

    62

        else:

    63

            self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())

    64

    65

        self._init_sizers()

    66

    67

    def __init__(self, parent, controler, default=None):

    68

        self._init_ctrls(parent)

    69

    70

        slaves_types = controler.GetSlaveTypesLibrary()

    71

    72

        root = self.SlaveTypesLibrary.AddRoot("")

    73

        self.GenerateSlaveTypesLibraryTreeBranch(root, slaves_types, default)

    74

    75

    def GenerateSlaveTypesLibraryTreeBranch(self, root, children, default):

    76

        for infos in children:

    77

            item = self.SlaveTypesLibrary.AppendItem(root, infos["name"])

    78

            if infos["type"] == ETHERCAT_DEVICE:

    79

                self.SlaveTypesLibrary.SetPyData(item, infos["infos"])

    80

                if infos["infos"] == default:

    81

                    self.SlaveTypesLibrary.SelectItem(item)

    82

                    self.SlaveTypesLibrary.EnsureVisible(item)

    83

            else:

    84

                self.GenerateSlaveTypesLibraryTreeBranch(item, infos["children"], default)

    85

    86

    def GetType(self):

    87

        selected = self.SlaveTypesLibrary.GetSelection()

    88

        return self.SlaveTypesLibrary.GetPyData(selected)

    89

    90

    def OnOK(self, event):

    91

        selected = self.SlaveTypesLibrary.GetSelection()

    92

        if not selected.IsOk() or self.SlaveTypesLibrary.GetPyData(selected) is None:

    93

            message = wx.MessageDialog(self, _("No valid slave type selected!"), _("Error"), wx.OK|wx.ICON_ERROR)

    94

            message.ShowModal()

    95

            message.Destroy()

    96

        else:

    97

            self.EndModal(wx.ID_OK)

    98

    99

   100

def GetVariablesTableColnames():

   101

    _ = lambda x : x

   102

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

   103

   104

class PDOsTable(CustomTable):

   105

   106

    def GetValue(self, row, col):

   107

        if row < self.GetNumberRows():

   108

            if col == 0:

   109

                return row + 1

   110

            colname = self.GetColLabelValue(col, False)

   111

            value = self.data[row].get(colname, "")

   112

            if colname == "Type":

   113

                value = _(value)

   114

            return value

   115

   116

class VariablesTable(CustomTable):

   117

   118

    def GetValue(self, row, col):

   119

        if row < self.GetNumberRows():

   120

            if col == 0:

   121

                return row + 1

   122

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

   123

   124

[ID_SLAVEPANEL, ID_SLAVEPANELTYPELABEL,

   125

 ID_SLAVEPANELTYPE, ID_SLAVEPANELTYPEBROWSE, 

   126

 ID_SLAVEPANELALIASLABEL, ID_SLAVEPANELALIAS, 

   127

 ID_SLAVEPANELPOSLABEL, ID_SLAVEPANELPOS, 

   128

 ID_SLAVEPANELSLAVEINFOSSTATICBOX, ID_SLAVEPANELVENDORLABEL, 

   129

 ID_SLAVEPANELVENDOR, ID_SLAVEPANELPRODUCTCODELABEL, 

   130

 ID_SLAVEPANELPRODUCTCODE, ID_SLAVEPANELREVISIONNUMBERLABEL, 

   131

 ID_SLAVEPANELREVISIONNUMBER, ID_SLAVEPANELPHYSICSLABEL, 

   132

 ID_SLAVEPANELPHYSICS, ID_SLAVEPANELPDOSLABEL, 

   133

 ID_SLAVEPANELPDOSGRID, ID_SLAVEPANELVARIABLESLABEL, 

   134

 ID_SLAVEPANELVARIABLESGRID, 

   135

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

   136

   137

class SlavePanel(wx.Panel):

   138

   139

    if wx.VERSION < (2, 6, 0):

   140

        def Bind(self, event, function, id = None):

   141

            if id is not None:

   142

                event(self, id, function)

   143

            else:

   144

                event(self, function)

   145

   146

    def _init_coll_MainSizer_Items(self, parent):

   147

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

   148

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

   149

   150

    def _init_coll_MainSizer_Growables(self, parent):

   151

        parent.AddGrowableCol(0)

   152

        parent.AddGrowableRow(1)

   153

   154

    def _init_coll_PositionSizer_Items(self, parent):

   155

        parent.AddWindow(self.TypeLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)

   156

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

   157

        parent.AddWindow(self.AliasLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)

   158

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

   159

        parent.AddWindow(self.PosLabel, 0, border=0, flag=wx.ALIGN_CENTER_VERTICAL)

   160

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

   161

   162

    def _init_coll_PositionSizer_Growables(self, parent):

   163

        parent.AddGrowableCol(1)

   164

        parent.AddGrowableCol(3)

   165

        parent.AddGrowableCol(5)

   166

        parent.AddGrowableRow(0)

   167

   168

    def _init_coll_TypeSizer_Items(self, parent):

   169

        parent.AddWindow(self.Type, 1, border=0, flag=0)

   170

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

   171

   172

    def _init_coll_SlaveInfosBoxSizer_Items(self, parent):

   173

        parent.AddSizer(self.SlaveInfosSizer, 1, border=5, flag=wx.GROW|wx.ALL)

   174

   175

    def _init_coll_SlaveInfosSizer_Items(self, parent):

   176

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

   177

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

   178

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

   179

   180

    def _init_coll_SlaveInfosSizer_Growables(self, parent):

   181

        parent.AddGrowableCol(0)

   182

        parent.AddGrowableRow(2)

   183

   184

    def _init_coll_SlaveInfosDetailsSizer_Items(self, parent):

   185

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

   186

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

   187

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

   188

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

   189

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

   190

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

   191

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

   192

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

   193

   194

    def _init_coll_SlaveInfosDetailsSizer_Growables(self, parent):

   195

        parent.AddGrowableCol(1)

   196

        parent.AddGrowableCol(3)

   197

   198

    def _init_sizers(self):

   199

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

   200

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

   201

        self.TypeSizer = wx.BoxSizer(wx.HORIZONTAL)

   202

        self.SlaveInfosBoxSizer = wx.StaticBoxSizer(self.SlaveInfosStaticBox, wx.VERTICAL)

   203

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

   204

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

   205

   206

        self._init_coll_MainSizer_Growables(self.MainSizer)

   207

        self._init_coll_MainSizer_Items(self.MainSizer)

   208

        self._init_coll_PositionSizer_Growables(self.PositionSizer)

   209

        self._init_coll_PositionSizer_Items(self.PositionSizer)

   210

        self._init_coll_TypeSizer_Items(self.TypeSizer)

   211

        self._init_coll_SlaveInfosBoxSizer_Items(self.SlaveInfosBoxSizer)

   212

        self._init_coll_SlaveInfosSizer_Growables(self.SlaveInfosSizer)

   213

        self._init_coll_SlaveInfosSizer_Items(self.SlaveInfosSizer)

   214

        self._init_coll_SlaveInfosDetailsSizer_Growables(self.SlaveInfosDetailsSizer)

   215

        self._init_coll_SlaveInfosDetailsSizer_Items(self.SlaveInfosDetailsSizer)

   216

   217

        self.SetSizer(self.MainSizer)

   218

   219

    def _init_ctrls(self, prnt):

   220

        wx.Panel.__init__(self, id=ID_SLAVEPANEL, name='SlavePanel', parent=prnt,

   221

              size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

   222

   223

        self.TypeLabel = wx.StaticText(id=ID_SLAVEPANELTYPELABEL,

   224

              label=_('Type:'), name='TypeLabel', parent=self,

   225

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

   226

   227

        self.Type = wx.TextCtrl(id=ID_SLAVEPANELTYPE, value='',

   228

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

   229

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

   230

   231

        self.TypeBrowse = wx.Button(id=ID_SLAVEPANELTYPEBROWSE, label='...',

   232

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

   233

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

   234

        self.Bind(wx.EVT_BUTTON, self.OnTypeBrowseClick, id=ID_SLAVEPANELTYPEBROWSE)

   235

   236

        self.AliasLabel = wx.StaticText(id=ID_SLAVEPANELALIASLABEL,

   237

              label=_('Alias:'), name='AliasLabel', parent=self,

   238

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

   239

   240

        self.Alias = wx.SpinCtrl(id=ID_SLAVEPANELALIAS,

   241

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

   242

              size=wx.Size(0, 24), style=wx.SP_ARROW_KEYS, min=0, max=0xffff)

   243

        self.Bind(wx.EVT_SPINCTRL, self.OnAliasChanged, id=ID_SLAVEPANELALIAS)

   244

   245

        self.PosLabel = wx.StaticText(id=ID_SLAVEPANELPOSLABEL,

   246

              label=_('Position:'), name='PositionLabel', parent=self,

   247

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

   248

   249

        self.Pos = wx.SpinCtrl(id=ID_SLAVEPANELPOS,

   250

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

   251

              size=wx.Size(0, 24), style=wx.SP_ARROW_KEYS, min=0, max=0xffff)

   252

        self.Bind(wx.EVT_SPINCTRL, self.OnPositionChanged, id=ID_SLAVEPANELPOS)

   253

   254

        self.SlaveInfosStaticBox = wx.StaticBox(id=ID_SLAVEPANELSLAVEINFOSSTATICBOX,

   255

              label=_('Slave infos:'), name='SlaveInfosStaticBox', parent=self,

   256

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

   257

   258

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

   259

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

   260

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

   261

   262

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

   263

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

   264

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

   265

   266

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

   267

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

   268

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

   269

   270

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

   271

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

   272

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

   273

   274

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

   275

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

   276

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

   277

   278

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

   279

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

   280

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

   281

   282

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

   283

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

   284

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

   285

   286

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

   287

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

   288

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

   289

   290

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

   291

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

   292

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

   293

   294

        self.VariablesGrid = CustomGrid(id=ID_SLAVEPANELPDOSGRID,

   295

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

   296

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

   297

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

   298

            self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)

   299

        else:

   300

            wx.grid.EVT_GRID_CELL_LEFT_CLICK(self.VariablesGrid, self.OnVariablesGridCellLeftClick)

   301

   302

        self._init_sizers()

   303

   304

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

   305

        self._init_ctrls(parent)

   306

   307

        self.Controler = controler

   308

        self.ParentWindow = window

   309

        self.Slave = slave

   310

   311

        self.VariablesTable = VariablesTable(self, [], GetVariablesTableColnames())

   312

        self.VariablesGrid.SetTable(self.VariablesTable)

   313

        self.VariablesGridColAlignements = [wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, wx.ALIGN_RIGHT, 

   314

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

   315

                                            wx.ALIGN_LEFT, wx.ALIGN_LEFT]

   316

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

   317

        self.VariablesGrid.SetRowLabelSize(0)

   318

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

   319

            attr = wx.grid.GridCellAttr()

   320

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

   321

            self.VariablesGrid.SetColAttr(col, attr)

   322

            self.VariablesGrid.SetColMinimalWidth(col, self.VariablesGridColSizes[col])

   323

            self.VariablesGrid.AutoSizeColumn(col, False)

   324

   325

        self.RefreshView()

   326

   327

    def GetSlaveTitle(self):

   328

        type_infos = self.Controler.GetSlaveType(self.Slave)

   329

        return "%s (%d:%d)" % (type_infos["device_type"], self.Slave[0], self.Slave[1])

   330

   331

    def GetSlave(self):

   332

        return self.Slave

   333

   334

    def SetSlave(self, slave):

   335

        if self.Slave != slave:

   336

            self.Slave = slave

   337

            self.RefreshView()

   338

   339

    def RefreshView(self):

   340

        self.Alias.SetValue(self.Slave[0])

   341

        self.Pos.SetValue(self.Slave[1])

   342

        slave_infos = self.Controler.GetSlaveInfos(self.Slave)

   343

        if slave_infos is not None:

   344

            self.Type.SetValue(slave_infos["device_type"])

   345

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

   346

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

   347

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

   348

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

   349

            self.VariablesTable.SetData(slave_infos["variables"])

   350

            self.VariablesTable.ResetView(self.VariablesGrid)

   351

        else:

   352

            type_infos = self.Controler.GetSlaveType(self.Slave)

   353

            self.Type.SetValue(type_infos["device_type"])

   354

   355

    def OnAliasChanged(self, event):

   356

        alias = self.Alias.GetValue()

   357

        if alias != self.Slave[0]:

   358

            result = self.Controler.SetSlavePos(self.Slave[:2], alias = alias)

   359

            if result is not None:

   360

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

   361

                message.ShowModal()

   362

                message.Destroy()

   363

            else:

   364

                wx.CallAfter(self.ParentWindow.RefreshView, (alias, self.Slave[1]))

   365

                wx.CallAfter(self.ParentWindow.RefreshParentWindow)

   366

        event.Skip()

   367

   368

    def OnPositionChanged(self, event):

   369

        position = self.Pos.GetValue()

   370

        if position != self.Slave[1]:

   371

            result = self.Controler.SetSlavePos(self.Slave, position = position)

   372

            if result is not None:

   373

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

   374

                message.ShowModal()

   375

                message.Destroy()

   376

            else:

   377

                wx.CallAfter(self.ParentWindow.RefreshView, (self.Slave[0], position))

   378

                wx.CallAfter(self.ParentWindow.RefreshParentWindow)

   379

        event.Skip()

   380

   381

    def OnTypeBrowseClick(self, event):

   382

        dialog = SlaveTypeChoiceDialog(self, self.Controler, self.Controler.GetSlaveType(self.Slave))

   383

        if dialog.ShowModal() == wx.ID_OK:

   384

            result = self.Controler.SetSlaveType(self.Slave, dialog.GetType())

   385

            if result is not None:

   386

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

   387

                message.ShowModal()

   388

                message.Destroy()

   389

            else:

   390

                wx.CallAfter(self.RefreshView)

   391

                wx.CallAfter(self.ParentWindow.RefreshSlaveNodesTitles)

   392

                wx.CallAfter(self.ParentWindow.RefreshParentWindow)

   393

        dialog.Destroy()

   394

        event.Skip()

   395

   396

    def OnVariablesGridCellLeftClick(self, event):

   397

        if event.GetCol() == 0:

   398

            row = event.GetRow()

   399

            data_type = self.VariablesTable.GetValueByName(row, "Type")

   400

            var_name = self.VariablesTable.GetValueByName(row, "Name")

   401

            entry_index = self.Controler.ExtractHexDecValue(self.VariablesTable.GetValueByName(row, "Index"))

   402

            entry_subindex = self.VariablesTable.GetValueByName(row, "SubIndex")

   403

            if self.VariablesTable.GetValueByName(row, "PDO type") == "Transmit":

   404

                dir = "%I"

   405

            else:

   406

                dir = "%Q"

   407

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

   408

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

   409

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

   410

            dragSource = wx.DropSource(self.VariablesGrid)

   411

            dragSource.SetData(data)

   412

            dragSource.DoDragDrop()

   413

        event.Skip()

   414

   415

[ID_CONFIGEDITOR, ID_CONFIGEDITORADDSLAVEBUTTON,

   416

 ID_CONFIGEDITORDELETESLAVEBUTTON, ID_CONFIGEDITORSLAVENODES,

   417

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

   418

   419

class ConfigEditor(EditorPanel):

   420

   421

    ID = ID_CONFIGEDITOR

   422

   423

    def _init_coll_MainSizer_Items(self, parent):

   424

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

   425

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

   426

   427

    def _init_coll_MainSizer_Growables(self, parent):

   428

        parent.AddGrowableCol(0)

   429

        parent.AddGrowableRow(1)

   430

   431

    def _init_coll_ButtonSizer_Items(self, parent):

   432

        parent.AddWindow(self.AddSlaveButton, 0, border=5, flag=wx.RIGHT)

   433

        parent.AddWindow(self.DeleteSlaveButton, 0, border=5, flag=0)

   434

   435

    def _init_sizers(self):

   436

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

   437

        self.ButtonSizer = wx.BoxSizer(wx.HORIZONTAL)

   438

   439

        self._init_coll_MainSizer_Items(self.MainSizer)

   440

        self._init_coll_MainSizer_Growables(self.MainSizer)

   441

        self._init_coll_ButtonSizer_Items(self.ButtonSizer)

   442

   443

        self.Editor.SetSizer(self.MainSizer)

   444

   445

    def _init_Editor(self, prnt):

   446

        self.Editor = wx.Panel(id=-1, parent=prnt, pos=wx.Point(0, 0), 

   447

                size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

   448

   449

        self.AddSlaveButton = wx.Button(id=ID_CONFIGEDITORADDSLAVEBUTTON, label=_('Add slave'),

   450

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

   451

              size=wx.DefaultSize, style=0)

   452

        self.Bind(wx.EVT_BUTTON, self.OnAddSlaveButtonClick, id=ID_CONFIGEDITORADDSLAVEBUTTON)

   453

   454

        self.DeleteSlaveButton = wx.Button(id=ID_CONFIGEDITORDELETESLAVEBUTTON, label=_('Delete slave'),

   455

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

   456

              size=wx.DefaultSize, style=0)

   457

        self.Bind(wx.EVT_BUTTON, self.OnDeleteSlaveButtonClick, id=ID_CONFIGEDITORDELETESLAVEBUTTON)

   458

   459

        self.SlaveNodes = wx.Notebook(id=ID_CONFIGEDITORSLAVENODES,

   460

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

   461

              size=wx.Size(0, 0), style=wx.NB_LEFT)

   462

   463

        self._init_sizers()

   464

   465

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

   466

        EditorPanel.__init__(self, parent, "", window, controler)

   467

   468

        img = wx.Bitmap(self.Controler.GetIconPath("Cfile.png"), wx.BITMAP_TYPE_PNG).ConvertToImage()

   469

        self.SetIcon(wx.BitmapFromImage(img.Rescale(16, 16)))

   474

    def GetTitle(self):

   475

        fullname = self.Controler.PlugFullName()

   476

        if not self.Controler.ConfigIsSaved():

   477

            return "~%s~" % fullname

   478

        return fullname

   479

   480

    def GetBufferState(self):

   481

        return self.Controler.GetBufferState()

   482

   483

    def Undo(self):

   484

        self.Controler.LoadPrevious()

   485

        self.RefreshView()

   486

   487

    def Redo(self):

   488

        self.Controler.LoadNext()

   489

        self.RefreshView()

   490

   491

    def RefreshView(self, slave_pos=None):

   492

        slaves = self.Controler.GetSlaves()

   493

        for i, slave in enumerate(slaves):

   494

            if i < self.SlaveNodes.GetPageCount():

   495

                panel = self.SlaveNodes.GetPage(i)

   496

                panel.SetSlave(slave)

   497

            else:

   498

                panel = SlavePanel(self.SlaveNodes, self.Controler, self, slave)

   499

                self.SlaveNodes.AddPage(panel, "")

   500

        while self.SlaveNodes.GetPageCount() > len(slaves):

   501

            self.SlaveNodes.RemovePage(len(slaves))

   502

        self.RefreshSlaveNodesTitles()

   503

        self.RefreshButtons()

   504

        if slave_pos is not None:

   505

            self.SelectSlave(slave_pos)

   506

   507

    def RefreshParentWindow(self):

   508

        self.ParentWindow.RefreshTitle()

   509

        self.ParentWindow.RefreshFileMenu()

   510

        self.ParentWindow.RefreshEditMenu()

   511

        self.ParentWindow.RefreshPageTitles()

   512

   513

    def RefreshSlaveNodesTitles(self):

   514

        for idx in xrange(self.SlaveNodes.GetPageCount()):

   515

            panel = self.SlaveNodes.GetPage(idx)

   516

            self.SlaveNodes.SetPageText(idx, panel.GetSlaveTitle())

   517

   518

    def RefreshButtons(self):

   519

        self.DeleteSlaveButton.Enable(self.SlaveNodes.GetPageCount() > 0)

   520

   521

    def SelectSlave(self, slave):

   522

        for idx in xrange(self.SlaveNodes.GetPageCount()):

   523

            panel = self.SlaveNodes.GetPage(idx)

   524

            if panel.GetSlave() == slave:

   525

                self.SlaveNodes.SetSelection(idx)

   526

                return

   527