45
class GAnnotateWindow(Window):
44
class GAnnotateWindow(gtk.Window):
46
45
"""Annotate window."""
48
def __init__(self, all=False, plain=False, parent=None, branch=None):
47
def __init__(self, all=False, plain=False):
53
Window.__init__(self, parent)
51
gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
55
53
self.set_icon(self.render_icon(gtk.STOCK_FIND, gtk.ICON_SIZE_BUTTON))
56
54
self.annotate_colormap = AnnotateColorSaturation()
59
self.span_selector.hide()
61
def annotate(self, branch, file_id, revision_id=None):
59
62
self.revisions = {}
63
def annotate(self, tree, branch, file_id):
64
63
self.annotations = []
65
64
self.branch = branch
67
65
self.file_id = file_id
68
self.revisionview.set_file_id(file_id)
69
self.revision_id = getattr(tree, 'get_revision_id',
70
lambda: CURRENT_REVISION)()
66
self.revision_id = revision_id
72
# [revision id, line number, author, revno, highlight color, line]
68
# [revision id, line number, committer, revno, highlight color, line]
73
69
self.annomodel = gtk.ListStore(gobject.TYPE_STRING,
74
70
gobject.TYPE_STRING,
75
71
gobject.TYPE_STRING,
132
123
self.annoview.set_cursor(row)
133
124
self.annoview.scroll_to_cell(row, use_align=True)
126
def _dotted_revnos(self, repository, revision_id):
127
"""Return a dict of revision_id -> dotted revno
129
:param repository: The repository to get the graph from
130
:param revision_id: The last revision for which this info is needed
132
graph = repository.get_revision_graph(revision_id)
134
for n, revision_id, d, revno, e in tsort.merge_sort(graph,
135
revision_id, generate_revno=True):
136
dotted[revision_id] = '.'.join(str(num) for num in revno)
136
def _annotate(self, tree, file_id):
137
current_revision = FakeRevision(CURRENT_REVISION)
138
current_revision.committer = self.branch.get_config().username()
139
current_revision.timestamp = time.time()
140
current_revision.message = '[Not yet committed]'
141
current_revision.parent_ids = tree.get_parent_ids()
142
current_revision.properties['branch-nick'] = self.branch.nick
143
current_revno = '%d?' % (self.branch.revno() + 1)
144
repository = self.branch.repository
145
if self.revision_id == CURRENT_REVISION:
146
revision_id = self.branch.last_revision()
148
revision_id = self.revision_id
149
revision_cache = RevisionCache(repository, self.revisions)
150
for origin, text in tree.annotate_iter(file_id):
139
def _annotate(self, branch, file_id, revision_id):
140
repository = branch.repository
141
if revision_id is None:
142
revision_id = branch.last_revision()
143
dotted = self._dotted_revnos(repository, revision_id)
144
rev_tree = repository.revision_tree(revision_id)
145
revision_id = rev_tree.inventory[file_id].revision
146
weave = repository.weave_store.get_weave(file_id,
147
branch.get_transaction())
149
revision_cache = RevisionCache(repository)
150
for origin, text in weave.annotate_iter(revision_id):
152
if rev_id == CURRENT_REVISION:
153
revision = current_revision
154
revno = current_revno
157
revision = revision_cache.get_revision(rev_id)
158
revno = self.dotted.get(rev_id, 'merge')
161
except NoSuchRevision:
162
revision = FakeRevision(rev_id)
153
revision = revision_cache.get_revision(rev_id)
154
revno = dotted.get(rev_id, 'merge')
157
except NoSuchRevision:
158
revision = NoneRevision(rev_id)
165
161
yield revision, revno, text
163
def _set_oldest_newest(self):
164
rev_dates = map(lambda i: self.revisions[i].timestamp, self.revisions)
165
if len(rev_dates) == 0:
167
oldest = min(rev_dates)
168
newest = max(rev_dates)
170
span = self._span_from_seconds(time.time() - oldest)
171
self.span_selector.set_to_oldest_span(span)
173
span = self._span_from_seconds(newest - oldest)
174
self.span_selector.set_newest_to_oldest_span(span)
176
def _span_from_seconds(self, seconds):
177
return (seconds / (24 * 60 * 60))
179
def _span_changed_cb(self, w, span):
180
self.annotate_colormap.set_span(span)
182
self.annomodel.foreach(self._highlight_annotation, now)
167
184
def _highlight_annotation(self, model, path, iter, now):
168
185
revision_id, = model.get(iter, REVISION_ID_COL)
169
186
revision = self.revisions[revision_id]
170
187
model.set(iter, HIGHLIGHT_COLOR_COL,
171
188
self.annotate_colormap.get_color(revision, now))
173
def _selected_revision(self):
190
def _show_log(self, w):
174
191
(path, col) = self.annoview.get_cursor()
177
return self.annomodel[path][REVISION_ID_COL]
179
def _activate_selected_revision(self, w):
180
rev_id = self._selected_revision()
181
if not rev_id or rev_id == NULL_REVISION:
183
selected = self.revisions[rev_id]
184
self.revisionview.set_revision(selected)
185
if (len(selected.parent_ids) != 0 and selected.parent_ids[0] not in
190
self.back_button.set_sensitive(enable_back)
194
rev_id = self.annomodel[path][REVISION_ID_COL]
195
self.logview.set_revision(self.revisions[rev_id])
192
197
def _create(self):
193
self.revisionview = self._create_log_view()
198
self.logview = self._create_log_view()
194
199
self.annoview = self._create_annotate_view()
200
self.span_selector = self._create_span_selector()
196
vbox = gtk.VBox(False)
202
vbox = gtk.VBox(False, 12)
203
vbox.set_border_width(12)
199
206
sw = gtk.ScrolledWindow()
202
209
sw.add(self.annoview)
203
210
self.annoview.gwindow = self
210
hbox = gtk.HBox(False, 6)
211
self.back_button = self._create_back_button()
212
hbox.pack_start(self.back_button, expand=False, fill=True)
213
self.forward_button = self._create_forward_button()
214
hbox.pack_start(self.forward_button, expand=False, fill=True)
216
vbox.pack_start(hbox, expand=False, fill=True)
218
213
self.pane = pane = gtk.VPaned()
220
pane.add2(self.revisionview)
215
pane.add2(self.logview)
222
217
vbox.pack_start(pane, expand=True, fill=True)
224
self._search = SearchBox()
225
swbox.pack_start(self._search, expand=False, fill=True)
226
accels = gtk.AccelGroup()
227
accels.connect_group(gtk.keysyms.f, gtk.gdk.CONTROL_MASK,
229
self._search_by_text)
230
accels.connect_group(gtk.keysyms.g, gtk.gdk.CONTROL_MASK,
232
self._search_by_line)
233
self.add_accel_group(accels)
219
hbox = gtk.HBox(True, 6)
220
hbox.pack_start(self.span_selector, expand=False, fill=True)
221
hbox.pack_start(self._create_button_box(), expand=False, fill=True)
223
vbox.pack_start(hbox, expand=False, fill=True)
237
def _search_by_text(self, accel_group, window, key, modifiers):
238
self._search.show_for('text')
239
self._search.set_target(self.annoview, TEXT_LINE_COL)
241
def _search_by_line(self, accel_group, window, key, modifiers):
242
self._search.show_for('line')
243
self._search.set_target(self.annoview, LINE_NUM_COL)
245
def line_diff(self, tv, path, tvc):
227
def row_diff(self, tv, path, tvc):
247
229
revision = self.annotations[row]
248
230
repository = self.branch.repository
249
if revision.revision_id == CURRENT_REVISION:
251
tree2 = self.tree.basis_tree()
231
tree1 = repository.revision_tree(revision.revision_id)
232
if len(revision.parent_ids) > 0:
233
tree2 = repository.revision_tree(revision.parent_ids[0])
253
tree1 = repository.revision_tree(revision.revision_id)
254
if len(revision.parent_ids) > 0:
255
tree2 = repository.revision_tree(revision.parent_ids[0])
257
tree2 = repository.revision_tree(NULL_REVISION)
258
from bzrlib.plugins.gtk.diff import DiffWindow
235
tree2 = repository.revision_tree(NULL_REVISION)
236
from bzrlib.plugins.gtk.viz.diffwin import DiffWindow
259
237
window = DiffWindow()
260
window.set_diff("Diff for line %d" % (row+1), tree1, tree2)
238
window.set_diff("Diff for row %d" % (row+1), tree1, tree2)
261
239
window.set_file(tree1.id2path(self.file_id))
314
292
col.add_attribute(cell, "text", TEXT_LINE_COL)
315
293
tv.append_column(col)
317
# FIXME: Now that C-f is now used for search by text we
318
# may as well disable the auto search.
319
295
tv.set_search_column(LINE_NUM_COL)
299
def _create_span_selector(self):
301
ss.connect("span-changed", self._span_changed_cb)
323
306
def _create_log_view(self):
324
lv = RevisionView(self._branch)
328
def _create_back_button(self):
329
button = gtk.Button()
330
button.set_use_stock(True)
331
button.set_label("gtk-go-back")
332
button.connect("clicked", lambda w: self.go_back())
333
button.set_relief(gtk.RELIEF_NONE)
337
def _create_forward_button(self):
338
button = gtk.Button()
339
button.set_use_stock(True)
340
button.set_label("gtk-go-forward")
341
button.connect("clicked", lambda w: self.go_forward())
342
button.set_relief(gtk.RELIEF_NONE)
344
button.set_sensitive(False)
348
last_tree = self.tree
349
rev_id = self._selected_revision()
350
parent_id = self.revisions[rev_id].parent_ids[0]
351
target_tree = self.branch.repository.revision_tree(parent_id)
352
if self._go(target_tree):
353
self.history.append(last_tree)
354
self.forward_button.set_sensitive(True)
356
self._no_back.add(parent_id)
357
self.back_button.set_sensitive(False)
359
def go_forward(self):
360
if len(self.history) == 0:
362
target_tree = self.history.pop()
363
if len(self.history) == 0:
364
self.forward_button.set_sensitive(False)
365
self._go(target_tree)
367
def _go(self, target_tree):
368
rev_id = self._selected_revision()
369
if self.file_id in target_tree:
370
offset = self.get_scroll_offset(target_tree)
371
(row,), col = self.annoview.get_cursor()
372
self.annotate(target_tree, self.branch, self.file_id)
376
self.annoview.set_cursor(new_row)
381
def get_scroll_offset(self, tree):
382
old = self.tree.get_file(self.file_id)
383
new = tree.get_file(self.file_id)
384
(row,), col = self.annoview.get_cursor()
385
matcher = patiencediff.PatienceSequenceMatcher(None, old.readlines(),
387
for i, j, n in matcher.get_matching_blocks():
312
def _create_button_box(self):
313
box = gtk.HButtonBox()
314
box.set_layout(gtk.BUTTONBOX_END)
317
button = gtk.Button()
318
button.set_use_stock(True)
319
button.set_label("gtk-close")
320
button.connect("clicked", lambda w: self.destroy())
323
box.pack_start(button, expand=False, fill=False)
393
329
""" A fake revision.
395
331
For when a revision is referenced but not present.
398
def __init__(self, revision_id, committer='?', nick=None):
334
def __init__(self, revision_id):
399
335
self.revision_id = revision_id
400
336
self.parent_ids = []
401
self.committer = committer
402
338
self.message = "?"
403
339
self.timestamp = 0.0
404
340
self.timezone = 0
407
def get_apparent_author(self):
408
return self.committer
411
343
class RevisionCache(object):
412
344
"""A caching revision source"""
413
def __init__(self, real_source, seed_cache=None):
345
def __init__(self, real_source):
414
346
self.__real_source = real_source
415
if seed_cache is None:
418
self.__cache = dict(seed_cache)
420
349
def get_revision(self, revision_id):
421
350
if revision_id not in self.__cache:
422
351
revision = self.__real_source.get_revision(revision_id)
423
352
self.__cache[revision_id] = revision
424
353
return self.__cache[revision_id]
426
class SearchBox(gtk.HBox):
427
"""A button box for searching in text or lines of annotations"""
429
gtk.HBox.__init__(self, False, 6)
432
button = gtk.Button()
434
image.set_from_stock('gtk-stop', gtk.ICON_SIZE_BUTTON)
435
button.set_image(image)
436
button.set_relief(gtk.RELIEF_NONE)
437
button.connect("clicked", lambda w: self.hide_all())
438
self.pack_start(button, expand=False, fill=False)
443
self.pack_start(label, expand=False, fill=False)
447
entry.connect("activate", lambda w, d: self._do_search(d),
449
self.pack_start(entry, expand=False, fill=False)
451
# Next/previous buttons
452
button = gtk.Button('_Next')
454
image.set_from_stock('gtk-go-forward', gtk.ICON_SIZE_BUTTON)
455
button.set_image(image)
456
button.connect("clicked", lambda w, d: self._do_search(d),
458
self.pack_start(button, expand=False, fill=False)
460
button = gtk.Button('_Previous')
462
image.set_from_stock('gtk-go-back', gtk.ICON_SIZE_BUTTON)
463
button.set_image(image)
464
button.connect("clicked", lambda w, d: self._do_search(d),
466
self.pack_start(button, expand=False, fill=False)
469
check = gtk.CheckButton('Match case')
470
self._match_case = check
471
self.pack_start(check, expand=False, fill=False)
473
check = gtk.CheckButton('Regexp')
474
check.connect("toggled", lambda w: self._set_label())
476
self.pack_start(check, expand=False, fill=False)
480
# Note that we stay hidden (we do not call self.show_all())
483
def show_for(self, kind):
487
# Hide unrelated buttons
489
self._match_case.hide()
492
self._entry.grab_focus()
494
def _set_label(self):
495
if self._kind == 'line':
496
self._label.set_text('Find Line: ')
498
if self._regexp.get_active():
499
self._label.set_text('Find Regexp: ')
501
self._label.set_text('Find Text: ')
503
def set_target(self, view,column):
505
self._column = column
507
def _match(self, model, iterator, column):
508
matching_case = self._match_case.get_active()
509
string, = model.get(iterator, column)
510
key = self._entry.get_text()
511
if self._regexp.get_active():
513
match = re.compile(key).search(string, 1)
515
match = re.compile(key, re.I).search(string, 1)
517
if not matching_case:
518
string = string.lower()
520
match = string.find(key) != -1
524
def _iterate_rows_forward(self, model, start):
525
model_size = len(model)
527
while model_size != 0:
528
if current >= model_size: current = 0
529
yield model.get_iter_from_string('%d' % current)
530
if current == start: raise StopIteration
533
def _iterate_rows_backward(self, model, start):
534
model_size = len(model)
536
while model_size != 0:
537
if current < 0: current = model_size - 1
538
yield model.get_iter_from_string('%d' % current)
539
if current == start: raise StopIteration
542
def _do_search(self, direction):
543
if direction == 'forward':
544
iterate = self._iterate_rows_forward
546
iterate = self._iterate_rows_backward
548
model, sel = self._view.get_selection().get_selected()
552
path = model.get_string_from_iter(sel)
555
for row in iterate(model, start):
556
if self._match(model, row, self._column):
557
path = model.get_path(row)
558
self._view.set_cursor(path)
559
self._view.scroll_to_cell(path, use_align=True)