/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 branchview/treemodel.py

  • Committer: Jelmer Vernooij
  • Date: 2011-11-02 11:11:06 UTC
  • mfrom: (734.1.55 gtk3)
  • Revision ID: jelmer@samba.org-20111102111106-7l0vso8eg24dpf87
Merge gtk3 support from Curtis.

Show diffs side-by-side

added added

removed removed

Lines of Context:
branchview/treemodel.py
1
1
# -*- coding: UTF-8 -*-
2
 
"""Tree model.
3
 
 
4
 
"""
 
2
"""BranchTreeModel."""
5
3
 
6
4
__copyright__ = "Copyright � 2005 Canonical Ltd."
7
 
__author__    = "Gary van der Merwe <garyvdm@gmail.com>"
 
5
__author__ = "Gary van der Merwe <garyvdm@gmail.com>"
8
6
 
9
7
 
10
8
from gi.repository import Gtk
19
17
    localtime,
20
18
    )
21
19
 
 
20
 
22
21
REVID = 0
23
22
NODE = 1
24
23
LINES = 2
34
33
TAGS = 12
35
34
AUTHORS = 13
36
35
 
37
 
class TreeModel(Gtk.TreeStore):
38
 
 
39
 
    def __init__ (self, branch, line_graph_data):
40
 
        Gtk.TreeStore.__init__(self)
 
36
 
 
37
class BranchTreeModel(Gtk.ListStore):
 
38
    """A model of branch's merge history."""
 
39
 
 
40
    def __init__(self, branch, line_graph_data):
 
41
        super(BranchTreeModel, self).__init__(
 
42
            GObject.TYPE_STRING,
 
43
            GObject.TYPE_PYOBJECT,
 
44
            GObject.TYPE_PYOBJECT,
 
45
            GObject.TYPE_PYOBJECT,
 
46
            GObject.TYPE_STRING,
 
47
            GObject.TYPE_STRING,
 
48
            GObject.TYPE_STRING,
 
49
            GObject.TYPE_STRING,
 
50
            GObject.TYPE_STRING,
 
51
            GObject.TYPE_PYOBJECT,
 
52
            GObject.TYPE_PYOBJECT,
 
53
            GObject.TYPE_PYOBJECT,
 
54
            GObject.TYPE_PYOBJECT,
 
55
            GObject.TYPE_STRING)
41
56
        self.revisions = {}
42
57
        self.branch = branch
43
58
        self.repository = branch.repository
44
 
        self.line_graph_data = line_graph_data
45
 
 
46
59
        if self.branch.supports_tags():
47
60
            self.tags = self.branch.tags.get_reverse_tag_dict()
48
61
        else:
49
62
            self.tags = {}
 
63
        self.set_line_graph_data(line_graph_data)
50
64
 
51
65
    def add_tag(self, tag, revid):
52
66
        self.branch.tags.set_tag(tag, revid)
55
69
        except KeyError:
56
70
            self.tags[revid] = [tag]
57
71
 
58
 
    def on_get_flags(self):
59
 
        return Gtk.TREE_MODEL_LIST_ONLY
60
 
 
61
 
    def on_get_n_columns(self):
62
 
        return 14
63
 
 
64
 
    def on_get_column_type(self, index):
65
 
        if index == REVID: return GObject.TYPE_STRING
66
 
        if index == NODE: return GObject.TYPE_PYOBJECT
67
 
        if index == LINES: return GObject.TYPE_PYOBJECT
68
 
        if index == LAST_LINES: return GObject.TYPE_PYOBJECT
69
 
        if index == REVNO: return GObject.TYPE_STRING
70
 
        if index == SUMMARY: return GObject.TYPE_STRING
71
 
        if index == MESSAGE: return GObject.TYPE_STRING
72
 
        if index == COMMITTER: return GObject.TYPE_STRING
73
 
        if index == TIMESTAMP: return GObject.TYPE_STRING
74
 
        if index == REVISION: return GObject.TYPE_PYOBJECT
75
 
        if index == PARENTS: return GObject.TYPE_PYOBJECT
76
 
        if index == CHILDREN: return GObject.TYPE_PYOBJECT
77
 
        if index == TAGS: return GObject.TYPE_PYOBJECT
78
 
        if index == AUTHORS: return GObject.TYPE_STRING
79
 
 
80
 
    def on_get_iter(self, path):
