/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 viz/branchwin.py

  • Committer: Aaron Bentley
  • Date: 2007-08-16 14:31:02 UTC
  • Revision ID: abentley@panoramicfeedback.com-20070816143102-27yzfhdgdv3ggo6e
Fix hue selection to use author in gannotate

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/branchwin.py
1
 
#!/usr/bin/python
2
1
# -*- coding: UTF-8 -*-
3
2
"""Branch window.
4
3
 
10
9
__author__    = "Scott James Remnant <scott@ubuntu.com>"
11
10
 
12
11
 
13
 
import os
14
 
 
15
12
import gtk
16
13
import gobject
17
14
import pango
52
49
 
53
50
    def construct(self):
54
51
        """Construct the window contents."""
 
52
        vbox = gtk.VBox(spacing=0)
 
53
        self.add(vbox)
 
54
 
 
55
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
56
 
55
57
        paned = gtk.VPaned()
56
58
        paned.pack1(self.construct_top(), resize=True, shrink=False)
57
59
        paned.pack2(self.construct_bottom(), resize=False, shrink=True)
58
 
        self.add(paned)
59
60
        paned.show()
 
61
        vbox.pack_start(paned, expand=True, fill=True)
 
62
        vbox.set_focus_child(paned)
 
63
 
 
64
        vbox.show()
60
65
 
61
66
    def construct_top(self):
62
67
        """Construct the top-half of the window."""
63
 
        vbox = gtk.VBox(spacing=6)
64
 
        vbox.set_border_width(12)
65
 
        vbox.show()
66
 
 
67
68
        scrollwin = gtk.ScrolledWindow()
68
69
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
69
70
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
70
 
        vbox.pack_start(scrollwin, expand=True, fill=True)
71
71
        scrollwin.show()
72
72
 
73
73
        self.treeview = gtk.TreeView()
74
74
        self.treeview.set_rules_hint(True)
75
75
        self.treeview.set_search_column(4)
76
76
        self.treeview.connect("cursor-changed", self._treeview_cursor_cb)
 
77
        self.treeview.connect("row-activated", self._treeview_row_activated_cb)
 
78
        self.treeview.connect("button-release-event", 
 
79
                self._treeview_row_mouseclick)
77
80
        scrollwin.add(self.treeview)
78
81
        self.treeview.show()
79
82
 
80
83
        cell = CellRendererGraph()
81
84
        column = gtk.TreeViewColumn()
82
 
        column.set_resizable(False)
 
85
        column.set_resizable(True)
83
86
        column.pack_start(cell, expand=False)
84
87
        column.add_attribute(cell, "node", 1)
85
88
        column.add_attribute(cell, "in-lines", 2)
112
115
        column.add_attribute(cell, "text", 6)
113
116
        self.treeview.append_column(column)
114
117
 
115
 
        hbox = gtk.HBox(False, spacing=6)
116
 
        vbox.pack_start(hbox, expand=False, fill=False)
 
118
        return scrollwin
 
119
 
 
120
    def construct_navigation(self):
 
121
        """Construct the navigation buttons."""
 
122
        frame = gtk.Frame()
 
123
        frame.set_shadow_type(gtk.SHADOW_OUT)
 
124
        frame.show()
 
125
        
 
126
        hbox = gtk.HBox(spacing=12)
 
127
        frame.add(hbox)
117
128
        hbox.show()
118
129
 
119
130
        self.back_button = gtk.Button(stock=gtk.STOCK_GO_BACK)
 
131
        self.back_button.set_relief(gtk.RELIEF_NONE)
120
132
        self.back_button.add_accelerator("clicked", self.accel_group, ord('['),
121
133
                                         0, 0)
122
134
        self.back_button.connect("clicked", self._back_clicked_cb)
124
136
        self.back_button.show()
125
137
 
126
138
        self.fwd_button = gtk.Button(stock=gtk.STOCK_GO_FORWARD)
 
139
        self.fwd_button.set_relief(gtk.RELIEF_NONE)
127
140
        self.fwd_button.add_accelerator("clicked", self.accel_group, ord(']'),
128
141
                                        0, 0)
129
142
        self.fwd_button.connect("clicked", self._fwd_clicked_cb)
130
143
        hbox.pack_start(self.fwd_button, expand=False, fill=True)
131
144
        self.fwd_button.show()
132
145
 
133
 
        return vbox
 
146
        return frame
134
147
 
135
148
    def construct_bottom(self):
136
149
        """Construct the bottom half of the window."""
137
 
        vbox = gtk.VBox(False, spacing=6)
138
 
        vbox.set_border_width(12)
 
150
        from bzrlib.plugins.gtk.logview import LogView
 
151
        self.logview = LogView()
139
152
        (width, height) = self.get_size()
140
 
        vbox.set_size_request(width, int(height / 2.5))
141
 
        vbox.show()
142
 
 
143
 
        self.table = gtk.Table(rows=4, columns=2)
144
 
        self.table.set_row_spacings(6)
145
 
        self.table.set_col_spacings(6)
146
 
        vbox.pack_start(self.table, expand=False, fill=True)
147
 
        self.table.show()
148
 
 
149
 
        align = gtk.Alignment(0.0, 0.5)
150
 
        label = gtk.Label()
151
 
        label.set_markup("<b>Revision:</b>")
152
 
        align.add(label)
153
 
        self.table.attach(align, 0, 1, 0, 1, gtk.FILL, gtk.FILL)
154
 
        label.show()
155
 
        align.show()
156
 
 
157
 
        align = gtk.Alignment(0.0, 0.5)
158
 
        self.revid_label = gtk.Label()
159
 
        self.revid_label.set_selectable(True)
160
 
        align.add(self.revid_label)
161
 
        self.table.attach(align, 1, 2, 0, 1, gtk.EXPAND | gtk.FILL, gtk.FILL)
162
 
        self.revid_label.show()
163
 
        align.show()
164
 
 
165
 
        align = gtk.Alignment(0.0, 0.5)
166
 
        label = gtk.Label()
167
 
        label.set_markup("<b>Committer:</b>")
168
 
        align.add(label)
169
 
        self.table.attach(align, 0, 1, 1, 2, gtk.FILL, gtk.FILL)
170
 
        label.show()
171
 
        align.show()
172
 
 
173
 
        align = gtk.Alignment(0.0, 0.5)
174
 
        self.committer_label = gtk.Label()
175
 
        self.committer_label.set_selectable(True)
176
 
        align.add(self.committer_label)
177
 
        self.table.attach(align, 1, 2, 1, 2, gtk.EXPAND | gtk.FILL, gtk.FILL)
178
 
        self.committer_label.show()
179
 
        align.show()
180
 
 
181
 
        align = gtk.Alignment(0.0, 0.5)
182
 
        label = gtk.Label()
183
 
        label.set_markup("<b>Timestamp:</b>")
184
 
        align.add(label)
185
 
        self.table.attach(align, 0, 1, 2, 3, gtk.FILL, gtk.FILL)
186
 
        label.show()
187
 
        align.show()
188
 
 
189
 
        align = gtk.Alignment(0.0, 0.5)
190
 
        self.timestamp_label = gtk.Label()
191
 
        self.timestamp_label.set_selectable(True)
192
 
        align.add(self.timestamp_label)
193
 
        self.table.attach(align, 1, 2, 2, 3, gtk.EXPAND | gtk.FILL, gtk.FILL)
194
 
        self.timestamp_label.show()
195
 
        align.show()
196
 
 
197
 
        align = gtk.Alignment(0.0, 0.5)
198
 
        label = gtk.Label()
199
 
        label.set_markup("<b>Parents:</b>")
200
 
        align.add(label)
201
 
        self.table.attach(align, 0, 1, 3, 4, gtk.FILL, gtk.FILL)
202
 
        label.show()
203
 
        align.show()
204
 
 
205
 
        self.parents_widgets = []
206
 
 
207
 
        scrollwin = gtk.ScrolledWindow()
208
 
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
209
 
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
210
 
        vbox.pack_start(scrollwin, expand=True, fill=True)
211
 
        scrollwin.show()
212
 
 
213
 
        self.message_buffer = gtk.TextBuffer()
214
 
        textview = gtk.TextView(self.message_buffer)
215
 
        textview.set_editable(False)
216
 
        textview.set_wrap_mode(gtk.WRAP_WORD)
217
 
        textview.modify_font(pango.FontDescription("Monospace"))
218
 
        scrollwin.add(textview)
219
 
        textview.show()
220
 
 
221
 
        return vbox
222
 
 
223
 
    def set_branch(self, branch, start):
 
153
        self.logview.set_size_request(width, int(height / 2.5))
 
154
        self.logview.show()
 
155
        self.logview.set_show_callback(self._show_clicked_cb)
 
156
        self.logview.set_go_callback(self._go_clicked_cb)
 
157
        return self.logview
 
158
 
 
159
    def set_branch(self, branch, start, maxnum):
224
160
        """Set the branch and start position for this window.
