/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 branchwin.py

  • Committer: Scott James Remnant
  • Date: 2005-10-28 16:29:57 UTC
  • Revision ID: scott@ubuntu.com-20051028162957-048fa447b8a8ed28
Bump version to 0.2 and mention use of bzr 0.6

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/usr/bin/python
 
2
# -*- coding: UTF-8 -*-
1
3
"""Branch window.
2
4
 
3
5
This module contains the code to manage the branch information window,
4
6
which contains both the revision graph and details panes.
5
7
"""
6
8
 
7
 
__copyright__ = "Copyright (c) 2005 Canonical Ltd."
 
9
__copyright__ = "Copyright © 2005 Canonical Ltd."
8
10
__author__    = "Scott James Remnant <scott@ubuntu.com>"
9
11
 
10
12
 
11
 
from gi.repository import Gdk
12
 
from gi.repository import Gtk
13
 
 
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
 
from bzrlib.plugins.gtk.window import Window
19
 
 
20
 
from bzrlib.config import GlobalConfig
21
 
from bzrlib.revision import NULL_REVISION
22
 
from bzrlib.trace import mutter
23
 
 
24
 
 
25
 
class BranchWindow(Window):
 
13
import os
 
14
 
 
15
import gtk
 
16
import gobject
 
17
import pango
 
18
 
 
19
from bzrlib.osutils import format_date
 
20
 
 
21
from graph import distances, graph, same_branch
 
22
from graphcell import CellRendererGraph
 
23
 
 
24
 
 
25
class BranchWindow(gtk.Window):
26
26
    """Branch window.
27
27
 
28
28
    This object represents and manages a single window containing information
29
29
    for a particular branch.
30
30
    """
31
31
 
32
 
    def __init__(self, branch, start_revs, maxnum, parent=None):
33
 
        """Create a new BranchWindow.
34
 
 
35
 
        :param branch: Branch object for branch to show.
36
 
        :param start_revs: Revision ids of top revisions.
37
 
        :param maxnum: Maximum number of revisions to display, 
38
 
                       None for no limit.
39
 
        """
40
 
 
41
 
        super(BranchWindow, self).__init__(parent=parent)
 
32
    def __init__(self, app=None):
 
33
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
42
34
        self.set_border_width(0)
43
 
 
44
 
        self.branch      = branch
45
 
        self.start_revs  = start_revs
46
 
        self.maxnum      = maxnum
47
 
        self.config      = GlobalConfig()
48
 
 
49
 
        if self.config.get_user_option('viz-compact-view') == 'yes':
50
 
            self.compact_view = True
51
 
        else:
52
 
            self.compact_view = False
53
 
 
54
 
        self.set_title(branch._get_nick(local=True) + " - revision history")
55
 
 
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)
 
35
        self.set_title("bzrk")
 
36
 
 
37
        self.app = app
 
38
 
 
39
        # Use three-quarters of the screen by default
 
40
        screen = self.get_screen()
 
41
        monitor = screen.get_monitor_geometry(0)
 
42
        width = int(monitor.width * 0.75)
 
43
        height = int(monitor.height * 0.75)
66
44
        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
45
 
70
46
        # FIXME AndyFitz!
71
 
        icon = self.render_icon_pixbuf(Gtk.STOCK_INDEX, Gtk.IconSize.BUTTON)
 
47
        icon = self.render_icon(gtk.STOCK_INDEX, gtk.ICON_SIZE_BUTTON)
72
48
        self.set_icon(icon)
73
49
 
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()
 
50
        self.accel_group = gtk.AccelGroup()
79
51
        self.add_accel_group(self.accel_group)
80
52
 
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)
 
53
        self.construct()
113
54
 
114
55
    def construct(self):
115
56
        """Construct the window contents."""
116
 
        vbox = Gtk.VBox(spacing=0)
117
 
        self.add(vbox)
118
 
 
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
 
 
129
 
 
130
 
        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
 
