objdictgen/networkedit.py

   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 import os, re, platform, sys, time, traceback, getopt
  29
  30 __version__ = "$Revision: 1.17 $"
  31
  32 from nodelist import *
  33 from nodemanager import *
  34 from subindextable import *
  35 from commondialogs import *
  36 from doc_index.DS301_index import *
  37
  38 ScriptDirectory = os.path.split(os.path.realpath(__file__))[0]
  39
  40 try:
  41     import wx.html
  42
  43     EVT_HTML_URL_CLICK = wx.NewId()
  44
  45     class HtmlWindowUrlClick(wx.PyEvent):
  46         def __init__(self, linkinfo):
  47             wx.PyEvent.__init__(self)
  48             self.SetEventType(EVT_HTML_URL_CLICK)
  49             self.linkinfo = (linkinfo.GetHref(), linkinfo.GetTarget())
  50
  51     class UrlClickHtmlWindow(wx.html.HtmlWindow):
  52         """ HTML window that generates and OnLinkClicked event.
  53
  54         Use this to avoid having to override HTMLWindow
  55         """
  56         def OnLinkClicked(self, linkinfo):
  57             wx.PostEvent(self, HtmlWindowUrlClick(linkinfo))
  58
  59         def Bind(self, event, handler, source=None, id=wx.ID_ANY, id2=wx.ID_ANY):
  60             if event == HtmlWindowUrlClick:
  61                 self.Connect(-1, -1, EVT_HTML_URL_CLICK, handler)
  62             else:
  63                 wx.html.HtmlWindow.Bind(event, handler, source=source, id=id, id2=id2)
  64
  65 #-------------------------------------------------------------------------------
  66 #                                Html Frame
  67 #-------------------------------------------------------------------------------
  68
  69     [ID_HTMLFRAME, ID_HTMLFRAMEHTMLCONTENT] = [wx.NewId() for _init_ctrls in range(2)]
  70
  71     class HtmlFrame(wx.Frame):
  72         def _init_ctrls(self, prnt):
  73             # generated method, don't edit
  74             wx.Frame.__init__(self, id=ID_HTMLFRAME, name='HtmlFrame',
  75                   parent=prnt, pos=wx.Point(320, 231), size=wx.Size(853, 616),
  76                   style=wx.DEFAULT_FRAME_STYLE, title='')
  77             self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)
  78
  79             self.HtmlContent = UrlClickHtmlWindow(id=ID_HTMLFRAMEHTMLCONTENT,
  80                   name='HtmlContent', parent=self, pos=wx.Point(0, 0),
  81                   size=wx.Size(-1, -1), style=wx.html.HW_SCROLLBAR_AUTO|wx.html.HW_NO_SELECTION)
  82             self.HtmlContent.Bind(HtmlWindowUrlClick, self.OnLinkClick)
  83
  84         def __init__(self, parent, opened):
  85             self._init_ctrls(parent)
  86             self.HtmlFrameOpened = opened
  87
  88         def SetHtmlCode(self, htmlcode):
  89             self.HtmlContent.SetPage(htmlcode)
  90
  91         def SetHtmlPage(self, htmlpage):
  92             self.HtmlContent.LoadPage(htmlpage)
  93
  94         def OnCloseFrame(self, event):
  95             self.HtmlFrameOpened.remove(self.GetTitle())
  96             event.Skip()
  97
  98         def OnLinkClick(self, event):
  99             url = event.linkinfo[0]
 100             try:
 101                 import webbrowser
 102             except ImportError:
 103                 wx.MessageBox('Please point your browser at: %s' % url)
 104             else:
 105                 webbrowser.open(url)
 106
 107     Html_Window = True
 108 except:
 109     Html_Window = False
 110
 111
 112 [ID_NETWORKEDIT, ID_NETWORKEDITNETWORKNODES,
 113  ID_NETWORKEDITHELPBAR,
 114 ] = [wx.NewId() for _init_ctrls in range(3)]
 115
 116 [ID_NETWORKEDITADDMENUITEMS0, ID_NETWORKEDITADDMENUITEMS1,
 117  ID_NETWORKEDITADDMENUITEMS2, ID_NETWORKEDITADDMENUITEMS3,
 118  ID_NETWORKEDITADDMENUITEMS4, ID_NETWORKEDITADDMENUITEMS5,
 119 ] = [wx.NewId() for _init_coll_AddMenu_Items in range(6)]
 120
 121 [ID_NETWORKEDITFILEMENUITEMS0, ID_NETWORKEDITFILEMENUITEMS1,
 122  ID_NETWORKEDITFILEMENUITEMS2, ID_NETWORKEDITFILEMENUITEMS4,
 123  ID_NETWORKEDITFILEMENUITEMS5, ID_NETWORKEDITFILEMENUITEMS6,
 124 ] = [wx.NewId() for _init_coll_FileMenu_Items in range(6)]
 125
 126 [ID_NETWORKEDITNETWORKMENUITEMS0, ID_NETWORKEDITNETWORKMENUITEMS1,
 127  ID_NETWORKEDITNETWORKMENUITEMS3,
 128 ] = [wx.NewId() for _init_coll_AddMenu_Items in range(3)]
 129
 130
 131 [ID_NETWORKEDITEDITMENUITEMS0, ID_NETWORKEDITEDITMENUITEMS1,
 132  ID_NETWORKEDITEDITMENUITEMS2, ID_NETWORKEDITEDITMENUITEMS4,
 133  ID_NETWORKEDITEDITMENUITEMS6, ID_NETWORKEDITEDITMENUITEMS7,
 134  ID_NETWORKEDITEDITMENUITEMS8,
 135 ] = [wx.NewId() for _init_coll_EditMenu_Items in range(7)]
 136
 137 [ID_NETWORKEDITHELPMENUITEMS0, ID_NETWORKEDITHELPMENUITEMS1,
 138  ID_NETWORKEDITHELPMENUITEMS2,
 139 ] = [wx.NewId() for _init_coll_HelpMenu_Items in range(3)]
 140
 141 class networkedit(wx.Frame):
 142     def _init_coll_menuBar1_Menus(self, parent):
 143         if self.ModeSolo:
 144             parent.Append(menu=self.FileMenu, title='File')
 145         parent.Append(menu=self.NetworkMenu, title='Network')
 146         parent.Append(menu=self.EditMenu, title='Edit')
 147         parent.Append(menu=self.AddMenu, title='Add')
 148         parent.Append(menu=self.HelpMenu, title='Help')
 149
 150     def _init_coll_EditMenu_Items(self, parent):
 151         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS4,
 152               kind=wx.ITEM_NORMAL, text='Refresh\tCTRL+R')
 153         parent.AppendSeparator()
 154         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS1,
 155               kind=wx.ITEM_NORMAL, text='Undo\tCTRL+Z')
 156         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS0,
 157               kind=wx.ITEM_NORMAL, text='Redo\tCTRL+Y')
 158         parent.AppendSeparator()
 159         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS6,
 160               kind=wx.ITEM_NORMAL, text='Node infos')
 161         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS2,
 162               kind=wx.ITEM_NORMAL, text='DS-301 Profile')
 163         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS8,
 164               kind=wx.ITEM_NORMAL, text='DS-302 Profile')
 165         parent.Append(help='', id=ID_NETWORKEDITEDITMENUITEMS7,
 166               kind=wx.ITEM_NORMAL, text='Other Profile')
 167         self.Bind(wx.EVT_MENU, self.OnUndoMenu,
 168               id=ID_NETWORKEDITEDITMENUITEMS1)
 169         self.Bind(wx.EVT_MENU, self.OnRedoMenu,
 170               id=ID_NETWORKEDITEDITMENUITEMS0)
 171         self.Bind(wx.EVT_MENU, self.OnCommunicationMenu,
 172               id=ID_NETWORKEDITEDITMENUITEMS2)
 173         self.Bind(wx.EVT_MENU, self.OnRefreshMenu,
 174               id=ID_NETWORKEDITEDITMENUITEMS4)
 175         self.Bind(wx.EVT_MENU, self.OnNodeInfosMenu,
 176               id=ID_NETWORKEDITEDITMENUITEMS6)
 177         self.Bind(wx.EVT_MENU, self.OnEditProfileMenu,
 178               id=ID_NETWORKEDITEDITMENUITEMS7)
 179         self.Bind(wx.EVT_MENU, self.OnOtherCommunicationMenu,
 180               id=ID_NETWORKEDITEDITMENUITEMS8)
 181
 182     def _init_coll_HelpMenu_Items(self, parent):
 183         parent.Append(help='', id=ID_NETWORKEDITHELPMENUITEMS0,
 184               kind=wx.ITEM_NORMAL, text='DS-301 Standard\tF1')
 185         self.Bind(wx.EVT_MENU, self.OnHelpDS301Menu,
 186               id=ID_NETWORKEDITHELPMENUITEMS0)
 187         parent.Append(help='', id=ID_NETWORKEDITHELPMENUITEMS1,
 188               kind=wx.ITEM_NORMAL, text='CAN Festival Docs\tF2')
 189         self.Bind(wx.EVT_MENU, self.OnHelpCANFestivalMenu,
 190               id=ID_NETWORKEDITHELPMENUITEMS1)
 191         if Html_Window and self.ModeSolo:
 192             parent.Append(help='', id=ID_NETWORKEDITHELPMENUITEMS2,
 193                   kind=wx.ITEM_NORMAL, text='About')
 194             self.Bind(wx.EVT_MENU, self.OnAboutMenu,
 195                   id=ID_NETWORKEDITHELPMENUITEMS2)
 196
 197     def _init_coll_FileMenu_Items(self, parent):
 198         parent.Append(help='', id=ID_NETWORKEDITFILEMENUITEMS5,
 199               kind=wx.ITEM_NORMAL, text='New\tCTRL+N')
 200         parent.Append(help='', id=ID_NETWORKEDITFILEMENUITEMS0,
 201               kind=wx.ITEM_NORMAL, text='Open\tCTRL+O')
 202         parent.Append(help='', id=ID_NETWORKEDITFILEMENUITEMS1,
 203               kind=wx.ITEM_NORMAL, text='Save\tCTRL+S')
 204         parent.Append(help='', id=ID_NETWORKEDITFILEMENUITEMS2,
 205               kind=wx.ITEM_NORMAL, text='Close\tCTRL+W')
 206         parent.AppendSeparator()
 207         parent.Append(help='', id=ID_NETWORKEDITFILEMENUITEMS4,
 208               kind=wx.ITEM_NORMAL, text='Exit')
 209         self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu,
 210               id=ID_NETWORKEDITFILEMENUITEMS0)
 211         self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu,
 212               id=ID_NETWORKEDITFILEMENUITEMS1)
 213         self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu,
 214               id=ID_NETWORKEDITFILEMENUITEMS2)
 215         self.Bind(wx.EVT_MENU, self.OnQuitMenu,
 216               id=ID_NETWORKEDITFILEMENUITEMS4)
 217         self.Bind(wx.EVT_MENU, self.OnNewProjectMenu,
 218               id=ID_NETWORKEDITFILEMENUITEMS5)
 219
 220     def _init_coll_NetworkMenu_Items(self, parent):
 221         parent.Append(help='', id=ID_NETWORKEDITNETWORKMENUITEMS0,
 222               kind=wx.ITEM_NORMAL, text='Add Slave Node')
 223         parent.Append(help='', id=ID_NETWORKEDITNETWORKMENUITEMS1,
 224               kind=wx.ITEM_NORMAL, text='Remove Slave Node')
 225         parent.AppendSeparator()
 226         parent.Append(help='', id=ID_NETWORKEDITNETWORKMENUITEMS3,
 227               kind=wx.ITEM_NORMAL, text='Build Master Dictionary')
 228         self.Bind(wx.EVT_MENU, self.OnAddSlaveMenu,
 229               id=ID_NETWORKEDITNETWORKMENUITEMS0)
 230         self.Bind(wx.EVT_MENU, self.OnRemoveSlaveMenu,
 231               id=ID_NETWORKEDITNETWORKMENUITEMS1)
 232 ##        self.Bind(wx.EVT_MENU, self.OnBuildMasterMenu,
 233 ##              id=ID_NETWORKEDITNETWORKMENUITEMS3)
 234
 235     def _init_coll_AddMenu_Items(self, parent):
 236         parent.Append(help='', id=ID_NETWORKEDITADDMENUITEMS0,
 237               kind=wx.ITEM_NORMAL, text='SDO Server')
 238         parent.Append(help='', id=ID_NETWORKEDITADDMENUITEMS1,
 239               kind=wx.ITEM_NORMAL, text='SDO Client')
 240         parent.Append(help='', id=ID_NETWORKEDITADDMENUITEMS2,
 241               kind=wx.ITEM_NORMAL, text='PDO Transmit')
 242         parent.Append(help='', id=ID_NETWORKEDITADDMENUITEMS3,
 243               kind=wx.ITEM_NORMAL, text='PDO Receive')
 244         parent.Append(help='', id=ID_NETWORKEDITADDMENUITEMS4,
 245               kind=wx.ITEM_NORMAL, text='Map Variable')
 246         parent.Append(help='', id=ID_NETWORKEDITADDMENUITEMS5,
 247               kind=wx.ITEM_NORMAL, text='User Type')
 248         self.Bind(wx.EVT_MENU, self.OnAddSDOServerMenu,
 249               id=ID_NETWORKEDITADDMENUITEMS0)
 250         self.Bind(wx.EVT_MENU, self.OnAddSDOClientMenu,
 251               id=ID_NETWORKEDITADDMENUITEMS1)
 252         self.Bind(wx.EVT_MENU, self.OnAddPDOTransmitMenu,
 253               id=ID_NETWORKEDITADDMENUITEMS2)
 254         self.Bind(wx.EVT_MENU, self.OnAddPDOReceiveMenu,
 255               id=ID_NETWORKEDITADDMENUITEMS3)
 256         self.Bind(wx.EVT_MENU, self.OnAddMapVariableMenu,
 257               id=ID_NETWORKEDITADDMENUITEMS4)
 258         self.Bind(wx.EVT_MENU, self.OnAddUserTypeMenu,
 259               id=ID_NETWORKEDITADDMENUITEMS5)
 260
 261     def _init_coll_HelpBar_Fields(self, parent):
 262         parent.SetFieldsCount(3)
 263
 264         parent.SetStatusText(number=0, text='')
 265         parent.SetStatusText(number=1, text='')
 266         parent.SetStatusText(number=2, text='')
 267
 268         parent.SetStatusWidths([100, 110, -1])
 269
 270     def _init_utils(self):
 271         self.menuBar1 = wx.MenuBar()
 272         self.menuBar1.SetEvtHandlerEnabled(True)
 273
 274         if self.ModeSolo:
 275             self.FileMenu = wx.Menu(title='')
 276         self.NetworkMenu = wx.Menu(title='')
 277         self.EditMenu = wx.Menu(title='')
 278         self.AddMenu = wx.Menu(title='')
 279         self.HelpMenu = wx.Menu(title='')
 280
 281         self._init_coll_menuBar1_Menus(self.menuBar1)
 282         if self.ModeSolo:
 283             self._init_coll_FileMenu_Items(self.FileMenu)
 284         self._init_coll_NetworkMenu_Items(self.NetworkMenu)
 285         self._init_coll_EditMenu_Items(self.EditMenu)
 286         self._init_coll_AddMenu_Items(self.AddMenu)
 287         self._init_coll_HelpMenu_Items(self.HelpMenu)
 288
 289     def _init_ctrls(self, prnt):
 290         wx.Frame.__init__(self, id=ID_NETWORKEDIT, name='networkedit',
 291               parent=prnt, pos=wx.Point(149, 178), size=wx.Size(1000, 700),
 292               style=wx.DEFAULT_FRAME_STYLE, title='Networkedit')
 293         self._init_utils()
 294         self.SetClientSize(wx.Size(1000, 700))
 295         self.SetMenuBar(self.menuBar1)
 296         self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)
 297         if not self.ModeSolo:
 298             self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=ID_NETWORKEDITFILEMENUITEMS1)
 299             accel = wx.AcceleratorTable([wx.AcceleratorEntry(wx.ACCEL_CTRL, 83, ID_NETWORKEDITFILEMENUITEMS1)])
 300             self.SetAcceleratorTable(accel)
 301
 302         self.NetworkNodes = wx.Notebook(id=ID_NETWORKEDITNETWORKNODES,
 303               name='NetworkNodes', parent=self, pos=wx.Point(0, 0),
 304               size=wx.Size(0, 0), style=wx.NB_LEFT)
 305         self.NetworkNodes.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED,
 306               self.OnNodeSelectedChanged, id=ID_NETWORKEDITNETWORKNODES)
 307
 308         self.HelpBar = wx.StatusBar(id=ID_NETWORKEDITHELPBAR, name='HelpBar',
 309               parent=self, style=wx.ST_SIZEGRIP)
 310         self._init_coll_HelpBar_Fields(self.HelpBar)
 311         self.SetStatusBar(self.HelpBar)
 312
 313     def __init__(self, parent, nodelist = None, projectOpen = None):
 314         self.ModeSolo = nodelist == None
 315         self._init_ctrls(parent)
 316         self.HtmlFrameOpened = []
 317         self.BusId = None
 318
 319         if self.ModeSolo:
 320             self.Manager = NodeManager()
 321             if projectOpen:
 322                 self.NodeList = NodeList(self.Manager)
 323                 result = self.NodeList.LoadProject(projectOpen)
 324                 if not result:
 325                     self.RefreshNetworkNodes()
 326             else:
 327                 self.NodeList = None
 328         else:
 329             self.NodeList = nodelist
 330             self.Manager = self.NodeList.GetManager()
 331             self.NodeList.SetCurrentSelected(0)
 332             self.RefreshNetworkNodes()
 333             self.RefreshProfileMenu()
 334             self.NetworkNodes.SetFocus()
 335
 336         self.RefreshBufferState()
 337         self.RefreshTitle()
 338         self.RefreshMainMenu()
 339
 340     def SetBusId(self, bus_id):
 341         self.BusId = bus_id
 342
 343     def GetBusId(self):
 344         return self.BusId
 345
 346     def GetCurrentNodeId(self):
 347         selected = self.NetworkNodes.GetSelection()
 348         # At init selected = -1
 349         if selected > 0:
 350             window = self.NetworkNodes.GetPage(selected)
 351             return window.GetIndex()
 352         else:
 353             return 0
 354
 355     def OnCloseFrame(self, event):
 356         if not self.ModeSolo and getattr(self, "_onclose", None) != None:
 357             self._onclose()
 358         event.Skip()
 359
 360     def OnChar(self, event):
 361         if event.ControlDown() and event.GetKeyCode() == 83 and getattr(self, "_onsave", None) != None:
 362             self._onsave()
 363         event.Skip()
 364
 365     def OnQuitMenu(self, event):
 366         self.Close()
 367         event.Skip()
 368
 369     def OnAddSDOServerMenu(self, event):
 370         self.Manager.AddSDOServerToCurrent()
 371         self.RefreshBufferState()
 372         self.RefreshCurrentIndexList()
 373         event.Skip()
 374
 375     def OnAddSDOClientMenu(self, event):
 376         self.Manager.AddSDOClientToCurrent()
 377         self.RefreshBufferState()
 378         self.RefreshCurrentIndexList()
 379         event.Skip()
 380
 381     def OnAddPDOTransmitMenu(self, event):
 382         self.Manager.AddPDOTransmitToCurrent()
 383         self.RefreshBufferState()
 384         self.RefreshCurrentIndexList()
 385         event.Skip()
 386
 387     def OnAddPDOReceiveMenu(self, event):
 388         self.Manager.AddPDOReceiveToCurrent()
 389         self.RefreshBufferState()
 390         self.RefreshCurrentIndexList()
 391         event.Skip()
 392
 393     def OnAddMapVariableMenu(self, event):
 394         self.AddMapVariable()
 395         event.Skip()
 396
 397     def OnAddUserTypeMenu(self, event):
 398         self.AddUserType()
 399         event.Skip()
 400
 401     def OnNodeSelectedChanged(self, event):
 402         selected = event.GetSelection()
 403         # At init selected = -1
 404         if selected >= 0:
 405             window = self.NetworkNodes.GetPage(selected)
 406             self.NodeList.SetCurrentSelected(window.GetIndex())
 407         wx.CallAfter(self.RefreshMainMenu)
 408         wx.CallAfter(self.RefreshStatusBar)
 409         event.Skip()
 410
 411 #-------------------------------------------------------------------------------
 412 #                         Load and Save Funtions
 413 #-------------------------------------------------------------------------------
 414
 415     def OnNewProjectMenu(self, event):
 416         if self.NodeList:
 417             defaultpath = os.path.dirname(self.NodeList.GetRoot())
 418         else:
 419             defaultpath = os.getcwd()
 420         dialog = wx.DirDialog(self , "Choose a project", defaultpath, wx.DD_NEW_DIR_BUTTON)
 421         if dialog.ShowModal() == wx.ID_OK:
 422             projectpath = dialog.GetPath()
 423             if os.path.isdir(projectpath) and len(os.listdir(projectpath)) == 0:
 424                 manager = NodeManager()
 425                 nodelist = NodeList(manager)
 426                 result = nodelist.LoadProject(projectpath)
 427                 if not result:
 428                     self.Manager = manager
 429                     self.NodeList = nodelist
 430                     self.NodeList.SetCurrentSelected(0)
 431
 432                     self.RefreshNetworkNodes()
 433                     self.RefreshBufferState()
 434                     self.RefreshTitle()
 435                     self.RefreshProfileMenu()
 436                     self.RefreshMainMenu()
 437                 else:
 438                     message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
 439                     message.ShowModal()
 440                     message.Destroy()
 441         event.Skip()
 442
 443     def OnOpenProjectMenu(self, event):
 444         if self.NodeList:
 445             defaultpath = os.path.dirname(self.NodeList.GetRoot())
 446         else:
 447             defaultpath = os.getcwd()
 448         dialog = wx.DirDialog(self , "Choose a project", defaultpath, 0)
 449         if dialog.ShowModal() == wx.ID_OK:
 450             projectpath = dialog.GetPath()
 451             if os.path.isdir(projectpath):
 452                 manager = NodeManager()
 453                 nodelist = NodeList(manager)
 454                 result = nodelist.LoadProject(projectpath)
 455                 if not result:
 456                     self.Manager = manager
 457                     self.NodeList = nodelist
 458                     self.NodeList.SetCurrentSelected(0)
 459
 460                     self.RefreshNetworkNodes()
 461                     self.RefreshBufferState()
 462                     self.RefreshTitle()
 463                     self.RefreshProfileMenu()
 464                     self.RefreshMainMenu()
 465                 else:
 466                     message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 467                     message.ShowModal()
 468                     message.Destroy()
 469         dialog.Destroy()
 470         event.Skip()
 471
 472     def OnSaveProjectMenu(self, event):
 473         if not self.ModeSolo and getattr(self, "_onsave", None) != None:
 474             self._onsave()
 475         else:
 476             result = self.NodeList.SaveProject()
 477             if result:
 478                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 479                 message.ShowModal()
 480                 message.Destroy()
 481         event.Skip()
 482
 483     def OnCloseProjectMenu(self, event):
 484         if self.NodeList:
 485             if self.NodeList.HasChanged():
 486                 dialog = wx.MessageDialog(self, "There are changes, do you want to save?",  "Close Project", wx.YES_NO|wx.CANCEL|wx.ICON_QUESTION)
 487                 answer = dialog.ShowModal()
 488                 dialog.Destroy()
 489                 if answer == wx.ID_YES:
 490                     result = self.NodeList.SaveProject()
 491                     if result:
 492                         message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 493                         message.ShowModal()
 494                         message.Destroy()
 495                 elif answer == wx.ID_NO:
 496                     self.NodeList.ForceChanged(False)
 497             if not self.NodeList.HasChanged():
 498                 self.Manager = None
 499                 self.NodeList = None
 500                 self.RefreshNetworkNodes()
 501                 self.RefreshTitle()
 502                 self.RefreshMainMenu()
 503         event.Skip()
 504
 505 #-------------------------------------------------------------------------------
 506 #                             Slave Nodes Management
 507 #-------------------------------------------------------------------------------
 508
 509     def OnAddSlaveMenu(self, event):
 510         dialog = AddSlaveDialog(self)
 511         dialog.SetNodeList(self.NodeList)
 512         if dialog.ShowModal() == wx.ID_OK:
 513             values = dialog.GetValues()
 514             result = self.NodeList.AddSlaveNode(values["slaveName"], values["slaveNodeID"], values["edsFile"])
 515             if not result:
 516                 new_editingpanel = EditingPanel(self.NetworkNodes, self, self.NodeList, False)
 517                 new_editingpanel.SetIndex(values["slaveNodeID"])
 518                 idx = self.NodeList.GetOrderNumber(values["slaveNodeID"])
 519                 self.NetworkNodes.InsertPage(idx, new_editingpanel, "")
 520                 self.NodeList.SetCurrentSelected(idx)
 521                 self.NetworkNodes.SetSelection(idx)
 522                 self.RefreshBufferState()
 523             else:
 524                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 525                 message.ShowModal()
 526                 message.Destroy()
 527         dialog.Destroy()
 528         event.Skip()
 529
 530     def OnRemoveSlaveMenu(self, event):
 531         slavenames = self.NodeList.GetSlaveNames()
 532         slaveids = self.NodeList.GetSlaveIDs()
 533         dialog = wx.SingleChoiceDialog(self, "Choose a slave to remove", "Remove slave", slavenames)
 534         if dialog.ShowModal() == wx.ID_OK:
 535             choice = dialog.GetSelection()
 536             result = self.NodeList.RemoveSlaveNode(slaveids[choice])
 537             if not result:
 538                 slaveids.pop(choice)
 539                 current = self.NetworkNodes.GetSelection()
 540                 self.NetworkNodes.DeletePage(choice + 1)
 541                 if self.NetworkNodes.GetPageCount() > 0:
 542                     new_selection = min(current, self.NetworkNodes.GetPageCount() - 1)
 543                     self.NetworkNodes.SetSelection(new_selection)
 544                     if new_selection > 0:
 545                         self.NodeList.SetCurrentSelected(slaveids[new_selection - 1])
 546                     self.RefreshBufferState()
 547             else:
 548                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 549                 message.ShowModal()
 550                 message.Destroy()
 551         event.Skip()
 552
 553 #-------------------------------------------------------------------------------
 554 #                             Refresh Functions
 555 #-------------------------------------------------------------------------------
 556
 557     def RefreshTitle(self):
 558         if self.NodeList != None:
 559             self.SetTitle("Networkedit - %s"%self.NodeList.GetNetworkName())
 560         else:
 561             self.SetTitle("Networkedit")
 562
 563     def OnRefreshMenu(self, event):
 564         self.RefreshCurrentIndexList()
 565         event.Skip()
 566
 567     def RefreshCurrentIndexList(self):
 568         selected = self.NetworkNodes.GetSelection()
 569         if selected == 0:
 570             window = self.NetworkNodes.GetPage(selected)
 571             window.RefreshIndexList()
 572         else:
 573             pass
 574
 575     def RefreshNetworkNodes(self):
 576         if self.NetworkNodes.GetPageCount() > 0:
 577             self.NetworkNodes.DeleteAllPages()
 578         if self.NodeList:
 579             new_editingpanel = EditingPanel(self.NetworkNodes, self, self.Manager)
 580             new_editingpanel.SetIndex(0)
 581             self.NetworkNodes.AddPage(new_editingpanel, "")
 582             for idx in self.NodeList.GetSlaveIDs():
 583                 new_editingpanel = EditingPanel(self.NetworkNodes, self, self.NodeList, False)
 584                 new_editingpanel.SetIndex(idx)
 585                 self.NetworkNodes.AddPage(new_editingpanel, "")
 586
 587     def RefreshStatusBar(self):
 588         if self:
 589             selected = self.NetworkNodes.GetSelection()
 590             if self.HelpBar and selected >= 0:
 591                 window = self.NetworkNodes.GetPage(selected)
 592                 selection = window.GetSelection()
 593                 if selection:
 594                     index, subIndex = selection
 595                     if self.NodeList.IsCurrentEntry(index):
 596                         self.HelpBar.SetStatusText("Index: 0x%04X"%index, 0)
 597                         self.HelpBar.SetStatusText("Subindex: 0x%02X"%subIndex, 1)
 598                         entryinfos = self.NodeList.GetEntryInfos(index)
 599                         name = entryinfos["name"]
 600                         category = "Optional"
 601                         if entryinfos["need"]:
 602                             category = "Mandatory"
 603                         struct = "VAR"
 604                         number = ""
 605                         if entryinfos["struct"] & OD_IdenticalIndexes:
 606                             number = " possibly defined %d times"%entryinfos["nbmax"]
 607                         if entryinfos["struct"] & OD_IdenticalSubindexes:
 608                             struct = "REC"
 609                         elif entryinfos["struct"] & OD_MultipleSubindexes:
 610                             struct = "ARRAY"
 611                         text = "%s: %s entry of struct %s%s."%(name,category,struct,number)
 612                         self.HelpBar.SetStatusText(text, 2)
 613                     else:
 614                         for i in xrange(3):
 615                             self.HelpBar.SetStatusText("", i)
 616                 else:
 617                     for i in xrange(3):
 618                         self.HelpBar.SetStatusText("", i)
 619
 620     def RefreshMainMenu(self):
 621         if self.menuBar1:
 622             self.NetworkMenu.Enable(ID_NETWORKEDITNETWORKMENUITEMS3, False)
 623             if self.NodeList == None:
 624                 if self.ModeSolo:
 625                     self.menuBar1.EnableTop(1, False)
 626                     self.menuBar1.EnableTop(2, False)
 627                     self.menuBar1.EnableTop(3, False)
 628                     if self.FileMenu:
 629                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS1, False)
 630                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS2, False)
 631                 else:
 632                     self.menuBar1.EnableTop(0, False)
 633                     self.menuBar1.EnableTop(1, False)
 634                     self.menuBar1.EnableTop(2, False)
 635             else:
 636                 if self.ModeSolo:
 637                     self.menuBar1.EnableTop(1, True)
 638                     if self.FileMenu:
 639                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS1, True)
 640                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS2, True)
 641                     if self.NetworkNodes.GetSelection() == 0:
 642                         self.menuBar1.EnableTop(2, True)
 643                         self.menuBar1.EnableTop(3, True)
 644                     else:
 645                         self.menuBar1.EnableTop(2, False)
 646                         self.menuBar1.EnableTop(3, False)
 647                 else:
 648                     self.menuBar1.EnableTop(0, True)
 649                     if self.NetworkNodes.GetSelection() == 0:
 650                         self.menuBar1.EnableTop(1, True)
 651                         self.menuBar1.EnableTop(2, True)
 652                     else:
 653                         self.menuBar1.EnableTop(1, False)
 654                         self.menuBar1.EnableTop(2, False)
 655
 656     def RefreshProfileMenu(self):
 657         if self.EditMenu:
 658             profile = self.Manager.GetCurrentProfileName()
 659             edititem = self.EditMenu.FindItemById(ID_NETWORKEDITEDITMENUITEMS7)
 660             if edititem:
 661                 length = self.AddMenu.GetMenuItemCount()
 662                 for i in xrange(length-6):
 663                     additem = self.AddMenu.FindItemByPosition(6)
 664                     self.AddMenu.Delete(additem.GetId())
 665                 if profile not in ("None", "DS-301"):
 666                     edititem.SetText("%s Profile"%profile)
 667                     edititem.Enable(True)
 668                     self.AddMenu.AppendSeparator()
 669                     for text, indexes in self.Manager.GetCurrentSpecificMenu():
 670                         new_id = wx.NewId()
 671                         self.AddMenu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=text)
 672                         self.Bind(wx.EVT_MENU, self.GetProfileCallBack(text), id=new_id)
 673                 else:
 674                     edititem.SetText("Other Profile")
 675                     edititem.Enable(False)
 676
 677     def GetProfileCallBack(self, text):
 678         def ProfileCallBack(event):
 679             self.Manager.AddSpecificEntryToCurrent(text)
 680             self.RefreshBufferState()
 681             self.RefreshCurrentIndexList()
 682             event.Skip()
 683         return ProfileCallBack
 684
 685 #-------------------------------------------------------------------------------
 686 #                              Buffer Functions
 687 #-------------------------------------------------------------------------------
 688
 689     def RefreshBufferState(self):
 690         if self.NodeList:
 691             nodeID = self.Manager.GetCurrentNodeID()
 692             if nodeID != None:
 693                 nodename = "0x%2.2X %s"%(nodeID, self.Manager.GetCurrentNodeName())
 694             else:
 695                 nodename = self.Manager.GetCurrentNodeName()
 696             self.NetworkNodes.SetPageText(0, nodename)
 697             for idx, name in enumerate(self.NodeList.GetSlaveNames()):
 698                 self.NetworkNodes.SetPageText(idx + 1, name)
 699             self.RefreshTitle()
 700
 701     def OnUndoMenu(self, event):
 702         self.Manager.LoadCurrentPrevious()
 703         self.RefreshCurrentIndexList()
 704         self.RefreshBufferState()
 705         event.Skip()
 706
 707     def OnRedoMenu(self, event):
 708         self.Manager.LoadCurrentNext()
 709         self.RefreshCurrentIndexList()
 710         self.RefreshBufferState()
 711         event.Skip()
 712
 713 #-------------------------------------------------------------------------------
 714 #                                Help Method
 715 #-------------------------------------------------------------------------------
 716
 717     def OnHelpDS301Menu(self, event):
 718         find_index = False
 719         selected = self.NetworkNodes.GetSelection()
 720         if selected >= 0:
 721             window = self.NetworkNodes.GetPage(selected)
 722             result = window.GetSelection()
 723             if result:
 724                 find_index = True
 725                 index, subIndex = result
 726                 result = OpenPDFDocIndex(index, ScriptDirectory)
 727                 if type(result) == StringType:
 728                     message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
 729                     message.ShowModal()
 730                     message.Destroy()
 731         if not find_index:
 732             result = OpenPDFDocIndex(None, ScriptDirectory)
 733             if type(result) == StringType:
 734                 message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
 735                 message.ShowModal()
 736                 message.Destroy()
 737         event.Skip()
 738
 739     def OnHelpCANFestivalMenu(self, event):
 740         #self.OpenHtmlFrame("CAN Festival Reference", os.path.join(ScriptDirectory, "doc/canfestival.html"), wx.Size(1000, 600))
 741         os.system("xpdf -remote CANFESTIVAL %s %d &"%(os.path.join(ScriptDirectory, "doc/manual_en.pdf"),16))
 742         event.Skip()
 743
 744     def OnAboutMenu(self, event):
 745         self.OpenHtmlFrame("About CAN Festival", os.path.join(ScriptDirectory, "doc/about.html"), wx.Size(500, 450))
 746         event.Skip()
 747
 748     def OpenHtmlFrame(self, title, file, size):
 749         if title not in self.HtmlFrameOpened:
 750             self.HtmlFrameOpened.append(title)
 751             window = HtmlFrame(self, self.HtmlFrameOpened)
 752             window.SetTitle(title)
 753             window.SetHtmlPage(file)
 754             window.SetClientSize(size)
 755             window.Show()
 756
 757 #-------------------------------------------------------------------------------
 758 #                          Editing Profiles functions
 759 #-------------------------------------------------------------------------------
 760
 761     def OnCommunicationMenu(self, event):
 762         dictionary,current = self.Manager.GetCurrentCommunicationLists()
 763         self.EditProfile("Edit DS-301 Profile", dictionary, current)
 764         event.Skip()
 765
 766     def OnOtherCommunicationMenu(self, event):
 767         dictionary,current = self.Manager.GetCurrentDS302Lists()
 768         self.EditProfile("Edit DS-301 Profile", dictionary, current)
 769         event.Skip()
 770
 771     def OnEditProfileMenu(self, event):
 772         title = "Edit %s Profile"%self.Manager.GetCurrentProfileName()
 773         dictionary,current = self.Manager.GetCurrentProfileLists()
 774         self.EditProfile(title, dictionary, current)
 775         event.Skip()
 776
 777     def EditProfile(self, title, dictionary, current):
 778         dialog = CommunicationDialog(self)
 779         dialog.SetTitle(title)
 780         dialog.SetIndexDictionary(dictionary)
 781         dialog.SetCurrentList(current)
 782         dialog.RefreshLists()
 783         if dialog.ShowModal() == wx.ID_OK:
 784             new_profile = dialog.GetCurrentList()
 785             addinglist = []
 786             removinglist = []
 787             for index in new_profile:
 788                 if index not in current:
 789                     addinglist.append(index)
 790             for index in current:
 791                 if index not in new_profile:
 792                     removinglist.append(index)
 793             self.Manager.ManageEntriesOfCurrent(addinglist, removinglist)
 794             self.Manager.BufferCurrentNode()
 795             self.RefreshBufferState()
 796             self.RefreshCurrentIndexList()
 797         dialog.Destroy()
 798
 799 #-------------------------------------------------------------------------------
 800 #                         Edit Node informations function
 801 #-------------------------------------------------------------------------------
 802
 803     def OnNodeInfosMenu(self, event):
 804         dialog = NodeInfosDialog(self)
 805         name, id, type, description = self.Manager.GetCurrentNodeInfos()
 806         dialog.SetValues(name, id, type, description)
 807         if dialog.ShowModal() == wx.ID_OK:
 808             name, id, type, description = dialog.GetValues()
 809             self.Manager.SetCurrentNodeInfos(name, id, type, description)
 810             self.RefreshBufferState()
 811             self.RefreshProfileMenu()
 812             selected = self.NetworkNodes.GetSelection()
 813             if selected >= 0:
 814                 window = self.NetworkNodes.GetPage(selected)
 815                 window.RefreshTable()
 816         event.Skip()
 817
 818
 819 #-------------------------------------------------------------------------------
 820 #                           Add User Types and Variables
 821 #-------------------------------------------------------------------------------
 822
 823     def AddMapVariable(self):
 824         index = self.Manager.GetCurrentNextMapIndex()
 825         if index:
 826             dialog = MapVariableDialog(self)
 827             dialog.SetIndex(index)
 828             if dialog.ShowModal() == wx.ID_OK:
 829                 index, name, struct, number = dialog.GetValues()
 830                 result = self.Manager.AddMapVariableToCurrent(index, name, struct, number)
 831                 if type(result) != StringType:
 832                     self.RefreshBufferState()
 833                     self.RefreshCurrentIndexList()
 834                 else:
 835                     message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 836                     message.ShowModal()
 837                     message.Destroy()
 838             dialog.Destroy()
 839         else:
 840             message = wx.MessageDialog(self, result, "No map variable index left!", wx.OK|wx.ICON_ERROR)
 841             message.ShowModal()
 842             message.Destroy()
 843
 844     def AddUserType(self):
 845         dialog = UserTypeDialog(self)
 846         dialog.SetTypeList(self.Manager.GetCustomisableTypes())
 847         if dialog.ShowModal() == wx.ID_OK:
 848             type, min, max, length = dialog.GetValues()
 849             result = self.Manager.AddUserTypeToCurrent(type, min, max, length)
 850             if not IsOfType(result, StringType):
 851                 self.RefreshBufferState()
 852                 self.RefreshCurrentIndexList()
 853             else:
 854                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 855                 message.ShowModal()
 856                 message.Destroy()
 857         dialog.Destroy()
 858
 859     def OpenMasterDCFDialog(self, node_id):
 860         self.NetworkNodes.SetSelection(0)
 861         self.NetworkNodes.GetPage(0).OpenDCFDialog(node_id)
 862
 863 #-------------------------------------------------------------------------------
 864 #                               Exception Handler
 865 #-------------------------------------------------------------------------------
 866
 867 Max_Traceback_List_Size = 20
 868
 869 def Display_Exception_Dialog(e_type,e_value,e_tb):
 870     trcbck_lst = []
 871     for i,line in enumerate(traceback.extract_tb(e_tb)):
 872         trcbck = " " + str(i+1) + ". "
 873         if line[0].find(os.getcwd()) == -1:
 874             trcbck += "file : " + str(line[0]) + ",   "
 875         else:
 876             trcbck += "file : " + str(line[0][len(os.getcwd()):]) + ",   "
 877         trcbck += "line : " + str(line[1]) + ",   " + "function : " + str(line[2])
 878         trcbck_lst.append(trcbck)
 879
 880     # Allow clicking....
 881     cap = wx.Window_GetCapture()
 882     if cap:
 883         cap.ReleaseMouse()
 884
 885     dlg = wx.SingleChoiceDialog(None,
 886         """
 887 An error happens.
 888
 889 Click on OK for saving an error report.
 890
 891 Please contact LOLITech at:
 892 +33 (0)3 29 52 95 67
 893 bugs_networkedit@lolitech.fr
 894
 895
 896 Error:
 897 """ +
 898         str(e_type) + " : " + str(e_value),
 899         "Error",
 900         trcbck_lst)
 901     try:
 902         res = (dlg.ShowModal() == wx.ID_OK)
 903     finally:
 904         dlg.Destroy()
 905
 906     return res
 907
 908 def Display_Error_Dialog(e_value):
 909     message = wx.MessageDialog(None, str(e_value), "Error", wx.OK|wx.ICON_ERROR)
 910     message.ShowModal()
 911     message.Destroy()
 912
 913 def get_last_traceback(tb):
 914     while tb.tb_next:
 915         tb = tb.tb_next
 916     return tb
 917
 918
 919 def format_namespace(d, indent='    '):
 920     return '\n'.join(['%s%s: %s' % (indent, k, repr(v)[:10000]) for k, v in d.iteritems()])
 921
 922
 923 ignored_exceptions = [] # a problem with a line in a module is only reported once per session
 924
 925 def AddExceptHook(path, app_version='[No version]'):#, ignored_exceptions=[]):
 926
 927     def handle_exception(e_type, e_value, e_traceback):
 928         traceback.print_exception(e_type, e_value, e_traceback) # this is very helpful when there's an exception in the rest of this func
 929         last_tb = get_last_traceback(e_traceback)
 930         ex = (last_tb.tb_frame.f_code.co_filename, last_tb.tb_frame.f_lineno)
 931         if str(e_value).startswith("!!!"):
 932             Display_Error_Dialog(e_value)
 933         elif ex not in ignored_exceptions:
 934             ignored_exceptions.append(ex)
 935             result = Display_Exception_Dialog(e_type,e_value,e_traceback)
 936             if result:
 937                 info = {
 938                     'app-title' : wx.GetApp().GetAppName(), # app_title
 939                     'app-version' : app_version,
 940                     'wx-version' : wx.VERSION_STRING,
 941                     'wx-platform' : wx.Platform,
 942                     'python-version' : platform.python_version(), #sys.version.split()[0],
 943                     'platform' : platform.platform(),
 944                     'e-type' : e_type,
 945                     'e-value' : e_value,
 946                     'date' : time.ctime(),
 947                     'cwd' : os.getcwd(),
 948                     }
 949                 if e_traceback:
 950                     info['traceback'] = ''.join(traceback.format_tb(e_traceback)) + '%s: %s' % (e_type, e_value)
 951                     last_tb = get_last_traceback(e_traceback)
 952                     exception_locals = last_tb.tb_frame.f_locals # the locals at the level of the stack trace where the exception actually occurred
 953                     info['locals'] = format_namespace(exception_locals)
 954                     if 'self' in exception_locals:
 955                         info['self'] = format_namespace(exception_locals['self'].__dict__)
 956
 957                 output = open(path+os.sep+"bug_report_"+info['date'].replace(':','-').replace(' ','_')+".txt",'w')
 958                 lst = info.keys()
 959                 lst.sort()
 960                 for a in lst:
 961                     output.write(a+":\n"+str(info[a])+"\n\n")
 962
 963     #sys.excepthook = lambda *args: wx.CallAfter(handle_exception, *args)
 964     sys.excepthook = handle_exception
 965
 966 if __name__ == '__main__':
 967     def usage():
 968         print "\nUsage of networkedit.py :"
 969         print "\n   %s [Projectpath]\n"%sys.argv[0]
 970
 971     try:
 972         opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
 973     except getopt.GetoptError:
 974         # print help information and exit:
 975         usage()
 976         sys.exit(2)
 977
 978     for o, a in opts:
 979         if o in ("-h", "--help"):
 980             usage()
 981             sys.exit()
 982
 983     if len(args) == 0:
 984         projectOpen = None
 985     elif len(args) == 1:
 986         projectOpen = args[0]
 987     else:
 988         usage()
 989         sys.exit(2)
 990
 991     app = wx.PySimpleApp()
 992     wx.InitAllImageHandlers()
 993
 994     # Install a exception handle for bug reports
 995     AddExceptHook(os.getcwd(),__version__)
 996
 997     frame = networkedit(None, projectOpen=projectOpen)
 998
 999     frame.Show()
1000     app.MainLoop()