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