225
161
 
226
162
        Creates a new TreeModel and populates it with information about
227
163
        the new branch before updating the window title and model of the
228
164
        treeview itself.
229
165
        """
 
166
        self.branch = branch
 
167
 
230
168
        # [ revision, node, last_lines, lines, message, committer, timestamp ]
231
169
        self.model = gtk.ListStore(gobject.TYPE_PYOBJECT,
232
170
                                   gobject.TYPE_PYOBJECT,
237
175
        index = 0
238
176
 
239
177
        last_lines = []
240
 
        (revids, self.revisions, colours, self.children) \
241
 
                 = distances(branch, start)
242
 
        for revision, node, lines in graph(revids, self.revisions, colours):
 
178
        (self.revisions, colours, self.children, self.parent_ids,
 
179
            merge_sorted) = distances(branch.repository, start)
 
180
        for (index, (revision, node, lines)) in enumerate(graph(
 
181
                self.revisions, colours, merge_sorted)):
 
182
            # FIXME: at this point we should be able to show the graph order
 
183
            # and lines with no message or commit data - and then incrementally
 
184
            # fill the timestamp, committer etc data as desired.
243
185
            message = revision.message.split("\n")[0]
244
186
            if revision.committer is not None:
245
187
                timestamp = format_date(revision.timestamp, revision.timezone)
246
188
            else:
247
189
                timestamp = None
248
 
 
249
 
            self.model.append([ revision, node, last_lines, lines,
250
 
                                message, revision.committer, timestamp ])
 
190
            self.model.append([revision, node, last_lines, lines,
 
191
                               message, revision.committer, timestamp])
251
192
            self.index[revision] = index
252
 
            index += 1
253
 
 
254
193
            last_lines = lines
 
194
            if maxnum is not None and index > maxnum:
 
195
                break
255
196
 
256
 
        self.set_title(os.path.basename(branch.base) + " - bzrk")
 
197
        self.set_title(branch.nick + " - bzrk")
257
198
        self.treeview.set_model(self.model)
258
199
 
259
200
    def _treeview_cursor_cb(self, *args):
261
202
        (path, col) = self.treeview.get_cursor()
262
203
        revision = self.model[path][0]
263
204
 
264
 
        self.back_button.set_sensitive(len(revision.parent_ids) > 0)
 
205
        self.back_button.set_sensitive(len(self.parent_ids[revision]) > 0)
265
206
        self.fwd_button.set_sensitive(len(self.children[revision]) > 0)
266
 
 
267
 
        if revision.committer is not None:
268
 
            committer = revision.committer
269
 
            timestamp = format_date(revision.timestamp, revision.timezone)
270
 
            message = revision.message
271
 
        else:
272
 
            committer = ""
273
 
            timestamp = ""
274
 
            message = ""
275
 
 
276
 
        self.revid_label.set_text(revision.revision_id)
277
 
        self.committer_label.set_text(committer)
278
 
        self.timestamp_label.set_text(timestamp)
279
 
        self.message_buffer.set_text(message)
280
 
 
281
 
        for widget in self.parents_widgets:
282
 
            self.table.remove(widget)
283
 
 
284
 
        self.parents_widgets = []
285
 
        self.table.resize(4 + len(revision.parent_ids) - 1, 2)
286
 
        for idx, parent_id in enumerate(revision.parent_ids):
287
 
            self.table.set_row_spacing(idx + 3, 0)
288
 
 
289
 
            align = gtk.Alignment(0.0, 0.0)
290
 
            self.parents_widgets.append(align)
291
 
            self.table.attach(align, 1, 2, idx + 3, idx + 4,
292
 
                              gtk.EXPAND | gtk.FILL, gtk.FILL)
293
 
            align.show()
294
 
 
295
 
            hbox = gtk.HBox(False, 0)
296
 
            align.add(hbox)
297
 
            hbox.show()
298
 
 
299
 
            label = gtk.Label(parent_id)
300
 
            label.set_selectable(True)
301
 
            hbox.pack_start(label, expand=False, fill=True)
302
 
            label.show()
303
 
 
304
 
            image = gtk.Image()
305
 
            image.set_from_stock(gtk.STOCK_JUMP_TO, gtk.ICON_SIZE_MENU)
306
 
            image.show()
307
 
 
308
 
            button = gtk.Button()
309
 
            button.add(image)
310
 
            button.set_relief(gtk.RELIEF_NONE)
311
 
            button.connect("clicked", self._go_clicked_cb, parent_id)
312
 
            hbox.pack_start(button, expand=False, fill=True)
313
 
            button.show()
314
 
 
315
 
            image = gtk.Image()
316
 
            image.set_from_stock(gtk.STOCK_FIND, gtk.ICON_SIZE_MENU)
317
 
            image.show()
318
 
 
319
 
            button = gtk.Button()
320
 
            button.add(image)
321
 
            button.set_relief(gtk.RELIEF_NONE)
322
 
            button.set_sensitive(False)
323
 
            button.connect("clicked", self._show_clicked_cb, parent_id)
324
 
            hbox.pack_start(button, expand=False, fill=True)
325
 
            button.show()
 
207
        tags = []
 
208
        if self.branch.supports_tags():
 
209
            tagdict = self.branch.tags.get_reverse_tag_dict()
 
210
            if tagdict.has_key(revision.revision_id):
 
211
                tags = tagdict[revision.revision_id]
 
212
        self.logview.set_revision(revision, tags)
326
213
 
327
214
    def _back_clicked_cb(self, *args):
328
215
        """Callback for when the back button is clicked."""
329
216
        (path, col) = self.treeview.get_cursor()
330
217
        revision = self.model[path][0]
331
 
        if not len(revision.parent_ids):
 
218
        if not len(self.parent_ids[revision]):
332
219
            return
333
220
 
334
 
        for parent_id in revision.parent_ids:
 
221
        for parent_id in self.parent_ids[revision]:
335
222
            parent = self.revisions[parent_id]
336
223
            if same_branch(revision, parent):
337
224
                self.treeview.set_cursor(self.index[parent])
338
225
                break
339
226
        else:
340
 
            next = self.revisions[revision.parent_ids[0]]
 
227
            next = self.revisions[self.parent_ids[revision][0]]
341
228
            self.treeview.set_cursor(self.index[next])
 
229
        self.treeview.grab_focus()
342
230
 
343
231
    def _fwd_clicked_cb(self, *args):
344
232
        """Callback for when the forward button is clicked."""
354
242
        else:
355
243
            prev = list(self.children[revision])[0]
356
244
            self.treeview.set_cursor(self.index[prev])
 
245
        self.treeview.grab_focus()
357
246
 
358
 
    def _go_clicked_cb(self, widget, revid):
 
247
    def _go_clicked_cb(self, revid):
359
248
        """Callback for when the go button for a parent is clicked."""
360
249
        self.treeview.set_cursor(self.index[self.revisions[revid]])
361
 
 
362
 
    def _show_clicked_cb(self, widget, revid):
 
250
        self.treeview.grab_focus()
 
251
 
 
252
    def show_diff(self, branch, revid, parentid):
 
253
        """Open a new window to show a diff between the given revisions."""
 
254
        from bzrlib.plugins.gtk.diff import DiffWindow
 
255
        window = DiffWindow()
 
256
        (parent_tree, rev_tree) = branch.repository.revision_trees([parentid, 
 
257
                                                                   revid])
 
258
        description = revid + " - " + branch.nick
 
259
        window.set_diff(description, rev_tree, parent_tree)
 
260
        window.show()
 
261
 
 
262
    def _show_clicked_cb(self, revid, parentid):
363
263
        """Callback for when the show button for a parent is clicked."""
364
 
        print "SHOW %s" % revid
 
264
        self.show_diff(self.branch, revid, parentid)
 
265
        self.treeview.grab_focus()
 
266
 
 
267
    def _treeview_row_mouseclick(self, widget, event):
 
268
        from bzrlib.plugins.gtk.revisionmenu import RevisionPopupMenu
 
269
        if event.button == 3:
 
270
            menu = RevisionPopupMenu(self.branch.repository, 
 
271
                [x.revision_id for x in self.selected_revisions()],
 
272
                self.branch)
 
273
            menu.popup(None, None, None, event.button, event.get_time())
 
274
 
 
275
    def selected_revision(self, path):
 
276
        return self.model[path][0]
 
277
 
 
278
    def selected_revisions(self):
 
279
        return [self.selected_revision(path) for path in \
 
280
                self.treeview.get_selection().get_selected_rows()[1]]
 
281
 
 
282
    def _treeview_row_activated_cb(self, widget, path, col):
 
283
        # TODO: more than one parent
 
284
        """Callback for when a treeview row gets activated."""
 
285
        revision = self.selected_revision(path)
 
286
        if len(self.parent_ids[revision]) == 0:
 
287
            # Ignore revisions without parent
 
288
            return
 
289
        parent_id = self.parent_ids[revision][0]
 
290
        self.show_diff(self.branch, revision.revision_id, parent_id)
 
291
        self.treeview.grab_focus()