1

import wx

     2

     3

from node import OD_Subindex, OD_MultipleSubindexes, OD_IdenticalSubindexes, OD_IdenticalIndexes

     4

from commondialogs import *

     5

     6

class NodeEditorTemplate:

     7

     8

    EDITMENU_ID = None

     9

    10

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

    11

        self.Manager = manager

    12

        self.Frame = frame

    13

        self.ModeSolo = mode_solo

    14

    15

        self.BusId = None

    16

        self.Closing = False

    17

    18

    def SetBusId(self, bus_id):

    19

        self.BusId = bus_id

    20

    21

    def GetBusId(self):

    22

        return self.BusId

    23

    24

    def IsClosing(self):

    25

        return self.Closing

    26

    27

    def OnAddSDOServerMenu(self, event):

    28

        self.Manager.AddSDOServerToCurrent()

    29

        self.RefreshBufferState()

    30

        self.RefreshCurrentIndexList()

    31

    32

    def OnAddSDOClientMenu(self, event):

    33

        self.Manager.AddSDOClientToCurrent()

    34

        self.RefreshBufferState()

    35

        self.RefreshCurrentIndexList()

    36

    37

    def OnAddPDOTransmitMenu(self, event):

    38

        self.Manager.AddPDOTransmitToCurrent()

    39

        self.RefreshBufferState()

    40

        self.RefreshCurrentIndexList()

    41

    42

    def OnAddPDOReceiveMenu(self, event):

    43

        self.Manager.AddPDOReceiveToCurrent()

    44

        self.RefreshBufferState()

    45

        self.RefreshCurrentIndexList()

    46

    47

    def OnAddMapVariableMenu(self, event):

    48

        self.AddMapVariable()

    49

    50

    def OnAddUserTypeMenu(self, event):

    51

        self.AddUserType()

    52

    53

    def OnRefreshMenu(self, event):

    54

        self.RefreshCurrentIndexList()

    55

    56

    def RefreshCurrentIndexList(self):

    57

        pass

    58

    59

    def RefreshStatusBar(self):

    60

        pass

    61

    62

    def SetStatusBarText(self, selection, manager):

    63

        if selection:

    64

            index, subIndex = selection

    65

            if manager.IsCurrentEntry(index):

    66

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

    67

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

    68

                entryinfos = manager.GetEntryInfos(index)

    69

                name = entryinfos["name"]

    70

                category = _("Optional")

    71

                if entryinfos["need"]:

    72

                    category = _("Mandatory")

    73

                struct = "VAR"

    74

                number = ""

    75

                if entryinfos["struct"] & OD_IdenticalIndexes:

    76

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

    77

                if entryinfos["struct"] & OD_IdenticalSubindexes:

    78

                    struct = "REC"

    79

                elif entryinfos["struct"] & OD_MultipleSubindexes:

    80

                    struct = "ARRAY"

    81

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

    82

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

    83

            else:

    84

                for i in xrange(3):

    85

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

    86

        else:

    87

            for i in xrange(3):

    88

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

    89

    90

    def RefreshProfileMenu(self):

    91

        if self.EDITMENU_ID is not None:

    92

            profile = self.Manager.GetCurrentProfileName()

    93

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

    94

            if edititem:

    95

                length = self.Frame.AddMenu.GetMenuItemCount()

    96

                for i in xrange(length-6):

    97

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

    98

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

    99

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

   100

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

   101

                    edititem.Enable(True)

   102

                    self.Frame.AddMenu.AppendSeparator()

   103

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

   104

                        new_id = wx.NewId()

   105

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

   106

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

   107

                else:

   108

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

   109

                    edititem.Enable(False)

   110

   111

#-------------------------------------------------------------------------------

   112

#                            Buffer Functions

   113

