1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

    26

from __future__ import absolute_import

    27

from six.moves import cPickle

    28

import wx

    29

    30

MAX_ITEM_COUNT = 10

    31

MAX_ITEM_SHOWN = 6

    32

if wx.Platform == '__WXMSW__':

    33

    LISTBOX_BORDER_HEIGHT = 2

    34

    LISTBOX_INTERVAL_HEIGHT = 0

    35

else:

    36

    LISTBOX_BORDER_HEIGHT = 4

    37

    LISTBOX_INTERVAL_HEIGHT = 6

    38

    39

    40

class PopupWithListbox(wx.PopupWindow):

    41

    42

    def __init__(self, parent, choices=None):

    43

        wx.PopupWindow.__init__(self, parent, wx.BORDER_SIMPLE)

    44

    45

        self.ListBox = wx.ListBox(self, -1, style=wx.LB_HSCROLL | wx.LB_SINGLE | wx.LB_SORT)

    46

    47

        choices = [] if choices is None else choices

    48

        self.SetChoices(choices)

    49

    50

        self.ListBox.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)

    51

        self.ListBox.Bind(wx.EVT_MOTION, self.OnMotion)

    52

    53

    def SetChoices(self, choices):

    54

        max_text_width = 0

    55

        max_text_height = 0

    56

    57

        self.ListBox.Clear()

    58

        for choice in choices:

    59

            self.ListBox.Append(choice)

    60

            w, h = self.ListBox.GetTextExtent(choice)

    61

            max_text_width = max(max_text_width, w)

    62

            max_text_height = max(max_text_height, h)

    63

    64

        itemcount = min(len(choices), MAX_ITEM_SHOWN)

    65

        width = self.Parent.GetSize()[0]

    66

        height = \

    67

            max_text_height * itemcount + \

    68

            LISTBOX_INTERVAL_HEIGHT * max(0, itemcount - 1) + \

    69

            2 * LISTBOX_BORDER_HEIGHT

    70

        if max_text_width + 10 > width:

    71

            height += 15

    72

        size = wx.Size(width, height)

    73

        if wx.Platform == '__WXMSW__':

    74

            size.width -= 2

    75

        self.ListBox.SetSize(size)

    76

        self.SetClientSize(size)

    77

    78

    def MoveSelection(self, direction):

    79

        selected = self.ListBox.GetSelection()

    80

        if selected == wx.NOT_FOUND:

    81

            if direction >= 0:

    82

                selected = 0

    83

            else:

    84

                selected = self.ListBox.GetCount() - 1

    85

        else:

    86

            selected = (selected + direction) % (self.ListBox.GetCount() + 1)

    87

        if selected == self.ListBox.GetCount():

    88

            selected = wx.NOT_FOUND

    89

        self.ListBox.SetSelection(selected)

    90

    91

    def GetSelection(self):

    92

        return self.ListBox.GetStringSelection()

    93

    94

    def OnLeftDown(self, event):

    95

        selected = self.ListBox.HitTest(wx.Point(event.GetX(), event.GetY()))

    96

        parent_size = self.Parent.GetSize()

    97

        parent_rect = wx.Rect(0, -parent_size[1], parent_size[0], parent_size[1])

    98

        if selected != wx.NOT_FOUND:

    99

            wx.CallAfter(self.Parent.SetValueFromSelected, self.ListBox.GetString(selected))

   100

        elif parent_rect.Contains(event.GetX(), event.GetY()):

   101

            result, x, y = self.Parent.HitTest(wx.Point(event.GetX(), event.GetY() + parent_size[1]))

   102

            if result != wx.TE_HT_UNKNOWN:

   103

                self.Parent.SetInsertionPoint(self.Parent.XYToPosition(x, y))

   104

        else:

   105

            wx.CallAfter(self.Parent.DismissListBox)

   106

        event.Skip()

   107

   108

    def OnMotion(self, event):

   109

        self.ListBox.SetSelection(

   110

            self.ListBox.HitTest(wx.Point(event.GetX(), event.GetY())))

   111

        event.Skip()

   112

   113

   114

