"""Один итератор на несколько потоков: «раздать работу» без очереди.

    it = iter(source)
    def worker():
        for x in it:
            out.append(x)

Четыре потока разбирают один итератор по миллиону элементов. Печатается:

  - выдано всего — сколько элементов получили потоки вместе;
  - повторов — сколько элементов выдано больше одного раза;
  - пропущено — сколько элементов источника не выдано ни разу;
  - вне источника — сколько выданных значений в источнике НЕТ вовсе;
    из них отдельно — за концом (x >= N) и до начала (x < 0).

Документация свободной сборки (howto/free-threading-python, раздел Iterators)
предупреждает о повторах и пропусках. Последнего столбца в ней нет; для
итератора range это гонка из cpython#129068, исправленная в 3.15.

Блок 3 — исправление: та же раздача через queue.Queue.
"""
import collections
import queue

from common import RUNS, THREADS, env_line, fmt_int, header, run_threads

N = 1_000_000


def trial(kind):
    if kind == "queue":
        q = queue.SimpleQueue()
        for x in range(N):
            q.put(x)
        for _ in range(THREADS):
            q.put(None)
    else:
        it = iter(range(N)) if kind == "range" else iter(list(range(N)))
    outs = [[] for _ in range(THREADS)]

    def worker(i):
        out = outs[i]
        if kind == "queue":
            while (x := q.get()) is not None:
                out.append(x)
        else:
            for x in it:
                out.append(x)

    run_threads(worker, args_for=lambda i: (i,))
    counts = collections.Counter(x for out in outs for x in out)
    given = sum(counts.values())
    dup = sum(c - 1 for c in counts.values() if c > 1)
    missing = sum(1 for x in range(N) if x not in counts)
    past_end = sum(1 for x in counts if x >= N)
    below = sum(1 for x in counts if x < 0)
    return given, dup, missing, past_end, below


def show(results):
    for given, dup, missing, past_end, below in results:
        print(
            f"handed out {fmt_int(given):>9}  duplicates {fmt_int(dup):>9}  "
            f"missing {fmt_int(missing):>7}  past the end {fmt_int(past_end):>7}  below zero {fmt_int(below):>3}"
        )


def main():
    print(env_line())
    print(f"threads {THREADS}, items {fmt_int(N)}, runs {RUNS}")

    header(1, "ONE RANGE ITERATOR, FOUR THREADS")
    show([trial("range") for _ in range(RUNS)])

    header(2, "ONE LIST ITERATOR, FOUR THREADS")
    show([trial("list") for _ in range(RUNS)])

    header(3, "FIX: A QUEUE INSTEAD OF A SHARED ITERATOR")
    show([trial("queue") for _ in range(RUNS)])


if __name__ == "__main__":
    main()