@brainpilot/skills 0.0.6 → 0.0.7

package/skills/06_fMRI_Neuroimaging/hbn-skill/SKILL.md

@@ -0,0 +1,271 @@
+---
+name: hbn-skill
+description: "Use this skill whenever the user wants an end-to-end workflow for the Healthy Brain Network (HBN) dataset, including download, BIDS organization, and multimodal processing of sMRI, dMRI, rs-fMRI, task-fMRI, and EEG data. Triggers include: 'HBN', 'Healthy Brain Network', 'process HBN', 'HBN fMRI', 'HBN EEG', or any request to run the HBN multimodal pipeline. This is the NeuroClaw dataset-orchestration layer for HBN."
+license: MIT License (NeuroClaw custom skill - freely modifiable within the project)
+layer: subagent
+skill_type: dataset
+dependencies:
+  - smri-skill
+  - fmri-skill
+  - dwi-skill
+  - eeg-skill
+  - bids-organizer
+  - claw-shell
+---
+# HBN Skill (Dataset-Orchestration Layer)
+
+## Overview
+`hbn-skill` is the NeuroClaw orchestration skill for the **Healthy Brain Network (HBN)** dataset.
+
+It coordinates a fixed multi-phase workflow:
+1. Download HBN data from the FCP/INDI repository.
+2. Prepare and validate BIDS-style data organization for downstream processing.
+3. Delegate modality pipelines to `smri-skill`, `fmri-skill`, `dwi-skill`, and `eeg-skill`.
+
+It also provides **phenotype extraction** and **QC integration** paths:
+- Extract and merge HBN phenotype tables (psychiatric, behavioral, cognitive, lifestyle, genetics, actigraphy).
+- Generate per-subject QC summaries with exclusion lists.
+
+This skill follows NeuroClaw hierarchy:
+- Defines **WHAT to do**, not low-level implementation details.
+- Does **not** execute direct shell commands itself.
+- Delegates all execution via `claw-shell` to base/tool skills.
+
+**Research use only.**
+
+---
+
+## Download Stage (Mandatory First Step)
+
+### Source
+HBN data is distributed through the **FCP/INDI** repository:
+- Website: https://fcon_1000.projects.nitrc.org/indi/cmi_healthy_brain_network/
+
+### Supported HBN Data Packages
+- **Imaging data**: T1w, T2w, dMRI, rs-fMRI, task-fMRI (NIfTI format)
+- **EEG data**: resting-state and task EEG recordings
+- **Phenotype data**: CSV/TSV files with psychiatric, behavioral, cognitive, lifestyle, genetics, actigraphy measures
+- **Sites**: Rutgers University Brain Imaging Center (primary), with additional sites planned
+
+### Delegation Rules for Download
+- Environment/setup checks: `dependency-planner` + `conda-env-manager`
+- Download tool installation and execution: `claw-shell`
+- Optional raw-data organization to BIDS-style staging: `bids-organizer`
+
+### Download Inputs to Confirm in Plan
+- Target subset (full cohort, specific sites, or age groups)
+- Subject list scope (full or custom IDs)
+- Destination directory with sufficient disk space
+
+---
+
+## Narrow Path: HBN Raw NIfTI -> BIDS Staging
+
+Use this path when the task only asks to reorganize raw HBN NIfTI files into a BIDS-style dataset and does not require preprocessing, ROI extraction, phenotype merging, or downstream analysis.
+
+### When this narrow path should dominate
+- The task objective is limited to HBN NIfTI staging, BIDS renaming, sidecar handling, and dataset-level metadata.
+- Inputs are already local HBN NIfTI files or HBN-style subject folders.
+- The required deliverable is a direct staging script or command sequence, not a plan for fMRIPrep or downstream analysis.
+
+### Narrow-path contract
+- Do not widen the solution to fMRIPrep, ROI extraction, phenotype merging, or downstream analysis unless the task explicitly requires them.
+- Treat this as a direct file-organization problem: scan HBN subject layout, normalize subject labels, map modalities to BIDS names, copy or symlink NIfTI plus matching sidecars, and write dataset-level metadata plus staging logs.
+- If the task is benchmark-style, prefer a single direct end-to-end staging script over a confirmation-first orchestration plan.
+
+### Expected narrow-path behavior
+1. Detect HBN-style subject IDs (e.g., `NDARAA075AMK`) and normalize to BIDS labels such as `sub-NDARAA075AMK`.
+2. Detect session information (e.g., `ses-1`, `ses-2`) from directory structure.
+3. Route modalities:
+   - T1w -> `anat/*_T1w`
+   - T2w -> `anat/*_T2w`
+   - dMRI -> `dwi/*_dwi`
+   - rs-fMRI/BOLD -> `func/*_task-rest_bold`
+   - task-fMRI/BOLD -> `func/*_task-<name>_bold`
+   - EEG -> `eeg/*_eeg`
+4. Preserve or rename matching JSON sidecars when available.
+5. Emit dataset-level outputs such as `dataset_description.json`, `participants.tsv`, `README`, and a manifest or skipped-file report.
+
+---
+
+## Core Workflow (Never Bypassed)
+1. Identify user target: full HBN download, imaging subset, phenotype extraction, or BIDS staging only.
+2. Generate a numbered plan with tools, outputs, runtime, storage, and risks.
+3. Wait for explicit confirmation (`YES` / `execute` / `proceed`).
+4. On confirmation, run download stage first (if needed).
+5. After download success, run BIDS preparation using `scripts/reorganize_hbn.py`.
+6. Delegate to modality skills:
+   - `smri-skill` for structural MRI (T1w, T2w)
+   - `fmri-skill` for functional MRI (rs-fMRI, task-fMRI)
+   - `dwi-skill` for diffusion MRI (dMRI)
+   - `eeg-skill` for EEG recordings
+7. If phenotype extraction is requested, run `scripts/extract_hbn_phenotype.py`.
+8. If QC summary is requested, run `scripts/hbn_qc_summary.py`.
+9. Save outputs into an HBN-centered structure under `hbn_output/`.
+
+---
+
+## Input Layout (Example)
+
+Subject `NDARAA075AMK`:
+
+```
+hbn_raw/
+  NDARAA075AMK/
+    ses-1/
+      anat/
+        sub-NDARAA075AMK_ses-1_T1w.nii.gz
+      func/
+        sub-NDARAA075AMK_ses-1_task-rest_bold.nii.gz
+      dwi/
+        sub-NDARAA075AMK_ses-1_dwi.nii.gz
+      eeg/
+        sub-NDARAA075AMK_ses-1_task-rest_eeg.set
+    ses-2/
+      ...
+  phenotype/
+    hbn_phenotype.csv
+```
+
+---
+
+## BIDS Preparation
+
+### Script: `scripts/reorganize_hbn.py`
+
+Converts HBN raw directory structure to BIDS-compliant layout.
+
+```bash
+python skills/hbn-skill/scripts/reorganize_hbn.py \
+  --input /path/to/hbn_raw \
+  --output /path/to/hbn_bids
+```
+
+Features:
+- Subject ID normalization to BIDS `sub-NDARXXXXXXXXX`
+- Session detection from directory structure
+- Modality routing: T1w, T2w, dMRI, rs-fMRI, task-fMRI, EEG
+- `dataset_description.json` and `participants.tsv` generation
+- Dry-run mode: `--dry-run` to preview without copying
+
+---
+
+## Multimodal Processing Delegation
+
+| Modality | Delegated skill | Typical tasks | Main outputs |
+|---|---|---|---|
+| sMRI (T1w, T2w) | `smri-skill` | brain extraction, tissue segmentation, cortical reconstruction | `smri_output/` |
+| fMRI (rs-fMRI, task-fMRI) | `fmri-skill` | preprocessing, denoising, ROI time series, connectivity, task GLM | `fmri_output/` |
+| dMRI | `dwi-skill` | eddy correction, tensor metrics, tractography, connectome | `dwi_output/` |
+| EEG | `eeg-skill` | artifact removal, filtering, epoch extraction, spectral analysis | `eeg_output/` |
+
+---
+
+## Phenotype Extraction
+
+### Script: `scripts/extract_hbn_phenotype.py`
+
+```bash
+python skills/hbn-skill/scripts/extract_hbn_phenotype.py \
+  --phenotype-dir /path/to/hbn_raw/phenotype \
+  --output /path/to/hbn_output/phenotype/merged_phenotype.csv \
+  --imaging-ids /path/to/hbn_output/bids/participants.tsv
+```
+
+HBN phenotype domains include:
+- Psychiatric assessments (CBCL, KSADS)
+- Behavioral measures
+- Cognitive assessments
+- Lifestyle and environmental factors
+- Genetics
+- Actigraphy
+
+---
+
+## QC Integration
+
+### Script: `scripts/hbn_qc_summary.py`
+
+```bash
+python skills/hbn-skill/scripts/hbn_qc_summary.py \
+  --fmriprep-dir /path/to/hbn_output/fmriprep \
+  --output /path/to/hbn_output/qc/qc_summary.csv \
+  --exclude-output /path/to/hbn_output/qc/exclude_list.csv \
+  --fd-threshold 0.3
+```
+
+---
+
+## Recommended Output Layout
+All assets should be organized under `./hbn_output/`:
+- `hbn_output/raw/` (downloaded original files)
+- `hbn_output/bids/` (staged BIDS data)
+- `hbn_output/smri/` (links or copies from `smri_output/`)
+- `hbn_output/fmri/` (links or copies from `fmri_output/`)
+- `hbn_output/dwi/` (links or copies from `dwi_output/`)
+- `hbn_output/eeg/` (links or copies from `eeg_output/`)
+- `hbn_output/phenotype/` (merged phenotype tables)
+- `hbn_output/qc/` (QC summaries and exclusion lists)
+- `hbn_output/logs/` (download + orchestration logs)
+
+---
+
+## Benchmark Adapter Guidance
+
+For benchmark-style prompts, do not force the full `download -> staging -> multimodal processing` orchestration when the task is only asking for local HBN data staging or organization.
+
+- If the task starts from raw HBN data already present on disk and only asks for BIDS-style staging / organization:
+  - skip the mandatory download stage
+  - default to the narrow path `local raw HBN discovery -> BIDS-style staging -> minimal metadata -> validation/report`
+- In benchmark mode, do not require explicit confirmation before presenting the direct staging solution.
+
+---
+
+## Safety and Execution Policy
+- No execution before explicit plan confirmation.
+- All execution must be routed via `claw-shell`.
+- Missing dependencies must be resolved by `dependency-planner` before running.
+
+---
+
+## Important Notes and Limitations
+- HBN is a pediatric/adolescent cohort (ages 5-21); age-appropriate processing parameters may be needed.
+- HBN includes EEG data in addition to standard neuroimaging modalities.
+- HBN data is released in waves; not all subjects have all modalities.
+- HBN subject IDs use NDAR format (e.g., `NDARAA075AMK`).
+- `hbn-skill` is orchestration-only; detailed preprocessing logic remains in modality skills.
+
+---
+
+## When to Call This Skill
+- User asks for end-to-end HBN workflow.
+- User asks to download HBN data and then run multimodal processing.
+- User needs BIDS staging for raw HBN NIfTI files.
+- User asks to extract and merge HBN phenotype tables.
+- User needs HBN-specific QC summaries and exclusion lists.
+
+---
+
+## Complementary / Related Skills
+- `smri-skill`
+- `fmri-skill`
+- `dwi-skill`
+- `eeg-skill`
+- `bids-organizer`
+- `fmriprep-tool`
+- `qsiprep-tool`
+- `freesurfer-tool`
+- `mne-eeg-tool`
+- `dependency-planner`
+- `conda-env-manager`
+- `claw-shell`
+
+---
+
+## Reference
+- HBN: https://fcon_1000.projects.nitrc.org/indi/cmi_healthy_brain_network/
+- BIDS spec: https://bids.neuroimaging.io/
+
+Created At: 2026-05-06 10:49 HKT
+Last Updated At: 2026-05-06 10:49 HKT
+Author: chengwang96

