/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: 2007-12-21 20:20:37 UTC
  • mto: This revision was merged to the branch mainline in revision 422.
  • Revision ID: jelmer@samba.org-20071221202037-6zvbwefvrte3e2zy
Fix URL.

Show diffs side-by-side

added added

removed removed

Lines of Context:
13
13
import gobject
14
14
import pango
15
15
 
16
 
from bzrlib.plugins.gtk import icon_path
17
 
from bzrlib.plugins.gtk.branchview import TreeView, treemodel
 
16
from bzrlib.plugins.gtk.window import Window
18
17
from bzrlib.plugins.gtk.tags import AddTagDialog
19
18
from bzrlib.plugins.gtk.preferences import PreferencesWindow
20
 
from bzrlib.plugins.gtk.revisionmenu import RevisionMenu
21
 
from bzrlib.plugins.gtk.window import Window
22
 
 
23
 
from bzrlib.config import BranchConfig, GlobalConfig
24
 
from bzrlib.revision import Revision, NULL_REVISION
25
 
from bzrlib.trace import mutter
 
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
26
24
 
27
25
class BranchWindow(Window):
28
26
    """Branch window.
31
29
    for a particular branch.
32
30
    """
33
31
 
34
 
    def __init__(self, branch, start_revs, maxnum, parent=None):
 
32
    def __init__(self, branch, start, maxnum, parent=None):
35
33
        """Create a new BranchWindow.
36
34
 
37
35
        :param branch: Branch object for branch to show.
38
 
        :param start_revs: Revision ids of top revisions.
 
36
        :param start: Revision id of top revision.
39
37
        :param maxnum: Maximum number of revisions to display, 
40
38
                       None for no limit.
41
39
        """
44
42
        self.set_border_width(0)
45
43
 
46
44
        self.branch      = branch
47
 
        self.start_revs  = start_revs
 
45
        self.start       = start
48
46
        self.maxnum      = maxnum
49
47
        self.config      = GlobalConfig()
50
48
 
53
51
        else:
54
52
            self.compact_view = False
55
53
 
56
 
        self.set_title(branch._get_nick(local=True) + " - revision history")
 
54
        self.set_title(branch.nick + " - revision history")
57
55
 
58
 
        # user-configured window size
59
 
        size = self._load_size('viz-window-size')
60
 
        if size:
61
 
            width, height = size
62
 
        else:
63
 
            # Use three-quarters of the screen by default
64
 
            screen = self.get_screen()
65
 
            monitor = screen.get_monitor_geometry(0)
66
 
            width = int(monitor.width * 0.75)
67
 
            height = int(monitor.height * 0.75)
 
56
        # Use three-quarters of the screen by default
 
57
        screen = self.get_screen()
 
58
        monitor = screen.get_monitor_geometry(0)
 
59
        width = int(monitor.width * 0.75)
 
60
        height = int(monitor.height * 0.75)
68
61
        self.set_default_size(width, height)
69
 
        self.set_size_request(width/3, height/3)
70
 
        self._save_size_on_destroy(self, 'viz-window-size')
71
62
 
72
63
        # FIXME AndyFitz!
73
64
        icon = self.render_icon(gtk.STOCK_INDEX, gtk.ICON_SIZE_BUTTON)
75
66
 
76
67
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
77
68
        gtk.accel_map_add_entry("<viz>/Go/Previous Revision", gtk.keysyms.Down, gtk.gdk.MOD1_MASK)
78
 
        gtk.accel_map_add_entry("<viz>/View/Refresh", gtk.keysyms.F5, 0)
79
69
 
80
70
        self.accel_group = gtk.AccelGroup()
81
71
        self.add_accel_group(self.accel_group)
94
84
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
95
85
        self.next_rev_action.connect_accelerator()
96
86
 
97
 
        self.refresh_action = gtk.Action("refresh", "_Refresh", "Refresh view", gtk.STOCK_REFRESH)
98
 
        self.refresh_action.set_accel_path("<viz>/View/Refresh")
99
 
        self.refresh_action.set_accel_group(self.accel_group)
100
 
        self.refresh_action.connect("activate", self._refresh_clicked)
101
 
        self.refresh_action.connect_accelerator()
102
 
 
103
87
        self.construct()
