casa6

import os

import re

import shutil

import time

import tempfile

import numpy as np

​

from casatools import calibrater, ms, table

from casatasks import casalog, virtualconcat

​

from .mstools import write_history

from .update_spw import *

from .parallel.parallel_data_helper import ParallelDataHelper

from .parallel.parallel_task_helper import ParallelTaskHelper

​

mytb = table( )

mycb = calibrater( )

myms = ms()

_ms = ms()       # task also referenced the old global ms object

​

def uvcontsub_old(vis, field, fitspw, excludechans, combine, solint, fitorder, spw, want_cont):

    if ParallelDataHelper.isMMSAndNotServer(vis):

        helper = ParallelTaskHelper('uvcontsub_old', locals())

        helper._consolidateOutput = False

        retVar = helper.go()

​

        # Gather the list of continuum subtraction-SubMSs 

        cont_subMS_list = []

        contsub_subMS_list = []

        for subMS in retVar:

            if retVar[subMS]:

                cont_subMS_list.append(subMS + ".cont")

                contsub_subMS_list.append(subMS + ".contsub")

        if len(cont_subMS_list) <= 0:

            raise ValueError("No continuum-subtracted sub-MSs for concatenation","SEVERE")

​

        # We have to sort the list because otherwise it  

        # depends on the time the engines dispatches their sub-MSs

        cont_subMS_list.sort()

        contsub_subMS_list.sort()

​

        # deal with the pointing table

        auxfile = "uvcontsub_aux2_"+str(time.time())

        pnrows = 0

        try:

            mytb.open(vis+'/POINTING')

            pnrows = mytb.nrows()

            mytb.close()

            if(pnrows>0): 

                shutil.copytree(os.path.realpath(vis+'/POINTING'), auxfile)

        except Exception as instance:

            raise RuntimeError("Error handling POINTING table %s: %s" %

                               (vis+'/POINTING',str(instance)))

​

        if want_cont:

            try:

                virtualconcat(concatvis=helper._arg['vis'] + ".cont",vis=cont_subMS_list,

                              copypointing=False)

            except Exception as instance:

                raise RuntimeError("Error concatenating continuum sub-MSs %s: %s" %

                                   (str(cont_subMS_list),str(instance)))

        try:

            virtualconcat(concatvis=helper._arg['vis'] + ".contsub",vis=contsub_subMS_list,

                          copypointing=False)

        except Exception as instance:

            raise RuntimeError("Error concatenating continuum-subtracted sub-MSs %s: %s" %

                               (str(contsub_subMS_list),str(instance)))

​

        # Remove continuum subtraction-SubMSs

        if want_cont:

            for subMS in cont_subMS_list:

                if (os.system("rm -rf " + subMS ) != 0):

                    casalog.post("Problem removing continuum sub-MS %s into working directory" % (subMS),'SEVERE')

        for subMS in contsub_subMS_list:

            if (os.system("rm -rf " + subMS ) != 0):

                casalog.post("Problem removing continuum-subtracted sub-MS %s into working directory" % (subMS),'SEVERE')

​

        if(pnrows>0): # put back the POINTING table

            casalog.post("Restoring the POINTING table ...",'NORMAL')

            try:

                if want_cont:

                    theptab = os.path.realpath(helper._arg['vis'] + ".cont/POINTING")

                    os.system('rm -rf '+theptab)

                    shutil.copytree(auxfile, theptab)

                theptab = os.path.realpath(helper._arg['vis'] + ".contsub/POINTING")

                os.system('rm -rf '+theptab)

                os.system('mv '+auxfile+' '+theptab)

            except Exception as instance:

                raise RuntimeError("Error restoring pointing table from %s: %s" %