1

     2

import wx

     3

     4

from subindextable import EditingPanel

     5

from networkedit import NetworkEditorTemplate

     6

from controls import EditorPanel

     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(EditorPanel, 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.Editor.SetSizer(self.MainSizer)

    43

    44

    def _init_Editor(self, prnt):

    45

        self.Editor = 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.Editor)

    49

    50

        self._init_sizers()

    51

    52

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

    53

        EditorPanel.__init__(self, parent, "", window, controler)

    54

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

    55

    56

        img = wx.Bitmap(controler.GetIconPath(), wx.BITMAP_TYPE_PNG).ConvertToImage()

    57

        self.SetIcon(wx.BitmapFromImage(img.Rescale(16, 16)))

    58

    59

        self.RefreshNetworkNodes()

    60

        self.RefreshBufferState()

    61

    62

    def __del__(self):

    63

        self.Controler.OnCloseEditor(self)

    64

    65

    def GetConfNodeMenuItems(self):

    66

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

    67

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

    68

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

    69

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

    70

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

    71

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

    72

    73

        profile = self.Manager.GetCurrentProfileName()

    74

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

    75

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

    76

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

    77

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

    78

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

    79

        else:

    80

            other_profile_text = _('Other Profile')

    81

    82

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

    83

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

    84

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

    85

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

    86

                       (wx.ITEM_SEPARATOR, None),

    87

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

    88

    89

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

    90

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

    91

                (wx.ITEM_SEPARATOR, None),

    92

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

    93

    94

    def RefreshMainMenu(self):

    95

        pass

    96

    97

    def RefreshConfNodeMenu(self, confnode_menu):

    98

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

    99

   100

    def GetTitle(self):

   101

        fullname = self.Controler.CTNFullName()

   102

        if not self.Manager.CurrentIsSaved():

   103

            return "~%s~" % fullname

   104

        return fullname

   105

   106

    def RefreshView(self):

   107

        self.RefreshCurrentIndexList()

   108

   109

    def RefreshBufferState(self):

   110

        NetworkEditorTemplate.RefreshBufferState(self)

   111

        self.ParentWindow.RefreshTitle()

   112

        self.ParentWindow.RefreshFileMenu()

   113

        self.ParentWindow.RefreshEditMenu()

   114

        self.ParentWindow.RefreshPageTitles()

   115

   116

    def OnNodeSelectedChanged(self, event):

   117

        NetworkEditorTemplate.OnNodeSelectedChanged(self, event)

   118

        wx.CallAfter(self.ParentWindow.RefreshConfNodeMenu)

   119