24

    25

import os

    26

    27

import wx

    28

    29

from plcopen.structures import LOCATIONDATATYPES

    30

from PLCControler import LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP, LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY

    31

from util.BitmapLibrary import GetBitmap

    32

    33

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

    34

#                                   Helpers

    35

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

    36

    37

def GetDirFilterChoiceOptions():

    38

    _ = lambda x : x

    39

    return [(_("All"), [LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY]), 

    40

            (_("Input"), [LOCATION_VAR_INPUT]), 

    41

            (_("Output"), [LOCATION_VAR_OUTPUT]), 

    42

            (_("Memory"), [LOCATION_VAR_MEMORY])]

    43

DIRFILTERCHOICE_OPTIONS = dict([(_(option), filter) for option, filter in GetDirFilterChoiceOptions()])

    44

    45

def GetTypeFilterChoiceOptions():

    46

    _ = lambda x : x

    47

    return [_("All"), 

    48

            _("Type and derivated"), 

    49

            _("Type strict")]

    50

    51

# turn LOCATIONDATATYPES inside-out

    52

LOCATION_SIZES = {}

    53

for size, types in LOCATIONDATATYPES.iteritems():

    54

    for type in types:

    55

        LOCATION_SIZES[type] = size

    56

    57

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

    58

#                            Browse Locations Dialog

    59

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

    60

    61