104
88
 
105
 
    def _save_size_on_destroy(self, widget, config_name):
106
 
        """Creates a hook that saves the size of widget to config option 
107
 
           config_name when the window is destroyed/closed."""
108
 
        def save_size(src):
109
 
            width, height = widget.allocation.width, widget.allocation.height
110
 
            value = '%sx%s' % (width, height)
111
 
            self.config.set_user_option(config_name, value)
112
 
        self.connect("destroy", save_size)
113
 
 
114
89
    def set_revision(self, revid):
115
90
        self.treeview.set_revision_id(revid)
116
91
 
120
95
        self.add(vbox)
121
96
 
122
97
        self.paned = gtk.VPaned()
123
 
        self.paned.pack1(self.construct_top(), resize=False, shrink=True)
124
 
        self.paned.pack2(self.construct_bottom(), resize=True, shrink=False)
 
98
        self.paned.pack1(self.construct_top(), resize=True, shrink=False)
 
99
        self.paned.pack2(self.construct_bottom(), resize=False, shrink=True)
125
100
        self.paned.show()
126
101
 
127
 
        nav = self.construct_navigation()
128
 
        menubar = self.construct_menubar()
129
 
        vbox.pack_start(menubar, expand=False, fill=True)
130
 
        vbox.pack_start(nav, expand=False, fill=True)
131
 
 
 
102
        vbox.pack_start(self.construct_menubar(), expand=False, fill=True)
 
103
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
104
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
 
105
        
132
106
        vbox.pack_start(self.paned, expand=True, fill=True)
133
107
        vbox.set_focus_child(self.paned)
134
108
 
135
 
        self.treeview.connect('revision-selected',
136
 
                self._treeselection_changed_cb)
137
 
        self.treeview.connect('revision-activated',
138
 
                self._tree_revision_activated)
139
 
 
140
 
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
141
109
        vbox.show()
142
 
    
 
110
 
143
111
    def construct_menubar(self):
144
112
        menubar = gtk.MenuBar()
145
113
 
161
129
        edit_menuitem = gtk.MenuItem("_Edit")
162
130
        edit_menuitem.set_submenu(edit_menu)
163
131
 
 
132
        edit_menu_find = gtk.ImageMenuItem(gtk.STOCK_FIND)
 
133
 
164
134
        edit_menu_branchopts = gtk.MenuItem("Branch Settings")
165
135
        edit_menu_branchopts.connect('activate', lambda x: PreferencesWindow(self.branch.get_config()).show())
166
136
 
167
137
        edit_menu_globopts = gtk.MenuItem("Global Settings")
168
138
        edit_menu_globopts.connect('activate', lambda x: PreferencesWindow().show())
169
139
 
 
140
        edit_menu.add(edit_menu_find)
170
141
        edit_menu.add(edit_menu_branchopts)
171
142
        edit_menu.add(edit_menu_globopts)
172
143
 
174
145
        view_menuitem = gtk.MenuItem("_View")
175
146
        view_menuitem.set_submenu(view_menu)
176
147
 
177
 
        view_menu_refresh = self.refresh_action.create_menu_item()
178
 
        view_menu_refresh.connect('activate', self._refresh_clicked)
179
 
 
180
 
        view_menu.add(view_menu_refresh)
181
 
        view_menu.add(gtk.SeparatorMenuItem())
182
 
 
183
148
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
184
149
        view_menu_toolbar.set_active(True)
185
 
        if self.config.get_user_option('viz-toolbar-visible') == 'False':
186
 
            view_menu_toolbar.set_active(False)
187
 
            self.toolbar.hide()
188
150
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
189
151
 
190
152
        view_menu_compact = gtk.CheckMenuItem("Show Compact Graph")
191
153
        view_menu_compact.set_active(self.compact_view)
192
154
        view_menu_compact.connect('activate', self._brokenlines_toggled_cb)
193
 
        
194
 
        view_menu_diffs = gtk.CheckMenuItem("Show Diffs")
195
 
        view_menu_diffs.set_active(False)
196
 
        if self.config.get_user_option('viz-show-diffs') == 'True':
197
 
            view_menu_diffs.set_active(True)
198
 
        view_menu_diffs.connect('toggled', self._diff_visibility_changed)
