casa6

from __future__ import absolute_import

import sys

import os

import numpy

import numpy.random as random

import shutil

​

# get is_CASA6 and is_python3

from casatasks.private.casa_transition import *

if is_CASA6:

    from casatools import calibrater, table, ms

    from casatasks import casalog, applycal

    from . import sdutil

    from .mstools import write_history

​

    # Table tool

    tb = table()

    # Calibrator tool

    cb = calibrater()

    # MS tool

    myms = ms()

else:

    from taskinit import *

    from applycal import applycal

    from mstools import write_history

    import sdutil

​

    # Calibrator tool

    (cb,myms) = gentools(['cb','ms'])

​

def sdcal(infile=None, calmode='tsys', fraction='10%', noff=-1,

           width=0.5, elongated=False, applytable='',interp='', spwmap={},

           outfile='', overwrite=False, field='', spw='', scan='',intent=''): 

    """ Externally specify calibration solutions of various types

    """

    # Composite mode: compute calibration table and calibrate

    if ',' in calmode:

        handle_composite_mode(locals())

        return

    # Single mode: either calibrate or compute calibration table

    try:

        # Parameters check

        if calmode == 'tsys':

            if scan != '':

                raise UserWarning("Scan input must be ''(=all) in calmode='tsys'.")

            if spw != '':

                raise UserWarning("Spw input must be ''(=all) in calmode='tsys'.")

​

        if isinstance(infile,str) and os.path.exists(infile):

            # check if CORRECTED_DATA is necessary

            addcorr = calmode == 'apply'

            cb.setvi(old=True)

            cb.open(filename=infile,compress=False,addcorr=addcorr,addmodel=False)

            cb.selectvis(spw=spw, scan=scan, field=field)

        else:

            raise Exception('Infile data set not found - please verify the name')

​

        if not isinstance(calmode,str):

            raise Exception("Calmode must be a string")

        if calmode.lower() not in ['tsys', 'ps', 'otfraster', 'otf', 'apply']: 

            raise Exception("Calmode must be either 'ps' or 'otfraster' or  'otf' or 'tsys' or 'apply'.")

​

        if (not overwrite) and os.path.exists(outfile):

            raise RuntimeError("overwrite is False and output file exists: {}".format(outfile))

​

        # Calibration

        if calmode == 'apply': # Calibrate using existing tables

            # single calibration table

            if isinstance(applytable, str):

                _table_list = [applytable]

​

            # multiple calibration tables

            if isinstance(applytable, list) or isinstance(applytable, numpy.ndarray):

                _table_list = applytable

            # no calibration table

            if len(_table_list) == 0:

                raise Exception('Applytable name must be specified.')

            # check calibration table files

            for table in _table_list:

                # empty string

                if len(table) == 0:

                    raise Exception('Applytable name must be specified.')

                # unexisting table

                if not os.path.exists(table):

                    raise Exception("Table doesn't exist: {}".format(table))