/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-06 18:39:30 UTC
  • Revision ID: daniel.schierbeck@gmail.com-20071106183930-ncfbl8rqg81omdna
Made revision menu use a try-finally block when locking the branch.

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
 
 
22
 
from bzrlib.config import BranchConfig, GlobalConfig
23
 
from bzrlib.revision import Revision, NULL_REVISION
24
 
from bzrlib.trace import mutter
 
19
from bzrlib.revision import Revision
 
20
from bzrlib.config import BranchConfig
 
21
from treeview import TreeView
25
22
 
26
23
class BranchWindow(Window):
27
24
    """Branch window.
30
27
    for a particular branch.
31
28
    """
32
29
 
33
 
    def __init__(self, branch, start_revs, maxnum, parent=None):
 
30
    def __init__(self, branch, start, maxnum, parent=None):
34
31
        """Create a new BranchWindow.
35
32
 
36
33
        :param branch: Branch object for branch to show.
37
 
        :param start_revs: Revision ids of top revisions.
 
34
        :param start: Revision id of top revision.
38
35
        :param maxnum: Maximum number of revisions to display, 
39
36
                       None for no limit.
40
37
        """
42
39
        Window.__init__(self, parent=parent)
43
40
        self.set_border_width(0)
44
41
 
45
 
        self.branch      = branch
46
 
        self.start_revs  = start_revs
47
 
        self.maxnum      = maxnum
48
 
        self.config      = GlobalConfig()
49
 
 
50
 
        if self.config.get_user_option('viz-compact-view') == 'yes':
51
 
            self.compact_view = True
52
 
        else:
53
 
            self.compact_view = False
 
42
        self.branch = branch
 
43
        self.start  = start
 
44
        self.maxnum = maxnum
54
45
 
55
46
        self.set_title(branch.nick + " - revision history")
56
47
 
65
56
        icon = self.render_icon(gtk.STOCK_INDEX, gtk.ICON_SIZE_BUTTON)
66
57
        self.set_icon(icon)
67
58
 
68
 
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
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)
71
 
 
72
59
        self.accel_group = gtk.AccelGroup()
73
60
        self.add_accel_group(self.accel_group)
74
61
 
75
 
        gtk.Action.set_tool_item_type(gtk.MenuToolButton)
76
 
 
77
 
        self.prev_rev_action = gtk.Action("prev-rev", "_Previous Revision", "Go to the previous revision", gtk.STOCK_GO_DOWN)
78
 
        self.prev_rev_action.set_accel_path("<viz>/Go/Previous Revision")
79
 
        self.prev_rev_action.set_accel_group(self.accel_group)
80
 
        self.prev_rev_action.connect("activate", self._back_clicked_cb)
81
 
        self.prev_rev_action.connect_accelerator()
82
 
 
83
 
        self.next_rev_action = gtk.Action("next-rev", "_Next Revision", "Go to the next revision", gtk.STOCK_GO_UP)
84
 
        self.next_rev_action.set_accel_path("<viz>/Go/Next Revision")
85
 
        self.next_rev_action.set_accel_group(self.accel_group)
86
 
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
87
 
        self.next_rev_action.connect_accelerator()
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
 
 
95
62
        self.construct()
96
63
 
97
 
    def set_revision(self, revid):
98
 
        self.treeview.set_revision_id(revid)
99
 
 
100
64
    def construct(self):
101
65
        """Construct the window contents."""
102
66
        vbox = gtk.VBox(spacing=0)
103
67
        self.add(vbox)
104
68
 
105
 
        self.paned = gtk.VPaned()
106
 
        self.paned.pack1(self.construct_top(), resize=True, shrink=False)
107
 
        self.paned.pack2(self.construct_bottom(), resize=False, shrink=True)
108
 
        self.paned.show()
 
69
        top = self.construct_top()
109
70
 
110
71
        vbox.pack_start(self.construct_menubar(), expand=False, fill=True)
111
72
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
73
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
112
74
        
113
 
        vbox.pack_start(self.paned, expand=True, fill=True)
114
 
        vbox.set_focus_child(self.paned)
 
75
        paned = gtk.VPaned()
 
76
        paned.pack1(top, resize=True, shrink=False)
 
