/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: Szilveszter Farkas (Phanatic)
  • Date: 2007-11-10 17:10:08 UTC
  • mfrom: (330.5.8 bzr-gtk-0.92.1)
  • mto: This revision was merged to the branch mainline in revision 395.
  • Revision ID: szilveszter.farkas@gmail.com-20071110171008-0lbcoysr75cmuwwd
Merge changes from the 0.92.1 release branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# -*- coding: UTF-8 -*-
1
2
"""Branch window.
2
3
 
3
4
This module contains the code to manage the branch information window,
4
5
which contains both the revision graph and details panes.
5
6
"""
6
7
 
7
 
__copyright__ = "Copyright (c) 2005 Canonical Ltd."
 
8
__copyright__ = "Copyright © 2005 Canonical Ltd."
8
9
__author__    = "Scott James Remnant <scott@ubuntu.com>"
9
10
 
10
11
 
11
 
from gi.repository import Gdk
12
 
from gi.repository import Gtk
 
12
import gtk
 
13
import gobject
 
14
import pango
 
15
import treemodel
13
16
 
14
 
from bzrlib.plugins.gtk import icon_path
15
 
from bzrlib.plugins.gtk.branchview import TreeView
16
 
from bzrlib.plugins.gtk.preferences import PreferencesWindow
17
 
from bzrlib.plugins.gtk.revisionmenu import RevisionMenu
18
17
from bzrlib.plugins.gtk.window import Window
19
 
 
 
18
from bzrlib.osutils import format_date
20
19
from bzrlib.config import GlobalConfig
21
 
from bzrlib.revision import NULL_REVISION
22
 
from bzrlib.trace import mutter
23
20
 
 
21
from linegraph import linegraph, same_branch
 
22
from graphcell import CellRendererGraph
 
23
from treemodel import TreeModel
 
24
from treeview  import TreeView
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_revs, maxnum, parent=None):
 
33
    def __init__(self, branch, start, maxnum, parent=None):
33
34
        """Create a new BranchWindow.
34
35
 
35
36
        :param branch: Branch object for branch to show.
36
 
        :param start_revs: Revision ids of top revisions.
 
37
        :param start: Revision id of top revision.
37
38
        :param maxnum: Maximum number of revisions to display, 
38
39
                       None for no limit.
39
40
        """
40
41
 
41
 
        super(BranchWindow, self).__init__(parent=parent)
 
42
        Window.__init__(self, parent=parent)
42
43
        self.set_border_width(0)
43
44
 
44
45
        self.branch      = branch
45
 
        self.start_revs  = start_revs
 
46
        self.start       = start
46
47
        self.maxnum      = maxnum
47
48
        self.config      = GlobalConfig()
48
49
 
51
52
        else:
52
53
            self.compact_view = False
53
54
 
54
 
        self.set_title(branch._get_nick(local=True) + " - revision history")
 
55
        self.set_title(branch.nick + " - revision history")
55
56
 
56
 
        # user-configured window size
57
 
        size = self._load_size('viz-window-size')
58
 
        if size:
59
 
            width, height = size
60
 
        else:
61
 
            # Use three-quarters of the screen by default
62
 
            screen = self.get_screen()
63
 
            monitor = screen.get_monitor_geometry(0)
64
 
            width = int(monitor.width * 0.75)
65
 
            height = int(monitor.height * 0.75)
 
57
        # Use three-quarters of the screen by default
 
58
        screen = self.get_screen()
 
59
        monitor = screen.get_monitor_geometry(0)
 
60
        width = int(monitor.width * 0.75)
 
61
        height = int(monitor.height * 0.75)
66
62
        self.set_default_size(width, height)
67
 
        self.set_size_request(width/3, height/3)
68
 
        self._save_size_on_destroy(self, 'viz-window-size')
69
63
 
70
64
        # FIXME AndyFitz!
71
 
        icon = self.render_icon_pixbuf(Gtk.STOCK_INDEX, Gtk.IconSize.BUTTON)
 
65
        icon = self.render_icon(gtk.STOCK_INDEX, gtk.ICON_SIZE_BUTTON)
72
66
        self.set_icon(icon)
73
67
 
74
 
        Gtk.AccelMap.add_entry("<viz>/Go/Next Revision", Gdk.KEY_Up, Gdk.ModifierType.MOD1_MASK)
75
 
        Gtk.AccelMap.add_entry("<viz>/Go/Previous Revision", Gdk.KEY_Down, Gdk.ModifierType.MOD1_MASK)
76
 
        Gtk.AccelMap.add_entry("<viz>/View/Refresh", Gdk.KEY_F5, 0)
77
 
 
78
 
        self.accel_group = Gtk.AccelGroup()
 
68
        self.accel_group = gtk.AccelGroup()
79
69
        self.add_accel_group(self.accel_group)
80
70
 
81
 
        self.prev_rev_action = Gtk.Action("prev-rev", "_Previous Revision", "Go to the previous revision", Gtk.STOCK_GO_DOWN)
