16
16
from bzrlib.plugins.gtk.window import Window
 
 
17
from bzrlib.plugins.gtk import icon_path
 
17
18
from bzrlib.plugins.gtk.tags import AddTagDialog
 
18
19
from bzrlib.plugins.gtk.preferences import PreferencesWindow
 
19
 
from bzrlib.plugins.gtk.branchview import TreeView
 
20
 
from bzrlib.revision import Revision
 
 
20
from bzrlib.plugins.gtk.branchview import TreeView, treemodel
 
 
21
from bzrlib.revision import Revision, NULL_REVISION
 
21
22
from bzrlib.config import BranchConfig
 
22
23
from bzrlib.config import GlobalConfig
 
 
28
29
    for a particular branch.
 
31
 
    def __init__(self, branch, start, maxnum, parent=None):
 
 
32
    def __init__(self, branch, start_revs, maxnum, parent=None):
 
32
33
        """Create a new BranchWindow.
 
34
35
        :param branch: Branch object for branch to show.
 
35
 
        :param start: Revision id of top revision.
 
 
36
        :param start_revs: Revision ids of top revisions.
 
36
37
        :param maxnum: Maximum number of revisions to display, 
 
 
66
67
        gtk.accel_map_add_entry("<viz>/Go/Next Revision", gtk.keysyms.Up, gtk.gdk.MOD1_MASK)
 
67
68
        gtk.accel_map_add_entry("<viz>/Go/Previous Revision", gtk.keysyms.Down, gtk.gdk.MOD1_MASK)
 
 
69
        gtk.accel_map_add_entry("<viz>/View/Refresh", gtk.keysyms.F5, 0)
 
69
71
        self.accel_group = gtk.AccelGroup()
 
70
72
        self.add_accel_group(self.accel_group)
 
 
83
85
        self.next_rev_action.connect("activate", self._fwd_clicked_cb)
 
84
86
        self.next_rev_action.connect_accelerator()
 
 
88
        self.refresh_action = gtk.Action("refresh", "_Refresh", "Refresh view", gtk.STOCK_REFRESH)
 
 
89
        self.refresh_action.set_accel_path("<viz>/View/Refresh")
 
 
90
        self.refresh_action.set_accel_group(self.accel_group)
 
 
91
        self.refresh_action.connect("activate", self._refresh_clicked)
 
 
92
        self.refresh_action.connect_accelerator()
 
88
96
    def set_revision(self, revid):
 
 
101
109
        vbox.pack_start(self.construct_menubar(), expand=False, fill=True)
 
102
110
        vbox.pack_start(self.construct_navigation(), expand=False, fill=True)
 
103
 
        vbox.pack_start(self.construct_loading_msg(), expand=False, fill=True)
 
105
112
        vbox.pack_start(self.paned, expand=True, fill=True)
 
106
113
        vbox.set_focus_child(self.paned)
 
 
144
151
        view_menuitem = gtk.MenuItem("_View")
 
145
152
        view_menuitem.set_submenu(view_menu)
 
 
154
        view_menu_refresh = self.refresh_action.create_menu_item()
 
 
155
        view_menu_refresh.connect('activate', self._refresh_clicked)
 
 
157
        view_menu.add(view_menu_refresh)
 
 
158
        view_menu.add(gtk.SeparatorMenuItem())
 
147
160
        view_menu_toolbar = gtk.CheckMenuItem("Show Toolbar")
 
148
161
        view_menu_toolbar.set_active(True)
 
149
162
        view_menu_toolbar.connect('toggled', self._toolbar_visibility_changed)
 
 
156
169
        view_menu.add(view_menu_compact)
 
157
170
        view_menu.add(gtk.SeparatorMenuItem())
 
159
 
        for (label, name) in [("Revision _Number", "revno"), ("_Date", "date")]:
 
160
 
            col = gtk.CheckMenuItem("Show " + label + " Column")
 
 
172
        self.mnu_show_revno_column = gtk.CheckMenuItem("Show Revision _Number Column")
 
 
173
        self.mnu_show_date_column = gtk.CheckMenuItem("Show _Date Column")
 
 
175
        # Revision numbers are pointless if there are multiple branches
 
 
176
        if len(self.start_revs) > 1:
 
 
177
            self.mnu_show_revno_column.set_sensitive(False)
 
 
178
            self.treeview.set_property('revno-column-visible', False)
 
 
180
        for (col, name) in [(self.mnu_show_revno_column, "revno"), 
 
 
181
                            (self.mnu_show_date_column, "date")]:
 
161
182
            col.set_active(self.treeview.get_property(name + "-column-visible"))
 
162
183
            col.connect('toggled', self._col_visibility_changed, name)
 
163
184
            view_menu.add(col)
 
 
170
191
        go_menu_next = self.next_rev_action.create_menu_item()
 
171
192
        go_menu_prev = self.prev_rev_action.create_menu_item()
 
173
 
        self.go_menu_tags = gtk.MenuItem("_Tags")
 
 
194
        tag_image = gtk.Image()
 
 
195
        tag_image.set_from_file(icon_path("tag-16.png"))
 
 
196
        self.go_menu_tags = gtk.ImageMenuItem("_Tags")
 
 
197
        self.go_menu_tags.set_image(tag_image)
 
174
198
        self._update_tags()
 
176
200
        go_menu.add(go_menu_next)
 
 
185
209
        revision_menu_diff = gtk.MenuItem("View Changes")
 
186
210
        revision_menu_diff.connect('activate', 
 
187
 
                lambda w: self.treeview.show_diff())
 
 
213
        revision_menu_compare = gtk.MenuItem("Compare with...")
 
 
214
        revision_menu_compare.connect('activate',
 
 
215
                self._compare_with_cb)
 
189
217
        revision_menu_tag = gtk.MenuItem("Tag Revision")
 
190
218
        revision_menu_tag.connect('activate', self._tag_revision_cb)
 
192
220
        revision_menu.add(revision_menu_tag)
 
193
221
        revision_menu.add(revision_menu_diff)
 
 
222
        revision_menu.add(revision_menu_compare)
 
195
224
        branch_menu = gtk.Menu()
 
196
225
        branch_menuitem = gtk.MenuItem("_Branch")
 
 
218
247
        menubar.show_all()
 
222
 
    def construct_loading_msg(self):
 
223
 
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
 
224
 
                                                 gtk.ICON_SIZE_BUTTON)
 
227
 
        label_loading = gtk.Label(_("Please wait, loading ancestral graph..."))
 
228
 
        label_loading.set_alignment(0.0, 0.5)
 
231
 
        self.loading_msg_box = gtk.HBox()
 
232
 
        self.loading_msg_box.set_spacing(5)
 
233
 
        self.loading_msg_box.set_border_width(5)        
 
234
 
        self.loading_msg_box.pack_start(image_loading, False, False)
 
235
 
        self.loading_msg_box.pack_start(label_loading, True, True)
 
236
 
        self.loading_msg_box.show()
 
238
 
        return self.loading_msg_box
 
240
251
    def construct_top(self):
 
241
252
        """Construct the top-half of the window."""
 
242
253
        # FIXME: Make broken_line_length configurable
 
244
 
        self.treeview = TreeView(self.branch, self.start, self.maxnum, self.compact_view)
 
 
255
        self.treeview = TreeView(self.branch, self.start_revs, self.maxnum, self.compact_view)
 
246
257
        self.treeview.connect('revision-selected',
 
247
258
                self._treeselection_changed_cb)
 
249
 
        self.treeview.connect('revisions-loaded', 
 
250
 
                lambda x: self.loading_msg_box.hide())
 
 
259
        self.treeview.connect('revision-activated',
 
 
260
                self._tree_revision_activated)
 
252
262
        self.treeview.connect('tag-added', lambda w, t, r: self._update_tags())
 
 
307
319
        parents  = self.treeview.get_parents()
 
308
320
        children = self.treeview.get_children()
 
310
 
        if revision is not None:
 
 
322
        if revision and revision != NULL_REVISION:
 
311
323
            prev_menu = gtk.Menu()
 
312
324
            if len(parents) > 0:
 
313
325
                self.prev_rev_action.set_sensitive(True)
 
314
326
                for parent_id in parents:
 
315
 
                    parent = self.branch.repository.get_revision(parent_id)
 
317
 
                        str = ' (' + parent.properties['branch-nick'] + ')'
 
 
327
                    if parent_id and parent_id != NULL_REVISION:
 
 
328
                        parent = self.branch.repository.get_revision(parent_id)
 
 
330
                            str = ' (' + parent.properties['branch-nick'] + ')'
 
321
 
                    item = gtk.MenuItem(parent.message.split("\n")[0] + str)
 
322
 
                    item.connect('activate', self._set_revision_cb, parent_id)
 
 
334
                        item = gtk.MenuItem(parent.message.split("\n")[0] + str)
 
 
335
                        item.connect('activate', self._set_revision_cb, parent_id)
 
324
337
                prev_menu.show_all()
 
326
339
                self.prev_rev_action.set_sensitive(False)
 
 
351
364
            self.revisionview.set_revision(revision)
 
352
365
            self.revisionview.set_children(children)
 
 
367
    def _tree_revision_activated(self, widget, path, col):
 
 
368
        # TODO: more than one parent
 
 
369
        """Callback for when a treeview row gets activated."""
 
 
370
        revision = self.treeview.get_revision()
 
 
371
        parents  = self.treeview.get_parents()
 
 
373
        if len(parents) == 0:
 
 
376
            parent_id = parents[0]
 
 
378
        self.show_diff(revision.revision_id, parent_id)
 
 
379
        self.treeview.grab_focus()
 
 
381
    def _menu_diff_cb(self,w):
 
 
382
        (path, focus) = self.treeview.treeview.get_cursor()
 
 
383
        revid = self.treeview.model[path][treemodel.REVID]
 
 
385
        parentids = self.branch.repository.revision_parents(revid)
 
 
387
        if len(parentids) == 0:
 
 
388
            parentid = NULL_REVISION
 
 
390
            parentid = parentids[0]
 
 
392
        self.show_diff(revid,parentid)    
 
354
394
    def _back_clicked_cb(self, *args):
 
355
395
        """Callback for when the back button is clicked."""
 
 
367
407
    def _show_clicked_cb(self, revid, parentid):
 
368
408
        """Callback for when the show button for a parent is clicked."""
 
369
 
        self.treeview.show_diff(revid, parentid)
 
 
409
        self.show_diff(revid, parentid)
 
370
410
        self.treeview.grab_focus()
 
 
412
    def _compare_with_cb(self,w):
 
 
413
        """Callback for revision 'compare with' menu. Will show a small
 
 