77
        paned.pack2(self.construct_bottom(), resize=False, shrink=True)
 
78
        paned.show()
 
79
        vbox.pack_start(paned, expand=True, fill=True)
 
80
        vbox.set_focus_child(paned)
115
81
 
116
82
        vbox.show()
117
83
 
128
94
        file_menu_quit = gtk.ImageMenuItem(gtk.STOCK_QUIT, self.accel_group)
129
95
        file_menu_quit.connect('activate', lambda x: gtk.main_quit())
130
96
        
131
 
        if self._parent is not None:
132
 
            file_menu.add(file_menu_close)
 
97
        file_menu.add(file_menu_close)
133
98
        file_menu.add(file_menu_quit)
134
99
 
135
100
        edit_menu = gtk.Menu()
141
106
        edit_menu_branchopts = gtk.MenuItem("Branch Settings")
142
107
        edit_menu_branchopts.connect('activate', lambda x: PreferencesWindow(self.branch.get_config()).show())
143
108
 
144
 
        edit_menu_globopts = gtk.MenuItem("Global Settings")
145
 
        edit_menu_globopts.connect('activate', lambda x: PreferencesWindow().show())
 
109
        edit_menu_prefs = gtk.ImageMenuItem(gtk.STOCK_PREFERENCES)
 
110
        edit_menu_prefs.connect('activate', lambda x: PreferencesWindow().show())
146
111
 
147
112
        edit_menu.add(edit_menu_find)
148
113
        edit_menu.add(edit_menu_branchopts)
149
 
        edit_menu.add(edit_menu_globopts)
 
114
        edit_menu.add(edit_menu_prefs)
150
115
 
151
116
        view_menu = gtk.Menu()
152
117
        view_menuitem = gtk.MenuItem("_View")
153
118
        view_menuitem.set_submenu(view_menu)
154
119
 
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
 
 
161
120
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
162
121
        view_menu_toolbar.set_active(True)
163
122
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
164
123
 
165
 
        view_menu_compact = gtk.CheckMenuItem("Show Compact Graph")
166
 
        view_menu_compact.set_active(self.compact_view)
167
 
        view_menu_compact.connect('activate', self._brokenlines_toggled_cb)
168
 
 
169
124
        view_menu.add(view_menu_toolbar)
170
 
        view_menu.add(view_menu_compact)
171
125
        view_menu.add(gtk.SeparatorMenuItem())
172
126
 
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")]:
 
127
        for (label, name) in [("Revision _Number", "revno"), ("_Date", "date")]:
 
128
            col = gtk.CheckMenuItem("Show " + label + " Column")
183
129
            col.set_active(self.treeview.get_property(name + "-column-visible"))
184
130
            col.connect('toggled', self._col_visibility_changed, name)
185
131
            view_menu.add(col)
186
132
 
187
133
        go_menu = gtk.Menu()
188
 
        go_menu.set_accel_group(self.accel_group)
189
134
        go_menuitem = gtk.MenuItem("_Go")
190
135
        go_menuitem.set_submenu(go_menu)
191
136
        
192
 
        go_menu_next = self.next_rev_action.create_menu_item()
193
 
        go_menu_prev = self.prev_rev_action.create_menu_item()
194
 
 
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()
200
 
 
201
 
        go_menu.add(go_menu_next)
202
 
        go_menu.add(go_menu_prev)
 
137
        go_menu_back = gtk.ImageMenuItem(gtk.STOCK_GO_DOWN)
 
138
        go_menu_back.connect("activate", self._back_clicked_cb)
 
139
 
 
140
        go_menu_forward = gtk.ImageMenuItem(gtk.STOCK_GO_UP)
 
141
        go_menu_forward.connect("activate", self._fwd_clicked_cb)
 
142
 
 
143
        tags_menu = gtk.Menu()
 
144
        go_menu_tags = gtk.MenuItem("_Tags")
 
145
        go_menu_tags.set_submenu(tags_menu)
 
146
 
 
147
        for (tag, revid) in self.branch.tags.get_tag_dict().items():
 
148
            tag_item = gtk.MenuItem(tag)
 
