/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)
121
128
        file_menu_quit = gtk.ImageMenuItem(gtk.STOCK_QUIT, self.accel_group)
122
129
        file_menu_quit.connect('activate', lambda x: gtk.main_quit())
123
130
        
124
 
        file_menu.add(file_menu_close)
 
131
        if self._parent is not None:
 
132
            file_menu.add(file_menu_close)
125
133
        file_menu.add(file_menu_quit)
126
134
 
127
135
        edit_menu = gtk.Menu()
133
141
        edit_menu_branchopts = gtk.MenuItem("Branch Settings")
134
142
        edit_menu_branchopts.connect('activate', lambda x: PreferencesWindow(self.branch.get_config()).show())
135
143
 
136
 
        edit_menu_prefs = gtk.ImageMenuItem(gtk.STOCK_PREFERENCES)
137
 
        edit_menu_prefs.connect('activate', lambda x: PreferencesWindow().show())
 
144
        edit_menu_globopts = gtk.MenuItem("Global Settings")
 
145
        edit_menu_globopts.connect('activate', lambda x: PreferencesWindow().show())
138
146
 
139
147
        edit_menu.add(edit_menu_find)
140
148
        edit_menu.add(edit_menu_branchopts)
141
 
        edit_menu.add(edit_menu_prefs)
 
149
        edit_menu.add(edit_menu_globopts)
142
150
 
143
151
        view_menu = gtk.Menu()
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
 
            for (tag, revid) in self.branch.tags.get_tag_dict().items():
179
 
                tag_item = gtk.MenuItem(tag)
180
 
                tag_item.connect('activate', self._tag_selected_cb, revid)
181
 
                tags_menu.add(tag_item)
 
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()
182
200
 
183
201
        go_menu.add(go_menu_next)
184
202
        go_menu.add(go_menu_prev)
185
203
        go_menu.add(gtk.SeparatorMenuItem())
186
 
        go_menu.add(go_menu_tags)
 
204
        go_menu.add(self.go_menu_tags)
187
205
 
188
206
        revision_menu = gtk.Menu()
189
207
        revision_menuitem = gtk.MenuItem("_Revision")
191
209
 
192
210
        revision_menu_diff = gtk.MenuItem("View Changes")
193
211
        revision_menu_diff.connect('activate', 
194
 
                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)
195
217
 
196
218
        revision_menu_tag = gtk.MenuItem("Tag Revision")
197
219
        revision_menu_tag.connect('activate', self._tag_revision_cb)
198
220
 
199
221
        revision_menu.add(revision_menu_tag)
200
222
        revision_menu.add(revision_menu_diff)
 
223
        revision_menu.add(revision_menu_compare)
201
224
 
202
225
        branch_menu = gtk.Menu()
203
226
        branch_menuitem = gtk.MenuItem("_Branch")
206
229
        branch_menu.add(gtk.MenuItem("Pu_ll Revisions"))
207
230
        branch_menu.add(gtk.MenuItem("Pu_sh Revisions"))
208
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
 
209
241
        help_menu = gtk.Menu()
210
242
        help_menuitem = gtk.MenuItem("_Help")
211
243
        help_menuitem.set_submenu(help_menu)
212
244
 
213
 
        help_menu_about = gtk.ImageMenuItem(gtk.STOCK_ABOUT)
214
 
        help_menu_about.connect('activate', self._show_about_cb)
215
 
 
216
 
        help_menu.add(help_menu_about)
217
 
       
 
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
 
218
250
        menubar.add(file_menuitem)
219
251
        menubar.add(edit_menuitem)
220
252
        menubar.add(view_menuitem)
225
257
        menubar.show_all()
226
258
 
227
259
        return menubar
228
 
    
229
 
    def construct_loading_msg(self):
230
 
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
231
 
                                                 gtk.ICON_SIZE_BUTTON)
232
 
        image_loading.show()
233
 
        
234
 
        label_loading = gtk.Label(_("Please wait, loading ancestral graph..."))
235
 
        label_loading.set_alignment(0.0, 0.5)
236
 
        label_loading.show()
