58
self.span_selector.hide()
60
def annotate(self, branch, file_id):
58
def annotate(self, tree, branch, file_id):
61
59
self.revisions = {}
62
60
self.annotations = []
63
61
self.branch = branch
64
63
self.file_id = file_id
64
self.revision_id = getattr(tree, 'get_revision_id',
65
lambda: CURRENT_REVISION)()
66
67
# [revision id, line number, committer, revno, highlight color, line]
67
68
self.annomodel = gtk.ListStore(gobject.TYPE_STRING,
77
78
branch.repository.lock_read()
78
79
for line_no, (revision, revno, line)\
79
in enumerate(self._annotate(branch, file_id)):
80
in enumerate(self._annotate(tree, file_id)):
80
81
if revision.revision_id == last_seen and not self.all:
81
82
revno = committer = ""
99
100
self._set_oldest_newest()
100
# Recall that calling activate_default will emit "span-changed",
101
# so self._span_changed_cb will take care of initial highlighting
102
self.span_selector.activate_default()
102
self.annomodel.foreach(self._highlight_annotation, now)
104
104
branch.repository.unlock()
121
121
self.annoview.set_cursor(row)
122
122
self.annoview.scroll_to_cell(row, use_align=True)
124
def _annotate(self, branch, file_id):
125
rev_hist = branch.revision_history()
126
repository = branch.repository
127
rev_tree = repository.revision_tree(branch.last_revision())
128
rev_id = rev_tree.inventory[file_id].revision
129
weave = repository.weave_store.get_weave(file_id,
130
branch.get_transaction())
124
def _dotted_revnos(self, repository, revision_id):
125
"""Return a dict of revision_id -> dotted revno
127
:param repository: The repository to get the graph from
128
:param revision_id: The last revision for which this info is needed
130
graph = repository.get_revision_graph(revision_id)
132
for n, revision_id, d, revno, e in tsort.merge_sort(graph,
133
revision_id, generate_revno=True):
134
dotted[revision_id] = '.'.join(str(num) for num in revno)
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_revno = '%d?' % (self.branch.revno() + 1)
143
repository = self.branch.repository
144
if self.revision_id == CURRENT_REVISION:
145
revision_id = self.branch.last_revision()
147
revision_id = self.revision_id
148
dotted = self._dotted_revnos(repository, revision_id)
132
149
revision_cache = RevisionCache(repository)
133
for origin, text in weave.annotate_iter(rev_id):
150
for origin, text in tree.annotate_iter(file_id):
136
153
revision = revision_cache.get_revision(rev_id)
137
if rev_id in rev_hist:
138
revno = branch.revision_id_to_revno(rev_id)
154
revno = dotted.get(rev_id, 'merge')
141
157
except NoSuchRevision:
142
revision = NoneRevision(rev_id)
159
if rev_id == CURRENT_REVISION:
160
revision = current_revision
161
revno = current_revno
163
revision = FakeRevision(rev_id)
145
166
yield revision, revno, text
151
172
oldest = min(rev_dates)
152
173
newest = max(rev_dates)
154
span = self._span_from_seconds(time.time() - oldest)
155
self.span_selector.set_to_oldest_span(span)
157
span = self._span_from_seconds(newest - oldest)
158
self.span_selector.set_newest_to_oldest_span(span)
160
def _span_from_seconds(self, seconds):
161
return (seconds / (24 * 60 * 60))
163
def _span_changed_cb(self, w, span):
164
self.annotate_colormap.set_span(span)
166
self.annomodel.foreach(self._highlight_annotation, now)
168
175
def _highlight_annotation(self, model, path, iter, now):
169
176
revision_id, = model.get(iter, REVISION_ID_COL)
170
177
revision = self.revisions[revision_id]
201
207
vbox.pack_start(pane, expand=True, fill=True)
203
209
hbox = gtk.HBox(True, 6)
204
hbox.pack_start(self.span_selector, expand=False, fill=True)
205
210
hbox.pack_start(self._create_button_box(), expand=False, fill=True)
207
212
vbox.pack_start(hbox, expand=False, fill=True)
213
218
revision = self.annotations[row]
214
219
repository = self.branch.repository
215
tree1 = repository.revision_tree(revision.revision_id)
216
if len(revision.parent_ids) > 0:
217
tree2 = repository.revision_tree(revision.parent_ids[0])
220
if revision.revision_id == CURRENT_REVISION:
222
tree2 = self.tree.basis_tree()
219
tree2 = repository.revision_tree(NULL_REVISION)
224
tree1 = repository.revision_tree(revision.revision_id)
225
if len(revision.parent_ids) > 0:
226
tree2 = repository.revision_tree(revision.parent_ids[0])
228
tree2 = repository.revision_tree(NULL_REVISION)
220
229
from bzrlib.plugins.gtk.viz.diffwin import DiffWindow
221
230
window = DiffWindow()
222
231
window.set_diff("Diff for row %d" % (row+1), tree1, tree2)