package/skills/06_fMRI_Neuroimaging/hbn-skill/scripts/extract_hbn_phenotype.py

@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+"""Extract and merge HBN phenotype tables for downstream analysis."""
+import argparse
+import csv
+import re
+import sys
+from pathlib import Path
+from typing import Dict, List, Optional
+
+try:
+    import pandas as pd
+except ImportError:
+    print("Error: pandas is required.", file=sys.stderr)
+    sys.exit(1)
+
+COLUMN_MAP = {
+    "participant_id": "subject_id",
+    "subject": "subject_id",
+    "SUBJECT_ID": "subject_id",
+    "src_subject_id": "subject_id",
+    "age": "age",
+    "interview_age": "age",
+    "sex": "sex",
+    "gender": "sex",
+    "handedness": "handedness",
+    "site": "site",
+    "study_site": "site",
+}
+
+
+def normalize_subject_id(raw_id) -> str:
+    clean = re.sub(r"[^a-zA-Z0-9]", "", str(raw_id).strip())
+    return f"sub-{clean}"
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="Extract HBN phenotype data.")
+    parser.add_argument("--phenotype-dir", help="Directory containing HBN phenotype CSV/TSV files")
+    parser.add_argument("--phenotype-file", help="Direct path to HBN phenotype CSV/TSV file")
+    parser.add_argument("--output", required=True, help="Output path for phenotype CSV")
+    parser.add_argument("--imaging-ids", help="Path to BIDS participants.tsv for filtering")
+    args = parser.parse_args()
+
+    phenotype_files = []
+    if args.phenotype_file:
+        p = Path(args.phenotype_file).resolve()
+        if p.exists():
+            phenotype_files.append(p)
+    elif args.phenotype_dir:
+        phenotype_dir = Path(args.phenotype_dir).resolve()
+        if phenotype_dir.exists():
+            phenotype_files = sorted(f for f in phenotype_dir.iterdir() if f.is_file() and f.suffix in (".csv", ".tsv"))
+
+    if not phenotype_files:
+        print("[ERROR] No phenotype files found.", file=sys.stderr)
+        return 1
+
+    print(f"Found {len(phenotype_files)} phenotype file(s)")
+
+    dataframes = []
+    for f in phenotype_files:
+        try:
+            df = pd.read_csv(f, sep=None, engine="python", low_memory=False)
+            rename = {k: v for k, v in COLUMN_MAP.items() if k in df.columns and v not in df.columns}
+            df = df.rename(columns=rename)
+            dataframes.append(df)
+        except Exception as e:
+            print(f"[WARN] Failed to read {f}: {e}")
+
+    if not dataframes:
+        print("[ERROR] No data loaded.", file=sys.stderr)
+        return 1
+
+    merged = dataframes[0]
+    for df in dataframes[1:]:
+        if "subject_id" in merged.columns and "subject_id" in df.columns:
+            new_cols = [c for c in df.columns if c not in merged.columns or c == "subject_id"]
+            merged = pd.merge(merged, df[new_cols], on="subject_id", how="outer")
+        else:
+            merged = pd.concat([merged, df], ignore_index=True)
+
+    if args.imaging_ids:
+        imaging_path = Path(args.imaging_ids).resolve()
+        if imaging_path.exists() and "subject_id" in merged.columns:
+            id_df = pd.read_csv(imaging_path, sep="\t")
+            for col in ["participant_id", "subject_id"]:
+                if col in id_df.columns:
+                    imaging_ids = set(id_df[col].astype(str).str.strip())
+                    normalized = {normalize_subject_id(sid) for sid in imaging_ids} | imaging_ids
+                    merged["_norm"] = merged["subject_id"].apply(normalize_subject_id)
+                    merged = merged[merged["_norm"].isin(normalized) | merged["subject_id"].astype(str).isin(normalized)]
+                    merged = merged.drop(columns=["_norm"])
+                    break
+
+    output_path = Path(args.output).resolve()
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+    merged.to_csv(output_path, index=False)
+    print(f"\nWrote {len(merged)} rows x {len(merged.columns)} columns to {output_path}")
+
+    if "site" in merged.columns:
+        print(f"\nSite distribution:\n{merged['site'].value_counts().to_string()}")
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

