#!/usr/bin/env python
from __future__ import annotations

import argparse
import json
import sys
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parents[1]
if str(PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(PROJECT_ROOT))

from dovla_cil.generation.tangent_chart_synthesis import (  # noqa: E402
    ChartSynthesisConfig,
    evaluate_chart_synthesis,
)


def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(
        description="Evaluate local chart synthesis for positive tangent support."
    )
    parser.add_argument("--targets", type=Path, required=True)
    parser.add_argument("--out", type=Path, required=True)
    parser.add_argument("--k-values", default="1,2,4,8,16")
    parser.add_argument("--thresholds", default="0.05,0.1,0.2,0.4")
    parser.add_argument("--obs-dim", type=int, default=96)
    parser.add_argument("--text-dim", type=int, default=64)
    parser.add_argument("--val-fraction", type=float, default=0.2)
    parser.add_argument("--seed", type=int, default=0)
    parser.add_argument("--neighbor-pool", type=int, default=64)
    parser.add_argument("--direct-count", type=int, default=12)
    parser.add_argument("--barycentric-windows", default="4,8,16,32")
    parser.add_argument("--barycentric-temperature", type=float, default=0.5)
    parser.add_argument("--utility-weight", type=float, default=0.0)
    parser.add_argument(
        "--order",
        choices=("direct_first", "means_first", "interleave"),
        default="direct_first",
    )
    parser.add_argument("--all-tasks", action="store_true")
    parser.add_argument("--no-groups", action="store_true", help="Drop per-group rows from output.")
    args = parser.parse_args(argv)

    payload = json.loads(args.targets.read_text())
    config = ChartSynthesisConfig(
        obs_dim=args.obs_dim,
        text_dim=args.text_dim,
        val_fraction=args.val_fraction,
        seed=args.seed,
        neighbor_pool=args.neighbor_pool,
        direct_count=args.direct_count,
        barycentric_windows=_parse_ints(args.barycentric_windows),
        barycentric_temperature=args.barycentric_temperature,
        utility_weight=args.utility_weight,
        order=args.order,
        same_task_only=not args.all_tasks,
    )
    report = evaluate_chart_synthesis(
        list(payload.get("targets", [])),
        config=config,
        k_values=_parse_ints(args.k_values),
        thresholds=_parse_floats(args.thresholds),
    )
    report["targets"] = str(args.targets)
    if args.no_groups:
        report.pop("groups", None)
    args.out.parent.mkdir(parents=True, exist_ok=True)
    args.out.write_text(json.dumps(report, indent=2) + "\n")
    summary = {key: value for key, value in report.items() if key != "groups"}
    print(json.dumps(summary, indent=2))
    print(f"Wrote {args.out}")
    return 0


def _parse_ints(value: str) -> tuple[int, ...]:
    items = tuple(int(item.strip()) for item in value.split(",") if item.strip())
    if not items or any(item <= 0 for item in items):
        raise ValueError("integer list must contain positive values")
    return items


def _parse_floats(value: str) -> tuple[float, ...]:
    items = tuple(float(item.strip()) for item in value.split(",") if item.strip())
    if not items or any(item < 0 for item in items):
        raise ValueError("float list must contain non-negative values")
    return items


if __name__ == "__main__":
    raise SystemExit(main())