1

     2

import wx

     3

     4

from subindextable import EditingPanel

     5

from networkedit import NetworkEditorTemplate

     6

from ConfTreeNodeEditor import ConfTreeNodeEditor

     7

     8

[ID_NETWORKEDITOR, 

     9

] = [wx.NewId() for _init_ctrls in range(1)]

    10

    11

[ID_NETWORKEDITORCONFNODEMENUADDSLAVE, ID_NETWORKEDITORCONFNODEMENUREMOVESLAVE, 

    12

 ID_NETWORKEDITORCONFNODEMENUMASTER, 

    13

] = [wx.NewId() for _init_coll_ConfNodeMenu_Items in range(3)]

    14

    15

[ID_NETWORKEDITORMASTERMENUNODEINFOS, ID_NETWORKEDITORMASTERMENUDS301PROFILE,

    16

 ID_NETWORKEDITORMASTERMENUDS302PROFILE, ID_NETWORKEDITORMASTERMENUDSOTHERPROFILE,

    17

 ID_NETWORKEDITORMASTERMENUADD, 

    18

] = [wx.NewId() for _init_coll_MasterMenu_Items in range(5)]

    19

    20

[ID_NETWORKEDITORADDMENUSDOSERVER, ID_NETWORKEDITORADDMENUSDOCLIENT,

    21

 ID_NETWORKEDITORADDMENUPDOTRANSMIT, ID_NETWORKEDITORADDMENUPDORECEIVE,

    22

 ID_NETWORKEDITORADDMENUMAPVARIABLE, ID_NETWORKEDITORADDMENUUSERTYPE,

    23

] = [wx.NewId() for _init_coll_AddMenu_Items in range(6)]

    24

    25

class NetworkEditor(ConfTreeNodeEditor, NetworkEditorTemplate):

    26

    27

    ID = ID_NETWORKEDITOR

    28

    29

    def _init_coll_MainSizer_Items(self, parent):

    30

        parent.AddWindow(self.NetworkNodes, 0, border=5, flag=wx.GROW|wx.ALL)

    31

    32

    def _init_coll_MainSizer_Growables(self, parent):

    33

        parent.AddGrowableCol(0)

    34

        parent.AddGrowableRow(0)

    35

    36

    def _init_sizers(self):

    37

        self.MainSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=1, vgap=0)

    38

    39

        self._init_coll_MainSizer_Items(self.MainSizer)

    40

        self._init_coll_MainSizer_Growables(self.MainSizer)

    41

    42

        self.ConfNodeEditor.SetSizer(self.MainSizer)

    43

    44

    def _init_ConfNodeEditor(self, prnt):

    45

        self.ConfNodeEditor = wx.Panel(id=-1, parent=prnt, pos=wx.Point(0, 0), 

    46

                size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

    47

    48

        NetworkEditorTemplate._init_ctrls(self, self.ConfNodeEditor)

    49

    50

        self._init_sizers()

    51

    52

    def __init__(self, parent, controler, window):

    53

        ConfTreeNodeEditor.__init__(self, parent, controler, window)

    54

        NetworkEditorTemplate.__init__(self, controler, window, False)

    55

    56

        self.RefreshNetworkNodes()

    57

        self.RefreshBufferState()

    58

    59

    def __del__(self):

    60

        self.Controler.OnCloseEditor(self)

    61

    62

    def GetConfNodeMenuItems(self):

    63

        add_menu = [(wx.ITEM_NORMAL, (_('SDO Server'), ID_NETWORKEDITORADDMENUSDOSERVER, '', self.OnAddSDOServerMenu)),

    64

                    (wx.ITEM_NORMAL, (_('SDO Client'), ID_NETWORKEDITORADDMENUSDOCLIENT, '', self.OnAddSDOClientMenu)),

    65

                    (wx.ITEM_NORMAL, (_('PDO Transmit'), ID_NETWORKEDITORADDMENUPDOTRANSMIT, '', self.OnAddPDOTransmitMenu)),

    66

                    (wx.ITEM_NORMAL, (_('PDO Receive'), ID_NETWORKEDITORADDMENUPDORECEIVE, '', self.OnAddPDOReceiveMenu)),

    67

                    (wx.ITEM_NORMAL, (_('Map Variable'), ID_NETWORKEDITORADDMENUMAPVARIABLE, '', self.OnAddMapVariableMenu)),

    68

                    (wx.ITEM_NORMAL, (_('User Type'), ID_NETWORKEDITORADDMENUUSERTYPE, '', self.OnAddUserTypeMenu))]

    69

    70

        profile = self.Manager.GetCurrentProfileName()

    71

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

    72

            other_profile_text = _("%s Profile") % profile

    73

            add_menu.append((wx.ITEM_SEPARATOR, None))

    74

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

    75

                add_menu.append((wx.ITEM_NORMAL, (text, wx.NewId(), '', self.GetProfileCallBack(text))))

    76

        else:

    77

            other_profile_text = _('Other Profile')

    78

    79

        master_menu = [(wx.ITEM_NORMAL, (_('Node infos'), ID_NETWORKEDITORMASTERMENUNODEINFOS, '', self.OnNodeInfosMenu)),

    80

                       (wx.ITEM_NORMAL, (_('DS-301 Profile'), ID_NETWORKEDITORMASTERMENUDS301PROFILE, '', self.OnCommunicationMenu)),

    81

                       (wx.ITEM_NORMAL, (_('DS-302 Profile'), ID_NETWORKEDITORMASTERMENUDS302PROFILE, '', self.OnOtherCommunicationMenu)),

    82

                       (wx.ITEM_NORMAL, (other_profile_text, ID_NETWORKEDITORMASTERMENUDSOTHERPROFILE, '', self.OnEditProfileMenu)),

    83

                       (wx.ITEM_SEPARATOR, None),

    84

                       (add_menu, (_('Add'), ID_NETWORKEDITORMASTERMENUADD))]

    85

    86

        return [(wx.ITEM_NORMAL, (_('Add slave'), ID_NETWORKEDITORCONFNODEMENUADDSLAVE, '', self.OnAddSlaveMenu)),

    87

                (wx.ITEM_NORMAL, (_('Remove slave'), ID_NETWORKEDITORCONFNODEMENUREMOVESLAVE, '', self.OnRemoveSlaveMenu)),

    88

                (wx.ITEM_SEPARATOR, None),

    89

                (master_menu, (_('Master'), ID_NETWORKEDITORCONFNODEMENUMASTER))]

    90

    91

    def RefreshMainMenu(self):

    92

        pass

    93

    94

    def RefreshConfNodeMenu(self, confnode_menu):

    95

        confnode_menu.Enable(ID_NETWORKEDITORCONFNODEMENUMASTER, self.NetworkNodes.GetSelection() == 0)

    96

    97

    def RefreshView(self):

    98

        ConfTreeNodeEditor.RefreshView(self)

    99

        self.RefreshCurrentIndexList()

   100

   101

    def RefreshBufferState(self):

   102

        NetworkEditorTemplate.RefreshBufferState(self)

   103

        self.ParentWindow.RefreshTitle()

   104

        self.ParentWindow.RefreshFileMenu()

   105

        self.ParentWindow.RefreshEditMenu()

   106

        self.ParentWindow.RefreshPageTitles()

   107

   108

    def OnNodeSelectedChanged(self, event):

   109

        NetworkEditorTemplate.OnNodeSelectedChanged(self, event)

   110

        wx.CallAfter(self.ParentWindow.RefreshEditMenu)