/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
23
 
from about import AboutDialog
 
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
24
25
 
25
26
class BranchWindow(Window):
26
27
    """Branch window.
29
30
    for a particular branch.
30
31
    """
31
32
 
32
 
    def __init__(self, branch, start, maxnum, parent=None):
 
33
    def __init__(self, branch, start_revs, maxnum, parent=None):
33
34
        """Create a new BranchWindow.
34
35
 
35
36
        :param branch: Branch object for branch to show.
36
 
        :param start: Revision id of top revision.
 
37
        :param start_revs: Revision ids of top revisions.
37
38
        :param maxnum: Maximum number of revisions to display, 
38
39
                       None for no limit.
39
40
        """
42
43
        self.set_border_width(0)
43
44
 
44
45
        self.branch      = branch
45
 
        self.start       = start
 
46
        self.start_revs  = start_revs
46
47
        self.maxnum      = maxnum
47
48
        self.config      = GlobalConfig()
48
49
 
66
67
 
67
68
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
68
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)
69
71
 
70
72
        self.accel_group = gtk.AccelGroup()
71
73
        self.add_accel_group(self.accel_group)
84
86
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
85
87
        self.next_rev_action.connect_accelerator()
86
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
 
87
95
        self.construct()
88
96
 
89
97
    def set_revision(self, revid):
101
109
 
102
110
        vbox.pack_start(self.construct_menubar(), expand=False, fill=True)
103
111
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
104
 
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
105
112
        
106
113
        vbox.pack_start(self.paned, expand=True, fill=True)
107
114
        vbox.set_focus_child(self.paned)
145
152
        view_menuitem = gtk.MenuItem("_View")
146
153
        view_menuitem.set_submenu(view_menu)
147
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
 
148
161
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
149
162
        view_menu_toolbar.set_active(True)
150
163
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
157
170
        view_menu.add(view_menu_compact)
158
171
        view_menu.add(gtk.SeparatorMenuItem())
159
172
 
160
 
        for (label, name) in [("Revision _Number", "revno"), ("_Date", "date")]:
161
 
            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")]:
162
183
            col.set_active(self.treeview.get_property(name + "-column-visible"))
163
184
            col.connect('toggled', self._col_visibility_changed, name)
164
185
            view_menu.add(col)
171
192
        go_menu_next = self.next_rev_action.create_menu_item()
172
193
        go_menu_prev = self.prev_rev_action.create_menu_item()
173
194
 
174
 
        tags_menu = gtk.Menu()
175
 
        go_menu_tags = gtk.MenuItem("_Tags")
176
 
        go_menu_tags.set_submenu(tags_menu)
177
 
 
178
 
        if self.branch.supports_tags():
179
 
            tags = self.branch.tags.get_tag_dict().items()
180
 
            tags.sort()
181
 
            tags.reverse()
182
 
            for tag, revid in tags:
183
 
                tag_item = gtk.MenuItem(tag)
184
 
                tag_item.connect('activate', self._tag_selected_cb, revid)
185
 
                tags_menu.add(tag_item)
186
 
 
187
 
            go_menu_tags.set_sensitive(len(tags) != 0)
188
 
        else:
189
 
            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()
190
200
 
191
201
        go_menu.add(go_menu_next)
192
202
        go_menu.add(go_menu_prev)
193
203
        go_menu.add(gtk.SeparatorMenuItem())
194
 
        go_menu.add(go_menu_tags)
 
204
        go_menu.add(self.go_menu_tags)
195
205
 
196
206
        revision_menu = gtk.Menu()
197
207
        revision_menuitem = gtk.MenuItem("_Revision")
199
209
 
200
210
        revision_menu_diff = gtk.MenuItem("View Changes")
201
211
        revision_menu_diff.connect('activate', 
202
 
                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)
203
217
 
204
218
        revision_menu_tag = gtk.MenuItem("Tag Revision")
205
219
        revision_menu_tag.connect('activate', self._tag_revision_cb)
206
220
 
207
221
        revision_menu.add(revision_menu_tag)
208
222
        revision_menu.add(revision_menu_diff)
 
223
        revision_menu.add(revision_menu_compare)
209
224
 
210
225
        branch_menu = gtk.Menu()
211
226
        branch_menuitem = gtk.MenuItem("_Branch")
214
229
        branch_menu.add(gtk.MenuItem("Pu_ll Revisions"))
215
230
        branch_menu.add(gtk.MenuItem("Pu_sh Revisions"))
216
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
 
217
250
        menubar.add(file_menuitem)
218
251
        menubar.add(edit_menuitem)
219
252
        menubar.add(view_menuitem)
220
253
        menubar.add(go_menuitem)
221
254
        menubar.add(revision_menuitem)
222
255
        menubar.add(branch_menuitem)
 
256
        menubar.add(help_menuitem)
223
257
        menubar.show_all()
224
258
 
225
259
        return menubar
226
 
    
227
 
    def construct_loading_msg(self):
228
 
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
229
 
                                                 gtk.ICON_SIZE_BUTTON)
230
 
        image_loading.show()
231
 
        
232
 
        label_loading = gtk.Label(_("Please wait, loading ancestral graph..."))
233
 
        label_loading.set_alignment(0.0, 0.5)
234
 
        label_loading.show()
235
 
        
236
 
        self.loading_msg_box = gtk.HBox()
237
 
        self.loading_msg_box.set_spacing(5)
238
 
        self.loading_msg_box.set_border_width(5)        
239
 
        self.loading_msg_box.pack_start(image_loading, False, False)
240
 
        self.loading_msg_box.pack_start(label_loading, True, True)
241
 
        self.loading_msg_box.show()