149
            tag_item.connect('activate', self._tag_selected_cb, revid)
 
150
            tags_menu.add(tag_item)
 
151
 
 
152
        go_menu.add(go_menu_back)
 
153
        go_menu.add(go_menu_forward)
203
154
        go_menu.add(gtk.SeparatorMenuItem())
204
 
        go_menu.add(self.go_menu_tags)
 
155
        go_menu.add(go_menu_tags)
205
156
 
206
157
        revision_menu = gtk.Menu()
207
158
        revision_menuitem = gtk.MenuItem("_Revision")
209
160
 
210
161
        revision_menu_diff = gtk.MenuItem("View Changes")
211
162
        revision_menu_diff.connect('activate', 
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)
 
163
                lambda w: self.treeview.show_diff())
217
164
 
218
165
        revision_menu_tag = gtk.MenuItem("Tag Revision")
219
166
        revision_menu_tag.connect('activate', self._tag_revision_cb)
220
167
 
221
168
        revision_menu.add(revision_menu_tag)
222
169
        revision_menu.add(revision_menu_diff)
223
 
        revision_menu.add(revision_menu_compare)
224
170
 
225
171
        branch_menu = gtk.Menu()
226
172
        branch_menuitem = gtk.MenuItem("_Branch")
228
174
 
229
175
        branch_menu.add(gtk.MenuItem("Pu_ll Revisions"))
230
176
        branch_menu.add(gtk.MenuItem("Pu_sh Revisions"))
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
 
 
 
177
       
250
178
        menubar.add(file_menuitem)
251
179
        menubar.add(edit_menuitem)
252
180
        menubar.add(view_menuitem)
253
181
        menubar.add(go_menuitem)
254
182
        menubar.add(revision_menuitem)
255
183
        menubar.add(branch_menuitem)
256
 
        menubar.add(help_menuitem)
257
184
        menubar.show_all()
258
185
 
259
186
        return menubar
 
187
    
 
188
    def construct_loading_msg(self):
 
189
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
 
190
                                                 gtk.ICON_SIZE_BUTTON)
 
191
        image_loading.show()
 
192
        
 
193
        label_loading = gtk.Label(_("Please wait, loading ancestral graph..."))
 
194
        label_loading.set_alignment(0.0, 0.5)  
 
195
        label_loading.show()
 
196
        
 
197
        self.loading_msg_box = gtk.HBox()
 
198
        self.loading_msg_box.set_spacing(5)
 
199
        self.loading_msg_box.set_border_width(5)        
 
200
        self.loading_msg_box.pack_start(image_loading, False, False)
 
201
        self.loading_msg_box.pack_start(label_loading, True, True)
 
202
        self.loading_msg_box.show()
 
203
        
 
204
        return self.loading_msg_box
260
205
 
261
206
    def construct_top(self):
262
207
        """Construct the top-half of the window."""
263
208
        # FIXME: Make broken_line_length configurable
