casa6

# sd task for imaging

from __future__ import absolute_import

from __future__ import print_function

​

import os

import re

import numpy

import shutil

import contextlib

import functools

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatasks import casalog

    from casatools import ms as mstool

    from casatools import quanta, imager, image, table

    from . import sdutil

    from . import sdbeamutil

    from .cleanhelper import cleanhelper

​

    ## (1) Import the python application layer

    from .imagerhelpers.imager_base import PySynthesisImager

    from .imagerhelpers.input_parameters import ImagerParameters

else:

    from taskinit import casalog

    from taskinit import qatool as quanta

    from taskinit import imtool as imager

    from taskinit import iatool as image

    from taskinit import mstool

    from taskinit import tbtool as table

    import sdutil

    import sdbeamutil

    from cleanhelper import cleanhelper

​

    ## (1) Import the python application layer

    from imagerhelpers.imager_base import PySynthesisImager

    from imagerhelpers.input_parameters import ImagerParameters

​

image_suffix = '.image'

residual_suffix = '.residual'

weight_suffix = '.weight'

associate_suffixes = ['.psf', '.sumwt', weight_suffix, residual_suffix]

​

@contextlib.contextmanager

def open_ia(imagename):

    ia = image()

    ia.open(imagename)

    try:

        yield ia

    finally:

        ia.close()

​

@contextlib.contextmanager

def open_ms(vis):

    ms = mstool()

    ms.open(vis)

    try:

        yield ms

    finally:

        ms.close()

​

@contextlib.contextmanager

def open_table(vis):

    tb = table()

    tb.open(vis)

    try:

        yield tb

    finally:

        tb.close()

​

class SelectionHandler(object):

    def __init__(self, sel):

        self.sel = sel

        if isinstance(self.sel, str):

            self.selector = self._select0

        elif len(self.sel) == 1:

            self.selector = self._select1

        else:

            self.selector = self._select2

​

    def __call__(self, i):

        return self.selector(i)

​

    def _select0(self, i):

        return self.sel

​

    def _select1(self, i):

        return self.sel[0]

​

    def _select2(self, i):

        return self.sel[i]