Staging
v0.8.1
v0.8.1
https://foss.heptapod.net/mercurial/hgview
Tip revision: 210f60b137f1379816944ed0942e25eae443d441 authored by Julien Cristau on 21 June 2013, 11:47:55 UTC
Upload to unstable
Upload to unstable
Tip revision: 210f60b
hggraph.py
# Copyright (c) 2003-2012 LOGILAB S.A. (Paris, FRANCE).
# http://www.logilab.fr/ -- mailto:contact@logilab.fr
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free Software
# Foundation; either version 2 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with
# this program. If not, see <http://www.gnu.org/licenses/>.
"""helper functions and classes to ease hg revision graph building
Based on graphlog's algorithm, with inspiration stolen to TortoiseHg
revision grapher.
"""
import os
from cStringIO import StringIO
import difflib
from itertools import chain, count
from time import strftime, localtime
from functools import partial
from mercurial.node import nullrev
from mercurial import patch, util, match, error, hg
import hgviewlib.hgpatches # force apply patches to mercurial
from hgviewlib.hgpatches import mqsupport, phases, hiddenrevs
from hgviewlib.util import tounicode, isbfile, first_known_precursors, build_repo
from hgviewlib.config import HgConfig
DATE_FMT = '%F %R'
def diff(repo, ctx1, ctx2=None, files=None):
"""
Compute the diff of ``files`` between the 2 contexts ``ctx1`` and ``ctx2``.
:Note: context may be a changectx or a filectx.
* If ``ctx2`` is None, the parent of ``ctx1`` is used. If ``ctx1`` is a
file ctx, the parent is the first ancestor that contains modification
on the given file
* If ``files`` is None, return the diff for all files.
"""
if not getattr(ctx1, 'applied', True): #no diff vs ctx2 on unapplied patch
return ''.join(chain(ctx1.filectx(fname).data() for fname in files))
if ctx2 is None:
ctx2 = ctx1.p1()
if files is None:
matchfn = match.always(repo.root, repo.getcwd())
else:
matchfn = match.exact(repo.root, repo.getcwd(), files)
# try/except for the sake of hg compatibility (API changes between
# 1.0 and 1.1)
diffopts = patch.diffopts(repo.ui)
try:
out = StringIO()
patch.diff(repo, ctx2.node(), ctx1.node(), match=matchfn, fp=out,
opts=diffopts)
diffdata = out.getvalue()
except:
diffdata = '\n'.join(patch.diff(repo, ctx2.node(), ctx1.node(),
match=matchfn, opts=diffopts))
# XXX how to deal diff encodings?
try:
diffdata = unicode(diffdata, "utf-8")
except UnicodeError:
# XXX use a default encoding from config?
diffdata = unicode(diffdata, "iso-8859-15", 'ignore')
return diffdata
def __get_parents(repo, rev, branch=None):
"""
Return non-null parents of `rev`. If branch is given, only return
parents that belongs to names branch `branch` (beware that this is
much slower).
"""
if not branch:
if rev is None:
return [x.rev() for x in repo.changectx(None).parents() if x]
return [x for x in repo.changelog.parentrevs(rev) if x != nullrev]
if rev is None:
return [x.rev() for x in repo.changectx(None).parents() \
if x and repo.changectx(rev).branch() == branch]
return [x for x in repo.changelog.parentrevs(rev) \
if (x != nullrev and repo.changectx(rev).branch() == branch)]
def getlog(model, ctx, gnode):
if ctx.rev() is not None:
msg = tounicode(ctx.description())
if msg:
msg = msg.splitlines()[0]
else:
msg = "WORKING DIRECTORY (locally modified)"
return msg
def gettags(model, ctx, gnode):
if ctx.rev() is None:
return ""
mqtags = ['qbase', 'qtip', 'qparent']
tags = ctx.tags()
if model.hide_mq_tags:
tags = [t for t in tags if t not in mqtags]
return ",".join(tags)
def getdate(model, ctx, gnode):
if not ctx.date():
return ""
return strftime(DATE_FMT, localtime(int(ctx.date()[0])))
def ismerge(ctx):
"""
Return True if the changecontext ctx is a merge mode (should work
with hg 1.0 and 1.2)
"""
if ctx:
return len(ctx.parents()) == 2 and ctx.parents()[1]
return False
def _graph_iterator(repo, start_rev, stop_rev, reorder=False):
"""Iter through revisions from start_rev to stop_rev (included)
Handle Working directory as None.
"""
# check parameters
assert start_rev is None or start_rev >= stop_rev
assert start_rev < len(repo)
if start_rev is None:
yield start_rev
start_rev = len(repo.changelog) -1
target_revs = xrange(start_rev, stop_rev-1, -1)
phaserevs = None
if getattr(repo, '_phasecache', None) is not None:
phaserevs = repo._phasecache._phaserevs
elif getattr(repo, '_phaserev', None) is not None:
phaserevs = repo._phaserev
if not reorder or phaserevs is None:
# old hg
for curr_rev in target_revs:
yield curr_rev
else:
_cmp = lambda a, b: cmp(phases.public == phaserevs[a],
phases.public == phaserevs[b])
for r in sorted(target_revs, cmp=_cmp):
yield r
def revision_grapher(repo, start_rev=None, stop_rev=0, branch=None, follow=False,
show_hidden=False, reorder=False, closed=False,
show_obsolete=False):
"""incremental revision grapher
This generator function walks through the revision history from
revision start_rev to revision stop_rev (which must be less than
or equal to start_rev) and for each revision emits tuples with the
following elements:
- current revision
- column of the current node in the set of ongoing edges
- color of the node (?)
- lines; a list of (col, next_col, color) indicating the edges between
the current row and the next row
- parent revisions of current revision
If start_rev is -1, shown unapplied mq patches (if available)
If start_rev is None, started from working directory node
If follow is True, only generated the subtree from the start_rev head.
If branch is set, only generated the subtree for the given named branch.
"""
# follow is disabled when start_rev is None (== not revision specified)
follow = start_rev is not None and follow # disable follow is start_rev is None
if show_hidden and start_rev == None and hasattr(repo, 'mq'):
series = list(reversed(repo.mq.series))
for patchname in series:
if not repo.mq.isapplied(patchname):
yield (patchname, 0, 0, [(0, 0 ,0, False)], [])
# No uncommitted change
if start_rev is None and repo.status() == ([],)*7:
start_rev = len(repo.changelog) -1
assert start_rev is None or start_rev >= stop_rev
# all known revs for this line. This is used to compute column index
# it's combined with next_revs to compute how we must draw lines
revs = []
levels = [] # a rev -> level mapping.
# level are True for real relation (parent),
# False for weak one (obsolete)
rev_color = {}
free_color = count(0)
excluded = () if show_hidden else hiddenrevs(repo)
closedbranches = [tag for tag, node in repo.branchtags().items()
if repo.lookup(node) not in repo.branchheads(tag, closed=False)]
for curr_rev in _graph_iterator(repo, start_rev, stop_rev, not show_hidden and reorder):
# Compute revs and next_revs.
if curr_rev in excluded:
continue
if curr_rev not in revs: # rev not ancestor of already processed node
# shall we ignore this new heads ?
if (branch and repo[curr_rev].branch() != branch) or \
(follow and curr_rev != start_rev) or \
(not closed and repo[curr_rev].branch() in closedbranches):
continue
# we add this new head to know revision
revs.append(curr_rev)
levels.append(True)
rev_color[curr_rev] = curcolor = free_color.next()
else:
curcolor = rev_color[curr_rev]
# copy known revisions for this line
next_revs = revs[:]
next_levels = levels[:]
# Add parents to next_revs.
parents = [(p, True) for p in __get_parents(repo, curr_rev, branch)]
if show_obsolete:
ctx = repo[curr_rev]
for prec in first_known_precursors(ctx, excluded):
parents.append((prec.rev(), False))
parents_to_add = []
max_levels = dict(zip(next_revs, next_levels))
for idx, (parent, level) in enumerate(parents):
# could have been added by another children
if parent not in next_revs:
parents_to_add.append(parent)
if idx == 0: # first parent inherit the color
rev_color[parent] = curcolor
else: # second don't
rev_color[parent] = free_color.next()
max_levels[parent] = level or max_levels.get(parent, False)
# rev_index is also the column index
rev_index = next_revs.index(curr_rev)
# replace curr_rev by its parents.
next_revs[rev_index:rev_index + 1] = parents_to_add
next_levels = [max_levels[r] for r in next_revs]
lines = []
for i, rev in enumerate(revs):
if rev == curr_rev:
# one or more line to parents
targets = parents
else:
# single line to the same rev
targets = [(rev, levels[i])]
for trg, level in targets:
color = rev_color[trg]
lines.append((i, next_revs.index(trg), color, level))
yield (curr_rev, rev_index, curcolor, lines, parents)
revs = next_revs
levels = next_levels
def filelog_grapher(repo, path):
'''
Graph the ancestry of a single file (log). Deletions show
up as breaks in the graph.
'''
filerev = len(repo.file(path)) - 1
fctx = repo.filectx(path, fileid=filerev)
rev = fctx.rev()
flog = fctx.filelog()
heads = [repo.filectx(path, fileid=flog.rev(x)).rev() for x in flog.heads()]
assert rev in heads
heads.remove(rev)
revs = []
rev_color = {}
nextcolor = 0
_paths = {}
while rev >= 0:
# Compute revs and next_revs
if rev not in revs:
revs.append(rev)
rev_color[rev] = nextcolor ; nextcolor += 1
curcolor = rev_color[rev]
index = revs.index(rev)
next_revs = revs[:]
# Add parents to next_revs
fctx = repo.filectx(_paths.get(rev, path), changeid=rev)
for pfctx in fctx.parents():
_paths[pfctx.rev()] = pfctx.path()
parents = [pfctx.rev() for pfctx in fctx.parents()]# if f.path() == path]
parents_to_add = []
for parent in parents:
if parent not in next_revs:
parents_to_add.append(parent)
if len(parents) > 1:
rev_color[parent] = nextcolor ; nextcolor += 1
else:
rev_color[parent] = curcolor
parents_to_add.sort()
next_revs[index:index + 1] = parents_to_add
lines = []
for i, nrev in enumerate(revs):
if nrev in next_revs:
color = rev_color[nrev]
lines.append( (i, next_revs.index(nrev), color, True) )
elif nrev == rev:
for parent in parents:
color = rev_color[parent]
lines.append( (i, next_revs.index(parent), color, True) )
pcrevs = [pfc.rev() for pfc in fctx.parents()]
yield (fctx.rev(), index, curcolor, lines, pcrevs,
_paths.get(fctx.rev(), path))
revs = next_revs
if revs:
rev = max(revs)
else:
rev = -1
if heads and rev <= heads[-1]:
rev = heads.pop()
class GraphNode(object):
"""
Simple class to encapsulate e hg node in the revision graph. Does
nothing but declaring attributes.
"""
def __init__(self, rev, xposition, color, lines, parents, ncols=None,
extra=None):
self.rev = rev
self.x = xposition
self.color = color
if ncols is None:
ncols = len(lines)
self.cols = ncols
if not parents:
self.cols += 1 # root misses its own column
self.parents = parents
self.bottomlines = lines
self.toplines = []
self.extra = extra
class Graph(object):
"""
Graph object to ease hg repo navigation. The Graph object
instantiate a `revision_grapher` generator, and provide a `fill`
method to build the graph progressively.
"""
#@timeit
def __init__(self, repo, grapher, maxfilesize=100000):
self.maxfilesize = maxfilesize
self.repo = repo
self.maxlog = len(self.repo.changelog)
self.grapher = grapher
self.nodes = []
self.nodesdict = {}
self.max_cols = 0
def build_nodes(self, nnodes=None, rev=None):
"""
Build up to `nnodes` more nodes in our graph, or build as many
nodes required to reach `rev`.
If both rev and nnodes are set, build as many nodes as
required to reach rev plus nnodes more.
"""
if self.grapher is None:
return False
stopped = False
mcol = [self.max_cols]
for vnext in self.grapher:
if vnext is None:
continue
nrev, xpos, color, lines, parents = vnext[:5]
if isinstance(nrev, int) and nrev >= self.maxlog:
continue
gnode = GraphNode(nrev, xpos, color, lines, parents,
extra=vnext[5:])
if self.nodes:
gnode.toplines = self.nodes[-1].bottomlines
self.nodes.append(gnode)
self.nodesdict[nrev] = gnode
mcol.append(gnode.cols)
if rev is not None and nrev <= rev:
rev = None # we reached rev, switching to nnodes counter
if rev is None:
if nnodes is not None:
nnodes -= 1
if not nnodes:
break
else:
break
else:
self.grapher = None
stopped = True
self.max_cols = max(mcol)
return not stopped
def isfilled(self):
return self.grapher is None
def fill(self, step=100):
"""
Return a generator that fills the graph by bursts of `step`
more nodes at each iteration.
"""
while self.build_nodes(step):
yield len(self)
yield len(self)
def __getitem__(self, idx):
if isinstance(idx, slice):
# XXX TODO: ensure nodes are built
return self.nodes.__getitem__(idx)
if idx >= len(self.nodes):
# build as many graph nodes as required to answer the
# requested idx
self.build_nodes(idx)
if idx > len(self):
return self.nodes[-1]
return self.nodes[idx]
def __len__(self):
# len(graph) is the number of actually built graph nodes
return len(self.nodes)
def index(self, rev):
if len(self) == 0: # graph is empty, let's build some nodes
self.build_nodes(10)
if rev is not None and rev < self.nodes[-1].rev:
self.build_nodes(self.nodes[-1].rev - rev)
if rev in self.nodesdict:
return self.nodes.index(self.nodesdict[rev])
return -1
def fileflags(self, filename, rev, _cache={}):
"""
Return a couple of flags ('=', '+', '-' or '?') depending on the nature
of the diff for filename between rev and its parents.
"""
if rev not in _cache:
ctx = self.repo.changectx(rev)
_cache.clear()
_cache[rev] = (ctx,
[self.repo.status(p.node(), ctx.node())[:5]
for p in ctx.parents()])
ctx, allchanges = _cache[rev]
flags = []
for changes in allchanges:
# changes = modified, added, removed, deleted, unknown
for flag, lst in zip(["=", "+", "-", "-", "?"], changes):
if filename in lst:
if flag == "+":
renamed = ctx.filectx(filename).renamed()
if renamed:
flags.append(renamed)
break
flags.append(flag)
break
else:
flags.append('')
return flags
def fileflag(self, filename, rev):
"""
Return a flag (see fileflags) between rev and its first parent (may be
long)
"""
return self.fileflags(filename, rev)[0]
def filename(self, rev):
return self.nodesdict[rev].extra[0]
def filedata(self, filename, rev, mode='diff', flag=None):
"""XXX written under dubious encoding assumptions
The modification flag is computed using *fileflag* if ``flag`` is None.
"""
# XXX This really begins to be a dirty mess...
data = ""
if flag is None:
flag = self.fileflag(filename, rev)
ctx = self.repo.changectx(rev)
filesize = 0
try:
fctx = ctx.filectx(filename)
filesize = fctx.size() # compute size here to lookup data securely
except (LookupError, OSError):
fctx = None # may happen for renamed/removed files or mq patch ?
if isbfile(filename):
data = "[bfile]\n"
if fctx:
data = fctx.data()
data += "footprint: %s\n" % data
return "+", data
if flag not in ('-', '?'):
if fctx is None:# or fctx.node() is None:
return '', None
if self.maxfilesize >= 0 and filesize > self.maxfilesize:
try:
div = int(filesize).bit_length() // 10
sym = ('', 'K', 'M', 'G', 'T', 'E')[div] # more, really ???
val = int(filesize / (2 ** (div * 10)))
except AttributeError: # py<2.7
val = filesize
sym = ''
data = "File too big ! (~%i%so)" % (val, sym)
return flag, data
if flag == "+" or mode == 'file':
flag = '+'
# return the whole file
data = fctx.data()
if util.binary(data):
data = "binary file"
else: # tries to convert to unicode
data = tounicode(data)
elif flag == "=" or isinstance(mode, int):
flag = "="
if isinstance(mode, int):
parentctx = self.repo.changectx(mode)
else:
parentctx = self.repo[self._fileparent(fctx)]
data = diff(self.repo, ctx, parentctx, files=[filename])
# we assume that \n@@ marks the end of the diff header
data = ''.join(data.partition(os.linesep + '@@')[1:]).lstrip()
elif flag == '':
data = ''
else: # file renamed
oldname, node = flag
newdata = fctx.data().splitlines()
olddata = self.repo.filectx(oldname, fileid=node)
olddata = olddata.data().splitlines()
data = list(difflib.unified_diff(olddata, newdata, oldname,
filename))[2:]
if data:
flag = "="
else:
data = newdata
flag = "+"
data = u'\n'.join(tounicode(elt) for elt in data)
return flag, data
def _fileparent(self, fctx):
try:
return fctx.p1().rev()
except IndexError: # reach bottom
return -1
def fileparent(self, filename, rev):
return self._fileparent(self.repo[rev].filectx(filename))
class HgRepoListWalker(object):
"""
Graph object to ease hg repo revision tree drawing depending on user's
configurations.
"""
_allcolumns = ('ID', 'Branch', 'Log', 'Author', 'Date', 'Tags',)
_columns = ('ID', 'Branch', 'Log', 'Author', 'Date', 'Tags', 'Bookmarks')
_stretchs = {'Log': 1, }
_getcolumns = "getChangelogColumns"
def __init__(self, repo, branch='', fromhead=None, follow=False, closed=False,
parent=None, *args, **kwargs):
"""
repo is a hg repo instance
"""
#XXX col radius
self._datacache = {}
self._hasmq = False
self.mqueues = []
self.wd_revs = []
self.graph = None
self.rowcount = 0
self.repo = repo
self.show_hidden = False
self.load_config()
self.setRepo(repo, branch=branch, fromhead=fromhead, follow=follow, closed=closed)
def setRepo(self, repo=None, branch='', fromhead=None, follow=False, closed=False):
if repo is None:
repo = build_repo(self.repo.ui, self.repo.root)
self._hasmq = hasattr(self.repo, "mq")
if not getattr(repo, '__hgview__', False) and self._hasmq:
mqsupport.reposetup(repo.ui, repo)
oldrepo = self.repo
self.repo = repo
if oldrepo.root != repo.root:
self.load_config()
self._datacache = {}
try:
wdctxs = self.repo.changectx(None).parents()
except error.Abort:
# might occur if reloading during a mq operation (or
# whatever operation playing with hg history)
return
if self._hasmq:
self.mqueues = self.repo.mq.series[:]
self.wd_revs = [ctx.rev() for ctx in wdctxs]
self.wd_status = [self.repo.status(ctx.node(), None)[:4] for ctx in wdctxs]
self._user_colors = {}
self._branch_colors = {}
# precompute named branch color for stable value.
for branch_name in chain(['default', 'stable'], sorted(repo.branchtags().keys())):
self.namedbranch_color(branch_name)
grapher = revision_grapher(self.repo, start_rev=fromhead,
follow=follow, branch=branch,
show_hidden=self.show_hidden,
reorder=self.reorder_changesets,
closed=closed,
show_obsolete=self.show_obsolete)
self.graph = Graph(self.repo, grapher, self.max_file_size)
self.rowcount = 0
self.heads = [self.repo.changectx(x).rev() for x in self.repo.heads()]
self.ensureBuilt(row=self.fill_step)
def ensureBuilt(self, rev=None, row=None):
"""
Make sure rev data is available (graph element created).
"""
if self.graph.isfilled():
return
required = 0
buildrev = rev
n = len(self.graph)
if rev is not None:
if n and self.graph[-1].rev <= rev:
buildrev = None
else:
required = self.fill_step / 2
elif row is not None and row > (n - self.fill_step / 2):
required = row - n + self.fill_step
if required or buildrev:
self.graph.build_nodes(nnodes=required, rev=buildrev)
self.updateRowCount()
elif row and row > self.rowcount:
# asked row was already built, but views where not aware of this
self.updateRowCount()
elif rev is not None and rev <= self.graph[self.rowcount].rev:
# asked rev was already built, but views where not aware of this
self.updateRowCount()
def updateRowCount(self):
self.rowcount = None
#raise NotImplementedError
def rowCount(self, parent=None):
return self.rowcount
def columnCount(self, parent=None):
return len(self._columns)
def load_config(self):
cfg = HgConfig(self.repo.ui)
self._users, self._aliases = cfg.getUsers()
self.dot_radius = cfg.getDotRadius(default=8)
self.rowheight = cfg.getRowHeight()
self.fill_step = cfg.getFillingStep()
self.max_file_size = cfg.getMaxFileSize()
self.hide_mq_tags = cfg.getMQHideTags()
self.show_hidden = cfg.getShowHidden()
self.reorder_changesets = cfg.getNonPublicOnTop()
self.show_obsolete = cfg.getShowObsolete()
cols = getattr(cfg, self._getcolumns)()
if cols is not None:
validcols = [col for col in cols if col in self._allcolumns]
if len(validcols) != len(cols):
wrongcols = [col for col in cols if col not in self._allcolumns]
#XXX
#print "WARNING! %s are not valid column names. Check your configuration." % ','.join(wrongcols)
#print " reverting to default columns configuration"
elif 'Log' not in validcols or 'ID' not in validcols:
pass
#print "WARNING! 'Log' and 'ID' are mandatory. Check your configuration."
#print " reverting to default columns configuration"
else:
self._columns = tuple(validcols)
@staticmethod
def get_color(n, ignore=()):
return []
def user_color(self, user):
if user not in self._user_colors:
self._user_colors[user] = self.get_color(len(self._user_colors),
self._user_colors.values())
return self._user_colors[user]
def user_name(self, user):
return self._aliases.get(user, user)
def namedbranch_color(self, branch):
if branch not in self._branch_colors:
self._branch_colors[branch] = self.get_color(len(self._branch_colors))
return self._branch_colors[branch]
def col2x(self, col):
return (1.2*self.dot_radius + 0) * col + self.dot_radius/2 + 3
def rowFromRev(self, rev):
row = self.graph.index(rev)
if row == -1:
row = None
return row
def clear(self):
"""empty the list"""
self.graph = None
self._datacache = {}
self.notify_data_changed()
def notify_data_changed(self):
pass