File: maparea_browser.py

package info (click to toggle)
djvusmooth 0.2.7-1
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 408 kB
  • ctags: 915
  • sloc: python: 4,595; xml: 97; makefile: 54
file content (205 lines) | stat: -rw-r--r-- 6,375 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
# encoding=UTF-8
# Copyright © 2008, 2009 Jakub Wilk <jwilk@jwilk.net>
# Copyright © 2009 Mateusz Turcza <mturcza@mimuw.edu.pl>
#
# This package is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 dated June, 1991.
#
# This package is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.

import wx
import wx.lib.mixins.listctrl

import djvusmooth.models.annotations
from djvusmooth import models
import djvusmooth.gui.maparea_menu
from djvusmooth import gui
from djvusmooth.i18n import _

class PageAnnotationsCallback(models.annotations.PageAnnotationsCallback):

    def __init__(self, owner):
        self._owner = owner
    
    def notify_node_change(self, node):
        wx.CallAfter(lambda: self._owner.on_node_change(node))
    
    def notify_node_select(self, node):
        wx.CallAfter(lambda: self._owner.on_node_select(node))
    
    def notify_node_deselect(self, node):
        pass
    
    def notify_node_add(self, node):
        wx.CallAfter(lambda: self._owner.on_node_add(node))
    
    def notify_node_replace(self, node, other_node):
        wx.CallAfter(lambda: self._owner.on_node_replace(node, other_node))
    
    def notify_node_delete(self, node):
        wx.CallAfter(lambda: self._owner.on_node_delete(node))

def item_to_id(item):
    try:
        return int(item)
    except TypeError:
        return item.GetId()

class MapAreaBrowser(
    wx.ListCtrl,
    wx.lib.mixins.listctrl.ListCtrlAutoWidthMixin,
    wx.lib.mixins.listctrl.TextEditMixin
):

    def __init__(self, parent, id = wx.ID_ANY, pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.LC_REPORT):
        wx.ListCtrl.__init__(self, parent, id, pos, size, style)
        self.InsertColumn(0, _('URI'))
        self.InsertColumn(1, _('Comment'))
        self._have_items = False
        self._data = {}
        self._data_map = {}
        self.page = None
        wx.lib.mixins.listctrl.ListCtrlAutoWidthMixin.__init__(self)
        wx.lib.mixins.listctrl.TextEditMixin.__init__(self)
        self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_selection_changed, self)
        self.Bind(wx.EVT_LIST_ITEM_RIGHT_CLICK, self.on_item_right_click, self)
        self.Bind(wx.EVT_CHAR, self.on_char, self)


    def do_remove_node(self, node):
        node.delete()

    _WXK_TO_METHOD = {
        wx.WXK_DELETE: do_remove_node
    }

    def on_char(self, event):
        key_code = event.GetKeyCode()
        try:
            method = self._WXK_TO_METHOD[key_code]
        except KeyError:
            return
        item = self.GetFirstSelected()
        node = self.GetPyData(item)
        if node is None:
            return
        method(self, node)

    def on_node_add(self, node):
        item = self._insert_item(node)
        self.Focus(item)
    
    def on_node_replace(self, node, other_node):
        item = self._insert_item(other_node, replace_node=node)
        self.Focus(item)

    def on_node_change(self, node):
        self.on_node_replace(node, node)
    
    def on_node_delete(self, node):
        self._remove_item(node)
    
    def on_node_select(self, node):
        try:
            current_item = self._data_map[node]
        except KeyError:
            return
        selected_item = self.GetFirstSelected()
        if selected_item != current_item:
            self.Select(selected_item, False)
            self.Select(current_item, True)
    
    def on_item_right_click(self, event):
        item = event.m_itemIndex
        node = self.GetPyData(item)
        # Yup, we accept the fact that `node` can be `None`
        self.show_menu(node, event.GetPoint())
    
    def show_menu(self, node, point):
        gui.maparea_menu.show_menu(self, self.page.annotations, node, point)

    def on_selection_changed(self, event):
        event.Skip()
        item = event.m_itemIndex
        node = self.GetPyData(item)
        if node is None:
            return
        node.notify_select()
    
    @apply
    def page():
        def get(self):
            return self._page
        def set(self, value):
            if value is not True:
                self._page = value
            if self._page is not None:
                self._callback = PageAnnotationsCallback(self)
                self._page.annotations.register_callback(self._callback)
                self._model = self.page.annotations
            self._recreate_items()
        return property(get, set)

    def SetStringItem(self, item, col, label, super = False):
        wx.ListCtrl.SetStringItem(self, item, col, label)
        if super:
            return
        node = self.GetPyData(item)
        if node is None:
            return
        if col == 0:
            node.uri = label
        elif col == 1:
            node.comment = label

    def GetPyData(self, item):
        id = item_to_id(item)
        return self._data.get(id)
    
    def SetPyData(self, item, data):
        id = item_to_id(item)
        try:
            del self._data_map[self._data[id]]
        except KeyError:
            pass
        self._data[id] = data
        self._data_map[data] = id
    
    def _remove_all_items(self):
        wx.ListCtrl.DeleteAllItems(self)
        self._data.clear()
        self._data_map.clear()

    def _remove_item(self, node):
        item = self._data_map.pop(node)
        i = item_to_id(item)
        del self._data[i]
        self.DeleteItem(i)

    def _insert_item(self, node, replace_node = None):
        if replace_node in self._data_map:
            item = self._data_map[replace_node]
            i = item_to_id(item)
        else:
            i = self.GetItemCount()
            item = self.InsertStringItem(i, '')
        for i, s in enumerate((node.uri, node.comment)):
            self.SetStringItem(item, i, s, super = True)
        self.SetPyData(item, node)
        return item

    def _recreate_items(self):
        self._remove_all_items()
        self._nodes = []
        if self.page is None:
            return
        for node in self._model.mapareas:
            self._insert_item(node)

__all__ = 'MapAreaBrowser',

# vim:ts=4 sw=4 et