orca.tasks.imaging_tasks
========================

.. py:module:: orca.tasks.imaging_tasks

.. autoapi-nested-parse::

   Imaging Celery tasks for snapshot and batch imaging.

   Provides Celery tasks for imaging pipelines including:
   - Single measurement set imaging with calibration
   - Batch imaging with shared NVMe workspace
   - PNG preview generation from FITS images



Attributes
----------

.. autoapisummary::

   orca.tasks.imaging_tasks.logger
   orca.tasks.imaging_tasks.NVME_ROOT


Functions
---------

.. autoapisummary::

   orca.tasks.imaging_tasks.imaging_pipeline_task
   orca.tasks.imaging_tasks.imaging_shared_pipeline_task
   orca.tasks.imaging_tasks.run_image_pipeline


Module Contents
---------------

.. py:data:: logger

.. py:data:: NVME_ROOT
   :value: '/fast/pipeline'


.. py:function:: imaging_pipeline_task(ms_path: str, delay_table: str, bandpass_table: str, final_dir: str, extra_wsclean: List[str] = None) -> str

   Runs copy→applycal→WSClean→PNG→remove extra files→export→purge NVMe
   and returns the path to the saved PNG.


.. py:function:: imaging_shared_pipeline_task(self, ms_path: str, delay_table: str, bandpass_table: str, final_dir: str, workdir_root: str, keep_full_products: bool = False, extra_wsclean: Optional[List[str]] = None, bad_corrs: Optional[List[int]] = None, peel_sources: Optional[str] = None) -> str

   Shared-workspace version:
     • All MSs in a batch share *workdir_root*.
     • Only the first MS (or any with keep_full_products=True) keeps the full
       WSClean output; others retain just -dirty FITS + PNG.
     • Automatic retries wipe the MS sub-dir first to avoid half-baked data.


.. py:function:: run_image_pipeline(self, ms_path: str, workdir: str, dest: str) -> Dict[str, str]

   1. Images one Measurement Set.
   2. Moves the temporary workdir to Lustre on the same node.

   :param ms_path: Absolute path to the averaged *.ms file.
   :type ms_path: str
   :param workdir: Fast scratch / NVMe directory to use for intermediate files.
   :type workdir: str
   :param dest: Final directory on Lustre where the whole workdir is moved.
   :type dest: str

   :returns: {"fits": <final FITS file>, "dest": <final workdir location>}
   :rtype: dict