#!/usr/bin/env python

import argparse
import sys
import os
from importlib import import_module
from timeit import default_timer
import random

_this_dir = os.path.dirname(__file__)
sys.path.append(os.path.join(_this_dir, ".."))
from common import do_benchmark


def parse_args_simple():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "mode", help="Mode to use for serialization, ie. object or bytes"
    )
    parser.add_argument("reuse", help="Currently ignored")
    parser.add_argument("compression", help="Valid values are none or packed")
    parser.add_argument("iters", help="Number of iterations to run for", type=int)
    parser.add_argument(
        "-I",
        "--includes",
        help="Directories to add to PYTHONPATH",
        default="/usr/local/include",
    )

    return parser.parse_args()


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "name",
        help="Name of the benchmark to run, eg. carsales",
        nargs="?",
        default="carsales",
    )
    parser.add_argument(
        "-c", "--compression", help="Specify the compression type", default=None
    )
    parser.add_argument(
        "-s", "--suffix", help="Choose the protocol type.", default="pycapnp"
    )
    parser.add_argument("-m", "--mode", help="Specify the mode", default="object")
    parser.add_argument(
        "-i",
        "--iters",
        help="Specify the number of iterations manually. By default, it will be looked up in preset table",
        default=10,
        type=int,
    )
    parser.add_argument(
        "-r",
        "--reuse",
        help="If this flag is passed, objects will be re-used",
        action="store_true",
    )
    parser.add_argument(
        "-I",
        "--includes",
        help="Directories to add to PYTHONPATH",
        default="/usr/local/include",
    )

    return parser.parse_args()


def run_test(name, mode, reuse, compression, iters, suffix, includes):
    tic = default_timer()

    name = name
    sys.path.append(includes)
    module = import_module(name + "_" + suffix)
    benchmark = module.Benchmark(compression=compression)

    do_benchmark(mode=mode, benchmark=benchmark, iters=iters, reuse=reuse)
    toc = default_timer()
    return toc - tic


def main():
    args = parse_args()
    run_test(**vars(args))


if __name__ == "__main__":
    main()