237
 
        
238
 
        self.loading_msg_box = gtk.HBox()
239
 
        self.loading_msg_box.set_spacing(5)
240
 
        self.loading_msg_box.set_border_width(5)        
241
 
        self.loading_msg_box.pack_start(image_loading, False, False)
242
 
        self.loading_msg_box.pack_start(label_loading, True, True)
243
 
        self.loading_msg_box.show()
244
 
        
245
 
        return self.loading_msg_box
246
260
 
247
261
    def construct_top(self):
248
262
        """Construct the top-half of the window."""
249
263
        # FIXME: Make broken_line_length configurable
250
 
        if self.compact_view:
251
 
            brokenlines = 32
252
 
        else:
253
 
            brokenlines = None
254
264
 
255
 
        self.treeview = TreeView(self.branch, self.start, self.maxnum, brokenlines)
 
265
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
256
266
 
257
267
        self.treeview.connect('revision-selected',
258
268
                self._treeselection_changed_cb)
 
269
        self.treeview.connect('revision-activated',
 
270
                self._tree_revision_activated)
259
271
 
260
 
        self.treeview.connect('revisions-loaded', 
261
 
                lambda x: self.loading_msg_box.hide())
 
272
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
262
273
 
263
274
        for col in ["revno", "date"]:
264
275
            option = self.config.get_user_option(col + '-column-visible')
265
276
            if option is not None:
266
277
                self.treeview.set_property(col + '-column-visible', option == 'True')
 
278
            else:
 
279
                self.treeview.set_property(col + '-column-visible', False)
267
280
 
268
281
        self.treeview.show()
269
282
 
285
298
        self.next_button = self.next_rev_action.create_tool_item()
286
299
        self.toolbar.insert(self.next_button, -1)
287
300
 
 
301
        self.toolbar.insert(gtk.SeparatorToolItem(), -1)
 
302
 
 
303
        refresh_button = gtk.ToolButton(gtk.STOCK_REFRESH)
 
304
        refresh_button.connect('clicked', self._refresh_clicked)
 
305
        self.toolbar.insert(refresh_button, -1)
 
306
 
288
307
        self.toolbar.show_all()
289
308
 
290
309
        return self.toolbar
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
 
386
447
            option = 'no'
387
448
 
388
449
        self.config.set_user_option('viz-compact-view', option)
389
 
 
390
 
        revision = self.treeview.get_revision()
391
 
 
392
 
        self.paned.get_child1().destroy()
393
 
        self.paned.pack1(self.construct_top(), resize=True, shrink=False)
394
 
 
395
 
        gobject.idle_add(self.set_revision, revision.revision_id)
 
450
        self.treeview.set_property('compact', self.compact_view)
 
451
        self.treeview.refresh()
396
452
 
397
453
    def _tag_revision_cb(self, w):
398
454
        try:
399
455
            self.treeview.set_sensitive(False)
400
 
            self.branch.unlock()
401
456
            dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
402
457
            response = dialog.run()
403
458
            if response != gtk.RESPONSE_NONE:
404
459
                dialog.hide()
405
460
            
406
461
                if response == gtk.RESPONSE_OK:
407
 
                    try:
408
 
                        self.branch.lock_write()
409
 
                        self.branch.tags.set_tag(dialog.tagname, dialog._revid)
410
 
                    finally:
411
 
                        self.branch.unlock()
 
462
                    self.treeview.add_tag(dialog.tagname, dialog._revid)
412
463
                
413
464
                dialog.destroy()
414
465
 
415
466
        finally:
416
 
            self.branch.lock_read()
417
 
            self.treeview.emit("revision-selected")
418
467
            self.treeview.set_sensitive(True)
419
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
 
420
478
    def _col_visibility_changed(self, col, property):
421
479
        self.config.set_user_option(property + '-column-visible', col.get_active())
422
480
        self.treeview.set_property(property + '-column-visible', col.get_active())
431
489
        dialog = AboutDialog()
432
490
        dialog.connect('response', lambda d,r: d.destroy())
433
491
        dialog.run()
 
492
 
 
493
    def _refresh_clicked(self, w):
 
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