class TextCtrlAutoComplete(wx.TextCtrl):

   115

   116

    def __init__(self, parent, choices=None, dropDownClick=True,

   117

                 element_path=None, **therest):

   118

        """

   119

        Constructor works just like wx.TextCtrl except you can pass in a

   120

        list of choices.  You can also change the choice list at any time

   121

        by calling setChoices.

   122

        """

   123

   124

        therest['style'] = wx.TE_PROCESS_ENTER | therest.get('style', 0)

   125

   126

        wx.TextCtrl.__init__(self, parent, **therest)

   127

   128

        # Some variables

   129

        self._dropDownClick = dropDownClick

   130

        self._lastinsertionpoint = None

   131

        self._hasfocus = False

   132

   133

        self._screenheight = wx.SystemSettings.GetMetric(wx.SYS_SCREEN_Y)

   134

        self.element_path = element_path

   135

   136

        self.listbox = None

   137

   138

        self.SetChoices(choices)

   139

   140

        # gp = self

   141

        # while ( gp != None ) :

   142

        #    gp.Bind ( wx.EVT_MOVE , self.onControlChanged, gp )

   143

        #    gp.Bind ( wx.EVT_SIZE , self.onControlChanged, gp )

   144

        #    gp = gp.GetParent()

   145

   146

        self.Bind(wx.EVT_KILL_FOCUS, self.OnControlChanged)

   147

        self.Bind(wx.EVT_TEXT_ENTER, self.OnControlChanged)

   148

        self.Bind(wx.EVT_TEXT, self.OnEnteredText)

   149

        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)

   150

   151

        # If need drop down on left click

   152

        if dropDownClick:

   153

            self.Bind(wx.EVT_LEFT_DOWN, self.OnClickToggleDown)

   154

            self.Bind(wx.EVT_LEFT_UP, self.OnClickToggleUp)

   155

   156

    def ChangeValue(self, value):

   157

        wx.TextCtrl.ChangeValue(self, value)

   158

        self.RefreshListBoxChoices()

   159

   160

    def OnEnteredText(self, event):

   161

        wx.CallAfter(self.RefreshListBoxChoices)

   162

        event.Skip()

   163

   164

    def OnKeyDown(self, event):

   165

        """ Do some work when the user press on the keys:

   166

            up and down: move the cursor

   167

        """

   168

        keycode = event.GetKeyCode()

   169

        if keycode in [wx.WXK_DOWN, wx.WXK_UP]:

   170

            self.PopupListBox()

   171

            if keycode == wx.WXK_DOWN:

   172

                self.listbox.MoveSelection(1)

   173

            else:

   174

                self.listbox.MoveSelection(-1)

   175

        elif keycode in [wx.WXK_LEFT, wx.WXK_RIGHT, wx.WXK_RETURN] and self.listbox is not None:

   176

            selected = self.listbox.GetSelection()

   177

            if selected != "":

   178

                self.SetValueFromSelected(selected)

   179

            else:

   180

                event.Skip()

   181

        elif event.GetKeyCode() == wx.WXK_ESCAPE:

   182

            self.DismissListBox()

   183

        else:

   184

            event.Skip()

   185

   186

    def OnClickToggleDown(self, event):

   187

        self._lastinsertionpoint = self.GetInsertionPoint()

   188

        event.Skip()

   189

   190

    def OnClickToggleUp(self, event):

   191

        if not self._hasfocus:

   192

            self._hasfocus = True

   193

        elif self.GetInsertionPoint() == self._lastinsertionpoint:

   194

            wx.CallAfter(self.PopupListBox)

   195

        self._lastinsertionpoint = None

   196

        event.Skip()

   197

   198

    def OnControlChanged(self, event):

   199

        res = self.GetValue()

   200

        config = wx.ConfigBase.Get()

   201

        listentries = cPickle.loads(str(config.Read(self.element_path, cPickle.dumps([]))))

   202

        if res and res not in listentries:

   203

            listentries = (listentries + [res])[-MAX_ITEM_COUNT:]

   204

            config.Write(self.element_path, cPickle.dumps(listentries))

   205

            config.Flush()

   206

            self.SetChoices(listentries)

   207

        self.DismissListBox()

   208

        self._hasfocus = False

   209

        event.Skip()

   210

   211

    def SetChoices(self, choices):

   212

        self._choices = choices

   213

        self.RefreshListBoxChoices()

   214

   215

    def GetChoices(self):

   216

        return self._choices

   217

   218

    def SetValueFromSelected(self, selected):

   219

        """

   220

        Sets the wx.TextCtrl value from the selected wx.ListCtrl item.

   221

        Will do nothing if no item is selected in the wx.ListCtrl.

   222

        """

   223

        if selected != "":

   224

            self.SetValue(selected)

   225

        self.DismissListBox()

   226

   227

    def RefreshListBoxChoices(self):

   228

        if self.listbox is not None:

   229

            text = self.GetValue()

   230

            choices = [choice for choice in self._choices if choice.startswith(text)]

   231

            self.listbox.SetChoices(choices)

   232

   233

    def PopupListBox(self):

   234

        if self.listbox is None:

   235

            self.listbox = PopupWithListbox(self)

   236

   237

            # Show the popup right below or above the button

   238

            # depending on available screen space...

   239

            pos = self.ClientToScreen((0, 0))

   240

            sz = self.GetSize()

   241

            if wx.Platform == '__WXMSW__':

   242

                pos.x -= 2

   243

                pos.y -= 2

   244

            self.listbox.Position(pos, (0, sz[1]))

   245

   246

            self.RefreshListBoxChoices()

   247

   248

            self.listbox.Show()

   249

   250

    def DismissListBox(self):

   251

        if self.listbox is not None:

   252

            if self.listbox.ListBox.HasCapture():

   253

                self.listbox.ListBox.ReleaseMouse()

   254

            self.listbox.Destroy()

   255

            self.listbox = None