/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: 2011-04-10 18:44:39 UTC
  • mto: This revision was merged to the branch mainline in revision 730.
  • Revision ID: jelmer@samba.org-20110410184439-g7hqaacexqtviq13
Move i18n support to a separate file, so gettext files aren't loaded unless bzr-gtk is used.

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/branchwin.py
1
 
#!/usr/bin/python
2
 
# -*- coding: UTF-8 -*-
3
1
"""Branch window.
4
2
 
5
3
This module contains the code to manage the branch information window,
6
4
which contains both the revision graph and details panes.
7
5
"""
8
6
 
9
 
__copyright__ = "Copyright © 2005 Canonical Ltd."
 
7
__copyright__ = "Copyright (c) 2005 Canonical Ltd."
10
8
__author__    = "Scott James Remnant <scott@ubuntu.com>"
11
9
 
12
10
 
13
11
import gtk
14
 
import gobject
15
 
import pango
16
 
 
17
 
from bzrlib.osutils import format_date
18
 
 
19
 
from graph import distances, graph, same_branch
20
 
from graphcell import CellRendererGraph
21
 
 
22
 
 
23
 
class BranchWindow(gtk.Window):
 
12
 
 
13
from bzrlib.plugins.gtk import icon_path
 
14
from bzrlib.plugins.gtk.branchview import TreeView
 
15
from bzrlib.plugins.gtk.preferences import PreferencesWindow
 
16
from bzrlib.plugins.gtk.revisionmenu import RevisionMenu
 
17
from bzrlib.plugins.gtk.window import Window
 
18
 
 
19
from bzrlib.config import GlobalConfig
 
20
from bzrlib.revision import NULL_REVISION
 
21
from bzrlib.trace import mutter
 
22
 
 
23
 
 
24
class BranchWindow(Window):
24
25
    """Branch window.
25
26
 
26
27
    This object represents and manages a single window containing information
27
28
    for a particular branch.
28
29
    """
29
30
 
30
 
    def __init__(self, app=None):
31
 
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
 
31
    def __init__(self, branch, start_revs, maxnum, parent=None):
 
32
        """Create a new BranchWindow.
 
33
 
 
34
        :param branch: Branch object for branch to show.
 
35
        :param start_revs: Revision ids of top revisions.
 
36
        :param maxnum: Maximum number of revisions to display, 
 
37
                       None for no limit.
 
38
        """
 
39
 
 
40
        Window.__init__(self, parent=parent)
32
41
        self.set_border_width(0)
33
 
        self.set_title("bzrk")
34
 
 
35
 
        self.app = app
36
 
 
37
 
        # Use three-quarters of the screen by default
38
 
        screen = self.get_screen()
39
 
        monitor = screen.get_monitor_geometry(0)
40
 
        width = int(monitor.width * 0.75)
41
 
        height = int(monitor.height * 0.75)
 
42
 
 
43
        self.branch      = branch
 
44
        self.start_revs  = start_revs
 
45
        self.maxnum      = maxnum
 
46
        self.config      = GlobalConfig()
 
47
 
 
48
        if self.config.get_user_option('viz-compact-view') == 'yes':
 
49
            self.compact_view = True
 
50
        else:
 
51
            self.compact_view = False
 
52
 
 
53
        self.set_title(branch._get_nick(local=True) + " - revision history")
 
54
 
 
55
        # user-configured window size
 
56
        size = self._load_size('viz-window-size')
 
57
        if size:
 
58
            width, height = size
 
59
        else:
 
60
            # Use three-quarters of the screen by default
 
61
            screen = self.get_screen()
 
62
            monitor = screen.get_monitor_geometry(0)
 
63
            width = int(monitor.width * 0.75)
 
64
            height = int(monitor.height * 0.75)
42
65
        self.set_default_size(width, height)
 
66
        self.set_size_request(width/3, height/3)
 
67
        self._save_size_on_destroy(self, 'viz-window-size')
43
68
 
44
69
        # FIXME AndyFitz!
45
70
        icon = self.render_icon(gtk.STOCK_INDEX, gtk.ICON_SIZE_BUTTON)
46
71
        self.set_icon(icon)
47
72
 
 
73
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
 
74
        gtk.accel_map_add_entry("<viz>/Go/Previous Revision", gtk.keysyms.Down, gtk.gdk.MOD1_MASK)
 
75
        gtk.accel_map_add_entry("<viz>/View/Refresh", gtk.keysyms.F5, 0)
 
76
 
48
77
        self.accel_group = gtk.AccelGroup()
49
78
        self.add_accel_group(self.accel_group)
50
79
 
 
80
        if getattr(gtk.Action, 'set_tool_item_type', None) is not None:
 
81
            # Not available before PyGtk-2.10
 
82
            gtk.Action.set_tool_item_type(gtk.MenuToolButton)
 
83
 
 
84
        self.prev_rev_action = gtk.Action("prev-rev", "_Previous Revision", "Go to the previous revision", gtk.STOCK_GO_DOWN)
 
85
        self.prev_rev_action.set_accel_path("<viz>/Go/Previous Revision")
 
86
        self.prev_rev_action.set_accel_group(self.accel_group)
 
87
        self.prev_rev_action.connect("activate", self._back_clicked_cb)
 
88
        self.prev_rev_action.connect_accelerator()
 
89
 
 
90
        self.next_rev_action = gtk.Action("next-rev", "_Next Revision", "Go to the next revision", gtk.STOCK_GO_UP)
 
91
        self.next_rev_action.set_accel_path("<viz>/Go/Next Revision")
 
92
        self.next_rev_action.set_accel_group(self.accel_group)
 
93
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
 
94
        self.next_rev_action.connect_accelerator()
 
95
 
 
96
        self.refresh_action = gtk.Action("refresh", "_Refresh", "Refresh view", gtk.STOCK_REFRESH)
 