82
 
        self.prev_rev_action.set_accel_path("<viz>/Go/Previous Revision")
83
 
        self.prev_rev_action.set_accel_group(self.accel_group)
84
 
        self.prev_rev_action.connect("activate", self._back_clicked_cb)
85
 
        self.prev_rev_action.connect_accelerator()
86
 
 
87
 
        self.next_rev_action = Gtk.Action("next-rev", "_Next Revision", "Go to the next revision", Gtk.STOCK_GO_UP)
88
 
        self.next_rev_action.set_accel_path("<viz>/Go/Next Revision")
89
 
        self.next_rev_action.set_accel_group(self.accel_group)
90
 
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
91
 
        self.next_rev_action.connect_accelerator()
92
 
 
93
 
        self.refresh_action = Gtk.Action("refresh", "_Refresh", "Refresh view", Gtk.STOCK_REFRESH)
94
 
        self.refresh_action.set_accel_path("<viz>/View/Refresh")
95
 
        self.refresh_action.set_accel_group(self.accel_group)
96
 
        self.refresh_action.connect("activate", self._refresh_clicked)
97
 
        self.refresh_action.connect_accelerator()
98
 
 
99
 
        self.vbox = self.construct()
100
 
 
101
 
    def _save_size_on_destroy(self, widget, config_name):
102
 
        """Creates a hook that saves the size of widget to config option 
103
 
           config_name when the window is destroyed/closed."""
104
 
        def save_size(src):
105
 
            allocation = widget.get_allocation()
106
 
            width, height = allocation.width, allocation.height
107
 
            value = '%sx%s' % (width, height)
108
 
            self.config.set_user_option(config_name, value)
109
 
        self.connect("destroy", save_size)
110
 
 
111
 
    def set_revision(self, revid):
112
 
        self.treeview.set_revision_id(revid)
 
71
        self.construct()
 
72
 
 
73
    def set_revision(self, revision):
 
74
        self.treeview.set_revision(revision)
113
75
 
114
76
    def construct(self):
115
77
        """Construct the window contents."""
116
 
        vbox = Gtk.VBox(spacing=0)
 
78
        vbox = gtk.VBox(spacing=5)
117
79
        self.add(vbox)
118
80
 
119
 
        # order is important here
120
 
        paned = self.construct_paned()
121
 
        nav = self.construct_navigation()
122
 
        menubar = self.construct_menubar()
123
 
 
124
 
        vbox.pack_start(menubar, False, True, 0)
125
 
        vbox.pack_start(nav, False, True, 0)
126
 
        vbox.pack_start(paned, True, True, 0)
127
 
        vbox.set_focus_child(paned)
128
 
 
 
81
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
82
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
 
83
        
 
84
        self.paned = gtk.VPaned()
 
85
        self.paned.pack1(self.construct_top(), resize=True, shrink=False)
 
86
        self.paned.pack2(self.construct_bottom(), resize=False, shrink=True)
 
87
        self.paned.show()
 
88
        vbox.pack_start(self.paned, expand=True, fill=True)
 
89
        vbox.set_focus_child(self.paned)
129
90
 
130
91
        vbox.show()
131
 
 
132
 
        return vbox
133
 
 
134
 
    def construct_paned(self):
135
 
        """Construct the main HPaned/VPaned contents."""
136
 
        if self.config.get_user_option('viz-vertical') == 'True':
137
 
            self.paned = Gtk.Paned.new(Gtk.Orientation.HORIZONTAL)
138
 
        else:
139
 
            self.paned = Gtk.Paned.new(Gtk.Orientation.VERTICAL)
140
 
 
141
 
        self.paned.pack1(self.construct_top(), resize=False, shrink=True)
142
 
        self.paned.pack2(self.construct_bottom(), resize=True, shrink=False)
143
 
        self.paned.show()
144
 
 
145
 
        return self.paned
146
 
 
147
 
    def construct_menubar(self):
148
 
        menubar = Gtk.MenuBar()
149
 
 
150
 
        file_menu = Gtk.Menu()
151
 
        file_menuitem = Gtk.MenuItem.new_with_mnemonic("_File")
152
 
        file_menuitem.set_submenu(file_menu)
153
 
 
154
 
        file_menu_close = Gtk.ImageMenuItem.new_from_stock(
155
 
            Gtk.STOCK_CLOSE, self.accel_group)
156
 
        file_menu_close.connect('activate', lambda x: self.destroy())
157
 
 
158
 
        file_menu_quit = Gtk.ImageMenuItem.new_from_stock(
159
 
            Gtk.STOCK_QUIT, self.accel_group)
160
 
        file_menu_quit.connect('activate', lambda x: Gtk.main_quit())
161
 
 
162
 
        if self._parent is not None:
163
 
            file_menu.add(file_menu_close)
164
 
        file_menu.add(file_menu_quit)
165
 
 
166
 
        edit_menu = Gtk.Menu()
167
 
        edit_menuitem = Gtk.MenuItem.new_with_mnemonic("_Edit")