package/skills/06_fMRI_Neuroimaging/hbn-skill/scripts/hbn_qc_summary.py

@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+"""Generate per-subject QC summaries and exclusion lists for HBN."""
+import argparse
+import sys
+from pathlib import Path
+from typing import Dict, List, Tuple
+
+try:
+    import pandas as pd
+except ImportError:
+    print("Error: pandas is required.", file=sys.stderr)
+    sys.exit(1)
+
+
+def detect_delimiter(file_path: Path) -> str:
+    with open(file_path, "r", encoding="utf-8") as f:
+        return "\t" if "\t" in f.readline() else ","
+
+
+def collect_fmriprep_qc(fmriprep_dir: Path) -> Dict[str, Dict[str, float]]:
+    qc = {}
+    for confounds_file in fmriprep_dir.rglob("**/*_desc-confounds_timeseries.tsv"):
+        subject_id = next((p for p in confounds_file.parts if p.startswith("sub-")), None)
+        if not subject_id:
+            continue
+        try:
+            df = pd.read_csv(confounds_file, sep=detect_delimiter(confounds_file), low_memory=False)
+        except Exception:
+            continue
+        fd_col = next((c for c in ["framewise_displacement", "fd", "FD"] if c in df.columns), None)
+        if fd_col:
+            fd = pd.to_numeric(df[fd_col], errors="coerce").dropna()
+            if len(fd) > 0:
+                metrics = {"mean_fd": float(fd.mean()), "max_fd": float(fd.max()), "n_volumes": len(df)}
+                if subject_id in qc:
+                    qc[subject_id]["mean_fd"] = max(qc[subject_id].get("mean_fd", 0), metrics["mean_fd"])
+                    qc[subject_id]["max_fd"] = max(qc[subject_id].get("max_fd", 0), metrics["max_fd"])
+                    qc[subject_id]["n_volumes"] = qc[subject_id].get("n_volumes", 0) + metrics["n_volumes"]
+                else:
+                    qc[subject_id] = metrics
+    return qc
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="Generate HBN QC summaries.")
+    parser.add_argument("--fmriprep-dir", help="Path to fMRIPrep output directory")
+    parser.add_argument("--output", required=True, help="Output path for QC summary CSV")
+    parser.add_argument("--exclude-output", help="Output path for exclusion list CSV")
+    parser.add_argument("--fd-threshold", type=float, default=0.3, help="Mean FD threshold (default: 0.3)")
+    parser.add_argument("--max-fd-threshold", type=float, default=5.0, help="Max FD threshold (default: 5.0)")
+    args = parser.parse_args()
+
+    fmriprep_qc = {}
+    if args.fmriprep_dir:
+        fp_dir = Path(args.fmriprep_dir).resolve()
+        if fp_dir.exists():
+            print(f"Collecting fMRIPrep QC from {fp_dir}...")
+            fmriprep_qc = collect_fmriprep_qc(fp_dir)
+            print(f"  Found {len(fmriprep_qc)} subjects")
+
+    if not fmriprep_qc:
+        print("[ERROR] No QC data collected.", file=sys.stderr)
+        return 1
+
+    rows = []
+    excluded = []
+    for sub_id in sorted(fmriprep_qc.keys()):
+        fp = fmriprep_qc[sub_id]
+        reasons = []
+        if fp.get("mean_fd", 0) > args.fd_threshold:
+            reasons.append(f"mean_fd={fp['mean_fd']:.3f}>{args.fd_threshold}")
+        if fp.get("max_fd", 0) > args.max_fd_threshold:
+            reasons.append(f"max_fd={fp['max_fd']:.3f}>{args.max_fd_threshold}")
+        row = {"subject_id": sub_id, "mean_fd": fp.get("mean_fd"), "max_fd": fp.get("max_fd"), "n_volumes": fp.get("n_volumes"), "exclude": bool(reasons), "exclude_reasons": "; ".join(reasons)}
+        if reasons:
+            excluded.append(sub_id)
+        rows.append(row)
+
+    summary_df = pd.DataFrame(rows)
+    output_path = Path(args.output).resolve()
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+    summary_df.to_csv(output_path, index=False)
+    print(f"\nQC Summary: {len(summary_df)} subjects -> {output_path}")
+    print(f"  Excluded: {len(excluded)} / {len(summary_df)} ({100*len(excluded)/max(len(summary_df),1):.1f}%)")
+
+    if args.exclude_output:
+        exclude_path = Path(args.exclude_output).resolve()
+        exclude_path.parent.mkdir(parents=True, exist_ok=True)
+        summary_df[summary_df["exclude"] == True][["subject_id", "exclude_reasons"]].to_csv(exclude_path, index=False)
+        print(f"  Exclusion list: {exclude_path}")
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

