atooms is a Python framework for classical simulations of interacting particles. It makes it easy to develop simulation and analysis tools using an expressive language, without sacrificing efficiency. To achieve this, atooms relies on backends written in C, CUDA or Fortran.
The goal of atooms is to provide a coherent interface to the basic objects of molecular dynamics or Monte Carlo simulations. The simulation data are stored in trajectory files, whose format usually depends on the code that produced the simulation.
In this simple example, we read a trajectory file in xyz format. Accessing the coordinates of the particles in a trajectory file goes like this:
from atooms.trajectory import Trajectory
with Trajectory('input.xyz') as trajectory:
for system in trajectory:
print('The position of particle 0 is', system.particle[0].position)
Note that trajectories support iteration and slicing, just like lists.
Here we pick the last frame of the trajectory, change the density of the system to unity and write this new configuration to a trajectory format suitable for the RUMD simulation package:
with Trajectory('input.xyz') as trajectory:
system = trajectory[-1]
system.density = 1.0
print('The new density is', len(system.particle) / system.cell.volume)
from atooms.trajectory import TrajectoryRUMD
with TrajectoryRUMD('rescaled.xyz.gz', 'w') as trajectory:
trajectory.write(system, step=0)
We can now run 1000 molecular dynamics steps using the Lennard-Jones potential:
from atooms.backends.rumd import RUMD
from atooms.simulation import Simulation
backend = RUMD('rescaled.xyz.gz', forcefield_file='lj_rumd.ff',
output_path='/tmp/outdir', integrator='nve')
sim = Simulation(backend)
sim.run(1000)
print('Final temperature and density', sim.system.temperature, sim.system.density)
The forcefield file lj_rumd.ff
(available in data/
) defines the interaction potential.
See the public API documentation for full details.
From the python package index
pip install atooms
From the code repository
git clone https://gitlab.info-ufr.univ-montp2.fr/atooms/atooms.git
cd atooms
make install
atooms has a generic simulation interface that abstracts out most of the common parts of particle-based simulations. The actual simulation code is wrapped by a simulation backend that exposes a minimal but consistent interface. This enables one to develop more complex simulation frameworks (e.g., parallel tempering) that are essentially decoupled from the underlying simulation code.
atooms provides a command line tool to convert between various trajectory formats. The following command will convert a trajectory file produced by RUMD into a simpler xyz format
$ trj.py convert -i rumd -o xyz trajectory.xyz.gz output.xyz
If you don't specify the output path, the trajectory is written to standard output. This is useful for quick inspection of complex trajectory formats or for piping into sed / awk.
trj.py
provides means to fine tune the format of the output file. Type trj.py --help
to get a list of options and supported trajectory formats.
It is easy to add new trajectory formats by subclassing existing trajectory classes. Just create a package called
atooms_plugins
and add your trajectory modules there. They will be automatically
available to all client codes that use atooms.
Suppose you wrote a custom trajectory class TrajectoryABC
in
atooms_plugins/test.py
(the last path is relative to the current
directory). You can now convert an existing xyz trajectory to your custom
format:
$ trj.py convert output.xyz output.abc
Remember to add an empty __init__.py
file at the root of atooms_plugins
.
Actually, the atooms_plugins
package can be put anywhere in your PYTHONPATH
.
Atooms is composable: it is easy to add new functionalities, and just those you actually need. Additional packages are available from the atooms main repository. These packages will be installed in the atooms namespace to prevent name clashing.
If you want to add your own package to the atooms namespace, structure it this way
atooms/your_package
atooms/your_package/__init__.py
where __init__.py
contains
from pkgutil import extend_path
__path__ = extend_path(__path__, __name__)
Add the package root folder to $PYTHONPATH. You can now import your package as
import atooms.your_package
Daniele Coslovich: http://www.coulomb.univ-montp2.fr/perso/daniele.coslovich/