/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: 2006-09-30 10:21:43 UTC
  • Revision ID: jelmer@samba.org-20060930102143-c0ef64d6ca860c21
Merge some files from Olive and bzr-gtk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/branchwin.py
13
13
import gobject
14
14
import pango
15
15
 
16
 
from bzrlib.plugins.gtk import icon_path
17
 
from bzrlib.plugins.gtk.branchview import TreeView, treemodel
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):
 
16
from bzrlib.osutils import format_date
 
17
 
 
18
from graph import distances, graph, same_branch
 
19
from graphcell import CellRendererGraph
 
20
 
 
21
 
 
22
class BranchWindow(gtk.Window):
28
23
    """Branch window.
29
24
 
30
25
    This object represents and manages a single window containing information
31
26
    for a particular branch.
32
27
    """
33
28
 
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)
 
29
    def __init__(self, app=None):
 
30
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
44
31
        self.set_border_width(0)
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)
 
32
        self.set_title("bzrk")
 
33
 
 
34
        self.app = app
 
35
 
 
36
        # Use three-quarters of the screen by default
 
37
        screen = self.get_screen()
 
38
        monitor = screen.get_monitor_geometry(0)
 
39
        width = int(monitor.width * 0.75)
 
40
        height = int(monitor.height * 0.75)
70
41
        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')
73
42
 
74
43
        # FIXME AndyFitz!
75
44
        icon = self.render_icon(gtk.STOCK_INDEX, gtk.ICON_SIZE_BUTTON)
76
45
        self.set_icon(icon)
77
46
 
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
 
 
82
47
        self.accel_group = gtk.AccelGroup()
83
48
        self.add_accel_group(self.accel_group)
84
49
 
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
 
 
105
50
        self.construct()
106
51
 
107
 
    def set_revision(self, revid):
108
 
        self.treeview.set_revision_id(revid)
109
 
 
110
52
    def construct(self):
111
53
        """Construct the window contents."""
112
54
        vbox = gtk.VBox(spacing=0)
113
55
        self.add(vbox)
114
56
 
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)
 
57
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
58
 
 
59
        paned = gtk.VPaned()
 
60
        paned.pack1(self.construct_top(), resize=True, shrink=False)
 
61
        paned.pack2(self.construct_bottom(), resize=False, shrink=True)
 
62
        paned.show()
 
63
        vbox.pack_start(paned, expand=True, fill=True)
 
64
        vbox.set_focus_child(paned)
127
65
 
128
66
        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
285
67
 
286
68
    def construct_top(self):
287
69
        """Construct the top-half of the window."""
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
 
 
 
70
        scrollwin = gtk.ScrolledWindow()
 
71
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
 
72
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
 
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
        self.treeview.connect("row-activated", self._treeview_row_activated_cb)
 
80
        scrollwin.add(self.treeview)
306
81
        self.treeview.show()
307
82
 
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
 
83
        cell = CellRendererGraph()
 
84
        column = gtk.TreeViewColumn()
 
85
        column.set_resizable(True)
 
86
        column.pack_start(cell, expand=False)
 
87
        column.add_attribute(cell, "node", 1)
 
88
        column.add_attribute(cell, "in-lines", 2)
 
89
        column.add_attribute(cell, "out-lines", 3)
 
90
        self.treeview.append_column(column)
 
91
 
 
92
        cell = gtk.CellRendererText()
 
93
        cell.set_property("width-chars", 40)
 
94
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
 
95
        column = gtk.TreeViewColumn("Message")
 
96
        column.set_resizable(True)
 
97
        column.pack_start(cell, expand=True)
 
98
        column.add_attribute(cell, "text", 4)
 
99
        self.treeview.append_column(column)
 
100
 
 
101
        cell = gtk.CellRendererText()
 
102
        cell.set_property("width-chars", 40)
 
103
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
 
104
        column = gtk.TreeViewColumn("Committer")
 
105
        column.set_resizable(True)
 
106
        column.pack_start(cell, expand=True)
 
107
        column.add_attribute(cell, "text", 5)
 
108
        self.treeview.append_column(column)
 
109
 
 
110
        cell = gtk.CellRendererText()
 
111
        cell.set_property("ellipsize", pango.ELLIPSIZE_END)
 
112
        column = gtk.TreeViewColumn("Date")
 
113
        column.set_resizable(True)
 
114
        column.pack_start(cell, expand=True)
 
115
        column.add_attribute(cell, "text", 6)
 
116
        self.treeview.append_column(column)
 
117
 
 
118
        return scrollwin
323
119
 
324
120
    def construct_navigation(self):
325
121
        """Construct the navigation buttons."""
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
 
122
        frame = gtk.Frame()
 
123
        frame.set_shadow_type(gtk.SHADOW_OUT)
 