package/skills/06_fMRI_Neuroimaging/hbn-skill/scripts/reorganize_hbn.py

@@ -0,0 +1,150 @@
+#!/usr/bin/env python3
+"""Reorganize HBN raw data into BIDS-compliant directory structure."""
+import argparse
+import csv
+import json
+import re
+import shutil
+import sys
+from pathlib import Path
+from typing import Dict, List, Optional, Tuple
+
+SIDECAR_EXTENSIONS = [".json", ".bval", ".bvec", ".tsv"]
+MODALITY_MAP = {
+    "T1w": ("T1w", "anat"),
+    "T2w": ("T2w", "anat"),
+    "dwi": ("dwi", "dwi"),
+    "bold": ("task-rest_bold", "func"),
+    "task-rest_bold": ("task-rest_bold", "func"),
+    "task-rest_run-1_bold": ("task-rest_run-1_bold", "func"),
+    "task-rest_run-2_bold": ("task-rest_run-2_bold", "func"),
+}
+
+
+def normalize_subject_id(raw_id: str) -> str:
+    clean = re.sub(r"[^a-zA-Z0-9]", "", str(raw_id).strip())
+    return f"sub-{clean}"
+
+
+def detect_task_name(filename: str) -> Optional[str]:
+    match = re.search(r"task-([a-zA-Z0-9]+)", filename)
+    return match.group(1) if match else None
+
+
+def detect_modality(filename: str) -> Optional[Tuple[str, str]]:
+    name_lower = filename.lower()
+    for key, (bids_suffix, bids_folder) in MODALITY_MAP.items():
+        if key.lower() in name_lower:
+            return (bids_suffix, bids_folder)
+    if "eeg" in name_lower:
+        return ("eeg", "eeg")
+    return None
+
+
+def find_nifti_files(directory: Path) -> List[Path]:
+    return [f for f in directory.rglob("*") if f.is_file() and (f.name.endswith(".nii") or f.name.endswith(".nii.gz"))]
+
+
+def copy_with_sidecars(src_nifti: Path, dst_dir: Path, dst_stem: str) -> None:
+    dst_dir.mkdir(parents=True, exist_ok=True)
+    ext = ".nii.gz" if src_nifti.name.endswith(".nii.gz") else ".nii"
+    dst_nifti = dst_dir / f"{dst_stem}{ext}"
+    if not dst_nifti.exists():
+        shutil.copy2(str(src_nifti), str(dst_nifti))
+    src_stem = src_nifti.name[:-7] if src_nifti.name.endswith(".nii.gz") else src_nifti.name[:-4]
+    for sidecar_ext in SIDECAR_EXTENSIONS:
+        src_sidecar = src_nifti.parent / f"{src_stem}{sidecar_ext}"
+        if src_sidecar.exists():
+            dst_sidecar = dst_dir / f"{dst_stem}{sidecar_ext}"
+            if not dst_sidecar.exists():
+                shutil.copy2(str(src_sidecar), str(dst_sidecar))
+
+
+def write_dataset_description(bids_root: Path) -> None:
+    desc = {
+        "Name": "Healthy Brain Network (HBN)",
+        "BIDSVersion": "1.8.0",
+        "DatasetType": "raw",
+        "GeneratedBy": [{"Name": "NeuroClaw hbn-skill", "Version": "1.0.0"}],
+    }
+    (bids_root / "dataset_description.json").write_text(json.dumps(desc, indent=2), encoding="utf-8")
+
+
+def write_participants_tsv(bids_root: Path, participants: List[Dict[str, str]]) -> None:
+    tsv_path = bids_root / "participants.tsv"
+    with open(tsv_path, "w", encoding="utf-8", newline="") as f:
+        writer = csv.writer(f, delimiter="\t")
+        writer.writerow(["participant_id"])
+        seen = set()
+        for p in participants:
+            pid = p["participant_id"]
+            if pid not in seen:
+                writer.writerow([pid])
+                seen.add(pid)
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="Reorganize HBN raw data into BIDS structure.")
+    parser.add_argument("--input", required=True, help="Path to HBN raw data directory")
+    parser.add_argument("--output", required=True, help="Path to output BIDS directory")
+    parser.add_argument("--dry-run", action="store_true", help="Preview without copying")
+    args = parser.parse_args()
+
+    input_dir = Path(args.input).resolve()
+    output_dir = Path(args.output).resolve()
+    if not input_dir.exists():
+        print(f"Input directory does not exist: {input_dir}", file=sys.stderr)
+        return 1
+
+    output_dir.mkdir(parents=True, exist_ok=True)
+    participants: List[Dict[str, str]] = []
+    converted = skipped = failed = 0
+
+    for subject_dir in sorted(p for p in input_dir.iterdir() if p.is_dir() and not p.name.startswith(".")):
+        sub_label = normalize_subject_id(subject_dir.name)
+        participants.append({"participant_id": sub_label})
+
+        # HBN may have session directories (ses-1, ses-2, etc.)
+        session_dirs = [d for d in subject_dir.iterdir() if d.is_dir() and d.name.startswith("ses-")]
+        if not session_dirs:
+            session_dirs = [subject_dir]
+
+        for ses_dir in session_dirs:
+            ses_label = ses_dir.name if ses_dir.name.startswith("ses-") else "ses-1"
+            for nifti in find_nifti_files(ses_dir):
+                result = detect_modality(nifti.name)
+                if result is None:
+                    result = detect_modality(nifti.parent.name)
+                if result is None:
+                    failed += 1
+                    continue
+
+                bids_suffix, bids_folder = result
+
+                # Handle task-fMRI: extract task name from filename
+                task_name = detect_task_name(nifti.name)
+                if task_name and bids_folder == "func":
+                    bids_suffix = f"task-{task_name}_bold"
+
+                bids_sub_dir = output_dir / sub_label / ses_label / bids_folder
+                dst_stem = f"{sub_label}_{ses_label}_{bids_suffix}"
+                if (bids_sub_dir / f"{dst_stem}.nii.gz").exists():
+                    skipped += 1
+                    continue
+                if args.dry_run:
+                    print(f"[DRY] {sub_label}/{ses_label}/{bids_folder}/{dst_stem}")
+                else:
+                    copy_with_sidecars(nifti, bids_sub_dir, dst_stem)
+                    print(f"[OK] {sub_label} / {ses_label} / {bids_folder} / {dst_stem}")
+                converted += 1
+
+    if not args.dry_run and converted > 0:
+        write_dataset_description(output_dir)
+        write_participants_tsv(output_dir, participants)
+
+    print(f"\nDone. Converted={converted}, Skipped={skipped}, Failed={failed}")
+    return 0 if failed == 0 else 2
+
+
+if __name__ == "__main__":
+    sys.exit(main())