/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to annotate/gannotate.py

  • Committer: Jelmer Vernooij
  • Date: 2006-05-19 16:56:46 UTC
  • mfrom: (0.1.25 gannotate)
  • Revision ID: jelmer@samba.org-20060519165646-0d867938fdbc9097
Merge in Dan Loda's gannotate plugin and put it in annotate/

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
import time
18
18
 
19
 
from gi.repository import GObject
20
 
from gi.repository import Gdk
21
 
from gi.repository import Gtk
22
 
from gi.repository import Pango
23
 
import re
 
19
import pygtk
 
20
pygtk.require("2.0")
 
21
import gobject
 
22
import gtk
 
23
import pango
24
24
 
25
 
from bzrlib import patiencediff
26
25
from bzrlib.errors import NoSuchRevision
27
 
from bzrlib.revision import NULL_REVISION, CURRENT_REVISION
28
26
 
29
 
from bzrlib.plugins.gtk.annotate.colormap import AnnotateColorSaturation
30
 
from bzrlib.plugins.gtk.i18n import _i18n
31
 
from bzrlib.plugins.gtk.revisionview import RevisionView
32
 
from bzrlib.plugins.gtk.window import Window
 
27
from colormap import AnnotateColorMap, AnnotateColorSaturation
 
28
from logview import LogView
 
29
from spanselector import SpanSelector
33
30
 
34
31
 
35
32
(
42
39
) = range(6)
43
40
 
44
41
 
45
 
class GAnnotateWindow(Window):
 
42
class GAnnotateWindow(gtk.Window):
46
43
    """Annotate window."""
47
44
 
48
 
    def __init__(self, all=False, plain=False, parent=None, branch=None):
 
45
    def __init__(self, all=False, plain=False):
49
46
        self.all = all
50
47
        self.plain = plain
51
 
        self._branch = branch
52
 
 
53
 
        super(GAnnotateWindow, self).__init__(parent=parent)
54
 
 
55
 
        self.set_icon(
56
 
            self.render_icon_pixbuf(Gtk.STOCK_FIND, Gtk.IconSize.BUTTON))
 
48
        
 
49
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
 
50
        
 
51
        self.set_icon(self.render_icon(gtk.STOCK_FIND, gtk.ICON_SIZE_BUTTON))
57
52
        self.annotate_colormap = AnnotateColorSaturation()
58
53
 
59
54
        self._create()
 
55
 
 
56
        if self.plain:
 
57
            self.span_selector.hide()
 
58
 
 
59
    def annotate(self, branch, file_id):
60
60
        self.revisions = {}
61
 
        self.history = []
62
 
        self._no_back = set()
63
 
 
64
 
    def annotate(self, tree, branch, file_id):
65
 
        self.annotations = []
66
 
        self.branch = branch
67
 
        self.tree = tree
68
 
        self.file_id = file_id
69
 
        self.revisionview.set_file_id(file_id)
70
 
        self.revision_id = getattr(tree, 'get_revision_id', 
71
 
                                   lambda: CURRENT_REVISION)()
72
 
 
73
 
        # [revision id, line number, author, revno, highlight color, line]
74
 
        self.annomodel = Gtk.ListStore(GObject.TYPE_STRING,
75
 
                                       GObject.TYPE_INT,
76
 
                                       GObject.TYPE_STRING,
77
 
                                       GObject.TYPE_STRING,
78
 
                                       GObject.TYPE_STRING,
79
 
                                       GObject.TYPE_STRING)
80
 
 
 
61
        
 
62
        # [revision id, line number, committer, revno, highlight color, line]
 
63
        self.annomodel = gtk.ListStore(gobject.TYPE_STRING,
 
64
                                       gobject.TYPE_STRING,
 
65
                                       gobject.TYPE_STRING,
 
66
                                       gobject.TYPE_STRING,
 
67
                                       gobject.TYPE_STRING,
 
68
                                       gobject.TYPE_STRING)
 
69
        
81
70
        last_seen = None
82
71
        try:
83
72
            branch.lock_read()
84
73
            branch.repository.lock_read()
