artistools
Artistools is collection of plotting, analysis, and file format conversion tools for the ARTIS radiative transfer code.
Installation
Requires Python >= 3.13. A free-threaded Python requires version 3.15 or later, e.g. Python 3.15t.
The artistools command can be invoked with uvx artistools (after installing uv), installed globally with uv tool install artistools, or installed into your environment with pip install artistools.
Development (editable installation)
For development, you will need a rust compiler and a clone of the repository:
git clone https://github.com/artis-mcrt/artistools.git
cd artistools
To make a uv virtual environment in the clone, with the development tools, run:
uv sync --all-extras
uv run -- prek install
uv run -- artistools then runs the command of the clone. To make the artistools command available outside the clone, also run uv tool install --editable .[extras].
Alternatively, to avoid uv and install into the system environment with pip:
pip install --group dev --editable .[extras]
prek install
To enable tab completion, run artistools completions. The command detects your shell and gives the
steps for it.
Citing artistools
If you use artistools for a paper or a presentation, please cite it. For details, see https://zenodo.org/badge/latestdoi/53433932.
Usage
Run "artistools" (or the short alias "at") at the command-line to get a full list of subcommands, and "artistools --version" to check the installed version. Some common commands are:
- artistools plotspectra (alias: at spec)
- artistools plotlightcurves (alias: at lc)
- artistools plotestimators (alias: at estimators)
- artistools plotnltepops
- artistools inputmodel describe
Use the -h option to get a list of command-line arguments for each subcommand. Set ARTISTOOLS_TRACEBACK=1 to get the full traceback of an error. Run a command in an ARTIS run folder, or give the path of the folder. A command that reads a model takes the folder with -modelpath, e.g. artistools timesteps -modelpath mymodel. Most of these commands also take the folder as the last positional argument, e.g. artistools plotspectra -t 300 mymodel.
A model on a different host
These commands accept a model path of the form host:path, where host is a name that ssh knows:
- plotestimators and its
--interactiveviewer; - plotlightcurves and its
--interactiveviewer; - plotspectra and its
--interactiveviewer; at.scan_estimatorsin a Python script.
artistools plotspectra -t 300 vae26:~/scratch/mymodel
The rule of rsync decides whether a path is remote: a colon before the first slash makes a remote path. A local folder with such a colon in its name needs "./" at the start, e.g. ./run:2.
artistools starts the server on that host through ssh, and it prints the command and the reason for it. A release of artistools starts uvx artistools@<version> server, with the versions of the local artistools and polars. Thus the host needs only uv. A clone whose package code differs from the release starts its git commit with uvx, if the commit is on a remote branch. The host then also needs git and Rust. For a commit that is only in the clone, the release starts, because the host cannot get the commit. The command sets POLARS_MAX_THREADS=16, because a login node can have hundreds of cores, and other users share them. The server reads the files and makes the data of each plot, e.g. it puts the packets into bins. Only the data to draw comes back, and the parquet caches stay beside the data on the host. A script sends the filter and the columns of each query to the host, and only the rows of the query come back.
An option that reads its files on the local host stops with an error. To start the server with a different command on the host, e.g. an artistools in a clone, set ARTISTOOLS_REMOTE_COMMAND:
export ARTISTOOLS_REMOTE_COMMAND='POLARS_MAX_THREADS=16 ~/artistools/.venv/bin/artistools server'
The environment of such a command must have the same version of polars as the client. The filter of a query goes to the host as a polars expression, and the format of an expression changes between two versions. If the versions differ, the client stops with an error that gives the uvx option.
Use from Python
artistools is mainly a set of commands, but a script or a notebook can call the same functions. import artistools as at gives a small set of names at the top level. Each package holds the other names, e.g. at.spectra and at.inputmodel.
from pathlib import Path
import artistools as at
import polars as pl
modelpath = Path("mymodel")
dfmodel, modelmeta = at.get_modeldata(modelpath)
times_days = at.get_timestep_times(modelpath, loc="mid")
dftemperature = at.scan_estimators(modelpath).filter(pl.col("timestep") == 40).select("modelgridindex", "Te").collect()
dfspectrum = at.spectra.get_spectra(modelpath, timestepmin=40, timestepmax=45)[-1].collect()
A function with the prefix scan_ returns a polars LazyFrame, or one LazyFrame for each direction bin, and reads nothing until .collect(). A function with the prefix read_ reads at once. A function with the prefix get_ can do either, and the tables below say which one returns a LazyFrame. The direction bin -1 is the average over all directions.
Names at the top level
| Name | Purpose |
|---|---|
at.get_modeldata |
Read model.txt. Return the cells as a LazyFrame, and the model parameters as a dictionary. |
at.add_derived_cols_to_modeldata |
Add columns that follow from the model file, e.g. the volume and the mass of each cell. |
at.get_inputparams |
Return the parameters of input.txt. |
at.get_model_name |
Return the name of the model that holds a path. |
at.get_nprocs |
Return the number of MPI processes of the run. |
at.get_timestep_times |
Return the time in days of each timestep (loc is "start", "mid", "end", or "delta"). |
at.get_timestep_of_timedays |
Return the timestep that holds a time in days. |
at.scan_estimators |
Read the estimators of a full run as a LazyFrame, with a row for each timestep and cell. |
at.get_deposition |
Return deposition.out as a LazyFrame. |
at.get_composition_data |
Return the elements and the ion stages of compositiondata.txt as a DataFrame. |
at.get_ionstring |
Return a text such as Fe II for an atomic number and an ion stage. |
at.get_ion_tuple |
Return (26, 2) for a text such as FeII, Fe II, or 26_2. |
at.get_elsymbol |
Return the element symbol of an atomic number. |
at.get_atomic_number |
Return the atomic number of an element symbol. |
at.get_z_a_nucname |
Return the atomic number and the mass number of a text such as Pb208. |
at.decode_roman_numeral |
Return the integer of a Roman numeral. |
at.get_path |
Return a known path by name, e.g. the package folder. |
at.firstexisting |
Return the first file of a list that exists, with a compressed copy as an alternative. |
at.zopen |
Open a file, or its .zst, .gz, or .xz copy. |
Main readers in the packages
| Name | Purpose |
|---|---|
at.estimators.read_estimators |
Read the estimators of a few cells into a dictionary. This is slow for many cells, thus prefer at.scan_estimators. |
at.spectra.get_spectra |
Return a dictionary with a LazyFrame of the spectrum of each direction bin, as an average over a range of timesteps. |
at.spectra.get_from_packets |
Return a dictionary with a LazyFrame of the spectrum of each direction bin from the packets files, for a range of arrival times. |
at.lightcurve.scan_lightcurve |
Read a light curve file. Return one LazyFrame for each direction bin. |
at.lightcurve.get_from_packets |
Return a dictionary with a LazyFrame of the luminosity against time of each direction bin, from the packets files. |
at.packets.get_packets |
Return the number of ranks and a LazyFrame of the packets of a run. |
at.nltepops.read_nltepops |
Read the NLTE populations of a timestep and of one or more cells. |
at.atomic.get_levels |
Return the energy levels of each ion, with the transitions as an option. |
at.inputmodel.save_modeldata |
Write model.txt.zst from a dataframe of the cells. |
at.plottools.set_mpl_style |
Apply the matplotlib style of artistools. |
The command of a package
A package that has a plot command gives it as plot, e.g. at.spectra.plot, at.lightcurve.plot, at.estimators.plot, at.nltepops.plot, at.packets.plot, at.nonthermal.plot, and at.gsinetwork.plot. This function is the command itself. Each keyword is one command-line argument, and argsraw gives more arguments as text, e.g. argsraw=["-t", "300"]:
at.spectra.plot(argsraw=[], specpath=[modelpath], timedays="300", outputfile="spectrum.pdf")
Example output
License
Distributed under the MIT license. See LICENSE for more information.
1 maintainer
Metadata
1 maintainer
Release files for artistools 2026.10.11
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| artistools-2026.10.11.tar.gz | 9.0 MB | Details |
Built distributions (wheels)
| File | Reset | |||
|---|---|---|---|---|
| artistools-2026.10.11-cp315-abi3.abi3t-manylinux_2_28_x86_64.whl | CPython 3.15 | abi3, abi3t | Linux glibc 2.28+ x86-64 | Details |
| artistools-2026.10.11-cp315-abi3.abi3t-manylinux_2_28_aarch64.whl | CPython 3.15 | abi3, abi3t | Linux glibc 2.28+ ARM64 | Details |
| artistools-2026.10.11-cp315-abi3.abi3t-macosx_14_0_arm64.whl | CPython 3.15 | abi3, abi3t | macOS 14.0+ ARM64 | Details |
| artistools-2026.10.11-cp313-abi3-manylinux_2_28_x86_64.whl | CPython 3.13 | abi3 | Linux glibc 2.28+ x86-64 | Details |
| artistools-2026.10.11-cp313-abi3-manylinux_2_28_aarch64.whl | CPython 3.13 | abi3 | Linux glibc 2.28+ ARM64 | Details |
| artistools-2026.10.11-cp313-abi3-macosx_14_0_arm64.whl | CPython 3.13 | abi3 | macOS 14.0+ ARM64 | Details |
Total release size: 100.5 MB
Release files / artistools-2026.10.11.tar.gz
| Download URL | artistools-2026.10.11.tar.gz |
|---|---|
| Size | 9.0 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
8684b769178c82c1616abba2b2d86ef9f4a643788640287cb4ddea3bef130570
|
|
BLAKE2b-256 checksum How to use checksums |
8f31e60defde665c9ccb0e1cb72ede28c5321b3d0b316655f9ea7eaa25ea6cdd
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency logRelease files / artistools-2026.10.11-cp315-abi3.abi3t-manylinux_2_28_x86_64.whl
| Download URL | artistools-2026.10.11-cp315-abi3.abi3t-manylinux_2_28_x86_64.whl |
|---|---|
| Size | 15.9 MB |
| Tags | CPython 3.15 Linux glibc 2.28+ x86-64 abi3 abi3t |
|
SHA-256 checksum How to use checksums |
66206391f4f54b0734c7fabfdda28f0b7a7746311eca06a2b51ca61fa4849988
|
|
BLAKE2b-256 checksum How to use checksums |
fc68748f1775b2dd74ef5ae75d47a3f6984502e76739e84421107d10df6e4bd3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency logRelease files / artistools-2026.10.11-cp315-abi3.abi3t-manylinux_2_28_aarch64.whl
| Download URL | artistools-2026.10.11-cp315-abi3.abi3t-manylinux_2_28_aarch64.whl |
|---|---|
| Size | 16.0 MB |
| Tags | CPython 3.15 Linux glibc 2.28+ ARM64 abi3 abi3t |
|
SHA-256 checksum How to use checksums |
97e4c2b9f93d0eb9bff544fb64c9b3cd04e2ce428040607e32c9536febb46c49
|
|
BLAKE2b-256 checksum How to use checksums |
f3bf5931a875309cb603c742a840e7313d10b0d92d9f396f094f93f77538623f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency logRelease files / artistools-2026.10.11-cp315-abi3.abi3t-macosx_14_0_arm64.whl
| Download URL | artistools-2026.10.11-cp315-abi3.abi3t-macosx_14_0_arm64.whl |
|---|---|
| Size | 13.8 MB |
| Tags | CPython 3.15 abi3 abi3t macOS 14.0+ ARM64 |
|
SHA-256 checksum How to use checksums |
87f4005c5731e7e58fcd4624658464e0cf20cf705a76457de75d0114e9dc4890
|
|
BLAKE2b-256 checksum How to use checksums |
fb21a263973c9c32eb2cf1f32d13c4e05e25bf26bf309a8e31d681d6a6046feb
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency logRelease files / artistools-2026.10.11-cp313-abi3-manylinux_2_28_x86_64.whl
| Download URL | artistools-2026.10.11-cp313-abi3-manylinux_2_28_x86_64.whl |
|---|---|
| Size | 15.9 MB |
| Tags | CPython 3.13 Linux glibc 2.28+ x86-64 abi3 |
|
SHA-256 checksum How to use checksums |
3dab04d95d60290034a241492bb0c3871050b539dbf69ad7f9957770356050a5
|
|
BLAKE2b-256 checksum How to use checksums |
930f798e0a0cc464db604677e7893167c50c0c1e0baf9fa6b7c0092e44c4bb63
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency logRelease files / artistools-2026.10.11-cp313-abi3-manylinux_2_28_aarch64.whl
| Download URL | artistools-2026.10.11-cp313-abi3-manylinux_2_28_aarch64.whl |
|---|---|
| Size | 16.0 MB |
| Tags | CPython 3.13 Linux glibc 2.28+ ARM64 abi3 |
|
SHA-256 checksum How to use checksums |
4aed242d37d650369f9da8bf7a1f121f337730c438d7982392fe4122e5fe9888
|
|
BLAKE2b-256 checksum How to use checksums |
afb5ef474fd8bfdb14db5ebe81c6bce8d7163470a4a080a6c008bbdf7ed089b6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency logRelease files / artistools-2026.10.11-cp313-abi3-macosx_14_0_arm64.whl
| Download URL | artistools-2026.10.11-cp313-abi3-macosx_14_0_arm64.whl |
|---|---|
| Size | 13.8 MB |
| Tags | CPython 3.13 abi3 macOS 14.0+ ARM64 |
|
SHA-256 checksum How to use checksums |
d8b51a5a4aa76d41fe1982690a57a2a0c8338be605024093a7bbc63fdd3ba467
|
|
BLAKE2b-256 checksum How to use checksums |
16d62badb5c93bad3816a08bbda47b973619c9df82ee519ae2597f0bce4b1030
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 11, 2026.
Transparency log
