casa6

from taskinit import casalog, gentools, qa

import pylab as pl

import numpy

import os

import contextlib

​

import asap as sd

​

tb = gentools(['tb'])[0]

​

def asdatestring(mjd, digit, timeonly=False):

    datedict = qa.splitdate(qa.quantity(mjd, 'd'))

    if digit > 10 : digit = 10

    sstr_tmp = str(numpy.round(datedict['s'], digit)).split('.')

    sstr = sstr_tmp[0] + '.' + sstr_tmp[1][0:digit]

    if timeonly:

        return '%s:%s:%s'%(datedict['hour'],datedict['min'],sstr)

    else:

        return '%s/%s/%s/%s:%s:%s'%(datedict['year'],datedict['month'],datedict['monthday'],datedict['hour'],datedict['min'],sstr)

​

def astimerange(tmargin, mjd0, mjd1):

    digit = numpy.abs(int(numpy.floor(numpy.log10(tmargin))))

    if int(mjd0) == int(mjd1): # same date, different time

        timeonly=True

    else: # different date

        timeonly=False

    return '%s~%s'%(asdatestring(mjd0,digit), asdatestring(mjd1,digit,timeonly=timeonly))

​

@contextlib.contextmanager

def selection_manager(scantab, original_selection, **kwargs):

    sel = sd.selector(original_selection)

    for (k,v) in kwargs.items():

        method_name = 'set_%s'%(k)

        if hasattr(sel, method_name):

            getattr(sel, method_name)(v)

    scantab.set_selection(sel)

    yield scantab

    scantab.set_selection(original_selection)

class Raster(object):

    def __init__(self, scantab):

        self.scantab = scantab

        self.original_selection = self.scantab.get_selection()

        self.rows = None

        self.rasters = None

        self.mjd_range = None

        self.mjd_range_raster = None

        self.mjd_range_nomargin = None

        self.mjd_range_nomargin_raster = None

        self.spw = None

        self.pol = None

        self.margin = None

        self.gaplist = []

        self.gaplist_raster = []

        self.ngap = 0

        self.ngap_raster = 0

​

    @property

    def nrow(self):

        return self.ngap - 1

​

    @property

    def nraster(self):

        return self.ngap_raster - 1

​

    @property

    def nominal_spw(self):

        if hasattr(self, '_nominal_spw') and getattr(self, '_nominal_spw') is not None:

            return self._nominal_spw

        self._nominal_spw = None

        with selection_manager(self.scantab, self.original_selection, types=0) as s:

            spw_list = s.getifnos()

        print spw_list

        for spw in spw_list:

            with selection_manager(self.scantab, self.original_selection, ifs=spw) as s:

                if self.scantab.nrow() > 0:

                    nchan = len(self.scantab._getspectrum(0))

                    if nchan > 4: # ignore channel-averaged and WVR

                        self._nominal_spw = spw

                        break

        return self._nominal_spw

​

    @property

    def nominal_pol(self):

        if hasattr(self, '_nominal_pol') and getattr(self, '_nominal_pol') is not None:

            return self._nominal_pol

​

        self._nominal_pol = None

        nominal_spw = self.nominal_spw

        with selection_manager(self.scantab, self.original_selection, types=0, ifs=nominal_spw) as s: