casa5

from taskinit import casalog, metool, qa, smtool, tbtool

from simutil import simutil

import pylab as pl

import os

import shutil

import tempfile

import matplotlib

​

def plotcomp(compdict, showplot=True, wantdict=False, symb=',',

             include0amp=False, include0bl=False, blunit='', bl0flux=0.0):

​

    """

    Given a dict including

    {'clist': component list,

     'objname': objname,

     'epoch': epoch,

     'shape': component shape dict, including direction.

     'freqs (GHz)': pl.array of frequencies,

     'antennalist': An array configuration file as used by simdata,

     'savedfig': False or, if specified, the filename to save the plot to,

     'standard': setjy fluxstandard type},

​

    and symb: One of matplotlib's codes for plot symbols: .:,o^v<>s+xDd234hH|_

          default: ',':  The smallest points I could find,

​

    make a plot of visibility amplitude vs. baseline length for clist at epoch.

​

    If antennalist is not found as is, it will look for antennalist in

    os.getenv('CASAPATH').split(' ')[0] + '/data/alma/simmos/'.

    showplot: Whether or not to show the plot on screen.

​

    If wantdict is True, it returns a dictionary with the amplitudes and

    baselines on success.  Otherwise, it returns True or False as its estimated

    success value.

​

    include0amp: Force the lower limit of the amplitude axis to 0.

    include0bl: Force the lower limit of the baseline length axis to 0.

    blunit: unit of the baseline length (='' used the unit in the data or klambda)

    bl0flux: Zero baseline flux

    """

    def failval():

        """

        Returns an appropriate failure value.

        Note that mydict.update(plotcomp(wantdict=True, ...)) would give a

        confusing error message if plotcomp returned False.

        """

        retval = False

        if wantdict:

            retval = {}

        return retval

    retval = failval()  # Default

    try:

        clist = compdict['clist']

        objname = compdict['objname']

        epoch = compdict['epoch']

        epstr = mepoch_to_str(epoch)

        antennalist = compdict['antennalist']

​

        # Read the configuration info.

        if not antennalist:

            print "compdict['antennalist'] must be set!"