pytao

pytao is a python binding for tao.

Setup

Build the latest bmad distribution. To enable position independent code in the static libraries, set

export ACC_ENABLE_FPIC=Y

before building the distribution.

Now, from the pytao base folder execute:

python setup.py install --bmad-dir=$DIST_BASE_DIR/production

If installing inside a virtualenv, the command develop is preferred over install.

Environment

Before running pytao, if you want to access the tao help system, set the TAO_DIR environment variable, e.g.:

export TAO_DIR=$DIST_BASE_DIR/tao

Usage

Basic commands:

from pytao.tao import Tao
tao = Tao('-lat my_lat.bmad')       # Init with standard command-line switches

tao.command('use var *')            # Issue a command, nothing returned

output = tao.capture('show top10')  # Issue a command, capturing stdout
print(output)                       # Get the stdout of command

result = tao.python('help')         # Issue a Tao python command.
print(result)

# parse output of the `Tao> python plot_graph beta.g` command as
# dictionary:
properties = tao.properties('plot_graph', 'beta.g')
print(properties['title'])

Some more specialized methods are available, e.g. curve_names or curve_data as shown here:

from pytao.tao import Tao
import matplotlib.pyplot as plt

tao = Tao('-lat my_lat.bmad')       # Init with standard command-line switches

for curve in tao.curve_names('beta'):
    x, y = tao.curve_data(curve).T
    plt.plot(x, y, label=curve)

plt.legend(loc='upper left')
plt.show()

For more information, it's best you browse the source code, as its changing quickly.