81
 
        # XXX sinzui 2011-08-12: maybe path.get_indices()[0]?
82
 
        return path[0]
83
 
 
84
 
    def on_get_path(self, rowref):
85
 
        return rowref
86
 
 
87
 
    def on_get_value(self, rowref, column):
88
 
        if len(self.line_graph_data) > 0:
89
 
            (revid, node, lines, parents,
90
 
             children, revno_sequence) = self.line_graph_data[rowref]
 
72
    def _line_graph_item_to_model_row(self, rowref, data):
 
73
        revid, node, lines, parents, children, revno_sequence = data
 
74
        if rowref > 0:
 
75
            last_lines = self.line_graph_data[rowref - 1][2]
91
76
        else:
92
 
            (revid, node, lines, parents,
93
 
             children, revno_sequence) = (None, (0, 0), (), (),
94
 
                                          (), ())
95
 
        if column == REVID: return revid
96
 
        if column == NODE: return node
97
 
        if column == LINES: return lines
98
 
        if column == PARENTS: return parents
99
 
        if column == CHILDREN: return children
100
 
        if column == LAST_LINES:
101
 
            if rowref>0:
102
 
                return self.line_graph_data[rowref-1][2]
103
 
            return []
104
 
        if column == REVNO: return ".".join(["%d" % (revno)
105
 
                                      for revno in revno_sequence])
106
 
 
107
 
        if column == TAGS: return self.tags.get(revid, [])
108
 
 
 
77
            last_lines = []
 
78
        revno = ".".join(["%d" % (revno) for revno in revno_sequence])
 
79
        tags = self.tags.get(revid, [])
109
80
        if not revid or revid == NULL_REVISION:
110
 
            return None
111
 
        if revid not in self.revisions:
 
81
            revision = None
 
82
        elif revid not in self.revisions:
112
83
            revision = self.repository.get_revisions([revid])[0]
113
84
            self.revisions[revid] = revision
114
85
        else:
115
86
            revision = self.revisions[revid]
116
 
 
117
 
        if column == REVISION: return revision
118
 
        if column == SUMMARY: return escape(revision.get_summary())
119
 
        if column == MESSAGE: return escape(revision.message)
120
 
        if column == COMMITTER: return parse_username(revision.committer)[0]
121
 
        if column == TIMESTAMP:
122
 
            return strftime("%Y-%m-%d %H:%M", localtime(revision.timestamp))
123
 
        if column == AUTHORS:
124
 
            return ", ".join([
125
 
                parse_username(author)[0] for author in revision.get_apparent_authors()])
126
 
 
127
 
    def on_iter_next(self, rowref):
128
 
        if rowref < len(self.line_graph_data) - 1:
129
 
            return rowref+1
130
 
        return None
131
 
 
132
 
    def on_iter_children(self, parent):
133
 
        if parent is None: return 0
134
 
        return None
135
 
 
136
 
    def on_iter_has_child(self, rowref):
137
 
        return False
138
 
 
139
 
    def on_iter_n_children(self, rowref):
140
 
        if rowref is None: return len(self.line_graph_data)
141
 
        return 0
142
 
 
143
 
    def on_iter_nth_child(self, parent, n):
144
 
        if parent is None: return n
145
 
        return None
146
 
 
147
 
    def on_iter_parent(self, child):
148
 
        return None
 
87
        if revision is None:
 
88
            summary = message = committer = timestamp = authors = None
 
89
        else:
 
90
            summary = escape(revision.get_summary())
 
91
            message = escape(revision.message)
 
92
            committer = parse_username(revision.committer)[0]
 
93
            timestamp = strftime(
 
94
                "%Y-%m-%d %H:%M", localtime(revision.timestamp))
 
95
            authors = ", ".join([
 
96
                parse_username(author)[0]
 
97
                for author in revision.get_apparent_authors()])
 
98
        return (revid, node, lines, last_lines, revno, summary, message,
 
99
                committer, timestamp, revision, parents, children, tags,
 
100
                authors)
 
101
 
 
102
    def set_line_graph_data(self, line_graph_data):
 
103
        self.clear()
 
104
        self.line_graph_data = line_graph_data
 
105
        for rowref, data in enumerate(self.line_graph_data):
 
106
            row = self._line_graph_item_to_model_row(rowref, data)
 
107
            self.append(row)