199
 
        
200
 
        view_menu_wide_diffs = gtk.CheckMenuItem("Wide Diffs")
201
 
        view_menu_wide_diffs.set_active(False)
202
 
        if self.config.get_user_option('viz-wide-diffs') == 'True':
203
 
            view_menu_wide_diffs.set_active(True)
204
 
        view_menu_wide_diffs.connect('toggled', self._diff_placement_changed)
205
 
        
206
 
        view_menu_wrap_diffs = gtk.CheckMenuItem("Wrap _Long Lines in Diffs")
207
 
        view_menu_wrap_diffs.set_active(False)
208
 
        if self.config.get_user_option('viz-wrap-diffs') == 'True':
209
 
            view_menu_wrap_diffs.set_active(True)
210
 
        view_menu_wrap_diffs.connect('toggled', self._diff_wrap_changed)
211
 
                
 
155
 
212
156
        view_menu.add(view_menu_toolbar)
213
157
        view_menu.add(view_menu_compact)
214
158
        view_menu.add(gtk.SeparatorMenuItem())
215
 
        view_menu.add(view_menu_diffs)
216
 
        view_menu.add(view_menu_wide_diffs)
217
 
        view_menu.add(view_menu_wrap_diffs)
218
 
        view_menu.add(gtk.SeparatorMenuItem())
219
 
 
220
 
        self.mnu_show_revno_column = gtk.CheckMenuItem("Show Revision _Number Column")
221
 
        self.mnu_show_date_column = gtk.CheckMenuItem("Show _Date Column")
222
 
 
223
 
        # Revision numbers are pointless if there are multiple branches
224
 
        if len(self.start_revs) > 1:
225
 
            self.mnu_show_revno_column.set_sensitive(False)
226
 
            self.treeview.set_property('revno-column-visible', False)
227
 
 
228
 
        for (col, name) in [(self.mnu_show_revno_column, "revno"), 
229
 
                            (self.mnu_show_date_column, "date")]:
 
159
 
 
160
        for (label, name) in [("Revision _Number", "revno"), ("_Date", "date")]:
 
161
            col = gtk.CheckMenuItem("Show " + label + " Column")
230
162
            col.set_active(self.treeview.get_property(name + "-column-visible"))
231
163
            col.connect('toggled', self._col_visibility_changed, name)
232
164
            view_menu.add(col)
239
171
        go_menu_next = self.next_rev_action.create_menu_item()
240
172
        go_menu_prev = self.prev_rev_action.create_menu_item()
241
173
 
242
 
        tag_image = gtk.Image()
243
 
        tag_image.set_from_file(icon_path("tag-16.png"))
244
 
        self.go_menu_tags = gtk.ImageMenuItem("_Tags")
245
 
        self.go_menu_tags.set_image(tag_image)
246
 
        self.treeview.connect('refreshed', lambda w: self._update_tags())
 
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)
247
190
 
248
191
        go_menu.add(go_menu_next)
249
192
        go_menu.add(go_menu_prev)
250
193
        go_menu.add(gtk.SeparatorMenuItem())
251
 
        go_menu.add(self.go_menu_tags)
 
194
        go_menu.add(go_menu_tags)
252
195
 
253
 
        self.revision_menu = RevisionMenu(self.branch.repository, [], self.branch, parent=self)
 
196
        revision_menu = gtk.Menu()
254
197
        revision_menuitem = gtk.MenuItem("_Revision")
255
 
        revision_menuitem.set_submenu(self.revision_menu)
 
198
        revision_menuitem.set_submenu(revision_menu)
 
199
 
 
200
        revision_menu_diff = gtk.MenuItem("View Changes")
 
201
        revision_menu_diff.connect('activate', 
 
202
                lambda w: self.treeview.show_diff())
 
203
 
 
204
        revision_menu_tag = gtk.MenuItem("Tag Revision")
 
205
        revision_menu_tag.connect('activate', self._tag_revision_cb)
 
206
 
 
207
        revision_menu.add(revision_menu_tag)
 
208
        revision_menu.add(revision_menu_diff)
256
209
 
257
210
        branch_menu = gtk.Menu()
258
211
        branch_menuitem = gtk.MenuItem("_Branch")
