47 lines
2.8 KiB
Python
47 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse, json
|
|
from pathlib import Path
|
|
from .pipeline import load_samples, run_stage, scan, select_samples
|
|
from .reports import generate_reports, read_json
|
|
|
|
DEFAULT_INPUT = Path("data/cadfs-sample/CADFS_test")
|
|
DEFAULT_OUTPUT = Path("cadfs_to_cdsl/output")
|
|
|
|
|
|
def _parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser(description="Convert CADFS FeatureScript to CDSL and validate against STEP")
|
|
commands = parser.add_subparsers(dest="command", required=True)
|
|
for name in ("scan", "convert", "rebuild", "compare", "report", "pipeline"):
|
|
command = commands.add_parser(name)
|
|
command.add_argument("--input", type=Path, default=DEFAULT_INPUT)
|
|
command.add_argument("--output", type=Path, default=DEFAULT_OUTPUT)
|
|
if name != "scan":
|
|
command.add_argument("--sample-id", action="append")
|
|
command.add_argument("--offset", type=int, default=0)
|
|
command.add_argument("--limit", type=int)
|
|
command.add_argument("--seed", type=int)
|
|
command.add_argument("--workers", type=int, default=1)
|
|
command.add_argument("--compare-mode", choices=("rp", "strict"), default="rp")
|
|
command.add_argument("--force", action="store_true")
|
|
command.add_argument("--timeout-seconds", type=float, default=30.0, help="per-model OCC timeout (default: 30)")
|
|
return parser
|
|
|
|
|
|
def main(argv: list[str] | None = None) -> int:
|
|
args = _parser().parse_args(argv); args.output.mkdir(parents=True, exist_ok=True)
|
|
if args.command == "scan":
|
|
records = scan(args.input, args.output); result = {"sample_count": len(records), "output": str(args.output / "dataset_index.json")}
|
|
elif args.command == "report":
|
|
manifest = args.output / "manifest.jsonl"; records = [json.loads(line) for line in manifest.read_text().splitlines() if line.strip()]
|
|
result = generate_reports(args.output, records)
|
|
else:
|
|
if not 1 <= args.workers <= 8: raise ValueError("--workers must be between 1 and 8")
|
|
samples = select_samples(load_samples(args.input, args.output), sample_ids=args.sample_id, offset=args.offset, limit=args.limit, seed=args.seed)
|
|
if args.timeout_seconds <= 0: raise ValueError("--timeout-seconds must be positive")
|
|
records = run_stage(args.command, samples, args.output, force=args.force, compare_mode=args.compare_mode, timeout_seconds=args.timeout_seconds, workers=args.workers)
|
|
counts: dict[str, int] = {}
|
|
for record in records: counts[record["status"]] = counts.get(record["status"], 0) + 1
|
|
result = {"sample_count": len(records), "statuses": counts, "summary": str(args.output / "summary.json")}
|
|
print(json.dumps(result, ensure_ascii=True, indent=2, sort_keys=True)); return 0
|