"""Benchmarks for the simpler-calling-convention changes.

Run with:
    python bench_call_conventions.py -o result.json
    python -m pyperf compare_to base.json new.json --table
"""
import datetime
import fractions

import pyperf
import numpy as np

N = 1000

obj_int = np.array([i + 10**30 for i in range(N)], dtype=object)
obj_int2 = np.array([(i + 7) * 3 + 10**30 for i in range(N)], dtype=object)
obj_complex = np.array([complex(i, -i) for i in range(N)], dtype=object)
obj_frac = np.array(
    [fractions.Fraction(3 * i + 1, 2) for i in range(N)], dtype=object
)

identity = np.frompyfunc(lambda x: x, 1, 1)
add2 = np.frompyfunc(lambda x, y: x, 2, 1)

dt_arr = np.array(["2020-05-17T12:00"] * N, dtype="M8[m]")
tzinfo = datetime.timezone(datetime.timedelta(hours=5, minutes=30))

mat = np.matrix([[1.0, 2.0], [3.0, 4.0]])


def bench_frompyfunc_1in(loops):
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        identity(obj_int)
    return pyperf.perf_counter() - t0


def bench_frompyfunc_2in(loops):
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        add2(obj_int, obj_int2)
    return pyperf.perf_counter() - t0


def bench_conjugate_object(loops):
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        np.conjugate(obj_complex)
    return pyperf.perf_counter() - t0


def bench_floor_object(loops):
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        np.floor(obj_frac)
    return pyperf.perf_counter() - t0


def bench_gcd_object(loops):
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        np.gcd(obj_int, obj_int2)
    return pyperf.perf_counter() - t0


def bench_datetime_as_string_tz(loops):
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        np.datetime_as_string(dt_arr, timezone=tzinfo)
    return pyperf.perf_counter() - t0


def bench_matrix_add(loops):
    m = mat
    t0 = pyperf.perf_counter()
    for _ in range(loops):
        m + m
    return pyperf.perf_counter() - t0


runner = pyperf.Runner()
runner.bench_time_func("frompyfunc_1in_1000", bench_frompyfunc_1in)
runner.bench_time_func("frompyfunc_2in_1000", bench_frompyfunc_2in)
runner.bench_time_func("conjugate_object_1000", bench_conjugate_object)
runner.bench_time_func("floor_object_1000", bench_floor_object)
runner.bench_time_func("gcd_object_1000", bench_gcd_object)
runner.bench_time_func("datetime_as_string_tz_1000", bench_datetime_as_string_tz)
runner.bench_time_func("matrix_add_2x2", bench_matrix_add)