/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: Jelmer Vernooij
  • Date: 2008-06-29 19:34:29 UTC
  • mto: This revision was merged to the branch mainline in revision 520.
  • Revision ID: jelmer@samba.org-20080629193429-ir2ilmbko6qkubg5
Add Branch/Index option if bzr-search is available.

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/branchwin.py
14
14
import pango
15
15
 
16
16
from bzrlib.plugins.gtk.window import Window
 
17
from bzrlib.plugins.gtk import icon_path
17
18
from bzrlib.plugins.gtk.tags import AddTagDialog
18
19
from bzrlib.plugins.gtk.preferences import PreferencesWindow
19
 
from bzrlib.revision import Revision
20
 
from bzrlib.config import BranchConfig
21
 
from bzrlib.config import GlobalConfig
22
 
from treeview import TreeView
 
20
from bzrlib.plugins.gtk.branchview import TreeView, treemodel
 
21
 
 
22
from bzrlib.config import BranchConfig, GlobalConfig
 
23
from bzrlib.revision import Revision, NULL_REVISION
 
24
from bzrlib.trace import mutter
23
25
 
24
26
class BranchWindow(Window):
25
27
    """Branch window.
28
30
    for a particular branch.
29
31
    """
30
32
 
31
 
    def __init__(self, branch, start, maxnum, parent=None):
 
33
    def __init__(self, branch, start_revs, maxnum, parent=None):
32
34
        """Create a new BranchWindow.
33
35
 
34
36
        :param branch: Branch object for branch to show.
35
 
        :param start: Revision id of top revision.
 
37
        :param start_revs: Revision ids of top revisions.
36
38
        :param maxnum: Maximum number of revisions to display, 
37
39
                       None for no limit.
38
40
        """
41
43
        self.set_border_width(0)
42
44
 
43
45
        self.branch      = branch
44
 
        self.start       = start
 
46
        self.start_revs  = start_revs
45
47
        self.maxnum      = maxnum
46
48
        self.config      = GlobalConfig()
47
49
 
65
67
 
66
68
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
67
69
        gtk.accel_map_add_entry("<viz>/Go/Previous Revision", gtk.keysyms.Down, gtk.gdk.MOD1_MASK)
 
70
        gtk.accel_map_add_entry("<viz>/View/Refresh", gtk.keysyms.F5, 0)
68
71
 
69
72
        self.accel_group = gtk.AccelGroup()
70
73
        self.add_accel_group(self.accel_group)
83
86
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
84
87
        self.next_rev_action.connect_accelerator()
85
88
 
 
89
        self.refresh_action = gtk.Action("refresh", "_Refresh", "Refresh view", gtk.STOCK_REFRESH)
 
90
        self.refresh_action.set_accel_path("<viz>/View/Refresh")
 
91
        self.refresh_action.set_accel_group(self.accel_group)
 
92
        self.refresh_action.connect("activate", self._refresh_clicked)
 
93
        self.refresh_action.connect_accelerator()
 
94
 
86
95
        self.construct()
87
96
 
88
97
    def set_revision(self, revid):
100
109
 
101
110
        vbox.pack_start(self.construct_menubar(), expand=False, fill=True)
102
111
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
103
 
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
104
112
        
105
113
        vbox.pack_start(self.paned, expand=True, fill=True)
106
114
        vbox.set_focus_child(self.paned)
144
152
        view_menuitem = gtk.MenuItem("_View")
145
153
        view_menuitem.set_submenu(view_menu)
146
154
 
 
155
        view_menu_refresh = self.refresh_action.create_menu_item()
 
156
        view_menu_refresh.connect('activate', self._refresh_clicked)
 
157
 
 
158
        view_menu.add(view_menu_refresh)
 
159
        view_menu.add(gtk.SeparatorMenuItem())
 
160
 
147
161
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
148
162
        view_menu_toolbar.set_active(True)
149
163
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
156
170
        view_menu.add(view_menu_compact)
157
171
        view_menu.add(gtk.SeparatorMenuItem())
158
172
 
159
 
        for (label, name) in [("Revision _Number", "revno"), ("_Date", "date")]:
160
 
            col = gtk.CheckMenuItem("Show " + label + " Column")
 
173
        self.mnu_show_revno_column = gtk.CheckMenuItem("Show Revision _Number Column")
 
174
        self.mnu_show_date_column = gtk.CheckMenuItem("Show _Date Column")
 
175
 
 
176
        # Revision numbers are pointless if there are multiple branches
 
177
        if len(self.start_revs) > 1:
 
178
            self.mnu_show_revno_column.set_sensitive(False)
 
179
            self.treeview.set_property('revno-column-visible', False)
 
