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.10 $"
  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)
 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)
 323         if not self.ModeSolo:
 324             self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=ID_NETWORKEDITFILEMENUITEMS1)
 325             accel = wx.AcceleratorTable([wx.AcceleratorEntry(wx.ACCEL_CTRL, 83, ID_NETWORKEDITFILEMENUITEMS1)])
 326             self.SetAcceleratorTable(accel)
 327
 328         self.NetworkNodes = wx.Notebook(id=ID_NETWORKEDITNETWORKNODES,
 329               name='NetworkNodes', parent=self, pos=wx.Point(0, 0),
 330               size=wx.Size(0, 0), style=wx.NB_LEFT)
 331         self.NetworkNodes.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED,
 332               self.OnNodeSelectedChanged, id=ID_NETWORKEDITNETWORKNODES)
 333
 334         self.HelpBar = wx.StatusBar(id=ID_NETWORKEDITHELPBAR, name='HelpBar',
 335               parent=self, style=wx.ST_SIZEGRIP)
 336         self._init_coll_HelpBar_Fields(self.HelpBar)
 337         self.SetStatusBar(self.HelpBar)
 338
 339     def __init__(self, parent, nodelist = None):
 340         self.ModeSolo = nodelist == None
 341         self._init_ctrls(parent)
 342         self.HtmlFrameOpened = []
 343         self.BusId = None
 344
 345         if self.ModeSolo:
 346             self.Manager = NodeManager()
 347             if projectOpen:
 348                 self.NodeList = NodeList(self.Manager)
 349                 result = self.NodeList.LoadProject(projectOpen)
 350                 if not result:
 351                     self.RefreshNetworkNodes()
 352             else:
 353                 self.NodeList = None
 354         else:
 355             self.NodeList = nodelist
 356             self.Manager = self.NodeList.GetManager()
 357             self.NodeList.SetCurrentSelected(0)
 358             self.RefreshNetworkNodes()
 359             self.RefreshProfileMenu()
 360             self.NetworkNodes.SetFocus()
 361
 362         self.RefreshBufferState()
 363         self.RefreshTitle()
 364         self.RefreshMainMenu()
 365
 366     def SetBusId(self, bus_id):
 367         self.BusId = bus_id
 368
 369     def GetBusId(self):
 370         return self.BusId
 371
 372     def GetCurrentNodeId(self):
 373         selected = self.NetworkNodes.GetSelection()
 374         # At init selected = -1
 375         if selected > 0:
 376             window = self.NetworkNodes.GetPage(selected)
 377             return window.GetIndex()
 378         else:
 379             return 0
 380
 381     def OnCloseFrame(self, event):
 382         if not self.ModeSolo and getattr(self, "_onclose", None) != None:
 383             self._onclose()
 384         event.Skip()
 385
 386     def OnChar(self, event):
 387         if event.ControlDown() and event.GetKeyCode() == 83 and getattr(self, "_onsave", None) != None:
 388             self._onsave()
 389         event.Skip()
 390
 391     def GetNoteBook(self):
 392         return self.NetworkNodes
 393
 394     def OnQuitMenu(self, event):
 395         self.Close()
 396         event.Skip()
 397
 398     def OnAddSDOServerMenu(self, event):
 399         self.Manager.AddSDOServerToCurrent()
 400         self.RefreshBufferState()
 401         self.RefreshCurrentIndexList()
 402         event.Skip()
 403
 404     def OnAddSDOClientMenu(self, event):
 405         self.Manager.AddSDOClientToCurrent()
 406         self.RefreshBufferState()
 407         self.RefreshCurrentIndexList()
 408         event.Skip()
 409
 410     def OnAddPDOTransmitMenu(self, event):
 411         self.Manager.AddPDOTransmitToCurrent()
 412         self.RefreshBufferState()
 413         self.RefreshCurrentIndexList()
 414         event.Skip()
 415
 416     def OnAddPDOReceiveMenu(self, event):
 417         self.Manager.AddPDOReceiveToCurrent()
 418         self.RefreshBufferState()
 419         self.RefreshCurrentIndexList()
 420         event.Skip()
 421
 422     def OnAddMapVariableMenu(self, event):
 423         self.AddMapVariable()
 424         event.Skip()
 425
 426     def OnAddUserTypeMenu(self, event):
 427         self.AddUserType()
 428         event.Skip()
 429
 430     def OnNodeSelectedChanged(self, event):
 431         selected = event.GetSelection()
 432         # At init selected = -1
 433         if selected > 0:
 434             window = self.NetworkNodes.GetPage(selected)
 435             self.NodeList.SetCurrentSelected(window.GetIndex())
 436         self.RefreshMainMenu()
 437         self.RefreshStatusBar()
 438         event.Skip()
 439
 440 #-------------------------------------------------------------------------------
 441 #                         Load and Save Funtions
 442 #-------------------------------------------------------------------------------
 443
 444     def OnNewProjectMenu(self, event):
 445         if self.NodeList:
 446             defaultpath = os.path.dirname(self.NodeList.GetRoot())
 447         else:
 448             defaultpath = os.getcwd()
 449         dialog = wx.DirDialog(self , "Choose a project", defaultpath, wx.DD_NEW_DIR_BUTTON)
 450         if dialog.ShowModal() == wx.ID_OK:
 451             projectpath = dialog.GetPath()
 452             if os.path.isdir(projectpath) and len(os.listdir(projectpath)) == 0:
 453                 manager = NodeManager()
 454                 nodelist = NodeList(manager)
 455                 result = nodelist.LoadProject(projectpath)
 456                 if not result:
 457                     self.Manager = manager
 458                     self.NodeList = nodelist
 459                     self.NodeList.SetCurrentSelected(0)
 460
 461                     self.RefreshNetworkNodes()
 462                     self.RefreshBufferState()
 463                     self.RefreshTitle()
 464                     self.RefreshProfileMenu()
 465                     self.RefreshMainMenu()
 466                 else:
 467                     message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
 468                     message.ShowModal()
 469                     message.Destroy()
 470         event.Skip()
 471
 472     def OnOpenProjectMenu(self, event):
 473         if self.NodeList:
 474             defaultpath = os.path.dirname(self.NodeList.GetRoot())
 475         else:
 476             defaultpath = os.getcwd()
 477         dialog = wx.DirDialog(self , "Choose a project", defaultpath, 0)
 478         if dialog.ShowModal() == wx.ID_OK:
 479             projectpath = dialog.GetPath()
 480             if os.path.isdir(projectpath):
 481                 manager = NodeManager()
 482                 nodelist = NodeList(manager)
 483                 result = nodelist.LoadProject(projectpath)
 484                 if not result:
 485                     self.Manager = manager
 486                     self.NodeList = nodelist
 487                     self.NodeList.SetCurrentSelected(0)
 488
 489                     self.RefreshNetworkNodes()
 490                     self.RefreshBufferState()
 491                     self.RefreshTitle()
 492                     self.RefreshProfileMenu()
 493                     self.RefreshMainMenu()
 494                 else:
 495                     message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 496                     message.ShowModal()
 497                     message.Destroy()
 498         dialog.Destroy()
 499         event.Skip()
 500
 501     def OnSaveProjectMenu(self, event):
 502         if not self.ModeSolo and getattr(self, "_onsave", None) != None:
 503             self._onsave()
 504         else:
 505             result = self.NodeList.SaveProject()
 506             if result:
 507                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 508                 message.ShowModal()
 509                 message.Destroy()
 510         event.Skip()
 511
 512     def OnCloseProjectMenu(self, event):
 513         if self.NodeList:
 514             if self.NodeList.HasChanged():
 515                 dialog = wx.MessageDialog(self, "There are changes, do you want to save?",  "Close Project", wx.YES_NO|wx.CANCEL|wx.ICON_QUESTION)
 516                 answer = dialog.ShowModal()
 517                 dialog.Destroy()
 518                 if answer == wx.ID_YES:
 519                     result = self.NodeList.SaveProject()
 520                     if result:
 521                         message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 522                         message.ShowModal()
 523                         message.Destroy()
 524                 elif answer == wx.ID_NO:
 525                     self.NodeList.ForceChanged(False)
 526             if not self.NodeList.HasChanged():
 527                 self.Manager = None
 528                 self.NodeList = None
 529                 self.RefreshNetworkNodes()
 530                 self.RefreshTitle()
 531                 self.RefreshMainMenu()
 532         event.Skip()
 533
 534 #-------------------------------------------------------------------------------
 535 #                             Slave Nodes Management
 536 #-------------------------------------------------------------------------------
 537
 538     def OnAddSlaveMenu(self, event):
 539         dialog = AddSlaveDialog(self)
 540         dialog.SetNodeList(self.NodeList)
 541         if dialog.ShowModal() == wx.ID_OK:
 542             values = dialog.GetValues()
 543             result = self.NodeList.AddSlaveNode(values["slaveName"], values["slaveNodeID"], values["edsFile"])
 544             if not result:
 545                 new_editingpanel = EditingPanel(self, self.NodeList, False)
 546                 new_editingpanel.SetIndex(values["slaveNodeID"])
 547                 idx = self.NodeList.GetOrderNumber(values["slaveNodeID"])
 548                 self.NetworkNodes.InsertPage(idx, new_editingpanel, "")
 549                 self.NodeList.SetCurrentSelected(idx)
 550                 self.NetworkNodes.SetSelection(idx)
 551                 self.RefreshBufferState()
 552             else:
 553                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 554                 message.ShowModal()
 555                 message.Destroy()
 556         dialog.Destroy()
 557         event.Skip()
 558
 559     def OnRemoveSlaveMenu(self, event):
 560         slavenames = self.NodeList.GetSlaveNames()
 561         slaveids = self.NodeList.GetSlaveIDs()
 562         dialog = wx.SingleChoiceDialog(self, "Choose a slave to remove", "Remove slave", slavenames)
 563         if dialog.ShowModal() == wx.ID_OK:
 564             choice = dialog.GetSelection()
 565             result = self.NodeList.RemoveSlaveNode(slaveids[choice])
 566             if not result:
 567                 slaveids.pop(choice)
 568                 current = self.NetworkNodes.GetSelection()
 569                 self.NetworkNodes.DeletePage(choice + 1)
 570                 if self.NetworkNodes.GetPageCount() > 0:
 571                     new_selection = min(current, self.NetworkNodes.GetPageCount() - 1)
 572                     self.NetworkNodes.SetSelection(new_selection)
 573                     if new_selection > 0:
 574                         self.NodeList.SetCurrentSelected(slaveids[new_selection - 1])
 575                     self.RefreshBufferState()
 576             else:
 577                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 578                 message.ShowModal()
 579                 message.Destroy()
 580         event.Skip()
 581
 582 #-------------------------------------------------------------------------------
 583 #                             Refresh Functions
 584 #-------------------------------------------------------------------------------
 585
 586     def RefreshTitle(self):
 587         if self.NodeList != None:
 588             self.SetTitle("Networkedit - %s"%self.NodeList.GetNetworkName())
 589         else:
 590             self.SetTitle("Networkedit")
 591
 592     def OnRefreshMenu(self, event):
 593         self.RefreshCurrentIndexList()
 594         event.Skip()
 595
 596     def RefreshCurrentIndexList(self):
 597         selected = self.NetworkNodes.GetSelection()
 598         if selected == 0:
 599             window = self.NetworkNodes.GetPage(selected)
 600             window.RefreshIndexList()
 601         else:
 602             pass
 603
 604     def RefreshNetworkNodes(self):
 605         if self.NetworkNodes.GetPageCount() > 0:
 606             self.NetworkNodes.DeleteAllPages()
 607         if self.NodeList:
 608             new_editingpanel = EditingPanel(self, self.Manager)
 609             new_editingpanel.SetIndex(0)
 610             self.NetworkNodes.AddPage(new_editingpanel, "")
 611             for idx in self.NodeList.GetSlaveIDs():
 612                 new_editingpanel = EditingPanel(self, self.NodeList, False)
 613                 new_editingpanel.SetIndex(idx)
 614                 self.NetworkNodes.AddPage(new_editingpanel, "")
 615
 616     def RefreshStatusBar(self):
 617         if self.HelpBar:
 618             window = self.NetworkNodes.GetPage(self.NetworkNodes.GetSelection())
 619             selection = window.GetSelection()
 620             if selection:
 621                 index, subIndex = selection
 622                 if self.NodeList.IsCurrentEntry(index):
 623                     self.HelpBar.SetStatusText("Index: 0x%04X"%index, 0)
 624                     self.HelpBar.SetStatusText("Subindex: 0x%02X"%subIndex, 1)
 625                     entryinfos = self.NodeList.GetEntryInfos(index)
 626                     name = entryinfos["name"]
 627                     category = "Optional"
 628                     if entryinfos["need"]:
 629                         category = "Mandatory"
 630                     struct = "VAR"
 631                     number = ""
 632                     if entryinfos["struct"] & OD_IdenticalIndexes:
 633                         number = " possibly defined %d times"%entryinfos["nbmax"]
 634                     if entryinfos["struct"] & OD_IdenticalSubindexes:
 635                         struct = "REC"
 636                     elif entryinfos["struct"] & OD_MultipleSubindexes:
 637                         struct = "ARRAY"
 638                     text = "%s: %s entry of struct %s%s."%(name,category,struct,number)
 639                     self.HelpBar.SetStatusText(text, 2)
 640                 else:
 641                     for i in xrange(3):
 642                         self.HelpBar.SetStatusText("", i)
 643             else:
 644                 for i in xrange(3):
 645                     self.HelpBar.SetStatusText("", i)
 646
 647     def RefreshMainMenu(self):
 648         if self.menuBar1:
 649             self.NetworkMenu.Enable(ID_NETWORKEDITNETWORKMENUITEMS3, False)
 650             if self.NodeList == None:
 651                 if self.ModeSolo:
 652                     self.menuBar1.EnableTop(1, False)
 653                     self.menuBar1.EnableTop(2, False)
 654                     self.menuBar1.EnableTop(3, False)
 655                     if self.FileMenu:
 656                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS1, False)
 657                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS2, False)
 658                 else:
 659                     self.menuBar1.EnableTop(0, False)
 660                     self.menuBar1.EnableTop(1, False)
 661                     self.menuBar1.EnableTop(2, False)
 662             else:
 663                 if self.ModeSolo:
 664                     self.menuBar1.EnableTop(1, True)
 665                     if self.FileMenu:
 666                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS1, True)
 667                         self.FileMenu.Enable(ID_NETWORKEDITFILEMENUITEMS2, True)
 668                     if self.NetworkNodes.GetSelection() == 0:
 669                         self.menuBar1.EnableTop(2, True)
 670                         self.menuBar1.EnableTop(3, True)
 671                     else:
 672                         self.menuBar1.EnableTop(2, False)
 673                         self.menuBar1.EnableTop(3, False)
 674                 else:
 675                     self.menuBar1.EnableTop(0, True)
 676                     if self.NetworkNodes.GetSelection() == 0:
 677                         self.menuBar1.EnableTop(1, True)
 678                         self.menuBar1.EnableTop(2, True)
 679                     else:
 680                         self.menuBar1.EnableTop(1, False)
 681                         self.menuBar1.EnableTop(2, False)
 682
 683     def RefreshProfileMenu(self):
 684         if self.EditMenu:
 685             profile = self.Manager.GetCurrentProfileName()
 686             edititem = self.EditMenu.FindItemById(ID_NETWORKEDITEDITMENUITEMS7)
 687             if edititem:
 688                 length = self.AddMenu.GetMenuItemCount()
 689                 for i in xrange(length-6):
 690                     additem = self.AddMenu.FindItemByPosition(6)
 691                     self.AddMenu.Delete(additem.GetId())
 692                 if profile not in ("None", "DS-301"):
 693                     edititem.SetText("%s Profile"%profile)
 694                     edititem.Enable(True)
 695                     self.AddMenu.AppendSeparator()
 696                     for text, indexes in self.Manager.GetCurrentSpecificMenu():
 697                         new_id = wx.NewId()
 698                         self.AddMenu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=text)
 699                         self.Bind(wx.EVT_MENU, self.GetProfileCallBack(text), id=new_id)
 700                 else:
 701                     edititem.SetText("Other Profile")
 702                     edititem.Enable(False)
 703
 704     def GetProfileCallBack(self, text):
 705         def ProfileCallBack(event):
 706             self.Manager.AddSpecificEntryToCurrent(text)
 707             self.RefreshBufferState()
 708             self.RefreshCurrentIndexList()
 709             event.Skip()
 710         return ProfileCallBack
 711
 712 #-------------------------------------------------------------------------------
 713 #                              Buffer Functions
 714 #-------------------------------------------------------------------------------
 715
 716     def RefreshBufferState(self):
 717         if self.NodeList:
 718             nodeID = self.Manager.GetCurrentNodeID()
 719             if nodeID != None:
 720                 nodename = "0x%2.2X %s"%(nodeID, self.Manager.GetCurrentNodeName())
 721             else:
 722                 nodename = self.Manager.GetCurrentNodeName()
 723             self.NetworkNodes.SetPageText(0, nodename)
 724             for idx, name in enumerate(self.NodeList.GetSlaveNames()):
 725                 self.NetworkNodes.SetPageText(idx + 1, name)
 726             self.RefreshTitle()
 727
 728     def OnUndoMenu(self, event):
 729         self.Manager.LoadCurrentPrevious()
 730         self.RefreshCurrentIndexList()
 731         self.RefreshBufferState()
 732         event.Skip()
 733
 734     def OnRedoMenu(self, event):
 735         self.Manager.LoadCurrentNext()
 736         self.RefreshCurrentIndexList()
 737         self.RefreshBufferState()
 738         event.Skip()
 739
 740 #-------------------------------------------------------------------------------
 741 #                                Help Method
 742 #-------------------------------------------------------------------------------
 743
 744     def OnHelpDS301Menu(self, event):
 745         find_index = False
 746         selected = self.NetworkNodes.GetSelection()
 747         if selected >= 0:
 748             window = self.NetworkNodes.GetPage(selected)
 749             result = window.GetSelection()
 750             if result:
 751                 find_index = True
 752                 index, subIndex = result
 753                 result = OpenPDFDocIndex(index, ScriptDirectory)
 754                 if type(result) == StringType:
 755                     message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
 756                     message.ShowModal()
 757                     message.Destroy()
 758         if not find_index:
 759             result = OpenPDFDocIndex(None, ScriptDirectory)
 760             if type(result) == StringType:
 761                 message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
 762                 message.ShowModal()
 763                 message.Destroy()
 764         event.Skip()
 765
 766     def OnHelpCANFestivalMenu(self, event):
 767         #self.OpenHtmlFrame("CAN Festival Reference", os.path.join(ScriptDirectory, "doc/canfestival.html"), wx.Size(1000, 600))
 768         os.system("xpdf -remote CANFESTIVAL %s %d &"%(os.path.join(ScriptDirectory, "doc/manual_en.pdf"),16))
 769         event.Skip()
 770
 771     def OnAboutMenu(self, event):
 772         self.OpenHtmlFrame("About CAN Festival", os.path.join(ScriptDirectory, "doc/about.html"), wx.Size(500, 450))
 773         event.Skip()
 774
 775     def OpenHtmlFrame(self, title, file, size):
 776         if title not in self.HtmlFrameOpened:
 777             self.HtmlFrameOpened.append(title)
 778             window = HtmlFrame(self, self.HtmlFrameOpened)
 779             window.SetTitle(title)
 780             window.SetHtmlPage(file)
 781             window.SetClientSize(size)
 782             window.Show()
 783
 784 #-------------------------------------------------------------------------------
 785 #                          Editing Profiles functions
 786 #-------------------------------------------------------------------------------
 787
 788     def OnCommunicationMenu(self, event):
 789         dictionary,current = self.Manager.GetCurrentCommunicationLists()
 790         self.EditProfile("Edit DS-301 Profile", dictionary, current)
 791         event.Skip()
 792
 793     def OnOtherCommunicationMenu(self, event):
 794         dictionary,current = self.Manager.GetCurrentDS302Lists()
 795         self.EditProfile("Edit DS-301 Profile", dictionary, current)
 796         event.Skip()
 797
 798     def OnEditProfileMenu(self, event):
 799         title = "Edit %s Profile"%self.Manager.GetCurrentProfileName()
 800         dictionary,current = self.Manager.GetCurrentProfileLists()
 801         self.EditProfile(title, dictionary, current)
 802         event.Skip()
 803
 804     def EditProfile(self, title, dictionary, current):
 805         dialog = CommunicationDialog(self)
 806         dialog.SetTitle(title)
 807         dialog.SetIndexDictionary(dictionary)
 808         dialog.SetCurrentList(current)
 809         dialog.RefreshLists()
 810         if dialog.ShowModal() == wx.ID_OK:
 811             new_profile = dialog.GetCurrentList()
 812             addinglist = []
 813             removinglist = []
 814             for index in new_profile:
 815                 if index not in current:
 816                     addinglist.append(index)
 817             for index in current:
 818                 if index not in new_profile:
 819                     removinglist.append(index)
 820             self.Manager.ManageEntriesOfCurrent(addinglist, removinglist)
 821             self.Manager.BufferCurrentNode()
 822             self.RefreshBufferState()
 823             self.RefreshCurrentIndexList()
 824         dialog.Destroy()
 825
 826 #-------------------------------------------------------------------------------
 827 #                         Edit Node informations function
 828 #-------------------------------------------------------------------------------
 829
 830     def OnNodeInfosMenu(self, event):
 831         dialog = NodeInfosDialog(self)
 832         name, id, type, description = self.Manager.GetCurrentNodeInfos()
 833         dialog.SetValues(name, id, type, description)
 834         if dialog.ShowModal() == wx.ID_OK:
 835             name, id, type, description = dialog.GetValues()
 836             self.Manager.SetCurrentNodeInfos(name, id, type, description)
 837             self.RefreshBufferState()
 838             self.RefreshProfileMenu()
 839             selected = self.NetworkNodes.GetSelection()
 840             if selected >= 0:
 841                 window = self.NetworkNodes.GetPage(selected)
 842                 window.RefreshTable()
 843         event.Skip()
 844
 845
 846 #-------------------------------------------------------------------------------
 847 #                           Add User Types and Variables
 848 #-------------------------------------------------------------------------------
 849
 850     def AddMapVariable(self):
 851         index = self.Manager.GetCurrentNextMapIndex()
 852         if index:
 853             dialog = MapVariableDialog(self)
 854             dialog.SetIndex(index)
 855             if dialog.ShowModal() == wx.ID_OK:
 856                 index, name, struct, number = dialog.GetValues()
 857                 result = self.Manager.AddMapVariableToCurrent(index, name, struct, number)
 858                 if type(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         else:
 867             message = wx.MessageDialog(self, result, "No map variable index left!", wx.OK|wx.ICON_ERROR)
 868             message.ShowModal()
 869             message.Destroy()
 870
 871     def AddUserType(self):
 872         dialog = UserTypeDialog(self)
 873         dialog.SetTypeList(self.Manager.GetCustomisableTypes())
 874         if dialog.ShowModal() == wx.ID_OK:
 875             type, min, max, length = dialog.GetValues()
 876             result = self.Manager.AddUserTypeToCurrent(type, min, max, length)
 877             if not IsOfType(result, StringType):
 878                 self.RefreshBufferState()
 879                 self.RefreshCurrentIndexList()
 880             else:
 881                 message = wx.MessageDialog(self, result, "Error", wx.OK|wx.ICON_ERROR)
 882                 message.ShowModal()
 883                 message.Destroy()
 884         dialog.Destroy()
 885
 886 #-------------------------------------------------------------------------------
 887 #                               Exception Handler
 888 #-------------------------------------------------------------------------------
 889
 890 Max_Traceback_List_Size = 20
 891
 892 def Display_Exception_Dialog(e_type,e_value,e_tb):
 893     trcbck_lst = []
 894     for i,line in enumerate(traceback.extract_tb(e_tb)):
 895         trcbck = " " + str(i+1) + ". "
 896         if line[0].find(os.getcwd()) == -1:
 897             trcbck += "file : " + str(line[0]) + ",   "
 898         else:
 899             trcbck += "file : " + str(line[0][len(os.getcwd()):]) + ",   "
 900         trcbck += "line : " + str(line[1]) + ",   " + "function : " + str(line[2])
 901         trcbck_lst.append(trcbck)
 902
 903     # Allow clicking....
 904     cap = wx.Window_GetCapture()
 905     if cap:
 906         cap.ReleaseMouse()
 907
 908     dlg = wx.SingleChoiceDialog(None,
 909         """
 910 An error happens.
 911
 912 Click on OK for saving an error report.
 913
 914 Please contact LOLITech at:
 915 +33 (0)3 29 52 95 67
 916 bugs_networkedit@lolitech.fr
 917
 918
 919 Error:
 920 """ +
 921         str(e_type) + " : " + str(e_value),
 922         "Error",
 923         trcbck_lst)
 924     try:
 925         res = (dlg.ShowModal() == wx.ID_OK)
 926     finally:
 927         dlg.Destroy()
 928
 929     return res
 930
 931 def Display_Error_Dialog(e_value):
 932     message = wx.MessageDialog(None, str(e_value), "Error", wx.OK|wx.ICON_ERROR)
 933     message.ShowModal()
 934     message.Destroy()
 935
 936 def get_last_traceback(tb):
 937     while tb.tb_next:
 938         tb = tb.tb_next
 939     return tb
 940
 941
 942 def format_namespace(d, indent='    '):
 943     return '\n'.join(['%s%s: %s' % (indent, k, repr(v)[:10000]) for k, v in d.iteritems()])
 944
 945
 946 ignored_exceptions = [] # a problem with a line in a module is only reported once per session
 947
 948 def AddExceptHook(path, app_version='[No version]'):#, ignored_exceptions=[]):
 949
 950     def handle_exception(e_type, e_value, e_traceback):
 951         traceback.print_exception(e_type, e_value, e_traceback) # this is very helpful when there's an exception in the rest of this func
 952         last_tb = get_last_traceback(e_traceback)
 953         ex = (last_tb.tb_frame.f_code.co_filename, last_tb.tb_frame.f_lineno)
 954         if str(e_value).startswith("!!!"):
 955             Display_Error_Dialog(e_value)
 956         elif ex not in ignored_exceptions:
 957             ignored_exceptions.append(ex)
 958             result = Display_Exception_Dialog(e_type,e_value,e_traceback)
 959             if result:
 960                 info = {
 961                     'app-title' : wx.GetApp().GetAppName(), # app_title
 962                     'app-version' : app_version,
 963                     'wx-version' : wx.VERSION_STRING,
 964                     'wx-platform' : wx.Platform,
 965                     'python-version' : platform.python_version(), #sys.version.split()[0],
 966                     'platform' : platform.platform(),
 967                     'e-type' : e_type,
 968                     'e-value' : e_value,
 969                     'date' : time.ctime(),
 970                     'cwd' : os.getcwd(),
 971                     }
 972                 if e_traceback:
 973                     info['traceback'] = ''.join(traceback.format_tb(e_traceback)) + '%s: %s' % (e_type, e_value)
 974                     last_tb = get_last_traceback(e_traceback)
 975                     exception_locals = last_tb.tb_frame.f_locals # the locals at the level of the stack trace where the exception actually occurred
 976                     info['locals'] = format_namespace(exception_locals)
 977                     if 'self' in exception_locals:
 978                         info['self'] = format_namespace(exception_locals['self'].__dict__)
 979
 980                 output = open(path+os.sep+"bug_report_"+info['date'].replace(':','-').replace(' ','_')+".txt",'w')
 981                 lst = info.keys()
 982                 lst.sort()
 983                 for a in lst:
 984                     output.write(a+":\n"+str(info[a])+"\n\n")
 985
 986     #sys.excepthook = lambda *args: wx.CallAfter(handle_exception, *args)
 987     sys.excepthook = handle_exception
 988
 989 if __name__ == '__main__':
 990     app = wx.PySimpleApp()
 991     wx.InitAllImageHandlers()
 992
 993     # Install a exception handle for bug reports
 994     AddExceptHook(os.getcwd(),__version__)
 995
 996     frame = networkedit(None)
 997
 998     frame.Show()
 999     app.MainLoop()