97
        self.refresh_action.set_accel_path("<viz>/View/Refresh")
 
98
        self.refresh_action.set_accel_group(self.accel_group)
 
99
        self.refresh_action.connect("activate", self._refresh_clicked)
 
100
        self.refresh_action.connect_accelerator()
 
101
 
51
102
        self.construct()
52
103
 
 
104
    def _save_size_on_destroy(self, widget, config_name):
 
105
        """Creates a hook that saves the size of widget to config option 
 
106
           config_name when the window is destroyed/closed."""
 
107
        def save_size(src):
 
108
            width, height = widget.allocation.width, widget.allocation.height
 
109
            value = '%sx%s' % (width, height)
 
110
            self.config.set_user_option(config_name, value)
 
111
        self.connect("destroy", save_size)
 
112
 
 
113
    def set_revision(self, revid):
 
114
        self.treeview.set_revision_id(revid)
 
115
 
53
116
    def construct(self):
54
117
        """Construct the window contents."""
55
118
        vbox = gtk.VBox(spacing=0)
56
119
        self.add(vbox)
57
120
 
58
 
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
59
 
 
60
 
        paned = gtk.VPaned()
61
 
        paned.pack1(self.construct_top(), resize=True, shrink=False)
62
 
        paned.pack2(self.construct_bottom(), resize=False, shrink=True)
63
 
        paned.show()
64
 
        vbox.pack_start(paned, expand=True, fill=True)
65
 
        vbox.set_focus_child(paned)
66
 
 
 
121
        self.paned = gtk.VPaned()
 
122
        self.paned.pack1(self.construct_top(), resize=False, shrink=True)
 
123
        self.paned.pack2(self.construct_bottom(), resize=True, shrink=False)
 
124
        self.paned.show()
 
125
 
 
126
        nav = self.construct_navigation()
 
127
        menubar = self.construct_menubar()
 
128
        vbox.pack_start(menubar, expand=False, fill=True)
 
129
        vbox.pack_start(nav, expand=False, fill=True)
 
130
 
 
131
        vbox.pack_start(self.paned, expand=True, fill=True)
 
132
        vbox.set_focus_child(self.paned)
 
133
 
 
134
        self.treeview.connect('revision-selected',
 
135
                self._treeselection_changed_cb)
 
136
        self.treeview.connect('revision-activated',
 
137
                self._tree_revision_activated)
 
138
 
 
139
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
67
140
        vbox.show()
68
141
 
 
142
    def construct_menubar(self):
 
143
        menubar = gtk.MenuBar()
 
144
 
 
145
        file_menu = gtk.Menu()
 
146
        file_menuitem = gtk.MenuItem("_File")
 
147
        file_menuitem.set_submenu(file_menu)
 
148
 
 
149
        file_menu_close = gtk.ImageMenuItem(gtk.STOCK_CLOSE, self.accel_group)
 
150
        file_menu_close.connect('activate', lambda x: self.destroy())
 
151
 
 
152
        file_menu_quit = gtk.ImageMenuItem(gtk.STOCK_QUIT, self.accel_group)
 
153
        file_menu_quit.connect('activate', lambda x: gtk.main_quit())
 
154
 
 
155
        if self._parent is not None:
 
156
            file_menu.add(file_menu_close)
 
157
        file_menu.add(file_menu_quit)
 
158
 
 
159
        edit_menu = gtk.Menu()
 
160
        edit_menuitem = gtk.MenuItem("_Edit")
 
161
        edit_menuitem.set_submenu(edit_menu)
 
162
 
 
163
        edit_menu_branchopts = gtk.MenuItem("Branch Settings")
 
164
        edit_menu_branchopts.connect('activate', lambda x: PreferencesWindow(self.branch.get_config()).show())
 
165
 
 
166
        edit_menu_globopts = gtk.MenuItem("Global Settings")
 
167
        edit_menu_globopts.connect('activate', lambda x: PreferencesWindow().show())
 
168
 
 
169
        edit_menu.add(edit_menu_branchopts)
 
170
        edit_menu.add(edit_menu_globopts)
 
171
 
 
172
        view_menu = gtk.Menu()
 
173
        view_menuitem = gtk.MenuItem("_View")
 
174
        view_menuitem.set_submenu(view_menu)
 
175
 
 
176
        view_menu_refresh = self.refresh_action.create_menu_item()
 
177
        view_menu_refresh.connect('activate', self._refresh_clicked)
 
178
 
 
179
        view_menu.add(view_menu_refresh)
 
180
        view_menu.add(gtk.SeparatorMenuItem())
 
181
 
 
182
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
 
183
        view_menu_toolbar.set_active(True)
 
184
        if self.config.get_user_option('viz-toolbar-visible') == 'False':
 
185
            view_menu_toolbar.set_active(False)
 
186
            self.toolbar.hide()
 
187
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
 
188
 
 
189
        view_menu_compact = gtk.CheckMenuItem("Show Compact Graph")
 
190
        view_menu_compact.set_active(self.compact_view)
 
191
        view_menu_compact.connect('activate', self._brokenlines_toggled_cb)
 
192
 
 
193
        view_menu_diffs = gtk.CheckMenuItem("Show Diffs")
 
194
        view_menu_diffs.set_active(False)
 
195
        if self.config.get_user_option('viz-show-diffs') == 'True':
 
196
            view_menu_diffs.set_active(True)
 
197
        view_menu_diffs.connect('toggled', self._diff_visibility_changed)
 
198
 
 
199
        view_menu_wide_diffs = gtk.CheckMenuItem("Wide Diffs")
 
200
        view_menu_wide_diffs.set_active(False)
 
201
        if self.config.get_user_option('viz-wide-diffs') == 'True':
 
202
            view_menu_wide_diffs.set_active(True)
 
203
        view_menu_wide_diffs.connect('toggled', self._diff_placement_changed)
 
204
 
 
205
        view_menu_wrap_diffs = gtk.CheckMenuItem("Wrap _Long Lines in Diffs")
 
206
        view_menu_wrap_diffs.set_active(False)
 
207
        if self.config.get_user_option('viz-wrap-diffs') == 'True':
 
208
            view_menu_wrap_diffs.set_active(True)
 
209
        view_menu_wrap_diffs.connect('toggled', self._diff_wrap_changed)
 
210
 
 
211
        view_menu.add(view_menu_toolbar)
 
212
        view_menu.add(view_menu_compact)
 
213
        view_menu.add(gtk.SeparatorMenuItem())
 
214
        view_menu.add(view_menu_diffs)
 
215
        view_menu.add(view_menu_wide_diffs)
 
216
        view_menu.add(view_menu_wrap_diffs)
 
217
        view_menu.add(gtk.SeparatorMenuItem())
 
218
 
 
219
        self.mnu_show_revno_column = gtk.CheckMenuItem("Show Revision _Number Column")
 
220
        self.mnu_show_date_column = gtk.CheckMenuItem("Show _Date Column")
 
221
 
 
222
        # Revision numbers are pointless if there are multiple branches
 
223
        if len(self.start_revs) > 1:
 
224
            self.mnu_show_revno_column.set_sensitive(False)
 
225
            self.treeview.set_property('revno-column-visible', False)
 
226
 
 
227
        for (col, name) in [(self.mnu_show_revno_column, "revno"), 
 
228
                            (self.mnu_show_date_column, "date")]:
 
229
            col.set_active(self.treeview.get_property(name + "-column-visible"))
 
230
            col.connect('toggled', self._col_visibility_changed, name)
 
231
            view_menu.add(col)
 
232
 
 
233
        go_menu = gtk.Menu()
 
234
        go_menu.set_accel_group(self.accel_group)
 
235
        go_menuitem = gtk.MenuItem("_Go")
 
236
        go_menuitem.set_submenu(go_menu)
 
237
 
 
238
        go_menu_next = self.next_rev_action.create_menu_item()
 
239
        go_menu_prev = self.prev_rev_action.create_menu_item()
 
240
 
 
241
        tag_image = gtk.Image()
 
242
        tag_image.set_from_file(icon_path("tag-16.png"))
 
243
        self.go_menu_tags = gtk.ImageMenuItem("_Tags")
 
244
        self.go_menu_tags.set_image(tag_image)
 
245
        self.treeview.connect('refreshed', lambda w: self._update_tags())
 
246
 
 
247
        go_menu.add(go_menu_next)
 
248
        go_menu.add(go_menu_prev)
 
249
        go_menu.add(gtk.SeparatorMenuItem())
 
250
        go_menu.add(self.go_menu_tags)
 
251
 
 
252
        self.revision_menu = RevisionMenu(self.branch.repository, [], self.branch, parent=self)
 
253
        revision_menuitem = gtk.MenuItem("_Revision")
 
254
        revision_menuitem.set_submenu(self.revision_menu)
 
255
 
 
256
        branch_menu = gtk.Menu()
 
257
        branch_menuitem = gtk.MenuItem("_Branch")
 
258
        branch_menuitem.set_submenu(branch_menu)
 
259
 
 
260
        branch_menu.add(gtk.MenuItem("Pu_ll Revisions"))
 
261
        branch_menu.add(gtk.MenuItem("Pu_sh Revisions"))
 
262
 
 
263
        try:
 
264
            from bzrlib.plugins import search
 
265
        except ImportError:
 
266
            mutter("Didn't find search plugin")
 
267
        else:
 
268
            branch_menu.add(gtk.SeparatorMenuItem())
 
269
 
 
270
            branch_index_menuitem = gtk.MenuItem("_Index")
 
271
            branch_index_menuitem.connect('activate', self._branch_index_cb)
 
272
            branch_menu.add(branch_index_menuitem)
 
273
 
 
274
            branch_search_menuitem = gtk.MenuItem("_Search")
 
275
            branch_search_menuitem.connect('activate', self._branch_search_cb)
 
276
            branch_menu.add(branch_search_menuitem)
 
277
 
 
278
        help_menu = gtk.Menu()
 
279
        help_menuitem = gtk.MenuItem("_Help")
 
280
        help_menuitem.set_submenu(help_menu)
 
281
 
 
282
        help_about_menuitem = gtk.ImageMenuItem(gtk.STOCK_ABOUT, self.accel_group)
 
283
        help_about_menuitem.connect('activate', self._about_dialog_cb)
 
284
 
 
285
        help_menu.add(help_about_menuitem)
 
286
 
 
287
        menubar.add(file_menuitem)
 
288
        menubar.add(edit_menuitem)
 
289
        menubar.add(view_menuitem)
 
290
        menubar.add(go_menuitem)
 
291
        menubar.add(revision_menuitem)
 
292
        menubar.add(branch_menuitem)
 
293
        menubar.add(help_menuitem)
 
294
        menubar.show_all()
 
295
 
 
296
        return menubar
 
297
 
69
298
    def construct_top(self):
70
299
        """Construct the top-half of the window."""
71
 
        scrollwin = gtk.ScrolledWindow()
72
 
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
73
 
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
74
 
        scrollwin.show()
75
 
 
76
 
        self.treeview = gtk.TreeView()
77
 
        self.treeview.set_rules_hint(True)
78
 
        self.treeview.set_search_column(4)
79
 
        self.treeview.connect("cursor-changed", self._treeview_cursor_cb)
80
 
        self.treeview.connect("row-activated", self._treeview_row_activated_cb)
81
 
        scrollwin.add(self.treeview)
 
300
        # FIXME: Make broken_line_length configurable
 
301
 
 
302
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
 