180
 
 
181
        for (col, name) in [(self.mnu_show_revno_column, "revno"), 
 
182
                            (self.mnu_show_date_column, "date")]:
161
183
            col.set_active(self.treeview.get_property(name + "-column-visible"))
162
184
            col.connect('toggled', self._col_visibility_changed, name)
163
185
            view_menu.add(col)
170
192
        go_menu_next = self.next_rev_action.create_menu_item()
171
193
        go_menu_prev = self.prev_rev_action.create_menu_item()
172
194
 
173
 
        tags_menu = gtk.Menu()
174
 
        go_menu_tags = gtk.MenuItem("_Tags")
175
 
        go_menu_tags.set_submenu(tags_menu)
176
 
 
177
 
        if self.branch.supports_tags():
178
 
            tags = self.branch.tags.get_tag_dict().items()
179
 
            tags.sort()
180
 
            tags.reverse()
181
 
            for tag, revid in tags:
182
 
                tag_item = gtk.MenuItem(tag)
183
 
                tag_item.connect('activate', self._tag_selected_cb, revid)
184
 
                tags_menu.add(tag_item)
185
 
 
186
 
            go_menu_tags.set_sensitive(len(tags) != 0)
187
 
        else:
188
 
            go_menu_tags.set_sensitive(False)
 
195
        tag_image = gtk.Image()
 
196
        tag_image.set_from_file(icon_path("tag-16.png"))
 
197
        self.go_menu_tags = gtk.ImageMenuItem("_Tags")
 
198
        self.go_menu_tags.set_image(tag_image)
 
199
        self._update_tags()
189
200
 
190
201
        go_menu.add(go_menu_next)
191
202
        go_menu.add(go_menu_prev)
192
203
        go_menu.add(gtk.SeparatorMenuItem())
193
 
        go_menu.add(go_menu_tags)
 
204
        go_menu.add(self.go_menu_tags)
194
205
 
195
206
        revision_menu = gtk.Menu()
196
207
        revision_menuitem = gtk.MenuItem("_Revision")
198
209
 
199
210
        revision_menu_diff = gtk.MenuItem("View Changes")
200
211
        revision_menu_diff.connect('activate', 
201
 
                lambda w: self.treeview.show_diff())
 
212
                self._menu_diff_cb)
 
213
        
 
214
        revision_menu_compare = gtk.MenuItem("Compare with...")
 
215
        revision_menu_compare.connect('activate',
 
216
                self._compare_with_cb)
202
217
 
203
218
        revision_menu_tag = gtk.MenuItem("Tag Revision")
204
219
        revision_menu_tag.connect('activate', self._tag_revision_cb)
205
220
 
206
221
        revision_menu.add(revision_menu_tag)
207
222
        revision_menu.add(revision_menu_diff)
 
223
        revision_menu.add(revision_menu_compare)
208
224
 
209
225
        branch_menu = gtk.Menu()
210
226
        branch_menuitem = gtk.MenuItem("_Branch")
213
229
        branch_menu.add(gtk.MenuItem("Pu_ll Revisions"))
214
230
        branch_menu.add(gtk.MenuItem("Pu_sh Revisions"))
215
231
 
 
232
        try:
 
233
            from bzrlib.plugins import search
 
234
        except ImportError:
 
235
            mutter("Didn't find search plugin")
 
236
        else:
 
237
            branch_index_menuitem = gtk.MenuItem("_Index")
 
238
            branch_index_menuitem.connect('activate', self._branch_index_cb)
 
239
            branch_menu.add(branch_index_menuitem)
 
240
 
 
241
        help_menu = gtk.Menu()
 
242
        help_menuitem = gtk.MenuItem("_Help")
 
243
        help_menuitem.set_submenu(help_menu)
 
244
 
 
245
        help_about_menuitem = gtk.ImageMenuItem(gtk.STOCK_ABOUT, self.accel_group)
 
246
        help_about_menuitem.connect('activate', self._about_dialog_cb)
 
247
 
 
248
        help_menu.add(help_about_menuitem)
 
249
 
216
250
        menubar.add(file_menuitem)
217
251
        menubar.add(edit_menuitem)
218
252
        menubar.add(view_menuitem)
219
253
        menubar.add(go_menuitem)
220
254
        menubar.add(revision_menuitem)
221
255
        menubar.add(branch_menuitem)
 
256
        menubar.add(help_menuitem)
222
257
        menubar.show_all()
223
258
 
224
259
        return menubar
225
 
    
226
 
    def construct_loading_msg(self):
227
 
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
228
 
                                                 gtk.ICON_SIZE_BUTTON)
229
 
        image_loading.show()
230
 
        
231
 
        label_loading = gtk.Label(_("Please wait, loading ancestral graph..."))