168
 
        edit_menuitem.set_submenu(edit_menu)
169
 
 
170
 
        edit_menu_branchopts = Gtk.MenuItem(label="Branch Settings")
171
 
        edit_menu_branchopts.connect('activate', lambda x: PreferencesWindow(self.branch.get_config()).show())
172
 
 
173
 
        edit_menu_globopts = Gtk.MenuItem(label="Global Settings")
174
 
        edit_menu_globopts.connect('activate', lambda x: PreferencesWindow().show())
175
 
 
176
 
        edit_menu.add(edit_menu_branchopts)
177
 
        edit_menu.add(edit_menu_globopts)
178
 
 
179
 
        view_menu = Gtk.Menu()
180
 
        view_menuitem = Gtk.MenuItem.new_with_mnemonic("_View")
181
 
        view_menuitem.set_submenu(view_menu)
182
 
 
183
 
        view_menu_refresh = self.refresh_action.create_menu_item()
184
 
        view_menu_refresh.connect('activate', self._refresh_clicked)
185
 
 
186
 
        view_menu.add(view_menu_refresh)
187
 
        view_menu.add(Gtk.SeparatorMenuItem())
188
 
 
189
 
        view_menu_toolbar = Gtk.CheckMenuItem(label="Show Toolbar")
190
 
        view_menu_toolbar.set_active(True)
191
 
        if self.config.get_user_option('viz-toolbar-visible') == 'False':
192
 
            view_menu_toolbar.set_active(False)
193
 
            self.toolbar.hide()
194
 
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
195
 
 
196
 
        view_menu_compact = Gtk.CheckMenuItem(label="Show Compact Graph")
197
 
        view_menu_compact.set_active(self.compact_view)
198
 
        view_menu_compact.connect('activate', self._brokenlines_toggled_cb)
199
 
 
200
 
        view_menu_vertical = Gtk.CheckMenuItem(label="Side-by-side Layout")
201
 
        view_menu_vertical.set_active(False)
202
 
        if self.config.get_user_option('viz-vertical') == 'True':
203
 
            view_menu_vertical.set_active(True)
204
 
        view_menu_vertical.connect('toggled', self._vertical_layout)
205
 
 
206
 
        view_menu_diffs = Gtk.CheckMenuItem(label="Show Diffs")
207
 
        view_menu_diffs.set_active(False)
208
 
        if self.config.get_user_option('viz-show-diffs') == 'True':
209
 
            view_menu_diffs.set_active(True)
210
 
        view_menu_diffs.connect('toggled', self._diff_visibility_changed)
211
 
 
212
 
        view_menu_wide_diffs = Gtk.CheckMenuItem(label="Wide Diffs")
213
 
        view_menu_wide_diffs.set_active(False)
214
 
        if self.config.get_user_option('viz-wide-diffs') == 'True':
215
 
            view_menu_wide_diffs.set_active(True)
216
 
        view_menu_wide_diffs.connect('toggled', self._diff_placement_changed)
217
 
 
218
 
        view_menu_wrap_diffs = Gtk.CheckMenuItem.new_with_mnemonic(
219
 
            "Wrap _Long Lines in Diffs")
220
 
        view_menu_wrap_diffs.set_active(False)
221
 
        if self.config.get_user_option('viz-wrap-diffs') == 'True':
222
 
            view_menu_wrap_diffs.set_active(True)
223
 
        view_menu_wrap_diffs.connect('toggled', self._diff_wrap_changed)
224
 
 
225
 
        view_menu.add(view_menu_toolbar)
226
 
        view_menu.add(view_menu_compact)
227
 
        view_menu.add(view_menu_vertical)
228
 
        view_menu.add(Gtk.SeparatorMenuItem())
229
 
        view_menu.add(view_menu_diffs)
230
 
        view_menu.add(view_menu_wide_diffs)
231
 
        view_menu.add(view_menu_wrap_diffs)
232
 
        view_menu.add(Gtk.SeparatorMenuItem())
233
 
 
234
 
        self.mnu_show_revno_column = Gtk.CheckMenuItem.new_with_mnemonic(
235
 
            "Show Revision _Number Column")
236
 
        self.mnu_show_date_column = Gtk.CheckMenuItem.new_with_mnemonic(
237
 
            "Show _Date Column")
238
 
 
239
 
        # Revision numbers are pointless if there are multiple branches
240
 
        if len(self.start_revs) > 1:
241
 
            self.mnu_show_revno_column.set_sensitive(False)
242
 
            self.treeview.set_property('revno-column-visible', False)
243
 
 
244
 
        for (col, name) in [(self.mnu_show_revno_column, "revno"), 
245
 
                            (self.mnu_show_date_column, "date")]:
246
 
            col.set_active(self.treeview.get_property(name + "-column-visible"))
247
 
            col.connect('toggled', self._col_visibility_changed, name)
248
 
            view_menu.add(col)
249
 
 
250
 
        go_menu = Gtk.Menu()
251
 
        go_menu.set_accel_group(self.accel_group)
