/b-gtk/fix-viz : contents of viz/treemodel.py at revision 290.1.1

: (revision 290.1.1)

To get this branch, use:

bzr branch
http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

# -*- coding: UTF-8 -*-
"""Tree model.

"""

__copyright__ = "Copyright © 2005 Canonical Ltd."
__author__    = "Gary van der Merwe <garyvdm@gmail.com>"


import gtk
import gobject
import pango

from bzrlib.osutils import format_date

REVID = 0
NODE = 1
LINES = 2
LAST_LINES = 3
REVNO = 4
MESSAGE = 5
COMMITER = 6
TIMESTAMP = 7
REVISION = 8
PARENTS = 9
CHILDREN = 10

class TreeModel(gtk.GenericTreeModel):

    
    def __init__ (self, branch, line_graph_data):
        gtk.GenericTreeModel.__init__(self)
        self.revisions = {}
        self.branch = branch
        self.line_graph_data = line_graph_data
    
    def on_get_flags(self):
        return gtk.TREE_MODEL_LIST_ONLY
    
    def on_get_n_columns(self):
        return 11
    
    def on_get_column_type(self, index):
        if index == REVID: return gobject.TYPE_STRING
        if index == NODE: return gobject.TYPE_PYOBJECT
        if index == LINES: return gobject.TYPE_PYOBJECT
        if index == LAST_LINES: return gobject.TYPE_PYOBJECT
        if index == REVNO: return gobject.TYPE_STRING
        if index == MESSAGE: return gobject.TYPE_STRING
        if index == COMMITER: return gobject.TYPE_STRING
        if index == TIMESTAMP: return gobject.TYPE_STRING
        if index == REVISION: return gobject.TYPE_PYOBJECT
        if index == PARENTS: return gobject.TYPE_PYOBJECT
        if index == CHILDREN: return gobject.TYPE_PYOBJECT
        
    def on_get_iter(self, path):
        return path[0]
    
    def on_get_path(self, rowref):
        return rowref
    
    def on_get_value(self, rowref, column):
        (revid, node, lines, parents,
         children, revno_sequence) = self.line_graph_data[rowref]
        if column == REVID: return revid
        if column == NODE: return node
        if column == LINES: return lines
        if column == PARENTS: return parents
        if column == CHILDREN: return children
        if column == LAST_LINES:
            if rowref>0:
                return self.line_graph_data[rowref-1][2]
            return []
        if column == REVNO: return ".".join(["%d" % (revno)
                                      for revno in revno_sequence])
        
        if revid not in self.revisions:
            revision = self.branch.repository.get_revisions([revid])[0]
            self.revisions[revid] = revision
        else:
            revision = self.revisions[revid]
        
        if column == REVISION: return revision
        if column == MESSAGE: return revision.message.split("\n")[0]
        if column == COMMITER: return revision.committer
        if column == TIMESTAMP: return format_date(revision.timestamp,
                                                   revision.timezone)
    
    def on_iter_next(self, rowref):
        if rowref < len(self.line_graph_data) - 1:
            return rowref+1
        return None
    
    def on_iter_children(self, parent):
        if parent is None: return 0
        return None
    
    def on_iter_has_child(self, rowref):
        return False
    
    def on_iter_n_children(self, rowref):
        if rowref is None: return len(self.line_graph_data)
        return 0
    
    def on_iter_nth_child(self, parent, n):
        if parent is None: return n
        return None
    
    def on_iter_parent(self, child):
        return None

256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	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
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	28	class TreeModel(gtk.GenericTreeModel):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	29
	30
	31	def __init__ (self, branch, line_graph_data):
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	32	gtk.GenericTreeModel.__init__(self)
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	33	self.revisions = {}
	34	self.branch = branch
	35	self.line_graph_data = line_graph_data
	36
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	37	def on_get_flags(self):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	38	return gtk.TREE_MODEL_LIST_ONLY
	39
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	40	def on_get_n_columns(self):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	41	return 11
	42
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	43	def on_get_column_type(self, index):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	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
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	56	def on_get_iter(self, path):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	57	return path[0]
	58
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	59	def on_get_path(self, rowref):
	60	return rowref
	61
	62	def on_get_value(self, rowref, column):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	63	(revid, node, lines, parents,
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	64	children, revno_sequence) = self.line_graph_data[rowref]
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	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:
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	71	if rowref>0:
	72	return self.line_graph_data[rowref-1][2]
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	73	return []
	74	if column == REVNO: return ".".join(["%d" % (revno)
	75	for revno in revno_sequence])
	76
256.2.41 by Gary van der Merwe Revert GTKTreeModel on_ref_node implementation.	77	if revid not in self.revisions:
	78	revision = self.branch.repository.get_revisions([revid])[0]
	79	self.revisions[revid] = revision
	80	else:
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	81	revision = self.revisions[revid]
256.2.41 by Gary van der Merwe Revert GTKTreeModel on_ref_node implementation.	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)
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	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):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	95	if parent is None: return 0
	96	return None
	97
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	98	def on_iter_has_child(self, rowref):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	99	return False
	100
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	101	def on_iter_n_children(self, rowref):
	102	if rowref is None: return len(self.line_graph_data)
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	103	return 0
	104
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	105	def on_iter_nth_child(self, parent, n):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	106	if parent is None: return n
	107	return None
	108
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	109	def on_iter_parent(self, child):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	110	return None