#-------------------------------------------------------------------------------

   114

   115

    def RefreshBufferState(self):

   116

        pass

   117

   118

    def OnUndoMenu(self, event):

   119

        self.Manager.LoadCurrentPrevious()

   120

        self.RefreshCurrentIndexList()

   121

        self.RefreshBufferState()

   122

   123

    def OnRedoMenu(self, event):

   124

        self.Manager.LoadCurrentNext()

   125

        self.RefreshCurrentIndexList()

   126

        self.RefreshBufferState()

   127

   128

#-------------------------------------------------------------------------------

   129

#                          Editing Profiles functions

   130

#-------------------------------------------------------------------------------

   131

   132

    def OnCommunicationMenu(self, event):

   133

        dictionary,current = self.Manager.GetCurrentCommunicationLists()

   134

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

   135

   136

    def OnOtherCommunicationMenu(self, event):

   137

        dictionary,current = self.Manager.GetCurrentDS302Lists()

   138

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

   139

   140

    def OnEditProfileMenu(self, event):

   141

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

   142

        dictionary,current = self.Manager.GetCurrentProfileLists()

   143

        self.EditProfile(title, dictionary, current)

   144

   145

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

   146

        dialog = CommunicationDialog(self.Frame)

   147

        dialog.SetTitle(title)

   148

        dialog.SetIndexDictionary(dictionary)

   149

        dialog.SetCurrentList(current)

   150

        dialog.RefreshLists()

   151

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

   152

            new_profile = dialog.GetCurrentList()

   153

            addinglist = []

   154

            removinglist = []

   155

            for index in new_profile:

   156

                if index not in current:

   157

                    addinglist.append(index)

   158

            for index in current:

   159

                if index not in new_profile:

   160

                    removinglist.append(index)

   161

            self.Manager.ManageEntriesOfCurrent(addinglist, removinglist)

   162

            self.Manager.BufferCurrentNode()

   163

            self.RefreshBufferState()

   164

   165

        dialog.Destroy()

   166

   167

    def GetProfileCallBack(self, text):

   168

        def ProfileCallBack(event):

   169

            self.Manager.AddSpecificEntryToCurrent(text)

   170

            self.RefreshBufferState()

   171

            self.RefreshCurrentIndexList()

   172

        return ProfileCallBack

   173

   174

#-------------------------------------------------------------------------------

   175

#                         Edit Node informations function

   176

#-------------------------------------------------------------------------------

   177

   178

    def OnNodeInfosMenu(self, event):

   179

        dialog = NodeInfosDialog(self.Frame)

   180

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

   181

        defaultstringsize = self.Manager.GetCurrentNodeDefaultStringSize()

   182

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

   183

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

   184

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

   185

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

   186

            self.Manager.SetCurrentNodeDefaultStringSize(defaultstringsize)

   187

            self.RefreshBufferState()

   188

            self.RefreshCurrentIndexList()

   189

            self.RefreshProfileMenu()

   190

   191

#-------------------------------------------------------------------------------

   192

#                           Add User Types and Variables

   193

#-------------------------------------------------------------------------------

   194

   195

    def AddMapVariable(self):

   196

        index = self.Manager.GetCurrentNextMapIndex()

   197

        if index:

   198

            dialog = MapVariableDialog(self.Frame)

   199

            dialog.SetIndex(index)

   200

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

   201

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

   202

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

   203

                    self.RefreshBufferState()

   204

                    self.RefreshCurrentIndexList()

   205

                else:

   206

                    self.ShowErrorMessage(result)

   207

            dialog.Destroy()

   208

        else:

   209

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

   210

   211

    def AddUserType(self):

   212

        dialog = UserTypeDialog(self)

   213

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

   214

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

   215

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

   216

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

   217

                self.RefreshBufferState()

   218

                self.RefreshCurrentIndexList()

   219

            else:

   220

                self.ShowErrorMessage(result)

   221

        dialog.Destroy()

   222

   223

    def ShowErrorMessage(self, message):

   224

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

   225

        message.ShowModal()

   226

        message.Destroy()