"""Сколько стоит каждый слой декоратора.

ПОЧЕМУ ЭТОТ ФАЙЛ ПОЯВИЛСЯ ПОЗЖЕ УРОКА. Числа обёрток были в уроке и в
визуализации `WrapperCost`, а скрипта, которым они получены, в репозитории не
было — то есть проверить их читатель не мог, хотя весь сайт обещает обратное.
Файл закрывает эту дыру: он и есть источник четырёх чисел.

ЧТО МЕРЯЕТСЯ. Одна и та же функция под нулём, одним, двумя и тремя слоями
обычной обёртки `*args, **kwargs` с `functools.wraps`. Каждый слой добавляет
вызов Python-функции и упаковку аргументов, поэтому рост должен быть линейным —
и в этом всё содержание замера.

ОБЩЕЕ ПРАВИЛО ЗАМЕРОВ: время между версиями не сравнивается вообще. Все
четыре формы измеряются одним процессом, и сравнивать их между собой можно
всегда.

ПОЧЕМУ ЗАМЕРЫ ЧЕРЕДУЮТСЯ, А НЕ ИДУТ ПОДРЯД. Первая версия мерила формы одну
за другой: сначала семь повторов голого вызова, потом семь повторов одного
слоя и так далее. На машине с переменной нагрузкой это даёт не шум, а СДВИГ:
просадка, случившаяся в окне одной формы, целиком достаётся ей, и кратность
между формами меняется от запуска к запуску. Измерено на четырёх запусках
подряд: отношение трёх слоёв к голому вызову вышло 5,9 / 7,1 / 7,5 / 9,2 —
полуторакратный разброс там, где меряется одно и то же.

Теперь формы чередуются: в каждом круге меряются все четыре, и минимум для
каждой берётся по кругам. Просадка круга бьёт по всем четырём сразу, поэтому
кратность её переживает, а минимум по кругам остаётся оценкой наименее
потревоженного выполнения.
"""
import functools
import sys
import timeit

N = 20_000
REPEAT = 100


def target(x):
    return x + 1


def wrap(fn):
    @functools.wraps(fn)
    def inner(*args, **kwargs):
        return fn(*args, **kwargs)

    return inner


one = wrap(target)
two = wrap(wrap(target))
three = wrap(wrap(wrap(target)))

ROWS = [
    ("без декораторов", "target(1)"),
    ("один слой", "one(1)"),
    ("два слоя", "two(1)"),
    ("три слоя", "three(1)"),
]


def measure(stmts):
    """Минимум по кругам для каждой формы; в каждом круге меряются все формы."""
    setup = "from __main__ import target, one, two, three"
    best = [float("inf")] * len(stmts)
    for _ in range(REPEAT):
        for i, stmt in enumerate(stmts):
            best[i] = min(best[i], timeit.timeit(stmt, setup=setup, number=N) / N)
    return best


print("PY", sys.version.split()[0], f"| лучшее из {REPEAT} чередующихся кругов по {N:,} вызовов".replace(",", " "))
values = [t * 1e9 for t in measure([stmt for _, stmt in ROWS])]
for (label, _), ns in zip(ROWS, values):
    print(f"    {label:<20} {ns:7.1f} нс")

base = values[0]
print("  прибавка на слой:")
for i in range(1, len(values)):
    print(f"    слой {i}: +{values[i] - values[i - 1]:6.1f} нс")
print(f"  доля работы самой функции при трёх слоях: {base / values[-1] * 100:.0f} %")
# Кратность печатается отдельной строкой, потому что именно она переносится на
# другую машину, а наносекунды — нет. На ней же стоит практическая задача урока
# («во сколько раз»), и сборочная проверка ищет это число здесь.
print(f"  три слоя дороже голого вызова в {values[-1] / base:.1f} раза")