/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: Daniel Schierbeck
  • Date: 2007-11-12 17:11:33 UTC
  • mfrom: (330.6.1 trunk)
  • Revision ID: daniel.schierbeck@gmail.com-20071112171133-ju9afbjziskeep2w
Merged with trunk.

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
18
17
from bzrlib.plugins.gtk.tags import AddTagDialog
19
18
from bzrlib.plugins.gtk.preferences import PreferencesWindow
20
 
from bzrlib.plugins.gtk.branchview import TreeView, treemodel
21
 
from bzrlib.revision import Revision, NULL_REVISION
 
19
from bzrlib.revision import Revision
22
20
from bzrlib.config import BranchConfig
23
21
from bzrlib.config import GlobalConfig
 
22
from treeview import TreeView
 
23
from about import AboutDialog
24
24
 
25
25
class BranchWindow(Window):
26
26
    """Branch window.
29
29
    for a particular branch.
30
30
    """
31
31
 
32
 
    def __init__(self, branch, start_revs, maxnum, parent=None):
 
32
    def __init__(self, branch, start, maxnum, parent=None):
33
33
        """Create a new BranchWindow.
34
34
 
35
35
        :param branch: Branch object for branch to show.
36
 
        :param start_revs: Revision ids of top revisions.
 
36
        :param start: Revision id of top revision.
37
37
        :param maxnum: Maximum number of revisions to display, 
38
38
                       None for no limit.
39
39
        """
42
42
        self.set_border_width(0)
43
43
 
44
44
        self.branch      = branch
45
 
        self.start_revs  = start_revs
 
45
        self.start       = start
46
46
        self.maxnum      = maxnum
47
47
        self.config      = GlobalConfig()
48
48
 
66
66
 
67
67
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
68
68
        gtk.accel_map_add_entry("<viz>/Go/Previous Revision", gtk.keysyms.Down, gtk.gdk.MOD1_MASK)
69
 
        gtk.accel_map_add_entry("<viz>/View/Refresh", gtk.keysyms.F5, 0)
70
69
 
71
70
        self.accel_group = gtk.AccelGroup()
72
71
        self.add_accel_group(self.accel_group)
85
84
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
86
85
        self.next_rev_action.connect_accelerator()
87
86
 
88
 
        self.refresh_action = gtk.Action("refresh", "_Refresh", "Refresh view", gtk.STOCK_REFRESH)
89
 
        self.refresh_action.set_accel_path("<viz>/View/Refresh")
90
 
        self.refresh_action.set_accel_group(self.accel_group)
91
 
        self.refresh_action.connect("activate", self._refresh_clicked)
92
 
        self.refresh_action.connect_accelerator()
93
 
 
94
87
        self.construct()
95
88
 
96
89
    def set_revision(self, revid):
108
101
 
109
102
        vbox.pack_start(self.construct_menubar(), expand=False, fill=True)
110
103
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
104
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
111
105
        
112
106
        vbox.pack_start(self.paned, expand=True, fill=True)
113
107
        vbox.set_focus_child(self.paned)
127
121
        file_menu_quit = gtk.ImageMenuItem(gtk.STOCK_QUIT, self.accel_group)
128
122
        file_menu_quit.connect('activate', lambda x: gtk.main_quit())
129
123
        
130
 
        if self._parent is not None:
131
 
            file_menu.add(file_menu_close)
 
124
        file_menu.add(file_menu_close)
132
125
        file_menu.add(file_menu_quit)
133
126
 
134
127
        edit_menu = gtk.Menu()
140
133
        edit_menu_branchopts = gtk.MenuItem("Branch Settings")
141
134
        edit_menu_branchopts.connect('activate', lambda x: PreferencesWindow(self.branch.get_config()).show())
142
135
 
143
 
        edit_menu_globopts = gtk.MenuItem("Global Settings")
144
 
        edit_menu_globopts.connect('activate', lambda x: PreferencesWindow().show())
 
136
        edit_menu_prefs = gtk.ImageMenuItem(gtk.STOCK_PREFERENCES)
 
137
        edit_menu_prefs.connect('activate', lambda x: PreferencesWindow().show())
145
138
 
146
139
        edit_menu.add(edit_menu_find)
147
140
        edit_menu.add(edit_menu_branchopts)
148
 
        edit_menu.add(edit_menu_globopts)
 
141
        edit_menu.add(edit_menu_prefs)
149
142
 
150
143
        view_menu = gtk.Menu()
151
144
        view_menuitem = gtk.MenuItem("_View")
152
145
        view_menuitem.set_submenu(view_menu)
153
146
 
154
 
        view_menu_refresh = self.refresh_action.create_menu_item()
155
 
        view_menu_refresh.connect('activate', self._refresh_clicked)
156
 
 
157
 
        view_menu.add(view_menu_refresh)