303
 
 
304
        for col in ["revno", "date"]:
 
305
            option = self.config.get_user_option(col + '-column-visible')
 
306
            if option is not None:
 
307
                self.treeview.set_property(col + '-column-visible', option == 'True')
 
308
            else:
 
309
                self.treeview.set_property(col + '-column-visible', False)
 
310
 
82
311
        self.treeview.show()
83
312
 
84
 
        cell = CellRendererGraph()
85
 
        column = gtk.TreeViewColumn()
86
 
        column.set_resizable(True)
87
 
        column.pack_start(cell, expand=False)
88
 
        column.add_attribute(cell, "node", 1)
89
 
        column.add_attribute(cell, "in-lines", 2)
90
 
        column.add_attribute(cell, "out-lines", 3)
91
 
        self.treeview.append_column(column)
92
 
 
93
 
        cell = gtk.CellRendererText()
94
 
        cell.set_property("width-chars", 40)
95
 
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
96
 
        column = gtk.TreeViewColumn("Message")
97
 
        column.set_resizable(True)
98
 
        column.pack_start(cell, expand=True)
99
 
        column.add_attribute(cell, "text", 4)
100
 
        self.treeview.append_column(column)
101
 
 
102
 
        cell = gtk.CellRendererText()
103
 
        cell.set_property("width-chars", 40)
104
 
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
105
 
        column = gtk.TreeViewColumn("Committer")
106
 
        column.set_resizable(True)
107
 
        column.pack_start(cell, expand=True)
108
 
        column.add_attribute(cell, "text", 5)
109
 
        self.treeview.append_column(column)
110
 
 
111
 
        cell = gtk.CellRendererText()
112
 
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
113
 
        column = gtk.TreeViewColumn("Date")
114
 
        column.set_resizable(True)
115
 
        column.pack_start(cell, expand=True)
116
 
        column.add_attribute(cell, "text", 6)
117
 
        self.treeview.append_column(column)
118
 
 
119
 
        return scrollwin
 
313
        align = gtk.Alignment(0.0, 0.0, 1.0, 1.0)
 
314
        align.set_padding(5, 0, 0, 0)
 
315
        align.add(self.treeview)
 
316
        # user-configured size
 
317
        size = self._load_size('viz-graph-size')
 
318
        if size:
 
319
            width, height = size
 
320
            align.set_size_request(width, height)
 
321
        else:
 
322
            (width, height) = self.get_size()
 
323
            align.set_size_request(width, int(height / 2.5))
 
324
        self._save_size_on_destroy(align, 'viz-graph-size')
 
325
        align.show()
 
326
 
 
327
        return align
120
328
 
121
329
    def construct_navigation(self):
122
330
        """Construct the navigation buttons."""
123
 
        frame = gtk.Frame()
124
 
        frame.set_shadow_type(gtk.SHADOW_OUT)
125
 
        frame.show()
126
 
        
127
 
        hbox = gtk.HBox(spacing=12)
128
 
        frame.add(hbox)
129
 
        hbox.show()
130
 
 
131
 
        self.back_button = gtk.Button(stock=gtk.STOCK_GO_BACK)
132
 
        self.back_button.set_relief(gtk.RELIEF_NONE)
133
 
        self.back_button.add_accelerator("clicked", self.accel_group, ord('['),
134
 
                                         0, 0)
135
 
        self.back_button.connect("clicked", self._back_clicked_cb)
136
 
        hbox.pack_start(self.back_button, expand=False, fill=True)
137
 
        self.back_button.show()
138
 
 
139
 
        self.fwd_button = gtk.Button(stock=gtk.STOCK_GO_FORWARD)
140
 
        self.fwd_button.set_relief(gtk.RELIEF_NONE)
141
 
        self.fwd_button.add_accelerator("clicked", self.accel_group, ord(']'),
142
 
                                        0, 0)
143
 
        self.fwd_button.connect("clicked", self._fwd_clicked_cb)
144
 
        hbox.pack_start(self.fwd_button, expand=False, fill=True)
145
 
        self.fwd_button.show()
146
 
 
147
 
        return frame
 
331
        self.toolbar = gtk.Toolbar()
 
332
        self.toolbar.set_style(gtk.TOOLBAR_BOTH_HORIZ)
 
333
 
 
334
        self.prev_button = self.prev_rev_action.create_tool_item()
 
335
        self.toolbar.insert(self.prev_button, -1)
 
336
 
 
337
        self.next_button = self.next_rev_action.create_tool_item()
 
338
        self.toolbar.insert(self.next_button, -1)
 
339
 
 
340
        self.toolbar.insert(gtk.SeparatorToolItem(), -1)
 
341
 
 
342
        refresh_button = gtk.ToolButton(gtk.STOCK_REFRESH)
 
343
        refresh_button.connect('clicked', self._refresh_clicked)
 
344
        self.toolbar.insert(refresh_button, -1)
 
345
 
 
346
        self.toolbar.show_all()
 
347
 
 
348
        return self.toolbar
148
349
 
149
350
    def construct_bottom(self):
150
351
        """Construct the bottom half of the window."""
151
 
        scrollwin = gtk.ScrolledWindow()
152
 
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
153
 
        scrollwin.set_shadow_type(gtk.SHADOW_NONE)
 
352
        if self.config.get_user_option('viz-wide-diffs') == 'True':
 
353
            self.diff_paned = gtk.VPaned()
 
354
        else:
 
355
            self.diff_paned = gtk.HPaned()
154
356
        (width, height) = self.get_size()
155
 
        scrollwin.set_size_request(width, int(height / 2.5))
156
 
        scrollwin.show()
157
 
 
158
 
        vbox = gtk.VBox(False, spacing=6)
159
 
        vbox.set_border_width(6)
160
 
        scrollwin.add_with_viewport(vbox)
161
 
        vbox.show()