252
 
        go_menuitem = Gtk.MenuItem.new_with_mnemonic("_Go")
253
 
        go_menuitem.set_submenu(go_menu)
254
 
 
255
 
        go_menu_next = self.next_rev_action.create_menu_item()
256
 
        go_menu_prev = self.prev_rev_action.create_menu_item()
257
 
 
258
 
        tag_image = Gtk.Image()
259
 
        tag_image.set_from_file(icon_path("tag-16.png"))
260
 
        self.go_menu_tags = Gtk.ImageMenuItem.new_with_mnemonic("_Tags")
261
 
        self.go_menu_tags.set_image(tag_image)
262
 
        self.treeview.connect('refreshed', lambda w: self._update_tags())
263
 
 
264
 
        go_menu.add(go_menu_next)
265
 
        go_menu.add(go_menu_prev)
266
 
        go_menu.add(Gtk.SeparatorMenuItem())
267
 
        go_menu.add(self.go_menu_tags)
268
 
 
269
 
        self.revision_menu = RevisionMenu(self.branch.repository, [],
270
 
            self.branch, parent=self)
271
 
        revision_menuitem = Gtk.MenuItem.new_with_mnemonic("_Revision")
272
 
        revision_menuitem.set_submenu(self.revision_menu)
273
 
 
274
 
        branch_menu = Gtk.Menu()
275
 
        branch_menuitem = Gtk.MenuItem.new_with_mnemonic("_Branch")
276
 
        branch_menuitem.set_submenu(branch_menu)
277
 
 
278
 
        branch_menu.add(Gtk.MenuItem.new_with_mnemonic("Pu_ll Revisions"))
279
 
        branch_menu.add(Gtk.MenuItem.new_with_mnemonic("Pu_sh Revisions"))
280
 
 
281
 
        try:
282
 
            from bzrlib.plugins import search
283
 
        except ImportError:
284
 
            mutter("Didn't find search plugin")
285
 
        else:
286
 
            branch_menu.add(Gtk.SeparatorMenuItem())
287
 
 
288
 
            branch_index_menuitem = Gtk.MenuItem.new_with_mnemonic("_Index")
289
 
            branch_index_menuitem.connect('activate', self._branch_index_cb)
290
 
            branch_menu.add(branch_index_menuitem)
291
 
 
292
 
            branch_search_menuitem = Gtk.MenuItem.new_with_mnemonic("_Search")
293
 
            branch_search_menuitem.connect('activate', self._branch_search_cb)
294
 
            branch_menu.add(branch_search_menuitem)
295
 
 
296
 
        help_menu = Gtk.Menu()
297
 
        help_menuitem = Gtk.MenuItem.new_with_mnemonic("_Help")
298
 
        help_menuitem.set_submenu(help_menu)
299
 
 
300
 
        help_about_menuitem = Gtk.ImageMenuItem.new_from_stock(
301
 
            Gtk.STOCK_ABOUT, self.accel_group)
302
 
        help_about_menuitem.connect('activate', self._about_dialog_cb)
303
 
 
304
 
        help_menu.add(help_about_menuitem)
305
 
 
306
 
        menubar.add(file_menuitem)
307
 
        menubar.add(edit_menuitem)
308
 
        menubar.add(view_menuitem)
309
 
        menubar.add(go_menuitem)
310
 
        menubar.add(revision_menuitem)
311
 
        menubar.add(branch_menuitem)
312
 
        menubar.add(help_menuitem)
313
 
        menubar.show_all()
314
 
 
315
 
        return menubar
 
92
    
 
93
    def construct_loading_msg(self):
 
94
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
 
95
                                                 gtk.ICON_SIZE_BUTTON)
 
96
        image_loading.show()
 
97
        
 
98
        label_loading = gtk.Label(_("Please wait, loading ancestral graph..."))
 
99
        label_loading.set_alignment(0.0, 0.5)        
 
100
        label_loading.show()
 
101
        
 
102
        self.loading_msg_box = gtk.HBox()
 
103
        self.loading_msg_box.set_spacing(5)
 
104
        self.loading_msg_box.set_border_width(5)        
 
105
        self.loading_msg_box.pack_start(image_loading, False, False)
 
106
        self.loading_msg_box.pack_start(label_loading, True, True)
 
107
        self.loading_msg_box.show()
 
108
        
 
109
        return self.loading_msg_box
316
110
 
317
111
    def construct_top(self):
318
112
        """Construct the top-half of the window."""
319
113
        # FIXME: Make broken_line_length configurable
320
 
 
321
 
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum,
322
 
            self.compact_view)
323
 
 
324
 
        for col in ["revno", "date"]:
325
 
            option = self.config.get_user_option(col + '-column-visible')
326
 
            if option is not None:
327
 
                self.treeview.set_property(col + '-column-visible',
328
 
                    option == 'True')
329
 
            else:
330
 
                self.treeview.set_property(col + '-column-visible', False)
 
114
        if self.compact_view:
 
115
            brokenlines = 32
 
116
        else:
 
117
            brokenlines = None
 