158
 
        view_menu.add(gtk.SeparatorMenuItem())
159
 
 
160
147
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
161
148
        view_menu_toolbar.set_active(True)
162
149
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
169
156
        view_menu.add(view_menu_compact)
170
157
        view_menu.add(gtk.SeparatorMenuItem())
171
158
 
172
 
        self.mnu_show_revno_column = gtk.CheckMenuItem("Show Revision _Number Column")
173
 
        self.mnu_show_date_column = gtk.CheckMenuItem("Show _Date Column")
174
 
 
175
 
        # Revision numbers are pointless if there are multiple branches
176
 
        if len(self.start_revs) > 1:
177
 
            self.mnu_show_revno_column.set_sensitive(False)
178
 
            self.treeview.set_property('revno-column-visible', False)
179
 
 
180
 
        for (col, name) in [(self.mnu_show_revno_column, "revno"), 
181
 
                            (self.mnu_show_date_column, "date")]:
 
159
        for (label, name) in [("Revision _Number", "revno"), ("_Date", "date")]:
 
160
            col = gtk.CheckMenuItem("Show " + label + " Column")
182
161
            col.set_active(self.treeview.get_property(name + "-column-visible"))
183
162
            col.connect('toggled', self._col_visibility_changed, name)
184
163
            view_menu.add(col)
191
170
        go_menu_next = self.next_rev_action.create_menu_item()
192
171
        go_menu_prev = self.prev_rev_action.create_menu_item()
193
172
 
194
 
        tag_image = gtk.Image()
195
 
        tag_image.set_from_file(icon_path("tag-16.png"))
196
 
        self.go_menu_tags = gtk.ImageMenuItem("_Tags")
197
 
        self.go_menu_tags.set_image(tag_image)
198
 
        self._update_tags()
 
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)
199
182
 
200
183
        go_menu.add(go_menu_next)
201
184
        go_menu.add(go_menu_prev)
202
185
        go_menu.add(gtk.SeparatorMenuItem())
203
 
        go_menu.add(self.go_menu_tags)
 
186
        go_menu.add(go_menu_tags)
204
187
 
205
188
        revision_menu = gtk.Menu()
206
189
        revision_menuitem = gtk.MenuItem("_Revision")
208
191
 
209
192
        revision_menu_diff = gtk.MenuItem("View Changes")
210
193
        revision_menu_diff.connect('activate', 
211
 
                self._menu_diff_cb)
212
 
        
213
 
        revision_menu_compare = gtk.MenuItem("Compare with...")
214
 
        revision_menu_compare.connect('activate',
215
 
                self._compare_with_cb)
 
194
                lambda w: self.treeview.show_diff())
216
195
 
217
196
        revision_menu_tag = gtk.MenuItem("Tag Revision")
218
197
        revision_menu_tag.connect('activate', self._tag_revision_cb)
219
198
 
220
199
        revision_menu.add(revision_menu_tag)
221
200
        revision_menu.add(revision_menu_diff)
222
 
        revision_menu.add(revision_menu_compare)
223
201
 
224
202
        branch_menu = gtk.Menu()
225
203
        branch_menuitem = gtk.MenuItem("_Branch")
232
210
        help_menuitem = gtk.MenuItem("_Help")
233
211
        help_menuitem.set_submenu(help_menu)
234
212
 
235
 
        help_about_menuitem = gtk.ImageMenuItem(gtk.STOCK_ABOUT, self.accel_group)
236
 
        help_about_menuitem.connect('activate', self._about_dialog_cb)
237
 
 
238
 
        help_menu.add(help_about_menuitem)
239
 
 
 
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
       
240
218
        menubar.add(file_menuitem)
241
219
        menubar.add(edit_menuitem)
242
220
        menubar.add(view_menuitem)
247
225
        menubar.show_all()
248
226
 
249
227
        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
250
246
 
251
247
    def construct_top(self):
252
248
        """Construct the top-half of the window."""
253
249
        # FIXME: Make broken_line_length configurable
 
250
        if self.compact_view:
 
251
            brokenlines = 32
 
252
        else:
 
253
            brokenlines = None
254
254
 
255
 
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
 
255
        self.treeview = TreeView(self.branch, self.start, self.maxnum, brokenlines)
256
256
 
257
257
        self.treeview.connect('revision-selected',
258
258
                self._treeselection_changed_cb)
259
 
        self.treeview.connect('revision-activated',
260
 
                self._tree_revision_activated)
261
259
 
262
 
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
 
260
        self.treeview.connect('revisions-loaded', 
 
261
                lambda x: self.loading_msg_box.hide())
263
262
 
264
263
        for col in ["revno", "date"]:
265
264
            option = self.config.get_user_option(col + '-column-visible')
266
265
            if option is not None:
267
266
                self.treeview.set_property(col + '-column-visible', option == 'True')
