"""Run one baseline strategy on one scenario and print the run.

Prints the tool log (tool, arguments, reason and a short result), the final
report, and then the revealed ground truth.

Usage::

    .venv\\Scripts\\python.exe scripts\\run_baseline.py clean scripted_iso
    .venv\\Scripts\\python.exe scripts\\run_baseline.py clean naive_uniform -s 2
"""

from __future__ import annotations

import argparse
import json
from typing import Any

from beampilot.baselines import naive_uniform, scripted_iso
from beampilot.session import SCENARIOS, MeasurementSession, reveal_truth

STRATEGIES = {"naive_uniform": naive_uniform, "scripted_iso": scripted_iso}


def short_result(result: dict[str, Any]) -> str:
    """One-line summary of a tool result."""
    if "error" in result or "internal_error" in result:
        return json.dumps(result, ensure_ascii=False)
    if "measurement_id" in result and "flags" in result:
        flags = [name for name, value in result["flags"].items() if value]
        return (
            f"{result['measurement_id']}: d_x {result['d_x_um']['mean']} um, "
            f"d_y {result['d_y_um']['mean']} um, peak "
            f"{result['details']['peak_counts']}, flags {flags or 'none'}, "
            f"suggested {result['suggested_verdict']}"
        )
    if "status" in result and "x" in result:
        return (
            f"M²x {result['x']['m2']}, M²y {result['y']['m2']}, "
            f"z0 {result['x']['z0_mm']}/{result['y']['z0_mm']} mm, "
            f"coverage met {result['iso_coverage_met']}"
        )
    if "report" in result:
        return f"verdict {result['report']['verdict']}"
    return json.dumps(result, ensure_ascii=False)


def main() -> None:
    parser = argparse.ArgumentParser(description=__doc__.splitlines()[0])
    parser.add_argument("scenario", choices=sorted(SCENARIOS))
    parser.add_argument("strategy", choices=sorted(STRATEGIES))
    parser.add_argument("-s", "--seed", type=int, default=0)
    args = parser.parse_args()

    session = MeasurementSession(SCENARIOS[args.scenario](seed=args.seed))
    STRATEGIES[args.strategy](session)

    print(f"== Tool log: {args.strategy} on {args.scenario} (seed {args.seed})")
    for i, entry in enumerate(session.log, start=1):
        arguments = {
            k: v for k, v in entry["arguments"].items() if k not in ("reason",)
        }
        print(f"{i:3d}. {entry['tool']}({arguments})")
        print(f"     reason: {entry['reason']}")
        print(f"     result: {short_result(entry['result'])}")
    print("\n== Report")
    print(json.dumps(session.report, indent=2, ensure_ascii=False))
    print("\n== Revealed truth")
    print(json.dumps(reveal_truth(session), indent=2, ensure_ascii=False))


if __name__ == "__main__":
    main()