comparison contrib/perf.py @ 40214:b456b2e0ad9f

py3: make test-contrib-perf.t work on python 3

Differential Revision: https://phab.mercurial-scm.org/D5025

40213:83912321bf09

    from mercurial import pycompat
    getargspec = pycompat.getargspec  # added to module after 4.5
    _byteskwargs = pycompat.byteskwargs  # since 4.1 (or fbc3f73dc802)
    _sysstr = pycompat.sysstr         # since 4.0 (or 2219f4f82ede)
    _xrange = pycompat.xrange         # since 4.8 (or 7eba8f83129b)
    if pycompat.ispy3:
        _maxint = sys.maxsize  # per py3 docs for replacing maxint
    else:
        _maxint = sys.maxint
except (ImportError, AttributeError):
    import inspect
    getargspec = inspect.getargspec
    _byteskwargs = identity
    _maxint = sys.maxint              # no py3 support
    _sysstr = lambda x: x             # no py3 support
    _xrange = xrange

try:

@command(b'perfstartup', formatteropts)
def perfstartup(ui, repo, **opts):
    opts = _byteskwargs(opts)
    timer, fm = gettimer(ui, opts)
    cmd = sys.argv[0]
    def d():
        if os.name != r'nt':
            os.system(b"HGRCPATH= %s version -q > /dev/null" % cmd)
        else:
            os.environ[r'HGRCPATH'] = r' '

def perfdiffwd(ui, repo, **opts):
    """Profile diff of working directory changes"""
    opts = _byteskwargs(opts)
    timer, fm = gettimer(ui, opts)
    options = {
        b'w': b'ignore_all_space',
        b'b': b'ignore_space_change',
        b'B': b'ignore_blank_lines',
        }

    for diffopt in (b'', b'w', b'b', b'B', b'wB'):
        opts = dict((options[c], b'1') for c in diffopt)
        def d():
            ui.pushbuffer()
            commands.diff(ui, repo, **opts)
            ui.popbuffer()
        title = b'diffopts: %s' % (diffopt and (b'-' + diffopt) or b'none')
        timer(d, title)
    fm.end()

@command(b'perfrevlogindex', revlogopts + formatteropts,

    chain = r._deltachain(rev)[0]
    data = segmentforrevs(chain[0], chain[-1])[1]
    rawchunks = getrawchunks(data, chain)
    bins = r._chunks(chain)
    text = str(bins[0])
    bins = bins[1:]
    text = mdiff.patches(text, bins)

    benches = [
        (lambda: dorevision(), b'full'),

40214:b456b2e0ad9f

    from mercurial import pycompat
    getargspec = pycompat.getargspec  # added to module after 4.5
    _byteskwargs = pycompat.byteskwargs  # since 4.1 (or fbc3f73dc802)
    _sysstr = pycompat.sysstr         # since 4.0 (or 2219f4f82ede)
    _xrange = pycompat.xrange         # since 4.8 (or 7eba8f83129b)
    fsencode = pycompat.fsencode      # since 3.9 (or f4a5e0e86a7e)
    if pycompat.ispy3:
        _maxint = sys.maxsize  # per py3 docs for replacing maxint
    else:
        _maxint = sys.maxint
except (ImportError, AttributeError):
    import inspect
    getargspec = inspect.getargspec
    _byteskwargs = identity
    fsencode = identity               # no py3 support
    _maxint = sys.maxint              # no py3 support
    _sysstr = lambda x: x             # no py3 support
    _xrange = xrange

try:

@command(b'perfstartup', formatteropts)
def perfstartup(ui, repo, **opts):
    opts = _byteskwargs(opts)
    timer, fm = gettimer(ui, opts)
    cmd = fsencode(sys.argv[0])
    def d():
        if os.name != r'nt':
            os.system(b"HGRCPATH= %s version -q > /dev/null" % cmd)
        else:
            os.environ[r'HGRCPATH'] = r' '

def perfdiffwd(ui, repo, **opts):
    """Profile diff of working directory changes"""
    opts = _byteskwargs(opts)
    timer, fm = gettimer(ui, opts)
    options = {
        'w': 'ignore_all_space',
        'b': 'ignore_space_change',
        'B': 'ignore_blank_lines',
        }

    for diffopt in ('', 'w', 'b', 'B', 'wB'):
        opts = dict((options[c], b'1') for c in diffopt)
        def d():
            ui.pushbuffer()
            commands.diff(ui, repo, **opts)
            ui.popbuffer()
        diffopt = diffopt.encode('ascii')
        title = b'diffopts: %s' % (diffopt and (b'-' + diffopt) or b'none')
        timer(d, title)
    fm.end()

@command(b'perfrevlogindex', revlogopts + formatteropts,

    chain = r._deltachain(rev)[0]
    data = segmentforrevs(chain[0], chain[-1])[1]
    rawchunks = getrawchunks(data, chain)
    bins = r._chunks(chain)
    text = bytes(bins[0])
    bins = bins[1:]
    text = mdiff.patches(text, bins)

    benches = [
        (lambda: dorevision(), b'full'),