242
 
        
243
 
        return self.loading_msg_box
244
260
 
245
261
    def construct_top(self):
246
262
        """Construct the top-half of the window."""
247
263
        # FIXME: Make broken_line_length configurable
248
264
 
249
 
        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)
250
266
 
251
267
        self.treeview.connect('revision-selected',
252
268
                self._treeselection_changed_cb)
 
269
        self.treeview.connect('revision-activated',
 
270
                self._tree_revision_activated)
253
271
 
254
 
        self.treeview.connect('revisions-loaded', 
255
 
                lambda x: self.loading_msg_box.hide())
 
272
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
256
273
 
257
274
        for col in ["revno", "date"]:
258
275
            option = self.config.get_user_option(col + '-column-visible')
259
276
            if option is not None:
260
277
                self.treeview.set_property(col + '-column-visible', option == 'True')
 
278
            else:
 
279
                self.treeview.set_property(col + '-column-visible', False)
261
280
 
262
281
        self.treeview.show()
263
282
 
292
311
    def construct_bottom(self):
293
312
        """Construct the bottom half of the window."""
294
313
        from bzrlib.plugins.gtk.revisionview import RevisionView
295
 
        self.revisionview = RevisionView(None, tags=[], show_children=True, branch=self.branch)
 
314
        self.revisionview = RevisionView(branch=self.branch)
296
315
        (width, height) = self.get_size()
297
316
        self.revisionview.set_size_request(width, int(height / 2.5))
298
317
        self.revisionview.show()
299
318
        self.revisionview.set_show_callback(self._show_clicked_cb)
300
 
        self.revisionview.set_go_callback(self._go_clicked_cb)
 
319
        self.revisionview.connect('notify::revision', self._go_clicked_cb)
 
320
        self.treeview.connect('tag-added', lambda w, t, r: self.revisionview.update_tags())
301
321
        return self.revisionview
302
322
 
303
323
    def _tag_selected_cb(self, menuitem, revid):
309
329
        parents  = self.treeview.get_parents()
310
330
        children = self.treeview.get_children()
311
331
 
312
 
        if revision is not None:
 
332
        if revision and revision != NULL_REVISION:
313
333
            prev_menu = gtk.Menu()
314
334
            if len(parents) > 0:
315
335
                self.prev_rev_action.set_sensitive(True)
316
336
                for parent_id in parents:
317
 
                    parent = self.branch.repository.get_revision(parent_id)
318
 
                    try:
319
 
                        str = ' (' + parent.properties['branch-nick'] + ')'
320
 
                    except KeyError:
321
 
                        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 = ""
322
343
 
323
 
                    item = gtk.MenuItem(parent.message.split("\n")[0] + str)
324
 
                    item.connect('activate', self._set_revision_cb, parent_id)
325
 
                    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)
326
347
                prev_menu.show_all()
327
348
            else:
328
349
                self.prev_rev_action.set_sensitive(False)
350
371
 
351
372
            self.next_button.set_menu(next_menu)
352
373
 
353
 
            tags = []
354
 
            if self.branch.supports_tags():
355
 
                tagdict = self.branch.tags.get_reverse_tag_dict()
356
 
                if tagdict.has_key(revision.revision_id):
357
 
                    tags = tagdict[revision.revision_id]
358
 
            self.revisionview.set_revision(revision, tags, children)
 
374
            self.revisionview.set_revision(revision)
 
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)    
359
403
 
360
404
    def _back_clicked_cb(self, *args):
361
405
        """Callback for when the back button is clicked."""
365
409
        """Callback for when the forward button is clicked."""
366
410
        self.treeview.forward()
367
411
 
368
 
    def _go_clicked_cb(self, revid):
 
412
    def _go_clicked_cb(self, w, p):
369
413
        """Callback for when the go button for a parent is clicked."""
370
 
        self.treeview.set_revision_id(revid)
 
414
        if self.revisionview.get_revision() is not None:
 
415
            self.treeview.set_revision(self.revisionview.get_revision())
371
416
 
372
417
    def _show_clicked_cb(self, revid, parentid):
373
418
        """Callback for when the show button for a parent is clicked."""
374
 
        self.treeview.show_diff(revid, parentid)
 
419
        self.show_diff(revid, parentid)
375
420
        self.treeview.grab_focus()
376
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
            
377
438
    def _set_revision_cb(self, w, revision_id):
378
439
        self.treeview.set_revision_id(revision_id)
379
440
 
392
453
    def _tag_revision_cb(self, w):
393
454
        try:
394
455
            self.treeview.set_sensitive(False)
395
 
            self.branch.unlock()
396
456
            dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
397
457
            response = dialog.run()
398
458
            if response != gtk.RESPONSE_NONE:
399
459
                dialog.hide()
400
460
            
401
461
                if response == gtk.RESPONSE_OK:
402
 
                    try:
403
 
                        self.branch.lock_write()
404
 
                        self.branch.tags.set_tag(dialog.tagname, dialog._revid)
405
 
                    finally:
406
 
                        self.branch.unlock()
 
462
                    self.treeview.add_tag(dialog.tagname, dialog._revid)
407
463
                
408
464
                dialog.destroy()
409
465
 
410
466
        finally:
411
 
            self.branch.lock_read()
412
 
            self.treeview.emit("revision-selected")
413
467
            self.treeview.set_sensitive(True)
414
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
 
415
478
    def _col_visibility_changed(self, col, property):
416
479
        self.config.set_user_option(property + '-column-visible', col.get_active())
417
480
        self.treeview.set_property(property + '-column-visible', col.get_active())
428
491
        dialog.run()
429
492
 
430
493
    def _refresh_clicked(self, w):
431
 
        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