268
 
            else:
269
 
                self.treeview.set_property(col + '-column-visible', False)
270
267
 
271
268
        self.treeview.show()
272
269
 
288
285
        self.next_button = self.next_rev_action.create_tool_item()
289
286
        self.toolbar.insert(self.next_button, -1)
290
287
 
291
 
        self.toolbar.insert(gtk.SeparatorToolItem(), -1)
292
 
 
293
 
        refresh_button = gtk.ToolButton(gtk.STOCK_REFRESH)
294
 
        refresh_button.connect('clicked', self._refresh_clicked)
295
 
        self.toolbar.insert(refresh_button, -1)
296
 
 
297
288
        self.toolbar.show_all()
298
289
 
299
290
        return self.toolbar
301
292
    def construct_bottom(self):
302
293
        """Construct the bottom half of the window."""
303
294
        from bzrlib.plugins.gtk.revisionview import RevisionView
304
 
        self.revisionview = RevisionView(branch=self.branch)
 
295
        self.revisionview = RevisionView(None, tags=[], show_children=True, branch=self.branch)
305
296
        (width, height) = self.get_size()
306
297
        self.revisionview.set_size_request(width, int(height / 2.5))
307
298
        self.revisionview.show()
308
299
        self.revisionview.set_show_callback(self._show_clicked_cb)
309
 
        self.revisionview.connect('notify::revision', self._go_clicked_cb)
310
 
        self.treeview.connect('tag-added', lambda w, t, r: self.revisionview.update_tags())
 
300
        self.revisionview.set_go_callback(self._go_clicked_cb)
311
301
        return self.revisionview
312
302
 
313
303
    def _tag_selected_cb(self, menuitem, revid):
319
309
        parents  = self.treeview.get_parents()
320
310
        children = self.treeview.get_children()
321
311
 
322
 
        if revision and revision != NULL_REVISION:
 
312
        if revision is not None:
323
313
            prev_menu = gtk.Menu()
324
314
            if len(parents) > 0:
325
315
                self.prev_rev_action.set_sensitive(True)
326
316
                for parent_id in parents:
327
 
                    if parent_id and parent_id != NULL_REVISION:
328
 
                        parent = self.branch.repository.get_revision(parent_id)
329
 
                        try:
330
 
                            str = ' (' + parent.properties['branch-nick'] + ')'
331
 
                        except KeyError:
332
 
                            str = ""
 
317
                    parent = self.branch.repository.get_revision(parent_id)
 
318
                    try:
 
319
                        str = ' (' + parent.properties['branch-nick'] + ')'
 
320
                    except KeyError:
 
321
                        str = ""
333
322
 
334
 
                        item = gtk.MenuItem(parent.message.split("\n")[0] + str)
335
 
                        item.connect('activate', self._set_revision_cb, parent_id)
336
 
                        prev_menu.add(item)
 
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)
337
326
                prev_menu.show_all()
338
327
            else:
339
328
                self.prev_rev_action.set_sensitive(False)
361
350
 
362
351
            self.next_button.set_menu(next_menu)
363
352
 
364
 
            self.revisionview.set_revision(revision)
365
 
            self.revisionview.set_children(children)
366
 
    
367
 
    def _tree_revision_activated(self, widget, path, col):
368
 
        # TODO: more than one parent
369
 
        """Callback for when a treeview row gets activated."""
370
 
        revision = self.treeview.get_revision()
371
 
        parents  = self.treeview.get_parents()
372
 
 
373
 
        if len(parents) == 0:
374
 
            parent_id = None
375
 
        else:
376
 
            parent_id = parents[0]
377
 
 
378
 
        self.show_diff(revision.revision_id, parent_id)
379
 
        self.treeview.grab_focus()
380
 
        
381
 
    def _menu_diff_cb(self,w):
382
 
        (path, focus) = self.treeview.treeview.get_cursor()
383
 
        revid = self.treeview.model[path][treemodel.REVID]
384
 
        
385
 
        parentids = self.branch.repository.revision_parents(revid)
386
 
 
387
 
        if len(parentids) == 0:
388
 
            parentid = NULL_REVISION
389
 
        else:
390
 
            parentid = parentids[0]
391
 
        
392
 
        self.show_diff(revid,parentid)    
 
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)
393
359
 
394
360
    def _back_clicked_cb(self, *args):
395
361
        """Callback for when the back button is clicked."""
399
365
        """Callback for when the forward button is clicked."""
400
366
        self.treeview.forward()
401
367
 
402
 
    def _go_clicked_cb(self, w, p):
 
368
    def _go_clicked_cb(self, revid):
403
369
        """Callback for when the go button for a parent is clicked."""
404
 
        if self.revisionview.get_revision() is not None:
405
 
            self.treeview.set_revision(self.revisionview.get_revision())
 