118
 
 
119
        self.treeview = TreeView(self.branch, self.start, self.maxnum, brokenlines)
 
120
 
 
121
        self.treeview.connect("revision-selected",
 
122
                self._treeselection_changed_cb)
 
123
 
 
124
        self.treeview.connect('revisions-loaded', 
 
125
                lambda x: self.loading_msg_box.hide())
331
126
 
332
127
        self.treeview.show()
333
128
 
334
 
        align = Gtk.Alignment.new(0.0, 0.0, 1.0, 1.0)
335
 
        align.set_padding(5, 0, 0, 0)
336
 
        align.add(self.treeview)
337
 
        # user-configured size
338
 
        size = self._load_size('viz-graph-size')
339
 
        if size:
340
 
            width, height = size
341
 
            align.set_size_request(width, height)
342
 
        else:
343
 
            (width, height) = self.get_size()
344
 
            align.set_size_request(width, int(height / 2.5))
345
 
        self._save_size_on_destroy(align, 'viz-graph-size')
346
 
        align.show()
347
 
 
348
 
        return align
 
129
        return self.treeview
349
130
 
350
131
    def construct_navigation(self):
351
132
        """Construct the navigation buttons."""
352
 
        self.toolbar = Gtk.Toolbar()
353
 
        self.toolbar.set_style(Gtk.ToolbarStyle.BOTH_HORIZ)
354
 
 
355
 
        self.prev_button = self.prev_rev_action.create_tool_item()
356
 
        self.toolbar.insert(self.prev_button, -1)
357
 
 
358
 
        self.next_button = self.next_rev_action.create_tool_item()
359
 
        self.toolbar.insert(self.next_button, -1)
360
 
 
361
 
        self.toolbar.insert(Gtk.SeparatorToolItem(), -1)
362
 
 
363
 
        refresh_button = Gtk.ToolButton.new_from_stock(Gtk.STOCK_REFRESH)
364
 
        refresh_button.connect('clicked', self._refresh_clicked)
365
 
        self.toolbar.insert(refresh_button, -1)
 
133
        self.toolbar = gtk.Toolbar()
 
134
        self.toolbar.set_style(gtk.TOOLBAR_BOTH_HORIZ)
 
135
 
 
136
        self.back_button = gtk.ToolButton(stock_id=gtk.STOCK_GO_BACK)
 
137
        self.back_button.set_is_important(True)
 
138
        self.back_button.add_accelerator("clicked", self.accel_group, ord('['),
 
139
                                         0, 0)
 
140
        self.back_button.connect("clicked", self._back_clicked_cb)
 
141
        self.toolbar.insert(self.back_button, -1)
 
142
 
 
143
        self.fwd_button = gtk.ToolButton(stock_id=gtk.STOCK_GO_FORWARD)
 
144
        self.fwd_button.set_is_important(True)
 
145
        self.fwd_button.add_accelerator("clicked", self.accel_group, ord(']'),
 
146
                                        0, 0)
 
147
        self.fwd_button.connect("clicked", self._fwd_clicked_cb)
 
148
        self.toolbar.insert(self.fwd_button, -1)
 
149
 
 
150
        self.toolbar.insert(gtk.SeparatorToolItem(), -1)
 
151
 
 
152
        brokenlines_button = gtk.ToggleToolButton()
 
153
        brokenlines_button.set_label("Compact View")
 
154
        brokenlines_button.set_active(self.compact_view)
 
155
        brokenlines_button.set_is_important(True)
 
156
        brokenlines_button.connect('toggled', self._brokenlines_toggled_cb)
 
157
        self.toolbar.insert(brokenlines_button, -1)
366
158
 
367
159
        self.toolbar.show_all()
368
160
 
370
162
 
371
163
    def construct_bottom(self):
372
164
        """Construct the bottom half of the window."""
373
 
        if self.config.get_user_option('viz-wide-diffs') == 'True':
374
 
            self.diff_paned = Gtk.Paned.new(Gtk.Orientation.VERTICAL)
375
 
        else:
376
 
            self.diff_paned = Gtk.Paned.new(Gtk.Orientation.HORIZONTAL)
 
165
        from bzrlib.plugins.gtk.revisionview import RevisionView
 
166
        self.revisionview = RevisionView(None, tags=[], show_children=True, branch=self.branch)
377
167
        (width, height) = self.get_size()
378
 
        self.diff_paned.set_size_request(20, 20) # shrinkable
379
 
 
380
 
        from bzrlib.plugins.gtk.revisionview import RevisionView
381
 
        self.revisionview = RevisionView(branch=self.branch)
382
 
        self.revisionview.set_size_request(width/3, int(height / 2.5))
383
 
        # user-configured size
384
 
        size = self._load_size('viz-revisionview-size')
385
 
        if size:
386
 
            width, height = size
387
 
            self.revisionview.set_size_request(width, height)
388
 
        self._save_size_on_destroy(self.revisionview, 'viz-revisionview-size')
 
168
        self.revisionview.set_size_request(width, int(height / 2.5))