124
        frame.show()
 
125
        
 
126
        hbox = gtk.HBox(spacing=12)
 
127
        frame.add(hbox)
 
128
        hbox.show()
 
129
 
 
130
        self.back_button = gtk.Button(stock=gtk.STOCK_GO_BACK)
 
131
        self.back_button.set_relief(gtk.RELIEF_NONE)
 
132
        self.back_button.add_accelerator("clicked", self.accel_group, ord('['),
 
133
                                         0, 0)
 
134
        self.back_button.connect("clicked", self._back_clicked_cb)
 
135
        hbox.pack_start(self.back_button, expand=False, fill=True)
 
136
        self.back_button.show()
 
137
 
 
138
        self.fwd_button = gtk.Button(stock=gtk.STOCK_GO_FORWARD)
 
139
        self.fwd_button.set_relief(gtk.RELIEF_NONE)
 
140
        self.fwd_button.add_accelerator("clicked", self.accel_group, ord(']'),
 
141
                                        0, 0)
 
142
        self.fwd_button.connect("clicked", self._fwd_clicked_cb)
 
143
        hbox.pack_start(self.fwd_button, expand=False, fill=True)
 
144
        self.fwd_button.show()
 
145
 
 
146
        return frame
344
147
 
345
148
    def construct_bottom(self):
346
149
        """Construct the bottom half of the window."""
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()
 
150
        scrollwin = gtk.ScrolledWindow()
 
151
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
 
152
        scrollwin.set_shadow_type(gtk.SHADOW_NONE)
351
153
        (width, height) = self.get_size()
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
 
154
        scrollwin.set_size_request(width, int(height / 2.5))
 
155
        scrollwin.show()
 
156
 
 
157
        vbox = gtk.VBox(False, spacing=6)
 
158
        vbox.set_border_width(6)
 
159
        scrollwin.add_with_viewport(vbox)
 
160
        vbox.show()
 
161
 
 
162
        table = gtk.Table(rows=4, columns=2)
 
163
        table.set_row_spacings(6)
 
164
        table.set_col_spacings(6)
 
165
        vbox.pack_start(table, expand=False, fill=True)
 
166
        table.show()
 
167
 
 
168
        align = gtk.Alignment(0.0, 0.5)
 
169
        label = gtk.Label()
 
170
        label.set_markup("<b>Revision:</b>")
 
171
        align.add(label)
 
172
        table.attach(align, 0, 1, 0, 1, gtk.FILL, gtk.FILL)
 
173
        label.show()
 
174
        align.show()
 
175
 
 
176
        align = gtk.Alignment(0.0, 0.5)
 
177
        self.revid_label = gtk.Label()
 
178
        self.revid_label.set_selectable(True)
 
179
        align.add(self.revid_label)
 