57
        paned = gtk.VPaned()
 
58
        paned.pack1(self.construct_top(), resize=True, shrink=False)
 
59
        paned.pack2(self.construct_bottom(), resize=False, shrink=True)
 
60
        self.add(paned)
 
61
        paned.show()
316
62
 
317
63
    def construct_top(self):
318
64
        """Construct the top-half of the window."""
319
 
        # 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)
331
 
 
 
65
        vbox = gtk.VBox(spacing=6)
 
66
        vbox.set_border_width(12)
 
67
        vbox.show()
 
68
 
 
69
        scrollwin = gtk.ScrolledWindow()
 
70
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
 
71
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
 
72
        vbox.pack_start(scrollwin, expand=True, fill=True)
 
73
        scrollwin.show()
 
74
 
 
75
        self.treeview = gtk.TreeView()
 
76
        self.treeview.set_rules_hint(True)
 
77
        self.treeview.set_search_column(4)
 
78
        self.treeview.connect("cursor-changed", self._treeview_cursor_cb)
 
79
        scrollwin.add(self.treeview)
332
80
        self.treeview.show()
333
81
 
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
349
 
 
350
 
    def construct_navigation(self):
351
 
        """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)
366
 
 
367
 
        self.toolbar.show_all()
368
 
 
369
 
        return self.toolbar
 
82
        cell = CellRendererGraph()
 
83
        column = gtk.TreeViewColumn()
 
84
        column.set_resizable(False)
 
85
        column.pack_start(cell, expand=False)
 
86
        column.add_attribute(cell, "node", 1)
 
87
        column.add_attribute(cell, "in-lines", 2)
 
88
        column.add_attribute(cell, "out-lines", 3)
 
89
        self.treeview.append_column(column)
 
90
 
 
91
        cell = gtk.CellRendererText()
 
92
        cell.set_property("width-chars", 40)
 
93
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
 
94
        column = gtk.TreeViewColumn("Message")
 
95
        column.set_resizable(True)
 
96
        column.pack_start(cell, expand=True)
 
97
        column.add_attribute(cell, "text", 4)
 
98
        self.treeview.append_column(column)
 
99
 
 
100
        cell = gtk.CellRendererText()
 
101
        cell.set_property("width-chars", 40)
 
102
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
 
103
        column = gtk.TreeViewColumn("Committer")
 
104
        column.set_resizable(True)
 
105
        column.pack_start(cell, expand=True)
 
106
        column.add_attribute(cell, "text", 5)
 
107
        self.treeview.append_column(column)
 
108
 
 
109
        cell = gtk.CellRendererText()
 
110
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
 
111
        column = gtk.TreeViewColumn("Date")
 
112
        column.set_resizable(True)
 
113
        column.pack_start(cell, expand=True)
 
114
        column.add_attribute(cell, "text", 6)
 
115
        self.treeview.append_column(column)
 
116
 
 
117
        hbox = gtk.HBox(False, spacing=6)
 
118
        vbox.pack_start(hbox, expand=False, fill=False)
 
119
        hbox.show()
 
120
 
 
121
        self.back_button = gtk.Button(stock=gtk.STOCK_GO_BACK)
 
122
        self.back_button.add_accelerator("clicked", self.accel_group, ord('['),
 
123
                                         0, 0)
 
124
        self.back_button.connect("clicked", self._back_clicked_cb)
 
125
        hbox.pack_start(self.back_button, expand=False, fill=True)
 
126
        self.back_button.show()
 
127
 
 
128
        self.fwd_button = gtk.Button(stock=gtk.STOCK_GO_FORWARD)
 
129
        self.fwd_button.add_accelerator("clicked", self.accel_group, ord(']'),
 
130
                                        0, 0)
 
131
        self.fwd_button.connect("clicked", self._fwd_clicked_cb)
 
132
        hbox.pack_start(self.fwd_button, expand=False, fill=True)
 
133
        self.fwd_button.show()
 
134
 
 
135
        return vbox
370
136
 
371
137
    def construct_bottom(self):
372
138
        """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)
 
139
        vbox = gtk.VBox(False, spacing=6)
 
140
        vbox.set_border_width(12)
377
141
        (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')
389
 
        self.revisionview.show()
390
 
        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
 
 
413
 
    def _treeselection_changed_cb(self, selection, *args):
414
 
        """callback for when the treeview changes."""
415
 
        revision = self.treeview.get_revision()
416
 
        parents  = self.treeview.get_parents()
417
 
        children = self.treeview.get_children()
418
 
 
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
 
        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()
 
142
        vbox.set_size_request(width, int(height / 2.5))
 
143
        vbox.show()
 
144
 
 
145
        self.table = gtk.Table(rows=4, columns=2)
 
146
        self.table.set_row_spacings(6)
 
147
        self.table.set_col_spacings(6)
 
148
        vbox.pack_start(self.table, expand=False, fill=True)
 
149
        self.table.show()
 
150
 
 
151
        align = gtk.Alignment(0.0, 0.5)
 
152
        label = gtk.Label()
 
153
        label.set_markup("<b>Revision:</b>")
 
154
        align.add(label)
 
155
        self.table.attach(align, 0, 1, 0, 1, gtk.FILL, gtk.FILL)
 
156
        label.show()
 
157
        align.show()
 
158
 
 
159
        align = gtk.Alignment(0.0, 0.5)
 
160
        self.revid_label = gtk.Label()
 
161
        self.revid_label.set_selectable(True)
 
162
        align.add(self.revid_label)
 
163
        self.table.attach(align, 1, 2, 0, 1, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
164
        self.revid_label.show()
 
165
        align.show()
 
166
 
 
167
        align = gtk.Alignment(0.0, 0.5)
 
168
        label = gtk.Label()
 
169
        label.set_markup("<b>Committer:</b>")
 
170
        align.add(label)
 
171
        self.table.attach(align, 0, 1, 1, 2, gtk.FILL, gtk.FILL)
 
172
        label.show()
 
173
        align.show()
 
174
 
 
175
        align = gtk.Alignment(0.0, 0.5)
 
176
        self.committer_label = gtk.Label()
 
177
        self.committer_label.set_selectable(True)
 
178
        align.add(self.committer_label)
 
179
        self.table.attach(align, 1, 2, 1, 2, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
180
        self.committer_label.show()
 
181
        align.show()
 
182
 
 
183
        align = gtk.Alignment(0.0, 0.5)
 
184
        label = gtk.Label()
 
185
        label.set_markup("<b>Timestamp:</b>")
 
186
        align.add(label)
 
187
        self.table.attach(align, 0, 1, 2, 3, gtk.FILL, gtk.FILL)
 
188
        label.show()
 
189
        align.show()
 
190
 
 
191
        align = gtk.Alignment(0.0, 0.5)
 
192
        self.timestamp_label = gtk.Label()
 
193
        self.timestamp_label.set_selectable(True)
 
194
        align.add(self.timestamp_label)
 
195
        self.table.attach(align, 1, 2, 2, 3, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
196
        self.timestamp_label.show()
 
197
        align.show()
 
198
 
 
199
        align = gtk.Alignment(0.0, 0.5)
 
200
        label = gtk.Label()
 
201
        label.set_markup("<b>Parents:</b>")
 
202
        align.add(label)
 
203
        self.table.attach(align, 0, 1, 3, 4, gtk.FILL, gtk.FILL)
 
204
        label.show()
 
205
        align.show()
 
206
 
 
207
        self.parents_widgets = []
 
208
 
 
209
        scrollwin = gtk.ScrolledWindow()
 
210
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
 
211
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
 
212
        vbox.pack_start(scrollwin, expand=True, fill=True)
 
213
        scrollwin.show()
 
214
 
 
215
        self.message_buffer = gtk.TextBuffer()
 
216
        textview = gtk.TextView(self.message_buffer)
 
217
        textview.set_editable(False)
 
218
        textview.set_wrap_mode(gtk.WRAP_WORD)
 
219
        textview.modify_font(pango.FontDescription("Monospace"))
 
220
        scrollwin.add(textview)
 
221
        textview.show()
 
222
 
 
223
        return vbox
 
224
 
 
225
    def set_branch(self, branch, start):
 
226
        """Set the branch and start position for this window.
 
227
 
 
228
        Creates a new TreeModel and populates it with information about
 
229
        the new branch before updating the window title and model of the
 
230
        treeview itself.
 
231
        """
 
232
        self.branch = branch
 
233
 
 
234
        # [ revision, node, last_lines, lines, message, committer, timestamp ]
 
235
        self.model = gtk.ListStore(gobject.TYPE_PYOBJECT,
 
236
                                   gobject.TYPE_PYOBJECT,
 
237
                                   gobject.TYPE_PYOBJECT,
 
238
                                   gobject.TYPE_PYOBJECT,
 
239
                                   str, str, str)
 
240
        self.index = {}
 
241
        index = 0
 
242
 
 
243
        last_lines = []
 
244
        (revids, self.revisions, colours, self.children) \
 
245
                 = distances(branch, start)
 
246
        for revision, node, lines in graph(revids, self.revisions, colours):
 
247
            message = revision.message.split("\n")[0]
 
248
            if revision.committer is not None:
 
249
                timestamp = format_date(revision.timestamp, revision.timezone)
 
250
            else:
 
251
                timestamp = None
 
252
 
 
253
            self.model.append([ revision, node, last_lines, lines,
 
254
                                message, revision.committer, timestamp ])
 
255
            self.index[revision] = index
 
256
            index += 1
 
257
 
 
258
            last_lines = lines
 
259
 
 
260
        self.set_title(os.path.basename(branch.base) + " - bzrk")
 
261
        self.treeview.set_model(self.model)
 
262
 
 
263
    def _treeview_cursor_cb(self, *args):
 
264
        """Callback for when the treeview cursor changes."""
 
265
        (path, col) = self.treeview.get_cursor()
 
266
        revision = self.model[path][0]
 
267
 
 
268
        self.back_button.set_sensitive(len(revision.parent_ids) > 0)
 
269
        self.fwd_button.set_sensitive(len(self.children[revision]) > 0)
 
270
 
 
271
        if revision.committer is not None:
 
272
            committer = revision.committer
 
273
            timestamp = format_date(revision.timestamp, revision.timezone)
 
274
            message = revision.message
 
275
        else:
 
276
            committer = ""
 
277
            timestamp = ""
 
278
            message = ""
 
279
 
 
280
        self.revid_label.set_text(revision.revision_id)
 
281
        self.committer_label.set_text(committer)
 
282
        self.timestamp_label.set_text(timestamp)
 
283
        self.message_buffer.set_text(message)
 
284
 
 
285
        for widget in self.parents_widgets:
 
286
            self.table.remove(widget)
 
287
 
 
288
        self.parents_widgets = []
 
289
        self.table.resize(4 + len(revision.parent_ids) - 1, 2)
 
290
        for idx, parent_id in enumerate(revision.parent_ids):
 
291
            self.table.set_row_spacing(idx + 3, 0)
 
292
 
 
293
            align = gtk.Alignment(0.0, 0.0)
 
294
            self.parents_widgets.append(align)
 
295
            self.table.attach(align, 1, 2, idx + 3, idx + 4,
 
296
                              gtk.EXPAND | gtk.FILL, gtk.FILL)
 
297
            align.show()
 
298
 
 
299
            hbox = gtk.HBox(False, 0)
 
300
            align.add(hbox)
 
301
            hbox.show()
 
302
 
 
303
            label = gtk.Label(parent_id)
 
304
            label.set_selectable(True)
 
305
            hbox.pack_start(label, expand=False, fill=True)
 
306
            label.show()
 
307
 
 
308
            image = gtk.Image()
 
309
            image.set_from_stock(gtk.STOCK_JUMP_TO, gtk.ICON_SIZE_MENU)
 
310
            image.show()
 
311
 
 
312
            button = gtk.Button()
 
313
            button.add(image)
 
314
            button.set_relief(gtk.RELIEF_NONE)
 
315
            button.connect("clicked", self._go_clicked_cb, parent_id)
 
316
            hbox.pack_start(button, expand=False, fill=True)
 
317
            button.show()
 
318
 
 
319
            image = gtk.Image()
 
320
            image.set_from_stock(gtk.STOCK_FIND, gtk.ICON_SIZE_MENU)
 
321
            image.show()
 
322
 
 
323
            button = gtk.Button()
 
324
            button.add(image)
 
325
            button.set_relief(gtk.RELIEF_NONE)
 
326
            button.set_sensitive(self.app is not None)
 
327
            button.connect("clicked", self._show_clicked_cb,
 
328
                           revision.revision_id, parent_id)
 
329
            hbox.pack_start(button, expand=False, fill=True)
 
330
            button.show()
486
331
 
487
332
    def _back_clicked_cb(self, *args):
488
333
        """Callback for when the back button is clicked."""
489
 
        self.treeview.back()
 
334
        (path, col) = self.treeview.get_cursor()
 
335
        revision = self.model[path][0]
 
336
        if not len(revision.parent_ids):
 
337
            return
 
338
 
 
339
        for parent_id in revision.parent_ids:
 
340
            parent = self.revisions[parent_id]
 
341
            if same_branch(revision, parent):
 
342
                self.treeview.set_cursor(self.index[parent])
 
343
                break
 
344
        else:
 
345
            next = self.revisions[revision.parent_ids[0]]
 
346
            self.treeview.set_cursor(self.index[next])
 
347
        self.treeview.grab_focus()
490
348
 
491
349
    def _fwd_clicked_cb(self, *args):
492
350
        """Callback for when the forward button is clicked."""
493
 
        self.treeview.forward()
494
 
 
495
 
    def _go_clicked_cb(self, w, p):
 
351
        (path, col) = self.treeview.get_cursor()
 
352
        revision = self.model[path][0]
 
353
        if not len(self.children[revision]):
 
354
            return
 
355
 
 
356
        for child in self.children[revision]:
 
357
            if same_branch(child, revision):
 
358
                self.treeview.set_cursor(self.index[child])
 
359
                break
 
360
        else:
 
361
            prev = list(self.children[revision])[0]
 
362
            self.treeview.set_cursor(self.index[prev])
 
363
        self.treeview.grab_focus()
 
364
 
 
365
    def _go_clicked_cb(self, widget, revid):
496
366
        """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())
 
367
        self.treeview.set_cursor(self.index[self.revisions[revid]])
 
368
        self.treeview.grab_focus()
499
369
 
500
 
    def _show_clicked_cb(self, revid, parentid):
 
370
    def _show_clicked_cb(self, widget, revid, parentid):
501
371
        """Callback for when the show button for a parent is clicked."""
502
 
        self.show_diff(revid, parentid)
 
372
        if self.app is not None:
 
373
            self.app.show_diff(self.branch, revid, parentid)
503
374
        self.treeview.grab_focus()
504
 
 
505
 
    def _set_revision_cb(self, w, revision_id):
506
 
        self.treeview.set_revision_id(revision_id)
507
 
 
508
 
    def _brokenlines_toggled_cb(self, button):
509
 
        self.compact_view = button.get_active()
510
 
 
511
 
        if self.compact_view:
512
 
            option = 'yes'
513
 
        else:
514
 
            option = 'no'
515
 
 
516
 
        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()