389
169
        self.revisionview.show()
390
170
        self.revisionview.set_show_callback(self._show_clicked_cb)
391
 
        self.revisionview.connect('notify::revision', self._go_clicked_cb)
392
 
        self.treeview.connect('tag-added',
393
 
            lambda w, t, r: self.revisionview.update_tags())
394
 
        self.treeview.connect('revision-selected',
395
 
                self._treeselection_changed_cb)
396
 
        self.treeview.connect('revision-activated',
397
 
                self._tree_revision_activated)
398
 
        self.diff_paned.pack1(self.revisionview)
399
 
 
400
 
        from bzrlib.plugins.gtk.diff import DiffWidget
401
 
        self.diff = DiffWidget()
402
 
        self.diff_paned.pack2(self.diff)
403
 
 
404
 
        self.diff_paned.show_all()
405
 
        if self.config.get_user_option('viz-show-diffs') != 'True':
406
 
            self.diff.hide()
407
 
 
408
 
        return self.diff_paned
409
 
 
410
 
    def _tag_selected_cb(self, menuitem, revid):
411
 
        self.treeview.set_revision_id(revid)
412
 
 
 
171
        self.revisionview.set_go_callback(self._go_clicked_cb)
 
172
        return self.revisionview
 
173
    
413
174
    def _treeselection_changed_cb(self, selection, *args):
414
175
        """callback for when the treeview changes."""
415
176
        revision = self.treeview.get_revision()
416
177
        parents  = self.treeview.get_parents()
417
178
        children = self.treeview.get_children()
418
179
 
419
 
        if revision and revision.revision_id != NULL_REVISION:
420
 
            self.revision_menu.set_revision_ids([revision.revision_id])
421
 
            prev_menu = Gtk.Menu()
422
 
            if len(parents) > 0:
423
 
                self.prev_rev_action.set_sensitive(True)
424
 
                for parent_id in parents:
425
 
                    if parent_id and parent_id != NULL_REVISION:
426
 
                        parent = self.branch.repository.get_revision(parent_id)
427
 
                        try:
428
 
                            str = ' (%s)' % parent.properties['branch-nick']
429
 
                        except KeyError:
430
 
                            str = ""
431
 
 
432
 
                        item = Gtk.MenuItem(
433
 
                            label=parent.message.split("\n")[0] + str)
434
 
                        item.connect('activate', self._set_revision_cb, parent_id)
435
 
                        prev_menu.add(item)
436
 
                prev_menu.show_all()
437
 
            else:
438
 
                self.prev_rev_action.set_sensitive(False)
439
 
                prev_menu.hide()
440
 
 
441
 
            if getattr(self.prev_button, 'set_menu', None) is not None:
442
 
                self.prev_button.set_menu(prev_menu)
443
 
 
444
 
            next_menu = Gtk.Menu()
445
 
            if len(children) > 0:
446
 
                self.next_rev_action.set_sensitive(True)
447
 
                for child_id in children:
448
 
                    child = self.branch.repository.get_revision(child_id)
449
 
                    try:
450
 
                        str = ' (%s)' % child.properties['branch-nick']
451
 
                    except KeyError:
452
 
                        str = ""
453
 
 
454
 
                    item = Gtk.MenuItem(
455
 
                        label=child.message.split("\n")[0] + str)
456
 
                    item.connect('activate', self._set_revision_cb, child_id)
457
 
                    next_menu.add(item)
458
 
                next_menu.show_all()
459
 
            else:
460
 
                self.next_rev_action.set_sensitive(False)
461
 
                next_menu.hide()
462
 
 
463
 
            if getattr(self.next_button, 'set_menu', None) is not None:
464
 
                self.next_button.set_menu(next_menu)
465
 
 
466
 
            self.revisionview.set_revision(revision)
467
 
            self.revisionview.set_children(children)
468
 
            self.update_diff_panel(revision, parents)
469
 
 
470
 
    def _tree_revision_activated(self, widget, path, col):
471
 
        # TODO: more than one parent
472
 
        """Callback for when a treeview row gets activated."""
473
 
        revision = self.treeview.get_revision()
474
 
        parents  = self.treeview.get_parents()
475
 
 
476
 
        if len(parents) == 0:
477
 
            parent_id = NULL_REVISION
478
 
        else:
479
 
            parent_id = parents[0]
480
 
 
481
180
        if revision is not None:
482
 
            self.show_diff(revision.revision_id, parent_id)
483
 
        else:
484
 
            self.show_diff(NULL_REVISION)
485
 
        self.treeview.grab_focus()
 
181
            self.back_button.set_sensitive(len(parents) > 0)
 
182
            self.fwd_button.set_sensitive(len(children) > 0)
 
183
            tags = []
 
184
            if self.branch.supports_tags():
 
185
                tagdict = self.branch.tags.get_reverse_tag_dict()
 
186
                if tagdict.has_key(revision.revision_id):
 
187
                    tags = tagdict[revision.revision_id]
 
188
            self.revisionview.set_revision(revision, tags, children)
