complexity_metrics_analysis.py

import argparse

import pydicom

from analysis_batch import process_batch
from analysis_exports import DUAL_MLC_CSV_HEADER, build_dual_mlc_row
from analysis_logging import configure_logging
from DicomParse.utilities import retrieve_dcm_filenames
from ucomx_models import AnalysisMode


def build_parser():
    parser = argparse.ArgumentParser(description="Batch export complexity metrics for dual-MLC plans.")
    parser.add_argument("--input-dir", required=True, help="Directory containing RT Plan DICOM files.")
    parser.add_argument("--output-csv", required=True, help="Output CSV file path.")
    parser.add_argument(
        "--recursive",
        action="store_true",
        help="Recursively search for DICOM files under the input directory.",
    )
    parser.add_argument("--verbose", action="store_true", help="Enable debug logging.")
    return parser


def main():
    args = build_parser().parse_args()
    logger = configure_logging(args.verbose)
    filepaths = retrieve_dcm_filenames(args.input_dir, recursive=args.recursive)

    def precheck(logger, pfile):
        try:
            ds = pydicom.dcmread(pfile, force=True)
        except Exception as exc:
            logger.exception("Skipping file due to DICOM read failure %s: %s", pfile, exc)
            return False

        if 'Modality' not in ds or 'RTPlanName' not in ds:
            logger.warning("Skipping file missing Modality/RTPlanName: %s", pfile)
            return False
        if str(ds.Modality).upper() != 'RTPLAN' or 'reconstructed' in str(ds.RTPlanName).lower():
            logger.warning("Skipping non-target RT Plan file: %s", pfile)
            return False
        return True

    def metadata_filter(metadata):
        if metadata["beam_type"] not in ["STATIC", "DYNAMIC"]:
            return f"unsupported beam type {metadata['beam_type']}"
        return None

    process_batch(
        logger=logger,
        filepaths=filepaths,
        output_csv=args.output_csv,
        header=DUAL_MLC_CSV_HEADER,
        build_row=build_dual_mlc_row,
        metadata_filter=metadata_filter,
        precheck=precheck,
        requested_mode=AnalysisMode.VMAT_IMRT,
    )


if __name__ == '__main__':
    main()