Python Basics for CASA
Python Basics for CASA
Within CASA, you use Python to interact with the system. This does not mean an extensive Python course is necessary - basic interaction with the system (assigning parameters, running tasks) is straightforward. At the same time, the full potential of Python is at the more experienced user’s disposal. Some further details about Python, IPython, and the interaction between Python and CASA can be found in Section "Python and CASA".
The following are some examples of helpful hints and tricks on making Python work for you in CASA.
Variables
Python variables are set using the = syntax. Python assigns the type dynamically as you set the value, and thus you can easily give it a nonsensical value, e.g.
vis = 1
The CASA parameter system will check types when you run a task or tool, or more helpfully when you set inputs using inp (see below). CASA will check and protect the assignments of the global parameters in its namespace.
Note that Python variable names are case-sensitive:
CASA <110>: foo = 'Bar'
CASA <111>: foo
Out[111]: 'Bar'
CASA <112>: Foo
Out[112]: 'bar'
so be careful.
Also note that misspelling a variable assignment will not be noticed (as long as it is a valid Python variable name) by the interface. For example, if you wish to set correlation=’RR’ but instead type correllation=’RR’ you will find correlation unset and a new correllation variable set. Command completion (see "Getting Help in CASA") should help you avoid this.
Lists and Ranges
Sometimes, you need to give a task a list of indices. If these are consecutive, you can use the Python range function to generate this list:
CASA <2>: print iflist
[4, 5, 6, 7]
CASA <3>: iflist=range(4)
CASA <4>: print iflist
[0, 1, 2, 3]
See "Python and CASA" for more information.
Indexes
As in C, Python indices are 0-based. For example, the first element in a list antlist would be antlist[0]:
CASA <114>: antlist
Out[114]: [0, 1, 2, 3, 4]
CASA <115>: antlist[0]
Out[115]: 0
CASA <116>: antlist[4]
Out[116]: 4
CASA also uses 0-based indexing internally for elements in the MeasurementSet (MS – the basic construct that contains visibility and/or single dish data). Thus, we will often talk about Field or Antenna “ID”s which will be indexed starting at 0. For example, the first field in an MS would have FIELD_ID==0 in the MSselect syntax, and can be indexed as field=’0’ in most tasks, as well as by name, e.g; field=’0137+331’. You will see these indices in the MS summary from the task listobs.
Indentation
Python pays attention to the indentation of lines, as it uses indentation to determine the level of nesting in loops.
See Section "Python and CASA" for more information.
System & shell access
If you want to access system commands from a script, use the os.system command ("Python and CASA").
In interactive mode, any input line beginning with a ! character is passed verbatim to the underlying operating system. Also, several common commands (ls, pwd, less) may be executed with or without the ’!’, although the cp command must use ’!’ and cd must be executed without the ’!’. For example:
Note that if you want to access a Unix environment variable, you will need to prefix with a double $$ instead of a single $ — for example, to print the value of the $PAGER variable, you would use
See "Python and CASA" for more information.
Executing Python scripts
You can execute Python scripts (ASCII text files containing Python or CASA commands) using the execfile command. For example, to execute the script contained in the file myscript.py (in the current directory), you would type
or
which will invoke the IPython auto-parenthesis feature.
NOTE: in some cases, you can use the IPython run command instead, e.g.
In this case, you do not need the quotes around the filename. This is most useful for re-initializing the task parameters, e.g.
(see "CASA Tasks").
See Section "Python and CASA" for more information.
A script can also be executed from the command line without starting CASA first. To do so, use the -c CASA startup option:
casa -c myscript.py