/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: Jasper Groenewegen
  • Date: 2008-07-27 12:01:40 UTC
  • mfrom: (576.3.2 improve-merge)
  • mto: This revision was merged to the branch mainline in revision 579.
  • Revision ID: colbrac@xs4all.nl-20080727120140-1agdlzkc9fumjk5f
Merge merge dialog improvements

Show diffs side-by-side

added added

removed removed

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