85
 
            self.dotted = {}
86
 
            revno_map = self.branch.get_revision_id_to_revno_map()
87
 
            for revision_id, revno in revno_map.iteritems():
88
 
                self.dotted[revision_id] = '.'.join(str(num) for num in revno)
89
74
            for line_no, (revision, revno, line)\
90
 
                in enumerate(self._annotate(tree, file_id)):
 
75
                    in enumerate(self._annotate(branch, file_id)):
91
76
                if revision.revision_id == last_seen and not self.all:
92
 
                    revno = author = ""
 
77
                    revno = committer = ""
93
78
                else:
94
79
                    last_seen = revision.revision_id
95
 
                    author = ", ".join(revision.get_apparent_authors())
 
80
                    committer = revision.committer
96
81
 
97
82
                if revision.revision_id not in self.revisions:
98
83
                    self.revisions[revision.revision_id] = revision
99
84
 
100
85
                self.annomodel.append([revision.revision_id,
101
86
                                       line_no + 1,
102
 
                                       author,
 
87
                                       committer,
103
88
                                       revno,
104
89
                                       None,
105
90
                                       line.rstrip("\r\n")
106
 
                                       ])
107
 
                self.annotations.append(revision)
 
91
                                      ])
108
92
 
109
93
            if not self.plain:
110
 
                now = time.time()
111
 
                self.annomodel.foreach(self._highlight_annotation, now)
 
94
                self._set_oldest_newest()
 
95
                # Recall that calling activate_default will emit "span-changed",
 
96
                # so self._span_changed_cb will take care of initial highlighting
 
97
                self.span_selector.activate_default()
112
98
        finally:
113
99
            branch.repository.unlock()
114
100
            branch.unlock()
115
101
 
116
102
        self.annoview.set_model(self.annomodel)
117
103
        self.annoview.grab_focus()
118
 
        my_revno = self.dotted.get(self.revision_id, 'current')
119
 
        title = '%s (%s) - gannotate' % (self.tree.id2path(file_id), my_revno)
120
 
        self.set_title(title)
121
104
 
122
105
    def jump_to_line(self, lineno):
123
106
        if lineno > len(self.annomodel) or lineno < 1:
126
109
            # bar?
127
110
            print("gannotate: Line number %d does't exist. Defaulting to "
128
111
                  "line 1." % lineno)
129
 
            return
130
112
        else:
131
113
            row = lineno - 1
132
114
 
133
 
        tree_path = Gtk.TreePath(path=row)
134
 
        self.annoview.set_cursor(tree_path, None, False)
135
 
        self.annoview.scroll_to_cell(tree_path, use_align=True)
 
115
        self.annoview.set_cursor(row)
136
116
 
137
 
    def _annotate(self, tree, file_id):
138
 
        current_revision = FakeRevision(CURRENT_REVISION)
139
 
        current_revision.committer = self.branch.get_config().username()
140
 
        current_revision.timestamp = time.time()
141
 
        current_revision.message = '[Not yet committed]'
142
 
        current_revision.parent_ids = tree.get_parent_ids()
143
 
        current_revision.properties['branch-nick'] = self.branch._get_nick(local=True)
144
 
        current_revno = '%d?' % (self.branch.revno() + 1)
145
 
        repository = self.branch.repository
146
 
        if self.revision_id == CURRENT_REVISION:
147
 
            revision_id = self.branch.last_revision()
148
 
        else:
149
 
            revision_id = self.revision_id
150
 
        revision_cache = RevisionCache(repository, self.revisions)
151
 
        for origin, text in tree.annotate_iter(file_id):
 
117
    def _annotate(self, branch, file_id):
 
118
        rev_hist = branch.revision_history()
 
119
        repository = branch.repository
 
120
        rev_tree = repository.revision_tree(branch.last_revision())
 
121
        rev_id = rev_tree.inventory[file_id].revision
 
122
        weave = repository.weave_store.get_weave(file_id,
 
123
                                                 branch.get_transaction())
 
124
        
 
125
        revision_cache = RevisionCache(repository)
 
126
        for origin, text in weave.annotate_iter(rev_id):
