"""Build/rebuild the nnU-Net raw dataset (Dataset210_NTUH_T1C_PL) from rows jsonl.

rows: {key, pimg, label} where label is an absolute path to a 0/1 nifti mask.
Rebuilds imagesTr/labelsTr as symlinks and writes dataset.json.

Usage: python scripts_nnu/01_nnu_prepare_dataset.py --rows <rows.jsonl>
"""
import argparse
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from src.common import load_jsonl
from nnu_common import make_raw_dataset, raw_ds


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("--rows", required=True, help="jsonl with {key, pimg, label}")
    args = ap.parse_args()
    rows = load_jsonl(args.rows)
    n = make_raw_dataset(rows)
    print(f"[nnu:prepare] {raw_ds()}: {n} cases", flush=True)


if __name__ == "__main__":
    main()