"""Во что обходится место вызова, на котором предположение не держится.

ЗАЧЕМ ЭТОТ ЗАМЕР. Специализацию легко показать через `dis`: инструкция была
одна, стала другая. Гораздо труднее показать, что она СТОИТ, — а без этого
рассказ про адаптивный интерпретатор остаётся описанием устройства, из которого
для читателя ничего не следует.

ЧЕСТНОЕ СРАВНЕНИЕ. Сравнивать «сложение целых» с «сложением строк» нельзя:
разная работа. Здесь сравнивается ОДНА И ТА ЖЕ работа — чтение атрибута `x` —
на местах вызова, различающихся только тем, сколько разных типов через них
проходит. Классы `A`, `B`, `C`, `D`, `E` объявлены одинаково и делают одно и то
же; отличается лишь то, сколько их видит одна строка кода.

  1. ЦЕНА ПОЛИМОРФИЗМА. Один тип, два типа вперемешку, пять типов вперемешку.
  2. СКОЛЬКО РАЗ ИНСТРУКЦИЯ МЕНЯЕТСЯ. Место вызова опрашивается на каждом шаге:
     сколько раз интерпретатор переписал инструкцию за прогон.
  3. ПОСЛЕДНЯЯ ИНСТРУКЦИЯ. Чем всё закончилось.

ЧЕГО ЗДЕСЬ НЕТ. Сравнения версий по времени. Строки одного прогона сравнимы
между собой, потому что меряются подряд в одном процессе; колонки разных
версий — нет.

    python3.13 bench/adaptive/cost.py
    python3.14 bench/adaptive/cost.py
"""

import dis
import sys
import timeit

REPEAT = 7
NUMBER = 200_000


def make_classes(n):
    """n одинаковых классов: работа одна и та же, типы разные."""
    out = []
    for i in range(n):
        ns = {}
        exec(f"class C{i}:\n    def __init__(self):\n        self.x = 1\n", ns)
        out.append(ns[f"C{i}"])
    return out


CLASSES = make_classes(5)


def make_reader():
    """Свежий объект кода: специализация живёт в нём, а не в имени.

    Чтение идёт внутри цикла, а не через вызов на каждый элемент: иначе цену
    самого вызова (несколько десятков наносекунд) пришлось бы вычитать, и
    разница между случаями утонула бы в ней. Здесь во всех случаях выполняется
    один и тот же цикл по списку одной и той же длины — различается только
    последовательность типов, проходящих через ОДНУ строку `o.x`.
    """
    ns = {}
    exec(
        "def read_all(objects):\n"
        "    total = 0\n"
        "    for o in objects:\n"
        "        total += o.x\n"
        "    return total\n",
        ns,
    )
    return ns["read_all"]


def opname(fn):
    for i in dis.get_instructions(fn, adaptive=True):
        if i.opname.startswith("LOAD_ATTR"):
            return i.opname
    return "-"


LENGTH = 5000


def measure(kinds: int, rounds: int = 60):
    """Читаем o.x на месте вызова, через которое проходит `kinds` типов."""
    read_all = make_reader()
    objects = [CLASSES[i % kinds]() for i in range(LENGTH)]

    changes = 0
    previous = opname(read_all)
    for _ in range(rounds):
        read_all(objects)
        now = opname(read_all)
        if now != previous:
            changes += 1
            previous = now

    times = timeit.repeat(
        "read_all(objects)",
        globals={"read_all": read_all, "objects": objects},
        repeat=REPEAT,
        number=NUMBER // LENGTH,
    )
    per_access = min(times) / (NUMBER // LENGTH * LENGTH) * 1e9
    return per_access, changes, previous


def main() -> None:
    print("PY", sys.version.split()[0])
    print(f"лучшее из {REPEAT} прогонов по {NUMBER:,} чтений атрибута в цикле".replace(",", " "))
    print()
    print(f"  {'типов на месте вызова':<24} {'нс/чтение':>10} {'переписей':>10}  инструкция в конце")

    base = None
    for kinds in (1, 2, 3, 5):
        ns, changes, last = measure(kinds)
        if base is None:
            base = ns
        print(f"  {kinds:<24} {ns:>10.2f} {changes:>10}  {last}  ×{ns / base:.2f}")

    print()
    print("Классы объявлены одинаково и делают одно и то же — различается только")
    print("число типов, проходящих через ОДНУ строку кода.")


if __name__ == "__main__":
    main()