152
127
            rev_id = origin
153
 
            if rev_id == CURRENT_REVISION:
154
 
                revision = current_revision
155
 
                revno = current_revno
156
 
            else:
157
 
                try:
158
 
                    revision = revision_cache.get_revision(rev_id)
159
 
                    revno = self.dotted.get(rev_id, 'merge')
160
 
                    if len(revno) > 15:
161
 
                        revno = 'merge'
162
 
                except NoSuchRevision:
163
 
                    revision = FakeRevision(rev_id)
164
 
                    revno = "?"
 
128
            try:
 
129
                revision = revision_cache.get_revision(rev_id)
 
130
                if rev_id in rev_hist:
 
131
                    revno = branch.revision_id_to_revno(rev_id)
 
132
                else:
 
133
                    revno = "merge"
 
134
            except NoSuchRevision:
 
135
                revision = NoneRevision(rev_id)
 
136
                revno = "?"
165
137
 
166
138
            yield revision, revno, text
167
139
 
 
140
    def _set_oldest_newest(self):
 
141
        rev_dates = map(lambda i: self.revisions[i].timestamp, self.revisions)
 
142
        oldest = min(rev_dates)
 
143
        newest = max(rev_dates)
 
144
 
 
145
        span = self._span_from_seconds(time.time() - oldest)
 
146
        self.span_selector.set_to_oldest_span(span)
 
147
        
 
148
        span = self._span_from_seconds(newest - oldest)
 
149
        self.span_selector.set_newest_to_oldest_span(span)
 
150
 
 
151
    def _span_from_seconds(self, seconds):
 
152
        return (seconds / (24 * 60 * 60))
 
153
    
 
154
    def _span_changed_cb(self, w, span):
 
155
        self.annotate_colormap.set_span(span)
 
156
        now = time.time()
 
157
        self.annomodel.foreach(self._highlight_annotation, now)
 
158
 
168
159
    def _highlight_annotation(self, model, path, iter, now):
169
160
        revision_id, = model.get(iter, REVISION_ID_COL)
170
161
        revision = self.revisions[revision_id]
171
 
        # XXX sinzui 2011-08-12: What does get_color return?
172
 
        color = self.annotate_colormap.get_color(revision, now)
173
 
        model.set_value(iter, HIGHLIGHT_COLOR_COL, color)
 
162
        model.set(iter, HIGHLIGHT_COLOR_COL,
 
163
                  self.annotate_colormap.get_color(revision, now))
174
164
 
175
 
    def _selected_revision(self):
 
165
    def _show_log(self, w):
176
166
        (path, col) = self.annoview.get_cursor()
177
 
        if path is None:
178
 
            return None
179
 
        return self.annomodel[path][REVISION_ID_COL]
180
 
 
181
 
    def _activate_selected_revision(self, w):
182
 
        rev_id = self._selected_revision()
183
 
        if not rev_id or rev_id == NULL_REVISION:
184
 
            return
185
 
        selected = self.revisions[rev_id]
186
 
        self.revisionview.set_revision(selected)
187
 
        if (len(selected.parent_ids) != 0 and selected.parent_ids[0] not in
188
 
            self._no_back):
189
 
            enable_back = True
190
 
        else:
191
 
            enable_back = False
192
 
        self.back_button.set_sensitive(enable_back)
 
167
        rev_id = self.annomodel[path][REVISION_ID_COL]
 
168
        self.logview.set_revision(self.revisions[rev_id])
193
169
 
194
170
    def _create(self):
195
 
        self.revisionview = self._create_log_view()
 
171
        self.logview = self._create_log_view()
196
172
        self.annoview = self._create_annotate_view()
 
173
        self.span_selector = self._create_span_selector()
197
174
 
198
 
        vbox = Gtk.VBox(homogeneous=False, spacing=0)
 
175
        vbox = gtk.VBox(False, 12)
 
176
        vbox.set_border_width(12)
199
177
        vbox.show()
200
178
 
201
 
        sw = Gtk.ScrolledWindow()
202
 
        sw.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
