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

  • Committer: Szilveszter Farkas (Phanatic)
  • Date: 2007-09-30 21:58:29 UTC
  • mfrom: (256.2.64 gtk)
  • Revision ID: szilveszter.farkas@gmail.com-20070930215829-ty0js2529sni7ziv
Merge the vizchanges/brokenlines branch from Gary.

Show diffs side-by-side

added added

removed removed

Lines of Context:
viz/treemodel.py
 
1
# -*- coding: UTF-8 -*-
 
2
"""Tree model.
 
3
 
 
4
"""
 
5
 
 
6
__copyright__ = "Copyright � 2005 Canonical Ltd."
 
7
__author__    = "Gary van der Merwe <garyvdm@gmail.com>"
 
8
 
 
9
 
 
10
import gtk
 
11
import gobject
 
12
import pango
 
13
 
 
14
from bzrlib.osutils import format_date
 
15
 
 
16
REVID = 0
 
17
NODE = 1
 
18
LINES = 2
 
19
LAST_LINES = 3
 
20
REVNO = 4
 
21
MESSAGE = 5
 
22
COMMITER = 6
 
23
TIMESTAMP = 7
 
24
REVISION = 8
 
25
PARENTS = 9
 
26
CHILDREN = 10
 
27
 
 
28
class TreeModel(gtk.GenericTreeModel):
 
29
 
 
30
    
 
31
    def __init__ (self, branch, line_graph_data):
 
32
        gtk.GenericTreeModel.__init__(self)
 
33
        self.revisions = {}
 
34
        self.branch = branch
 
35
        self.line_graph_data = line_graph_data
 
36
    
 
37
    def on_get_flags(self):
 
38
        return gtk.TREE_MODEL_LIST_ONLY
 
39
    
 
40
    def on_get_n_columns(self):
 
41
        return 11
 
42
    
 
43
    def on_get_column_type(self, index):
 
44
        if index == REVID: return gobject.TYPE_STRING
 
45
        if index == NODE: return gobject.TYPE_PYOBJECT
 
46
        if index == LINES: return gobject.TYPE_PYOBJECT
 
47
        if index == LAST_LINES: return gobject.TYPE_PYOBJECT
 
48
        if index == REVNO: return gobject.TYPE_STRING
 
49
        if index == MESSAGE: return gobject.TYPE_STRING
 
50
        if index == COMMITER: return gobject.TYPE_STRING
 
51
        if index == TIMESTAMP: return gobject.TYPE_STRING
 
52
        if index == REVISION: return gobject.TYPE_PYOBJECT
 
53
        if index == PARENTS: return gobject.TYPE_PYOBJECT
 
54
        if index == CHILDREN: return gobject.TYPE_PYOBJECT
 
55
        
 
56
    def on_get_iter(self, path):
 
57
        return path[0]
 
58
    
 
59
    def on_get_path(self, rowref):
 
60
        return rowref
 
61
    
 
62
    def on_get_value(self, rowref, column):
 
63
        (revid, node, lines, parents,
 
64
         children, revno_sequence) = self.line_graph_data[rowref]
 
65
        if column == REVID: return revid
 
66
        if column == NODE: return node
 
67
        if column == LINES: return lines
 
68
        if column == PARENTS: return parents
 
69
        if column == CHILDREN: return children
 
70
        if column == LAST_LINES:
 
71
            if rowref>0:
 
72
                return self.line_graph_data[rowref-1][2]
 
73
            return []
 
74
        if column == REVNO: return ".".join(["%d" % (revno)
 
75
                                      for revno in revno_sequence])
 
76
        
 
77
        if revid not in self.revisions:
 
78
            revision = self.branch.repository.get_revisions([revid])[0]
 
79
            self.revisions[revid] = revision
 
80
        else:
 
81
            revision = self.revisions[revid]
 
82
        
 
83
        if column == REVISION: return revision
 
84
        if column == MESSAGE: return revision.message.split("\n")[0]
 
85
        if column == COMMITER: return revision.committer
 
86
        if column == TIMESTAMP: return format_date(revision.timestamp,
 
87
                                                   revision.timezone)
 
88
    
 
89
    def on_iter_next(self, rowref):
 
90
        if rowref < len(self.line_graph_data) - 1:
 
91
            return rowref+1
 
92
        return None
 
93
    
 
94
    def on_iter_children(self, parent):
 
95
        if parent is None: return 0
 
96
        return None
 
97
    
 
98
    def on_iter_has_child(self, rowref):
 
99
        return False
 
100
    
 
101
    def on_iter_n_children(self, rowref):
 
102
        if rowref is None: return len(self.line_graph_data)
 
103
        return 0
 
104
    
 
105
    def on_iter_nth_child(self, parent, n):
 
106
        if parent is None: return n
 
107
        return None
 
108
    
 
109
    def on_iter_parent(self, child):
 
110
        return None