414
            dialog with branch revisions to compare with selected revision in TreeView"""
 
 
416
        from bzrlib.plugins.gtk.revbrowser import RevisionBrowser
 
 
418
        rb = RevisionBrowser(self.branch,self)
 
 
421
        if ret == gtk.RESPONSE_OK:          
 
 
422
            (path, focus) = self.treeview.treeview.get_cursor()
 
 
423
            revid = self.treeview.model[path][treemodel.REVID]
 
 
424
            self.show_diff(revid, rb.selected_revid)
 
372
428
    def _set_revision_cb(self, w, revision_id):
 
373
429
        self.treeview.set_revision_id(revision_id)
 
 
433
489
            for tag, revid in tags:
 
434
 
                tag_item = gtk.MenuItem(tag)
 
 
490
                tag_image = gtk.Image()
 
 
491
                tag_image.set_from_file(icon_path('tag-16.png'))
 
 
492
                tag_item = gtk.ImageMenuItem(tag.replace('_', '__'))
 
 
493
                tag_item.set_image(tag_image)
 
435
494
                tag_item.connect('activate', self._tag_selected_cb, revid)
 
436
495
                menu.add(tag_item)
 
437
496
            self.go_menu_tags.set_submenu(menu)
 
 
443
502
        self.go_menu_tags.show_all()
 
 
504
    def show_diff(self, revid=None, parentid=None):
 
 
505
        """Open a new window to show a diff between the given revisions."""
 
 
506
        from bzrlib.plugins.gtk.diff import DiffWindow
 
 
507
        window = DiffWindow(parent=self)
 
 
510
            parentid = NULL_REVISION
 
 
512
        rev_tree    = self.branch.repository.revision_tree(revid)
 
 
513
        parent_tree = self.branch.repository.revision_tree(parentid)
 
 
515
        description = revid + " - " + self.branch.nick
 
 
516
        window.set_diff(description, rev_tree, parent_tree)