203
 
        sw.set_shadow_type(Gtk.ShadowType.IN)
 
179
        sw = gtk.ScrolledWindow()
 
180
        sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
 
181
        sw.set_shadow_type(gtk.SHADOW_IN)
204
182
        sw.add(self.annoview)
205
 
        self.annoview.gwindow = self
206
183
        sw.show()
207
 
 
208
 
        swbox = Gtk.VBox()
209
 
        swbox.pack_start(sw, True, True, 0)
210
 
        swbox.show()
211
 
 
212
 
        hbox = Gtk.HBox(homogeneous=False, spacing=6)
213
 
        self.back_button = self._create_back_button()
214
 
        hbox.pack_start(self.back_button, False, True, 0)
215
 
        self.forward_button = self._create_forward_button()
216
 
        hbox.pack_start(self.forward_button, False, True, 0)
217
 
        self.find_button = self._create_find_button()
218
 
        hbox.pack_start(self.find_button, False, True, 0)
219
 
        self.goto_button = self._create_goto_button()
220
 
        hbox.pack_start(self.goto_button, False, True, 0)
 
184
        
 
185
        self.pane = pane = gtk.VPaned()
 
186
        pane.add1(sw)
 
187
        pane.add2(self.logview)
 
188
        pane.show()
 
189
        vbox.pack_start(pane, expand=True, fill=True)
 
190
        
 
191
        hbox = gtk.HBox(True, 6)
 
192
        hbox.pack_start(self.span_selector, expand=False, fill=True)
 
193
        hbox.pack_start(self._create_button_box(), expand=False, fill=True)
221
194
        hbox.show()
222
 
        vbox.pack_start(hbox, False, True, 0)
223
 
 
224
 
        self.pane = pane = Gtk.Paned.new(Gtk.Orientation.VERTICAL)
225
 
        pane.add1(swbox)
226
 
        pane.add2(self.revisionview)
227
 
        pane.show()
228
 
        vbox.pack_start(pane, True, True, 0)
229
 
 
230
 
        self._search = SearchBox()
231
 
        swbox.pack_start(self._search, False, True, 0)
232
 
        accels = Gtk.AccelGroup()
233
 
        accels.connect(Gdk.KEY_f, Gdk.ModifierType.CONTROL_MASK,
234
 
                             Gtk.AccelFlags.LOCKED,
235
 
                             self._search_by_text)
236
 
        accels.connect(Gdk.KEY_g, Gdk.ModifierType.CONTROL_MASK,
237
 
                             Gtk.AccelFlags.LOCKED,
238
 
                             self._search_by_line)
239
 
        self.add_accel_group(accels)
 
195
        vbox.pack_start(hbox, expand=False, fill=True)
240
196
 
241
197
        self.add(vbox)
242
198
 
243
 
    def _search_by_text(self, *ignored): # (accel_group, window, key, modifiers):
244
 
        self._search.show_for('text')
245
 
        self._search.set_target(self.annoview, TEXT_LINE_COL)
246
 
 
247
 
    def _search_by_line(self, *ignored): # accel_group, window, key, modifiers):
248
 
        self._search.show_for('line')
249
 
        self._search.set_target(self.annoview, LINE_NUM_COL)
250
 
 
251
 
    def line_diff(self, tv, path, tvc):
252
 
        row = path.get_indices()[0]
253
 
        revision = self.annotations[row]
254
 
        repository = self.branch.repository
255
 
        if revision.revision_id == CURRENT_REVISION:
256
 
            tree1 = self.tree
257
 
            tree2 = self.tree.basis_tree()
258
 
        else:
259
 
            tree1 = repository.revision_tree(revision.revision_id)
260
 
            if len(revision.parent_ids) > 0:
261
 
                tree2 = repository.revision_tree(revision.parent_ids[0])
262
 
            else:
263
 
                tree2 = repository.revision_tree(NULL_REVISION)
264
 
        from bzrlib.plugins.gtk.diff import DiffWindow
265
 
        window = DiffWindow(self)
266
 
        window.set_diff("Diff for line %d" % (row+1), tree1, tree2)
