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

: (revision 330.1.2)

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
import re

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, repository, line_graph_data):
        gtk.GenericTreeModel.__init__(self)
        self.revisions = {}
        self.repository = repository
        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.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 re.sub('<.*@.*>', '', 
                                             revision.committer).strip(' ')
        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
292.1.1 by Daniel Schierbeck Removed email address from committer column in the revision history window.	13	import re
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	14
	15	from bzrlib.osutils import format_date
	16
	17	REVID = 0
	18	NODE = 1
	19	LINES = 2
	20	LAST_LINES = 3
	21	REVNO = 4
	22	MESSAGE = 5
	23	COMMITER = 6
	24	TIMESTAMP = 7
	25	REVISION = 8
	26	PARENTS = 9
	27	CHILDREN = 10
	28
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	29	class TreeModel(gtk.GenericTreeModel):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	30
	31
328 by Jelmer Vernooij Use repository instead of branch in more places, to make it easier to support multiple branches in viz.	32	def __init__ (self, repository, line_graph_data):
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	33	gtk.GenericTreeModel.__init__(self)
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	34	self.revisions = {}
328 by Jelmer Vernooij Use repository instead of branch in more places, to make it easier to support multiple branches in viz.	35	self.repository = repository
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	36	self.line_graph_data = line_graph_data
	37
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	38	def on_get_flags(self):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	39	return gtk.TREE_MODEL_LIST_ONLY
	40
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	41	def on_get_n_columns(self):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	42	return 11
	43
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	44	def on_get_column_type(self, index):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	45	if index == REVID: return gobject.TYPE_STRING
	46	if index == NODE: return gobject.TYPE_PYOBJECT
	47	if index == LINES: return gobject.TYPE_PYOBJECT
	48	if index == LAST_LINES: return gobject.TYPE_PYOBJECT
	49	if index == REVNO: return gobject.TYPE_STRING
	50	if index == MESSAGE: return gobject.TYPE_STRING
	51	if index == COMMITER: return gobject.TYPE_STRING
	52	if index == TIMESTAMP: return gobject.TYPE_STRING
	53	if index == REVISION: return gobject.TYPE_PYOBJECT
	54	if index == PARENTS: return gobject.TYPE_PYOBJECT
	55	if index == CHILDREN: return gobject.TYPE_PYOBJECT
	56
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	57	def on_get_iter(self, path):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	58	return path[0]
	59
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	60	def on_get_path(self, rowref):
	61	return rowref
	62
	63	def on_get_value(self, rowref, column):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	64	(revid, node, lines, parents,
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	65	children, revno_sequence) = self.line_graph_data[rowref]
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	66	if column == REVID: return revid
	67	if column == NODE: return node
	68	if column == LINES: return lines
	69	if column == PARENTS: return parents
	70	if column == CHILDREN: return children
	71	if column == LAST_LINES:
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	72	if rowref>0:
	73	return self.line_graph_data[rowref-1][2]
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	74	return []
	75	if column == REVNO: return ".".join(["%d" % (revno)
	76	for revno in revno_sequence])
	77
256.2.41 by Gary van der Merwe Revert GTKTreeModel on_ref_node implementation.	78	if revid not in self.revisions:
328 by Jelmer Vernooij Use repository instead of branch in more places, to make it easier to support multiple branches in viz.	79	revision = self.repository.get_revisions([revid])[0]
256.2.41 by Gary van der Merwe Revert GTKTreeModel on_ref_node implementation.	80	self.revisions[revid] = revision
	81	else:
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	82	revision = self.revisions[revid]
256.2.41 by Gary van der Merwe Revert GTKTreeModel on_ref_node implementation.	83
	84	if column == REVISION: return revision
	85	if column == MESSAGE: return revision.message.split("\n")[0]
292.1.1 by Daniel Schierbeck Removed email address from committer column in the revision history window.	86	if column == COMMITER: return re.sub('<.@.>', '',
	87	revision.committer).strip(' ')
256.2.41 by Gary van der Merwe Revert GTKTreeModel on_ref_node implementation.	88	if column == TIMESTAMP: return format_date(revision.timestamp,
	89	revision.timezone)
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	90
	91	def on_iter_next(self, rowref):
	92	if rowref < len(self.line_graph_data) - 1:
	93	return rowref+1
	94	return None
	95
	96	def on_iter_children(self, parent):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	97	if parent is None: return 0
	98	return None
	99
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	100	def on_iter_has_child(self, rowref):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	101	return False
	102
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	103	def on_iter_n_children(self, rowref):
	104	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.	105	return 0
	106
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	107	def on_iter_nth_child(self, parent, n):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	108	if parent is None: return n
	109	return None
	110
256.2.36 by Gary van der Merwe Revert back to GenericTreeModel and implement on_ref_node	111	def on_iter_parent(self, child):
256.2.29 by Gary van der Merwe Implement a TreeModel that loads revisions incrementaly.	112	return None