casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import glob

import sys

import shutil

import numpy

import unittest

from numpy import array

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import ctsys, table

    from casatasks import sdbaseline

    from casatasks.private.sdutil import tbmanager

​

    ### for selection_syntax import

    from casatestutils import selection_syntax

​

    tb = table( )

​

    ctsys_resolve = ctsys.resolve

​

    # default is not necessary in CASA6

    def default(atask):

        pass

else:

    from __main__ import default

    from tasks import *

    from taskinit import *

    from sdbaseline import sdbaseline

    from sdutil import tbmanager

    # the global tb tool is used here as is

​

    try:

        from casatestutils import selection_syntax

    except:

        import tests.selection_syntax as selection_syntax

​

    dataRoot = os.path.join(os.environ.get('CASAPATH').split()[0],'casatestdata/')

    def ctsys_resolve(apath):

        return os.path.join(dataRoot,apath)

​

​

### Utilities for reading blparam file

class FileReader(object):

    def __init__(self, filename):

        self.__filename = filename

        self.__data = None

        self.__nline = None

​

    def read(self):

        if self.__data is None:

            f = open(self.__filename, 'r')

            self.__data = f.readlines()

            f.close()

            self.__nline = len(self.__data)

        return

​

    def nline(self):

        self.read()

        return self.__nline

​

    def index(self, txt, start):

        return self.__data[start:].index(txt) + 1 + start

​

    def getline(self, idx):

        return self.__data[idx]

​

​

class BlparamFileParser(FileReader):

    def __init__(self, blfile):

        FileReader.__init__(self, blfile)

        self.__nrow = None

        self.__coeff = None

        self.__rms = None

        self.__ctxt = 'Baseline parameters\n'

        self.__rtxt = 'Results of baseline fit\n'

​

    def nrow(self):

        self.read()

        if self.__nrow is None:

            return self._nrow()

        else:

            return self.__nrow

​

    def coeff(self):

        self.read()

        if self.__coeff is None:

            self.parseCoeff()

        return self.__coeff