180
        table.attach(align, 1, 2, 0, 1, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
181
        self.revid_label.show()
 
182
        align.show()
 
183
 
 
184
        align = gtk.Alignment(0.0, 0.5)
 
185
        label = gtk.Label()
 
186
        label.set_markup("<b>Committer:</b>")
 
187
        align.add(label)
 
188
        table.attach(align, 0, 1, 1, 2, gtk.FILL, gtk.FILL)
 
189
        label.show()
 
190
        align.show()
 
191
 
 
192
        align = gtk.Alignment(0.0, 0.5)
 
193
        self.committer_label = gtk.Label()
 
194
        self.committer_label.set_selectable(True)
 
195
        align.add(self.committer_label)
 
196
        table.attach(align, 1, 2, 1, 2, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
197
        self.committer_label.show()
 
198
        align.show()
 
199
 
 
200
        align = gtk.Alignment(0.0, 0.5)
 
201
        label = gtk.Label()
 
202
        label.set_markup("<b>Branch nick:</b>")
 
203
        align.add(label)
 
204
        table.attach(align, 0, 1, 2, 3, gtk.FILL, gtk.FILL)
 
205
        label.show()
 
206
        align.show()
 
207
 
 
208
        align = gtk.Alignment(0.0, 0.5)
 
209
        self.branchnick_label = gtk.Label()
 
210
        self.branchnick_label.set_selectable(True)
 
211
        align.add(self.branchnick_label)
 
212
        table.attach(align, 1, 2, 2, 3, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
213
        self.branchnick_label.show()
 
214
        align.show()
 
215
 
 
216
        align = gtk.Alignment(0.0, 0.5)
 
217
        label = gtk.Label()
 
218
        label.set_markup("<b>Timestamp:</b>")
 
219
        align.add(label)
 
220
        table.attach(align, 0, 1, 3, 4, gtk.FILL, gtk.FILL)
 
221
        label.show()
 
222
        align.show()
 
223
 
 
224
        align = gtk.Alignment(0.0, 0.5)
 
225
        self.timestamp_label = gtk.Label()
 
226
        self.timestamp_label.set_selectable(True)
 
227
        align.add(self.timestamp_label)
 
228
        table.attach(align, 1, 2, 3, 4, gtk.EXPAND | gtk.FILL, gtk.FILL)
 
229
        self.timestamp_label.show()
 
230
        align.show()
 
231
 
 
232
        self.parents_table = gtk.Table(rows=1, columns=2)
 
233
        self.parents_table.set_row_spacings(3)
 
234
        self.parents_table.set_col_spacings(6)
 
235
        self.parents_table.show()
 
236
        vbox.pack_start(self.parents_table, expand=False, fill=True)
 
237
        self.parents_widgets = []
 
238
 
 
239
        label = gtk.Label()
 
240
        label.set_markup("<b>Parents:</b>")
 
241
        align = gtk.Alignment(0.0, 0.5)
 
242
        align.add(label)
 
243
        self.parents_table.attach(align, 0, 1, 0, 1, gtk.FILL, gtk.FILL)
 
244
        label.show()
 
245
        align.show()
 
246
 
 
247
        self.message_buffer = gtk.TextBuffer()
 
248
        textview = gtk.TextView(self.message_buffer)
 
249
        textview.set_editable(False)
 
250
        textview.set_wrap_mode(gtk.WRAP_WORD)
 
251
        textview.modify_font(pango.FontDescription("Monospace"))
 
252
        vbox.pack_start(textview, expand=True, fill=True)
 
253
        textview.show()
 
254
 
 
255
        return scrollwin
 
256
 
 
257
    def set_branch(self, branch, start, maxnum):
 
258
        """Set the branch and start position for this window.
 
259
 
 
260
        Creates a new TreeModel and populates it with information about
 
261
        the new branch before updating the window title and model of the
 
262
        treeview itself.
 
263
        """
 
264
        self.branch = branch
 
265
 
 
266
        # [ revision, node, last_lines, lines, message, committer, timestamp ]
 
267
        self.model = gtk.ListStore(gobject.TYPE_PYOBJECT,
 
268
                                   gobject.TYPE_PYOBJECT,
 
269
                                   gobject.TYPE_PYOBJECT,
 
270
                                   gobject.TYPE_PYOBJECT,
 
271
                                   str, str, str)
 
272
        self.index = {}
 
273
        index = 0
 
274
 
 
275
        last_lines = []
 
276
        (self.revisions, colours, self.children, self.parent_ids,
 
277
         merge_sorted) = distances(branch, start)
 
278
        for (index, (revision, node, lines)) in enumerate(graph(
 
279
                self.revisions, colours, merge_sorted)):
 
280
            # FIXME: at this point we should be able to show the graph order
 
281
            # and lines with no message or commit data - and then incrementally
 
282
            # fill the timestamp, committer etc data as desired.
 
283
            message = revision.message.split("\n")[0]
 
284
            if revision.committer is not None:
 
285
                timestamp = format_date(revision.timestamp, revision.timezone)
 
286
            else:
 
287
                timestamp = None
 
288
            self.model.append([revision, node, last_lines, lines,
 
289
                               message, revision.committer, timestamp])
 
290
            self.index[revision] = index
 
291
            last_lines = lines
 
292
            if maxnum is not None and index > maxnum:
 
293
                break
 
294
 
 
295
        self.set_title(branch.nick + " - bzrk")
 
296
        self.treeview.set_model(self.model)
 
297
 
 
298
    def _treeview_cursor_cb(self, *args):
 
299
        """Callback for when the treeview cursor changes."""
 
300
        (path, col) = self.treeview.get_cursor()
 
301
        revision = self.model[path][0]
 
302
 
 
303
        self.back_button.set_sensitive(len(self.parent_ids[revision]) > 0)
 
304
        self.fwd_button.set_sensitive(len(self.children[revision]) > 0)
 
305
 
 
306
        if revision.committer is not None:
 
307
            branchnick = ""
 
308
            committer = revision.committer
 
309
            timestamp = format_date(revision.timestamp, revision.timezone)
 
310
            message = revision.message
 
311
            try:
 
312
                branchnick = revision.properties['branch-nick']
 
313
            except KeyError:
 
314
                pass
 
315
 
444
316
        else:
445
 
            parent_id = parents[0]
446
 
 
447
 
        self.show_diff(revision.revision_id, parent_id)
448
 
        self.treeview.grab_focus()
 
317
            committer = ""
 
318
            timestamp = ""
 
319
            message = ""
 
320
            branchnick = ""
 
321
 
 
322
        self.revid_label.set_text(revision.revision_id)
 
323
        self.branchnick_label.set_text(branchnick)
 
324
 
 
325
        self.committer_label.set_text(committer)
 
326
        self.timestamp_label.set_text(timestamp)
 
327
        self.message_buffer.set_text(message)
 
328
 
 
329
        for widget in self.parents_widgets:
 
330
            self.parents_table.remove(widget)
 
331
 
 
332
        self.parents_widgets = []
 
333
        self.parents_table.resize(max(len(self.parent_ids[revision]), 1), 2)
449
334
        
 
335
        for idx, parent_id in enumerate(self.parent_ids[revision]):
 
336
            align = gtk.Alignment(0.0, 0.0)
 
337
            self.parents_widgets.append(align)
 
338
            self.parents_table.attach(align, 1, 2, idx, idx + 1,
 
339
                                      gtk.EXPAND | gtk.FILL, gtk.FILL)
 
340
            align.show()
 
341
 
 
342
            hbox = gtk.HBox(False, spacing=6)
 
343
            align.add(hbox)
 
344
            hbox.show()
 
345
 
 
346
            image = gtk.Image()
 
347
            image.set_from_stock(
 
348
                gtk.STOCK_FIND, gtk.ICON_SIZE_SMALL_TOOLBAR)
 
349
            image.show()
 
350
 
 
351
            button = gtk.Button()
 
352
            button.add(image)
 
353
            button.set_sensitive(self.app is not None)
 
354
            button.connect("clicked", self._show_clicked_cb,
 
355
                           revision.revision_id, parent_id)
 
356
            hbox.pack_start(button, expand=False, fill=True)
 
357
            button.show()
 
358
 
 
359
            button = gtk.Button(parent_id)
 
360
            button.set_use_underline(False)
 
361
            button.connect("clicked", self._go_clicked_cb, parent_id)
 
362
            hbox.pack_start(button, expand=False, fill=True)
 
363
            button.show()
 
364
 
 
365
 
450
366
    def _back_clicked_cb(self, *args):
451
367
        """Callback for when the back button is clicked."""
452
 
        self.treeview.back()
453
 
        
 
368
        (path, col) = self.treeview.get_cursor()
 
369
        revision = self.model[path][0]
 
370
        if not len(self.parent_ids[revision]):
 
371
            return
 
372
 
 
373
        for parent_id in self.parent_ids[revision]:
 
374
            parent = self.revisions[parent_id]
 
375
            if same_branch(revision, parent):
 
376
                self.treeview.set_cursor(self.index[parent])
 
377
                break
 
378
        else:
 
379
            next = self.revisions[self.parent_ids[revision][0]]
 
380
            self.treeview.set_cursor(self.index[next])
 
381
        self.treeview.grab_focus()
 
382
 
454
383
    def _fwd_clicked_cb(self, *args):
455
384
        """Callback for when the forward button is clicked."""
456
 
        self.treeview.forward()
457
 
 
458
 
    def _go_clicked_cb(self, w, p):
 
385
        (path, col) = self.treeview.get_cursor()
 
386
        revision = self.model[path][0]
 
387
        if not len(self.children[revision]):
 
388
            return
 
389
 
 
390
        for child in self.children[revision]:
 
391
            if same_branch(child, revision):
 
392
                self.treeview.set_cursor(self.index[child])
 
393
                break
 
394
        else:
 
395
            prev = list(self.children[revision])[0]
 
396
            self.treeview.set_cursor(self.index[prev])
 
397
        self.treeview.grab_focus()
 
398
 
 
399
    def _go_clicked_cb(self, widget, revid):
459
400
        """Callback for when the go button for a parent is clicked."""
460
 
        if self.revisionview.get_revision() is not None:
461
 
            self.treeview.set_revision(self.revisionview.get_revision())
 
401
        self.treeview.set_cursor(self.index[self.revisions[revid]])
 
402
        self.treeview.grab_focus()
462
403
 
463
 
    def _show_clicked_cb(self, revid, parentid):
 
404
    def _show_clicked_cb(self, widget, revid, parentid):
464
405
        """Callback for when the show button for a parent is clicked."""
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()
 
406
        if self.app is not None:
 
407
            self.app.show_diff(self.branch, revid, parentid)
 
408
        self.treeview.grab_focus()
 
409
 
 
410
    def _treeview_row_activated_cb(self, widget, path, col):
 
411
        # TODO: more than one parent
 
412
        """Callback for when a treeview row gets activated."""
 
413
        revision = self.model[path][0]
 
414
        if len(self.parent_ids[revision]) == 0:
 
415
            # Ignore revisions without parent
 
416
            return
 
417
        parent_id = self.parent_ids[revision][0]
 
418
        if self.app is not None:
 
419
            self.app.show_diff(self.branch, revision.revision_id, parent_id)
 
420
        self.treeview.grab_focus()