casa5

import os

from taskinit import *

from parallel.parallel_task_helper import ParallelTaskHelper

​

def initweights(vis=None,wtmode=None,tsystable=None,gainfield=None,interp=None,spwmap=None,dowtsp=None):

​

    casalog.origin('initweights')

​

    # Do the trivial parallelization

    if ParallelTaskHelper.isMPIEnabled() and ParallelTaskHelper.isParallelMS(vis):

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

        helper.go()

        # Write history to MS.

        try:

            param_names = initweights.func_code.co_varnames[:initweights.func_code.co_argcount]

            param_vals = [eval(p) for p in param_names]

            casalog.post('Updating the history in the output', 'DEBUG1')

            write_history(ms, vis, 'initweights', param_names,

                          param_vals, casalog)

        except Exception, instance:

            casalog.post("*** Error \'%s\' updating HISTORY" % (instance),

                         'WARN')

        return

​

​

    #Python script

    try:

        mycb=cbtool()

        myms=mstool()

​

        # only if vis exists...

        if ((type(vis)==str) & (os.path.exists(vis))):

            if wtmode.upper().find("TSYS") > -1:

                if not os.path.exists(tsystable):

                    raise Exception, 'Tsys calibration table %s not found' % tsystable

                if len(spwmap)==0:

                    spwmap=[-1]

                if interp=="":

                    interp="linear"

            # ... and we are asked to do something...

            # open without adding anything!

            mycb.open(vis,compress=False,addcorr=False,addmodel=False)

            mycb.initweights(wtmode=wtmode,dowtsp=dowtsp,tsystable=tsystable,gainfield=gainfield,interp=interp,spwmap=spwmap)

            mycb.close()

        else:

            raise Exception, 'Visibility data set not found - please verify the name'

        # Write history to MS.

        # When running in parallel, history will be written in the parallel section above

        # normal MSs should write the history here

        if ParallelTaskHelper.isMPIClient():

            try:

                param_names = initweights.func_code.co_varnames[:initweights.func_code.co_argcount]

                param_vals = [eval(p) for p in param_names]

                casalog.post('Updating the history in the output', 'DEBUG1')

                write_history(myms, vis, 'initweights', param_names,

                              param_vals, casalog)

            except Exception, instance:

                casalog.post("*** Error \'%s\' updating HISTORY" % (instance),

                             'WARN')

​

​