232
 
        label_loading.set_alignment(0.0, 0.5)
233
 
        label_loading.show()
234
 
        
235
 
        self.loading_msg_box = gtk.HBox()
236
 
        self.loading_msg_box.set_spacing(5)
237
 
        self.loading_msg_box.set_border_width(5)        
238
 
        self.loading_msg_box.pack_start(image_loading, False, False)
239
 
        self.loading_msg_box.pack_start(label_loading, True, True)
240
 
        self.loading_msg_box.show()
241
 
        
242
 
        return self.loading_msg_box
243
260
 
244
261
    def construct_top(self):
245
262
        """Construct the top-half of the window."""
246
263
        # FIXME: Make broken_line_length configurable
247
264
 
248
 
        self.treeview = TreeView(self.branch, self.start, self.maxnum, self.compact_view)
 
265
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
249
266
 
250
267
        self.treeview.connect('revision-selected',
251
268
                self._treeselection_changed_cb)
 
269
        self.treeview.connect('revision-activated',
 
270
                self._tree_revision_activated)
252
271
 
253
 
        self.treeview.connect('revisions-loaded', 
254
 
                lambda x: self.loading_msg_box.hide())
 
272
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
255
273
 
256
274
        for col in ["revno", "date"]:
257
275
            option = self.config.get_user_option(col + '-column-visible')
258
276
            if option is not None:
259
277
                self.treeview.set_property(col + '-column-visible', option == 'True')
 
278
            else:
 
279
                self.treeview.set_property(col + '-column-visible', False)
260
280
 
261
281
        self.treeview.show()
262
282
 
297
317
        self.revisionview.show()
298
318
        self.revisionview.set_show_callback(self._show_clicked_cb)
299
319
        self.revisionview.connect('notify::revision', self._go_clicked_cb)
 
320
        self.treeview.connect('tag-added', lambda w, t, r: self.revisionview.update_tags())
300
321
        return self.revisionview
301
322
 
302
323
    def _tag_selected_cb(self, menuitem, revid):
308
329
        parents  = self.treeview.get_parents()
309
330
        children = self.treeview.get_children()
310
331
 
311
 
        if revision is not None:
 
332
        if revision and revision != NULL_REVISION:
312
333
            prev_menu = gtk.Menu()
313
334
            if len(parents) > 0:
314
335
                self.prev_rev_action.set_sensitive(True)
315
336
                for parent_id in parents:
316
 
                    parent = self.branch.repository.get_revision(parent_id)
317
 
                    try:
318
 
                        str = ' (' + parent.properties['branch-nick'] + ')'
319
 
                    except KeyError:
320
 
                        str = ""
 
337
                    if parent_id and parent_id != NULL_REVISION:
 
338
                        parent = self.branch.repository.get_revision(parent_id)
 
339
                        try:
 
340
                            str = ' (' + parent.properties['branch-nick'] + ')'
 
341
                        except KeyError:
 
342
                            str = ""
321
343
 
322
 
                    item = gtk.MenuItem(parent.message.split("\n")[0] + str)
323
 
                    item.connect('activate', self._set_revision_cb, parent_id)
324
 
                    prev_menu.add(item)
 
344
                        item = gtk.MenuItem(parent.message.split("\n")[0] + str)
 
345
                        item.connect('activate', self._set_revision_cb, parent_id)
 
346
                        prev_menu.add(item)
325
347
                prev_menu.show_all()
326
348
            else:
327
349
                self.prev_rev_action.set_sensitive(False)
351
373
 
352
374
            self.revisionview.set_revision(revision)
353
375
            self.revisionview.set_children(children)
 
376
    
 
377
    def _tree_revision_activated(self, widget, path, col):
 
378
        # TODO: more than one parent
 
379
        """Callback for when a treeview row gets activated."""
 
380
        revision = self.treeview.get_revision()
 
381
        parents  = self.treeview.get_parents()
 
382
 
 
383
        if len(parents) == 0:
 
384
            parent_id = None
 
385
        else:
 
386
            parent_id = parents[0]
 
387
 
 
388
        self.show_diff(revision.revision_id, parent_id)
 
389
        self.treeview.grab_focus()
 
390
        
 
391
    def _menu_diff_cb(self,w):
 
392
        (path, focus) = self.treeview.treeview.get_cursor()
 
393
        revid = self.treeview.model[path][treemodel.REVID]
 
394
        
 
395
        parentids = self.branch.repository.revision_parents(revid)
 
396
 
 
397
        if len(parentids) == 0:
 
398
            parentid = NULL_REVISION
 
399
        else:
 
400
            parentid = parentids[0]
 
401
        
 
402
        self.show_diff(revid,parentid)    
354
403
 
355
404
    def _back_clicked_cb(self, *args):
