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

  • Committer: Jelmer Vernooij
  • Date: 2006-08-21 13:40:08 UTC
  • Revision ID: jelmer@samba.org-20060821134008-444fdf4dce5ed555
Fix exception when double-clicking revision without parents.

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/diffwin.py
 
1
#!/usr/bin/python
 
2
# -*- coding: UTF-8 -*-
 
3
"""Difference window.
 
4
 
 
5
This module contains the code to manage the diff window which shows
 
6
the changes made between two revisions on a branch.
 
7
"""
 
8
 
 
9
__copyright__ = "Copyright © 2005 Canonical Ltd."
 
10
__author__    = "Scott James Remnant <scott@ubuntu.com>"
 
11
 
 
12
 
 
13
from cStringIO import StringIO
 
14
 
 
15
import gtk
 
16
import pango
 
17
 
 
18
try:
 
19
    import gtksourceview
 
20
    have_gtksourceview = True
 
21
except ImportError:
 
22
    have_gtksourceview = False
 
23
 
 
24
import bzrlib
 
25
if bzrlib.version_info < (0, 9):
 
26
    # function deprecated in 0.9
 
27
    from bzrlib.delta import compare_trees
 
28
 
 
29
from bzrlib.diff import show_diff_trees
 
30
from bzrlib.errors import NoSuchFile
 
31
 
 
32
 
 
33
class DiffWindow(gtk.Window):
 
34
    """Diff window.
 
35
 
 
36
    This object represents and manages a single window containing the
 
37
    differences between two revisions on a branch.
 
38
    """
 
39
 
 
40
    def __init__(self):
 
41
        gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
 
42
        self.set_border_width(0)
 
43
        self.set_title("bzrk diff")
 
44
 
 
45
        # Use two thirds of the screen by default
 
46
        screen = self.get_screen()
 
47
        monitor = screen.get_monitor_geometry(0)
 
48
        width = int(monitor.width * 0.66)
 
49
        height = int(monitor.height * 0.66)
 
50
        self.set_default_size(width, height)
 
51
 
 
52
        self.construct()
 
53
 
 
54
    def construct(self):
 
55
        """Construct the window contents."""
 
56
        hbox = gtk.HBox(spacing=6)
 
57
        hbox.set_border_width(0)
 
58
        self.add(hbox)
 
59
        hbox.show()
 
60
 
 
61
        scrollwin = gtk.ScrolledWindow()
 
62
        scrollwin.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
 
63
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
 
64
        hbox.pack_start(scrollwin, expand=False, fill=True)
 
65
        scrollwin.show()
 
66
 
 
67
        self.model = gtk.TreeStore(str, str)
 
68
        self.treeview = gtk.TreeView(self.model)
 
69
        self.treeview.set_headers_visible(False)
 
70
        self.treeview.set_search_column(1)
 
71
        self.treeview.connect("cursor-changed", self._treeview_cursor_cb)
 
72
        scrollwin.add(self.treeview)
 
73
        self.treeview.show()
 
74
 
 
75
        cell = gtk.CellRendererText()
 
76
        cell.set_property("width-chars", 20)
 
77
        column = gtk.TreeViewColumn()
 
78
        column.pack_start(cell, expand=True)
 
79
        column.add_attribute(cell, "text", 0)
 
80
        self.treeview.append_column(column)
 
81
 
 
82
 
 
83
        scrollwin = gtk.ScrolledWindow()
 
84
        scrollwin.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
 
85
        scrollwin.set_shadow_type(gtk.SHADOW_IN)
 
86
        hbox.pack_start(scrollwin, expand=True, fill=True)
 
87
        scrollwin.show()
 
88
 
 
89
        if have_gtksourceview:
 
90
            self.buffer = gtksourceview.SourceBuffer()
 
91
            slm = gtksourceview.SourceLanguagesManager()
 
92
            gsl = slm.get_language_from_mime_type("text/x-patch")
 
93
            self.buffer.set_language(gsl)
 
94
            self.buffer.set_highlight(True)
 
95
 
 
96
            sourceview = gtksourceview.SourceView(self.buffer)
 
97
        else:
 
98
            self.buffer = gtk.TextBuffer()
 
99
            sourceview = gtk.TextView(self.buffer)
 
100
 
 
101
        sourceview.set_editable(False)
 
102
        sourceview.modify_font(pango.FontDescription("Monospace"))
 
103
        scrollwin.add(sourceview)
 
104
        sourceview.show()
 
105
 
 
106
    def set_diff(self, description, rev_tree, parent_tree):
 
107
        """Set the differences showed by this window.
 
108
 
 
109
        Compares the two trees and populates the window with the
 
110
        differences.
 
111
        """
 
112
        self.rev_tree = rev_tree
 
113
        self.parent_tree = parent_tree
 
114
 
 
115
        self.model.clear()
 
116
        if bzrlib.version_info < (0, 9):
 
117
            delta = compare_trees(self.parent_tree, self.rev_tree)
 
118
        else:
 
119
            delta = self.rev_tree.changes_from(self.parent_tree)
 
120
 
 
121
        self.model.append(None, [ "Complete Diff", "" ])
 
122
 
 
123
        if len(delta.added):
 
124
            titer = self.model.append(None, [ "Added", None ])
 
125
            for path, id, kind in delta.added:
 
126
                self.model.append(titer, [ path, path ])
 
127
 
 
128
        if len(delta.removed):
 
129
            titer = self.model.append(None, [ "Removed", None ])
 
130
            for path, id, kind in delta.removed:
 
131
                self.model.append(titer, [ path, path ])
 
132
 
 
133
        if len(delta.renamed):
 
134
            titer = self.model.append(None, [ "Renamed", None ])
 
135
            for oldpath, newpath, id, kind, text_modified, meta_modified \
 
136
                    in delta.renamed:
 
137
                self.model.append(titer, [ oldpath, newpath ])
 
138
 
 
139
        if len(delta.modified):
 
140
            titer = self.model.append(None, [ "Modified", None ])
 
141
            for path, id, kind, text_modified, meta_modified in delta.modified:
 
142
                self.model.append(titer, [ path, path ])
 
143
 
 
144
        self.treeview.expand_all()
 
145
        self.set_title(description + " - bzrk diff")
 
146
 
 
147
    def set_file(self, file_path):
 
148
        tv_path = None
 
149
        for data in self.model:
 
150
            for child in data.iterchildren():
 
151
                if child[0] == file_path or child[1] == file_path:
 
152
                    tv_path = child.path
 
153
                    break
 
154
        if tv_path is None:
 
155
            raise NoSuchFile(file_path)
 
156
        self.treeview.set_cursor(tv_path)
 
157
        self.treeview.scroll_to_cell(tv_path)
 
158
 
 
159
    def _treeview_cursor_cb(self, *args):
 
160
        """Callback for when the treeview cursor changes."""
 
161
        (path, col) = self.treeview.get_cursor()
 
162
        specific_files = [ self.model[path][1] ]
 
163
        if specific_files == [ None ]:
 
164
            return
 
165
        elif specific_files == [ "" ]:
 
166
            specific_files = []
 
167
 
 
168
        s = StringIO()
 
169
        show_diff_trees(self.parent_tree, self.rev_tree, s, specific_files)
 
170
        self.buffer.set_text(s.getvalue())