Research projects can often take months to years to complete, but the precise
version of software they use will often have a shorter lifetime than this. These
new versions of software will often include new features which might be of great
use, but they might also introduce changes which break your existing work and
introduce compatibility issues with other pieces of software. So whilst for
newer projects we might want to use the most up-to-date versions of software,
for existing projects we want to be able to freeze the versions of software that
are in use. This leads to us needing to install and manage multiple versions of
software across our various research projects.
With the upcoming release of 0.16.0 of MDAnalysis, alongside various
improvements, we are also introducing some changes which could break existing
code. In this post we will explain how conda environments
or Python virtual environments can be used to
manage this transition, allowing you to finish existing projects with version
0.15.0, while also enjoying the benefits provided in version 0.16.0.
Conda Environments
Conda is a general package manager for
scientific applications. It is mostly used for Python packages but the system
can be used with any program. The conda-forge
community also provides a large collection of scientific software for Python, R
, Perl and others.
In this guide we will concentrate only on creating and managing environments
with conda. For more general information on installing conda please refer
to the official documentation.
Software is made available through different conda channels, which each act as a
source for different software. When attempting to install packages into a conda
environment, these channels are searched. In this post we will be using the
conda-forge channel which you can add to your configuration like so:
conda config --add channels conda-forge
For each research project, it is advised that you create a new environment so that
the software used in each project does not interfere across different projects.
To create a new environment for your next project that uses MDAnalysis in version
0.15.0 run:
conda create -n myproject mdanalysis=0.15.0 -y
This has created a new software environment called myproject but has not affected
anything currently! To have access to all the software installed within it we
must first activate it
source activate myproject
To list your available environments
conda env list
A nice feature of using conda-environments is that they are easy to share with
colleagues or transferred to other computers. This allows all collaborators on a
project to use an identical set of software and makes your research projects
reproducible. To store the state of the environment we created in a file called
myproject-environment
conda list --explicit--md5-n myproject > myproject-environment
You can now copy this file to a colleague or onto another computer. The first 3
lines also contain instructions how this file can be used with conda.
# This file may be used to create an environment using:# $ conda create --name <env> --file <this file># platform: linux-64
Like conda, virtual environments managed with virtualenv allow you to use
different versions of python and python packages for your different project. On
the contrary to conda, however, virtualenv is not a general-purpose package
manager; it leverages what is available on your system, and let you install
python packages using pip.
To use virtual environments you have to install the virtualenv package first.
This can be done with either pip or the package manager of your system:
pip install virtualenv
# or on ubuntusudo apt install virtualenv
# or on fedorasudo dnf install python-virtualenv
Virtual environments can be created per project directory.
cd myproject
# if you want to use the default python interpretor
virtualenv myproject-env
# or if you want to explicitly use python 2.7
virtualenv -p /usr/bin/python2.7 myproject-env
# or if you want to use python 3.5
virtualenv -p /usr/bin/python3.5 myproject-env
# or maybe you want to use pypy
virtualenv -p /usr/bin/pypy myproject-env
This will create a new folder myproject-env. This folder contains the virtual
environment and all packages you have installed in it. To activate it in the
current terminal run:
source myproject-env/bin/activate
Now you can install packages via pip without affecting your global
environment. The packages you install when the environment is activated will be
available in terminal sessions that have the environment activated. You can
deactivate the virtual environment by running
deactivate
The virtualenvwrapper package
makes virtual environment easier to use. It provides some very useful features:
it organizes the virtual environment in a single directory, avoiding to have
them scattered throughout the file system;
it defines command to easy the creation, deletion, and copy of virtual
environments;
it defines a command to activate a virtual environment using its name;
all commands defined by virtualenvwrapper have tab-completion for virtual
environment names.
To use virtualenvwrapper you first need to install it outside of a virtual
environment:
pip install virtualenvwrapper
# or on ubuntusudo apt install virtualenvwrapper
# or on fedorasudo dnf install python-virtualenvwrapper
Then, you need to have it loaded in your terminal session. Add the following
lines in ~/.bashrc, they will be executed every time you open a new terminal
session:
# Decide where to store the virtual environmentsexport WORKON_HOME=~/Envs
# Make sure the directory existsmkdir-p${WORKON_HOME}# Load virtualenvwrappersource /usr/local/bin/virtualenvwrapper.sh
Open a new terminal or run source ~/.bashrc to update your session. You can
now create a virtual environment with
mkvirtualenv my-project
Like the virtualenv command we saw earlier, mkvirtualenv lets you choose
your python interpretor with the -p option. Regardless of your current
working directory, we created the virtual environment in ~/Envs/ and it is
now loaded in our terminal session.
You can load your virtual environments by running workon my-project, and
exit them by running deactivate.
Virtual environments, especially with virtualenvwrapper, can do much more.
The Hitchhikers Guide to Python has a good
tutorial that
gives a more in depth explanation of virtual environments. The
virtualenvwrapper documentation
is also a good resource to read.
With MDAnalysis 0.16.0 on the horizon, we wanted to showcase a major development.
In fall 2015, we (@richardjgowers and @dotsdl) set to work on redesigning the topology system from scratch.
This system determines how atom, residue, and segment information is internally represented and exposed to everything in the API (Universe, AtomGroup, etc.), and the old scheme had issues with data duplication, maintaining consistency between atom and residue attributes, and performance for large systems.
We hoped to resolve all of these issues with our new design.
The starting point of this work was (the now infamous) issue 363, which floated the idea of holding all atom, residue, and segment attributes in arrays instead of lists of Atom, Residue, and Segment objects.
This approach turned the way topology data such as atom names, resids, masses, etc. are stored in a Universe on its head, going from an array of structs (list of Atom objects with individual attributes) to a struct of arrays (an array for each attribute, one entry per Atom).
Now, over a year later, the finishing touches on this work are being prepared for release.
This post is meant to serve as a brief view to what has changed internally, what has changed externally, and what benefits this gives us looking forward to the future.
Invisible changes to make working with MDAnalysis faster
Most of the changes are (or should be) invisible to the user. But they made some of the most fundamental operations in MDAnalysis quite a bit faster. Although this section is mostly of interest to developers, it is useful for all users to know the operations that MDAnalysis can now do much faster than before (and why).
In the new system, each atom is a member of exactly one residue, and each residue is a member of exactly one segment.
The new Topology object keeps an array giving the residue membership of each atom, and likewise an array giving segment membership of each residue.
Getting the resname of the residue of a group of atoms, then, is achieved by taking the indices of these atoms to fancy-index the Atoms->Residues array, and then using the result of this to fancy-index the Resnames array.
For example, if the Topology has 5 atoms and 3 residues, with membership (Atoms->Residues) and Resnames arrays as below:
Atoms->Residues Resnames
index --------------- index --------
0 0 0 GLU
1 2 1 LYS
2 1 2 ALA
3 1
4 2
calling AtomGroup.resnames for an AtomGroup with atoms [2, 0, 1, 2] will yield (pseudocode):
This scheme only works if each atom is a member of one and only one residue, and likewise if residues are members of one and only one segment.
Furthermore, AtomGroups, ResidueGroups, and SegmentGroups are very thin, storing only the indices of their members as a numpy array.
This gives a number of advantages:
Performance. We get up to an 8x speedup over the old scheme when accessing attributes. Setting attributes can give up to a 40x speedup.
Memory. We don’t store, for example, a resname for each atom, but instead store attributes at the level they make sense for.
Consistency. Since attributes are stored in one place, we avoid cases where the topology is in an inconsistent state, e.g. two atoms in the same residue give a different resname.
No staleness. Because e.g. ResidueGroups are only an array of indices, not a list of Residue objects generated upon creation of the group, changes of resiude-level properties by another ResidueGroup are always reflected consistently by every other one. Data is not duplicated anywhere in this scheme, and is all contained in the Topology object.
Serialization. Topologies become serializable and changes to topologies can be easily saved and communicated around. This is an important step towards implementing parallel algorithms in MDAnalysis.
For further performance comparisons, check out this notebook.
External changes that may affect how you use MDAnalysis
Previously, every object except Atom subclassed from AtomGroup.
This meant that calling .positions of would give you the positions of the Atoms contained within that group.
Previous class structure:
Atom
AtomGroup -> Residue
-> ResidueGroup -> Segment
-> SegmentGroup
New class structure:
Group -> AtomGroup
-> ResidueGroup
-> SegmentGroup
Atom
Residue
Segment
Now each object only contains information pertaining to that particular object.
A Residue object only yields information about the residue; to get to the atoms, use Residue.atoms.
Similarly, to get the atoms from a Segment or a SegmentGroup use Segment.atoms or SegmentGroup.atoms.
As before, you can get all residues associated with a group with Group.residues (which returns a ResidueGroup) and all segments with Group.segments (a SegmentGroup).
Bottom line: you should now always be explicit about what you want.
Why this was changed
Previously everything inheriting from AtomGroup made it unclear at what level of topology a given method or attribute was working on.
For example, does ResidueGroup.charges give the charge of the residues or the atoms?
Also, it was unclear what size a given output would be (see issue 411).
How to work with the new system
To access atom-level information from anything that isn’t an AtomGroup, use the .atoms level accessor.
For example, changing all .positions calls on anything that isn’t an AtomGroup to .atoms.positions.
Going forward: what does this mean for MDAnalysis as a project?
A major benefit of the new topology system is that information about the topology of a Universe is now completely encapsulated in the Topology object.
This not only makes development and maintenance easier, but also opens the door to some exciting new possibilities as simulation systems grow larger.
A single Topology object can now be cleanly shared by multiple Universe instances, each with their own trajectory reader(s).
This could make common operations such as fitting a trajectory to a reference structure or doing parallel analysis of many trajectories more efficient for large systems.
The Topology object can also be serialized more easily.
This should enable parallelization on workers without shared memory (using libraries such as distributed) out-of-the-box.
Making these things work is an ongoing effort, but the MDAnalysis coredevs are working to take advantage of all these possibilities.
We look forward to the benefits this brings not only to the project, but also to all our users going forward.
We hope you like what we’ve done here.
MDAnalysis has been accepted as a sub-org of the NumFOCUS foundation,
for Google Summer of Code 2017. If you are interested in working with us
this summer as a student read the advice and links below and write to us on the
mailing list.
We are looking forward to all applications from interested students
(undergraduates and graduates).
The application window deadline is April 3, 2017 at 12:00 (MST). As part
of the application process you must familiarize yourself
with Google Summer of Code 2017. Apply as soon as possible.
Project Ideas
We have listed several possible projects for you to work on on our
wiki. Each project is rated with a difficulty and lists the possible mentors for
it.
Alternatively, if you have another idea about a project please write to us on the
developer list and we can discuss it there.
Information for Students
You must meet our own requirements if you want to be a student with MDAnalysis
this year (read all the docs behind these links!). You must also meet the
eligibility criteria.
As a start to get familiar with MDAnalysis and open source development you
should follow these steps:
Complete the Tutorial
We have a tutorial explaining the basics of MDAnalysis. You should go through
the tutorial at least once to understand how MDAnalysis is used.
Introduce yourself to us
Introduce yourself on the mailing list. Tell us what you plan to work
on during the summer or what you have already done with MDAnalysis
Close an issue of MDAnalysis
You must have at least one commit in the development branch of
MDAnalysis in order to be eligible, i.e.. you must demonstrate that
you have been seriously engaged with the MDAnalysis project.
We have a list of easy bugs to work on in our issue tracker on
GitHub. We also appreciate if you write more tests or update/improve
our documentation. To start developing for MDAnalysis have a look at
our guide for developers and write us on the
mailing list if you have more questions about setting up a
development environment.