261
214
        branch_menu.add(gtk.MenuItem("Pu_ll Revisions"))
262
215
        branch_menu.add(gtk.MenuItem("Pu_sh Revisions"))
263
216
 
264
 
        try:
265
 
            from bzrlib.plugins import search
266
 
        except ImportError:
267
 
            mutter("Didn't find search plugin")
268
 
        else:
269
 
            branch_menu.add(gtk.SeparatorMenuItem())
270
 
 
271
 
            branch_index_menuitem = gtk.MenuItem("_Index")
272
 
            branch_index_menuitem.connect('activate', self._branch_index_cb)
273
 
            branch_menu.add(branch_index_menuitem)
274
 
 
275
 
            branch_search_menuitem = gtk.MenuItem("_Search")
276
 
            branch_search_menuitem.connect('activate', self._branch_search_cb)
277
 
            branch_menu.add(branch_search_menuitem)
278
 
 
279
 
        help_menu = gtk.Menu()
280
 
        help_menuitem = gtk.MenuItem("_Help")
281
 
        help_menuitem.set_submenu(help_menu)
282
 
 
283
 
        help_about_menuitem = gtk.ImageMenuItem(gtk.STOCK_ABOUT, self.accel_group)
284
 
        help_about_menuitem.connect('activate', self._about_dialog_cb)
285
 
 
286
 
        help_menu.add(help_about_menuitem)
287
 
 
288
217
        menubar.add(file_menuitem)
289
218
        menubar.add(edit_menuitem)
290
219
        menubar.add(view_menuitem)
291
220
        menubar.add(go_menuitem)
292
221
        menubar.add(revision_menuitem)
293
222
        menubar.add(branch_menuitem)
294
 
        menubar.add(help_menuitem)
295
223
        menubar.show_all()
296
224
 
297
225
        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
298
244
 
299
245
    def construct_top(self):
300
246
        """Construct the top-half of the window."""
301
247
        # FIXME: Make broken_line_length configurable
302
248
 
303
 
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
 
249
        self.treeview = TreeView(self.branch, self.start, self.maxnum, self.compact_view)
 
250
 
 
251
        self.treeview.connect('revision-selected',
 
252
                self._treeselection_changed_cb)
 
253
 
 
254
        self.treeview.connect('revisions-loaded', 
 
255
                lambda x: self.loading_msg_box.hide())
304
256
 
305
257
        for col in ["revno", "date"]:
306
258
            option = self.config.get_user_option(col + '-column-visible')
307
259
            if option is not None:
308
260
                self.treeview.set_property(col + '-column-visible', option == 'True')
309
 
            else:
310
 
                self.treeview.set_property(col + '-column-visible', False)
311
261
 
312
262
        self.treeview.show()
313
263
 
314
264
        align = gtk.Alignment(0.0, 0.0, 1.0, 1.0)
315
265
        align.set_padding(5, 0, 0, 0)
316
266
        align.add(self.treeview)
317
 
        # user-configured size
318
 
        size = self._load_size('viz-graph-size')
319
 
        if size:
320
 
            width, height = size
321
 
            align.set_size_request(width, height)
322
 
        else:
323
 
            (width, height) = self.get_size()
324
 
            align.set_size_request(width, int(height / 2.5))
325
 
        self._save_size_on_destroy(align, 'viz-graph-size')
326
267
        align.show()
327
268
 
328
269
        return align
350
291
 
351
292
    def construct_bottom(self):
352
293
        """Construct the bottom half of the window."""
353
 
        if self.config.get_user_option('viz-wide-diffs') == 'True':
354
 
            self.diff_paned = gtk.VPaned()
355
 
        else:
356
 
            self.diff_paned = gtk.HPaned()
 
294
        from bzrlib.plugins.gtk.revisionview import RevisionView
 
295
        self.revisionview = RevisionView(None, tags=[], show_children=True, branch=self.branch)
357
296
        (width, height) = self.get_size()
358
 
        self.diff_paned.set_size_request(20, 20) # shrinkable
359
 
 
360
 
        from bzrlib.plugins.gtk.revisionview import RevisionView
361
 
        self.revisionview = RevisionView(branch=self.branch)
362
 
        self.revisionview.set_size_request(width/3, int(height / 2.5))
