1 from types import *

     2 

     3 import wx

     4 

     5 from node import OD_Subindex, OD_MultipleSubindexes, OD_IdenticalSubindexes, OD_IdenticalIndexes

     6 from commondialogs import *

     7 

     8 class NodeEditorTemplate:

     9     

    10     EDITMENU_ID = None

    11     

    12     def __init__(self, manager, frame, mode_solo):

    13         self.Manager = manager

    14         self.Frame = frame

    15         self.ModeSolo = mode_solo

    16         

    17         self.BusId = None

    18         self.Closing = False

    19     

    20     def SetBusId(self, bus_id):

    21         self.BusId = bus_id

    22 

    23     def GetBusId(self):

    24         return self.BusId

    25 

    26     def IsClosing(self):

    27         return self.Closing

    28     

    29     def OnAddSDOServerMenu(self, event):

    30         self.Manager.AddSDOServerToCurrent()

    31         self.RefreshBufferState()

    32         self.RefreshCurrentIndexList()

    33         

    34     def OnAddSDOClientMenu(self, event):

    35         self.Manager.AddSDOClientToCurrent()

    36         self.RefreshBufferState()

    37         self.RefreshCurrentIndexList()

    38         

    39     def OnAddPDOTransmitMenu(self, event):

    40         self.Manager.AddPDOTransmitToCurrent()

    41         self.RefreshBufferState()

    42         self.RefreshCurrentIndexList()

    43         

    44     def OnAddPDOReceiveMenu(self, event):

    45         self.Manager.AddPDOReceiveToCurrent()

    46         self.RefreshBufferState()

    47         self.RefreshCurrentIndexList()

    48         

    49     def OnAddMapVariableMenu(self, event):

    50         self.AddMapVariable()

    51         

    52     def OnAddUserTypeMenu(self, event):

    53         self.AddUserType()

    54     

    55     def OnRefreshMenu(self, event):

    56         self.RefreshCurrentIndexList()

    57         

    58     def RefreshCurrentIndexList(self):

    59         pass

    60     

    61     def RefreshStatusBar(self):

    62         pass

    63     

    64     def SetStatusBarText(self, selection, manager):

    65         if selection:

    66             index, subIndex = selection

    67             if manager.IsCurrentEntry(index):

    68                 self.Frame.HelpBar.SetStatusText(_("Index: 0x%04X")%index, 0)

    69                 self.Frame.HelpBar.SetStatusText(_("Subindex: 0x%02X")%subIndex, 1)

    70                 entryinfos = manager.GetEntryInfos(index)

    71                 name = entryinfos["name"]

    72                 category = _("Optional")

    73                 if entryinfos["need"]:

    74                     category = _("Mandatory")

    75                 struct = "VAR"

    76                 number = ""

    77                 if entryinfos["struct"] & OD_IdenticalIndexes:

    78                     number = _(" possibly defined %d times")%entryinfos["nbmax"]

    79                 if entryinfos["struct"] & OD_IdenticalSubindexes:

    80                     struct = "REC"

    81                 elif entryinfos["struct"] & OD_MultipleSubindexes:

    82                     struct = "ARRAY"

    83                 text = _("%s: %s entry of struct %s%s.")%(name,category,struct,number)

    84                 self.Frame.HelpBar.SetStatusText(text, 2)

    85             else:

    86                 for i in xrange(3):

    87                     self.Frame.HelpBar.SetStatusText("", i)

    88         else:

    89             for i in xrange(3):

    90                 self.Frame.HelpBar.SetStatusText("", i)

    91     

    92     def RefreshProfileMenu(self):

    93         if self.EDITMENU_ID is not None:

    94             profile = self.Manager.GetCurrentProfileName()

    95             edititem = self.Frame.EditMenu.FindItemById(self.EDITMENU_ID)

    96             if edititem:

    97                 length = self.Frame.AddMenu.GetMenuItemCount()

    98                 for i in xrange(length-6):

    99                     additem = self.Frame.AddMenu.FindItemByPosition(6)

   100                     self.Frame.AddMenu.Delete(additem.GetId())

   101                 if profile not in ("None", "DS-301"):

   102                     edititem.SetText(_("%s Profile")%profile)

   103                     edititem.Enable(True)

   104                     self.Frame.AddMenu.AppendSeparator()

   105                     for text, indexes in self.Manager.GetCurrentSpecificMenu():

   106                         new_id = wx.NewId()

   107                         self.Frame.AddMenu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=text)

   108                         self.Frame.Bind(wx.EVT_MENU, self.GetProfileCallBack(text), id=new_id)

   109                 else:

   110                     edititem.SetText(_("Other Profile"))

   111                     edititem.Enable(False)

   112         

   113 #-------------------------------------------------------------------------------

   114 #                            Buffer Functions

   115 #-------------------------------------------------------------------------------

   116 

   117     def RefreshBufferState(self):

   118         pass

   119 

   120     def OnUndoMenu(self, event):

   121         self.Manager.LoadCurrentPrevious()

   122         self.RefreshCurrentIndexList()

   123         self.RefreshBufferState()

   124         

   125     def OnRedoMenu(self, event):

   126         self.Manager.LoadCurrentNext()

   127         self.RefreshCurrentIndexList()

   128         self.RefreshBufferState()

   129 

   130 #-------------------------------------------------------------------------------

   131 #                          Editing Profiles functions

   132 #-------------------------------------------------------------------------------

   133 

   134     def OnCommunicationMenu(self, event):

   135         dictionary,current = self.Manager.GetCurrentCommunicationLists()

   136         self.EditProfile(_("Edit DS-301 Profile"), dictionary, current)

   137     

   138     def OnOtherCommunicationMenu(self, event):

   139         dictionary,current = self.Manager.GetCurrentDS302Lists()

   140         self.EditProfile(_("Edit DS-302 Profile"), dictionary, current)

   141     

   142     def OnEditProfileMenu(self, event):

   143         title = _("Edit %s Profile") % self.Manager.GetCurrentProfileName()

   144         dictionary,current = self.Manager.GetCurrentProfileLists()

   145         self.EditProfile(title, dictionary, current)

   146     

   147     def EditProfile(self, title, dictionary, current):

   148         dialog = CommunicationDialog(self.Frame)

   149         dialog.SetTitle(title)

   150         dialog.SetIndexDictionary(dictionary)

   151         dialog.SetCurrentList(current)

   152         dialog.RefreshLists()

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

   154             new_profile = dialog.GetCurrentList()

   155             addinglist = []

   156             removinglist = []

   157             for index in new_profile:

   158                 if index not in current:

   159                     addinglist.append(index)

   160             for index in current:

   161                 if index not in new_profile:

   162                     removinglist.append(index)

   163             self.Manager.ManageEntriesOfCurrent(addinglist, removinglist)

   164             self.Manager.BufferCurrentNode()

   165             self.RefreshBufferState()

   166             

   167         dialog.Destroy()

   168 

   169     def GetProfileCallBack(self, text):

   170         def ProfileCallBack(event):

   171             self.Manager.AddSpecificEntryToCurrent(text)

   172             self.RefreshBufferState()

   173             self.RefreshCurrentIndexList()

   174         return ProfileCallBack

   175 

   176 #-------------------------------------------------------------------------------

   177 #                         Edit Node informations function

   178 #-------------------------------------------------------------------------------

   179 

   180     def OnNodeInfosMenu(self, event):

   181         dialog = NodeInfosDialog(self.Frame)

   182         name, id, type, description = self.Manager.GetCurrentNodeInfos()

   183         defaultstringsize = self.Manager.GetCurrentNodeDefaultStringSize()

   184         dialog.SetValues(name, id, type, description, defaultstringsize)

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

   186             name, id, type, description, defaultstringsize = dialog.GetValues()

   187             self.Manager.SetCurrentNodeInfos(name, id, type, description)

   188             self.Manager.SetCurrentNodeDefaultStringSize(defaultstringsize)

   189             self.RefreshBufferState()

   190             self.RefreshCurrentIndexList()

   191             self.RefreshProfileMenu()

   192 

   193 #-------------------------------------------------------------------------------

   194 #                           Add User Types and Variables

   195 #-------------------------------------------------------------------------------

   196         

   197     def AddMapVariable(self):

   198         index = self.Manager.GetCurrentNextMapIndex()

   199         if index:

   200             dialog = MapVariableDialog(self.Frame)

   201             dialog.SetIndex(index)

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

   203                 result = self.Manager.AddMapVariableToCurrent(*dialog.GetValues())

   204                 if not isinstance(result, (StringType, UnicodeType)):

   205                     self.RefreshBufferState()

   206                     self.RefreshCurrentIndexList()

   207                 else:

   208                     self.ShowErrorMessage(result)

   209             dialog.Destroy()

   210         else:

   211             self.ShowErrorMessage(_("No map variable index left!"))

   212         

   213     def AddUserType(self):

   214         dialog = UserTypeDialog(self)

   215         dialog.SetTypeList(self.Manager.GetCustomisableTypes())

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

   217             result = self.Manager.AddUserTypeToCurrent(*dialog.GetValues())

   218             if not isinstance(result, (StringType, UnicodeType)):

   219                 self.RefreshBufferState()

   220                 self.RefreshCurrentIndexList()

   221             else:

   222                 self.ShowErrorMessage(result)

   223         dialog.Destroy()

   224 

   225     def ShowErrorMessage(self, message):

   226         message = wx.MessageDialog(self.Frame, message, _("Error"), wx.OK|wx.ICON_ERROR)

   227         message.ShowModal()

   228         message.Destroy()