486
189
 
487
190
    def _back_clicked_cb(self, *args):
488
191
        """Callback for when the back button is clicked."""
489
192
        self.treeview.back()
490
 
 
 
193
        
491
194
    def _fwd_clicked_cb(self, *args):
492
195
        """Callback for when the forward button is clicked."""
493
196
        self.treeview.forward()
494
197
 
495
 
    def _go_clicked_cb(self, w, p):
 
198
    def _go_clicked_cb(self, revid):
496
199
        """Callback for when the go button for a parent is clicked."""
497
 
        if self.revisionview.get_revision() is not None:
498
 
            self.treeview.set_revision(self.revisionview.get_revision())
 
200
        self.treeview.set_revision(revid)
499
201
 
500
202
    def _show_clicked_cb(self, revid, parentid):
501
203
        """Callback for when the show button for a parent is clicked."""
502
 
        self.show_diff(revid, parentid)
 
204
        self.treeview.show_diff(self.branch, revid, parentid)
503
205
        self.treeview.grab_focus()
504
206
 
505
 
    def _set_revision_cb(self, w, revision_id):
506
 
        self.treeview.set_revision_id(revision_id)
507
 
 
508
207
    def _brokenlines_toggled_cb(self, button):
509
208
        self.compact_view = button.get_active()
510
209
 
514
213
            option = 'no'
515
214
 
516
215
        self.config.set_user_option('viz-compact-view', option)
517
 
        self.treeview.set_property('compact', self.compact_view)
518
 
        self.treeview.refresh()
519
 
 
520
 
    def _branch_index_cb(self, w):
521
 
        from bzrlib.plugins.search import index as _mod_index
522
 
        _mod_index.index_url(self.branch.base)
523
 
 
524
 
    def _branch_search_cb(self, w):
525
 
        from bzrlib.plugins.search import (
526
 
            index as _mod_index,
527
 
            errors as search_errors,
528
 
            )
529
 
        from bzrlib.plugins.gtk.search import SearchDialog
530
 
 
531
 
        try:
532
 
            index = _mod_index.open_index_url(self.branch.base)
533
 
        except search_errors.NoSearchIndex:
534
 
            dialog = Gtk.MessageDialog(self, type=Gtk.MessageType.QUESTION, 
535
 
                buttons=Gtk.ButtonsType.OK_CANCEL, 
536
 
                message_format="This branch has not been indexed yet. "
537
 
                               "Index now?")
538
 
            if dialog.run() == Gtk.ResponseType.OK:
539
 
                dialog.destroy()
540
 
                index = _mod_index.index_url(self.branch.base)
541
 
            else:
542
 
                dialog.destroy()
543
 
                return
544
 
 
545
 
        dialog = SearchDialog(index)
546
 
 
547
 
        if dialog.run() == Gtk.ResponseType.OK:
548
 
            revid = dialog.get_revision()
549
 
            if revid is not None:
550
 
                self.set_revision(revid)
551
 
 
552
 
        dialog.destroy()
553
 
 
554
 
    def _about_dialog_cb(self, w):
555
 
        from bzrlib.plugins.gtk.about import AboutDialog
556
 
        AboutDialog().run()
557
 
 
558
 
    def _col_visibility_changed(self, col, property):
559
 
        self.config.set_user_option(property + '-column-visible', col.get_active())
560
 
        self.treeview.set_property(property + '-column-visible', col.get_active())
561
 
 
562
 
    def _toolbar_visibility_changed(self, col):
563
 
        if col.get_active():
564
 
            self.toolbar.show()
565
 
        else:
566
 
            self.toolbar.hide()
567
 
        self.config.set_user_option('viz-toolbar-visible', col.get_active())
568
 
 
569
 
    def _vertical_layout(self, col):
570
 
        """Toggle the layout vertical/horizontal"""
571
 
        self.config.set_user_option('viz-vertical', str(col.get_active()))
572
 
 
573
 
        old = self.paned
574
 
        self.vbox.remove(old)
575
 
        self.vbox.pack_start(
576
 
            self.construct_paned(), True, True, 0)
577
 
        self._make_diff_paned_nonzero_size()
578
 
        self._make_diff_nonzero_size()
579
 
 
580
 
        self.treeview.emit('revision-selected')
581
 
 
582
 
    def _make_diff_paned_nonzero_size(self):
583
 
        """make sure the diff/revision pane isn't zero-width or zero-height"""
584
 
        alloc = self.diff_paned.get_allocation()
585
 
        if (alloc.width < 10) or (alloc.height < 10):
586
 
            width, height = self.get_size()
587
 
            self.diff_paned.set_size_request(width/3, int(height / 2.5))
588
 
 
589
 
    def _make_diff_nonzero_size(self):
590
 
        """make sure the diff isn't zero-width or zero-height"""
591
 
        alloc = self.diff.get_allocation()
592
 
        if (alloc.width < 10) or (alloc.height < 10):
593
 
            width, height = self.get_size()