356
405
        """Callback for when the back button is clicked."""
367
416
 
368
417
    def _show_clicked_cb(self, revid, parentid):
369
418
        """Callback for when the show button for a parent is clicked."""
370
 
        self.treeview.show_diff(revid, parentid)
 
419
        self.show_diff(revid, parentid)
371
420
        self.treeview.grab_focus()
372
421
 
 
422
    def _compare_with_cb(self,w):
 
423
        """Callback for revision 'compare with' menu. Will show a small
 
424
            dialog with branch revisions to compare with selected revision in TreeView"""
 
425
        
 
426
        from bzrlib.plugins.gtk.revbrowser import RevisionBrowser
 
427
        
 
428
        rb = RevisionBrowser(self.branch,self)
 
429
        ret = rb.run()
 
430
        
 
431
        if ret == gtk.RESPONSE_OK:          
 
432
            (path, focus) = self.treeview.treeview.get_cursor()
 
433
            revid = self.treeview.model[path][treemodel.REVID]
 
434
            self.show_diff(revid, rb.selected_revid)
 
435
            
 
436
        rb.destroy()
 
437
            
373
438
    def _set_revision_cb(self, w, revision_id):
374
439
        self.treeview.set_revision_id(revision_id)
375
440
 
388
453
    def _tag_revision_cb(self, w):
389
454
        try:
390
455
            self.treeview.set_sensitive(False)
391
 
            self.branch.unlock()
392
456
            dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
393
457
            response = dialog.run()
394
458
            if response != gtk.RESPONSE_NONE:
395
459
                dialog.hide()
396
460
            
397
461
                if response == gtk.RESPONSE_OK:
398
 
                    try:
399
 
                        self.branch.lock_write()
400
 
                        self.branch.tags.set_tag(dialog.tagname, dialog._revid)
401
 
                    finally:
402
 
                        self.branch.unlock()
 
462
                    self.treeview.add_tag(dialog.tagname, dialog._revid)
403
463
                
404
464
                dialog.destroy()
405
465
 
406
466
        finally:
407
 
            self.branch.lock_read()
408
 
            self.treeview.emit("revision-selected")
409
467
            self.treeview.set_sensitive(True)
410
468
 
 
469
    def _branch_index_cb(self, w):
 
470
        from bzrlib.plugins.search import index as _mod_index
 
471
        _mod_index.index_url(self.branch.base)
 
472
 
 
473
    def _about_dialog_cb(self, w):
 
474
        from bzrlib.plugins.gtk.about import AboutDialog
 
475
 
 
476
        AboutDialog().run()
 
477
 
411
478
    def _col_visibility_changed(self, col, property):
412
479
        self.config.set_user_option(property + '-column-visible', col.get_active())
413
480
        self.treeview.set_property(property + '-column-visible', col.get_active())
424
491
        dialog.run()
425
492
 
426
493
    def _refresh_clicked(self, w):
427
 
        self.treeview.update()
 
494
        self.treeview.refresh()
 
495
 
 
496
    def _update_tags(self):
 
497
        menu = gtk.Menu()
 
498
 
 
499
        if self.branch.supports_tags():
 
500
            tags = self.branch.tags.get_tag_dict().items()
 
501
            tags.sort()
 
502
            tags.reverse()
 
503
            for tag, revid in tags:
 
504
                tag_image = gtk.Image()
 
505
                tag_image.set_from_file(icon_path('tag-16.png'))
 
506
                tag_item = gtk.ImageMenuItem(tag.replace('_', '__'))
 
507
                tag_item.set_image(tag_image)
 
508
                tag_item.connect('activate', self._tag_selected_cb, revid)
 
509
                menu.add(tag_item)
 
510
            self.go_menu_tags.set_submenu(menu)
 
511
 
 
512
            self.go_menu_tags.set_sensitive(len(tags) != 0)
 
513
        else:
 
514
            self.go_menu_tags.set_sensitive(False)
 
515
 
 
516
        self.go_menu_tags.show_all()
 
517
 
 
518
    def show_diff(self, revid=None, parentid=None):
 
519
        """Open a new window to show a diff between the given revisions."""
 
520
        from bzrlib.plugins.gtk.diff import DiffWindow
 
521
        window = DiffWindow(parent=self)
 
522
 
 
523
        if parentid is None:
 
524
            parentid = NULL_REVISION
 
525
 
 
526
        rev_tree    = self.branch.repository.revision_tree(revid)
 
527
        parent_tree = self.branch.repository.revision_tree(parentid)
 
528
 
 
529
        description = revid + " - " + self.branch.nick
 
530
        window.set_diff(description, rev_tree, parent_tree)
 
531
        window.show()
 
532
 
 
533