#!/usr/bin/env python3

"""Print smallest-detectable-delta and required-n guidance for benchmark cells."""

from __future__ import annotations

import argparse

import json

import sys

from pathlib import Path

from typing import Any

ROOT = Path(__file__).resolve().parents[1]

sys.path.insert(0, str(ROOT / "scripts"))

from oakbench.stats import required_n_for_relative_delta  # noqa: E402

def parse_args() -> argparse.Namespace:

    parser = argparse.ArgumentParser(description=__doc__)

    parser.add_argument("--noise-floor-pct", type=float, required=True)

    parser.add_argument("--target-delta-pct", type=float, required=True)

    parser.add_argument("--current-n", type=int, required=True)

    parser.add_argument("--json", action="store_true")

    return parser.parse_args()

def main() -> int:

    args = parse_args()

    required = required_n_for_relative_delta(args.noise_floor_pct, args.target_delta_pct, args.current_n)

    row: dict[str, Any] = {

        "noise_floor_pct": args.noise_floor_pct,

        "target_delta_pct": args.target_delta_pct,

        "current_n": args.current_n,

        "required_n": required,

        "undecidable": required is None,

    }

    if args.json:

        print(json.dumps(row, sort_keys=True))

    else:

        if required is None:

            print("Target delta must be positive; cell is undecidable.")

        elif required <= args.current_n:

            print(f"current n={args.current_n} can resolve target delta {args.target_delta_pct:.2f}%")

        else:

            print(

                f"current n={args.current_n} is underpowered for target delta "

                f"{args.target_delta_pct:.2f}%; estimated required n={required}"

            )

    return 0

if __name__ == "__main__":

    raise SystemExit(main())