casa6

# sd task for imaging

from __future__ import absolute_import

import os

import re

import numpy

import shutil

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import quanta, ms, image, table, msmetadata

    from casatasks import casalog

    from . import sdutil

    from . import sdbeamutil

    from .cleanhelper import cleanhelper

else:

    from taskinit import casalog

    from taskinit import msmdtool as msmetadata

    from taskinit import tbtool as table

    from taskinit import mstool as ms

    from taskinit import iatool as image

    from taskinit import qatool as quanta

    import sdutil

    import sdbeamutil

    from cleanhelper import cleanhelper

​

​

@sdutil.sdtask_decorator

def sdimaging(infiles, outfile, overwrite, field, spw, antenna, scan, intent,

              mode, nchan, start, width, veltype, outframe,

              gridfunction, convsupport, truncate, gwidth, jwidth,

              imsize, cell, phasecenter, projection, ephemsrcname,

              pointingcolumn, restfreq, stokes, minweight, brightnessunit, clipminmax):

    with sdutil.sdtask_manager(sdimaging_worker, locals()) as worker:

        worker.initialize()

        worker.execute()

        worker.finalize()

​

​

def is_string_type(val):

    """

    Returns True if the argument is string type.

    """

    return type(val) in [str, numpy.string_]

​

​

def smart_remove(path):

    if os.path.isdir(path):

        shutil.rmtree(path)

    else:

        os.remove(path)

​

​

class sdimaging_worker(sdutil.sdtask_template_imaging):

    def __init__(self, **kwargs):

        super(sdimaging_worker,self).__init__(**kwargs)

        self.imager_param = {}

        self.sorted_idx = []

        self.image_unit = ""

​

    def parameter_check(self):

        # outfile check

        sdutil.assert_outfile_canoverwrite_or_nonexistent(self.outfile,

                                                          'im',

                                                          self.overwrite)

        sdutil.assert_outfile_canoverwrite_or_nonexistent(self.outfile+'.weight',

                                                          'im',

                                                          self.overwrite)

        # fix spw

        if type(self.spw) == str:

            self.spw = self.__format_spw_string(self.spw)

        # check unit of start and width

        # fix default

        if self.mode == 'channel':

            if self.start == '': self.start = 0

            if self.width == '': self.width = 1

        else:

            if self.start == 0: self.start = ''

            if self.width == 1: self.width = ''

        # fix unit

        if self.mode == 'frequency':

            myunit = 'Hz'

        elif self.mode == 'velocity':

            myunit = 'km/s'

        else: # channel

            myunit = ''

​

        for name in ['start', 'width']:

            param = getattr(self, name)

            new_param = self.__format_quantum_unit(param, myunit)

            if new_param == None:

                raise ValueError("Invalid unit for %s in mode %s: %s" % \