370
        self.treeview.set_revision_id(revid)
406
371
 
407
372
    def _show_clicked_cb(self, revid, parentid):
408
373
        """Callback for when the show button for a parent is clicked."""
409
 
        self.show_diff(revid, parentid)
 
374
        self.treeview.show_diff(revid, parentid)
410
375
        self.treeview.grab_focus()
411
376
 
412
 
    def _compare_with_cb(self,w):
413
 
        """Callback for revision 'compare with' menu. Will show a small
414
 
            dialog with branch revisions to compare with selected revision in TreeView"""
415
 
        
416
 
        from bzrlib.plugins.gtk.revbrowser import RevisionBrowser
417
 
        
418
 
        rb = RevisionBrowser(self.branch,self)
419
 
        ret = rb.run()
420
 
        
421
 
        if ret == gtk.RESPONSE_OK:          
422
 
            (path, focus) = self.treeview.treeview.get_cursor()
423
 
            revid = self.treeview.model[path][treemodel.REVID]
424
 
            self.show_diff(revid, rb.selected_revid)
425
 
            
426
 
        rb.destroy()
427
 
            
428
377
    def _set_revision_cb(self, w, revision_id):
429
378
        self.treeview.set_revision_id(revision_id)
430
379
 
437
386
            option = 'no'
438
387
 
439
388
        self.config.set_user_option('viz-compact-view', option)
440
 
        self.treeview.set_property('compact', self.compact_view)
441
 
        self.treeview.refresh()
 
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)
442
396
 
443
397
    def _tag_revision_cb(self, w):
444
398
        try:
445
399
            self.treeview.set_sensitive(False)
 
400
            self.branch.unlock()
446
401
            dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
447
402
            response = dialog.run()
448
403
            if response != gtk.RESPONSE_NONE:
449
404
                dialog.hide()
450
405
            
451
406
                if response == gtk.RESPONSE_OK:
452
 
                    self.treeview.add_tag(dialog.tagname, dialog._revid)
 
407
                    try:
 
408
                        self.branch.lock_write()
 
409
                        self.branch.tags.set_tag(dialog.tagname, dialog._revid)
 
410
                    finally:
 
411
                        self.branch.unlock()
453
412
                
454
413
                dialog.destroy()
455
414
 
456
415
        finally:
 
416
            self.branch.lock_read()
 
417
            self.treeview.emit("revision-selected")
457
418
            self.treeview.set_sensitive(True)
458
419
 
459
 
    def _about_dialog_cb(self, w):
460
 
        from bzrlib.plugins.gtk.about import AboutDialog
461
 
 
462
 
        AboutDialog().run()
463
 
 
464
420
    def _col_visibility_changed(self, col, property):
465
421
        self.config.set_user_option(property + '-column-visible', col.get_active())
466
422
        self.treeview.set_property(property + '-column-visible', col.get_active())
475
431
        dialog = AboutDialog()
476
432
        dialog.connect('response', lambda d,r: d.destroy())
477
433
        dialog.run()
478
 
 
479
 
    def _refresh_clicked(self, w):
480
 
        self.treeview.refresh()
481
 
 
482
 
    def _update_tags(self):
483
 
        menu = gtk.Menu()
484
 
 
485
 
        if self.branch.supports_tags():
486
 
            tags = self.branch.tags.get_tag_dict().items()
487
 
            tags.sort()
488
 
            tags.reverse()
489
 
            for tag, revid in tags:
490
 
                tag_image = gtk.Image()
491
 
                tag_image.set_from_file(icon_path('tag-16.png'))
492
 
                tag_item = gtk.ImageMenuItem(tag.replace('_', '__'))
493
 
                tag_item.set_image(tag_image)
494
 
                tag_item.connect('activate', self._tag_selected_cb, revid)
495
 
                menu.add(tag_item)
496
 
            self.go_menu_tags.set_submenu(menu)
497
 
 
498
 
            self.go_menu_tags.set_sensitive(len(tags) != 0)
499
 
        else:
500
 
            self.go_menu_tags.set_sensitive(False)
501
 
 
502
 
        self.go_menu_tags.show_all()
503
 
 
504
 
    def show_diff(self, revid=None, parentid=None):
505
 
        """Open a new window to show a diff between the given revisions."""
506
 
        from bzrlib.plugins.gtk.diff import DiffWindow
507
 
        window = DiffWindow(parent=self)
508
 
 
509
 
        if parentid is None:
510
 
            parentid = NULL_REVISION
511
 
 
512
 
        rev_tree    = self.branch.repository.revision_tree(revid)
513
 
        parent_tree = self.branch.repository.revision_tree(parentid)
514
 
 
515
 
        description = revid + " - " + self.branch.nick
516
 
        window.set_diff(description, rev_tree, parent_tree)
517
 
        window.show()
518
 
 
519