casa6

from __future__ import absolute_import

import numpy

import os

import time, datetime

from numpy import ma, array, logical_not, logical_and

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import table, singledishms

    from casatools import ms as mstool

    from casatasks import casalog

    from . import sdutil

​

    ms = mstool( )

    sdms = singledishms( )

    tb = table( )

else:

    from taskinit import gentools, casalog

    import sdutil

    ms, sdms, tb = gentools(['ms','sdms','tb'])

​

@sdutil.sdtask_decorator

def sdfit(infile=None, datacolumn=None, antenna=None, field=None, spw=None,

           timerange=None, scan=None, pol=None, intent=None,

           timebin=None, timespan=None,

           polaverage=None, 

           fitfunc=None, fitmode=None, nfit=None, thresh=None, avg_limit=None,

           minwidth=None, edge=None, outfile=None, overwrite=None):

​

    casalog.origin('sdfit')

​

    try:

        if os.path.exists(outfile):

            if overwrite:

                os.system('rm -rf %s' % outfile)

            else:

                raise ValueError(outfile + ' exists.')

        if (fitmode not in  ['list', 'auto']):

            raise ValueError("fitmode='%s' is not supported yet" % fitmode)

        if (spw == ''): spw = '*'

​

        selection = ms.msseltoindex(vis=infile, spw=spw, field=field, 

                                    baseline=antenna, time=timerange, 

                                    scan=scan)

​

        sdms.open(infile)

        sdms.set_selection(spw=sdutil.get_spwids(selection), field=field, 

                           antenna=antenna, timerange=timerange, 

                           scan=scan, polarization=pol, intent=intent)

​

        tempfile = 'temp_sdfit_'+str(datetime.datetime.fromtimestamp(time.time())).replace('-','').replace(' ','').replace(':','')

        if os.path.exists(tempfile):

            tempfile += str(datetime.datetime.fromtimestamp(time.time())).replace('-','').replace(' ','').replace(':','')

            if os.path.exists(tempfile):

                raise Exception('temporary file ' + tempfile + ' exists...')

        tempoutfile = tempfile + '_temp_output_ms'

        if os.path.exists(tempoutfile):

            tempoutfile += str(datetime.datetime.fromtimestamp(time.time())).replace('-','').replace(' ','').replace(':','')

            if os.path.exists(tempoutfile):

                raise Exception('temporary ms file ' + tempoutfile + ' exists...')

        if fitmode=='auto': nfit = [-1]

        num_fit_str = str(',').join(map(str, nfit))