267
 
        window.set_file(tree1.id2path(self.file_id))
268
 
        window.show()
269
 
 
270
 
 
271
199
    def _create_annotate_view(self):
272
 
        tv = Gtk.TreeView()
 
200
        tv = gtk.TreeView()
273
201
        tv.set_rules_hint(False)
274
 
        tv.connect("cursor-changed", self._activate_selected_revision)
 
202
        tv.connect("cursor-changed", self._show_log)
275
203
        tv.show()
276
 
        tv.connect("row-activated", self.line_diff)
277
204
 
278
 
        cell = Gtk.CellRendererText()
 
205
        cell = gtk.CellRendererText()
279
206
        cell.set_property("xalign", 1.0)
280
207
        cell.set_property("ypad", 0)
281
208
        cell.set_property("family", "Monospace")
282
209
        cell.set_property("cell-background-gdk",
283
 
                          tv.get_style().bg[Gtk.StateType.NORMAL])
284
 
        col = Gtk.TreeViewColumn()
 
210
                          tv.get_style().bg[gtk.STATE_NORMAL])
 
211
        col = gtk.TreeViewColumn()
285
212
        col.set_resizable(False)
286
 
        col.pack_start(cell, True)
 
213
        col.pack_start(cell, expand=True)
287
214
        col.add_attribute(cell, "text", LINE_NUM_COL)
288
215
        tv.append_column(col)
289
216
 
290
 
        cell = Gtk.CellRendererText()
 
217
        cell = gtk.CellRendererText()
291
218
        cell.set_property("ypad", 0)
292
 
        cell.set_property("ellipsize", Pango.EllipsizeMode.END)
 
219
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
293
220
        cell.set_property("cell-background-gdk",
294
 
                          self.get_style().bg[Gtk.StateType.NORMAL])
295
 
        col = Gtk.TreeViewColumn("Committer")
 
221
                          self.get_style().bg[gtk.STATE_NORMAL])
 
222
        col = gtk.TreeViewColumn("Committer")
296
223
        col.set_resizable(True)
297
 
        col.pack_start(cell, True)
 
224
        col.pack_start(cell, expand=True)
298
225
        col.add_attribute(cell, "text", COMMITTER_COL)
299
226
        tv.append_column(col)
300
227
 
301
 
        cell = Gtk.CellRendererText()
 
228
        cell = gtk.CellRendererText()
302
229
        cell.set_property("xalign", 1.0)
303
230
        cell.set_property("ypad", 0)
304
231
        cell.set_property("cell-background-gdk",
305
 
                          self.get_style().bg[Gtk.StateType.NORMAL])
306
 
        col = Gtk.TreeViewColumn("Revno")
 
232
                          self.get_style().bg[gtk.STATE_NORMAL])
 
233
        col = gtk.TreeViewColumn("Revno")
307
234
        col.set_resizable(False)
308
 
        col.pack_start(cell, True)
 
235
        col.pack_start(cell, expand=True)
309
236
        col.add_attribute(cell, "markup", REVNO_COL)
310
237
        tv.append_column(col)
311
238
 
312
 
        cell = Gtk.CellRendererText()
 
239
        cell = gtk.CellRendererText()
313
240
        cell.set_property("ypad", 0)
314
241
        cell.set_property("family", "Monospace")
315
 
        col = Gtk.TreeViewColumn()
 
242
        col = gtk.TreeViewColumn()
316
243
        col.set_resizable(False)
317
 
        col.pack_start(cell, True)
 
244
        col.pack_start(cell, expand=True)
318
245
#        col.add_attribute(cell, "foreground", HIGHLIGHT_COLOR_COL)
319
246
        col.add_attribute(cell, "background", HIGHLIGHT_COLOR_COL)
320
247
        col.add_attribute(cell, "text", TEXT_LINE_COL)
321
248
        tv.append_column(col)
322
249
 
323
 
        # interactive substring search
324
 
        def search_equal_func(model, column, key, iter):
325
 
            return model.get_value(iter, TEXT_LINE_COL).lower().find(key.lower()) == -1
326
 
 
327
 
        tv.set_enable_search(True)
328
 
        tv.set_search_equal_func(search_equal_func, None)
329
 
 
 
250
        tv.set_search_column(LINE_NUM_COL)
 
251
        
330
252
        return tv
331
253
 
 
254
    def _create_span_selector(self):
 
255
        ss = SpanSelector()
 
256
        ss.connect("span-changed", self._span_changed_cb)
 
257
        ss.show()
 
258
 
 
259
        return ss
 
260
 
332
261
    def _create_log_view(self):
333
 
        lv = RevisionView(self._branch)
 
262
        lv = LogView()
334
263
        lv.show()
 
264
 
335
265
        return lv
336
266
 
337
 
    def _create_back_button(self):
338
 
        button = Gtk.Button()
339
 
        button.set_use_stock(True)
340
 
        button.set_label("gtk-go-back")
341
 
        button.connect("clicked", lambda w: self.go_back())
342
 
        button.set_relief(Gtk.ReliefStyle.NONE)
343
 
        button.show()
344
 
        return button
345
 
 
346
 
    def _create_forward_button(self):
347
 
        button = Gtk.Button()
348
 
        button.set_use_stock(True)
349
 
        button.set_label("gtk-go-forward")
350
 
        button.connect("clicked", lambda w: self.go_forward())
351
 
        button.set_relief(Gtk.ReliefStyle.NONE)
352
 
        button.show()
353
 
        button.set_sensitive(False)
354
 
        return button
355
 
 
356
 
    def _create_find_button(self):
357
 
        button = Gtk.Button()
358
 
        button.set_use_stock(True)
359
 
        button.set_label("gtk-find")
360
 
        button.set_tooltip_text("Search for text (Ctrl+F)")
361
 
        button.connect("clicked", self._search_by_text)
362
 
        button.set_relief(Gtk.ReliefStyle.NONE)
363
 
        button.show()
364
 
        button.set_sensitive(True)
365
 
        return button
366
 
 
367
 
    def _create_goto_button(self):
368
 
        button = Gtk.Button()
369
 
        button.set_label("Goto Line")
370
 
        button.set_tooltip_text("Scroll to a line by entering its number (Ctrl+G)")
371
 
        button.connect("clicked", self._search_by_line)
372
 
        button.set_relief(Gtk.ReliefStyle.NONE)
373
 
        button.show()
374
 
        button.set_sensitive(True)
375
 
        return button
376
 
 
377
 
    def go_back(self):
378
 
        last_tree = self.tree
379
 
        rev_id = self._selected_revision()
380
 
        parent_id = self.revisions[rev_id].parent_ids[0]
381
 
        target_tree = self.branch.repository.revision_tree(parent_id)
382
 
        if self._go(target_tree):
383
 
            self.history.append(last_tree)
384
 
            self.forward_button.set_sensitive(True)
385
 
        else:
386
 
            self._no_back.add(parent_id)
387
 
            self.back_button.set_sensitive(False)
388
 
 
389
 
    def go_forward(self):
390
 
        if len(self.history) == 0:
391
 
            return
392
 
        target_tree = self.history.pop()
393
 
        if len(self.history) == 0:
394
 
            self.forward_button.set_sensitive(False)
395
 
        self._go(target_tree)
396
 
 
397
 
    def _go(self, target_tree):
398
 
        rev_id = self._selected_revision()
399
 
        if self.file_id in target_tree:
400
 
            offset = self.get_scroll_offset(target_tree)
401
 
            path, col = self.annoview.get_cursor()
402
 
            (row,) = path.get_indices()
403
 
            self.annotate(target_tree, self.branch, self.file_id)
404
 
            new_row = row+offset
405
 
            if new_row < 0:
406
 
                new_row = 0
407
 
            new_path = Gtk.TreePath(path=new_row)
408
 
            self.annoview.set_cursor(new_path, None, False)
409
 
            return True
410
 
        else:
411
 
            return False
412
 
 
413
 
    def get_scroll_offset(self, tree):
414
 
        old = self.tree.get_file(self.file_id)
415
 
        new = tree.get_file(self.file_id)
416
 
        path, col = self.annoview.get_cursor()
417
 
        (row,) = path.get_indices()
418
 
        matcher = patiencediff.PatienceSequenceMatcher(None, old.readlines(),
419
 
                                                       new.readlines())
420
 
        for i, j, n in matcher.get_matching_blocks():
421
 
            if i + n >= row:
422
 
                return j - i
423
 
 
424
 
 
425
 
class FakeRevision(object):
 
267
    def _create_button_box(self):
 
268
        box = gtk.HButtonBox()
 
269
        box.set_layout(gtk.BUTTONBOX_END)
 
270
        box.show()
 
271
        
 
272
        button = gtk.Button()
 
273
        button.set_use_stock(True)
 
274
        button.set_label("gtk-close")
 
275
        button.connect("clicked", lambda w: self.destroy())
 
276
        button.show()
 
277
        
 
278
        box.pack_start(button, expand=False, fill=False)
 
279
 
 
280
        return box
 
281
 
 
282
 
 
283
class NoneRevision:
426
284
    """ A fake revision.
427
285
 
428
286
    For when a revision is referenced but not present.
429
287
    """
430
288
 
431
 
    def __init__(self, revision_id, committer='?', nick=None):
 
289
    def __init__(self, revision_id):
432
290
        self.revision_id = revision_id
433
291
        self.parent_ids = []
434
 
        self.committer = committer
 
292
        self.committer = "?"
435
293
        self.message = "?"
436
294
        self.timestamp = 0.0
437
295
        self.timezone = 0
438
 
        self.properties = {}
439
 
 
440
 
    def get_apparent_authors(self):
441
 
        return [self.committer]
442
296
 
443
297
 
444
298
class RevisionCache(object):
445
299
    """A caching revision source"""
446
 
 
447
 
    def __init__(self, real_source, seed_cache=None):
 
300
    def __init__(self, real_source):
448
301
        self.__real_source = real_source
449
 
        if seed_cache is None:
450
 
            self.__cache = {}
451
 
        else:
452
 
            self.__cache = dict(seed_cache)
 
302
        self.__cache = {}
453
303
 
454
304
    def get_revision(self, revision_id):
455
305
        if revision_id not in self.__cache:
456
306
            revision = self.__real_source.get_revision(revision_id)
457
307
            self.__cache[revision_id] = revision
458
308
        return self.__cache[revision_id]
459
 
 
460
 
 
461
 
class SearchBox(Gtk.HBox):
462
 
    """A button box for searching in text or lines of annotations"""
463
 
    def __init__(self):
464
 
        super(SearchBox, self).__init__(homogeneous=False, spacing=6)
465
 
 
466
 
        # Close button
467
 
        button = Gtk.Button()
468
 
        image = Gtk.Image()
469
 
        image.set_from_stock('gtk-stop', Gtk.IconSize.BUTTON)
470
 
        button.set_image(image)
471
 
        button.set_relief(Gtk.ReliefStyle.NONE)
472
 
        button.connect("clicked", lambda w: self.hide())
473
 
        self.pack_start(button, False, False, 0)
474
 
 
475
 
        # Search entry
476
 
        label = Gtk.Label()
477
 
        self._label = label
478
 
        self.pack_start(label, False, False, 0)
479
 
 
480
 
        entry = Gtk.Entry()
481
 
        self._entry = entry
482
 
        entry.connect("activate", lambda w, d: self._do_search(d),
483
 
                      'forward')
484
 
        self.pack_start(entry, False, False, 0)
485
 
 
486
 
        # Next/previous buttons
487
 
        button = Gtk.Button(_i18n('_Next'), use_underline=True)
488
 
        image = Gtk.Image()
489
 
        image.set_from_stock('gtk-go-forward', Gtk.IconSize.BUTTON)
490
 
        button.set_image(image)
491
 
        button.connect("clicked", lambda w, d: self._do_search(d),
492
 
                       'forward')
493
 
        self.pack_start(button, False, False, 0)
494
 
 
495
 
        button = Gtk.Button(_i18n('_Previous'), use_underline=True)
