disco.cli

This module contains the full implementation of the DISCO automated pipeline, including FITS file discovery, group management, and the five-phase processing function. It is invoked when disco-start is called without the gui argument.

disco.cli.discover_groups(base_dir)[source]

Traverse the directory tree rooted at base_dir and group discovered FITS files by their common source prefix.

For each directory containing FITS files, filenames are split on the pattern _?[Bb]and_?\d+ to extract a source prefix. Files sharing the same prefix are aggregated into a group. The group name is prefixed with the parent directory name to disambiguate sources at different directory levels.

Parameters:

base_dir (str) – Root directory to search.

Returns:

List of group dictionaries, each with keys: "name" (str), "files" (list of str), "output_dir" (str).

Return type:

list[dict]

disco.cli.run_pipeline(files_to_process, group_name, output_dir, args, cnn_model)[source]

Execute the five-phase DISCO pipeline for a single group of FITS files.

Parameters:

  • files_to_process (list) – Ordered list of FITS file paths.

  • group_name (str) – Human-readable group identifier (used in output filenames).

  • output_dir (str) – Directory where output files are written.

  • args (argparse.Namespace) – Parsed command-line arguments. Relevant attributes: rout, rmin, incl, pa, beam, homobeam, csv, debug.

  • cnn_model (DiscoNet | None) – Pre-loaded DiscoNet model, or None to use analytical-only geometry.

Writes the following files to output_dir:

  • RP_<group_name>.PNG — normalised radial profile plot

  • RP_<group_name>_global.csv (if --csv on)

  • RP_<group_name>_bands.csv (if --csv on)

  • RP_<group_name>_profile.csv (if --csv on)

  • debug_pipeline/<group_name>_debug_center_rout.png (if --debug on)

disco.cli.main()[source]

Parse command-line arguments and orchestrate the full pipeline execution across all matched groups. Loads the DiscoNet model once before iterating over groups. Groups are processed sequentially with a tqdm outer progress bar. Processing failures for individual groups are logged but do not abort the remaining groups.