264
 
 
265
 
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
266
 
 
267
 
        self.treeview.connect('revision-selected',
 
209
        self.treeview = TreeView(self.branch, self.start, self.maxnum, 32)
 
210
 
 
211
        self.treeview.connect("revision-selected",
268
212
                self._treeselection_changed_cb)
269
 
        self.treeview.connect('revision-activated',
270
 
                self._tree_revision_activated)
271
 
 
272
 
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
273
 
 
274
 
        for col in ["revno", "date"]:
275
 
            option = self.config.get_user_option(col + '-column-visible')
276
 
            if option is not None:
277
 
                self.treeview.set_property(col + '-column-visible', option == 'True')
278
 
            else:
279
 
                self.treeview.set_property(col + '-column-visible', False)
 
213
 
 
214
        self.treeview.connect('revisions-loaded', 
 
215
                lambda x: self.loading_msg_box.hide())
280
216
 
281
217
        self.treeview.show()
282
218
 
283
 
        align = gtk.Alignment(0.0, 0.0, 1.0, 1.0)
284
 
        align.set_padding(5, 0, 0, 0)
285
 
        align.add(self.treeview)
286
 
        align.show()
287
 
 
288
 
        return align
 
219
        return self.treeview
289
220
 
290
221
    def construct_navigation(self):
291
222
        """Construct the navigation buttons."""
292
223
        self.toolbar = gtk.Toolbar()
293
224
        self.toolbar.set_style(gtk.TOOLBAR_BOTH_HORIZ)
294
225
 
295
 
        self.prev_button = self.prev_rev_action.create_tool_item()
296
 
        self.toolbar.insert(self.prev_button, -1)
297
 
 
298
 
        self.next_button = self.next_rev_action.create_tool_item()
299
 
        self.toolbar.insert(self.next_button, -1)
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
 
 
307
 
        self.toolbar.show_all()
 
226
        self.back_button = gtk.MenuToolButton(stock_id=gtk.STOCK_GO_DOWN)
 
227
        self.back_button.add_accelerator("clicked", self.accel_group, ord('['),
 
228
                                         0, 0)
 
229
        self.back_button.connect("clicked", self._back_clicked_cb)
 
230
        self.toolbar.insert(self.back_button, -1)
 
231
        self.back_button.show()
 
232
 
 
233
        self.fwd_button = gtk.MenuToolButton(stock_id=gtk.STOCK_GO_UP)
 
234
        self.fwd_button.add_accelerator("clicked", self.accel_group, ord(']'),
 
235
                                        0, 0)
 
236
        self.fwd_button.connect("clicked", self._fwd_clicked_cb)
 
237
        self.toolbar.insert(self.fwd_button, -1)
 
238
        self.fwd_button.show()
 
239
 
 
240
        self.toolbar.show()
308
241
 
309
242
        return self.toolbar
310
243
 
311
244
    def construct_bottom(self):
312
245
        """Construct the bottom half of the window."""
313
246
        from bzrlib.plugins.gtk.revisionview import RevisionView
314
 
        self.revisionview = RevisionView(branch=self.branch)
 
247
        self.revisionview = RevisionView(None, tags=[], show_children=True, branch=self.branch)
315
248
        (width, height) = self.get_size()
316
249
        self.revisionview.set_size_request(width, int(height / 2.5))
317
250
        self.revisionview.show()
318
251
        self.revisionview.set_show_callback(self._show_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())
 
252
        self.revisionview.set_go_callback(self._go_clicked_cb)
321
253
        return self.revisionview
322
254
 
323
255
    def _tag_selected_cb(self, menuitem, revid):
329
261
        parents  = self.treeview.get_parents()
330
262
        children = self.treeview.get_children()
331
263
 
332
 
        if revision and revision != NULL_REVISION:
333
 
            prev_menu = gtk.Menu()
 
264
        if revision is not None:
 
265
            back_menu = gtk.Menu()
334
266
            if len(parents) > 0:
335
 
                self.prev_rev_action.set_sensitive(True)
 
267
                self.back_button.set_sensitive(True)
336
268
                for parent_id in parents:
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 = ""
 
269
                    parent = self.branch.repository.get_revision(parent_id)
 
270
                    try:
 
271
                        str = ' (' + parent.properties['branch-nick'] + ')'
 
272
                    except KeyError:
 
273
                        str = ""
343
274
 
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)
347
 
                prev_menu.show_all()
 
275
                    item = gtk.MenuItem(parent.message.split("\n")[0] + str)
 
276
                    item.connect('activate', self._set_revision_cb, parent_id)
 
277
                    back_menu.add(item)
 
278
                back_menu.show_all()
348
279
            else:
349
 
                self.prev_rev_action.set_sensitive(False)
350
 
                prev_menu.hide()
351
 
 
352
 
            self.prev_button.set_menu(prev_menu)
353
 
 
354
 
            next_menu = gtk.Menu()
 
280
                self.back_button.set_sensitive(False)
 
281
                back_menu.hide()
 
282
 
 
283
            self.back_button.set_menu(back_menu)
 
284
 
 
285
            fwd_menu = gtk.Menu()
355
286
            if len(children) > 0:
356
 
                self.next_rev_action.set_sensitive(True)
 
287
                self.fwd_button.set_sensitive(True)
357
288
                for child_id in children:
358
289
                    child = self.branch.repository.get_revision(child_id)
359
290
                    try:
363
294
 