496
 
        image = Gtk.Image()
497
 
        image.set_from_stock('gtk-go-back', Gtk.IconSize.BUTTON)
498
 
        button.set_image(image)
499
 
        button.connect("clicked", lambda w, d: self._do_search(d),
500
 
                       'backward')
501
 
        self.pack_start(button, False, False, 0)
502
 
 
503
 
        # Search options
504
 
        check = Gtk.CheckButton('Match case')
505
 
        self._match_case = check
506
 
        self.pack_start(check, False, False, 0)
507
 
 
508
 
        check = Gtk.CheckButton('Regexp')
509
 
        check.connect("toggled", lambda w: self._set_label())
510
 
        self._regexp = check
511
 
        self.pack_start(check, False, False, 0)
512
 
 
513
 
        self._view = None
514
 
        self._column = None
515
 
        # Note that we stay hidden (we do not call self.show_all())
516
 
 
517
 
 
518
 
    def show_for(self, kind):
519
 
        self._kind = kind
520
 
        self.show_all()
521
 
        self._set_label()
522
 
        # Hide unrelated buttons
523
 
        if kind == 'line':
524
 
            self._match_case.hide()
525
 
            self._regexp.hide()
526
 
        # Be ready
527
 
        self._entry.grab_focus()
528
 
 
529
 
    def _set_label(self):
530
 
        if self._kind == 'line':
531
 
            self._label.set_text('Find Line: ')
532
 
        else:
533
 
            if self._regexp.get_active():
534
 
                self._label.set_text('Find Regexp: ')
535
 
            else:
536
 
                self._label.set_text('Find Text: ')
537
 
 
538
 
    def set_target(self, view,column):
539
 
        self._view = view
540
 
        self._column = column
541
 
 
542
 
    def _match(self, model, iterator, column):
543
 
        matching_case = self._match_case.get_active()
544
 
        cell_value, = model.get(iterator, column)
545
 
        key = self._entry.get_text()
546
 
        if column == LINE_NUM_COL:
547
 
            # FIXME: For goto-line there are faster algorithms than searching 
548
 
            # every line til we find the right one! -- mbp 2011-01-27
549
 
            return key.strip() == str(cell_value)
550
 
        elif self._regexp.get_active():
551
 
            if matching_case:
552
 
                match = re.compile(key).search(cell_value, 1)
553
 
            else:
554
 
                match = re.compile(key, re.I).search(cell_value, 1)
555
 
        else:
556
 
            if not matching_case:
557
 
                cell_value = cell_value.lower()
558
 
                key = key.lower()
559
 
            match = cell_value.find(key) != -1
560
 
 
561
 
        return match
562
 
 
563
 
    def _iterate_rows_forward(self, model, start):
564
 
        model_size = len(model)
565
 
        current = start + 1
566
 
        while model_size != 0:
567
 
            if current >= model_size: current =  0
568
 
            yield model.get_iter_from_string('%d' % current)
569
 
            if current == start: raise StopIteration
570
 
            current += 1
571
 
 
572
 
    def _iterate_rows_backward(self, model, start):
573
 
        model_size = len(model)
574
 
        current = start - 1
575
 
        while model_size != 0:
576
 
            if current < 0: current = model_size - 1
577
 
            yield model.get_iter_from_string('%d' % current)
578
 
            if current == start: raise StopIteration
579
 
            current -= 1
580
 
 
581
 
    def _do_search(self, direction):
582
 
        if direction == 'forward':
583
 
            iterate = self._iterate_rows_forward
584
 
        else:
585
 
            iterate = self._iterate_rows_backward
586
 
 
587
 
        model, sel = self._view.get_selection().get_selected()
588
 
        if sel is None:
589
 
            start = 0
590
 
        else:
591
 
            path = model.get_string_from_iter(sel)
592
 
            start = int(path)
593
 
 
594
 
        for row in iterate(model, start):
595
 
            if self._match(model, row, self._column):
596
 
                path = model.get_path(row)
597
 
                self._view.set_cursor(path, None, False)
598
 
                self._view.scroll_to_cell(path, use_align=True)
599
 
                break