Writing a CASA Task: Difference between revisions

From CASA Guides
Jump to navigationJump to search
Mkrauss (talk | contribs)
No edit summary
Mkrauss (talk | contribs)
No edit summary
Line 123: Line 123:
That's it -- a new task!
That's it -- a new task!


=== Using the CASA toolkit ===
== Using the CASA toolkit ==


Ideally, a task uses the CASA "toolkit" for functionality, rather than calling other tasks.  This can help optimize performance, as well as take advantage of options that might not exist within the framework of currently existing tasks.
At their most basic level, CASA tasks are Python wrappers for the deeper functionality available via the CASA "toolkit".  (This is not to say the wrappers are always simple -- see [Exploring the built-in CASA tasks] for a look at the Python code in the built-in tasks.)


A good place to learn about the available toolkit functions is in the [http://casa.nrao.edu/docs/CasaRef/CasaRef.html CASA Toolkit Reference Manual].  Note that the tools are subdivided into different modules, and in CASA, these are called using an abbreviation of the module along with the name of the function.  For example, to call a tool from the <tt>image</tt> module, type
Although it's possible for a user-created tasks to call other tasks, ideally, it calls the toolkit for functionality. This can help optimize performance, as well as take advantage of options that might not exist within the framework of currently existing tasks. 
 
A good place to learn about the available toolkit functions is in the [http://casa.nrao.edu/docs/CasaRef/CasaRef.html CASA Toolkit Reference Manual].  Note that the toolkit is subdivided into different "tools", and in CASA, these are called using an abbreviation of the tool along with the name of the function.  For example, to call a function from the <tt>image</tt> tool, type


<source lang="python">
<source lang="python">
Line 136: Line 138:
Typing "ia." then hitting <TAB> will display all the possibilities.   
Typing "ia." then hitting <TAB> will display all the possibilities.   


The most complete help files can be found in the [http://casa.nrao.edu/docs/CasaRef/CasaRef.html CASA Toolkit Reference Manual] at the moment; we're working to get all this information into the inline CASA help as well.  '''Note that the functions are not listed in alphabetical order within a toolkit module.'''  This means you might need to search a bit to find what you're looking for.  Also, if you find documentation that's out of date or missing, please let us know via the [http://casa.nrao.edu/help_desk_all.shtml NRAO Helpdesk] so that we can fix it!
The most complete help files can be found in the [http://casa.nrao.edu/docs/CasaRef/CasaRef.html CASA Toolkit Reference Manual] at the moment; we're working to get all this information into the inline CASA help as well.  '''Note that the functions are not listed in alphabetical order within the tools.'''  This means you might need to search a bit to find what you're looking for.  Also, if you find documentation that's out of date or missing, please let us know via the [http://casa.nrao.edu/help_desk_all.shtml NRAO Helpdesk] so that we can fix it!


Here is a table of the tool names, and their abbreviations within CASA:


{| class="wikitable"
|-
! Toolkit tool
! CASA abbreviation
|-
| image
| ia
|-
| regionmanager
| rg
|-
| coordsys
| cs
|-
| imagepol
| po
|-
| componentlist
| cl
|-
| ms
| ms
|-
| msplot
|


== Simple example: mkmodelimage ==
== Simple example: mkmodelimage ==

Revision as of 23:43, 16 January 2012

Overview

In CASA, it is relatively simple to create your own tasks -- these can be as simple or as complex as you desire, and with the ease of scripting in Python, the possibilities are almost endless. You can also configure your CASA setup to automatically load these tasks on startup, so that they're always available to you.

Writing a task isn't all that much more work than writing a script, and it gives a nice interface through which to interact (along with the same functionality).

This tutorial will step you through the process of creating a task, and also point you to some examples of other tasks (some within CASA itself, and others written by non-developers). We hope you find this useful in creating your own!

The basics of writing and running a task

A task comprises an XML file, in which the parameter interface and help file are coded, and a Python file, in which the main body of code lives. They must follow a particular naming scheme, so that the program which creates all the task files for CASA to read (called buildmytasks) knows what it's looking for. So, if you want to write a task called newtask, you will need to have:

newtask.xml
task_newtask.py

The XML file

Getting the XML file right can sometimes be a bit tricky, since the parser is very particular. Here is a very simple example:

<?xml version="1.0" encoding="UTF-8"?>
<?xml-stylesheet type="text/xsl" ?>

<casaxml xmlns="http://casa.nrao.edu/schema/psetTypes.html"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://casa.nrao.edu/schema/casa.xsd
file:///opt/casa/code/xmlcasa/xml/casa.xsd">

<task type="function" name="newtask" category="editing">

  <shortdescription>Does nothing in particular.</shortdescription>
      
  <description>As I said, this task doesn't really do much.</description>

  <input>

    <param type="string" name="imagein" mustexist="true">
      <description>Input image name</description>
      <value></value>
    </param>

  <constraints>
  </constraints>

  </input>

  <returns type="void"/>

  <example>

   As I said, this task doesn't do much.

   Keyword arguments:

	imagein -- name of input image file

  </example>

</task>

</casaxml>

Note that all blocks must be closed by a matching "/" tag.

  • <shortdescription>: appears in the 'taskhelp' listing
  • <description>: I'm really not sure where this goes. If anyone can figure it out, let me know!
  • <input>: this is where the parameters which appear when you type "inp newtask" go
    • <param>: each param/description/value set defines a particular parameter.
      • In this case, we have defined that the parameter input needs to be of type "string" (other options include "int", for integer; "bool", for boolean True/False; "double" for double-precision floating point, or "any" for anything)
      • Setting mustexist="true" means that CASA will check that this file actually exists before proceeding (and exit with an error if it doesn't)
      • <description>: this is the information that follows the hash (#) in the parameter list
      • <value>: a default value to set for this parameter
  • <constraints>: this is where you would enter any constraints you want for the input parameters. It's not necessary to add any, but can be a nice way to control sub-parameters.
  • <example>: this is where the text for "help newtask" should be placed.

The Python file

This is where the "meat" of the task goes. In addition to the actual code, it also requires a few elements. Again, the simplest example:

from taskinit import *

def newtask(myinput = None):

    casalog.origin('newtask')
    casalog.post('Remember to '+myinput)

Building the task

Once you have the XML and Python files, run buildmytasks to compile them into a form that can be imported into CASA. This can be done on the command line (by simply typing "buildmytasks", or within a CASA session:

# In CASA
os.system('buildmytasks')

After buildmytasks has completed, you will see that there are a number of new files, e.g. "newtask.py-e", "newtask_cli.pyc", etc. Also, there is now a file called "mytasks.py", which is the meta-file for all the tasks you might have in this directory. This is what you import into CASA to include these tasks:

# In CASA
execfile('/<path_to_task_directory>/mytasks.py')

Note that if you rebuild a task after having already imported it into CASA, you will need to restart CASA and reimport for the changes to be incorporated.

Running the task

Now that the task has been imported into CASA, it can be run like any other task. For example,

# In CASA
inp newtask
help newtask
myinput = '"Stay hungry. Stay foolish." - Steve Jobs'
go

That's it -- a new task!

Using the CASA toolkit

At their most basic level, CASA tasks are Python wrappers for the deeper functionality available via the CASA "toolkit". (This is not to say the wrappers are always simple -- see [Exploring the built-in CASA tasks] for a look at the Python code in the built-in tasks.)

Although it's possible for a user-created tasks to call other tasks, ideally, it calls the toolkit for functionality. This can help optimize performance, as well as take advantage of options that might not exist within the framework of currently existing tasks.

A good place to learn about the available toolkit functions is in the CASA Toolkit Reference Manual. Note that the toolkit is subdivided into different "tools", and in CASA, these are called using an abbreviation of the tool along with the name of the function. For example, to call a function from the image tool, type

# In CASA
ia.<tool_function_name>

Typing "ia." then hitting <TAB> will display all the possibilities.

The most complete help files can be found in the CASA Toolkit Reference Manual at the moment; we're working to get all this information into the inline CASA help as well. Note that the functions are not listed in alphabetical order within the tools. This means you might need to search a bit to find what you're looking for. Also, if you find documentation that's out of date or missing, please let us know via the NRAO Helpdesk so that we can fix it!

Here is a table of the tool names, and their abbreviations within CASA:

Toolkit tool CASA abbreviation
image ia
regionmanager rg
coordsys cs
imagepol po
componentlist cl
ms ms
msplot

Simple example: mkmodelimage

Complex example: mkpipeline

Exploring the built-in CASA tasks

One great way to learn to tricks (and understand CASA more completely) is to look at the tasks within CASA. Finding the directories where the XML and Python code live can be a little tricky, since it varies for different operating systems. Try going to the directory where the CASA installation lives:

# In CASA
os.environ.get('CASAPATH').split()[0]

Then, in a terminal in this directory, type:

find . -name task_clean.py
find . -name clean.xml

This should point you to the directories where the relevant files live for all CASA tasks. Have fun exploring!

Automatically loading tasks on startup

So now you've written a task, and want to always have it available to you when you start CASA. There's a file that CASA will source every time on startup, which resides in your home directory in the hidden subdirectory .casa. Go to this directory, and create a file called init.py. In this file, put the line:

execfile('/<path_to_task_directory>/mytasks.py')

(In other words, the same command you used to load the task(s) into CASA before.) Note that this can sometimes cause problems, in case you update your version of CASA and it's incompatible with some tasks you have. However, it's easy enough to comment out this line before starting CASA, if you desire.