"""Байткод двух форм приращения: где у потока есть шанс уступить GIL.

В сборке с GIL поток отдаёт блокировку только в точках проверки
(`_CHECK_PERIODIC` в Python/bytecodes.c): на RESUME при входе в функцию,
после CALL, на JUMP_BACKWARD в конце тела цикла. Скрипт печатает инструкции
между чтением атрибута и записью в него и отмечает, есть ли среди них такая
точка. CALL помечен как точка проверки потому, что вызов её содержит: у
функции на C проверка выполняется в хвосте CALL, у функции на Python — в
RESUME на входе в неё (для one() срабатывает именно эта, и в этот список она
не попадает: она уже в теле вызываемой функции).
"""
import dis

from common import env_line, header

CHECKPOINTS = {"CALL", "RESUME", "JUMP_BACKWARD"}


class Box:
    pass


def one():
    return 1


def plus_equals(c):
    c.n += 1


def call_between(c):
    c.n = c.n + one()


def window(fn):
    """Инструкции от первого LOAD_ATTR n до STORE_ATTR n включительно."""
    ins = [i for i in dis.get_instructions(fn)]
    start = next(k for k, i in enumerate(ins) if i.opname == "LOAD_ATTR" and i.argval == "n")
    end = next(k for k, i in enumerate(ins) if i.opname == "STORE_ATTR" and i.argval == "n")
    return ins[start:end + 1]


def main():
    print(env_line())
    for n, title, fn in [
        (1, "INCREMENT WITH NO CALL BETWEEN READ AND WRITE", plus_equals),
        (2, "INCREMENT WITH A CALL BETWEEN READ AND WRITE", call_between),
    ]:
        header(n, title)
        span = window(fn)
        for i in span:
            mark = "  <- checkpoint" if i.opname in CHECKPOINTS else ""
            arg = f" {i.argrepr}" if i.argrepr else ""
            print(f"  {i.opname}{arg}{mark}")
        hits = sum(1 for i in span if i.opname in CHECKPOINTS)
        print(f"checkpoints between the read and the write: {hits}")


if __name__ == "__main__":
    main()