# revset.py - asv revset benchmarks

#

# Copyright 2016 Logilab SA <contact@logilab.fr>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

'''ASV revset benchmarks generated from contrib/base-revsets.txt

Each revset benchmark is parameterized with variants (first, last, sort, ...)

'''

from __future__ import absolute_import

import os

import string

import sys

from . import basedir, perfbench

def createrevsetbenchmark(baseset, variants=None):

    if variants is None:

        # Default variants

        variants = ["plain", "first", "last", "sort", "sort+first", "sort+last"]

    fname = "track_" + "_".join(

        "".join(

            [c if c in string.digits + string.letters else " " for c in baseset]

        ).split()

    )

    def wrap(fname, baseset):

        @perfbench(name=baseset, params=[("variant", variants)])

        def f(perf, variant):

            revset = baseset

            if variant != "plain":

                for var in variant.split("+"):

                    revset = "%s(%s)" % (var, revset)

            return perf("perfrevset", revset)

        f.__name__ = fname

        return f

    return wrap(fname, baseset)

def initializerevsetbenchmarks():

    mod = sys.modules[__name__]

    with open(os.path.join(basedir, 'contrib', 'base-revsets.txt'), 'rb') as fh:

        for line in fh:

            baseset = line.strip()

            if baseset and not baseset.startswith('#'):

                func = createrevsetbenchmark(baseset)

                setattr(mod, func.__name__, func)

initializerevsetbenchmarks()