casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import sys

import shutil

import unittest

import time

import numpy

import re

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import ctsys, quanta

    from casatools import image as iatool

    from casatasks import sdfixscan

​

    _ia = iatool( )

    qa = quanta( )

​

    # default is not used in casatasks

    def default(atask):

        pass

else:

    from __main__ import default

    from tasks import *

    from taskinit import iatool

    from sdfixscan import sdfixscan

    from taskinit import *

    _ia = iatool()

    # global qa tool is used here

​

#

# Unit test of sdfixscan task.

# 

​

### Utility

def drop_stokes_axis(imagename, outimagename):

    """

    create 3d image from 4d.

    The function intends to drop Stokes axis and assumes that

    the third axis is Stokes.

    """

    myia = iatool( )

    myia.open(imagename)

    subimage = myia.subimage(outfile=outimagename, dropdeg=True,

                             keepaxes=[0,1,3])

    subimage.close()

    myia.close()

def drop_deg_axes(imagename, outimagename):

    """

    create 2d image from 4d.

    The function intends to drop both Stokes and Spectral axis and assumes that

    the third axis is Spectral and fourth axis is Stokes.

    """

    myia = iatool( )

    myia.open(imagename)

    subimage = myia.subimage(outfile=outimagename, dropdeg=True,

                             keepaxes=[0,1])

    subimage.close()

    myia.close()