162
 
 
163
 
        table = gtk.Table(rows=4, columns=2)
164
 
        table.set_row_spacings(6)
165
 
        table.set_col_spacings(6)
166
 
        vbox.pack_start(table, expand=False, fill=True)
167
 
        table.show()
168
 
 
169
 
        align = gtk.Alignment(0.0, 0.5)
170
 
        label = gtk.Label()
171
 
        label.set_markup("<b>Revision:</b>")
172
 
        align.add(label)
173
 
        table.attach(align, 0, 1, 0, 1, gtk.FILL, gtk.FILL)
174
 
        label.show()
175
 
        align.show()
176
 
 
177
 
        align = gtk.Alignment(0.0, 0.5)
178
 
        self.revid_label = gtk.Label()
179
 
        self.revid_label.set_selectable(True)
180
 
        align.add(self.revid_label)
181
 
        table.attach(align, 1, 2, 0, 1, gtk.EXPAND | gtk.FILL, gtk.FILL)
182
 
        self.revid_label.show()
183
 
        align.show()
184
 
 
185
 
        align = gtk.Alignment(0.0, 0.5)
186
 
        label = gtk.Label()
187
 
        label.set_markup("<b>Committer:</b>")
188
 
        align.add(label)
189
 
        table.attach(align, 0, 1, 1, 2, gtk.FILL, gtk.FILL)
190
 
        label.show()
191
 
        align.show()
192
 
 
193
 
        align = gtk.Alignment(0.0, 0.5)
194
 
        self.committer_label = gtk.Label()
195
 
        self.committer_label.set_selectable(True)
196
 
        align.add(self.committer_label)
197
 
        table.attach(align, 1, 2, 1, 2, gtk.EXPAND | gtk.FILL, gtk.FILL)
198
 
        self.committer_label.show()
199
 
        align.show()
200
 
 
201
 
        align = gtk.Alignment(0.0, 0.5)
202
 
        label = gtk.Label()
203
 
        label.set_markup("<b>Branch nick:</b>")
204
 
        align.add(label)
205
 
        table.attach(align, 0, 1, 2, 3, gtk.FILL, gtk.FILL)
206
 
        label.show()
207
 
        align.show()
208
 
 
209
 
        align = gtk.Alignment(0.0, 0.5)
210
 
        self.branchnick_label = gtk.Label()
211
 
        self.branchnick_label.set_selectable(True)
212
 
        align.add(self.branchnick_label)
213
 
        table.attach(align, 1, 2, 2, 3, gtk.EXPAND | gtk.FILL, gtk.FILL)
214
 
        self.branchnick_label.show()
215
 
        align.show()
216
 
 
217
 
        align = gtk.Alignment(0.0, 0.5)
218
 
        label = gtk.Label()
219
 
        label.set_markup("<b>Timestamp:</b>")
220
 
        align.add(label)
221
 
        table.attach(align, 0, 1, 3, 4, gtk.FILL, gtk.FILL)
222
 
        label.show()
223
 
        align.show()
224
 
 
225
 
        align = gtk.Alignment(0.0, 0.5)
226
 
        self.timestamp_label = gtk.Label()
227
 
        self.timestamp_label.set_selectable(True)
228
 
        align.add(self.timestamp_label)
229
 
        table.attach(align, 1, 2, 3, 4, gtk.EXPAND | gtk.FILL, gtk.FILL)
230
 
        self.timestamp_label.show()
231
 
        align.show()
232
 
 
233
 
        self.parents_table = gtk.Table(rows=1, columns=2)
234
 
        self.parents_table.set_row_spacings(3)
235
 
        self.parents_table.set_col_spacings(6)
236
 
        self.parents_table.show()
237
 
        vbox.pack_start(self.parents_table, expand=False, fill=True)
238
 
        self.parents_widgets = []
239
 
 
240
 
        label = gtk.Label()
241
 
        label.set_markup("<b>Parents:</b>")
242
 
        align = gtk.Alignment(0.0, 0.5)
243
 
        align.add(label)
244
 
        self.parents_table.attach(align, 0, 1, 0, 1, gtk.FILL, gtk.FILL)
245
 
        label.show()
246
 
        align.show()
247
 
 
248
 
        self.message_buffer = gtk.TextBuffer()
249
 
        textview = gtk.TextView(self.message_buffer)
250
 
        textview.set_editable(False)
251
 
        textview.set_wrap_mode(gtk.WRAP_WORD)
252
 
        textview.modify_font(pango.FontDescription("Monospace"))
253
 
        vbox.pack_start(textview, expand=True, fill=True)
254
 
        textview.show()
255
 
 
256
 
        return scrollwin
257
 
 
258
 
    def set_branch(self, branch, start, maxnum):
259
 
        """Set the branch and start position for this window.
260
 
 
261
 
        Creates a new TreeModel and populates it with information about
262
 
        the new branch before updating the window title and model of the
263
 
        treeview itself.
264
 
        """
265
 
        self.branch = branch
266
 
 
267
 
        # [ revision, node, last_lines, lines, message, committer, timestamp ]
268
 
        self.model = gtk.ListStore(gobject.TYPE_PYOBJECT,
269
 
                                   gobject.TYPE_PYOBJECT,
270
 
                                   gobject.TYPE_PYOBJECT,
271
 
                                   gobject.TYPE_PYOBJECT,
272
 
                                   str, str, str)
273
 
        self.index = {}
274
 
        index = 0
275
 
 
276
 
        last_lines = []
277
 
        (self.revisions, colours, self.children, self.parent_ids,
278
 
         merge_sorted) = distances(branch, start)
279
 
        for (index, (revision, node, lines)) in enumerate(graph(
280
 
                self.revisions, colours, merge_sorted)):
281
 
            # FIXME: at this point we should be able to show the graph order
282
 
            # and lines with no message or commit data - and then incrementally
283
 
            # fill the timestamp, committer etc data as desired.
284
 
            message = revision.message.split("\n")[0]
285
 
            if revision.committer is not None:
286
 
                timestamp = format_date(revision.timestamp, revision.timezone)
287
 
            else:
288
 
                timestamp = None
289
 
            self.model.append([revision, node, last_lines, lines,
290
 
                               message, revision.committer, timestamp])
291
 
            self.index[revision] = index
292
 
            last_lines = lines
293
 
            if maxnum is not None and index > maxnum:
294
 
                break
295
 
 
296
 
        self.set_title(branch.nick + " - bzrk")
297
 
        self.treeview.set_model(self.model)
298
 
 
299
 
    def _treeview_cursor_cb(self, *args):
300
 
        """Callback for when the treeview cursor changes."""
301
 
        (path, col) = self.treeview.get_cursor()
302
 
        revision = self.model[path][0]
303
 
 
304
 
        self.back_button.set_sensitive(len(self.parent_ids[revision]) > 0)
305
 
        self.fwd_button.set_sensitive(len(self.children[revision]) > 0)
306
 
 
307
 
        if revision.committer is not None:
308
 
            branchnick = ""
309
 
            committer = revision.committer
310
 
            timestamp = format_date(revision.timestamp, revision.timezone)
311
 
            message = revision.message
312
 
            try:
313
 
                branchnick = revision.properties['branch-nick']
314
 
            except KeyError:
315
 
                pass
316
 
 
 
357
        self.diff_paned.set_size_request(20, 20) # shrinkable
 
358
 
 
359
        from bzrlib.plugins.gtk.revisionview import RevisionView
 
360
        self.revisionview = RevisionView(branch=self.branch)
 
361
        self.revisionview.set_size_request(width/3, int(height / 2.5))
 
362
        # user-configured size
 
363
        size = self._load_size('viz-revisionview-size')
 
364
        if size:
 
365
            width, height = size
 
366
            self.revisionview.set_size_request(width, height)
 
367
        self._save_size_on_destroy(self.revisionview, 'viz-revisionview-size')
 
368
        self.revisionview.show()
 
369
        self.revisionview.set_show_callback(self._show_clicked_cb)
 
370
        self.revisionview.connect('notify::revision', self._go_clicked_cb)
 
371
        self.treeview.connect('tag-added', lambda w, t, r: self.revisionview.update_tags())
 
372
        self.diff_paned.pack1(self.revisionview)
 
373
 
 
374
        from bzrlib.plugins.gtk.diff import DiffWidget
 
375
        self.diff = DiffWidget()
 
376
        self.diff_paned.pack2(self.diff)
 
377
 
 
378
        self.diff_paned.show_all()
 
379
        if self.config.get_user_option('viz-show-diffs') != 'True':
 
380
            self.diff.hide()
 
381
 
 
382
        return self.diff_paned
 
383
 
 
384
    def _tag_selected_cb(self, menuitem, revid):
 
385
        self.treeview.set_revision_id(revid)
 
386
 
 
387
    def _treeselection_changed_cb(self, selection, *args):
 
388
        """callback for when the treeview changes."""
 
389
        revision = self.treeview.get_revision()
 
390
        parents  = self.treeview.get_parents()
 
391
        children = self.treeview.get_children()
 
392
 
 
393
        self.revision_menu.set_revision_ids([revision.revision_id])
 
394
 
 
395
        if revision and revision != NULL_REVISION:
 
396
            prev_menu = gtk.Menu()
 
397
            if len(parents) > 0:
 
398
                self.prev_rev_action.set_sensitive(True)
 
399
                for parent_id in parents:
 
400
                    if parent_id and parent_id != NULL_REVISION:
 
401
                        parent = self.branch.repository.get_revision(parent_id)
 
402
                        try:
 
403
                            str = ' (' + parent.properties['branch-nick'] + ')'
 
404
                        except KeyError:
 
405
                            str = ""
 
406
 
 
407
                        item = gtk.MenuItem(parent.message.split("\n")[0] + str)
 
408
                        item.connect('activate', self._set_revision_cb, parent_id)
 
409
                        prev_menu.add(item)
 
410
                prev_menu.show_all()
 
411
            else:
 
412
                self.prev_rev_action.set_sensitive(False)
 
413
                prev_menu.hide()
 
414
 
 
415
            if getattr(self.prev_button, 'set_menu', None) is not None:
 
416
                self.prev_button.set_menu(prev_menu)
 
417
 
 
418
            next_menu = gtk.Menu()
 
419
            if len(children) > 0:
 
420
                self.next_rev_action.set_sensitive(True)
 
421
                for child_id in children:
 
422
                    child = self.branch.repository.get_revision(child_id)
 
423
                    try:
 
424
                        str = ' (' + child.properties['branch-nick'] + ')'
 
425
                    except KeyError:
 
426
                        str = ""
 
427
 
 
428
                    item = gtk.MenuItem(child.message.split("\n")[0] + str)
 
429
                    item.connect('activate', self._set_revision_cb, child_id)
 
430
                    next_menu.add(item)
 
431
                next_menu.show_all()
 
432
            else:
 
433
                self.next_rev_action.set_sensitive(False)
 
434
                next_menu.hide()
 
435
 
 
436
            if getattr(self.next_button, 'set_menu', None) is not None:
 
437
                self.next_button.set_menu(next_menu)
 
438
 
 
439
            self.revisionview.set_revision(revision)
 
440
            self.revisionview.set_children(children)
 
441
            self.update_diff_panel(revision, parents)
 
442
 
 
443
    def _tree_revision_activated(self, widget, path, col):
 
444
        # TODO: more than one parent
 
445
        """Callback for when a treeview row gets activated."""
 
446
        revision = self.treeview.get_revision()
 
447
        parents  = self.treeview.get_parents()
 
448
 
 
449
        if len(parents) == 0:
 
450
            parent_id = NULL_REVISION
317
451
        else:
318
 
            committer = ""
319
 
            timestamp = ""
320
 
            message = ""
321
 
            branchnick = ""
322
 
 
323
 
        self.revid_label.set_text(revision.revision_id)
324
 
        self.branchnick_label.set_text(branchnick)
325
 
 
326
 
        self.committer_label.set_text(committer)
327
 
        self.timestamp_label.set_text(timestamp)
328
 
        self.message_buffer.set_text(message)
329
 
 
330
 
        for widget in self.parents_widgets:
331
 
            self.parents_table.remove(widget)
332
 
 
333
 
        self.parents_widgets = []
334
 
        self.parents_table.resize(max(len(self.parent_ids[revision]), 1), 2)
335
 
        
336
 
        for idx, parent_id in enumerate(self.parent_ids[revision]):
337
 
            align = gtk.Alignment(0.0, 0.0)
338
 
            self.parents_widgets.append(align)
339
 
            self.parents_table.attach(align, 1, 2, idx, idx + 1,
340
 
                                      gtk.EXPAND | gtk.FILL, gtk.FILL)
341
 
            align.show()
342
 
 
343
 
            hbox = gtk.HBox(False, spacing=6)
344
 
            align.add(hbox)
345
 
            hbox.show()
346
 
 
347
 
            image = gtk.Image()
348
 
            image.set_from_stock(
349
 
                gtk.STOCK_JUMP_TO, gtk.ICON_SIZE_SMALL_TOOLBAR)
350
 
            image.show()
351
 
 
352
 
            button = gtk.Button()
353
 
            button.add(image)
354
 
            button.connect("clicked", self._go_clicked_cb, parent_id)
355
 
            hbox.pack_start(button, expand=False, fill=True)
356
 
            button.show()
357
 
 
358
 
            image = gtk.Image()
359
 
            image.set_from_stock(
360
 
                gtk.STOCK_FIND, gtk.ICON_SIZE_SMALL_TOOLBAR)
361
 
            image.show()
362
 
 
363
 
            button = gtk.Button()
364
 
            button.add(image)
365
 
            button.set_sensitive(self.app is not None)
366
 
            button.connect("clicked", self._show_clicked_cb,
367
 
                           revision.revision_id, parent_id)
368
 
            hbox.pack_start(button, expand=False, fill=True)
369
 
            button.show()
370
 
 
371
 
            label = gtk.Label(parent_id)
372
 
            label.set_selectable(True)
373
 
            hbox.pack_start(label, expand=False, fill=True)
374
 
            label.show()
375
 
 
 
452
            parent_id = parents[0]
 
453
 
 
454
        self.show_diff(revision.revision_id, parent_id)
 
455
        self.treeview.grab_focus()
376
456
 
377
457
    def _back_clicked_cb(self, *args):
378
458
        """Callback for when the back button is clicked."""
379
 
        (path, col) = self.treeview.get_cursor()
380
 
        revision = self.model[path][0]
381
 
        if not len(self.parent_ids[revision]):
382
 
            return
383
 
 
384
 
        for parent_id in self.parent_ids[revision]:
385
 
            parent = self.revisions[parent_id]
386
 
            if same_branch(revision, parent):
387
 
                self.treeview.set_cursor(self.index[parent])
388
 
                break
389
 
        else:
390
 
            next = self.revisions[self.parent_ids[revision][0]]
391
 
            self.treeview.set_cursor(self.index[next])
392
 
        self.treeview.grab_focus()
 
459
        self.treeview.back()
393
460
 
394
461
    def _fwd_clicked_cb(self, *args):
395
462
        """Callback for when the forward button is clicked."""
396
 
        (path, col) = self.treeview.get_cursor()
397
 
        revision = self.model[path][0]
398
 
        if not len(self.children[revision]):
399
 
            return
400
 
 
401
 
        for child in self.children[revision]:
402
 
            if same_branch(child, revision):
403
 
                self.treeview.set_cursor(self.index[child])
404
 
                break
405
 
        else:
406
 
            prev = list(self.children[revision])[0]
407
 
            self.treeview.set_cursor(self.index[prev])
408
 
        self.treeview.grab_focus()
409
 
 
410
 
    def _go_clicked_cb(self, widget, revid):
 
463
        self.treeview.forward()
 
464
 
 
465
    def _go_clicked_cb(self, w, p):
411
466
        """Callback for when the go button for a parent is clicked."""
412
 
        self.treeview.set_cursor(self.index[self.revisions[revid]])
413
 
        self.treeview.grab_focus()
 
467
        if self.revisionview.get_revision() is not None:
 
468
            self.treeview.set_revision(self.revisionview.get_revision())
414
469
 
415
 
    def _show_clicked_cb(self, widget, revid, parentid):
 
470
    def _show_clicked_cb(self, revid, parentid):
416
471
        """Callback for when the show button for a parent is clicked."""
417
 
        if self.app is not None:
418
 
            self.app.show_diff(self.branch, revid, parentid)
419
 
        self.treeview.grab_focus()
420
 
 
421
 
    def _treeview_row_activated_cb(self, widget, path, col):
422
 
        # TODO: more than one parent
423
 
        """Callback for when a treeview row gets activated."""
424
 
        revision = self.model[path][0]
425
 
        parent_id = self.parent_ids[revision][0]
426
 
        if self.app is not None:
427
 
            self.app.show_diff(self.branch, revision.revision_id, parent_id)
428
 
        self.treeview.grab_focus()
 
472
        self.show_diff(revid, parentid)
 
473
        self.treeview.grab_focus()
 
474
 
 
475
    def _set_revision_cb(self, w, revision_id):
 
476
        self.treeview.set_revision_id(revision_id)
 