363
 
        # user-configured size
364
 
        size = self._load_size('viz-revisionview-size')
365
 
        if size:
366
 
            width, height = size
367
 
            self.revisionview.set_size_request(width, height)
368
 
        self._save_size_on_destroy(self.revisionview, 'viz-revisionview-size')
 
297
        self.revisionview.set_size_request(width, int(height / 2.5))
369
298
        self.revisionview.show()
370
299
        self.revisionview.set_show_callback(self._show_clicked_cb)
371
 
        self.revisionview.connect('notify::revision', self._go_clicked_cb)
372
 
        self.treeview.connect('tag-added', lambda w, t, r: self.revisionview.update_tags())
373
 
        self.diff_paned.pack1(self.revisionview)
374
 
 
375
 
        from bzrlib.plugins.gtk.diff import DiffWidget
376
 
        self.diff = DiffWidget()
377
 
        self.diff_paned.pack2(self.diff)
378
 
 
379
 
        self.diff_paned.show_all()
380
 
        if self.config.get_user_option('viz-show-diffs') != 'True':
381
 
            self.diff.hide()
382
 
 
383
 
        return self.diff_paned
 
300
        self.revisionview.set_go_callback(self._go_clicked_cb)
 
301
        return self.revisionview
384
302
 
385
303
    def _tag_selected_cb(self, menuitem, revid):
386
304
        self.treeview.set_revision_id(revid)
391
309
        parents  = self.treeview.get_parents()
392
310
        children = self.treeview.get_children()
393
311
 
394
 
        self.revision_menu.set_revision_ids([revision.revision_id])
395
 
 
396
 
        if revision and revision != NULL_REVISION:
 
312
        if revision is not None:
397
313
            prev_menu = gtk.Menu()
398
314
            if len(parents) > 0:
399
315
                self.prev_rev_action.set_sensitive(True)
400
316
                for parent_id in parents:
401
 
                    if parent_id and parent_id != NULL_REVISION:
402
 
                        parent = self.branch.repository.get_revision(parent_id)
403
 
                        try:
404
 
                            str = ' (' + parent.properties['branch-nick'] + ')'
405
 
                        except KeyError:
406
 
                            str = ""
 
317
                    parent = self.branch.repository.get_revision(parent_id)
 
318
                    try:
 
319
                        str = ' (' + parent.properties['branch-nick'] + ')'
 
320
                    except KeyError:
 
321
                        str = ""
407
322
 
408
 
                        item = gtk.MenuItem(parent.message.split("\n")[0] + str)
409
 
                        item.connect('activate', self._set_revision_cb, parent_id)
410
 
                        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)
411
326
                prev_menu.show_all()
412
327
            else:
413
328
                self.prev_rev_action.set_sensitive(False)
435
350
 
436
351
            self.next_button.set_menu(next_menu)
437
352
 
438
 
            self.revisionview.set_revision(revision)
439
 
            self.revisionview.set_children(children)
440
 
            self.update_diff_panel(revision, parents)
441
 
 
442
 
    def _tree_revision_activated(self, widget, path, col):
443
 
        # TODO: more than one parent
444
 
        """Callback for when a treeview row gets activated."""
445
 
        revision = self.treeview.get_revision()
446
 
        parents  = self.treeview.get_parents()
447
 
 
448
 
        if len(parents) == 0:
449
 
            parent_id = NULL_REVISION
450
 
        else:
451
 
            parent_id = parents[0]
452
 
 
453
 
        self.show_diff(revision.revision_id, parent_id)
454
 
        self.treeview.grab_focus()
455
 
        
 
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)
 
359
 
456
360
    def _back_clicked_cb(self, *args):
457
361
        """Callback for when the back button is clicked."""
458
362
        self.treeview.back()
461
365
        """Callback for when the forward button is clicked."""
462
366
        self.treeview.forward()
463
367
 
464
 
    def _go_clicked_cb(self, w, p):
 
368
    def _go_clicked_cb(self, revid):
465
369
        """Callback for when the go button for a parent is clicked."""
466
 
        if self.revisionview.get_revision() is not None:
467
 
            self.treeview.set_revision(self.revisionview.get_revision())
 
370
        self.treeview.set_revision_id(revid)
468
371
 
