mercurial-scm/hg-stable: hgext/fastannotate/context.py annotate

annotate hgext/fastannotate/context.py @ 43087:66f2cc210a29

py3: manually import pycompat.setattr where it is needed Continuing to eliminate the implicit import of symbols in the Python 3 source transformer so we can eliminate it. Differential Revision: https://phab.mercurial-scm.org/D7007

author	Gregory Szorc <gregory.szorc@gmail.com>
date	Sun, 06 Oct 2019 14:58:42 -0400
parents	eef9a2d67051
children	c59eb1560c44

rev	line source
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	1 # Copyright 2016-present Facebook. All Rights Reserved.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	2 #
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	3 # context: context needed to annotate a file
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	4 #
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	5 # This software may be used and distributed according to the terms of the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	6 # GNU General Public License version 2 or any later version.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	7
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	8 from __future__ import absolute_import
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	9
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	10 import collections
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	11 import contextlib
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	12 import hashlib
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	13 import os
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	14
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	15 from mercurial.i18n import _
43087 66f2cc210a29 py3: manually import pycompat.setattr where it is needed Gregory Szorc <gregory.szorc@gmail.com> parents: 43085 diff changeset	16 from mercurial.pycompat import (
66f2cc210a29 py3: manually import pycompat.setattr where it is needed Gregory Szorc <gregory.szorc@gmail.com> parents: 43085 diff changeset	17 open,
66f2cc210a29 py3: manually import pycompat.setattr where it is needed Gregory Szorc <gregory.szorc@gmail.com> parents: 43085 diff changeset	18 setattr,
66f2cc210a29 py3: manually import pycompat.setattr where it is needed Gregory Szorc <gregory.szorc@gmail.com> parents: 43085 diff changeset	19 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	20 from mercurial import (
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	21 error,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	22 linelog as linelogmod,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	23 lock as lockmod,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	24 mdiff,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	25 node,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	26 pycompat,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	27 scmutil,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	28 util,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	29 )
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	30 from mercurial.utils import stringutil
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	31
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	32 from . import (
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	33 error as faerror,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	34 revmap as revmapmod,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	35 )
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	36
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	37 # given path, get filelog, cached
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	38 @util.lrucachefunc
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	39 def _getflog(repo, path):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	40 return repo.file(path)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	41
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	42
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	43 # extracted from mercurial.context.basefilectx.annotate
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	44 def _parents(f, follow=True):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	45 # Cut _descendantrev here to mitigate the penalty of lazy linkrev
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	46 # adjustment. Otherwise, p._adjustlinkrev() would walk changelog
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	47 # from the topmost introrev (= srcrev) down to p.linkrev() if it
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	48 # isn't an ancestor of the srcrev.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	49 f._changeid
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	50 pl = f.parents()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	51
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	52 # Don't return renamed parents if we aren't following.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	53 if not follow:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	54 pl = [p for p in pl if p.path() == f.path()]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	55
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	56 # renamed filectx won't have a filelog yet, so set it
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	57 # from the cache to save time
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	58 for p in pl:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	59 if not b'_filelog' in p.__dict__:
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	60 p._filelog = _getflog(f._repo, p.path())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	61
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	62 return pl
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	63
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	64
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	65 # extracted from mercurial.context.basefilectx.annotate. slightly modified
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	66 # so it takes a fctx instead of a pair of text and fctx.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	67 def _decorate(fctx):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	68 text = fctx.data()
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	69 linecount = text.count(b'\n')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	70 if text and not text.endswith(b'\n'):
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	71 linecount += 1
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	72 return ([(fctx, i) for i in pycompat.xrange(linecount)], text)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	73
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	74
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	75 # extracted from mercurial.context.basefilectx.annotate. slightly modified
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	76 # so it takes an extra "blocks" parameter calculated elsewhere, instead of
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	77 # calculating diff here.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	78 def _pair(parent, child, blocks):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	79 for (a1, a2, b1, b2), t in blocks:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	80 # Changed blocks ('!') or blocks made only of blank lines ('~')
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	81 # belong to the child.
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	82 if t == b'=':
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	83 child[0][b1:b2] = parent[0][a1:a2]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	84 return child
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	85
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	86
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	87 # like scmutil.revsingle, but with lru cache, so their states (like manifests)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	88 # could be reused
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	89 _revsingle = util.lrucachefunc(scmutil.revsingle)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	90
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	91
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	92 def resolvefctx(repo, rev, path, resolverev=False, adjustctx=None):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	93 """(repo, str, str) -> fctx
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	94
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	95 get the filectx object from repo, rev, path, in an efficient way.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	96
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	97 if resolverev is True, "rev" is a revision specified by the revset
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	98 language, otherwise "rev" is a nodeid, or a revision number that can
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	99 be consumed by repo.__getitem__.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	100
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	101 if adjustctx is not None, the returned fctx will point to a changeset
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	102 that introduces the change (last modified the file). if adjustctx
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	103 is 'linkrev', trust the linkrev and do not adjust it. this is noticeably
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	104 faster for big repos but is incorrect for some cases.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	105 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	106 if resolverev and not isinstance(rev, int) and rev is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	107 ctx = _revsingle(repo, rev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	108 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	109 ctx = repo[rev]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	110
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	111 # If we don't need to adjust the linkrev, create the filectx using the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	112 # changectx instead of using ctx[path]. This means it already has the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	113 # changectx information, so blame -u will be able to look directly at the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	114 # commitctx object instead of having to resolve it by going through the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	115 # manifest. In a lazy-manifest world this can prevent us from downloading a
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	116 # lot of data.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	117 if adjustctx is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	118 # ctx.rev() is None means it's the working copy, which is a special
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	119 # case.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	120 if ctx.rev() is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	121 fctx = ctx[path]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	122 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	123 fctx = repo.filectx(path, changeid=ctx.rev())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	124 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	125 fctx = ctx[path]
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	126 if adjustctx == b'linkrev':
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	127 introrev = fctx.linkrev()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	128 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	129 introrev = fctx.introrev()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	130 if introrev != ctx.rev():
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	131 fctx._changeid = introrev
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	132 fctx._changectx = repo[introrev]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	133 return fctx
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	134
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	135
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	136 # like mercurial.store.encodedir, but use linelog suffixes: .m, .l, .lock
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	137 def encodedir(path):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	138 return (
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	139 path.replace(b'.hg/', b'.hg.hg/')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	140 .replace(b'.l/', b'.l.hg/')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	141 .replace(b'.m/', b'.m.hg/')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	142 .replace(b'.lock/', b'.lock.hg/')
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	143 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	144
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	145
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	146 def hashdiffopts(diffopts):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	147 diffoptstr = stringutil.pprint(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	148 sorted((k, getattr(diffopts, k)) for k in mdiff.diffopts.defaults)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	149 )
40714 9fcf8084ada8 py3: use node.hex(m.digest()) instead of m.hexdigest() Pulkit Goyal <pulkit@yandex-team.ru> parents: 40710 diff changeset	150 return node.hex(hashlib.sha1(diffoptstr).digest())[:6]
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	151
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	152
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	153 _defaultdiffopthash = hashdiffopts(mdiff.defaultopts)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	154
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	155
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	156 class annotateopts(object):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	157 """like mercurial.mdiff.diffopts, but is for annotate
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	158
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	159 followrename: follow renames, like "hg annotate -f"
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	160 followmerge: follow p2 of a merge changeset, otherwise p2 is ignored
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	161 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	162
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	163 defaults = {
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	164 b'diffopts': None,
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	165 b'followrename': True,
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	166 b'followmerge': True,
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	167 }
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	168
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	169 def __init__(self, **opts):
40710 d6a13f6db69c py3: use pycompat.byteskwargs() to covert keys of kwargs to bytes Pulkit Goyal <pulkit@yandex-team.ru> parents: 39744 diff changeset	170 opts = pycompat.byteskwargs(opts)
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	171 for k, v in self.defaults.iteritems():
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	172 setattr(self, k, opts.get(k, v))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	173
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	174 @util.propertycache
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	175 def shortstr(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	176 """represent opts in a short string, suitable for a directory name"""
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	177 result = b''
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	178 if not self.followrename:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	179 result += b'r0'
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	180 if not self.followmerge:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	181 result += b'm0'
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	182 if self.diffopts is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	183 assert isinstance(self.diffopts, mdiff.diffopts)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	184 diffopthash = hashdiffopts(self.diffopts)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	185 if diffopthash != _defaultdiffopthash:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	186 result += b'i' + diffopthash
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	187 return result or b'default'
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	188
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	189
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	190 defaultopts = annotateopts()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	191
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	192
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	193 class _annotatecontext(object):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	194 """do not use this class directly as it does not use lock to protect
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	195 writes. use "with annotatecontext(...)" instead.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	196 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	197
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	198 def __init__(self, repo, path, linelogpath, revmappath, opts):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	199 self.repo = repo
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	200 self.ui = repo.ui
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	201 self.path = path
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	202 self.opts = opts
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	203 self.linelogpath = linelogpath
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	204 self.revmappath = revmappath
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	205 self._linelog = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	206 self._revmap = None
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	207 self._node2path = {} # {str: str}
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	208
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	209 @property
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	210 def linelog(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	211 if self._linelog is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	212 if os.path.exists(self.linelogpath):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	213 with open(self.linelogpath, b'rb') as f:
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	214 try:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	215 self._linelog = linelogmod.linelog.fromdata(f.read())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	216 except linelogmod.LineLogError:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	217 self._linelog = linelogmod.linelog()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	218 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	219 self._linelog = linelogmod.linelog()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	220 return self._linelog
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	221
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	222 @property
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	223 def revmap(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	224 if self._revmap is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	225 self._revmap = revmapmod.revmap(self.revmappath)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	226 return self._revmap
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	227
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	228 def close(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	229 if self._revmap is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	230 self._revmap.flush()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	231 self._revmap = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	232 if self._linelog is not None:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	233 with open(self.linelogpath, b'wb') as f:
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	234 f.write(self._linelog.encode())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	235 self._linelog = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	236
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	237 __del__ = close
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	238
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	239 def rebuild(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	240 """delete linelog and revmap, useful for rebuilding"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	241 self.close()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	242 self._node2path.clear()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	243 _unlinkpaths([self.revmappath, self.linelogpath])
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	244
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	245 @property
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	246 def lastnode(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	247 """return last node in revmap, or None if revmap is empty"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	248 if self._revmap is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	249 # fast path, read revmap without loading its full content
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	250 return revmapmod.getlastnode(self.revmappath)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	251 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	252 return self._revmap.rev2hsh(self._revmap.maxrev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	253
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	254 def isuptodate(self, master, strict=True):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	255 """return True if the revmap / linelog is up-to-date, or the file
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	256 does not exist in the master revision. False otherwise.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	257
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	258 it tries to be fast and could return false negatives, because of the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	259 use of linkrev instead of introrev.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	260
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	261 useful for both server and client to decide whether to update
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	262 fastannotate cache or not.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	263
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	264 if strict is True, even if fctx exists in the revmap, but is not the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	265 last node, isuptodate will return False. it's good for performance - no
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	266 expensive check was done.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	267
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	268 if strict is False, if fctx exists in the revmap, this function may
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	269 return True. this is useful for the client to skip downloading the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	270 cache if the client's master is behind the server's.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	271 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	272 lastnode = self.lastnode
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	273 try:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	274 f = self._resolvefctx(master, resolverev=True)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	275 # choose linkrev instead of introrev as the check is meant to be
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	276 # fast.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	277 linknode = self.repo.changelog.node(f.linkrev())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	278 if not strict and lastnode and linknode != lastnode:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	279 # check if f.node() is in the revmap. note: this loads the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	280 # revmap and can be slow.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	281 return self.revmap.hsh2rev(linknode) is not None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	282 # avoid resolving old manifest, or slow adjustlinkrev to be fast,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	283 # false negatives are acceptable in this case.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	284 return linknode == lastnode
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	285 except LookupError:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	286 # master does not have the file, or the revmap is ahead
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	287 return True
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	288
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	289 def annotate(self, rev, master=None, showpath=False, showlines=False):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	290 """incrementally update the cache so it includes revisions in the main
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	291 branch till 'master'. and run annotate on 'rev', which may or may not be
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	292 included in the main branch.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	293
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	294 if master is None, do not update linelog.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	295
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	296 the first value returned is the annotate result, it is [(node, linenum)]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	297 by default. [(node, linenum, path)] if showpath is True.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	298
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	299 if showlines is True, a second value will be returned, it is a list of
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	300 corresponding line contents.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	301 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	302
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	303 # the fast path test requires commit hash, convert rev number to hash,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	304 # so it may hit the fast path. note: in the "fctx" mode, the "annotate"
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	305 # command could give us a revision number even if the user passes a
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	306 # commit hash.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	307 if isinstance(rev, int):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	308 rev = node.hex(self.repo.changelog.node(rev))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	309
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	310 # fast path: if rev is in the main branch already
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	311 directly, revfctx = self.canannotatedirectly(rev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	312 if directly:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	313 if self.ui.debugflag:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	314 self.ui.debug(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	315 b'fastannotate: %s: using fast path '
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	316 b'(resolved fctx: %s)\n'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	317 % (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	318 self.path,
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	319 stringutil.pprint(util.safehasattr(revfctx, b'node')),
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	320 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	321 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	322 return self.annotatedirectly(revfctx, showpath, showlines)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	323
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	324 # resolve master
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	325 masterfctx = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	326 if master:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	327 try:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	328 masterfctx = self._resolvefctx(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	329 master, resolverev=True, adjustctx=True
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	330 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	331 except LookupError: # master does not have the file
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	332 pass
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	333 else:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	334 if masterfctx in self.revmap: # no need to update linelog
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	335 masterfctx = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	336
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	337 # ... - @ <- rev (can be an arbitrary changeset,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	338 # / not necessarily a descendant
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	339 # master -> o of master)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	340 # \|
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	341 # a merge -> o 'o': new changesets in the main branch
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	342 # \|\ '#': revisions in the main branch that
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	343 # o * exist in linelog / revmap
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	344 # \| . '*': changesets in side branches, or
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	345 # last master -> # . descendants of master
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	346 # \| .
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	347 # # * joint: '#', and is a parent of a '*'
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	348 # \|/
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	349 # a joint -> # ^^^^ --- side branches
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	350 # \|
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	351 # ^ --- main branch (in linelog)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	352
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	353 # these DFSes are similar to the traditional annotate algorithm.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	354 # we cannot really reuse the code for perf reason.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	355
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	356 # 1st DFS calculates merges, joint points, and needed.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	357 # "needed" is a simple reference counting dict to free items in
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	358 # "hist", reducing its memory usage otherwise could be huge.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	359 initvisit = [revfctx]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	360 if masterfctx:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	361 if masterfctx.rev() is None:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	362 raise error.Abort(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	363 _(b'cannot update linelog to wdir()'),
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	364 hint=_(b'set fastannotate.mainbranch'),
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	365 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	366 initvisit.append(masterfctx)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	367 visit = initvisit[:]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	368 pcache = {}
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	369 needed = {revfctx: 1}
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	370 hist = {} # {fctx: ([(llrev or fctx, linenum)], text)}
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	371 while visit:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	372 f = visit.pop()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	373 if f in pcache or f in hist:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	374 continue
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	375 if f in self.revmap: # in the old main branch, it's a joint
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	376 llrev = self.revmap.hsh2rev(f.node())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	377 self.linelog.annotate(llrev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	378 result = self.linelog.annotateresult
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	379 hist[f] = (result, f.data())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	380 continue
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	381 pl = self._parentfunc(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	382 pcache[f] = pl
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	383 for p in pl:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	384 needed[p] = needed.get(p, 0) + 1
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	385 if p not in pcache:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	386 visit.append(p)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	387
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	388 # 2nd (simple) DFS calculates new changesets in the main branch
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	389 # ('o' nodes in # the above graph), so we know when to update linelog.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	390 newmainbranch = set()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	391 f = masterfctx
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	392 while f and f not in self.revmap:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	393 newmainbranch.add(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	394 pl = pcache[f]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	395 if pl:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	396 f = pl[0]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	397 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	398 f = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	399 break
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	400
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	401 # f, if present, is the position where the last build stopped at, and
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	402 # should be the "master" last time. check to see if we can continue
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	403 # building the linelog incrementally. (we cannot if diverged)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	404 if masterfctx is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	405 self._checklastmasterhead(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	406
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	407 if self.ui.debugflag:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	408 if newmainbranch:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	409 self.ui.debug(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	410 b'fastannotate: %s: %d new changesets in the main'
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	411 b' branch\n' % (self.path, len(newmainbranch))
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	412 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	413 elif not hist: # no joints, no updates
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	414 self.ui.debug(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	415 b'fastannotate: %s: linelog cannot help in '
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	416 b'annotating this revision\n' % self.path
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	417 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	418
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	419 # prepare annotateresult so we can update linelog incrementally
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	420 self.linelog.annotate(self.linelog.maxrev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	421
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	422 # 3rd DFS does the actual annotate
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	423 visit = initvisit[:]
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	424 progress = self.ui.makeprogress(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	425 b'building cache', total=len(newmainbranch)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	426 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	427 while visit:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	428 f = visit[-1]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	429 if f in hist:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	430 visit.pop()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	431 continue
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	432
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	433 ready = True
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	434 pl = pcache[f]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	435 for p in pl:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	436 if p not in hist:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	437 ready = False
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	438 visit.append(p)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	439 if not ready:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	440 continue
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	441
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	442 visit.pop()
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	443 blocks = None # mdiff blocks, used for appending linelog
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	444 ismainbranch = f in newmainbranch
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	445 # curr is the same as the traditional annotate algorithm,
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	446 # if we only care about linear history (do not follow merge),
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	447 # then curr is not actually used.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	448 assert f not in hist
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	449 curr = _decorate(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	450 for i, p in enumerate(pl):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	451 bs = list(self._diffblocks(hist[p][1], curr[1]))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	452 if i == 0 and ismainbranch:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	453 blocks = bs
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	454 curr = _pair(hist[p], curr, bs)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	455 if needed[p] == 1:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	456 del hist[p]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	457 del needed[p]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	458 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	459 needed[p] -= 1
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	460
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	461 hist[f] = curr
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	462 del pcache[f]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	463
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	464 if ismainbranch: # need to write to linelog
40857 dd1006874a5a fastannotate: use progress helper Martin von Zweigbergk <martinvonz@google.com> parents: 40714 diff changeset	465 progress.increment()
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	466 bannotated = None
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	467 if len(pl) == 2 and self.opts.followmerge: # merge
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	468 bannotated = curr[0]
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	469 if blocks is None: # no parents, add an empty one
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	470 blocks = list(self._diffblocks(b'', curr[1]))
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	471 self._appendrev(f, blocks, bannotated)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	472 elif showpath: # not append linelog, but we need to record path
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	473 self._node2path[f.node()] = f.path()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	474
40857 dd1006874a5a fastannotate: use progress helper Martin von Zweigbergk <martinvonz@google.com> parents: 40714 diff changeset	475 progress.complete()
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	476
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	477 result = [
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	478 ((self.revmap.rev2hsh(fr) if isinstance(fr, int) else fr.node()), l)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	479 for fr, l in hist[revfctx][0]
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	480 ] # [(node, linenumber)]
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	481 return self._refineannotateresult(result, revfctx, showpath, showlines)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	482
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	483 def canannotatedirectly(self, rev):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	484 """(str) -> bool, fctx or node.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	485 return (True, f) if we can annotate without updating the linelog, pass
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	486 f to annotatedirectly.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	487 return (False, f) if we need extra calculation. f is the fctx resolved
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	488 from rev.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	489 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	490 result = True
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	491 f = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	492 if not isinstance(rev, int) and rev is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	493 hsh = {20: bytes, 40: node.bin}.get(len(rev), lambda x: None)(rev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	494 if hsh is not None and (hsh, self.path) in self.revmap:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	495 f = hsh
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	496 if f is None:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	497 adjustctx = b'linkrev' if self._perfhack else True
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	498 f = self._resolvefctx(rev, adjustctx=adjustctx, resolverev=True)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	499 result = f in self.revmap
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	500 if not result and self._perfhack:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	501 # redo the resolution without perfhack - as we are going to
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	502 # do write operations, we need a correct fctx.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	503 f = self._resolvefctx(rev, adjustctx=True, resolverev=True)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	504 return result, f
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	505
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	506 def annotatealllines(self, rev, showpath=False, showlines=False):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	507 """(rev : str) -> [(node : str, linenum : int, path : str)]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	508
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	509 the result has the same format with annotate, but include all (including
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	510 deleted) lines up to rev. call this after calling annotate(rev, ...) for
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	511 better performance and accuracy.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	512 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	513 revfctx = self._resolvefctx(rev, resolverev=True, adjustctx=True)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	514
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	515 # find a chain from rev to anything in the mainbranch
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	516 if revfctx not in self.revmap:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	517 chain = [revfctx]
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	518 a = b''
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	519 while True:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	520 f = chain[-1]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	521 pl = self._parentfunc(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	522 if not pl:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	523 break
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	524 if pl[0] in self.revmap:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	525 a = pl[0].data()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	526 break
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	527 chain.append(pl[0])
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	528
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	529 # both self.linelog and self.revmap is backed by filesystem. now
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	530 # we want to modify them but do not want to write changes back to
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	531 # files. so we create in-memory objects and copy them. it's like
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	532 # a "fork".
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	533 linelog = linelogmod.linelog()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	534 linelog.copyfrom(self.linelog)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	535 linelog.annotate(linelog.maxrev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	536 revmap = revmapmod.revmap()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	537 revmap.copyfrom(self.revmap)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	538
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	539 for f in reversed(chain):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	540 b = f.data()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	541 blocks = list(self._diffblocks(a, b))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	542 self._doappendrev(linelog, revmap, f, blocks)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	543 a = b
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	544 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	545 # fastpath: use existing linelog, revmap as we don't write to them
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	546 linelog = self.linelog
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	547 revmap = self.revmap
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	548
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	549 lines = linelog.getalllines()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	550 hsh = revfctx.node()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	551 llrev = revmap.hsh2rev(hsh)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	552 result = [(revmap.rev2hsh(r), l) for r, l in lines if r <= llrev]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	553 # cannot use _refineannotateresult since we need custom logic for
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	554 # resolving line contents
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	555 if showpath:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	556 result = self._addpathtoresult(result, revmap)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	557 if showlines:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	558 linecontents = self._resolvelines(result, revmap, linelog)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	559 result = (result, linecontents)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	560 return result
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	561
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	562 def _resolvelines(self, annotateresult, revmap, linelog):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	563 """(annotateresult) -> [line]. designed for annotatealllines.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	564 this is probably the most inefficient code in the whole fastannotate
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	565 directory. but we have made a decision that the linelog does not
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	566 store line contents. so getting them requires random accesses to
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	567 the revlog data, since they can be many, it can be very slow.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	568 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	569 # [llrev]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	570 revs = [revmap.hsh2rev(l[0]) for l in annotateresult]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	571 result = [None] * len(annotateresult)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	572 # {(rev, linenum): [lineindex]}
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	573 key2idxs = collections.defaultdict(list)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	574 for i in pycompat.xrange(len(result)):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	575 key2idxs[(revs[i], annotateresult[i][1])].append(i)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	576 while key2idxs:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	577 # find an unresolved line and its linelog rev to annotate
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	578 hsh = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	579 try:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	580 for (rev, _linenum), idxs in key2idxs.iteritems():
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	581 if revmap.rev2flag(rev) & revmapmod.sidebranchflag:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	582 continue
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	583 hsh = annotateresult[idxs[0]][0]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	584 break
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	585 except StopIteration: # no more unresolved lines
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	586 return result
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	587 if hsh is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	588 # the remaining key2idxs are not in main branch, resolving them
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	589 # using the hard way...
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	590 revlines = {}
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	591 for (rev, linenum), idxs in key2idxs.iteritems():
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	592 if rev not in revlines:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	593 hsh = annotateresult[idxs[0]][0]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	594 if self.ui.debugflag:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	595 self.ui.debug(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	596 b'fastannotate: reading %s line #%d '
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	597 b'to resolve lines %r\n'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	598 % (node.short(hsh), linenum, idxs)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	599 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	600 fctx = self._resolvefctx(hsh, revmap.rev2path(rev))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	601 lines = mdiff.splitnewlines(fctx.data())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	602 revlines[rev] = lines
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	603 for idx in idxs:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	604 result[idx] = revlines[rev][linenum]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	605 assert all(x is not None for x in result)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	606 return result
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	607
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	608 # run the annotate and the lines should match to the file content
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	609 self.ui.debug(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	610 b'fastannotate: annotate %s to resolve lines\n'
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	611 % node.short(hsh)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	612 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	613 linelog.annotate(rev)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	614 fctx = self._resolvefctx(hsh, revmap.rev2path(rev))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	615 annotated = linelog.annotateresult
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	616 lines = mdiff.splitnewlines(fctx.data())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	617 if len(lines) != len(annotated):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	618 raise faerror.CorruptedFileError(b'unexpected annotated lines')
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	619 # resolve lines from the annotate result
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	620 for i, line in enumerate(lines):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	621 k = annotated[i]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	622 if k in key2idxs:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	623 for idx in key2idxs[k]:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	624 result[idx] = line
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	625 del key2idxs[k]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	626 return result
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	627
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	628 def annotatedirectly(self, f, showpath, showlines):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	629 """like annotate, but when we know that f is in linelog.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	630 f can be either a 20-char str (node) or a fctx. this is for perf - in
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	631 the best case, the user provides a node and we don't need to read the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	632 filelog or construct any filecontext.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	633 """
41261 1198c86beb73 fastannotate: fix isinstance checks to be against bytes instead of str Augie Fackler <augie@google.com> parents: 40857 diff changeset	634 if isinstance(f, bytes):
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	635 hsh = f
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	636 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	637 hsh = f.node()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	638 llrev = self.revmap.hsh2rev(hsh)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	639 if not llrev:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	640 raise faerror.CorruptedFileError(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	641 b'%s is not in revmap' % node.hex(hsh)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	642 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	643 if (self.revmap.rev2flag(llrev) & revmapmod.sidebranchflag) != 0:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	644 raise faerror.CorruptedFileError(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	645 b'%s is not in revmap mainbranch' % node.hex(hsh)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	646 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	647 self.linelog.annotate(llrev)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	648 result = [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	649 (self.revmap.rev2hsh(r), l) for r, l in self.linelog.annotateresult
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	650 ]
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	651 return self._refineannotateresult(result, f, showpath, showlines)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	652
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	653 def _refineannotateresult(self, result, f, showpath, showlines):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	654 """add the missing path or line contents, they can be expensive.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	655 f could be either node or fctx.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	656 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	657 if showpath:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	658 result = self._addpathtoresult(result)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	659 if showlines:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	660 if isinstance(f, bytes): # f: node or fctx
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	661 llrev = self.revmap.hsh2rev(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	662 fctx = self._resolvefctx(f, self.revmap.rev2path(llrev))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	663 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	664 fctx = f
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	665 lines = mdiff.splitnewlines(fctx.data())
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	666 if len(lines) != len(result): # linelog is probably corrupted
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	667 raise faerror.CorruptedFileError()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	668 result = (result, lines)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	669 return result
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	670
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	671 def _appendrev(self, fctx, blocks, bannotated=None):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	672 self._doappendrev(self.linelog, self.revmap, fctx, blocks, bannotated)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	673
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	674 def _diffblocks(self, a, b):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	675 return mdiff.allblocks(a, b, self.opts.diffopts)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	676
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	677 @staticmethod
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	678 def _doappendrev(linelog, revmap, fctx, blocks, bannotated=None):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	679 """append a revision to linelog and revmap"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	680
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	681 def getllrev(f):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	682 """(fctx) -> int"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	683 # f should not be a linelog revision
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	684 if isinstance(f, int):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	685 raise error.ProgrammingError(b'f should not be an int')
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	686 # f is a fctx, allocate linelog rev on demand
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	687 hsh = f.node()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	688 rev = revmap.hsh2rev(hsh)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	689 if rev is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	690 rev = revmap.append(hsh, sidebranch=True, path=f.path())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	691 return rev
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	692
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	693 # append sidebranch revisions to revmap
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	694 siderevs = []
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	695 siderevmap = {} # node: int
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	696 if bannotated is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	697 for (a1, a2, b1, b2), op in blocks:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	698 if op != b'=':
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	699 # f could be either linelong rev, or fctx.
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	700 siderevs += [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	701 f
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	702 for f, l in bannotated[b1:b2]
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	703 if not isinstance(f, int)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	704 ]
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	705 siderevs = set(siderevs)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	706 if fctx in siderevs: # mainnode must be appended seperately
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	707 siderevs.remove(fctx)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	708 for f in siderevs:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	709 siderevmap[f] = getllrev(f)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	710
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	711 # the changeset in the main branch, could be a merge
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	712 llrev = revmap.append(fctx.node(), path=fctx.path())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	713 siderevmap[fctx] = llrev
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	714
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	715 for (a1, a2, b1, b2), op in reversed(blocks):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	716 if op == b'=':
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	717 continue
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	718 if bannotated is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	719 linelog.replacelines(llrev, a1, a2, b1, b2)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	720 else:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	721 blines = [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	722 ((r if isinstance(r, int) else siderevmap[r]), l)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	723 for r, l in bannotated[b1:b2]
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	724 ]
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	725 linelog.replacelines_vec(llrev, a1, a2, blines)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	726
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	727 def _addpathtoresult(self, annotateresult, revmap=None):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	728 """(revmap, [(node, linenum)]) -> [(node, linenum, path)]"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	729 if revmap is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	730 revmap = self.revmap
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	731
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	732 def _getpath(nodeid):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	733 path = self._node2path.get(nodeid)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	734 if path is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	735 path = revmap.rev2path(revmap.hsh2rev(nodeid))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	736 self._node2path[nodeid] = path
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	737 return path
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	738
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	739 return [(n, l, _getpath(n)) for n, l in annotateresult]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	740
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	741 def _checklastmasterhead(self, fctx):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	742 """check if fctx is the master's head last time, raise if not"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	743 if fctx is None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	744 llrev = 0
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	745 else:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	746 llrev = self.revmap.hsh2rev(fctx.node())
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	747 if not llrev:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	748 raise faerror.CannotReuseError()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	749 if self.linelog.maxrev != llrev:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	750 raise faerror.CannotReuseError()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	751
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	752 @util.propertycache
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	753 def _parentfunc(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	754 """-> (fctx) -> [fctx]"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	755 followrename = self.opts.followrename
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	756 followmerge = self.opts.followmerge
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	757
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	758 def parents(f):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	759 pl = _parents(f, follow=followrename)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	760 if not followmerge:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	761 pl = pl[:1]
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	762 return pl
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	763
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	764 return parents
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	765
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	766 @util.propertycache
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	767 def _perfhack(self):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	768 return self.ui.configbool(b'fastannotate', b'perfhack')
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	769
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	770 def _resolvefctx(self, rev, path=None, **kwds):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	771 return resolvefctx(self.repo, rev, (path or self.path), **kwds)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	772
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	773
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	774 def _unlinkpaths(paths):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	775 """silent, best-effort unlink"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	776 for path in paths:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	777 try:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	778 util.unlink(path)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	779 except OSError:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	780 pass
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	781
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	782
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	783 class pathhelper(object):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	784 """helper for getting paths for lockfile, linelog and revmap"""
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	785
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	786 def __init__(self, repo, path, opts=defaultopts):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	787 # different options use different directories
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	788 self._vfspath = os.path.join(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	789 b'fastannotate', opts.shortstr, encodedir(path)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	790 )
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	791 self._repo = repo
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	792
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	793 @property
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	794 def dirname(self):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	795 return os.path.dirname(self._repo.vfs.join(self._vfspath))
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	796
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	797 @property
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	798 def linelogpath(self):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	799 return self._repo.vfs.join(self._vfspath + b'.l')
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	800
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	801 def lock(self):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	802 return lockmod.lock(self._repo.vfs, self._vfspath + b'.lock')
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	803
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	804 @property
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	805 def revmappath(self):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	806 return self._repo.vfs.join(self._vfspath + b'.m')
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	807
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	808
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	809 @contextlib.contextmanager
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	810 def annotatecontext(repo, path, opts=defaultopts, rebuild=False):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	811 """context needed to perform (fast) annotate on a file
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	812
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	813 an annotatecontext of a single file consists of two structures: the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	814 linelog and the revmap. this function takes care of locking. only 1
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	815 process is allowed to write that file's linelog and revmap at a time.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	816
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	817 when something goes wrong, this function will assume the linelog and the
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	818 revmap are in a bad state, and remove them from disk.
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	819
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	820 use this function in the following way:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	821
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	822 with annotatecontext(...) as actx:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	823 actx. ....
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	824 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	825 helper = pathhelper(repo, path, opts)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	826 util.makedirs(helper.dirname)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	827 revmappath = helper.revmappath
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	828 linelogpath = helper.linelogpath
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	829 actx = None
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	830 try:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	831 with helper.lock():
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	832 actx = _annotatecontext(repo, path, linelogpath, revmappath, opts)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	833 if rebuild:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	834 actx.rebuild()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	835 yield actx
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	836 except Exception:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	837 if actx is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	838 actx.rebuild()
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	839 repo.ui.debug(b'fastannotate: %s: cache broken and deleted\n' % path)
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	840 raise
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	841 finally:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	842 if actx is not None:
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	843 actx.close()
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	844
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42951 diff changeset	845
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	846 def fctxannotatecontext(fctx, follow=True, diffopts=None, rebuild=False):
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	847 """like annotatecontext but get the context from a fctx. convenient when
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	848 used in fctx.annotate
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	849 """
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	850 repo = fctx._repo
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	851 path = fctx._path
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	852 if repo.ui.configbool(b'fastannotate', b'forcefollow', True):
39238 1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	853 follow = True
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	854 aopts = annotateopts(diffopts=diffopts, followrename=follow)
1ddb296e0dee fastannotate: initial import from Facebook's hg-experimental Augie Fackler <augie@google.com> parents: diff changeset	855 return annotatecontext(repo, path, aopts, rebuild)

Mercurial > public > mercurial-scm > hg-stable

annotate hgext/fastannotate/context.py @ 43087:66f2cc210a29