477
 
 
478
    def _brokenlines_toggled_cb(self, button):
 
479
        self.compact_view = button.get_active()
 
480
 
 
481
        if self.compact_view:
 
482
            option = 'yes'
 
483
        else:
 
484
            option = 'no'
 
485
 
 
486
        self.config.set_user_option('viz-compact-view', option)
 
487
        self.treeview.set_property('compact', self.compact_view)
 
488
        self.treeview.refresh()
 
489
 
 
490
    def _branch_index_cb(self, w):
 
491
        from bzrlib.plugins.search import index as _mod_index
 
492
        _mod_index.index_url(self.branch.base)
 
493
 
 
494
    def _branch_search_cb(self, w):
 
495
        from bzrlib.plugins.search import index as _mod_index
 
496
        from bzrlib.plugins.gtk.search import SearchDialog
 
497
        from bzrlib.plugins.search import errors as search_errors
 
498
 
 
499
        try:
 
500
            index = _mod_index.open_index_url(self.branch.base)
 
501
        except search_errors.NoSearchIndex:
 
502
            dialog = gtk.MessageDialog(self, type=gtk.MESSAGE_QUESTION, 
 
503
                buttons=gtk.BUTTONS_OK_CANCEL, 
 
504
                message_format="This branch has not been indexed yet. "
 
505
                               "Index now?")
 
506
            if dialog.run() == gtk.RESPONSE_OK:
 
507
                dialog.destroy()
 
508
                index = _mod_index.index_url(self.branch.base)
 
509
            else:
 
510
                dialog.destroy()
 
511
                return
 
512
 
 
513
        dialog = SearchDialog(index)
 
514
 
 
515
        if dialog.run() == gtk.RESPONSE_OK:
 
516
            self.set_revision(dialog.get_revision())
 
517
 
 
518
        dialog.destroy()
 
519
 
 
520
    def _about_dialog_cb(self, w):
 
521
        from bzrlib.plugins.gtk.about import AboutDialog
 
522
        AboutDialog().run()
 
523
 
 
524
    def _col_visibility_changed(self, col, property):
 
525
        self.config.set_user_option(property + '-column-visible', col.get_active())
 
526
        self.treeview.set_property(property + '-column-visible', col.get_active())
 
527
 
 
528
    def _toolbar_visibility_changed(self, col):
 
529
        if col.get_active():
 
530
            self.toolbar.show()
 
531
        else:
 
532
            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
 
 
568
    def _refresh_clicked(self, w):
 
569
        self.treeview.refresh()
 
570
 
 
571
    def _update_tags(self):
 
572
        menu = gtk.Menu()
 
573
 
 
574
        if self.branch.supports_tags():
 
575
            tags = self.branch.tags.get_tag_dict().items()
 
576
            tags.sort(reverse=True)
 
577
            for tag, revid in tags:
 
578
                tag_image = gtk.Image()
 
579
                tag_image.set_from_file(icon_path('tag-16.png'))
 
580
                tag_item = gtk.ImageMenuItem(tag.replace('_', '__'))
 
581
                tag_item.set_image(tag_image)
 
582
                tag_item.connect('activate', self._tag_selected_cb, revid)
 
583
                tag_item.set_sensitive(self.treeview.has_revision_id(revid))
 
584
                menu.add(tag_item)
 
585
            self.go_menu_tags.set_submenu(menu)
 
586
 
 
587
            self.go_menu_tags.set_sensitive(len(tags) != 0)
 
588
        else:
 
589
            self.go_menu_tags.set_sensitive(False)
 
590
 
 
591
        self.go_menu_tags.show_all()
 
592
 
 
593
    def _load_size(self, name):
 
594
        """Read and parse 'name' from self.config.
 
595
        The value is a string, formatted as WIDTHxHEIGHT
 
596
        Returns None, or (width, height)
 
597
        """
 
598
        size = self.config.get_user_option(name)
 
599
        if size:
 
600
            width, height = [int(num) for num in size.split('x')]
 
601
            # avoid writing config every time we start
 
602
            return width, height
 
603
        return None
 
604
 
 
605
    def show_diff(self, revid=None, parentid=NULL_REVISION):
 
606
        """Open a new window to show a diff between the given revisions."""
 
607
        from bzrlib.plugins.gtk.diff import DiffWindow
 
608
        window = DiffWindow(parent=self)
 
609
 
 
610
        rev_tree    = self.branch.repository.revision_tree(revid)
 
611
        parent_tree = self.branch.repository.revision_tree(parentid)
 
612
 
 
613
        description = revid + " - " + self.branch._get_nick(local=True)
 
614
        window.set_diff(description, rev_tree, parent_tree)
 
615
        window.show()
 
616
 
 
617
    def update_diff_panel(self, revision=None, parents=None):
 
618
        """Show the current revision in the diff panel."""
 
619
        if self.config.get_user_option('viz-show-diffs') != 'True':
 
620
            return
 
621
 
 
622
        if not revision: # default to selected row
 
623
            revision = self.treeview.get_revision()
 
624
        if revision == NULL_REVISION:
 
625
            return
 
626
 
 
627
        if not parents: # default to selected row's parents
 
628
            parents  = self.treeview.get_parents()
 
629
        if len(parents) == 0:
 
630
            parent_id = NULL_REVISION
 
631
        else:
 
632
            parent_id = parents[0]
 
633
 
 
634
        rev_tree    = self.branch.repository.revision_tree(revision.revision_id)
 
635
        parent_tree = self.branch.repository.revision_tree(parent_id)
 
636
 
 
637
        self.diff.set_diff(rev_tree, parent_tree)
 
638
        if self.config.get_user_option('viz-wrap-diffs') == 'True':
 
639
            self.diff._on_wraplines_toggled(wrap=True)
 
640
        self.diff.show_all()