"""Второй уровень: даёт ли он что-нибудь на этой машине.

ЗАЧЕМ. Про JIT в 3.13–3.14 пишут много, и почти всё — пересказ анонсов. Здесь
единственный вопрос, на который можно ответить своими руками: включение второго
уровня на ТОЙ ЖЕ сборке меняет время или нет.

ПОЧЕМУ ЭТО СРАВНЕНИЕ ЗАКОННО, хотя общее правило замеров запрещает сравнивать
версии по времени. Правило запрещает сравнивать РАЗНЫЕ СБОРКИ: у них разный
компилятор и разные флаги. Здесь сборка одна и та же, и различается ровно одна
переменная окружения — `PYTHON_JIT`. Это и есть контролируемый опыт.

КАК ЗАПУСКАТЬ. Не дважды, а несколькими ЧЕРЕДУЮЩИМИСЯ парами: машина дрейфует,
и один и тот же режим от запуска к запуску различается на 8–17 %. Чередование
нужно, чтобы дрейф попал в оба режима одинаково.

    for i in 1 2 3; do
      python3.14 bench/adaptive/jit.py --tsv
      PYTHON_JIT=1 python3.14 bench/adaptive/jit.py --tsv
    done | sort

Сравнивать надо не средние, а МИНИМУМЫ по рангам: три минимума одного режима
против трёх минимумов другого, первый с первым, второй со вторым. Если знак
разницы одинаков во всех трёх позициях — это уже не дрейф.

ЧТО НЕЛЬЗЯ ИЗ ЭТОГО ВЫВОДИТЬ. Приговор JIT вообще. Здесь четыре коротких цикла
на одной машине; ни pyperformance, ни настоящей программы тут нет. Вывод, на
который эти числа дают право, ровно один: на таком коде и на этой машине
разницы либо нет, либо она не в пользу второго уровня.
"""

import sys
import timeit

REPEAT = 11


def jit_status() -> str:
    jit = getattr(sys, "_jit", None)
    if jit is None:
        return "второго уровня в этой сборке нет"
    parts = []
    for name in ("is_available", "is_enabled", "is_active"):
        fn = getattr(jit, name, None)
        if fn is not None:
            parts.append(f"{name}={fn()}")
    return ", ".join(parts)


WORKLOADS = {
    "счётный цикл (i*i)": (
        "def hot(n):\n    t = 0\n    for i in range(n):\n        t += i * i\n    return t\n",
        "hot(20000)",
    ),
    "цикл по списку с атрибутом": (
        "class P:\n"
        "    def __init__(self):\n"
        "        self.x = 1\n"
        "OBJ = [P() for _ in range(20000)]\n"
        "def hot(seq):\n"
        "    t = 0\n"
        "    for o in seq:\n"
        "        t += o.x\n"
        "    return t\n",
        "hot(OBJ)",
    ),
    "вызов функции в цикле": (
        "def inner(a):\n    return a + 1\n"
        "def hot(n):\n"
        "    t = 0\n"
        "    for i in range(n):\n"
        "        t = inner(t)\n"
        "    return t\n",
        "hot(20000)",
    ),
    "словарь в цикле": (
        "D = {i: i for i in range(1000)}\n"
        "def hot(n):\n"
        "    t = 0\n"
        "    for i in range(n):\n"
        "        t += D[i % 1000]\n"
        "    return t\n",
        "hot(20000)",
    ),
}


def main() -> None:
    tsv = "--tsv" in sys.argv
    mode = "on" if getattr(sys, "_jit", None) and sys._jit.is_enabled() else "off"

    if tsv:
        for label, (setup, stmt) in WORKLOADS.items():
            ns: dict = {}
            exec(setup, ns)
            times = timeit.repeat(stmt, globals=ns, repeat=REPEAT, number=50)
            print(f"{mode}\t{label}\t{min(times) / 50 * 1e6:.1f}")
        return

    print("PY", sys.version.split()[0])
    print("  JIT:", jit_status())
    print(f"  лучшее из {REPEAT} прогонов, микросекунды на один вызов hot()")
    print()

    for label, (setup, stmt) in WORKLOADS.items():
        ns: dict = {}
        exec(setup, ns)
        times = timeit.repeat(stmt, globals=ns, repeat=REPEAT, number=50)
        best = min(times) / 50 * 1e6
        spread = (max(times) - min(times)) / min(times) * 100
        print(f"  {label:<28} {best:8.1f} мкс   разброс по прогонам {spread:4.1f} %")


if __name__ == "__main__":
    main()