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