class BrowseLocationsDialog(wx.Dialog):

    62

    63

    def __init__(self, parent, var_type, controller):

    64

        wx.Dialog.__init__(self, parent,  

    65

              size=wx.Size(600, 400), title=_('Browse Locations'),

    66

              style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER)

    67

    68

        main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=10)

    69

        main_sizer.AddGrowableCol(0)

    70

        main_sizer.AddGrowableRow(1)

    71

    72

        locations_label = wx.StaticText(self, label=_('Locations available:'))

    73

        main_sizer.AddWindow(locations_label, border=20, 

    74

              flag=wx.TOP|wx.LEFT|wx.RIGHT|wx.GROW)

    75

    76

        self.LocationsTree = wx.TreeCtrl(self, 

    77

              style=wx.TR_HAS_BUTTONS|wx.TR_SINGLE|wx.SUNKEN_BORDER|wx.TR_HIDE_ROOT|wx.TR_LINES_AT_ROOT)

    78

        self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnLocationsTreeItemActivated, 

    79

                  self.LocationsTree)

    80

        main_sizer.AddWindow(self.LocationsTree, border=20, 

    81

              flag=wx.LEFT|wx.RIGHT|wx.GROW)

    82

    83

        button_gridsizer = wx.FlexGridSizer(cols=5, hgap=5, rows=1, vgap=0)

    84

        button_gridsizer.AddGrowableCol(1)

    85

        button_gridsizer.AddGrowableCol(3)

    86

        button_gridsizer.AddGrowableRow(0)

    87

        main_sizer.AddSizer(button_gridsizer, border=20, 

    88

              flag=wx.BOTTOM|wx.LEFT|wx.RIGHT|wx.GROW)

    89

    90

        direction_label = wx.StaticText(self, label=_('Direction:'))

    91

        button_gridsizer.AddWindow(direction_label,

    92

              flag=wx.ALIGN_CENTER_VERTICAL)

    93

    94

        self.DirFilterChoice = wx.ComboBox(self, size=wx.Size(0, -1), style=wx.CB_READONLY)

    95

        self.Bind(wx.EVT_COMBOBOX, self.OnFilterChoice, self.DirFilterChoice)

    96

        button_gridsizer.AddWindow(self.DirFilterChoice,

    97

              flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL)

    98

    99

        filter_label = wx.StaticText(self, label=_('Type:'))

   100

        button_gridsizer.AddWindow(filter_label,

   101

              flag=wx.ALIGN_CENTER_VERTICAL)

   102

   103

        self.TypeFilterChoice = wx.ComboBox(self, size=wx.Size(0, -1), style=wx.CB_READONLY)

   104

        self.Bind(wx.EVT_COMBOBOX, self.OnFilterChoice, self.TypeFilterChoice)

   105

        button_gridsizer.AddWindow(self.TypeFilterChoice,

   106

              flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL)

   107

   108

        button_sizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   109

        self.Bind(wx.EVT_BUTTON, self.OnOK, button_sizer.GetAffirmativeButton())

   110

        button_gridsizer.AddSizer(button_sizer, flag=wx.ALIGN_RIGHT)

   111

   112

        self.SetSizer(main_sizer)

   113

   114

        self.Controller = controller

   115

        self.VarType = var_type

   116

        self.BaseVarType = self.Controller.GetBaseType(self.VarType)

   117

        self.VarTypeSize = LOCATION_SIZES[self.BaseVarType]

   118

        self.Locations = self.Controller.GetVariableLocationTree()

   119

   120

        # Define Tree item icon list

   121

        self.TreeImageList = wx.ImageList(16, 16)

   122

        self.TreeImageDict = {}

   123

   124

        # Icons for items

   125

        for imgname, itemtype in [

   126

            ("CONFIGURATION", LOCATION_CONFNODE), 

   127

            ("RESOURCE",      LOCATION_MODULE), 

   128

            ("PROGRAM",       LOCATION_GROUP), 

   129

            ("VAR_INPUT",     LOCATION_VAR_INPUT), 

   130

            ("VAR_OUTPUT",    LOCATION_VAR_OUTPUT), 

   131

            ("VAR_LOCAL",     LOCATION_VAR_MEMORY)]:

   132

            self.TreeImageDict[itemtype]=self.TreeImageList.Add(GetBitmap(imgname))

   133

   134

        # Assign icon list to TreeCtrls

   135

        self.LocationsTree.SetImageList(self.TreeImageList)

   136

   137

        # Set a options for the choice

   138

        for option, filter in GetDirFilterChoiceOptions():

   139

            self.DirFilterChoice.Append(_(option))

   140

        self.DirFilterChoice.SetStringSelection(_("All"))

   141

        for option in GetTypeFilterChoiceOptions():

   142

            self.TypeFilterChoice.Append(_(option))

   143

        self.TypeFilterChoice.SetStringSelection(_("All"))

   144

        self.RefreshFilters()

   145

   146

        self.RefreshLocationsTree()

   147

   148

    def RefreshFilters(self):

   149

        self.DirFilter = DIRFILTERCHOICE_OPTIONS[self.DirFilterChoice.GetStringSelection()]

   150

        self.TypeFilter = self.TypeFilterChoice.GetSelection()

   151

   152

    def RefreshLocationsTree(self):

   153

        root = self.LocationsTree.GetRootItem()

   154

        if not root.IsOk():

   155

            root = self.LocationsTree.AddRoot("")

   156

        self.GenerateLocationsTreeBranch(root, self.Locations)

   157

   158

    def FilterType(self, location_type, location_size):

   159

        if self.TypeFilter == 0:

   160

            return True

   161

   162

        if location_size != self.VarTypeSize:

   163

            return False

   164

   165

        if self.TypeFilter == 1:

   166

            return self.Controller.IsOfType(location_type, self.BaseVarType)

   167

        elif self.TypeFilter == 2:

   168

            return location_type == self.VarType

   169

   170

        return True

   171

   172

    def GenerateLocationsTreeBranch(self, root, locations):

   173

        to_delete = []

   174

        item, root_cookie = self.LocationsTree.GetFirstChild(root)

   175

        for loc_infos in locations:

   176

            infos = loc_infos.copy()

   177

            if infos["type"] in [LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP] or\

   178

               infos["type"] in self.DirFilter and self.FilterType(infos["IEC_type"], infos["size"]):

   179

                children = [child for child in infos.pop("children")]

   180

                if not item.IsOk():

   181

                    item = self.LocationsTree.AppendItem(root, infos["name"])

   182

                    if wx.Platform != '__WXMSW__':

   183

                        item, root_cookie = self.LocationsTree.GetNextChild(root, root_cookie)

   184

                else:

   185

                    self.LocationsTree.SetItemText(item, infos["name"])

   186

                self.LocationsTree.SetPyData(item, infos)

   187

                self.LocationsTree.SetItemImage(item, self.TreeImageDict[infos["type"]])

   188

                self.GenerateLocationsTreeBranch(item, children)

   189

                item, root_cookie = self.LocationsTree.GetNextChild(root, root_cookie)

   190

        while item.IsOk():

   191

            to_delete.append(item)

   192

            item, root_cookie = self.LocationsTree.GetNextChild(root, root_cookie)

   193

        for item in to_delete:

   194

            self.LocationsTree.Delete(item)

   195

   196

    def OnLocationsTreeItemActivated(self, event):

   197

        infos = self.LocationsTree.GetPyData(event.GetItem())

   198

        if infos["type"] not in [LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP]:

   199

            wx.CallAfter(self.EndModal, wx.ID_OK)

   200

        event.Skip()

   201

   202

    def OnFilterChoice(self, event):

   203

        self.RefreshFilters()

   204

        self.RefreshLocationsTree()

   205

   206

    def GetValues(self):

   207

        selected = self.LocationsTree.GetSelection()

   208

        return self.LocationsTree.GetPyData(selected)

   209

   210

    def OnOK(self, event):

   211

        selected = self.LocationsTree.GetSelection()

   212

        var_infos = None

   213

        if selected.IsOk():

   214

            var_infos = self.LocationsTree.GetPyData(selected)

   215

        if var_infos is None or var_infos["type"] in [LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP]:

   216

            dialog = wx.MessageDialog(self, _("A location must be selected!"), _("Error"), wx.OK|wx.ICON_ERROR)

   217

            dialog.ShowModal()

   218

            dialog.Destroy()

   219

        else:

   220

            self.EndModal(wx.ID_OK)