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']: