casa6

from math import *

from tasks import *

from parameter_dictionary import *

from task_help import *

import numpy as np

from casa_stack_manip import stack_frame_find

​

from parameter_check import *

​

##allow globals for taskby default

casaglobals=True

​

####################

def go(taskname=None):

    """ Execute taskname: """

    myf=stack_frame_find( )

    if taskname==None: taskname=myf['taskname']

    oldtaskname=taskname

    if(myf.has_key('taskname')):

        oldtaskname=myf['taskname']

    #myf['taskname']=taskname

    if type(taskname)!=str:

        taskname=taskname.__name__

        myf['taskname']=taskname

    try:

        parameter_checktype(['taskname'],[taskname],str)

    except TypeError, e:

        print "go -- TypeError: ",e

        return

    fulltaskname=taskname+'()'

    print 'Executing: ',fulltaskname

    exec(fulltaskname)

    myf['taskname']=oldtaskname

​

​

# from RR

def announce_async_task(taskname):

    """Use the logger to herald the beginning of an asynchronous task."""

    casalog.origin(taskname)

    casalog.post('')

    casalog.post('###############################################')

    casalog.post('###  Begin Task: ' + taskname + ' ' * (27 - len(taskname)) + '###')

    casalog.post("")

    casalog.post("Use: ")

    casalog.post("      tm.retrieve(return_value) # to retrieve the status")

    casalog.post("")

def write_task_obit(taskname):

    """Eulogize the task in the logger."""

    casalog.post('###  End Task: ' + taskname + ' ' * (29 - len(taskname)) + '###')

    casalog.post('###############################################')

    casalog.post('')

​

​

def inp(taskname=None, page=False):

    """

    Function to browse input parameters of a given task

    taskname: name of task of interest

    page: use paging if True, useful if list of parameters is longer than terminal height

    """

    try:

        ####paging contributed by user Ramiro Hernandez

        if(page):

            #########################

            class TemporaryRedirect(object):

                def __init__(self, stdout=None, stderr=None):

                    self._stdout = stdout or sys.stdout

                    self._stderr = stderr or sys.stderr

                def __enter__(self):

                    self.old_stdout, self.old_stderr = sys.stdout, sys.stderr

                    self.old_stdout.flush(); self.old_stderr.flush()

                    sys.stdout, sys.stderr = self._stdout, self._stderr

                def __exit__(self, exc_type, exc_value, traceback):

                    self._stdout.flush(); self._stderr.flush()

                    sys.stdout = self.old_stdout

                    sys.stderr = self.old_stderr

            #######################end class

            tempfile="__temp_input.casa"

            temporal = open(tempfile, 'w')    

​

            with TemporaryRedirect(stdout=temporal):

                inp(taskname, False)

            temporal.close()

            os.system('more '+tempfile)

            os.system('rm '+tempfile)

            return

        ####

        myf=stack_frame_find( )

        if((taskname==None) and (not myf.has_key('taskname'))):

            print 'No task name defined for inputs display'

            return