1

import os

     2

     6

import wx.lib.buttons

     7

     8

from controls import CustomGrid, CustomTable, FolderTree

    10

from util.BitmapLibrary import GetBitmap

    22

    return ["#", _("Name"), _("Index"), _("SubIndex"), _("Type"), _("Access")]

    23

    24

ACCESS_TYPES = {

    25

    'ro': 'R',

    26

    'wo': 'W',

    27

    'rw': 'R/W'}

    28

    29

def GetAccessValue(access, pdo_mapping):

    30

    value = ACCESS_TYPES.get(access)

    31

    if pdo_mapping != "":

    32

        value += "/P"

    33

    return value

    41

        self.EthercatNodeEditor = wx.Panel(prnt, style=wx.TAB_TRAVERSAL)

    42

    43

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

    44

        main_sizer.AddGrowableCol(0)

    45

        main_sizer.AddGrowableRow(1)

    46

    47

        variables_label = wx.StaticText(self.EthercatNodeEditor,

    48

              label=_('Variable entries:'))

    49

        main_sizer.AddWindow(variables_label, border=10, flag=wx.TOP|wx.LEFT|wx.RIGHT)

    50

    51

        self.VariablesGrid = wx.gizmos.TreeListCtrl(self.EthercatNodeEditor,

    52

              style=wx.TR_DEFAULT_STYLE |

    53

                    wx.TR_ROW_LINES |

    54

                    wx.TR_COLUMN_LINES |

    55

                    wx.TR_HIDE_ROOT |

    56

                    wx.TR_FULL_ROW_HIGHLIGHT)

    57

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

    58

            self.OnVariablesGridLeftClick)

    59

        main_sizer.AddWindow(self.VariablesGrid, border=10, 

    60

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

    62

        self.EthercatNodeEditor.SetSizer(main_sizer)

    63

    70

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

    72

                                               wx.ALIGN_RIGHT, wx.ALIGN_LEFT, wx.ALIGN_LEFT]):

    99

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

   110

                    value = entry.get(colname, "")

   111

                    if colname == "Access":

   112

                        value = GetAccessValue(value, entry.get("PDOMapping", ""))

   113

                    self.VariablesGrid.SetItemText(item, value, col)

   160

CIA402NodeEditor = NodeEditor

   161

   162

   163

def GetModulesTableColnames():

   164

    _ = lambda x : x

   165

    return [_("Name"), _("PDO alignment (bits)")]

   166

   167

class LibraryEditorPanel(wx.ScrolledWindow):

   168

   169

    def __init__(self, parent, module_library, buttons):

   170

        wx.ScrolledWindow.__init__(self, parent,

   171

            style=wx.TAB_TRAVERSAL|wx.HSCROLL|wx.VSCROLL)

   172

        self.Bind(wx.EVT_SIZE, self.OnResize)

   173

   174

        self.ModuleLibrary = module_library

   175

   176

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

   177

        main_sizer.AddGrowableCol(0)

   178

        main_sizer.AddGrowableRow(1)

   179

        main_sizer.AddGrowableRow(3)

   180

   181

        ESI_files_label = wx.StaticText(self, 

   182

            label=_("ESI Files:"))

   183

        main_sizer.AddWindow(ESI_files_label, border=10, 

   184

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

   185

   186

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

   187

        folder_tree_sizer.AddGrowableCol(0)

   188

        folder_tree_sizer.AddGrowableRow(0)

   189

        main_sizer.AddSizer(folder_tree_sizer, border=10, 

   190

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

   191

   192

        self.ESIFiles = FolderTree(self, self.GetPath(), editable=False)

   193

        self.ESIFiles.SetFilter(".xml")

   194

        self.ESIFiles.SetMinSize(wx.Size(600, 300))

   195

        folder_tree_sizer.AddWindow(self.ESIFiles, flag=wx.GROW)

   196

   197

        buttons_sizer = wx.BoxSizer(wx.VERTICAL)

   198

        folder_tree_sizer.AddSizer(buttons_sizer, 

   199

            flag=wx.ALIGN_CENTER_VERTICAL)

   200

   201

        for idx, (name, bitmap, help, callback) in enumerate(buttons):

   202

            button = wx.lib.buttons.GenBitmapButton(self, 

   203

                  bitmap=GetBitmap(bitmap), 

   204

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

   205

            button.SetToolTipString(help)

   206

            setattr(self, name, button)

   207

            if idx > 0:

   208

                flag = wx.TOP

   209

            else:

   210

                flag = 0

   211

            if callback is None:

   212

                callback = getattr(self, "On" + name, None)

   213

            if callback is not None:

   214

                self.Bind(wx.EVT_BUTTON, callback, button)

   215

            buttons_sizer.AddWindow(button, border=10, flag=flag)

   216

   217

        modules_label = wx.StaticText(self, 

   218

            label=_("Modules library:"))

   219

        main_sizer.AddSizer(modules_label, border=10, 

   220

            flag=wx.LEFT|wx.RIGHT)

   221

   222

        self.ModulesGrid = wx.gizmos.TreeListCtrl(self,

   223

              style=wx.TR_DEFAULT_STYLE |

   224

                    wx.TR_ROW_LINES |

   225

                    wx.TR_COLUMN_LINES |

   226

                    wx.TR_HIDE_ROOT |

   227

                    wx.TR_FULL_ROW_HIGHLIGHT)

   228

        self.ModulesGrid.SetMinSize(wx.Size(600, 300))

   229

        self.ModulesGrid.GetMainWindow().Bind(wx.EVT_LEFT_DCLICK,

   230

            self.OnModulesGridLeftDClick)

   231

        main_sizer.AddWindow(self.ModulesGrid, border=10, 

   232

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

   233

   234

        self.SetSizer(main_sizer)

   235

   236

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

   237

                                              [400, 150],

   238

                                              [wx.ALIGN_LEFT, wx.ALIGN_RIGHT]):

   239

            self.ModulesGrid.AddColumn(_(colname), colsize, colalign)

   240

        self.ModulesGrid.SetMainColumn(0)

   241

   242

    def GetPath(self):

   243

        return self.ModuleLibrary.GetPath()

   244

   245

    def GetSelectedFilePath(self):

   246

        return self.ESIFiles.GetPath()

   247

   248

    def RefreshView(self):

   249

        self.ESIFiles.RefreshTree()

   250

        self.RefreshModulesGrid()

   251

   252

    def RefreshModulesGrid(self):

   253

        root = self.ModulesGrid.GetRootItem()

   254

        if not root.IsOk():

   255

            root = self.ModulesGrid.AddRoot("Modules")

   256

        self.GenerateModulesGridBranch(root, 

   257

            self.ModuleLibrary.GetModulesLibrary(), 

   258

            GetVariablesTableColnames())

   259

        self.ModulesGrid.Expand(root)

   260

   261

    def GenerateModulesGridBranch(self, root, modules, colnames):

   262

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

   263

   264

        no_more_items = not item.IsOk()

   265

        for module in modules:

   266

            if no_more_items:

   267

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

   268

            self.ModulesGrid.SetItemText(item, module["name"], 0)

   269

            if module["infos"] is not None:

   270

                self.ModulesGrid.SetItemText(item, str(module["infos"]["alignment"]), 1)

   271

            else:

   272

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

   273

            self.ModulesGrid.SetItemPyData(item, module["infos"])

   274

            self.GenerateModulesGridBranch(item, module["children"], colnames)

   275

            if not no_more_items:

   276

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

   277

                no_more_items = not item.IsOk()

   278

   279

        if not no_more_items:

   280

            to_delete = []

   281

            while item.IsOk():

   282

                to_delete.append(item)

   283

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

   284

            for item in to_delete:

   285

                self.ModulesGrid.Delete(item)

   286

   287

    def OnImportButton(self, event):

   288

        dialog = wx.FileDialog(self,

   289

             _("Choose an XML file"), 

   290

             os.getcwd(), "",  

   291

             _("XML files (*.xml)|*.xml|All files|*.*"), wx.OPEN)

   292

   293

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

   294

            filepath = dialog.GetPath()

   295

            if self.ModuleLibrary.ImportModuleLibrary(filepath):

   296

                wx.CallAfter(self.RefreshView)

   297

            else:

   298

                message = wx.MessageDialog(self, 

   299

                    _("No such XML file: %s\n") % filepath, 

   300

                    _("Error"), wx.OK|wx.ICON_ERROR)

   301

                message.ShowModal()

   302

                message.Destroy()

   303

        dialog.Destroy()

   304

   305

        event.Skip()

   306

   307

    def OnDeleteButton(self, event):

   308

        filepath = self.GetSelectedFilePath()

   309

        if os.path.isfile(filepath):

   310

            folder, filename = os.path.split(filepath)

   311

   312

            dialog = wx.MessageDialog(self, 

   313

                  _("Do you really want to delete the file '%s'?") % filename, 

   314

                  _("Delete File"), wx.YES_NO|wx.ICON_QUESTION)

   315

            remove = dialog.ShowModal() == wx.ID_YES

   316

            dialog.Destroy()

   317

   318

            if remove:

   319

                os.remove(filepath)

   320

                self.ModuleLibrary.LoadModules()

   321

                wx.CallAfter(self.RefreshView)

   322

        event.Skip()

   323

   324

    def OnModulesGridLeftDClick(self, event):

   325

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

   326

        if item.IsOk():

   327

            entry_infos = self.ModulesGrid.GetItemPyData(item)

   328

            if entry_infos is not None and col == 1:

   329

                dialog = wx.TextEntryDialog(self, 

   330

                    _("Set PDO alignment (bits):"),

   331

                    _("%s PDO alignment") % self.ModulesGrid.GetItemText(item), 

   332

                    str(entry_infos["alignment"]))

   333

   334

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

   335

                    try:

   336

                        self.ModuleLibrary.SetAlignment(

   337

                            entry_infos["vendor"],

   338

                            entry_infos["product_code"],

   339

                            entry_infos["revision_number"],

   340

                            int(dialog.GetValue()))

   341

                        wx.CallAfter(self.RefreshModulesGrid)

   342

                    except ValueError:

   343

                        message = wx.MessageDialog(self, 

   344

                            _("Module PDO alignment must be an integer!"), 

   345

                            _("Error"), wx.OK|wx.ICON_ERROR)

   346

                        message.ShowModal()

   347

                        message.Destroy()

   348

   349

                dialog.Destroy()

   350

   351

        event.Skip()

   352

   353

    def OnResize(self, event):

   354

        self.GetBestSize()

   355

        xstart, ystart = self.GetViewStart()

   356

        window_size = self.GetClientSize()

   357

        maxx, maxy = self.GetMinSize()

   360

        self.Scroll(posx, posy)

   361

        self.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, 

   365

class DatabaseManagementDialog(wx.Dialog):

   366

   367

    def __init__(self, parent, database):

   368

        wx.Dialog.__init__(self, parent,

   369

              size=wx.Size(700, 500), title=_('ESI Files Database management'),

   370

              style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER)

   371

   372

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

   373

        main_sizer.AddGrowableCol(0)

   374

        main_sizer.AddGrowableRow(0)

   375

   376

        self.DatabaseEditor = LibraryEditorPanel(self, database,

   377

            [("ImportButton", "ImportESI", _("Import file to ESI files database"), None),

   378

             ("DeleteButton", "remove_element", _("Remove file from database"), None)])

   379

        main_sizer.AddWindow(self.DatabaseEditor, border=10,

   380

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

   381

   382

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

   383

        button_sizer.GetAffirmativeButton().SetLabel(_("Add file to project"))

   384

        button_sizer.GetCancelButton().SetLabel(_("Close"))

   385

        main_sizer.AddSizer(button_sizer, border=10, 

   386

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

   387

   388

        self.SetSizer(main_sizer)

   389

   390

        self.DatabaseEditor.RefreshView()

   391

   392

    def GetValue(self):

   393

        return self.DatabaseEditor.GetSelectedFilePath()

   394

   395

class LibraryEditor(ConfTreeNodeEditor):

   396

   397

    CONFNODEEDITOR_TABS = [

   398

        (_("Modules Library"), "_create_ModuleLibraryEditor")]

   399

   400

    def _create_ModuleLibraryEditor(self, prnt):

   401

        self.ModuleLibraryEditor = LibraryEditorPanel(prnt,

   402

            self.Controler.GetModulesLibraryInstance(),

   403

            [("ImportButton", "ImportESI", _("Import ESI file"), None),

   404

             ("AddButton", "ImportDatabase", _("Add file from ESI files database"), self.OnAddButton),

   405

             ("DeleteButton", "remove_element", _("Remove file from library"), None)])

   406

   407

        return self.ModuleLibraryEditor

   408

   409

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

   410

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

   411

   412

        self.RefreshView()

   413

   414

    def RefreshView(self):

   415

        ConfTreeNodeEditor.RefreshView(self)

   416

        self.ModuleLibraryEditor.RefreshView()

   417

   418

    def OnAddButton(self, event):

   419

        dialog = DatabaseManagementDialog(self, 

   420

            self.Controler.GetModulesDatabaseInstance())

   421

   422

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

   423

            module_library = self.Controler.GetModulesLibraryInstance()

   424

            module_library.ImportModuleLibrary(dialog.GetValue())

   425

   426

        dialog.Destroy()

   427

   428

        wx.CallAfter(self.ModuleLibraryEditor.RefreshView)

   429

   430

        event.Skip()

   431