594
 
            self.revisionview.set_size_request(width/3, int(height / 2.5))
595
 
 
596
 
    def _diff_visibility_changed(self, col):
597
 
        """Hide or show the diff panel."""
598
 
        if col.get_active():
599
 
            self.diff.show()
600
 
            self._make_diff_nonzero_size()
601
 
        else:
602
 
            self.diff.hide()
603
 
        self.config.set_user_option('viz-show-diffs', str(col.get_active()))
604
 
        self.update_diff_panel()
605
 
 
606
 
    def _diff_placement_changed(self, col):
607
 
        """Toggle the diff panel's position."""
608
 
        self.config.set_user_option('viz-wide-diffs', str(col.get_active()))
609
 
 
610
 
        old = self.paned.get_child2()
611
 
        self.paned.remove(old)
612
 
        self.paned.pack2(self.construct_bottom(), resize=True, shrink=False)
613
 
        self._make_diff_nonzero_size()
614
 
 
615
 
        self.treeview.emit('revision-selected')
616
 
 
617
 
    def _diff_wrap_changed(self, widget):
618
 
        """Toggle word wrap in the diff widget."""
619
 
        self.config.set_user_option('viz-wrap-diffs', widget.get_active())
620
 
        self.diff._on_wraplines_toggled(widget)
621
 
 
622
 
    def _refresh_clicked(self, w):
623
 
        self.treeview.refresh()
624
 
 
625
 
    def _update_tags(self):
626
 
        menu = Gtk.Menu()
627
 
 
628
 
        if self.branch.supports_tags():
629
 
            tags = self.branch.tags.get_tag_dict().items()
630
 
            tags.sort(reverse=True)
631
 
            for tag, revid in tags:
632
 
                tag_image = Gtk.Image()
633
 
                tag_image.set_from_file(icon_path('tag-16.png'))
634
 
                tag_item = Gtk.ImageMenuItem.new_with_mnemonic(
635
 
                    tag.replace('_', '__'))
636
 
                tag_item.set_image(tag_image)
637
 
                tag_item.connect('activate', self._tag_selected_cb, revid)
638
 
                tag_item.set_sensitive(self.treeview.has_revision_id(revid))
639
 
                menu.add(tag_item)
640
 
            self.go_menu_tags.set_submenu(menu)
641
 
 
642
 
            self.go_menu_tags.set_sensitive(len(tags) != 0)
643
 
        else:
644
 
            self.go_menu_tags.set_sensitive(False)
645
 
 
646
 
        self.go_menu_tags.show_all()
647
 
 
648
 
    def _load_size(self, name):
649
 
        """Read and parse 'name' from self.config.
650
 
        The value is a string, formatted as WIDTHxHEIGHT
651
 
        Returns None, or (width, height)
652
 
        """
653
 
        size = self.config.get_user_option(name)
654
 
        if size:
655
 
            width, height = [int(num) for num in size.split('x')]
656
 
            # avoid writing config every time we start
657
 
            return width, height
658
 
        return None
659
 
 
660
 
    def show_diff(self, revid, parentid=NULL_REVISION):
661
 
        """Open a new window to show a diff between the given revisions."""
662
 
        from bzrlib.plugins.gtk.diff import DiffWindow
663
 
        window = DiffWindow(parent=self)
664
 
 
665
 
        rev_tree    = self.branch.repository.revision_tree(revid)
666
 
        parent_tree = self.branch.repository.revision_tree(parentid)
667
 
 
668
 
        description = revid + " - " + self.branch._get_nick(local=True)
669
 
        window.set_diff(description, rev_tree, parent_tree)
670
 
        window.show()
671
 
 
672
 
    def update_diff_panel(self, revision=None, parents=None):
673
 
        """Show the current revision in the diff panel."""
674
 
        if self.config.get_user_option('viz-show-diffs') != 'True':
675
 
            return
676
 
 
677
 
        if not revision: # default to selected row
678
 
            revision = self.treeview.get_revision()
679
 
        if revision == NULL_REVISION:
680
 
            return
681
 
 
682
 
        if not parents: # default to selected row's parents
683
 
            parents  = self.treeview.get_parents()
684
 
        if len(parents) == 0:
685
 
            parent_id = NULL_REVISION
686
 
        else:
687
 
            parent_id = parents[0]
688
 
 
689
 
        rev_tree    = self.branch.repository.revision_tree(revision.revision_id)
690
 
        parent_tree = self.branch.repository.revision_tree(parent_id)
691
 
 
692
 
        self.diff.set_diff(rev_tree, parent_tree)
693
 
        if self.config.get_user_option('viz-wrap-diffs') == 'True':
694
 
            self.diff._on_wraplines_toggled(wrap=True)
695
 
        self.diff.show_all()
 
216
 
 
217
        revision = self.treeview.get_revision()
 
218
 
 
219
        self.treeview.destroy()
 
220
        self.paned.pack1(self.construct_top(), resize=True, shrink=False)
 
221
 
 
222
        gobject.idle_add(self.set_revision, revision.revision_id)