469
372
    def _show_clicked_cb(self, revid, parentid):
470
373
        """Callback for when the show button for a parent is clicked."""
471
 
        self.show_diff(revid, parentid)
 
374
        self.treeview.show_diff(revid, parentid)
472
375
        self.treeview.grab_focus()
473
376
 
474
377
    def _set_revision_cb(self, w, revision_id):
486
389
        self.treeview.set_property('compact', self.compact_view)
487
390
        self.treeview.refresh()
488
391
 
489
 
    def _branch_index_cb(self, w):
490
 
        from bzrlib.plugins.search import index as _mod_index
491
 
        _mod_index.index_url(self.branch.base)
492
 
 
493
 
    def _branch_search_cb(self, w):
494
 
        from bzrlib.plugins.search import index as _mod_index
495
 
        from bzrlib.plugins.gtk.search import SearchDialog
496
 
        from bzrlib.plugins.search import errors as search_errors
497
 
 
 
392
    def _tag_revision_cb(self, w):
498
393
        try:
499
 
            index = _mod_index.open_index_url(self.branch.base)
500
 
        except search_errors.NoSearchIndex:
501
 
            dialog = gtk.MessageDialog(self, type=gtk.MESSAGE_QUESTION, 
502
 
                buttons=gtk.BUTTONS_OK_CANCEL, 
503
 
                message_format="This branch has not been indexed yet. "
504
 
                               "Index now?")
505
 
            if dialog.run() == gtk.RESPONSE_OK:
506
 
                dialog.destroy()
507
 
                index = _mod_index.index_url(self.branch.base)
508
 
            else:
509
 
                dialog.destroy()
510
 
                return
511
 
 
512
 
        dialog = SearchDialog(index)
513
 
        
514
 
        if dialog.run() == gtk.RESPONSE_OK:
515
 
            self.set_revision(dialog.get_revision())
516
 
 
517
 
        dialog.destroy()
518
 
 
519
 
    def _about_dialog_cb(self, w):
520
 
        from bzrlib.plugins.gtk.about import AboutDialog
521
 
 
522
 
        AboutDialog().run()
 
394
            self.treeview.set_sensitive(False)
 
395
            self.branch.unlock()
 
396
            dialog = AddTagDialog(self.branch.repository, self.treeview.get_revision().revision_id, self.branch)
 
397
            response = dialog.run()
 
398
            if response != gtk.RESPONSE_NONE:
 
399
                dialog.hide()
 
400
            
 
401
                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()
 
407
                
 
408
                dialog.destroy()
 
409
 
 
410
        finally:
 
411
            self.branch.lock_read()
 
412
            self.treeview.emit("revision-selected")
 
413
            self.treeview.set_sensitive(True)
523
414
 
524
415
    def _col_visibility_changed(self, col, property):
525
416
        self.config.set_user_option(property + '-column-visible', col.get_active())
527
418
 
528
419
    def _toolbar_visibility_changed(self, col):
529
420
        if col.get_active():
530
 
            self.toolbar.show()
 
421
            self.toolbar.show() 
531
422
        else:
532
423
            self.toolbar.hide()
533
 
        self.config.set_user_option('viz-toolbar-visible', col.get_active())
534
 
 
535
 
    def _make_diff_nonzero_size(self):
536
 
        """make sure the diff isn't zero-width or zero-height"""
537
 
        alloc = self.diff.get_allocation()
538
 
        if (alloc.width < 10) or (alloc.height < 10):
539
 
            width, height = self.get_size()
540
 
            self.revisionview.set_size_request(width/3, int(height / 2.5))
541
 
 
542
 
    def _diff_visibility_changed(self, col):
543
 
        """Hide or show the diff panel."""
544
 
        if col.get_active():
545
 
            self.diff.show()
546
 
            self._make_diff_nonzero_size()
547
 
        else:
548
 
            self.diff.hide()
549
 
        self.config.set_user_option('viz-show-diffs', str(col.get_active()))
550
 
        self.update_diff_panel()
551
 
 
552
 
    def _diff_placement_changed(self, col):
553
 
        """Toggle the diff panel's position."""
554
 
        self.config.set_user_option('viz-wide-diffs', str(col.get_active()))
555
 
 
556
 
        old = self.paned.get_child2()