364
295
                    item = gtk.MenuItem(child.message.split("\n")[0] + str)
365
296
                    item.connect('activate', self._set_revision_cb, child_id)
366
 
                    next_menu.add(item)
367
 
                next_menu.show_all()
 
297
                    fwd_menu.add(item)
 
298
                fwd_menu.show_all()
368
299
            else:
369
 
                self.next_rev_action.set_sensitive(False)
370
 
                next_menu.hide()
371
 
 
372
 
            self.next_button.set_menu(next_menu)
373
 
 
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)    
 
300
                self.fwd_button.set_sensitive(False)
 
301
                fwd_menu.hide()
 
302
 
 
303
            self.fwd_button.set_menu(fwd_menu)
 
304
 
 
305
            tags = []
 
306
            if self.branch.supports_tags():
 
307
                tagdict = self.branch.tags.get_reverse_tag_dict()
 
308
                if tagdict.has_key(revision.revision_id):
 
309
                    tags = tagdict[revision.revision_id]
 
310
            self.revisionview.set_revision(revision, tags, children)
403
311
 
404
312
    def _back_clicked_cb(self, *args):
405
313
        """Callback for when the back button is clicked."""
409
317
        """Callback for when the forward button is clicked."""
410
318
        self.treeview.forward()
411
319
 
412
 
    def _go_clicked_cb(self, w, p):
 
320
    def _go_clicked_cb(self, revid):
413
321
        """Callback for when the go button for a parent is clicked."""
414
 
        if self.revisionview.get_revision() is not None:
415
 
            self.treeview.set_revision(self.revisionview.get_revision())
 
322
        self.treeview.set_revision_id(revid)
416
323
 
417
324
    def _show_clicked_cb(self, revid, parentid):
418
325
        """Callback for when the show button for a parent is clicked."""
419
 
        self.show_diff(revid, parentid)
 
326
        self.treeview.show_diff(revid, parentid)
420
327
        self.treeview.grab_focus()
421
328
 
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
 
            
438
329
    def _set_revision_cb(self, w, revision_id):
439
330
        self.treeview.set_revision_id(revision_id)
440
331
 
441
 
    def _brokenlines_toggled_cb(self, button):
442
 
        self.compact_view = button.get_active()
443
 
 
444
 
        if self.compact_view:
445
 
            option = 'yes'
446
 
        else:
447
 
            option = 'no'
448
 
 
449
 
        self.config.set_user_option('viz-compact-view', option)
450
 
        self.treeview.set_property('compact', self.compact_view)
451
 
        self.treeview.refresh()
452
 
 
453
332
    def _tag_revision_cb(self, w):
454
 
        try:
455
 
            self.treeview.set_sensitive(False)
456
 
            dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
457
 
            response = dialog.run()
458
 
            if response != gtk.RESPONSE_NONE:
459
 
                dialog.hide()
 
333
        dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
 
334
        response = dialog.run()
 
335
        if response != gtk.RESPONSE_NONE:
 
336
            dialog.hide()
 
337
        
 
338
            if response == gtk.RESPONSE_OK:
 
339
                try:
 
340
                    self.branch.lock_write()
 
341
                    self.branch.tags.set_tag(dialog.tagname, dialog._revid)
 
342
                finally:
 
343
                    self.branch.unlock()
460
344
            
461
 
                if response == gtk.RESPONSE_OK:
462
 
                    self.treeview.add_tag(dialog.tagname, dialog._revid)
463
 
                
464
 
                dialog.destroy()
465
 
 
466
 
        finally:
467
 
            self.treeview.set_sensitive(True)
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()
 
345
            dialog.destroy()
477
346
 
478
347
    def _col_visibility_changed(self, col, property):
479
 
        self.config.set_user_option(property + '-column-visible', col.get_active())
480
348
        self.treeview.set_property(property + '-column-visible', col.get_active())
481
349
 
482
350
    def _toolbar_visibility_changed(self, col):
484
352
            self.toolbar.show() 
485
353
        else:
486
354
            self.toolbar.hide()
487
 
 
488
 
    def _show_about_cb(self, w):
489
 
        dialog = AboutDialog()
490
 
        dialog.connect('response', lambda d,r: d.destroy())
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