557
 
        self.paned.remove(old)
558
 
        self.paned.pack2(self.construct_bottom(), resize=True, shrink=False)
559
 
        self._make_diff_nonzero_size()
560
 
 
561
 
        self.treeview.emit('revision-selected')
562
 
    
563
 
    def _diff_wrap_changed(self, widget):
564
 
        """Toggle word wrap in the diff widget."""
565
 
        self.config.set_user_option('viz-wrap-diffs', widget.get_active())
566
 
        self.diff._on_wraplines_toggled(widget)
567
 
    
 
424
 
568
425
    def _show_about_cb(self, w):
569
426
        dialog = AboutDialog()
570
427
        dialog.connect('response', lambda d,r: d.destroy())
571
428
        dialog.run()
572
429
 
573
430
    def _refresh_clicked(self, w):
574
 
        self.treeview.refresh()
575
 
 
576
 
    def _update_tags(self):
577
 
        menu = gtk.Menu()
578
 
 
579
 
        if self.branch.supports_tags():
580
 
            tags = self.branch.tags.get_tag_dict().items()
581
 
            tags.sort(reverse=True)
582
 
            for tag, revid in tags:
583
 
                tag_image = gtk.Image()
584
 
                tag_image.set_from_file(icon_path('tag-16.png'))
585
 
                tag_item = gtk.ImageMenuItem(tag.replace('_', '__'))
586
 
                tag_item.set_image(tag_image)
587
 
                tag_item.connect('activate', self._tag_selected_cb, revid)
588
 
                tag_item.set_sensitive(self.treeview.has_revision_id(revid))
589
 
                menu.add(tag_item)
590
 
            self.go_menu_tags.set_submenu(menu)
591
 
 
592
 
            self.go_menu_tags.set_sensitive(len(tags) != 0)
593
 
        else:
594
 
            self.go_menu_tags.set_sensitive(False)
595
 
 
596
 
        self.go_menu_tags.show_all()
597
 
 
598
 
    def _load_size(self, name):
599
 
        """Read and parse 'name' from self.config.
600
 
        The value is a string, formatted as WIDTHxHEIGHT
601
 
        Returns None, or (width, height)
602
 
        """
603
 
        size = self.config.get_user_option(name)
604
 
        if size:
605
 
            width, height = [int(num) for num in size.split('x')]
606
 
            # avoid writing config every time we start
607
 
            return width, height
608
 
        return None
609
 
 
610
 
    def show_diff(self, revid=None, parentid=NULL_REVISION):
611
 
        """Open a new window to show a diff between the given revisions."""
612
 
        from bzrlib.plugins.gtk.diff import DiffWindow
613
 
        window = DiffWindow(parent=self)
614
 
 
615
 
        rev_tree    = self.branch.repository.revision_tree(revid)
616
 
        parent_tree = self.branch.repository.revision_tree(parentid)
617
 
 
618
 
        description = revid + " - " + self.branch._get_nick(local=True)
619
 
        window.set_diff(description, rev_tree, parent_tree)
620
 
        window.show()
621
 
 
622
 
    def update_diff_panel(self, revision=None, parents=None):
623
 
        """Show the current revision in the diff panel."""
624
 
        if self.config.get_user_option('viz-show-diffs') != 'True':
625
 
            return
626
 
 
627
 
        if not revision: # default to selected row
628
 
            revision = self.treeview.get_revision()
629
 
        if revision == NULL_REVISION:
630
 
            return
631
 
 
632
 
        if not parents: # default to selected row's parents
633
 
            parents  = self.treeview.get_parents()
634
 
        if len(parents) == 0:
635
 
            parent_id = NULL_REVISION
636
 
        else:
637
 
            parent_id = parents[0]
638
 
 
639
 
        rev_tree    = self.branch.repository.revision_tree(revision.revision_id)
640
 
        parent_tree = self.branch.repository.revision_tree(parent_id)
641
 
 
642
 
        self.diff.set_diff(rev_tree, parent_tree)
643
 
        if self.config.get_user_option('viz-wrap-diffs') == 'True':
644
 
            self.diff._on_wraplines_toggled(wrap=True)
645
 
        self.diff.show_all()
 
431
        self.treeview.update()